2017-07-03 00:46:33 +00:00
|
|
|
/*
|
|
|
|
* Modern effects for a modern Streamer
|
2018-04-28 11:59:54 +00:00
|
|
|
* Copyright (C) 2017-2018 Michael Fabian Dirks
|
2017-07-03 00:46:33 +00:00
|
|
|
*
|
|
|
|
* 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-blur.h"
|
2018-09-30 18:50:00 +00:00
|
|
|
#include <inttypes.h>
|
|
|
|
#include <map>
|
|
|
|
#include <math.h>
|
2017-07-03 00:46:33 +00:00
|
|
|
#include "strings.h"
|
2017-12-14 00:39:33 +00:00
|
|
|
#include "util-math.h"
|
2017-07-03 00:46:33 +00:00
|
|
|
|
|
|
|
extern "C" {
|
2018-09-30 18:50:00 +00:00
|
|
|
#pragma warning(push)
|
|
|
|
#pragma warning(disable : 4201)
|
2018-04-23 15:53:27 +00:00
|
|
|
#include "graphics/graphics.h"
|
|
|
|
#include "graphics/matrix4.h"
|
2018-09-30 18:50:00 +00:00
|
|
|
#include "util/platform.h"
|
|
|
|
#pragma warning(pop)
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
// Translation Strings
|
|
|
|
#define SOURCE_NAME "Filter.Blur"
|
|
|
|
|
|
|
|
#define P_TYPE "Filter.Blur.Type"
|
|
|
|
#define P_TYPE_BOX "Filter.Blur.Type.Box"
|
|
|
|
#define P_TYPE_GAUSSIAN "Filter.Blur.Type.Gaussian"
|
|
|
|
#define P_TYPE_BILATERAL "Filter.Blur.Type.Bilateral"
|
|
|
|
#define P_SIZE "Filter.Blur.Size"
|
|
|
|
#define P_BILATERAL_SMOOTHING "Filter.Blur.Bilateral.Smoothing"
|
|
|
|
#define P_BILATERAL_SHARPNESS "Filter.Blur.Bilateral.Sharpness"
|
|
|
|
#define P_COLORFORMAT "Filter.Blur.ColorFormat"
|
2018-09-30 20:48:47 +00:00
|
|
|
#define P_MASK "Filter.Blur.Mask"
|
|
|
|
#define P_MASK_TYPE "Filter.Blur.Mask.Type"
|
|
|
|
#define P_MASK_TYPE_REGION "Filter.Blur.Mask.Type.Region"
|
|
|
|
#define P_MASK_TYPE_IMAGE "Filter.Blur.Mask.Type.Image"
|
|
|
|
#define P_MASK_TYPE_SOURCE "Filter.Blur.Mask.Type.Source"
|
|
|
|
#define P_MASK_REGION_LEFT "Filter.Blur.Mask.Region.Left"
|
|
|
|
#define P_MASK_REGION_RIGHT "Filter.Blur.Mask.Region.Right"
|
|
|
|
#define P_MASK_REGION_TOP "Filter.Blur.Mask.Region.Top"
|
|
|
|
#define P_MASK_REGION_BOTTOM "Filter.Blur.Mask.Region.Bottom"
|
|
|
|
#define P_MASK_REGION_FEATHER "Filter.Blur.Mask.Region.Feather"
|
|
|
|
#define P_MASK_REGION_FEATHER_SHIFT "Filter.Blur.Mask.Region.Feather.Shift"
|
|
|
|
#define P_MASK_REGION_INVERT "Filter.Blur.Mask.Region.Invert"
|
|
|
|
#define P_MASK_IMAGE "Filter.Blur.Mask.Image"
|
|
|
|
#define P_MASK_SOURCE "Filter.Blur.Mask.Source"
|
|
|
|
#define P_MASK_COLOR "Filter.Blur.Mask.Color"
|
2018-09-30 22:54:36 +00:00
|
|
|
#define P_MASK_ALPHA "Filter.Blur.Mask.Alpha"
|
2018-09-30 20:48:47 +00:00
|
|
|
#define P_MASK_MULTIPLIER "Filter.Blur.Mask.Multiplier"
|
2018-04-28 11:59:54 +00:00
|
|
|
|
2018-01-16 10:47:24 +00:00
|
|
|
// Initializer & Finalizer
|
2018-09-30 18:50:00 +00:00
|
|
|
INITIALIZER(filterBlurFactoryInitializer)
|
|
|
|
{
|
|
|
|
initializerFunctions.push_back([] { filter::blur::factory::initialize(); });
|
|
|
|
finalizerFunctions.push_back([] { filter::blur::factory::finalize(); });
|
2017-12-13 23:24:37 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
enum ColorFormat : uint64_t { // ToDo: Refactor into full class.
|
2017-07-06 04:02:41 +00:00
|
|
|
RGB,
|
|
|
|
YUV, // 701
|
|
|
|
};
|
2017-12-13 23:24:37 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
static uint8_t const max_kernel_size = 25;
|
2017-07-18 14:47:29 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
bool filter::blur::instance::apply_shared_param(gs_texture_t* input, float texelX, float texelY)
|
|
|
|
{
|
|
|
|
bool result = true;
|
2017-07-03 05:20:15 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
result = result && gs_set_param_texture(blur_effect->get_object(), "u_image", input);
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
vec2 imageSize;
|
|
|
|
vec2_set(&imageSize, (float)gs_texture_get_width(input), (float)gs_texture_get_height(input));
|
|
|
|
result = result && gs_set_param_float2(blur_effect->get_object(), "u_imageSize", &imageSize);
|
2018-04-28 12:41:18 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
vec2 imageTexelDelta;
|
|
|
|
vec2_set(&imageTexelDelta, 1.0f, 1.0f);
|
|
|
|
vec2_div(&imageTexelDelta, &imageTexelDelta, &imageSize);
|
|
|
|
result = result && gs_set_param_float2(blur_effect->get_object(), "u_imageTexel", &imageTexelDelta);
|
2018-04-28 12:41:18 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
vec2 texel;
|
|
|
|
vec2_set(&texel, texelX, texelY);
|
|
|
|
result = result && gs_set_param_float2(blur_effect->get_object(), "u_texelDelta", &texel);
|
2018-04-28 12:41:18 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
result = result && gs_set_param_int(blur_effect->get_object(), "u_radius", (int)size);
|
|
|
|
result = result && gs_set_param_int(blur_effect->get_object(), "u_diameter", (int)(1 + (size * 2)));
|
2018-04-28 12:41:18 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
return result;
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
bool filter::blur::instance::apply_bilateral_param()
|
|
|
|
{
|
|
|
|
if (type != type::Bilateral)
|
|
|
|
return false;
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (blur_effect->has_parameter("bilateralSmoothing")) {
|
|
|
|
blur_effect->get_parameter("bilateralSmoothing").set_float((float)(bilateral_smoothing * (1 + size * 2)));
|
|
|
|
}
|
2017-07-06 04:02:41 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (blur_effect->has_parameter("bilateralSharpness")) {
|
|
|
|
blur_effect->get_parameter("bilateralSharpness").set_float((float)(1.0 - bilateral_sharpness));
|
|
|
|
}
|
2017-07-06 04:02:41 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
return true;
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
bool filter::blur::instance::apply_gaussian_param()
|
|
|
|
{
|
|
|
|
std::shared_ptr<gs::texture> kernel = filter::blur::factory::get()->get_kernel(filter::blur::type::Gaussian);
|
2017-07-03 05:20:15 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (blur_effect->has_parameter("kernel")) {
|
|
|
|
blur_effect->get_parameter("kernel").set_texture(kernel);
|
|
|
|
} else {
|
|
|
|
return false;
|
2017-07-03 05:20:15 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (blur_effect->has_parameter("kernelTexel")) {
|
|
|
|
float_t wb = 1.0f / kernel->get_width();
|
|
|
|
float_t hb = 1.0f / kernel->get_height();
|
|
|
|
blur_effect->get_parameter("kernelTexel").set_float2(wb, hb);
|
|
|
|
}
|
2017-07-06 04:02:41 +00:00
|
|
|
|
2017-07-03 05:20:15 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2018-09-30 22:54:36 +00:00
|
|
|
bool filter::blur::instance::apply_mask_parameters(std::shared_ptr<gs::effect> effect, gs_texture_t* original_texture,
|
|
|
|
gs_texture_t* blurred_texture)
|
2018-09-30 20:48:47 +00:00
|
|
|
{
|
|
|
|
if (effect->has_parameter("image_orig")) {
|
|
|
|
effect->get_parameter("image_orig").set_texture(original_texture);
|
|
|
|
}
|
|
|
|
if (effect->has_parameter("image_blur")) {
|
|
|
|
effect->get_parameter("image_blur").set_texture(blurred_texture);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Region
|
|
|
|
if (mask.type == mask_type::Region) {
|
|
|
|
if (effect->has_parameter("mask_region_left")) {
|
|
|
|
effect->get_parameter("mask_region_left").set_float(mask.region.left);
|
|
|
|
}
|
|
|
|
if (effect->has_parameter("mask_region_right")) {
|
|
|
|
effect->get_parameter("mask_region_right").set_float(mask.region.right);
|
|
|
|
}
|
|
|
|
if (effect->has_parameter("mask_region_top")) {
|
|
|
|
effect->get_parameter("mask_region_top").set_float(mask.region.top);
|
|
|
|
}
|
|
|
|
if (effect->has_parameter("mask_region_bottom")) {
|
|
|
|
effect->get_parameter("mask_region_bottom").set_float(mask.region.bottom);
|
|
|
|
}
|
|
|
|
if (effect->has_parameter("mask_region_feather")) {
|
|
|
|
effect->get_parameter("mask_region_feather").set_float(mask.region.feather);
|
|
|
|
}
|
|
|
|
if (effect->has_parameter("mask_region_feather_shift")) {
|
|
|
|
effect->get_parameter("mask_region_feather_shift").set_float(mask.region.feather_shift);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Image
|
|
|
|
if (mask.type == mask_type::Image) {
|
|
|
|
if (effect->has_parameter("mask_image")) {
|
|
|
|
if (mask.image.texture) {
|
|
|
|
effect->get_parameter("mask_image").set_texture(mask.image.texture);
|
|
|
|
} else {
|
|
|
|
effect->get_parameter("mask_image").set_texture(nullptr);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Source
|
|
|
|
if (mask.type == mask_type::Source) {
|
|
|
|
if (effect->has_parameter("mask_image")) {
|
|
|
|
if (mask.source.texture) {
|
|
|
|
effect->get_parameter("mask_image").set_texture(mask.source.texture);
|
|
|
|
} else {
|
|
|
|
effect->get_parameter("mask_image").set_texture(nullptr);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Shared
|
|
|
|
if (effect->has_parameter("mask_color")) {
|
2018-09-30 22:54:36 +00:00
|
|
|
effect->get_parameter("mask_color").set_float4(mask.color.r, mask.color.g, mask.color.b, mask.color.a);
|
2018-09-30 20:48:47 +00:00
|
|
|
}
|
|
|
|
if (effect->has_parameter("mask_multiplier")) {
|
|
|
|
effect->get_parameter("mask_multiplier").set_float(mask.multiplier);
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
bool filter::blur::instance::modified_properties(void* ptr, obs_properties_t* props, obs_property* prop,
|
|
|
|
obs_data_t* settings)
|
|
|
|
{
|
|
|
|
bool showBilateral = (obs_data_get_int(settings, P_TYPE) == type::Bilateral);
|
|
|
|
|
2018-09-30 20:48:47 +00:00
|
|
|
prop;
|
|
|
|
ptr;
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
// bilateral blur
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_BILATERAL_SMOOTHING), showBilateral);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_BILATERAL_SHARPNESS), showBilateral);
|
|
|
|
|
|
|
|
// region
|
2018-09-30 20:48:47 +00:00
|
|
|
bool show_mask = obs_data_get_bool(settings, P_MASK);
|
|
|
|
mask_type mtype = static_cast<mask_type>(obs_data_get_int(settings, P_MASK_TYPE));
|
|
|
|
bool show_region = (mtype == mask_type::Region) && show_mask;
|
|
|
|
bool show_image = (mtype == mask_type::Image) && show_mask;
|
|
|
|
bool show_source = (mtype == mask_type::Source) && show_mask;
|
|
|
|
|
2018-09-30 22:54:36 +00:00
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_TYPE), show_mask);
|
2018-09-30 20:48:47 +00:00
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_REGION_LEFT), show_region);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_REGION_TOP), show_region);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_REGION_RIGHT), show_region);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_REGION_BOTTOM), show_region);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_REGION_FEATHER), show_region);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_REGION_FEATHER_SHIFT), show_region);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_REGION_INVERT), show_region);
|
|
|
|
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_IMAGE), show_image);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_SOURCE), show_source);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_COLOR), show_image || show_source);
|
2018-09-30 22:54:36 +00:00
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_ALPHA), show_image || show_source);
|
2018-09-30 20:48:47 +00:00
|
|
|
obs_property_set_visible(obs_properties_get(props, P_MASK_MULTIPLIER), show_image || show_source);
|
2018-09-30 18:50:00 +00:00
|
|
|
|
|
|
|
// advanced
|
|
|
|
bool showAdvanced = obs_data_get_bool(settings, S_ADVANCED);
|
|
|
|
obs_property_set_visible(obs_properties_get(props, P_COLORFORMAT), showAdvanced);
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
return true;
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
filter::blur::instance::instance(obs_data_t* settings, obs_source_t* parent)
|
|
|
|
{
|
|
|
|
m_source = parent;
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
obs_enter_graphics();
|
|
|
|
blur_effect = filter::blur::factory::get()->get_effect(filter::blur::type::Box);
|
|
|
|
primary_rendertarget = gs_texrender_create(GS_RGBA, GS_ZS_NONE);
|
|
|
|
secondary_rendertarget = gs_texrender_create(GS_RGBA, GS_ZS_NONE);
|
|
|
|
horizontal_rendertarget = gs_texrender_create(GS_RGBA, GS_ZS_NONE);
|
|
|
|
vertical_rendertarget = gs_texrender_create(GS_RGBA, GS_ZS_NONE);
|
|
|
|
obs_leave_graphics();
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (!primary_rendertarget) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Instance '%s' failed to create primary rendertarget.",
|
|
|
|
obs_source_get_name(m_source));
|
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (!secondary_rendertarget) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Instance '%s' failed to create secondary rendertarget.",
|
|
|
|
obs_source_get_name(m_source));
|
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (!horizontal_rendertarget) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Instance '%s' failed to create horizontal rendertarget.",
|
|
|
|
obs_source_get_name(m_source));
|
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (!vertical_rendertarget) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Instance '%s' failed to create vertical rendertarget.",
|
|
|
|
obs_source_get_name(m_source));
|
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
update(settings);
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
filter::blur::instance::~instance()
|
|
|
|
{
|
2017-07-03 00:46:33 +00:00
|
|
|
obs_enter_graphics();
|
2018-09-30 18:50:00 +00:00
|
|
|
gs_texrender_destroy(primary_rendertarget);
|
|
|
|
gs_texrender_destroy(secondary_rendertarget);
|
|
|
|
gs_texrender_destroy(horizontal_rendertarget);
|
|
|
|
gs_texrender_destroy(vertical_rendertarget);
|
2017-07-03 00:46:33 +00:00
|
|
|
obs_leave_graphics();
|
2018-09-30 18:50:00 +00:00
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
obs_properties_t* filter::blur::instance::get_properties()
|
|
|
|
{
|
|
|
|
obs_properties_t* pr = obs_properties_create();
|
|
|
|
obs_property_t* p = NULL;
|
2017-12-14 02:52:29 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
p = obs_properties_add_list(pr, P_TYPE, P_TRANSLATE(P_TYPE), OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_INT);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_TYPE)));
|
|
|
|
obs_property_set_modified_callback2(p, modified_properties, this);
|
|
|
|
obs_property_list_add_int(p, P_TRANSLATE(P_TYPE_BOX), filter::blur::type::Box);
|
|
|
|
obs_property_list_add_int(p, P_TRANSLATE(P_TYPE_GAUSSIAN), filter::blur::type::Gaussian);
|
|
|
|
obs_property_list_add_int(p, P_TRANSLATE(P_TYPE_BILATERAL), filter::blur::type::Bilateral);
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
p = obs_properties_add_int_slider(pr, P_SIZE, P_TRANSLATE(P_SIZE), 1, 25, 1);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_SIZE)));
|
|
|
|
//obs_property_set_modified_callback(p, modified_properties);
|
|
|
|
|
|
|
|
// bilateral Only
|
|
|
|
p = obs_properties_add_float_slider(pr, P_BILATERAL_SMOOTHING, P_TRANSLATE(P_BILATERAL_SMOOTHING), 0.0, 100.0,
|
|
|
|
0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_BILATERAL_SMOOTHING)));
|
|
|
|
p = obs_properties_add_float_slider(pr, P_BILATERAL_SHARPNESS, P_TRANSLATE(P_BILATERAL_SHARPNESS), 0.0, 100.0,
|
|
|
|
0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_BILATERAL_SHARPNESS)));
|
|
|
|
|
2018-09-30 20:48:47 +00:00
|
|
|
// Mask
|
|
|
|
p = obs_properties_add_bool(pr, P_MASK, P_TRANSLATE(P_MASK));
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK)));
|
2018-09-30 18:50:00 +00:00
|
|
|
obs_property_set_modified_callback2(p, modified_properties, this);
|
2018-09-30 20:48:47 +00:00
|
|
|
p = obs_properties_add_list(pr, P_MASK_TYPE, P_TRANSLATE(P_MASK_TYPE), OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_INT);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_TYPE)));
|
|
|
|
obs_property_set_modified_callback2(p, modified_properties, this);
|
|
|
|
obs_property_list_add_int(p, P_TRANSLATE(P_MASK_TYPE_REGION), mask_type::Region);
|
|
|
|
obs_property_list_add_int(p, P_TRANSLATE(P_MASK_TYPE_IMAGE), mask_type::Image);
|
|
|
|
obs_property_list_add_int(p, P_TRANSLATE(P_MASK_TYPE_SOURCE), mask_type::Source);
|
|
|
|
/// Region
|
|
|
|
p = obs_properties_add_float_slider(pr, P_MASK_REGION_LEFT, P_TRANSLATE(P_MASK_REGION_LEFT), 0.0, 100.0, 0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_REGION_LEFT)));
|
|
|
|
p = obs_properties_add_float_slider(pr, P_MASK_REGION_TOP, P_TRANSLATE(P_MASK_REGION_TOP), 0.0, 100.0, 0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_REGION_TOP)));
|
|
|
|
p = obs_properties_add_float_slider(pr, P_MASK_REGION_RIGHT, P_TRANSLATE(P_MASK_REGION_RIGHT), 0.0, 100.0, 0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_REGION_RIGHT)));
|
|
|
|
p = obs_properties_add_float_slider(pr, P_MASK_REGION_BOTTOM, P_TRANSLATE(P_MASK_REGION_BOTTOM), 0.0, 100.0, 0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_REGION_BOTTOM)));
|
|
|
|
p = obs_properties_add_float_slider(pr, P_MASK_REGION_FEATHER, P_TRANSLATE(P_MASK_REGION_FEATHER), 0.0, 50.0, 0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_REGION_FEATHER)));
|
|
|
|
p = obs_properties_add_float_slider(pr, P_MASK_REGION_FEATHER_SHIFT, P_TRANSLATE(P_MASK_REGION_FEATHER_SHIFT),
|
|
|
|
-100.0, 100.0, 0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_REGION_FEATHER_SHIFT)));
|
|
|
|
p = obs_properties_add_bool(pr, P_MASK_REGION_INVERT, P_TRANSLATE(P_MASK_REGION_INVERT));
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_REGION_INVERT)));
|
|
|
|
/// Image
|
|
|
|
p = obs_properties_add_path(pr, P_MASK_IMAGE, P_TRANSLATE(P_MASK_IMAGE), OBS_PATH_FILE,
|
|
|
|
P_TRANSLATE(S_FILEFILTERS_IMAGES), nullptr);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_IMAGE)));
|
|
|
|
/// Source
|
|
|
|
p = obs_properties_add_list(pr, P_MASK_SOURCE, P_TRANSLATE(P_MASK_SOURCE), OBS_COMBO_TYPE_LIST,
|
|
|
|
OBS_COMBO_FORMAT_STRING);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_SOURCE)));
|
|
|
|
obs_enum_sources(
|
|
|
|
[](void* ptr, obs_source_t* source) {
|
|
|
|
obs_property_t* p = reinterpret_cast<obs_property_t*>(ptr);
|
2018-09-30 22:54:36 +00:00
|
|
|
obs_property_list_add_string(p, std::string(std::string(obs_source_get_name(source)) + " (Source)").c_str(),
|
|
|
|
obs_source_get_name(source));
|
2018-09-30 20:48:47 +00:00
|
|
|
return true;
|
|
|
|
},
|
|
|
|
p);
|
|
|
|
/// Shared
|
|
|
|
p = obs_properties_add_color(pr, P_MASK_COLOR, P_TRANSLATE(P_MASK_COLOR));
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_COLOR)));
|
2018-09-30 22:54:36 +00:00
|
|
|
p = obs_properties_add_float_slider(pr, P_MASK_ALPHA, P_TRANSLATE(P_MASK_ALPHA), 0.0, 100.0, 0.1);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_ALPHA)));
|
2018-09-30 20:48:47 +00:00
|
|
|
p = obs_properties_add_float_slider(pr, P_MASK_MULTIPLIER, P_TRANSLATE(P_MASK_MULTIPLIER), 0.0, 10.0, 0.01);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_MASK_MULTIPLIER)));
|
2018-09-30 18:50:00 +00:00
|
|
|
|
|
|
|
// advanced
|
|
|
|
p = obs_properties_add_bool(pr, S_ADVANCED, P_TRANSLATE(S_ADVANCED));
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(S_ADVANCED)));
|
|
|
|
obs_property_set_modified_callback2(p, modified_properties, this);
|
|
|
|
|
|
|
|
p = obs_properties_add_list(pr, P_COLORFORMAT, P_TRANSLATE(P_COLORFORMAT), OBS_COMBO_TYPE_LIST,
|
|
|
|
OBS_COMBO_FORMAT_INT);
|
|
|
|
obs_property_set_long_description(p, P_TRANSLATE(P_DESC(P_COLORFORMAT)));
|
|
|
|
obs_property_list_add_int(p, "RGB", ColorFormat::RGB);
|
|
|
|
obs_property_list_add_int(p, "YUV", ColorFormat::YUV);
|
|
|
|
|
|
|
|
return pr;
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::instance::update(obs_data_t* settings)
|
|
|
|
{
|
|
|
|
type = (blur::type)obs_data_get_int(settings, P_TYPE);
|
|
|
|
blur_effect = factory::get()->get_effect(type);
|
|
|
|
size = (uint64_t)obs_data_get_int(settings, P_SIZE);
|
|
|
|
|
|
|
|
// bilateral blur
|
|
|
|
bilateral_smoothing = obs_data_get_double(settings, P_BILATERAL_SMOOTHING) / 100.0;
|
|
|
|
bilateral_sharpness = obs_data_get_double(settings, P_BILATERAL_SHARPNESS) / 100.0;
|
|
|
|
|
|
|
|
// region
|
2018-09-30 20:48:47 +00:00
|
|
|
mask.enabled = obs_data_get_bool(settings, P_MASK);
|
|
|
|
if (mask.enabled) {
|
|
|
|
mask.type = static_cast<mask_type>(obs_data_get_int(settings, P_MASK_TYPE));
|
|
|
|
switch (mask.type) {
|
|
|
|
case mask_type::Region:
|
|
|
|
mask.region.left = float_t(obs_data_get_double(settings, P_MASK_REGION_LEFT) / 100.0);
|
|
|
|
mask.region.top = float_t(obs_data_get_double(settings, P_MASK_REGION_TOP) / 100.0);
|
|
|
|
mask.region.right = 1.0f - float_t(obs_data_get_double(settings, P_MASK_REGION_RIGHT) / 100.0);
|
|
|
|
mask.region.bottom = 1.0f - float_t(obs_data_get_double(settings, P_MASK_REGION_BOTTOM) / 100.0);
|
|
|
|
mask.region.feather = float_t(obs_data_get_double(settings, P_MASK_REGION_FEATHER) / 100.0);
|
|
|
|
mask.region.feather_shift = float_t(obs_data_get_double(settings, P_MASK_REGION_FEATHER_SHIFT) / 100.0);
|
|
|
|
mask.region.invert = obs_data_get_bool(settings, P_MASK_REGION_INVERT);
|
|
|
|
break;
|
|
|
|
case mask_type::Image:
|
|
|
|
mask.image.path = obs_data_get_string(settings, P_MASK_IMAGE);
|
|
|
|
break;
|
|
|
|
case mask_type::Source:
|
|
|
|
mask.source.name = obs_data_get_string(settings, P_MASK_SOURCE);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if ((mask.type == mask_type::Image) || (mask.type == mask_type::Source)) {
|
2018-09-30 22:54:36 +00:00
|
|
|
uint32_t color = static_cast<uint32_t>(obs_data_get_int(settings, P_MASK_COLOR));
|
|
|
|
mask.color.r = ((color >> 0) & 0xFF) / 255.0f;
|
|
|
|
mask.color.g = ((color >> 8) & 0xFF) / 255.0f;
|
|
|
|
mask.color.b = ((color >> 16) & 0xFF) / 255.0f;
|
|
|
|
mask.color.a = static_cast<float_t>(obs_data_get_double(settings, P_MASK_ALPHA));
|
2018-09-30 20:48:47 +00:00
|
|
|
mask.multiplier = float_t(obs_data_get_double(settings, P_MASK_MULTIPLIER));
|
|
|
|
}
|
2018-04-28 22:14:29 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
// advanced
|
|
|
|
if (obs_data_get_bool(settings, S_ADVANCED)) {
|
|
|
|
color_format = obs_data_get_int(settings, P_COLORFORMAT);
|
2018-04-28 22:14:29 +00:00
|
|
|
} else {
|
2018-09-30 18:50:00 +00:00
|
|
|
color_format = obs_data_get_default_int(settings, P_COLORFORMAT);
|
2018-04-28 22:14:29 +00:00
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
uint32_t filter::blur::instance::get_width()
|
|
|
|
{
|
2018-09-30 22:54:36 +00:00
|
|
|
return uint32_t(0);
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
uint32_t filter::blur::instance::get_height()
|
|
|
|
{
|
2018-09-30 22:54:36 +00:00
|
|
|
return uint32_t(0);
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::instance::activate() {}
|
|
|
|
|
|
|
|
void filter::blur::instance::deactivate() {}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 22:54:36 +00:00
|
|
|
void filter::blur::instance::video_tick(float)
|
|
|
|
{
|
|
|
|
if (mask.type == mask_type::Image) {
|
|
|
|
if (mask.image.path_old != mask.image.path) {
|
|
|
|
try {
|
|
|
|
mask.image.texture = std::make_shared<gs::texture>(mask.image.path);
|
|
|
|
mask.image.path_old = mask.image.path;
|
|
|
|
} catch (...) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Instance '%s' failed to load image '%s'.", obs_source_get_name(m_source),
|
|
|
|
mask.image.path.c_str());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else if (mask.type == mask_type::Source) {
|
|
|
|
if (mask.source.name_old != mask.source.name) {
|
|
|
|
try {
|
|
|
|
mask.source.source_texture = std::make_shared<gfx::source_texture>(mask.source.name, m_source);
|
|
|
|
mask.source.name_old = mask.source.name;
|
|
|
|
} catch (...) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Instance '%s' failed to grab source '%s'.", obs_source_get_name(m_source),
|
|
|
|
mask.source.name.c_str());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::instance::video_render(gs_effect_t* effect)
|
|
|
|
{
|
|
|
|
obs_source_t* parent = obs_filter_get_parent(m_source);
|
|
|
|
obs_source_t* target = obs_filter_get_target(m_source);
|
|
|
|
uint32_t baseW = obs_source_get_base_width(target);
|
|
|
|
uint32_t baseH = obs_source_get_base_height(target);
|
|
|
|
vec4 black = {0};
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2017-07-03 05:20:15 +00:00
|
|
|
bool failed = false;
|
2018-09-30 18:50:00 +00:00
|
|
|
|
|
|
|
std::shared_ptr<gs::effect> colorConversionEffect = factory::get()->get_color_converter_effect();
|
2017-07-03 00:46:33 +00:00
|
|
|
|
|
|
|
// Skip rendering if our target, parent or context is not valid.
|
2017-12-14 00:39:33 +00:00
|
|
|
if (!target || !parent || !m_source) {
|
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
}
|
2018-09-30 22:54:36 +00:00
|
|
|
if ((baseW == 0) || (baseH == 0)) {
|
2018-09-30 18:50:00 +00:00
|
|
|
if (!have_logged_error)
|
|
|
|
P_LOG_ERROR("<filter-blur> Instance '%s' has invalid size source '%s'.", obs_source_get_name(m_source),
|
|
|
|
obs_source_get_name(target));
|
|
|
|
have_logged_error = true;
|
2017-12-14 00:39:33 +00:00
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
if (!primary_rendertarget || !blur_effect) {
|
|
|
|
if (!have_logged_error)
|
|
|
|
P_LOG_ERROR("<filter-blur> Instance '%s' is unable to render.", obs_source_get_name(m_source),
|
|
|
|
obs_source_get_name(target));
|
|
|
|
have_logged_error = true;
|
2017-07-03 00:46:33 +00:00
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
have_logged_error = false;
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
gs_effect_t* defaultEffect = obs_get_base_effect(obs_base_effect::OBS_EFFECT_DEFAULT);
|
|
|
|
gs_texture_t* sourceTexture = nullptr;
|
2017-07-03 00:46:33 +00:00
|
|
|
|
|
|
|
#pragma region Source To Texture
|
2018-09-30 18:50:00 +00:00
|
|
|
gs_texrender_reset(primary_rendertarget);
|
|
|
|
if (!gs_texrender_begin(primary_rendertarget, baseW, baseH)) {
|
2017-08-19 22:19:58 +00:00
|
|
|
P_LOG_ERROR("<filter-blur> Failed to set up base texture.");
|
2017-07-03 00:46:33 +00:00
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
} else {
|
2017-07-03 05:20:15 +00:00
|
|
|
gs_ortho(0, (float)baseW, 0, (float)baseH, -1, 1);
|
2017-07-03 00:46:33 +00:00
|
|
|
|
|
|
|
// Clear to Black
|
|
|
|
gs_clear(GS_CLEAR_COLOR | GS_CLEAR_DEPTH, &black, 0, 0);
|
|
|
|
|
|
|
|
// Render
|
|
|
|
if (obs_source_process_filter_begin(m_source, GS_RGBA, OBS_NO_DIRECT_RENDERING)) {
|
|
|
|
obs_source_process_filter_end(m_source, effect ? effect : defaultEffect, baseW, baseH);
|
|
|
|
} else {
|
2017-08-19 22:19:58 +00:00
|
|
|
P_LOG_ERROR("<filter-blur> Unable to render source.");
|
2017-07-03 05:20:15 +00:00
|
|
|
failed = true;
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
gs_texrender_end(primary_rendertarget);
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
|
|
|
|
2017-07-03 05:20:15 +00:00
|
|
|
if (failed) {
|
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
sourceTexture = gs_texrender_get_texture(primary_rendertarget);
|
2017-07-03 00:46:33 +00:00
|
|
|
if (!sourceTexture) {
|
2017-08-19 22:19:58 +00:00
|
|
|
P_LOG_ERROR("<filter-blur> Failed to get source texture.");
|
2017-07-03 00:46:33 +00:00
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
#pragma endregion Source To Texture
|
|
|
|
|
2017-07-06 04:02:41 +00:00
|
|
|
// Conversion
|
2018-09-30 18:50:00 +00:00
|
|
|
#pragma region RGB->YUV
|
|
|
|
if ((color_format == ColorFormat::YUV) && colorConversionEffect) {
|
|
|
|
gs_texrender_reset(secondary_rendertarget);
|
|
|
|
if (!gs_texrender_begin(secondary_rendertarget, baseW, baseH)) {
|
2017-08-19 22:19:58 +00:00
|
|
|
P_LOG_ERROR("<filter-blur> Failed to set up base texture.");
|
2017-07-06 04:02:41 +00:00
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
} else {
|
|
|
|
gs_ortho(0, (float)baseW, 0, (float)baseH, -1, 1);
|
|
|
|
|
|
|
|
// Clear to Black
|
|
|
|
gs_clear(GS_CLEAR_COLOR | GS_CLEAR_DEPTH, &black, 0, 0);
|
|
|
|
|
|
|
|
// Set up camera stuff
|
|
|
|
gs_set_cull_mode(GS_NEITHER);
|
|
|
|
gs_reset_blend_state();
|
2017-08-19 22:19:58 +00:00
|
|
|
gs_enable_blending(false);
|
|
|
|
gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);
|
2017-07-06 04:02:41 +00:00
|
|
|
gs_enable_depth_test(false);
|
|
|
|
gs_enable_stencil_test(false);
|
|
|
|
gs_enable_stencil_write(false);
|
|
|
|
gs_enable_color(true, true, true, true);
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if (colorConversionEffect->has_parameter("image")) {
|
|
|
|
colorConversionEffect->get_parameter("image").set_texture(sourceTexture);
|
2017-07-06 04:02:41 +00:00
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
while (gs_effect_loop(colorConversionEffect->get_object(), "RGBToYUV")) {
|
2017-07-06 04:02:41 +00:00
|
|
|
gs_draw_sprite(sourceTexture, 0, baseW, baseH);
|
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
gs_texrender_end(secondary_rendertarget);
|
2017-07-06 04:02:41 +00:00
|
|
|
}
|
2017-07-06 03:57:57 +00:00
|
|
|
|
2017-07-06 04:02:41 +00:00
|
|
|
if (failed) {
|
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
sourceTexture = gs_texrender_get_texture(secondary_rendertarget);
|
2017-07-06 04:02:41 +00:00
|
|
|
if (!sourceTexture) {
|
2017-08-19 22:19:58 +00:00
|
|
|
P_LOG_ERROR("<filter-blur> Failed to get source texture.");
|
2017-07-06 04:02:41 +00:00
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
#pragma endregion RGB->YUV
|
2017-08-19 22:19:58 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
#pragma region blur
|
2017-08-19 22:19:58 +00:00
|
|
|
// Set up camera stuff
|
|
|
|
gs_set_cull_mode(GS_NEITHER);
|
|
|
|
gs_reset_blend_state();
|
|
|
|
gs_enable_blending(true);
|
|
|
|
gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);
|
|
|
|
gs_enable_depth_test(false);
|
|
|
|
gs_enable_stencil_test(false);
|
|
|
|
gs_enable_stencil_write(false);
|
|
|
|
gs_enable_color(true, true, true, true);
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
gs_texture_t* blurred = nullptr;
|
|
|
|
gs_texture_t* intermediate = sourceTexture;
|
|
|
|
|
2017-08-19 22:19:58 +00:00
|
|
|
std::tuple<const char*, gs_texrender_t*, float, float> kvs[] = {
|
2018-09-30 18:50:00 +00:00
|
|
|
std::make_tuple("Horizontal", horizontal_rendertarget, 1.0f / baseW, 0.0f),
|
|
|
|
std::make_tuple("Vertical", vertical_rendertarget, 0.0f, 1.0f / baseH),
|
2017-08-19 22:19:58 +00:00
|
|
|
};
|
2018-09-30 18:50:00 +00:00
|
|
|
|
2018-04-28 22:14:29 +00:00
|
|
|
std::string pass = "Draw";
|
2018-09-30 20:48:47 +00:00
|
|
|
|
2017-08-19 22:19:58 +00:00
|
|
|
for (auto v : kvs) {
|
2018-09-30 18:50:00 +00:00
|
|
|
const char* name = std::get<0>(v);
|
|
|
|
gs_texrender_t* rt = std::get<1>(v);
|
|
|
|
float xpel = std::get<2>(v), ypel = std::get<3>(v);
|
2017-08-19 22:19:58 +00:00
|
|
|
|
|
|
|
if (!apply_shared_param(intermediate, xpel, ypel))
|
|
|
|
break;
|
2018-09-30 18:50:00 +00:00
|
|
|
switch (type) {
|
|
|
|
case Gaussian:
|
|
|
|
apply_gaussian_param();
|
|
|
|
break;
|
|
|
|
case Bilateral:
|
|
|
|
apply_bilateral_param();
|
|
|
|
break;
|
2017-08-19 22:19:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
gs_texrender_reset(rt);
|
|
|
|
if (!gs_texrender_begin(rt, baseW, baseH)) {
|
|
|
|
P_LOG_ERROR("<filter-blur:%s> Failed to begin rendering.", name);
|
|
|
|
break;
|
|
|
|
}
|
2017-07-06 03:57:57 +00:00
|
|
|
|
2017-08-19 22:19:58 +00:00
|
|
|
// Camera
|
|
|
|
gs_ortho(0, (float)baseW, 0, (float)baseH, -1, 1);
|
|
|
|
gs_clear(GS_CLEAR_COLOR | GS_CLEAR_DEPTH, &black, 0, 0);
|
|
|
|
|
|
|
|
// Render
|
2018-09-30 18:50:00 +00:00
|
|
|
while (gs_effect_loop(blur_effect->get_object(), pass.c_str())) {
|
2017-08-19 22:19:58 +00:00
|
|
|
gs_draw_sprite(intermediate, 0, baseW, baseH);
|
|
|
|
}
|
|
|
|
|
|
|
|
gs_texrender_end(rt);
|
|
|
|
intermediate = gs_texrender_get_texture(rt);
|
|
|
|
if (!intermediate) {
|
2018-09-30 18:50:00 +00:00
|
|
|
P_LOG_ERROR("<filter-blur:%s> Failed to get intermediate texture.", name);
|
2017-08-19 22:19:58 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
blurred = intermediate;
|
|
|
|
}
|
2017-07-06 03:57:57 +00:00
|
|
|
if (blurred == nullptr) {
|
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
#pragma endregion blur
|
2017-07-06 03:57:57 +00:00
|
|
|
|
2018-09-30 20:48:47 +00:00
|
|
|
#pragma region Mask
|
|
|
|
if (mask.enabled) {
|
|
|
|
std::string technique = "";
|
|
|
|
switch (mask.type) {
|
|
|
|
case Region:
|
|
|
|
if (mask.region.feather > 0.001) {
|
|
|
|
if (mask.region.invert) {
|
|
|
|
technique = "RegionFeatherInverted";
|
|
|
|
} else {
|
|
|
|
technique = "RegionFeather";
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if (mask.region.invert) {
|
|
|
|
technique = "RegionInverted";
|
|
|
|
} else {
|
|
|
|
technique = "Region";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case Image:
|
|
|
|
case Source:
|
|
|
|
technique = "Image";
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2018-09-30 22:54:36 +00:00
|
|
|
if (mask.source.source_texture) {
|
|
|
|
uint32_t source_width = obs_source_get_width(mask.source.source_texture->get_object());
|
|
|
|
uint32_t source_height = obs_source_get_height(mask.source.source_texture->get_object());
|
|
|
|
|
|
|
|
if (source_width == 0) {
|
|
|
|
source_width = baseW;
|
|
|
|
}
|
|
|
|
if (source_height == 0) {
|
|
|
|
source_height = baseH;
|
|
|
|
}
|
|
|
|
|
|
|
|
mask.source.texture = mask.source.source_texture->render(source_width, source_height);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<gs::effect> mask_effect = factory::get()->get_mask_effect();
|
|
|
|
apply_mask_parameters(mask_effect, sourceTexture, blurred);
|
|
|
|
|
2018-09-30 20:48:47 +00:00
|
|
|
gs_texrender_reset(horizontal_rendertarget);
|
|
|
|
if (gs_texrender_begin(horizontal_rendertarget, baseW, baseH)) {
|
|
|
|
// Camera
|
|
|
|
gs_ortho(0, (float)baseW, 0, (float)baseH, -1, 1);
|
|
|
|
gs_clear(GS_CLEAR_COLOR | GS_CLEAR_DEPTH, &black, 0, 0);
|
|
|
|
|
|
|
|
// Render
|
|
|
|
while (gs_effect_loop(mask_effect->get_object(), technique.c_str())) {
|
|
|
|
gs_draw_sprite(blurred, 0, baseW, baseH);
|
|
|
|
}
|
|
|
|
|
|
|
|
gs_texrender_end(horizontal_rendertarget);
|
|
|
|
|
|
|
|
blurred = gs_texrender_get_texture(horizontal_rendertarget);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#pragma endregion
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
#pragma region YUV->RGB or straight draw
|
2017-07-06 03:57:57 +00:00
|
|
|
// Draw final effect
|
|
|
|
{
|
2017-07-06 04:02:41 +00:00
|
|
|
gs_effect_t* finalEffect = defaultEffect;
|
2018-09-30 18:50:00 +00:00
|
|
|
const char* technique = "Draw";
|
2017-07-06 04:02:41 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
if ((color_format == ColorFormat::YUV) && colorConversionEffect) {
|
|
|
|
finalEffect = colorConversionEffect->get_object();
|
|
|
|
technique = "YUVToRGB";
|
2017-07-06 04:02:41 +00:00
|
|
|
}
|
|
|
|
|
2017-08-19 22:19:58 +00:00
|
|
|
// Set up camera stuff
|
|
|
|
gs_set_cull_mode(GS_NEITHER);
|
|
|
|
gs_reset_blend_state();
|
|
|
|
gs_enable_blending(true);
|
|
|
|
gs_blend_function(GS_BLEND_SRCALPHA, GS_BLEND_INVSRCALPHA);
|
|
|
|
gs_enable_depth_test(false);
|
|
|
|
gs_enable_stencil_test(false);
|
|
|
|
gs_enable_stencil_write(false);
|
|
|
|
gs_enable_color(true, true, true, true);
|
|
|
|
|
2017-07-06 04:02:41 +00:00
|
|
|
gs_eparam_t* param = gs_effect_get_param_by_name(finalEffect, "image");
|
2017-07-06 03:57:57 +00:00
|
|
|
if (!param) {
|
2017-08-19 22:19:58 +00:00
|
|
|
P_LOG_ERROR("<filter-blur:Final> Failed to set image param.");
|
2017-07-06 03:57:57 +00:00
|
|
|
failed = true;
|
|
|
|
} else {
|
|
|
|
gs_effect_set_texture(param, blurred);
|
|
|
|
}
|
2017-07-06 04:02:41 +00:00
|
|
|
while (gs_effect_loop(finalEffect, technique)) {
|
2017-07-06 03:57:57 +00:00
|
|
|
gs_draw_sprite(blurred, 0, baseW, baseH);
|
|
|
|
}
|
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
#pragma endregion YUV->RGB or straight draw
|
2017-07-06 03:57:57 +00:00
|
|
|
|
|
|
|
if (failed) {
|
2017-07-03 00:46:33 +00:00
|
|
|
obs_source_skip_video_filter(m_source);
|
|
|
|
return;
|
2017-07-06 03:57:57 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
filter::blur::factory::factory()
|
|
|
|
{
|
|
|
|
memset(&source_info, 0, sizeof(obs_source_info));
|
|
|
|
source_info.id = "obs-stream-effects-filter-blur";
|
|
|
|
source_info.type = OBS_SOURCE_TYPE_FILTER;
|
|
|
|
source_info.output_flags = OBS_SOURCE_VIDEO;
|
|
|
|
source_info.get_name = get_name;
|
|
|
|
source_info.get_defaults = get_defaults;
|
|
|
|
source_info.get_properties = get_properties;
|
|
|
|
|
|
|
|
source_info.create = create;
|
|
|
|
source_info.destroy = destroy;
|
|
|
|
source_info.update = update;
|
|
|
|
source_info.activate = activate;
|
|
|
|
source_info.deactivate = deactivate;
|
|
|
|
source_info.video_tick = video_tick;
|
|
|
|
source_info.video_render = video_render;
|
|
|
|
|
|
|
|
obs_register_source(&source_info);
|
|
|
|
}
|
2017-07-06 03:57:57 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
filter::blur::factory::~factory() {}
|
2017-07-06 03:57:57 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::factory::on_list_fill()
|
|
|
|
{
|
|
|
|
obs_enter_graphics();
|
2018-09-30 20:48:47 +00:00
|
|
|
|
|
|
|
{
|
|
|
|
char* file = obs_module_file("effects/box-blur.effect");
|
2018-09-30 18:50:00 +00:00
|
|
|
try {
|
2018-09-30 20:48:47 +00:00
|
|
|
effects.insert_or_assign(type::Box, std::make_shared<gs::effect>(file));
|
2018-09-30 18:50:00 +00:00
|
|
|
} catch (std::runtime_error ex) {
|
2018-09-30 20:48:47 +00:00
|
|
|
P_LOG_ERROR("<filter-blur> Loading effect '%s' failed with error(s): %s", file, ex.what());
|
2018-09-30 18:50:00 +00:00
|
|
|
}
|
2018-09-30 20:48:47 +00:00
|
|
|
bfree(file);
|
|
|
|
}
|
|
|
|
{
|
|
|
|
char* file = obs_module_file("effects/gaussian-blur.effect");
|
|
|
|
try {
|
|
|
|
effects.insert_or_assign(type::Gaussian, std::make_shared<gs::effect>(file));
|
|
|
|
} catch (std::runtime_error ex) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Loading effect '%s' failed with error(s): %s", file, ex.what());
|
|
|
|
}
|
|
|
|
bfree(file);
|
|
|
|
}
|
|
|
|
{
|
|
|
|
char* file = obs_module_file("effects/bilateral-blur.effect");
|
|
|
|
try {
|
|
|
|
effects.insert_or_assign(type::Bilateral, std::make_shared<gs::effect>(file));
|
|
|
|
} catch (std::runtime_error ex) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Loading effect '%s' failed with error(s): %s", file, ex.what());
|
|
|
|
}
|
|
|
|
bfree(file);
|
|
|
|
}
|
|
|
|
{
|
|
|
|
char* file = obs_module_file("effects/color-conversion.effect");
|
|
|
|
try {
|
|
|
|
color_converter_effect = std::make_shared<gs::effect>(file);
|
|
|
|
} catch (std::runtime_error ex) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Loading effect '%s' failed with error(s): %s", file, ex.what());
|
|
|
|
}
|
|
|
|
bfree(file);
|
|
|
|
}
|
|
|
|
{
|
|
|
|
char* file = obs_module_file("effects/mask.effect");
|
|
|
|
try {
|
|
|
|
mask_effect = std::make_shared<gs::effect>(file);
|
|
|
|
} catch (std::runtime_error ex) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Loading effect '%s' failed with error(s): %s", file, ex.what());
|
|
|
|
}
|
|
|
|
bfree(file);
|
2018-09-30 18:50:00 +00:00
|
|
|
}
|
2017-07-06 03:57:57 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
generate_kernel_textures();
|
|
|
|
obs_leave_graphics();
|
|
|
|
}
|
2017-07-03 05:20:15 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::factory::on_list_empty()
|
|
|
|
{
|
|
|
|
obs_enter_graphics();
|
|
|
|
effects.clear();
|
|
|
|
kernels.clear();
|
2018-09-30 20:48:47 +00:00
|
|
|
color_converter_effect.reset();
|
|
|
|
mask_effect.reset();
|
2018-09-30 18:50:00 +00:00
|
|
|
obs_leave_graphics();
|
|
|
|
}
|
2017-07-06 03:57:57 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::factory::generate_gaussian_kernels()
|
|
|
|
{
|
|
|
|
// 2D texture, horizontal is value, vertical is kernel size.
|
2018-09-30 20:48:47 +00:00
|
|
|
size_t size_power_of_two = size_t(pow(2, util::math::get_power_of_two_exponent_ceil(max_kernel_size)));
|
2017-07-06 03:57:57 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
std::vector<float_t> texture_Data(size_power_of_two * size_power_of_two);
|
|
|
|
std::vector<float_t> math_data(size_power_of_two);
|
|
|
|
|
|
|
|
for (size_t width = 1; width <= max_kernel_size; width++) {
|
|
|
|
size_t v = (width - 1) * size_power_of_two;
|
|
|
|
|
|
|
|
// Calculate and normalize
|
|
|
|
float_t sum = 0;
|
|
|
|
for (size_t p = 0; p <= width; p++) {
|
|
|
|
math_data[p] = float_t(Gaussian1D(double_t(p), double_t(width)));
|
|
|
|
sum += math_data[p] * (p > 0 ? 2 : 1);
|
2018-04-28 22:14:29 +00:00
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
|
|
|
|
// Normalize to Texture Buffer
|
|
|
|
double_t inverse_sum = 1.0 / sum;
|
|
|
|
for (size_t p = 0; p <= width; p++) {
|
|
|
|
texture_Data[v + p] = float_t(math_data[p] * inverse_sum);
|
2018-04-28 22:14:29 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
// Create Texture
|
|
|
|
try {
|
|
|
|
auto texture_buffer = reinterpret_cast<uint8_t*>(texture_Data.data());
|
|
|
|
auto unsafe_buffer = const_cast<const uint8_t**>(&texture_buffer);
|
2017-07-03 05:20:15 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
kernels.insert_or_assign(filter::blur::type::Gaussian,
|
|
|
|
std::make_shared<gs::texture>(uint32_t(size_power_of_two), uint32_t(size_power_of_two),
|
|
|
|
GS_R32F, 1, unsafe_buffer, gs::texture::flags::None));
|
|
|
|
} catch (std::runtime_error ex) {
|
|
|
|
P_LOG_ERROR("<filter-blur> Failed to create gaussian kernel texture.");
|
|
|
|
}
|
|
|
|
}
|
2017-07-03 05:20:15 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::factory::generate_kernel_textures()
|
|
|
|
{
|
|
|
|
generate_gaussian_kernels();
|
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void* filter::blur::factory::create(obs_data_t* data, obs_source_t* parent)
|
|
|
|
{
|
|
|
|
if (get()->sources.empty()) {
|
|
|
|
get()->on_list_fill();
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
filter::blur::instance* ptr = new filter::blur::instance(data, parent);
|
|
|
|
get()->sources.push_back(ptr);
|
|
|
|
return ptr;
|
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::factory::destroy(void* inptr)
|
|
|
|
{
|
|
|
|
filter::blur::instance* ptr = reinterpret_cast<filter::blur::instance*>(inptr);
|
|
|
|
get()->sources.remove(ptr);
|
|
|
|
if (get()->sources.empty()) {
|
|
|
|
get()->on_list_empty();
|
2017-07-03 05:20:15 +00:00
|
|
|
}
|
2018-09-30 18:50:00 +00:00
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::factory::get_defaults(obs_data_t* data)
|
|
|
|
{
|
|
|
|
obs_data_set_default_int(data, P_TYPE, filter::blur::type::Box);
|
|
|
|
obs_data_set_default_int(data, P_SIZE, 5);
|
|
|
|
|
|
|
|
// bilateral Only
|
|
|
|
obs_data_set_default_double(data, P_BILATERAL_SMOOTHING, 50.0);
|
|
|
|
obs_data_set_default_double(data, P_BILATERAL_SHARPNESS, 90.0);
|
|
|
|
|
|
|
|
// region
|
2018-09-30 20:48:47 +00:00
|
|
|
obs_data_set_default_bool(data, P_MASK, false);
|
|
|
|
obs_data_set_default_int(data, P_MASK_TYPE, mask_type::Region);
|
|
|
|
obs_data_set_default_double(data, P_MASK_REGION_LEFT, 0.0f);
|
|
|
|
obs_data_set_default_double(data, P_MASK_REGION_RIGHT, 0.0f);
|
|
|
|
obs_data_set_default_double(data, P_MASK_REGION_TOP, 0.0f);
|
|
|
|
obs_data_set_default_double(data, P_MASK_REGION_BOTTOM, 0.0f);
|
|
|
|
obs_data_set_default_double(data, P_MASK_REGION_FEATHER, 0.0f);
|
|
|
|
obs_data_set_default_double(data, P_MASK_REGION_FEATHER_SHIFT, 0.0f);
|
|
|
|
obs_data_set_default_bool(data, P_MASK_REGION_INVERT, false);
|
|
|
|
char* default_file = obs_module_file("white.png");
|
|
|
|
obs_data_set_default_string(data, P_MASK_IMAGE, default_file);
|
|
|
|
bfree(default_file);
|
|
|
|
obs_data_set_default_string(data, P_MASK_SOURCE, "");
|
2018-09-30 22:54:36 +00:00
|
|
|
obs_data_set_default_int(data, P_MASK_COLOR, 0xFFFFFFFFull);
|
2018-09-30 20:48:47 +00:00
|
|
|
obs_data_set_default_double(data, P_MASK_MULTIPLIER, 1.0);
|
2018-09-30 18:50:00 +00:00
|
|
|
|
|
|
|
// advanced
|
|
|
|
obs_data_set_default_bool(data, S_ADVANCED, false);
|
|
|
|
obs_data_set_default_int(data, P_COLORFORMAT, ColorFormat::RGB);
|
2017-08-19 22:19:58 +00:00
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
obs_properties_t* filter::blur::factory::get_properties(void* inptr)
|
|
|
|
{
|
|
|
|
return reinterpret_cast<filter::blur::instance*>(inptr)->get_properties();
|
|
|
|
}
|
2017-07-03 00:46:33 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
void filter::blur::factory::update(void* inptr, obs_data_t* settings)
|
|
|
|
{
|
|
|
|
reinterpret_cast<filter::blur::instance*>(inptr)->update(settings);
|
|
|
|
}
|
2017-08-19 22:19:58 +00:00
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
const char* filter::blur::factory::get_name(void* inptr)
|
|
|
|
{
|
2018-09-30 20:48:47 +00:00
|
|
|
inptr;
|
2018-09-30 18:50:00 +00:00
|
|
|
return P_TRANSLATE(SOURCE_NAME);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t filter::blur::factory::get_width(void* inptr)
|
|
|
|
{
|
|
|
|
return reinterpret_cast<filter::blur::instance*>(inptr)->get_width();
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t filter::blur::factory::get_height(void* inptr)
|
|
|
|
{
|
|
|
|
return reinterpret_cast<filter::blur::instance*>(inptr)->get_height();
|
|
|
|
}
|
|
|
|
|
|
|
|
void filter::blur::factory::activate(void* inptr)
|
|
|
|
{
|
|
|
|
reinterpret_cast<filter::blur::instance*>(inptr)->activate();
|
|
|
|
}
|
|
|
|
|
|
|
|
void filter::blur::factory::deactivate(void* inptr)
|
|
|
|
{
|
|
|
|
reinterpret_cast<filter::blur::instance*>(inptr)->deactivate();
|
|
|
|
}
|
|
|
|
|
|
|
|
void filter::blur::factory::video_tick(void* inptr, float delta)
|
|
|
|
{
|
|
|
|
reinterpret_cast<filter::blur::instance*>(inptr)->video_tick(delta);
|
|
|
|
}
|
|
|
|
|
|
|
|
void filter::blur::factory::video_render(void* inptr, gs_effect_t* effect)
|
|
|
|
{
|
|
|
|
reinterpret_cast<filter::blur::instance*>(inptr)->video_render(effect);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<gs::effect> filter::blur::factory::get_effect(filter::blur::type type)
|
|
|
|
{
|
|
|
|
return effects.at(type);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<gs::effect> filter::blur::factory::get_color_converter_effect()
|
|
|
|
{
|
|
|
|
return color_converter_effect;
|
|
|
|
}
|
|
|
|
|
2018-09-30 20:48:47 +00:00
|
|
|
std::shared_ptr<gs::effect> filter::blur::factory::get_mask_effect()
|
|
|
|
{
|
|
|
|
return mask_effect;
|
|
|
|
}
|
|
|
|
|
2018-09-30 18:50:00 +00:00
|
|
|
std::shared_ptr<gs::texture> filter::blur::factory::get_kernel(filter::blur::type type)
|
|
|
|
{
|
|
|
|
return kernels.at(type);
|
|
|
|
}
|
|
|
|
|
|
|
|
static filter::blur::factory* factory_instance = nullptr;
|
|
|
|
|
|
|
|
void filter::blur::factory::initialize()
|
|
|
|
{
|
|
|
|
factory_instance = new filter::blur::factory();
|
|
|
|
}
|
|
|
|
|
|
|
|
void filter::blur::factory::finalize()
|
|
|
|
{
|
|
|
|
delete factory_instance;
|
|
|
|
}
|
|
|
|
|
|
|
|
filter::blur::factory* filter::blur::factory::get()
|
|
|
|
{
|
|
|
|
return factory_instance;
|
2017-07-03 00:46:33 +00:00
|
|
|
}
|