obs-StreamFX/source/gfx/blur/gfx-blur-box-linear.cpp

371 lines
11 KiB
C++
Raw Normal View History

// AUTOGENERATED COPYRIGHT HEADER START
// Copyright (C) 2019-2023 Michael Fabian 'Xaymar' Dirks <info@xaymar.com>
// Copyright (C) 2022 lainon <GermanAizek@yandex.ru>
// AUTOGENERATED COPYRIGHT HEADER END
#include "gfx-blur-box-linear.hpp"
#include "common.hpp"
#include "obs/gs/gs-helper.hpp"
#include "plugin.hpp"
#include "warning-disable.hpp"
#include <cmath>
#include <memory>
2019-09-04 01:03:41 +00:00
#include <stdexcept>
#include "warning-enable.hpp"
2021-06-08 02:47:04 +00:00
#define ST_MAX_BLUR_SIZE 128 // Also change this in box-linear.effect if modified.
streamfx::gfx::blur::box_linear_data::box_linear_data() : _gfx_util(::streamfx::gfx::util::get())
{
2021-06-08 02:38:24 +00:00
auto gctx = streamfx::obs::gs::context();
{
auto file = streamfx::data_file_path("effects/blur/box-linear.effect");
try {
_effect = streamfx::obs::gs::effect::create(file);
} catch (const std::exception& ex) {
DLOG_ERROR("Error loading '%s': %s", file.generic_u8string().c_str(), ex.what());
}
}
}
2021-06-08 02:47:04 +00:00
streamfx::gfx::blur::box_linear_data::~box_linear_data()
{
2021-06-08 02:38:24 +00:00
auto gctx = streamfx::obs::gs::context();
2019-08-04 14:20:26 +00:00
_effect.reset();
}
std::shared_ptr<streamfx::gfx::util> streamfx::gfx::blur::box_linear_data::get_gfx_util()
{
return _gfx_util;
}
2021-06-08 02:47:04 +00:00
streamfx::obs::gs::effect streamfx::gfx::blur::box_linear_data::get_effect()
{
2019-08-04 14:20:26 +00:00
return _effect;
}
2021-06-08 02:47:04 +00:00
streamfx::gfx::blur::box_linear_factory::box_linear_factory() {}
2021-06-08 02:47:04 +00:00
streamfx::gfx::blur::box_linear_factory::~box_linear_factory() {}
2021-06-08 02:47:04 +00:00
bool streamfx::gfx::blur::box_linear_factory::is_type_supported(::streamfx::gfx::blur::type type)
{
switch (type) {
2021-06-08 02:47:04 +00:00
case ::streamfx::gfx::blur::type::Area:
case ::streamfx::gfx::blur::type::Directional:
return true;
default:
return false;
}
}
2021-06-08 02:47:04 +00:00
std::shared_ptr<::streamfx::gfx::blur::base>
streamfx::gfx::blur::box_linear_factory::create(::streamfx::gfx::blur::type type)
{
switch (type) {
2021-06-08 02:47:04 +00:00
case ::streamfx::gfx::blur::type::Area:
return std::make_shared<::streamfx::gfx::blur::box_linear>();
case ::streamfx::gfx::blur::type::Directional:
return std::make_shared<::streamfx::gfx::blur::box_linear_directional>();
default:
throw std::runtime_error("Invalid type.");
}
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_min_size(::streamfx::gfx::blur::type)
{
return double_t(1.0);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_step_size(::streamfx::gfx::blur::type)
{
return double_t(1.0);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_max_size(::streamfx::gfx::blur::type)
{
2021-06-08 02:47:04 +00:00
return double_t(ST_MAX_BLUR_SIZE);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_min_angle(::streamfx::gfx::blur::type v)
{
switch (v) {
2021-06-08 02:47:04 +00:00
case ::streamfx::gfx::blur::type::Directional:
case ::streamfx::gfx::blur::type::Rotational:
return -180.0;
default:
return 0;
}
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_step_angle(::streamfx::gfx::blur::type)
{
return double_t(0.01);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_max_angle(::streamfx::gfx::blur::type v)
{
switch (v) {
2021-06-08 02:47:04 +00:00
case ::streamfx::gfx::blur::type::Directional:
case ::streamfx::gfx::blur::type::Rotational:
return 180.0;
default:
return 0;
}
}
2021-06-08 02:47:04 +00:00
bool streamfx::gfx::blur::box_linear_factory::is_step_scale_supported(::streamfx::gfx::blur::type v)
{
switch (v) {
2021-06-08 02:47:04 +00:00
case ::streamfx::gfx::blur::type::Area:
case ::streamfx::gfx::blur::type::Zoom:
case ::streamfx::gfx::blur::type::Directional:
return true;
default:
return false;
}
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_min_step_scale_x(::streamfx::gfx::blur::type)
{
return double_t(0.01);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_step_step_scale_x(::streamfx::gfx::blur::type)
{
return double_t(0.01);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_max_step_scale_x(::streamfx::gfx::blur::type)
{
return double_t(1000.0);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_min_step_scale_y(::streamfx::gfx::blur::type)
{
return double_t(0.01);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_step_step_scale_y(::streamfx::gfx::blur::type)
{
return double_t(0.01);
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_factory::get_max_step_scale_y(::streamfx::gfx::blur::type)
{
return double_t(1000.0);
}
2021-06-08 02:47:04 +00:00
std::shared_ptr<::streamfx::gfx::blur::box_linear_data> streamfx::gfx::blur::box_linear_factory::data()
{
2021-06-08 02:47:04 +00:00
std::unique_lock<std::mutex> ulock(_data_lock);
std::shared_ptr<::streamfx::gfx::blur::box_linear_data> data = _data.lock();
if (!data) {
2021-06-08 02:47:04 +00:00
data = std::make_shared<::streamfx::gfx::blur::box_linear_data>();
2019-08-04 14:20:26 +00:00
_data = data;
}
return data;
}
2021-06-08 02:47:04 +00:00
::streamfx::gfx::blur::box_linear_factory& streamfx::gfx::blur::box_linear_factory::get()
{
2021-06-08 02:47:04 +00:00
static ::streamfx::gfx::blur::box_linear_factory instance;
return instance;
}
2021-06-08 02:47:04 +00:00
streamfx::gfx::blur::box_linear::box_linear()
: _data(::streamfx::gfx::blur::box_linear_factory::get().data()), _size(1.), _step_scale({1., 1.})
{
2021-06-08 02:38:24 +00:00
_rendertarget = std::make_shared<::streamfx::obs::gs::rendertarget>(GS_RGBA, GS_ZS_NONE);
_rendertarget2 = std::make_shared<::streamfx::obs::gs::rendertarget>(GS_RGBA, GS_ZS_NONE);
}
2021-06-08 02:47:04 +00:00
streamfx::gfx::blur::box_linear::~box_linear() {}
2021-06-08 02:47:04 +00:00
void streamfx::gfx::blur::box_linear::set_input(std::shared_ptr<::streamfx::obs::gs::texture> texture)
{
_input_texture = std::move(texture);
}
2021-06-08 02:47:04 +00:00
::streamfx::gfx::blur::type streamfx::gfx::blur::box_linear::get_type()
{
2021-06-08 02:47:04 +00:00
return ::streamfx::gfx::blur::type::Area;
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear::get_size()
{
2019-08-04 14:20:26 +00:00
return _size;
}
2021-06-08 02:47:04 +00:00
void streamfx::gfx::blur::box_linear::set_size(double_t width)
{
2019-08-04 14:20:26 +00:00
_size = width;
if (_size < 1.0) {
_size = 1.0;
}
2021-06-08 02:47:04 +00:00
if (_size > ST_MAX_BLUR_SIZE) {
_size = ST_MAX_BLUR_SIZE;
}
}
2021-06-08 02:47:04 +00:00
void streamfx::gfx::blur::box_linear::set_step_scale(double_t x, double_t y)
{
2019-08-04 14:20:26 +00:00
_step_scale = {x, y};
}
2021-06-08 02:47:04 +00:00
void streamfx::gfx::blur::box_linear::get_step_scale(double_t& x, double_t& y)
{
2019-08-04 14:20:26 +00:00
x = _step_scale.first;
y = _step_scale.second;
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear::get_step_scale_x()
{
2019-08-04 14:20:26 +00:00
return _step_scale.first;
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear::get_step_scale_y()
{
2019-08-04 14:20:26 +00:00
return _step_scale.second;
}
2021-06-08 02:47:04 +00:00
std::shared_ptr<::streamfx::obs::gs::texture> streamfx::gfx::blur::box_linear::render()
{
2021-06-08 02:38:24 +00:00
auto gctx = streamfx::obs::gs::context();
#if defined(ENABLE_PROFILING) && !defined(D_PLATFORM_MAC) && _DEBUG
2021-06-08 02:38:24 +00:00
auto gdmp = streamfx::obs::gs::debug_marker(streamfx::obs::gs::debug_color_azure_radiance, "Box Linear Blur");
#endif
2019-08-04 14:20:26 +00:00
float_t width = float_t(_input_texture->get_width());
float_t height = float_t(_input_texture->get_height());
gs_set_cull_mode(GS_NEITHER);
gs_enable_color(true, true, true, true);
gs_enable_depth_test(false);
gs_depth_function(GS_ALWAYS);
gs_blend_state_push();
gs_reset_blend_state();
gs_enable_blending(false);
gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);
gs_enable_stencil_test(false);
gs_enable_stencil_write(false);
gs_stencil_function(GS_STENCIL_BOTH, GS_ALWAYS);
gs_stencil_op(GS_STENCIL_BOTH, GS_ZERO, GS_ZERO, GS_ZERO);
// Two Pass Blur
2021-06-08 02:38:24 +00:00
streamfx::obs::gs::effect effect = _data->get_effect();
if (effect) {
// Pass 1
effect.get_parameter("pImage").set_texture(_input_texture);
effect.get_parameter("pImageTexel").set_float2(float_t(1.f / width), 0.f);
effect.get_parameter("pStepScale").set_float2(float_t(_step_scale.first), float_t(_step_scale.second));
effect.get_parameter("pSize").set_float(float_t(_size));
effect.get_parameter("pSizeInverseMul").set_float(float_t(1.0f / (float_t(_size) * 2.0f + 1.0f)));
{
#if defined(ENABLE_PROFILING) && !defined(D_PLATFORM_MAC) && _DEBUG
2021-06-08 02:38:24 +00:00
auto gdm = streamfx::obs::gs::debug_marker(streamfx::obs::gs::debug_color_azure_radiance, "Horizontal");
#endif
auto op = _rendertarget2->render(uint32_t(width), uint32_t(height));
gs_ortho(0, 1., 0, 1., 0, 1.);
while (gs_effect_loop(effect.get_object(), "Draw")) {
_data->get_gfx_util()->draw_fullscreen_triangle();
}
}
// Pass 2
effect.get_parameter("pImage").set_texture(_rendertarget2->get_texture());
effect.get_parameter("pImageTexel").set_float2(0., float_t(1.f / height));
{
#if defined(ENABLE_PROFILING) && !defined(D_PLATFORM_MAC) && _DEBUG
2021-06-08 02:38:24 +00:00
auto gdm = streamfx::obs::gs::debug_marker(streamfx::obs::gs::debug_color_azure_radiance, "Vertical");
#endif
auto op = _rendertarget->render(uint32_t(width), uint32_t(height));
gs_ortho(0, 1., 0, 1., 0, 1.);
while (gs_effect_loop(effect.get_object(), "Draw")) {
_data->get_gfx_util()->draw_fullscreen_triangle();
}
}
}
gs_blend_state_pop();
2019-08-04 14:20:26 +00:00
return _rendertarget->get_texture();
}
2021-06-08 02:47:04 +00:00
std::shared_ptr<::streamfx::obs::gs::texture> streamfx::gfx::blur::box_linear::get()
{
2019-08-04 14:20:26 +00:00
return _rendertarget->get_texture();
}
2021-06-08 02:47:04 +00:00
streamfx::gfx::blur::box_linear_directional::box_linear_directional() : _angle(0) {}
2021-06-08 02:47:04 +00:00
::streamfx::gfx::blur::type streamfx::gfx::blur::box_linear_directional::get_type()
{
2021-06-08 02:47:04 +00:00
return ::streamfx::gfx::blur::type::Directional;
}
2021-06-08 02:47:04 +00:00
double_t streamfx::gfx::blur::box_linear_directional::get_angle()
{
2019-08-04 14:20:26 +00:00
return D_RAD_TO_DEG(_angle);
}
2021-06-08 02:47:04 +00:00
void streamfx::gfx::blur::box_linear_directional::set_angle(double_t angle)
{
2019-08-04 14:20:26 +00:00
_angle = D_DEG_TO_RAD(angle);
}
2021-06-08 02:47:04 +00:00
std::shared_ptr<::streamfx::obs::gs::texture> streamfx::gfx::blur::box_linear_directional::render()
{
2021-06-08 02:38:24 +00:00
auto gctx = streamfx::obs::gs::context();
#if defined(ENABLE_PROFILING) && !defined(D_PLATFORM_MAC) && _DEBUG
2021-06-08 02:38:24 +00:00
auto gdmp =
streamfx::obs::gs::debug_marker(streamfx::obs::gs::debug_color_azure_radiance, "Box Linear Directional Blur");
#endif
2019-08-04 14:20:26 +00:00
float_t width = float_t(_input_texture->get_width());
float_t height = float_t(_input_texture->get_height());
gs_blend_state_push();
gs_reset_blend_state();
gs_enable_color(true, true, true, true);
gs_enable_blending(false);
gs_enable_depth_test(false);
gs_enable_stencil_test(false);
gs_enable_stencil_write(false);
gs_set_cull_mode(GS_NEITHER);
gs_depth_function(GS_ALWAYS);
gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);
gs_stencil_function(GS_STENCIL_BOTH, GS_ALWAYS);
gs_stencil_op(GS_STENCIL_BOTH, GS_ZERO, GS_ZERO, GS_ZERO);
// One Pass Blur
2021-06-08 02:38:24 +00:00
streamfx::obs::gs::effect effect = _data->get_effect();
if (effect) {
effect.get_parameter("pImage").set_texture(_input_texture);
effect.get_parameter("pImageTexel")
.set_float2(float_t(1. / width * cos(_angle)), float_t(1.f / height * sin(_angle)));
effect.get_parameter("pStepScale").set_float2(float_t(_step_scale.first), float_t(_step_scale.second));
effect.get_parameter("pSize").set_float(float_t(_size));
effect.get_parameter("pSizeInverseMul").set_float(float_t(1.0f / (float_t(_size) * 2.0f + 1.0f)));
{
2019-08-04 14:20:26 +00:00
auto op = _rendertarget->render(uint32_t(width), uint32_t(height));
gs_ortho(0, 1., 0, 1., 0, 1.);
while (gs_effect_loop(effect.get_object(), "Draw")) {
_data->get_gfx_util()->draw_fullscreen_triangle();
}
}
}
gs_blend_state_pop();
2019-08-04 14:20:26 +00:00
return _rendertarget->get_texture();
}