#include <obs-module.h>
#include <obs-frontend-api.h>
#include <QMainWindow>
#include <QAction>
#include <util/util.hpp>
#include <util/platform.h>
#include <media-io/video-io.h>
#include <media-io/video-frame.h>
#include "DecklinkOutputUI.h"
#include "../../../plugins/decklink/const.h"

OBS_DECLARE_MODULE()
OBS_MODULE_USE_DEFAULT_LOCALE("decklink-output-ui", "en-US")

DecklinkOutputUI *doUI;

bool main_output_running = false;
bool preview_output_running = false;

obs_output_t *output;

struct preview_output {
	bool enabled;
	obs_source_t *current_source;
	obs_output_t *output;

	video_t *video_queue;
	gs_texrender_t *texrender;
	gs_stagesurf_t *stagesurface;
	uint8_t *video_data;
	uint32_t video_linesize;

	obs_video_info ovi;
};

static struct preview_output context = {0};

OBSData load_settings()
{
	BPtr<char> path = obs_module_get_config_path(obs_current_module(),
			"decklinkOutputProps.json");
	BPtr<char> jsonData = os_quick_read_utf8_file(path);
	if (!!jsonData) {
		obs_data_t *data = obs_data_create_from_json(jsonData);
		OBSData dataRet(data);
		obs_data_release(data);

		return dataRet;
	}

	return nullptr;
}

void output_start()
{
	if (!main_output_running) {
		OBSData settings = load_settings();

		if (settings != nullptr) {
			output = obs_output_create("decklink_output", 
					"decklink_output", settings, NULL);

			obs_output_start(output);
			obs_data_release(settings);

			main_output_running = true;
		}
	}
}

void output_stop()
{
	if (main_output_running) {
		obs_output_stop(output);
		obs_output_release(output);
		main_output_running = false;
	}
}


OBSData load_preview_settings()
{
	BPtr<char> path = obs_module_get_config_path(obs_current_module(),
			"decklinkPreviewOutputProps.json");
	BPtr<char> jsonData = os_quick_read_utf8_file(path);
	if (!!jsonData) {
		obs_data_t *data = obs_data_create_from_json(jsonData);
		OBSData dataRet(data);
		obs_data_release(data);

		return dataRet;
	}

	return nullptr;
}

void on_preview_scene_changed(enum obs_frontend_event event, void *param);
void render_preview_source(void *param, uint32_t cx, uint32_t cy);

void preview_output_start()
{
	if (!preview_output_running) {
		OBSData settings = load_preview_settings();

		if (settings != nullptr) {
			context.output = obs_output_create("decklink_output",
					"decklink_preview_output", settings, NULL);

			obs_get_video_info(&context.ovi);

			uint32_t width = context.ovi.base_width;
			uint32_t height = context.ovi.base_height;

			obs_enter_graphics();
			context.texrender = gs_texrender_create(GS_BGRA, GS_ZS_NONE);
			context.stagesurface = gs_stagesurface_create(width, height, GS_BGRA);
			obs_leave_graphics();

			const video_output_info *mainVOI = video_output_get_info(obs_get_video());

			video_output_info vi = {0};
			vi.format = VIDEO_FORMAT_BGRA;
			vi.width = width;
			vi.height = height;
			vi.fps_den = context.ovi.fps_den;
			vi.fps_num = context.ovi.fps_num;
			vi.cache_size = 16;
			vi.colorspace = mainVOI->colorspace;
			vi.range = mainVOI->range;
			vi.name = "decklink_preview_output";

			video_output_open(&context.video_queue, &vi);

			obs_frontend_add_event_callback(on_preview_scene_changed, &context);
			if (obs_frontend_preview_program_mode_active()) {
				context.current_source = obs_frontend_get_current_preview_scene();
			} else {
				context.current_source = obs_frontend_get_current_scene();
			}
			obs_add_main_render_callback(render_preview_source, &context);

			obs_output_set_media(context.output, context.video_queue, obs_get_audio());
			obs_output_start(context.output);

			preview_output_running = true;
		}
	}
}

