obs-StreamFX/source/gfx-effect-source.h

// 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 <string>
#include <obs.h>
#include <memory>
#include "gs-effect.h"
#include "gs-rendertarget.h"
#include "gs-texture.h"
#include "gs-vertexbuffer.h"
#include "gfx-source-texture.h"
#include <vector>
#include <map>
#include <utility>

// Data Defines
#define D_TYPE			"CustomShader.Type"
#define D_INPUT_TEXT		"CustomShader.Input.Text"
#define D_INPUT_FILE		"CustomShader.Input.File"

// Translation Defines
#define T_TYPE			"CustomShader.Type"
#define T_TYPE_TEXT		"CustomShader.Type.Text"
#define T_TYPE_FILE		"CustomShader.Type.File"
#define T_INPUT_TEXT		"CustomShader.Input.Text"
#define T_INPUT_FILE		"CustomShader.Input.File"

namespace gfx {
	class effect_source {
		public:
		struct parameter {
			std::string name = "";
			std::shared_ptr<gs::effect_parameter> param;

			struct {
				std::vector<char> buffer;
				std::vector<const char*> names;
				std::vector<const char*> descs;
			} ui;
		};
		struct bool_parameter : parameter {
			bool value = false;
		};
		struct int_parameter : parameter {
			int32_t value[4] = { 0,0,0,0 };
		};
		struct float_parameter : parameter {
			float_t value[4] = { 0,0,0,0 };
		};
		struct texture_parameter : parameter {
			bool isSource = false;

			struct {
				std::string path = "";
				std::shared_ptr<gs::texture> tex;
				struct {
					float_t time_updated = 0;
					time_t time_create = 0;
					time_t time_modified = 0;
					size_t file_size = 0;
					bool modified = true;
				} info;
			} file;
			struct {
				std::string name = "";
				std::shared_ptr<gfx::source_texture> tex;
			} source;

			struct {
				bool doResample = false;
				std::pair<uint32_t, uint32_t> resolution = { 10, 10 };
				std::shared_ptr<gs::rendertarget> rt;
			} resample;
		};
		struct matrix_parameter : parameter {

		};
		typedef std::pair<std::string, gs::effect_parameter::type> paramident_t;

		private:


		protected:
		obs_source_t* m_source;
		std::shared_ptr<gs::vertex_buffer> m_quadBuffer;

		// Effect Information
		struct {
			std::shared_ptr<gs::effect> effect;
			std::string text;
			std::string path;
			struct {
				float_t time_updated;
				time_t time_create;
				time_t time_modified;
				size_t file_size;
				bool modified;
			} file_info;
		} m_shader;
		std::map<paramident_t, std::shared_ptr<parameter>> m_parameters;

		// Status
		float_t m_timeExisting;
		float_t m_timeActive;

		std::string m_defaultShaderPath = "shaders/";

		static bool property_type_modified(void* priv, obs_properties_t* props, obs_property_t* prop, obs_data_t* sett);
		static bool property_input_modified(void* priv, obs_properties_t* props, obs_property_t* prop, obs_data_t* sett);
		
		virtual bool is_special_parameter(std::string name, gs::effect_parameter::type type) = 0;

		virtual bool video_tick_impl(float_t time) = 0;
		virtual bool video_render_impl(gs_effect_t* parent_effect, uint32_t viewW, uint32_t viewH) = 0;

		public:
		effect_source(obs_data_t* data, obs_source_t* owner);
		~effect_source();

		void get_properties(obs_properties_t* properties);
		static void get_defaults(obs_data_t* data);
		void update(obs_data_t* data);
		bool test_for_updates(const char* text, const char* path);
		void update_parameters(obs_data_t* data);
		void apply_parameters();

		void activate();
		void deactivate();
		
		std::string get_shader_file();

		uint32_t get_width();
		uint32_t get_height();
		void video_tick(float time);
		void video_render(gs_effect_t* parent_effect);


		public:
		enum class InputTypes {
			Text,
			File
		};
	};
}