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

#include "gfx-effect-source.h"
#include "strings.h"
#include <util/platform.h>
#include <fstream>

bool gfx::effect_source::property_type_modified(void*, obs_properties_t* props, obs_property_t*, obs_data_t* sett) {
	switch ((InputTypes)obs_data_get_int(sett, D_TYPE)) {
		default:
		case InputTypes::Text:
			obs_property_set_visible(obs_properties_get(props, D_INPUT_TEXT), true);
			obs_property_set_visible(obs_properties_get(props, D_INPUT_FILE), false);
			break;
		case InputTypes::File:
			obs_property_set_visible(obs_properties_get(props, D_INPUT_TEXT), false);
			obs_property_set_visible(obs_properties_get(props, D_INPUT_FILE), true);
			break;
	}
	return true;
}

bool gfx::effect_source::property_input_modified(void* obj, obs_properties_t*, obs_property_t*, obs_data_t* sett) {
	const char* text = nullptr;
	const char* file = nullptr;

	switch ((InputTypes)obs_data_get_int(sett, D_TYPE)) {
		default:
		case InputTypes::Text:
			text = obs_data_get_string(sett, D_INPUT_TEXT);
			break;
		case InputTypes::File:
			file = obs_data_get_string(sett, D_INPUT_FILE);
			break;
	}

	return reinterpret_cast<gfx::effect_source*>(obj)->test_for_updates(text, file);
}

void gfx::effect_source::video_tick_impl(float time) {
	// Shader Timers
	time_existing += time;
	time_active += time;

	// File Timer
	shader.file_info.time_updated -= time;
}

void gfx::effect_source::video_render_impl(gs_effect_t* parent_effect) {
	parent_effect;
}

gfx::effect_source::effect_source(obs_data_t* data, obs_source_t* owner) {
	m_source = owner;
	time_existing = 0;
	time_active = 0;
	update(data);
}

gfx::effect_source::~effect_source() {}

void gfx::effect_source::get_properties(obs_properties_t* properties) {
	obs_property_t* p = nullptr;

	p = obs_properties_add_list(properties, D_TYPE, P_TRANSLATE(T_TYPE), OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_INT);
	obs_property_set_long_description(p, P_TRANSLATE(P_DESC(T_TYPE)));
	obs_property_list_add_int(p, P_TRANSLATE(T_TYPE_TEXT), (long long)InputTypes::Text);
	obs_property_list_add_int(p, P_TRANSLATE(T_TYPE_FILE), (long long)InputTypes::File);
	obs_property_set_modified_callback2(p, property_type_modified, this);

	p = obs_properties_add_text(properties, D_INPUT_TEXT, P_TRANSLATE(T_INPUT_TEXT), OBS_TEXT_MULTILINE);
	obs_property_set_long_description(p, P_TRANSLATE(P_DESC(T_INPUT_TEXT)));
	obs_property_set_modified_callback2(p, property_input_modified, this);

	{
		char* tmp_path = obs_module_file(default_shader_path.c_str());
		p = obs_properties_add_path(properties, D_INPUT_FILE, P_TRANSLATE(T_INPUT_FILE), OBS_PATH_FILE,
			"Any (*.effect *.shader *.hlsl);;Effect (*.effect);;Shader (*.shader);;DirectX (*.hlsl)", tmp_path);
		obs_property_set_long_description(p, P_TRANSLATE(P_DESC(T_INPUT_FILE)));
		obs_property_set_modified_callback2(p, property_input_modified, this);
		bfree(tmp_path);
	}

	// ToDo: Place updated properties here or somewhere else?
	for (auto prm : parameters) {
		if (prm.first.second == gs::effect_parameter::type::Boolean) {
			obs_properties_add_bool(properties, prm.second->ui.names[0], prm.second->ui.descs[0]);
		} else if (prm.first.second >= gs::effect_parameter::type::Integer && prm.first.second <= gs::effect_parameter::type::Integer4) {
			size_t cnt = (size_t)prm.first.second - (size_t)gs::effect_parameter::type::Integer;

			for (size_t idx = 0; idx <= cnt; idx++) {
				obs_properties_add_int(properties, prm.second->ui.names[idx], prm.second->ui.descs[idx], INT_MIN, INT_MAX, 1);
			}
		} else if (prm.first.second >= gs::effect_parameter::type::Float && prm.first.second <= gs::effect_parameter::type::Float4) {
			size_t cnt = (size_t)prm.first.second - (size_t)gs::effect_parameter::type::Float;

			for (size_t idx = 0; idx <= cnt; idx++) {
				obs_properties_add_float(properties, prm.second->ui.names[idx], prm.second->ui.descs[idx], FLT_MIN, FLT_MAX, 1);
			}
		}
	}
}