void preview_output_stop()
{
	if (preview_output_running) {
		obs_output_stop(context.output);
		video_output_stop(context.video_queue);

		obs_remove_main_render_callback(render_preview_source, &context);
		obs_frontend_remove_event_callback(on_preview_scene_changed, &context);

		obs_source_release(context.current_source);

		obs_enter_graphics();
		gs_stagesurface_destroy(context.stagesurface);
		gs_texrender_destroy(context.texrender);
		obs_leave_graphics();

		video_output_close(context.video_queue);

		preview_output_running = false;
	}
}

void on_preview_scene_changed(enum obs_frontend_event event, void *param)
{
	auto ctx = (struct preview_output*)param;
	switch (event) {
		case OBS_FRONTEND_EVENT_STUDIO_MODE_ENABLED:
		case OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED:
			obs_source_release(ctx->current_source);
			ctx->current_source = obs_frontend_get_current_preview_scene();
			break;
		case OBS_FRONTEND_EVENT_STUDIO_MODE_DISABLED:
			obs_source_release(ctx->current_source);
			ctx->current_source = obs_frontend_get_current_scene();
			break;
		case OBS_FRONTEND_EVENT_SCENE_CHANGED:
			if (!obs_frontend_preview_program_mode_active()) {
				obs_source_release(ctx->current_source);
				ctx->current_source = obs_frontend_get_current_scene();
			}
			break;
		default:
			break;
	}
}

void render_preview_source(void *param, uint32_t cx, uint32_t cy)
{
	auto ctx = (struct preview_output*)param;

	if (!ctx->current_source) return;

	uint32_t width = obs_source_get_base_width(ctx->current_source);
	uint32_t height = obs_source_get_base_height(ctx->current_source);

	gs_texrender_reset(ctx->texrender);

	if (gs_texrender_begin(ctx->texrender, width, height)) {
		struct vec4 background;
		vec4_zero(&background);

		gs_clear(GS_CLEAR_COLOR, &background, 0.0f, 0);
		gs_ortho(0.0f, (float)width, 0.0f, (float)height, -100.0f, 100.0f);

		gs_blend_state_push();
		gs_blend_function(GS_BLEND_ONE, GS_BLEND_ZERO);

		obs_source_video_render(ctx->current_source);

		gs_blend_state_pop();
		gs_texrender_end(ctx->texrender);

		struct video_frame output_frame;
		if (video_output_lock_frame(ctx->video_queue, &output_frame, 1, os_gettime_ns()))
		{
			gs_stage_texture(ctx->stagesurface, gs_texrender_get_texture(ctx->texrender));

			if (gs_stagesurface_map(ctx->stagesurface, &ctx->video_data, &ctx->video_linesize)) {
				uint32_t linesize = output_frame.linesize[0];
				for (uint32_t i = 0; i < ctx->ovi.base_height; i++) {
					uint32_t dst_offset = linesize * i;
					uint32_t src_offset = ctx->video_linesize * i;
					memcpy(output_frame.data[0] + dst_offset,
					ctx->video_data + src_offset,
					linesize);
				}

				gs_stagesurface_unmap(ctx->stagesurface);
				ctx->video_data = nullptr;
			}

			video_output_unlock_frame(ctx->video_queue);
		}
	}
}

void addOutputUI(void)
{
	QAction *action = (QAction*)obs_frontend_add_tools_menu_qaction(
			obs_module_text("Decklink Output"));

	QMainWindow *window = (QMainWindow*)obs_frontend_get_main_window();

	obs_frontend_push_ui_translation(obs_module_get_string);
	doUI = new DecklinkOutputUI(window);
	obs_frontend_pop_ui_translation();

	auto cb = []() {
		doUI->ShowHideDialog();
	};

	action->connect(action, &QAction::triggered, cb);
}

static void OBSEvent(enum obs_frontend_event event, void *)
{
	if (event == OBS_FRONTEND_EVENT_FINISHED_LOADING) {
		OBSData settings = load_settings();

		if (settings && obs_data_get_bool(settings, "auto_start"))
			output_start();

		OBSData previewSettings = load_preview_settings();

		if (previewSettings && obs_data_get_bool(previewSettings, "auto_start"))
			preview_output_start();
	}
}

bool obs_module_load(void)
{
	addOutputUI();

	obs_frontend_add_event_callback(OBSEvent, nullptr);

	return true;
}