nvidia/vfx/superresolution: Reduce overall CPU usage by caching results

This commit is contained in:
Michael Fabian 'Xaymar' Dirks 2021-10-06 04:11:26 +02:00
parent d1c8cda0a3
commit d0762e7150
2 changed files with 52 additions and 31 deletions

View file

@ -189,11 +189,20 @@ void streamfx::nvidia::vfx::superresolution::size(std::pair<uint32_t, uint32_t>
std::pair<uint32_t, uint32_t>& input_size, std::pair<uint32_t, uint32_t>& input_size,
std::pair<uint32_t, uint32_t>& output_size) std::pair<uint32_t, uint32_t>& output_size)
{ {
// Check if the size has actually changed at all.
if ((input_size.first == _cache_input_size.first) && (input_size.second == _cache_input_size.second)
&& (_scale == _cache_scale)) {
input_size = _cache_input_size;
output_size = _cache_output_size;
_scale = _cache_scale;
return;
}
// Define lower and upper boundaries for resolution.
constexpr uint32_t min_width = 160; constexpr uint32_t min_width = 160;
constexpr uint32_t min_height = 90; constexpr uint32_t min_height = 90;
uint32_t max_width = 0; uint32_t max_width = 0;
uint32_t max_height = 0; uint32_t max_height = 0;
if (_scale > 3.0) { if (_scale > 3.0) {
max_width = 960; max_width = 960;
max_height = 540; max_height = 540;
@ -239,6 +248,11 @@ void streamfx::nvidia::vfx::superresolution::size(std::pair<uint32_t, uint32_t>
this->size(size, input_size, output_size); this->size(size, input_size, output_size);
} }
} }
// Update last stored values.
_cache_input_size = input_size;
_cache_output_size = output_size;
_cache_scale = _scale;
} }
std::shared_ptr<::streamfx::obs::gs::texture> std::shared_ptr<::streamfx::obs::gs::texture>
@ -338,41 +352,43 @@ std::shared_ptr<::streamfx::obs::gs::texture>
void streamfx::nvidia::vfx::superresolution::resize(uint32_t width, uint32_t height) void streamfx::nvidia::vfx::superresolution::resize(uint32_t width, uint32_t height)
{ {
uint32_t out_width = static_cast<uint32_t>(std::lround(width * _scale));
uint32_t out_height = static_cast<uint32_t>(std::lround(height * _scale));
auto gctx = ::streamfx::obs::gs::context(); auto gctx = ::streamfx::obs::gs::context();
auto cctx = ::streamfx::nvidia::cuda::obs::get()->get_context()->enter(); auto cctx = ::streamfx::nvidia::cuda::obs::get()->get_context()->enter();
_cache_input_size = {width, height};
this->size(_cache_input_size, _cache_input_size, _cache_output_size);
if (!_tmp) { if (!_tmp) {
_tmp = std::make_shared<::streamfx::nvidia::cv::image>( _tmp = std::make_shared<::streamfx::nvidia::cv::image>(
out_width, out_height, ::streamfx::nvidia::cv::pixel_format::RGBA, _cache_output_size.first, _cache_output_size.second, ::streamfx::nvidia::cv::pixel_format::RGBA,
::streamfx::nvidia::cv::component_type::UINT8, ::streamfx::nvidia::cv::component_layout::PLANAR, ::streamfx::nvidia::cv::component_type::UINT8, ::streamfx::nvidia::cv::component_layout::PLANAR,
::streamfx::nvidia::cv::memory_location::GPU, 1); ::streamfx::nvidia::cv::memory_location::GPU, 1);
} }
// Input Size was changed. // Input Size was changed.
if (!_input || !_source || (width != _input->get_texture()->get_width()) if (!_input || !_source || (_cache_input_size.first != _input->get_texture()->get_width())
|| (height != _input->get_texture()->get_height())) { || (_cache_input_size.second != _input->get_texture()->get_height())) {
if (_input) { if (_input) {
_input->resize(width, height); _input->resize(_cache_input_size.first, _cache_input_size.second);
} else { } else {
_input = std::make_shared<::streamfx::nvidia::cv::texture>(width, height, GS_RGBA_UNORM); _input = std::make_shared<::streamfx::nvidia::cv::texture>(_cache_input_size.first,
} _cache_input_size.second, GS_RGBA_UNORM);
if (_source) {
_source->resize(width, height);
} else {
_source = std::make_shared<::streamfx::nvidia::cv::image>(
width, height, ::streamfx::nvidia::cv::pixel_format::BGR, ::streamfx::nvidia::cv::component_type::FP32,
::streamfx::nvidia::cv::component_layout::PLANAR, ::streamfx::nvidia::cv::memory_location::GPU, 1);
} }
if (_convert_to_fp32) { if (_convert_to_fp32) {
_convert_to_fp32->resize(out_width, out_height); _convert_to_fp32->resize(_cache_input_size.first, _cache_input_size.second);
} else { } else {
_convert_to_fp32 = std::make_shared<::streamfx::nvidia::cv::image>( _convert_to_fp32 = std::make_shared<::streamfx::nvidia::cv::image>(
out_width, out_height, ::streamfx::nvidia::cv::pixel_format::RGBA, _cache_input_size.first, _cache_input_size.second, ::streamfx::nvidia::cv::pixel_format::RGBA,
::streamfx::nvidia::cv::component_type::FP32, ::streamfx::nvidia::cv::component_layout::PLANAR,
::streamfx::nvidia::cv::memory_location::GPU, 1);
}
if (_source) {
_source->resize(_cache_input_size.first, _cache_input_size.second);
} else {
_source = std::make_shared<::streamfx::nvidia::cv::image>(
_cache_input_size.first, _cache_input_size.second, ::streamfx::nvidia::cv::pixel_format::BGR,
::streamfx::nvidia::cv::component_type::FP32, ::streamfx::nvidia::cv::component_layout::PLANAR, ::streamfx::nvidia::cv::component_type::FP32, ::streamfx::nvidia::cv::component_layout::PLANAR,
::streamfx::nvidia::cv::memory_location::GPU, 1); ::streamfx::nvidia::cv::memory_location::GPU, 1);
} }
@ -388,32 +404,33 @@ void streamfx::nvidia::vfx::superresolution::resize(uint32_t width, uint32_t hei
} }
// Input Size or Scale was changed. // Input Size or Scale was changed.
if (!_destination || !_output || (out_width != _output->get_texture()->get_width()) if (!_destination || !_output || (_cache_output_size.first != _output->get_texture()->get_width())
|| (out_height != _output->get_texture()->get_height())) { || (_cache_output_size.second != _output->get_texture()->get_height())) {
if (_destination) { if (_destination) {
_destination->resize(out_width, out_height); _destination->resize(_cache_output_size.first, _cache_output_size.second);
} else { } else {
_destination = std::make_shared<::streamfx::nvidia::cv::image>( _destination = std::make_shared<::streamfx::nvidia::cv::image>(
out_width, out_height, ::streamfx::nvidia::cv::pixel_format::BGR, _cache_output_size.first, _cache_output_size.second, ::streamfx::nvidia::cv::pixel_format::BGR,
::streamfx::nvidia::cv::component_type::FP32, ::streamfx::nvidia::cv::component_layout::PLANAR, ::streamfx::nvidia::cv::component_type::FP32, ::streamfx::nvidia::cv::component_layout::PLANAR,
::streamfx::nvidia::cv::memory_location::GPU, 1); ::streamfx::nvidia::cv::memory_location::GPU, 1);
} }
if (_output) {
_output->resize(out_width, out_height);
} else {
_output = std::make_shared<::streamfx::nvidia::cv::texture>(out_width, out_height, GS_RGBA_UNORM);
}
if (_convert_to_u8) { if (_convert_to_u8) {
_convert_to_u8->resize(out_width, out_height); _convert_to_u8->resize(_cache_output_size.first, _cache_output_size.second);
} else { } else {
_convert_to_u8 = std::make_shared<::streamfx::nvidia::cv::image>( _convert_to_u8 = std::make_shared<::streamfx::nvidia::cv::image>(
out_width, out_height, ::streamfx::nvidia::cv::pixel_format::RGBA, _cache_output_size.first, _cache_output_size.second, ::streamfx::nvidia::cv::pixel_format::RGBA,
::streamfx::nvidia::cv::component_type::UINT8, ::streamfx::nvidia::cv::component_layout::INTERLEAVED, ::streamfx::nvidia::cv::component_type::UINT8, ::streamfx::nvidia::cv::component_layout::INTERLEAVED,
::streamfx::nvidia::cv::memory_location::GPU, 1); ::streamfx::nvidia::cv::memory_location::GPU, 1);
} }
if (_output) {
_output->resize(_cache_output_size.first, _cache_output_size.second);
} else {
_output = std::make_shared<::streamfx::nvidia::cv::texture>(_cache_output_size.first,
_cache_output_size.second, GS_RGBA_UNORM);
}
if (auto res = _nvvfx->NvVFX_SetImage(_fx.get(), ::streamfx::nvidia::vfx::PARAMETER_OUTPUT_IMAGE_0, if (auto res = _nvvfx->NvVFX_SetImage(_fx.get(), ::streamfx::nvidia::vfx::PARAMETER_OUTPUT_IMAGE_0,
_destination->get_image()); _destination->get_image());
res != ::streamfx::nvidia::cv::result::SUCCESS) { res != ::streamfx::nvidia::cv::result::SUCCESS) {

View file

@ -47,6 +47,10 @@ namespace streamfx::nvidia::vfx {
bool _dirty; bool _dirty;
std::pair<uint32_t, uint32_t> _cache_input_size;
std::pair<uint32_t, uint32_t> _cache_output_size;
float _cache_scale;
public: public:
~superresolution(); ~superresolution();
superresolution(); superresolution();