obs-StreamFX/source/filters/filter-shader.cpp

301 lines
9.3 KiB
C++
Raw Normal View History

/*
* Modern effects for a modern Streamer
* Copyright (C) 2017 Michael Fabian Dirks
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include "filter-shader.hpp"
#include "strings.hpp"
#include "utility.hpp"
#define ST "Filter.Shader"
P_INITIALIZER(FilterShaderInit)
2018-11-07 14:24:25 +00:00
{
initializer_functions.push_back([] { filter::shader::shader_factory::initialize(); });
finalizer_functions.push_back([] { filter::shader::shader_factory::finalize(); });
}
static std::shared_ptr<filter::shader::shader_factory> factory_instance = nullptr;
void filter::shader::shader_factory::initialize()
2018-11-07 14:24:25 +00:00
{
factory_instance = std::make_shared<filter::shader::shader_factory>();
}
void filter::shader::shader_factory::finalize()
2018-11-07 14:24:25 +00:00
{
factory_instance.reset();
}
std::shared_ptr<filter::shader::shader_factory> filter::shader::shader_factory::get()
2018-11-07 14:24:25 +00:00
{
return factory_instance;
}
static void get_defaults(obs_data_t* data)
{
obs_data_set_default_string(data, S_SHADER_FILE, obs_module_file("shaders/filter/example.effect"));
obs_data_set_default_string(data, S_SHADER_TECHNIQUE, "Draw");
}
filter::shader::shader_factory::shader_factory()
2018-11-07 14:24:25 +00:00
{
memset(&_source_info, 0, sizeof(obs_source_info));
_source_info.id = "obs-stream-effects-filter-shader";
_source_info.type = OBS_SOURCE_TYPE_FILTER;
_source_info.output_flags = OBS_SOURCE_VIDEO;
_source_info.get_name = [](void*) { return D_TRANSLATE(ST); };
_source_info.get_defaults = get_defaults;
_source_info.create = [](obs_data_t* data, obs_source_t* self) {
try {
return static_cast<void*>(new filter::shader::shader_instance(data, self));
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to create source, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to create source.");
}
return static_cast<void*>(nullptr);
};
_source_info.destroy = [](void* ptr) {
try {
delete reinterpret_cast<filter::shader::shader_instance*>(ptr);
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to delete source, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to delete source.");
}
};
_source_info.get_properties = [](void* ptr) {
obs_properties_t* pr = obs_properties_create();
try {
if (ptr)
reinterpret_cast<filter::shader::shader_instance*>(ptr)->properties(pr);
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to retrieve options, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to retrieve options.");
}
return pr;
};
_source_info.get_width = [](void* ptr) {
try {
if (ptr)
return reinterpret_cast<filter::shader::shader_instance*>(ptr)->width();
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to retrieve width, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to retrieve width.");
}
return uint32_t(0);
};
_source_info.get_height = [](void* ptr) {
try {
if (ptr)
return reinterpret_cast<filter::shader::shader_instance*>(ptr)->height();
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to retrieve height, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to retrieve height.");
}
return uint32_t(0);
};
_source_info.update = [](void* ptr, obs_data_t* data) {
try {
if (ptr)
reinterpret_cast<filter::shader::shader_instance*>(ptr)->update(data);
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to update, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to update.");
}
};
_source_info.activate = [](void* ptr) {
try {
if (ptr)
reinterpret_cast<filter::shader::shader_instance*>(ptr)->activate();
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to activate, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to activate.");
}
};
_source_info.deactivate = [](void* ptr) {
try {
if (ptr)
reinterpret_cast<filter::shader::shader_instance*>(ptr)->deactivate();
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to deactivate, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to deactivate.");
}
};
_source_info.video_tick = [](void* ptr, float_t time) {
try {
if (ptr)
reinterpret_cast<filter::shader::shader_instance*>(ptr)->video_tick(time);
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to tick, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to tick.");
}
};
_source_info.video_render = [](void* ptr, gs_effect_t* effect) {
try {
if (ptr)
reinterpret_cast<filter::shader::shader_instance*>(ptr)->video_render(effect);
} catch (std::exception& ex) {
P_LOG_ERROR("<filter-shader> Failed to render, error: %s", ex.what());
} catch (...) {
P_LOG_ERROR("<filter-shader> Failed to render.");
}
};
obs_register_source(&_source_info);
}
filter::shader::shader_factory::~shader_factory() {}
filter::shader::shader_instance::shader_instance(obs_data_t* data, obs_source_t* self)
: _self(self), _active(true), _width(0), _height(0)
{
_fx = std::make_shared<gfx::effect_source::effect_source>();
_fx->set_valid_property_cb(std::bind(&filter::shader::shader_instance::valid_param, this, std::placeholders::_1));
_fx->set_override_cb(std::bind(&filter::shader::shader_instance::override_param, this, std::placeholders::_1));
_rt = std::make_shared<gs::rendertarget>(GS_RGBA, GS_ZS_NONE);
update(data);
}
filter::shader::shader_instance::~shader_instance() {}
uint32_t filter::shader::shader_instance::width()
2018-11-07 14:24:25 +00:00
{
return _width;
}
uint32_t filter::shader::shader_instance::height()
2018-11-07 14:24:25 +00:00
{
return _height;
}
void filter::shader::shader_instance::properties(obs_properties_t* props)
{
_fx->properties(props);
}
void filter::shader::shader_instance::update(obs_data_t* data)
{
_fx->update(data);
}
void filter::shader::shader_instance::activate()
2018-11-07 14:24:25 +00:00
{
_active = true;
}
void filter::shader::shader_instance::deactivate()
2018-11-07 14:24:25 +00:00
{
_active = false;
}
bool filter::shader::shader_instance::valid_param(std::shared_ptr<gs::effect_parameter> param)
{
if (strcmpi(param->get_name().c_str(), "ImageSource"))
return false;
if (strcmpi(param->get_name().c_str(), "ImageSource_Size"))
return false;
if (strcmpi(param->get_name().c_str(), "ImageSource_Texel"))
return false;
return true;
}
void filter::shader::shader_instance::override_param(std::shared_ptr<gs::effect> effect)
{
auto p_source = effect->get_parameter("ImageSource");
auto p_source_size = effect->get_parameter("ImageSource_Size");
auto p_source_texel = effect->get_parameter("ImageSource_Texel");
if (p_source && (p_source->get_type() == gs::effect_parameter::type::Texture)) {
p_source->set_texture(_rt_tex);
}
if (p_source_size && (p_source_size->get_type() == gs::effect_parameter::type::Float2)) {
p_source_size->set_float2(_width, _height);
}
if (p_source_texel && (p_source_size->get_type() == gs::effect_parameter::type::Float2)) {
p_source_texel->set_float2(1.0f / _width, 1.0f / _height);
}
}
void filter::shader::shader_instance::video_tick(float_t sec_since_last)
2018-11-07 14:24:25 +00:00
{
obs_source_t* target = obs_filter_get_target(_self);
{ // Update width and height.
_width = obs_source_get_base_width(target);
_height = obs_source_get_base_height(target);
}
if (_fx->tick(sec_since_last)) {
obs_data_t* data = obs_source_get_settings(_self);
update(data);
obs_data_release(data);
}
_rt_updated = false;
}
void filter::shader::shader_instance::video_render(gs_effect_t* effect)
2018-11-07 14:24:25 +00:00
{
// Grab initial values.
obs_source_t* parent = obs_filter_get_parent(_self);
obs_source_t* target = obs_filter_get_target(_self);
gs_effect_t* effect_default = obs_get_base_effect(obs_base_effect::OBS_EFFECT_DEFAULT);
// Skip filter if anything is wrong.
if (!_active || !parent || !target || !_width || !_height || !effect_default) {
obs_source_skip_video_filter(_self);
return;
}
if (!_rt_updated) {
if (obs_source_process_filter_begin(_self, GS_RGBA, OBS_ALLOW_DIRECT_RENDERING)) {
auto op = _rt->render(_width, _height);
gs_blend_state_push();
gs_reset_blend_state();
gs_set_cull_mode(GS_NEITHER);
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_ortho(0, static_cast<float_t>(_width), 0, static_cast<float_t>(_height), -1., 1.);
obs_source_process_filter_end(_self, effect_default, _width, _height);
gs_blend_state_pop();
}
_rt_tex = _rt->get_texture();
_rt_updated = true;
}
try {
_fx->render();
} catch (...) {
obs_source_skip_video_filter(_self);
}
}