source-mirror: Use util:SourceTexture instead of custom code

This commit is contained in:
Michael Fabian 'Xaymar' Dirks 2018-03-05 16:50:19 +01:00
parent 304db23335
commit 65a7ac696b
2 changed files with 40 additions and 50 deletions

View file

@ -200,11 +200,9 @@ void Source::MirrorAddon::video_render(void *p, gs_effect_t *ef) {
Source::Mirror::Mirror(obs_data_t* data, obs_source_t* src) { Source::Mirror::Mirror(obs_data_t* data, obs_source_t* src) {
m_active = true; m_active = true;
m_source = src; m_source = src;
m_target = nullptr;
m_rescale = false; m_rescale = false;
m_width = m_height = 1; m_width = m_height = 1;
m_renderTarget = std::make_unique<GS::RenderTarget>(GS_RGBA, GS_ZS_NONE);
m_renderTargetScale = std::make_unique<GS::RenderTarget>(GS_RGBA, GS_ZS_NONE); m_renderTargetScale = std::make_unique<GS::RenderTarget>(GS_RGBA, GS_ZS_NONE);
m_sampler = std::make_shared<GS::Sampler>(); m_sampler = std::make_shared<GS::Sampler>();
m_scalingEffect = obs_get_base_effect(obs_base_effect::OBS_EFFECT_DEFAULT); m_scalingEffect = obs_get_base_effect(obs_base_effect::OBS_EFFECT_DEFAULT);
@ -212,34 +210,34 @@ Source::Mirror::Mirror(obs_data_t* data, obs_source_t* src) {
update(data); update(data);
} }
Source::Mirror::~Mirror() { Source::Mirror::~Mirror() {}
if (m_target)
obs_source_release(m_target);
}
uint32_t Source::Mirror::get_width() { uint32_t Source::Mirror::get_width() {
if (m_rescale && m_width > 0 && !m_keepOriginalSize) if (m_rescale && m_width > 0 && !m_keepOriginalSize)
return m_width; return m_width;
if (m_target && m_target != m_source) if (m_mirrorSource)
return obs_source_get_width(m_target); return obs_source_get_width(m_mirrorSource->GetObject());
return 1; return 1;
} }
uint32_t Source::Mirror::get_height() { uint32_t Source::Mirror::get_height() {
if (m_rescale && m_height > 0 && !m_keepOriginalSize) if (m_rescale && m_height > 0 && !m_keepOriginalSize)
return m_height; return m_height;
if (m_target && m_target != m_source) if (m_mirrorSource)
return obs_source_get_height(m_target); return obs_source_get_height(m_mirrorSource->GetObject());
return 1; return 1;
} }
void Source::Mirror::update(obs_data_t* data) { void Source::Mirror::update(obs_data_t* data) {
// Update selected source. // Update selected source.
const char* source = obs_data_get_string(data, P_SOURCE); const char* sourceName = obs_data_get_string(data, P_SOURCE);
if (m_target) if (sourceName != m_mirrorName) {
obs_source_release(m_target); try {
m_target = obs_get_source_by_name(source); m_mirrorSource = std::make_unique<util::SourceTexture>(sourceName);
m_targetName = source; m_mirrorName = sourceName;
} catch (...) {
}
}
// Rescaling // Rescaling
m_rescale = obs_data_get_bool(data, P_SCALING); m_rescale = obs_data_get_bool(data, P_SCALING);
@ -310,30 +308,22 @@ void Source::Mirror::deactivate() {
} }
void Source::Mirror::video_tick(float) { void Source::Mirror::video_tick(float) {
if (!m_target) if (m_mirrorSource)
m_target = obs_get_source_by_name(m_targetName.c_str()); m_mirrorName = obs_source_get_name(m_mirrorSource->GetObject());
} }
void Source::Mirror::video_render(gs_effect_t* effect) { void Source::Mirror::video_render(gs_effect_t* effect) {
if (!m_active || (m_source == m_target) || (m_width == 0 || m_height == 0)) if ((m_width == 0) || (m_height == 0) || !m_mirrorSource) {
return; return;
}
if (m_rescale && m_width > 0 && m_height > 0 && m_scalingEffect) { if (m_rescale && m_width > 0 && m_height > 0 && m_scalingEffect) {
uint32_t sw, sh; uint32_t sw, sh;
sw = obs_source_get_width(m_target); sw = obs_source_get_width(m_mirrorSource->GetObject());
sh = obs_source_get_height(m_target); sh = obs_source_get_height(m_mirrorSource->GetObject());
// Store original Source Texture // Store original Source Texture
try { std::shared_ptr<GS::Texture> tex = m_mirrorSource->Render(sw, sh);
vec4 black; vec4_zero(&black);
auto op = m_renderTarget->Render(sw, sh);
gs_ortho(0, (float_t)sw, 0, (float_t)sh, 0, 1);
gs_clear(GS_CLEAR_COLOR, &black, 0, 0);
obs_source_video_render(m_target);
} catch (...) {
return;
}
gs_eparam_t *scale_param = gs_effect_get_param_by_name(m_scalingEffect, "base_dimension_i"); gs_eparam_t *scale_param = gs_effect_get_param_by_name(m_scalingEffect, "base_dimension_i");
if (scale_param) { if (scale_param) {
@ -353,7 +343,7 @@ void Source::Mirror::video_render(gs_effect_t* effect) {
while (gs_effect_loop(m_scalingEffect, "Draw")) { while (gs_effect_loop(m_scalingEffect, "Draw")) {
gs_eparam_t* image = gs_effect_get_param_by_name(m_scalingEffect, "image"); gs_eparam_t* image = gs_effect_get_param_by_name(m_scalingEffect, "image");
gs_effect_set_next_sampler(image, m_sampler->GetObject()); gs_effect_set_next_sampler(image, m_sampler->GetObject());
obs_source_draw(m_renderTarget->GetTextureObject(), 0, 0, m_width, m_height, false); obs_source_draw(tex->GetObject(), 0, 0, m_width, m_height, false);
} }
} }
while (gs_effect_loop(obs_get_base_effect(OBS_EFFECT_DEFAULT), "Draw")) { while (gs_effect_loop(obs_get_base_effect(OBS_EFFECT_DEFAULT), "Draw")) {
@ -365,10 +355,10 @@ void Source::Mirror::video_render(gs_effect_t* effect) {
while (gs_effect_loop(m_scalingEffect, "Draw")) { while (gs_effect_loop(m_scalingEffect, "Draw")) {
gs_eparam_t* image = gs_effect_get_param_by_name(m_scalingEffect, "image"); gs_eparam_t* image = gs_effect_get_param_by_name(m_scalingEffect, "image");
gs_effect_set_next_sampler(image, m_sampler->GetObject()); gs_effect_set_next_sampler(image, m_sampler->GetObject());
obs_source_draw(m_renderTarget->GetTextureObject(), 0, 0, m_width, m_height, false); obs_source_draw(tex->GetObject(), 0, 0, m_width, m_height, false);
} }
} }
} else { } else {
obs_source_video_render(m_target); obs_source_video_render(m_mirrorSource->GetObject());
} }
} }

View file

@ -21,10 +21,13 @@
#include "plugin.h" #include "plugin.h"
#include "gs-rendertarget.h" #include "gs-rendertarget.h"
#include "gs-sampler.h" #include "gs-sampler.h"
#include "util-source-texture.h"
#include <memory> #include <memory>
namespace Source { namespace Source {
class MirrorAddon { class MirrorAddon {
obs_source_info osi;
public: public:
MirrorAddon(); MirrorAddon();
~MirrorAddon(); ~MirrorAddon();
@ -45,12 +48,23 @@ namespace Source {
static void deactivate(void *); static void deactivate(void *);
static void video_tick(void *, float); static void video_tick(void *, float);
static void video_render(void *, gs_effect_t *); static void video_render(void *, gs_effect_t *);
private:
obs_source_info osi;
}; };
class Mirror { class Mirror {
bool m_active;
obs_source_t* m_source = nullptr;
bool m_rescale = false;
bool m_keepOriginalSize = false;
uint32_t m_width, m_height;
std::unique_ptr<GS::RenderTarget> m_renderTargetScale;
std::shared_ptr<GS::Sampler> m_sampler;
gs_effect_t* m_scalingEffect = nullptr;
std::unique_ptr<util::SourceTexture> m_mirrorSource;
std::string m_mirrorName;
public: public:
Mirror(obs_data_t*, obs_source_t*); Mirror(obs_data_t*, obs_source_t*);
~Mirror(); ~Mirror();
@ -63,19 +77,5 @@ namespace Source {
void deactivate(); void deactivate();
void video_tick(float); void video_tick(float);
void video_render(gs_effect_t*); void video_render(gs_effect_t*);
private:
bool m_active;
obs_source_t* m_source = nullptr;
obs_source_t* m_target = nullptr;
std::string m_targetName;
bool m_rescale = false;
bool m_keepOriginalSize = false;
uint32_t m_width, m_height;
std::unique_ptr<GS::RenderTarget> m_renderTarget;
std::unique_ptr<GS::RenderTarget> m_renderTargetScale;
std::shared_ptr<GS::Sampler> m_sampler;
gs_effect_t* m_scalingEffect = nullptr;
}; };
}; };