void gfx::effect_source::get_defaults(obs_data_t* data) {
	obs_data_set_default_int(data, D_TYPE, (long long)InputTypes::Text);
	obs_data_set_default_string(data, D_INPUT_TEXT, "");
	obs_data_set_default_string(data, D_INPUT_FILE, "");
}

void gfx::effect_source::update(obs_data_t* data) {
	obs_data_addref(data);

	// Update Shader
	InputTypes input_type = (InputTypes)obs_data_get_int(data, D_TYPE);
	if (input_type == InputTypes::Text) {
		const char* text = obs_data_get_string(data, D_INPUT_TEXT);
		test_for_updates(text, nullptr);
	} else if (input_type == InputTypes::File) {
		const char* path = obs_data_get_string(data, D_INPUT_FILE);
		test_for_updates(nullptr, path);
	}

	update_parameters(data);

	obs_data_release(data);
}

bool gfx::effect_source::test_for_updates(const char* text, const char* path) {
	bool is_shader_different = false;
	if (text != nullptr) {
		if (text != shader.text) {
			shader.text = text;
			is_shader_different = true;
		}

		if (is_shader_different) {
			shader.effect = std::make_unique<gs::effect>(shader.text, "Text");
		}
	} else if (path != nullptr) {
		if (path != this->shader.path) {
			this->shader.path = path;
			this->shader.file_info.time_updated = 0;
			this->shader.file_info.time_create = 0;
			this->shader.file_info.time_modified = 0;
			this->shader.file_info.file_size = 0;
			is_shader_different = true;
		}

		// If the update timer is 0 or less, grab new file information.
		if (shader.file_info.time_updated <= 0) {
			struct stat stats;
			if (os_stat(shader.path.c_str(), &stats) == 0) {
				shader.file_info.modified = (shader.file_info.time_create != stats.st_ctime)
					| (shader.file_info.time_modified != stats.st_mtime)
					| (shader.file_info.file_size != stats.st_size);

				// Mark shader as different if the file was changed.
				is_shader_different =
					is_shader_different
					| shader.file_info.modified;

				// Update own information
				shader.file_info.time_create = stats.st_ctime;
				shader.file_info.time_modified = stats.st_mtime;
				shader.file_info.file_size = stats.st_size;
			}

			// Increment timer so that the next check is a reasonable timespan away.
			shader.file_info.time_updated += 0.1f;
		}

		if (is_shader_different || shader.file_info.modified) {
			// gs_effect_create_from_file caches results, which is bad for us.
			std::vector<char> content;
			std::ifstream fs(shader.path.c_str(), std::ios::binary);

			if (fs.good()) {
				size_t beg = fs.tellg();
				fs.seekg(0, std::ios::end);
				size_t sz = size_t(fs.tellg()) - beg;
				content.resize(sz + 1);
				fs.seekg(0, std::ios::beg);
				fs.read(content.data(), sz);
				fs.close();
				content[sz] = '\0';

				shader.effect = std::make_unique<gs::effect>(std::string(content.data()), shader.path);
			}
		}
	}

	// If the shader is different, rebuild the parameter list.
	if (is_shader_different) {
		if (shader.effect) {
			// ToDo: Figure out if a recycling approach would work.
			//  Might improve stability in low memory situations.
			std::map<paramident_t, std::shared_ptr<parameter>> new_params;
			auto effect_param_list = shader.effect->get_parameters();
			for (auto effect_param : effect_param_list) {
				paramident_t ident;
				ident.first = effect_param.get_name();
				ident.second = effect_param.get_type();

				if (is_special_parameter(ident.first, ident.second))
					continue;

				auto entry = parameters.find(ident);
				if (entry != parameters.end()) {
					entry->second->param = std::make_shared<gs::effect_parameter>(effect_param);
					new_params.insert_or_assign(ident, entry->second);
					parameters.erase(entry);
				} else {
					std::shared_ptr<parameter> param;

					if (ident.second == gs::effect_parameter::type::Boolean) {
						std::shared_ptr<bool_parameter> nparam = std::make_shared<bool_parameter>();

						std::string ui_name, ui_desc;
						ui_name = ident.first;
						ui_desc = ident.first;

						nparam->ui.buffer.resize(ui_name.size() + 1 + ui_desc.size() + 1);
						memset(nparam->ui.buffer.data(), 0, nparam->ui.buffer.size());
						memcpy(nparam->ui.buffer.data(), ui_name.c_str(), ui_name.size());
						memcpy(nparam->ui.buffer.data() + ui_name.size() + 1, ui_desc.c_str(), ui_desc.size());

						nparam->ui.names.resize(1);
						nparam->ui.names[0] = nparam->ui.buffer.data();

						nparam->ui.descs.resize(1);
						nparam->ui.descs[0] = nparam->ui.buffer.data() + ui_name.size() + 1;

						param = std::dynamic_pointer_cast<parameter>(nparam);
					} else if (ident.second >= gs::effect_parameter::type::Integer && ident.second <= gs::effect_parameter::type::Integer4) {
						std::shared_ptr<int_parameter> nparam = std::make_shared<int_parameter>();

						size_t cnt = (size_t)ident.second - (size_t)gs::effect_parameter::type::Integer;

						std::string ui_name[4], ui_desc[4];
						size_t bufsize = 0;
						if (cnt > 0) {
							for (size_t idx = 0; idx <= cnt; idx++) {
								ui_name[idx] = ident.first + (char)(48 + idx);
								ui_desc[idx] = ident.first + "[" + (char)(48 + idx) + "]";

								bufsize += ui_name[idx].size() + 1;
								bufsize += ui_desc[idx].size() + 1;
							}
						} else {
							ui_name[0] = ident.first;
							ui_desc[0] = ident.first;
							bufsize += ui_name[0].size() + 1;
							bufsize += ui_desc[0].size() + 1;
						}

						nparam->ui.names.resize(cnt + 1);
						nparam->ui.descs.resize(cnt + 1);

						nparam->ui.buffer.resize(bufsize);
						memset(nparam->ui.buffer.data(), 0, bufsize);
						size_t off = 0;
						for (size_t idx = 0; idx <= cnt; idx++) {
							memcpy(nparam->ui.buffer.data() + off, ui_name[idx].c_str(), ui_name[idx].size());
							nparam->ui.names[idx] = nparam->ui.buffer.data() + off;
							off += ui_name[idx].size() + 1;

							memcpy(nparam->ui.buffer.data() + off, ui_desc[idx].c_str(), ui_desc[idx].size());
							nparam->ui.descs[idx] = nparam->ui.buffer.data() + off;
							off += ui_desc[idx].size() + 1;
						}

						param = std::dynamic_pointer_cast<parameter>(nparam);
					} else if (ident.second >= gs::effect_parameter::type::Float && ident.second <= gs::effect_parameter::type::Float4) {
						std::shared_ptr<float_parameter> nparam = std::make_shared<float_parameter>();

						size_t cnt = (size_t)ident.second - (size_t)gs::effect_parameter::type::Float;

						std::string ui_name[4], ui_desc[4];
						size_t bufsize = 0;
						if (cnt > 0) {
							for (size_t idx = 0; idx <= cnt; idx++) {
								ui_name[idx] = ident.first + (char)(48 + idx);
								ui_desc[idx] = ident.first + "[" + (char)(48 + idx) + "]";

								bufsize += ui_name[idx].size() + 1;
								bufsize += ui_desc[idx].size() + 1;
							}
						} else {
							ui_name[0] = ident.first;
							ui_desc[0] = ident.first;
							bufsize += ui_name[0].size() + 1;
							bufsize += ui_desc[0].size() + 1;
						}

						nparam->ui.names.resize(cnt + 1);
						nparam->ui.descs.resize(cnt + 1);

						nparam->ui.buffer.resize(bufsize);
						memset(nparam->ui.buffer.data(), 0, bufsize);
						size_t off = 0;
						for (size_t idx = 0; idx <= cnt; idx++) {
							memcpy(nparam->ui.buffer.data() + off, ui_name[idx].c_str(), ui_name[idx].size());
							nparam->ui.names[idx] = nparam->ui.buffer.data() + off;
							off += ui_name[idx].size() + 1;

							memcpy(nparam->ui.buffer.data() + off, ui_desc[idx].c_str(), ui_desc[idx].size());
							nparam->ui.descs[idx] = nparam->ui.buffer.data() + off;
							off += ui_desc[idx].size() + 1;
						}

						param = std::dynamic_pointer_cast<parameter>(nparam);
					} else {

					}

					if (param) {
						param->name = ident.first;
						param->param = std::make_shared<gs::effect_parameter>(effect_param);
						new_params.insert_or_assign(ident, param);
					}
				}
			}
			parameters = std::move(new_params);
		} else {
			parameters.clear();
		}
	}

	return is_shader_different;
}

