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.
95 lines
4.8 KiB
C++
95 lines
4.8 KiB
C++
// AUTOGENERATED COPYRIGHT HEADER START
|
|
// Copyright (C) 2021-2023 Michael Fabian 'Xaymar' Dirks <info@xaymar.com>
|
|
// AUTOGENERATED COPYRIGHT HEADER END
|
|
|
|
#pragma once
|
|
#include "nvidia/cv/nvidia-cv.hpp"
|
|
|
|
#include "warning-disable.hpp"
|
|
#include <cinttypes>
|
|
#include "warning-enable.hpp"
|
|
|
|
#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;
|
|
#ifdef WIN32
|
|
void* _extra;
|
|
#endif
|
|
std::filesystem::path _model_path;
|
|
|
|
public:
|
|
~vfx();
|
|
vfx();
|
|
|
|
std::filesystem::path const& model_path();
|
|
|
|
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);
|
|
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);
|
|
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);
|
|
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);
|
|
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
|