/*  rendering_server_wrap_mt.cpp                                            */


#include "rendering_server_wrap_mt.h"
#include "core/os/os.h"
#include "core/config/project_settings.h"

void RenderingServerWrapMT::thread_exit() {
	exit.set();
}

void RenderingServerWrapMT::thread_draw(bool p_swap_buffers, double frame_step) {
	if (!draw_pending.decrement()) {
		rendering_server->draw(p_swap_buffers, frame_step);
	}
}

void RenderingServerWrapMT::thread_flush() {
	draw_pending.decrement();
}

void RenderingServerWrapMT::_thread_callback(void *_instance) {
	RenderingServerWrapMT *vsmt = reinterpret_cast<RenderingServerWrapMT *>(_instance);

	vsmt->thread_loop();
}

void RenderingServerWrapMT::thread_loop() {
	server_thread = Thread::get_caller_id();

	OS::get_singleton()->make_rendering_thread();

	rendering_server->init();

	exit.clear();
	draw_thread_up.set();
	while (!exit.is_set()) {
		// flush commands one by one, until exit is requested
		command_queue.wait_and_flush_one();
	}

	command_queue.flush_all(); // flush all

	rendering_server->finish();
}

/* EVENT QUEUING */

void RenderingServerWrapMT::tick() {
	if (Thread::get_caller_id() != server_thread) {
		command_queue.push(rendering_server, &RenderingServer::tick);
	} else {
		rendering_server->tick();
	}
}

void RenderingServerWrapMT::pre_draw(bool p_will_draw) {
	if (Thread::get_caller_id() != server_thread) {
		command_queue.push(rendering_server, &RenderingServer::pre_draw, p_will_draw);
	} else {
		rendering_server->pre_draw(p_will_draw);
	}
}

void RenderingServerWrapMT::sync() {
	if (create_thread) {
		draw_pending.increment();
		command_queue.push_and_sync(this, &RenderingServerWrapMT::thread_flush);
	} else {
		command_queue.flush_all(); //flush all pending from other threads
	}
}

void RenderingServerWrapMT::draw(bool p_swap_buffers, double frame_step) {
	if (create_thread) {
		draw_pending.increment();
		command_queue.push(this, &RenderingServerWrapMT::thread_draw, p_swap_buffers, frame_step);
	} else {
		rendering_server->draw(p_swap_buffers, frame_step);
	}
}

void RenderingServerWrapMT::init() {
	if (create_thread) {
		print_verbose("RenderingServerWrapMT: Creating render thread");
		OS::get_singleton()->release_rendering_thread();
		if (create_thread) {
			thread.start(_thread_callback, this);
			print_verbose("RenderingServerWrapMT: Starting render thread");
		}
		while (!draw_thread_up.is_set()) {
			OS::get_singleton()->delay_usec(1000);
		}
		print_verbose("RenderingServerWrapMT: Finished render thread");
	} else {
		rendering_server->init();
	}
}

void RenderingServerWrapMT::finish() {
	if (create_thread) {
		command_queue.push(this, &RenderingServerWrapMT::thread_exit);
		thread.wait_to_finish();
	} else {
		rendering_server->finish();
	}

	texture_free_cached_ids();
	shader_free_cached_ids();
	material_free_cached_ids();
	mesh_free_cached_ids();
	multimesh_free_cached_ids();
	viewport_free_cached_ids();
	canvas_free_cached_ids();
	canvas_item_free_cached_ids();
}

void RenderingServerWrapMT::set_use_vsync_callback(bool p_enable) {
	singleton_mt->call_set_use_vsync(p_enable);
}

RenderingServerWrapMT *RenderingServerWrapMT::singleton_mt = nullptr;

RenderingServerWrapMT::RenderingServerWrapMT(RenderingServer *p_contained, bool p_create_thread) :
		command_queue(p_create_thread) {
	singleton_mt = this;
	OS::switch_vsync_function = set_use_vsync_callback; //as this goes to another thread, make sure it goes properly

	rendering_server = p_contained;
	create_thread = p_create_thread;
	pool_max_size = GLOBAL_GET("memory/limits/multithreaded_server/rid_pool_prealloc");

	if (!p_create_thread) {
		server_thread = Thread::get_caller_id();
	} else {
		server_thread = 0;
	}
}

RenderingServerWrapMT::~RenderingServerWrapMT() {
	memdelete(rendering_server);
	//finish();
}