void gfx::effect_source::update_parameters(obs_data_t* data) {
	for (auto prm : parameters) {
		if (prm.first.second == gs::effect_parameter::type::Boolean) {
			auto param = std::static_pointer_cast<bool_parameter>(prm.second);
			param->value = obs_data_get_bool(data, prm.second->ui.names[0]);
		} else if (prm.first.second >= gs::effect_parameter::type::Integer && prm.first.second <= gs::effect_parameter::type::Integer4) {
			auto param = std::static_pointer_cast<int_parameter>(prm.second);
			for (size_t idx = 0; idx < prm.second->ui.names.size(); idx++) {
				param->value[idx] = obs_data_get_int(data, prm.second->ui.names[idx]);
			}
		} else if (prm.first.second >= gs::effect_parameter::type::Float && prm.first.second <= gs::effect_parameter::type::Float4) {
			auto param = std::static_pointer_cast<float_parameter>(prm.second);
			for (size_t idx = 0; idx < prm.second->ui.names.size(); idx++) {
				param->value[idx] = obs_data_get_double(data, prm.second->ui.names[idx]);
			}
		}
	}
}

void gfx::effect_source::apply_parameters() {
	for (auto prm : parameters) {
		if (prm.first.second == gs::effect_parameter::type::Boolean) {
			auto param = std::static_pointer_cast<bool_parameter>(prm.second);

		} else if (prm.first.second >= gs::effect_parameter::type::Integer && prm.first.second <= gs::effect_parameter::type::Integer4) {
			auto param = std::static_pointer_cast<int_parameter>(prm.second);

		} else if (prm.first.second >= gs::effect_parameter::type::Float && prm.first.second <= gs::effect_parameter::type::Float4) {
			auto param = std::static_pointer_cast<float_parameter>(prm.second);

		}
	}
}

void gfx::effect_source::activate() {
	time_active = 0;
}

void gfx::effect_source::deactivate() {
	time_active = 0;
}

std::string gfx::effect_source::get_shader_file() {
	return shader.path;
}

uint32_t gfx::effect_source::get_width() {
	return 0;
}

uint32_t gfx::effect_source::get_height() {
	return 0;
}

void gfx::effect_source::video_tick(float time) {
	video_tick_impl(time);
}

void gfx::effect_source::video_render(gs_effect_t* parent_effect) {
	video_render_impl(parent_effect);
}