etc/obs-StreamFX
0
0
Fork 0
mirror of https://github.com/Xaymar/obs-StreamFX synced 2024-11-15 08:15:06 +00:00
Code Issues Projects Releases Packages Wiki Activity
obs-StreamFX/source/nvidia/vfx/nvidia-vfx.hpp

92 lines
4.8 KiB
C++
Raw Normal View History

project: Fix License, License headers and Copyright information 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.
2023-02-28 01:15:26 +00:00
// AUTOGENERATED COPYRIGHT HEADER START
// Copyright (C) 2021-2023 Michael Fabian 'Xaymar' Dirks <info@xaymar.com>
// AUTOGENERATED COPYRIGHT HEADER END
nvidia/vfx: Add wrapper for NVIDIA Video Effects SDK
2021-05-01 16:31:13 +00:00
#pragma once
#include "nvidia/cv/nvidia-cv.hpp"
cmake, code: Abuse pre-processor to disable/enable warnings
2022-08-29 10:29:44 +00:00
#include "warning-disable.hpp"
#include <cinttypes>
#include "warning-enable.hpp"
nvidia/vfx: Add wrapper for NVIDIA Video Effects SDK
2021-05-01 16:31:13 +00:00
#define NVVFX_DEFINE_FUNCTION(name, ...) \
private: \
typedef ::streamfx::nvidia::cv::result(__cdecl* t##name)(__VA_ARGS__); \
\
public: \
t##name name = nullptr;
namespace streamfx::nvidia::vfx {
typedef const char* effect_t;
typedef const char* parameter_t;
typedef void* object_t;
typedef object_t handle_t;
static constexpr effect_t EFFECT_TRANSFER = "Transfer";
static constexpr effect_t EFFECT_GREEN_SCREEN = "GreenScreen";
static constexpr effect_t EFFECT_BACKGROUND_BLUR = "BackgroundBlur";
static constexpr effect_t EFFECT_ARTIFACT_REDUCTION = "ArtifactReduction";
static constexpr effect_t EFFECT_SUPERRESOLUTION = "SuperRes";
static constexpr effect_t EFFECT_UPSCALE = "Upscale";
static constexpr effect_t EFFECT_DENOISING = "Denoising";
static constexpr parameter_t PARAMETER_INPUT_IMAGE_0 = "SrcImage0";
static constexpr parameter_t PARAMETER_INPUT_IMAGE_1 = "SrcImage1";
static constexpr parameter_t PARAMETER_OUTPUT_IMAGE_0 = "DstImage0";
static constexpr parameter_t PARAMETER_MODEL_DIRECTORY = "ModelDir";
static constexpr parameter_t PARAMETER_CUDA_STREAM = "CudaStream";
static constexpr parameter_t PARAMETER_INFO = "Info";
static constexpr parameter_t PARAMETER_SCALE = "Scale";
static constexpr parameter_t PARAMETER_STRENGTH = "Strength";
static constexpr parameter_t PARAMETER_STRENGTH_LEVELS = "StrengthLevels";
static constexpr parameter_t PARAMETER_MODE = "Mode";
static constexpr parameter_t PARAMETER_TEMPORAL = "Temporal";
static constexpr parameter_t PARAMETER_GPU = "GPU";
static constexpr parameter_t PARAMETER_BATCH_SIZE = "BatchSize";
static constexpr parameter_t PARAMETER_MODEL_BATCH = "ModelBatch";
static constexpr parameter_t PARAMETER_STATE = "State";
static constexpr parameter_t PARAMETER_STATE_SIZE = "StateSize";
class vfx {
std::shared_ptr<::streamfx::util::library> _library;
nvidia/vfx: Prefer AddDllDirectory over SetDefaultDllDirectories
2021-10-24 12:18:32 +00:00
#ifdef WIN32
void* _extra;
#endif
std::filesystem::path _model_path;
nvidia/vfx: Add wrapper for NVIDIA Video Effects SDK
2021-05-01 16:31:13 +00:00
public:
~vfx();
vfx();
nvidia/vfx: Optimize by returning a const-reference Prevents a needless deep clone of the path object.
2021-10-26 20:59:02 +00:00
std::filesystem::path const& model_path();
nvidia/vfx: Add wrapper for NVIDIA Video Effects SDK
2021-05-01 16:31:13 +00:00
public:
NVVFX_DEFINE_FUNCTION(NvVFX_GetVersion, uint32_t* version);
NVVFX_DEFINE_FUNCTION(NvVFX_CreateEffect, effect_t effect, handle_t* handle);
NVVFX_DEFINE_FUNCTION(NvVFX_DestroyEffect, handle_t handle);
NVVFX_DEFINE_FUNCTION(NvVFX_SetU32, handle_t effect, parameter_t paramName, uint32_t val);
NVVFX_DEFINE_FUNCTION(NvVFX_SetS32, handle_t effect, parameter_t paramName, int32_t val);
NVVFX_DEFINE_FUNCTION(NvVFX_SetF32, handle_t effect, parameter_t paramName, float val);
NVVFX_DEFINE_FUNCTION(NvVFX_SetF64, handle_t effect, parameter_t paramName, double val);
NVVFX_DEFINE_FUNCTION(NvVFX_SetU64, handle_t effect, parameter_t paramName, uint64_t val);
NVVFX_DEFINE_FUNCTION(NvVFX_SetObject, handle_t effect, parameter_t paramName, void* ptr);
code: Disable useless alignment styling and update standard
2023-05-13 12:35:46 +00:00
NVVFX_DEFINE_FUNCTION(NvVFX_SetCudaStream, handle_t effect, parameter_t paramName, ::streamfx::nvidia::cuda::stream_t stream);
NVVFX_DEFINE_FUNCTION(NvVFX_SetImage, handle_t effect, parameter_t paramName, ::streamfx::nvidia::cv::image_t* im);
nvidia/vfx: Add wrapper for NVIDIA Video Effects SDK
2021-05-01 16:31:13 +00:00
NVVFX_DEFINE_FUNCTION(NvVFX_SetString, handle_t effect, parameter_t paramName, const char* str);
NVVFX_DEFINE_FUNCTION(NvVFX_GetU32, handle_t effect, parameter_t paramName, uint32_t* val);
NVVFX_DEFINE_FUNCTION(NvVFX_GetS32, handle_t effect, parameter_t paramName, int32_t* val);
NVVFX_DEFINE_FUNCTION(NvVFX_GetF32, handle_t effect, parameter_t paramName, float* val);
NVVFX_DEFINE_FUNCTION(NvVFX_GetF64, handle_t effect, parameter_t paramName, double* val);
NVVFX_DEFINE_FUNCTION(NvVFX_GetU64, handle_t effect, parameter_t paramName, uint64_t* val);
NVVFX_DEFINE_FUNCTION(NvVFX_GetObject, handle_t effect, parameter_t paramName, void** ptr);
code: Disable useless alignment styling and update standard
2023-05-13 12:35:46 +00:00
NVVFX_DEFINE_FUNCTION(NvVFX_GetCudaStream, handle_t effect, parameter_t paramName, ::streamfx::nvidia::cuda::stream_t stream);
NVVFX_DEFINE_FUNCTION(NvVFX_GetImage, handle_t effect, parameter_t paramName, ::streamfx::nvidia::cv::image_t* im);
nvidia/vfx: Add wrapper for NVIDIA Video Effects SDK
2021-05-01 16:31:13 +00:00
NVVFX_DEFINE_FUNCTION(NvVFX_GetString, handle_t effect, parameter_t paramName, const char** str);
NVVFX_DEFINE_FUNCTION(NvVFX_Run, handle_t effect, int32_t async);
NVVFX_DEFINE_FUNCTION(NvVFX_Load, handle_t effect);
public:
static std::shared_ptr<::streamfx::nvidia::vfx::vfx> get();
};
} // namespace streamfx::nvidia::vfx
Reference in a new issue Copy permalink