etc
/
obs-StreamFX
mirror of https://github.com/Xaymar/obs-StreamFX
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

nvidia/ar: Adjust to match coding guidelines

This commit is contained in:
Michael Fabian 'Xaymar' Dirks 2021-06-16 12:15:50 +02:00
parent cd8abac142
commit 7f5916dd72
4 changed files with 68 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
source/nvidia/ar/nvidia-ar-feature.cpp
View File

@ -19,7 +19,7 @@
#include "nvidia-ar-feature.hpp"
nvidia::ar::feature::feature(std::shared_ptr<::nvidia::ar::ar> ar, NvAR_FeatureID feature) : _ar(ar)
streamfx::nvidia::ar::feature::feature(std::shared_ptr<::streamfx::nvidia::ar::ar> ar, NvAR_FeatureID feature) : _ar(ar)
{
NvAR_FeatureHandle feat;
if (NvCV_Status res = _ar->create(feature, &feat); res != NVCV_SUCCESS) {
@ -29,7 +29,7 @@ nvidia::ar::feature::feature(std::shared_ptr<::nvidia::ar::ar> ar, NvAR_FeatureI
_feature = std::shared_ptr<nvAR_Feature>{feat, [this](NvAR_FeatureHandle v) { _ar->destroy(v); }};
}
nvidia::ar::feature::~feature()
streamfx::nvidia::ar::feature::~feature()
{
_feature.reset();
}

12
source/nvidia/ar/nvidia-ar-feature.hpp
View File

@ -22,13 +22,13 @@
#include "nvidia-ar.hpp"
#include "nvidia/cuda/nvidia-cuda-stream.hpp"
namespace nvidia::ar {
namespace streamfx::nvidia::ar {
class feature {
std::shared_ptr<::nvidia::ar::ar> _ar;
std::shared_ptr<::streamfx::nvidia::ar::ar> _ar;
std::shared_ptr<nvAR_Feature> _feature;
public:
feature(std::shared_ptr<::nvidia::ar::ar> ar, NvAR_FeatureID feature);
feature(std::shared_ptr<::streamfx::nvidia::ar::ar> ar, NvAR_FeatureID feature);
~feature();
public:
@ -144,13 +144,13 @@ namespace nvidia::ar {
}
template<>
inline NvCV_Status set(std::string name, std::shared_ptr<::nvidia::cuda::stream> value)
inline NvCV_Status set(std::string name, std::shared_ptr<::streamfx::nvidia::cuda::stream> value)
{
return _ar->set_cuda_stream(_feature.get(), name.c_str(), reinterpret_cast<CUstream>(value->get()));
}
template<>
inline NvCV_Status get(std::string name, std::shared_ptr<::nvidia::cuda::stream>& value)
inline NvCV_Status get(std::string name, std::shared_ptr<::streamfx::nvidia::cuda::stream>& value)
{}
};
} // namespace nvidia::ar
} // namespace streamfx::nvidia::ar

99
source/nvidia/ar/nvidia-ar.cpp
View File

@ -34,15 +34,15 @@
#undef nvGetProcAddress
#undef nvFreeLibrary
nvidia::ar::ar::ar()
streamfx::nvidia::ar::ar::ar()
{
if (!getNvARLib())
throw std::runtime_error("Failed to load NVIDIA AR SDK runtime.");
}
nvidia::ar::ar::~ar() {}
streamfx::nvidia::ar::ar::~ar() {}
std::filesystem::path nvidia::ar::ar::get_ar_sdk_path()
std::filesystem::path streamfx::nvidia::ar::ar::get_ar_sdk_path()
{
char* arsdk_path = getenv("NV_AR_SDK_PATH");
if (arsdk_path) {
@ -65,202 +65,205 @@ std::filesystem::path nvidia::ar::ar::get_ar_sdk_path()
}
}
NvCV_Status nvidia::ar::ar::image_init(NvCVImage* im, unsigned width, unsigned height, int pitch, void* pixels,
NvCVImage_PixelFormat format, NvCVImage_ComponentType type, unsigned isPlanar,
unsigned onGPU)
NvCV_Status streamfx::nvidia::ar::ar::image_init(NvCVImage* im, unsigned width, unsigned height, int pitch,
void* pixels, NvCVImage_PixelFormat format,
NvCVImage_ComponentType type, unsigned isPlanar, unsigned onGPU)
{
return NvCVImage_Init(im, width, height, pitch, pixels, format, type, isPlanar, onGPU);
}
void nvidia::ar::ar::image_init_view(NvCVImage* subImg, NvCVImage* fullImg, int x, int y, unsigned width,
void streamfx::nvidia::ar::ar::image_init_view(NvCVImage* subImg, NvCVImage* fullImg, int x, int y, unsigned width,
unsigned height)
{
NvCVImage_InitView(subImg, fullImg, x, y, width, height);
}
NvCV_Status nvidia::ar::ar::image_alloc(NvCVImage* im, unsigned width, unsigned height, NvCVImage_PixelFormat format,
NvCVImage_ComponentType type, unsigned isPlanar, unsigned onGPU,
unsigned alignment)
NvCV_Status streamfx::nvidia::ar::ar::image_alloc(NvCVImage* im, unsigned width, unsigned height,
NvCVImage_PixelFormat format, NvCVImage_ComponentType type,
unsigned isPlanar, unsigned onGPU, unsigned alignment)
{
return NvCVImage_Alloc(im, width, height, format, type, isPlanar, onGPU, alignment);
}
NvCV_Status nvidia::ar::ar::image_realloc(NvCVImage* im, unsigned width, unsigned height, NvCVImage_PixelFormat format,
NvCVImage_ComponentType type, unsigned isPlanar, unsigned onGPU,
unsigned alignment)
NvCV_Status streamfx::nvidia::ar::ar::image_realloc(NvCVImage* im, unsigned width, unsigned height,
NvCVImage_PixelFormat format, NvCVImage_ComponentType type,
unsigned isPlanar, unsigned onGPU, unsigned alignment)
{
return NvCVImage_Realloc(im, width, height, format, type, isPlanar, onGPU, alignment);
}
void nvidia::ar::ar::image_dealloc(NvCVImage* im)
void streamfx::nvidia::ar::ar::image_dealloc(NvCVImage* im)
{
NvCVImage_Dealloc(im);
}
NvCV_Status nvidia::ar::ar::image_create(unsigned width, unsigned height, NvCVImage_PixelFormat format,
NvCV_Status streamfx::nvidia::ar::ar::image_create(unsigned width, unsigned height, NvCVImage_PixelFormat format,
NvCVImage_ComponentType type, unsigned isPlanar, unsigned onGPU,
unsigned alignment, NvCVImage** out)
{
return NvCVImage_Create(width, height, format, type, isPlanar, onGPU, alignment, out);
}
void nvidia::ar::ar::image_destroy(NvCVImage* im)
void streamfx::nvidia::ar::ar::image_destroy(NvCVImage* im)
{
NvCVImage_Destroy(im);
}
void nvidia::ar::ar::image_component_offsets(NvCVImage_PixelFormat format, int* rOff, int* gOff, int* bOff, int* aOff,
int* yOff)
void streamfx::nvidia::ar::ar::image_component_offsets(NvCVImage_PixelFormat format, int* rOff, int* gOff, int* bOff,
int* aOff, int* yOff)
{
NvCVImage_ComponentOffsets(format, rOff, gOff, bOff, aOff, yOff);
}
NvCV_Status nvidia::ar::ar::image_transfer(const NvCVImage* src, NvCVImage* dst, float scale, CUstream_st* stream,
NvCVImage* tmp)
NvCV_Status streamfx::nvidia::ar::ar::image_transfer(const NvCVImage* src, NvCVImage* dst, float scale,
CUstream_st* stream, NvCVImage* tmp)
{
return NvCVImage_Transfer(src, dst, scale, stream, tmp);
}
NvCV_Status nvidia::ar::ar::image_composite(const NvCVImage* src, const NvCVImage* mat, NvCVImage* dst)
NvCV_Status streamfx::nvidia::ar::ar::image_composite(const NvCVImage* src, const NvCVImage* mat, NvCVImage* dst)
{
//return NvCVImage_Composite(src, mat, dst);
throw std::runtime_error("Not implemented.");
}
NvCV_Status nvidia::ar::ar::image_composite_over_constant(const NvCVImage* src, const NvCVImage* mat,
NvCV_Status streamfx::nvidia::ar::ar::image_composite_over_constant(const NvCVImage* src, const NvCVImage* mat,
const unsigned char bgColor[3], NvCVImage* dst)
{
return NvCVImage_CompositeOverConstant(src, mat, bgColor, dst);
}
NvCV_Status nvidia::ar::ar::image_flipy(const NvCVImage* src, NvCVImage* dst)
NvCV_Status streamfx::nvidia::ar::ar::image_flipy(const NvCVImage* src, NvCVImage* dst)
{
return NvCVImage_FlipY(src, dst);
}
NvCV_Status nvidia::ar::ar::create(NvAR_FeatureID featureID, NvAR_FeatureHandle* handle)
NvCV_Status streamfx::nvidia::ar::ar::create(NvAR_FeatureID featureID, NvAR_FeatureHandle* handle)
{
return NvAR_Create(featureID, handle);
}
NvCV_Status nvidia::ar::ar::destroy(NvAR_FeatureHandle handle)
NvCV_Status streamfx::nvidia::ar::ar::destroy(NvAR_FeatureHandle handle)
{
return NvAR_Destroy(handle);
}
NvCV_Status nvidia::ar::ar::set_uint32(NvAR_FeatureHandle handle, const char* name, unsigned int val)
NvCV_Status streamfx::nvidia::ar::ar::set_uint32(NvAR_FeatureHandle handle, const char* name, unsigned int val)
{
return NvAR_SetU32(handle, name, val);
}
NvCV_Status nvidia::ar::ar::set_int32(NvAR_FeatureHandle handle, const char* name, int val)
NvCV_Status streamfx::nvidia::ar::ar::set_int32(NvAR_FeatureHandle handle, const char* name, int val)
{
return NvAR_SetS32(handle, name, val);
}
NvCV_Status nvidia::ar::ar::set_float32(NvAR_FeatureHandle handle, const char* name, float val)
NvCV_Status streamfx::nvidia::ar::ar::set_float32(NvAR_FeatureHandle handle, const char* name, float val)
{
return NvAR_SetF32(handle, name, val);
}
NvCV_Status nvidia::ar::ar::set_float64(NvAR_FeatureHandle handle, const char* name, double val)
NvCV_Status streamfx::nvidia::ar::ar::set_float64(NvAR_FeatureHandle handle, const char* name, double val)
{
return NvAR_SetF64(handle, name, val);
}
NvCV_Status nvidia::ar::ar::set_uint64(NvAR_FeatureHandle handle, const char* name, unsigned long long val)
NvCV_Status streamfx::nvidia::ar::ar::set_uint64(NvAR_FeatureHandle handle, const char* name, unsigned long long val)
{
return NvAR_SetU64(handle, name, val);
}
NvCV_Status nvidia::ar::ar::set_object(NvAR_FeatureHandle handle, const char* name, void* ptr, unsigned long typeSize)
NvCV_Status streamfx::nvidia::ar::ar::set_object(NvAR_FeatureHandle handle, const char* name, void* ptr,
unsigned long typeSize)
{
return NvAR_SetObject(handle, name, ptr, typeSize);
}
NvCV_Status nvidia::ar::ar::set_string(NvAR_FeatureHandle handle, const char* name, const char* str)
NvCV_Status streamfx::nvidia::ar::ar::set_string(NvAR_FeatureHandle handle, const char* name, const char* str)
{
return NvAR_SetString(handle, name, str);
}
NvCV_Status nvidia::ar::ar::set_cuda_stream(NvAR_FeatureHandle handle, const char* name, CUstream stream)
NvCV_Status streamfx::nvidia::ar::ar::set_cuda_stream(NvAR_FeatureHandle handle, const char* name, CUstream stream)
{
return NvAR_SetCudaStream(handle, name, stream);
}
NvCV_Status nvidia::ar::ar::set_float32_array(NvAR_FeatureHandle handle, const char* name, float* val, int count)
NvCV_Status streamfx::nvidia::ar::ar::set_float32_array(NvAR_FeatureHandle handle, const char* name, float* val,
int count)
{
return NvAR_SetF32Array(handle, name, val, count);
}
NvCV_Status nvidia::ar::ar::get_uint32(NvAR_FeatureHandle handle, const char* name, unsigned int* val)
NvCV_Status streamfx::nvidia::ar::ar::get_uint32(NvAR_FeatureHandle handle, const char* name, unsigned int* val)
{
return NvAR_GetU32(handle, name, val);
}
NvCV_Status nvidia::ar::ar::get_int32(NvAR_FeatureHandle handle, const char* name, int* val)
NvCV_Status streamfx::nvidia::ar::ar::get_int32(NvAR_FeatureHandle handle, const char* name, int* val)
{
return NvAR_GetS32(handle, name, val);
}
NvCV_Status nvidia::ar::ar::get_float32(NvAR_FeatureHandle handle, const char* name, float* val)
NvCV_Status streamfx::nvidia::ar::ar::get_float32(NvAR_FeatureHandle handle, const char* name, float* val)
{
return NvAR_GetF32(handle, name, val);
}
NvCV_Status nvidia::ar::ar::get_float64(NvAR_FeatureHandle handle, const char* name, double* val)
NvCV_Status streamfx::nvidia::ar::ar::get_float64(NvAR_FeatureHandle handle, const char* name, double* val)
{
return NvAR_GetF64(handle, name, val);
}
NvCV_Status nvidia::ar::ar::get_uint64(NvAR_FeatureHandle handle, const char* name, unsigned long long* val)
NvCV_Status streamfx::nvidia::ar::ar::get_uint64(NvAR_FeatureHandle handle, const char* name, unsigned long long* val)
{
return NvAR_GetU64(handle, name, val);
}
NvCV_Status nvidia::ar::ar::get_object(NvAR_FeatureHandle handle, const char* name, const void** ptr,
NvCV_Status streamfx::nvidia::ar::ar::get_object(NvAR_FeatureHandle handle, const char* name, const void** ptr,
unsigned long typeSize)
{
return NvAR_GetObject(handle, name, ptr, typeSize);
}
NvCV_Status nvidia::ar::ar::get_string(NvAR_FeatureHandle handle, const char* name, const char** str)
NvCV_Status streamfx::nvidia::ar::ar::get_string(NvAR_FeatureHandle handle, const char* name, const char** str)
{
return NvAR_GetString(handle, name, str);
}
NvCV_Status nvidia::ar::ar::get_cuda_stream(NvAR_FeatureHandle handle, const char* name, const CUstream* stream)
NvCV_Status streamfx::nvidia::ar::ar::get_cuda_stream(NvAR_FeatureHandle handle, const char* name,
const CUstream* stream)
{
return NvAR_GetCudaStream(handle, name, stream);
}
NvCV_Status nvidia::ar::ar::get_float32_array(NvAR_FeatureHandle handle, const char* name, const float** vals,
NvCV_Status streamfx::nvidia::ar::ar::get_float32_array(NvAR_FeatureHandle handle, const char* name, const float** vals,
int* count)
{
return NvAR_GetF32Array(handle, name, vals, count);
}
NvCV_Status nvidia::ar::ar::run(NvAR_FeatureHandle handle)
NvCV_Status streamfx::nvidia::ar::ar::run(NvAR_FeatureHandle handle)
{
return NvAR_Run(handle);
}
NvCV_Status nvidia::ar::ar::load(NvAR_FeatureHandle handle)
NvCV_Status streamfx::nvidia::ar::ar::load(NvAR_FeatureHandle handle)
{
return NvAR_Load(handle);
}
NvCV_Status nvidia::ar::ar::cuda_stream_create(CUstream* stream)
NvCV_Status streamfx::nvidia::ar::ar::cuda_stream_create(CUstream* stream)
{
return NvAR_CudaStreamCreate(stream);
}
NvCV_Status nvidia::ar::ar::cuda_stream_destroy(CUstream stream)
NvCV_Status streamfx::nvidia::ar::ar::cuda_stream_destroy(CUstream stream)
{
return NvAR_CudaStreamDestroy(stream);
}
const char* nvidia::ar::ar::cv_get_error_string_from_code(NvCV_Status code)
const char* streamfx::nvidia::ar::ar::cv_get_error_string_from_code(NvCV_Status code)
{
return NvCV_GetErrorStringFromCode(code);
}

4
source/nvidia/ar/nvidia-ar.hpp
View File

@ -35,7 +35,7 @@
#pragma warning(pop)
#endif
namespace nvidia::ar {
namespace streamfx::nvidia::ar {
class ar {
public:
ar();
@ -91,4 +91,4 @@ namespace nvidia::ar {
NvCV_Status cuda_stream_destroy(CUstream stream);
const char* cv_get_error_string_from_code(NvCV_Status code);
};
} // namespace nvidia::ar
} // namespace streamfx::nvidia::ar