mirror of
https://github.com/Xaymar/obs-StreamFX
synced 2024-10-24 13:52:39 +00:00
5a3954ae0e
Fixes several files incorrectly stated a different license from the actual project, as well as the copyright headers included in all files. This change has no effect on the licensing terms, it should clear up a bit of confusion by contributors. Plus the files get a bit smaller, and we have less duplicated information across the entire project. Overall the project is GPLv2 if not built with Qt, and GPLv3 if it is built with Qt. There are no parts licensed under a different license, all have been adapted from other compatible licenses into GPLv2 or GPLv3.
107 lines
3.9 KiB
C++
107 lines
3.9 KiB
C++
// AUTOGENERATED COPYRIGHT HEADER START
|
|
// Copyright (C) 2021-2023 Michael Fabian 'Xaymar' Dirks <info@xaymar.com>
|
|
// AUTOGENERATED COPYRIGHT HEADER END
|
|
|
|
// NVIDIA CVImage is part of:
|
|
// - NVIDIA Video Effects SDK
|
|
// - NVIDIA Augmented Reality SDK
|
|
|
|
#include "nvidia-cv-texture.hpp"
|
|
#include "nvidia/cuda/nvidia-cuda-obs.hpp"
|
|
#include "obs/gs/gs-helper.hpp"
|
|
#include "util/util-logging.hpp"
|
|
|
|
#ifdef _DEBUG
|
|
#define ST_PREFIX "<%s> "
|
|
#define D_LOG_ERROR(x, ...) P_LOG_ERROR(ST_PREFIX##x, __FUNCTION_SIG__, __VA_ARGS__)
|
|
#define D_LOG_WARNING(x, ...) P_LOG_WARN(ST_PREFIX##x, __FUNCTION_SIG__, __VA_ARGS__)
|
|
#define D_LOG_INFO(x, ...) P_LOG_INFO(ST_PREFIX##x, __FUNCTION_SIG__, __VA_ARGS__)
|
|
#define D_LOG_DEBUG(x, ...) P_LOG_DEBUG(ST_PREFIX##x, __FUNCTION_SIG__, __VA_ARGS__)
|
|
#else
|
|
#define ST_PREFIX "<nvidia::cv::texture> "
|
|
#define D_LOG_ERROR(...) P_LOG_ERROR(ST_PREFIX __VA_ARGS__)
|
|
#define D_LOG_WARNING(...) P_LOG_WARN(ST_PREFIX __VA_ARGS__)
|
|
#define D_LOG_INFO(...) P_LOG_INFO(ST_PREFIX __VA_ARGS__)
|
|
#define D_LOG_DEBUG(...) P_LOG_DEBUG(ST_PREFIX __VA_ARGS__)
|
|
#endif
|
|
|
|
using ::streamfx::nvidia::cv::component_layout;
|
|
using ::streamfx::nvidia::cv::component_type;
|
|
using ::streamfx::nvidia::cv::pixel_format;
|
|
using ::streamfx::nvidia::cv::result;
|
|
using ::streamfx::nvidia::cv::texture;
|
|
|
|
texture::~texture()
|
|
{
|
|
auto gctx = ::streamfx::obs::gs::context();
|
|
auto cctx = ::streamfx::nvidia::cuda::obs::get()->get_context()->enter();
|
|
|
|
free();
|
|
_texture.reset();
|
|
}
|
|
|
|
texture::texture(uint32_t width, uint32_t height, gs_color_format pix_fmt)
|
|
{
|
|
auto gctx = ::streamfx::obs::gs::context();
|
|
auto cctx = ::streamfx::nvidia::cuda::obs::get()->get_context()->enter();
|
|
|
|
// Allocate a new Texture
|
|
_texture = std::make_shared<::streamfx::obs::gs::texture>(width, height, pix_fmt, 1, nullptr,
|
|
::streamfx::obs::gs::texture::flags::None);
|
|
alloc();
|
|
}
|
|
|
|
void texture::resize(uint32_t width, uint32_t height)
|
|
{
|
|
auto gctx = ::streamfx::obs::gs::context();
|
|
auto cctx = ::streamfx::nvidia::cuda::obs::get()->get_context()->enter();
|
|
|
|
D_LOG_DEBUG("Resizing object 0x%" PRIxPTR " to %" PRIu32 "x%" PRIu32 "...", this, width, height);
|
|
|
|
// Allocate a new Texture
|
|
free();
|
|
_texture = std::make_shared<::streamfx::obs::gs::texture>(width, height, _texture->get_color_format(), 1, nullptr,
|
|
::streamfx::obs::gs::texture::flags::None);
|
|
alloc();
|
|
}
|
|
|
|
std::shared_ptr<::streamfx::obs::gs::texture> texture::get_texture()
|
|
{
|
|
return _texture;
|
|
}
|
|
|
|
void streamfx::nvidia::cv::texture::alloc()
|
|
{
|
|
auto gctx = ::streamfx::obs::gs::context();
|
|
auto cctx = ::streamfx::nvidia::cuda::obs::get()->get_context()->enter();
|
|
auto nvobs = ::streamfx::nvidia::cuda::obs::get();
|
|
|
|
// Allocate any relevant CV buffers and Map it.
|
|
if (auto res = _cv->NvCVImage_InitFromD3D11Texture(
|
|
&_image, reinterpret_cast<ID3D11Texture2D*>(gs_texture_get_obj(_texture->get_object())));
|
|
res != result::SUCCESS) {
|
|
D_LOG_ERROR("Object 0x%" PRIxPTR " failed NvCVImage_InitFromD3D11Texture call with error: %s", this,
|
|
_cv->NvCV_GetErrorStringFromCode(res));
|
|
throw std::runtime_error("NvCVImage_InitFromD3D11Texture");
|
|
}
|
|
if (auto res = _cv->NvCVImage_MapResource(&_image, nvobs->get_stream()->get()); res != result::SUCCESS) {
|
|
D_LOG_ERROR("Object 0x%" PRIxPTR " failed NvCVImage_MapResource call with error: %s", this,
|
|
_cv->NvCV_GetErrorStringFromCode(res));
|
|
throw std::runtime_error("NvCVImage_MapResource");
|
|
}
|
|
}
|
|
|
|
void streamfx::nvidia::cv::texture::free()
|
|
{
|
|
auto gctx = ::streamfx::obs::gs::context();
|
|
auto cctx = ::streamfx::nvidia::cuda::obs::get()->get_context()->enter();
|
|
auto nvobs = ::streamfx::nvidia::cuda::obs::get();
|
|
|
|
// Unmap and deallocate any relevant CV buffers.
|
|
if (auto res = _cv->NvCVImage_UnmapResource(&_image, nvobs->get_stream()->get()); res != result::SUCCESS) {
|
|
D_LOG_ERROR("Object 0x%" PRIxPTR " failed NvCVImage_UnmapResource call with error: %s", this,
|
|
_cv->NvCV_GetErrorStringFromCode(res));
|
|
throw std::runtime_error("NvCVImage_UnmapResource");
|
|
}
|
|
_cv->NvCVImage_Dealloc(&_image);
|
|
}
|