mirror of https://github.com/Xaymar/obs-StreamFX
139 lines
4.6 KiB
C++
139 lines
4.6 KiB
C++
// AUTOGENERATED COPYRIGHT HEADER START
|
|
// Copyright (C) 2019-2023 Michael Fabian 'Xaymar' Dirks <info@xaymar.com>
|
|
// AUTOGENERATED COPYRIGHT HEADER END
|
|
|
|
#pragma once
|
|
#include "common.hpp"
|
|
#include "gfx/gfx-source-texture.hpp"
|
|
#include "gfx/gfx-util.hpp"
|
|
#include "obs/gs/gs-effect.hpp"
|
|
#include "obs/obs-source-active-child.hpp"
|
|
#include "obs/obs-source-active-reference.hpp"
|
|
#include "obs/obs-source-factory.hpp"
|
|
#include "obs/obs-source-showing-reference.hpp"
|
|
#include "obs/obs-source-tracker.hpp"
|
|
#include "obs/obs-source.hpp"
|
|
#include "obs/obs-tools.hpp"
|
|
|
|
#include "warning-disable.hpp"
|
|
#include <list>
|
|
#include <map>
|
|
#include "warning-enable.hpp"
|
|
|
|
namespace streamfx::filter::dynamic_mask {
|
|
enum class channel : int8_t { Invalid = -1, Red, Green, Blue, Alpha };
|
|
|
|
class data {
|
|
streamfx::obs::gs::effect _channel_mask_fx;
|
|
|
|
private:
|
|
data();
|
|
|
|
public:
|
|
~data();
|
|
|
|
streamfx::obs::gs::effect channel_mask_fx();
|
|
|
|
public:
|
|
static std::shared_ptr<streamfx::filter::dynamic_mask::data> get();
|
|
};
|
|
|
|
class dynamic_mask_instance : public obs::source_instance {
|
|
std::shared_ptr<streamfx::filter::dynamic_mask::data> _data;
|
|
std::shared_ptr<streamfx::gfx::util> _gfx_util;
|
|
|
|
std::map<std::tuple<channel, channel, std::string>, std::string> _translation_map;
|
|
|
|
streamfx::obs::weak_source _input;
|
|
std::unique_ptr<streamfx::obs::source_active_child> _input_child;
|
|
std::shared_ptr<streamfx::obs::source_showing_reference> _input_vs;
|
|
std::shared_ptr<streamfx::obs::source_active_reference> _input_ac;
|
|
|
|
// Base texture for filtering
|
|
bool _have_base;
|
|
std::shared_ptr<streamfx::obs::gs::rendertarget> _base_rt;
|
|
std::shared_ptr<streamfx::obs::gs::texture> _base_tex;
|
|
gs_color_space _base_color_space;
|
|
gs_color_format _base_color_format;
|
|
bool _base_srgb;
|
|
|
|
bool _have_input;
|
|
std::shared_ptr<streamfx::obs::gs::rendertarget> _input_rt;
|
|
std::shared_ptr<streamfx::obs::gs::texture> _input_tex;
|
|
gs_color_space _input_color_space;
|
|
gs_color_format _input_color_format;
|
|
bool _input_srgb;
|
|
|
|
bool _have_final;
|
|
std::shared_ptr<streamfx::obs::gs::rendertarget> _final_rt;
|
|
std::shared_ptr<streamfx::obs::gs::texture> _final_tex;
|
|
bool _final_srgb;
|
|
|
|
int64_t _debug_texture;
|
|
|
|
struct channel_data {
|
|
float_t value = 0.0;
|
|
float_t scale = 1.0;
|
|
vec4 values = {0, 0, 0, 0};
|
|
};
|
|
std::map<channel, channel_data> _channels;
|
|
|
|
struct _precalc {
|
|
vec4 base;
|
|
vec4 scale;
|
|
matrix4 matrix;
|
|
} _precalc;
|
|
|
|
public:
|
|
dynamic_mask_instance(obs_data_t* data, obs_source_t* self);
|
|
virtual ~dynamic_mask_instance();
|
|
|
|
virtual void load(obs_data_t* settings) override;
|
|
virtual void migrate(obs_data_t* data, uint64_t version) override;
|
|
virtual void update(obs_data_t* settings) override;
|
|
virtual void save(obs_data_t* settings) override;
|
|
|
|
virtual gs_color_space video_get_color_space(size_t count, const gs_color_space* preferred_spaces) override;
|
|
virtual void video_tick(float_t time) override;
|
|
virtual void video_render(gs_effect_t* effect) override;
|
|
|
|
void enum_active_sources(obs_source_enum_proc_t enum_callback, void* param) override;
|
|
void enum_all_sources(obs_source_enum_proc_t enum_callback, void* param) override;
|
|
|
|
void show() override;
|
|
void hide() override;
|
|
void activate() override;
|
|
void deactivate() override;
|
|
|
|
bool acquire(std::string_view name);
|
|
void release();
|
|
};
|
|
|
|
class dynamic_mask_factory : public obs::source_factory<filter::dynamic_mask::dynamic_mask_factory, filter::dynamic_mask::dynamic_mask_instance> {
|
|
std::list<std::string> _translation_cache;
|
|
|
|
public:
|
|
dynamic_mask_factory();
|
|
virtual ~dynamic_mask_factory() override;
|
|
|
|
virtual const char* get_name() override;
|
|
|
|
virtual void get_defaults2(obs_data_t* data) override;
|
|
|
|
virtual obs_properties_t* get_properties2(filter::dynamic_mask::dynamic_mask_instance* data) override;
|
|
|
|
std::string translate_string(const char* format, ...);
|
|
|
|
#ifdef ENABLE_FRONTEND
|
|
static bool on_manual_open(obs_properties_t* props, obs_property_t* property, void* data);
|
|
#endif
|
|
|
|
public: // Singleton
|
|
static void initialize();
|
|
|
|
static void finalize();
|
|
|
|
static std::shared_ptr<dynamic_mask_factory> get();
|
|
};
|
|
} // namespace streamfx::filter::dynamic_mask
|