obs-StreamFX/source/source-mirror.hpp

/*
* 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
*/

#pragma once
#include <condition_variable>
#include <memory>
#include <mutex>
#include <thread>
#include <vector>
#include "gfx-source-texture.hpp"
#include "gs-rendertarget.hpp"
#include "gs-sampler.hpp"
#include "obs-audio-capture.hpp"
#include "obs-source.hpp"
#include "plugin.hpp"

// OBS
#ifdef _MSC_VER
#pragma warning(push)
#pragma warning(disable : 4201)
#endif
#include <obs-source.h>
#ifdef _MSC_VER
#pragma warning(pop)
#endif

namespace Source {
	class MirrorAddon {
		obs_source_info osi;

		public:
		MirrorAddon();
		~MirrorAddon();

		static const char*       get_name(void*);
		static void              get_defaults(obs_data_t*);
		static bool              modified_properties(obs_properties_t*, obs_property_t*, obs_data_t*);
		static obs_properties_t* get_properties(void*);

		static void* create(obs_data_t*, obs_source_t*);
		static void  destroy(void*);

		static uint32_t get_width(void*);
		static uint32_t get_height(void*);

		static void update(void*, obs_data_t*);
		static void activate(void*);
		static void deactivate(void*);
		static void video_tick(void*, float);
		static void video_render(void*, gs_effect_t*);
		static void enum_active_sources(void*, obs_source_enum_proc_t, void*);
		static void load(void*, obs_data_t*);
		static void save(void*, obs_data_t*);
	};

	class Mirror {
		bool          m_active;
		obs_source_t* m_self;
		float_t       m_tick;

		// Video Rendering
		std::shared_ptr<obs::source>         m_scene;
		std::shared_ptr<gfx::source_texture> m_scene_texture_renderer;

		// Rescaling
		std::shared_ptr<gs::rendertarget> m_rescale_rt;
		bool                              m_rescale_enabled;
		gs_effect_t*                      m_rescale_effect;
		bool                              m_rescale_keep_orig_size;
		uint32_t                          m_rescale_width;
		uint32_t                          m_rescale_height;
		std::shared_ptr<gs::sampler>      m_sampler;

		// Audio Rendering
		bool                                m_audio_enabled;
		std::mutex                          m_audio_lock;
		std::condition_variable             m_audio_notify;
		obs_source_audio                    m_audio_output;
		std::vector<std::vector<float_t>>   m_audio_data;
		std::thread                         m_audio_thread;
		bool                                m_audio_kill_thread;
		bool                                m_audio_have_output;

		// Input
		std::shared_ptr<obs::source>         m_source;
		obs_sceneitem_t*                     m_source_item;
		std::string                          m_source_name;
		std::shared_ptr<obs::audio_capture>  m_source_audio;

		private:
		void release_input();
		void acquire_input(std::string source_name);

		public:
		Mirror(obs_data_t*, obs_source_t*);
		~Mirror();

		uint32_t get_width();
		uint32_t get_height();

		void update(obs_data_t*);
		void activate();
		void deactivate();
		void video_tick(float);
		void video_render(gs_effect_t*);
		void audio_capture_cb(std::shared_ptr<obs::source> source, audio_data const* const audio, bool muted);
		void audio_output_cb();
		void enum_active_sources(obs_source_enum_proc_t, void*);
		void load(obs_data_t*);
		void save(obs_data_t*);

		void on_source_rename(obs::source* source, std::string new_name, std::string old_name);
	};
}; // namespace Source