#include "state.h"

static void _tick(State* self)
{
	preview_tick(&self->preview);
}

static void _update(State* self)
{
	SDL_GetWindowSize(self->window, &self->settings.windowSize.x, &self->settings.windowSize.y);
	
	imgui_update(&self->imgui);

	if (self->imgui.isQuit) 
		self->isRunning = false;
}

static void _draw(State* self)
{
	imgui_draw();

	SDL_GL_SwapWindow(self->window);
}

void init(State* self)
{

	log_info(STATE_INIT_INFO);
	
	settings_init(&self->settings);

	if (!SDL_Init(SDL_INIT_VIDEO))
	{
		log_error(std::format(STATE_SDL_INIT_ERROR, SDL_GetError()));
		quit(self);
	}

	log_info(STATE_SDL_INIT_INFO);
	
	// Todo, when sdl3 mixer is released officially
	/*
    if ((Mix_Init(STATE_MIX_FLAGS) & mixFlags) != mixFlags) 
		log_warning(std::format(STATE_MIX_INIT_WARNING, Mix_GetError()));
		
    if 
	(
		Mix_OpenAudioDevice
		(
			STATE_MIX_SAMPLE_RATE, 
			STATE_MIX_FORMAT, 
			STATE_MIX_CHANNELS, 
			STATE_CHUNK_SIZE, 
			STATE_MIX_DEVICE, 
			STATE_MIX_ALLOWED_CHANGES
		) 
		< 0 
	)
	{
		log_warning(std::format(STATE_MIX_INIT_WARNING, Mix_GetError()));
        Mix_Quit();
    }
	else
		log_info(STATE_MIX_INIT_INFO);
	*/

	SDL_CreateWindowAndRenderer
	(
		WINDOW_TITLE, 
		self->settings.windowSize.x, 
		self->settings.windowSize.y, 
		WINDOW_FLAGS, 
		&self->window, 
		&self->renderer
	);

	SDL_GL_SetAttribute(SDL_GL_CONTEXT_PROFILE_MASK, SDL_GL_CONTEXT_PROFILE_CORE);
   	SDL_GL_SetAttribute(SDL_GL_CONTEXT_MAJOR_VERSION, 3);
   	SDL_GL_SetAttribute(SDL_GL_CONTEXT_MINOR_VERSION, 3);

	self->glContext = SDL_GL_CreateContext(self->window);

	window_vsync_set(self->settings.isVsync);
	
	if (!self->glContext)
	{
		log_error(std::format(STATE_GL_CONTEXT_INIT_ERROR, SDL_GetError()));
		quit(self);
	}

	glewInit();

	log_info(std::format(STATE_GL_CONTEXT_INIT_INFO, (const char*)glGetString(GL_VERSION)));
	
	glEnable(GL_BLEND);
	glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);  
	glLineWidth(STATE_GL_LINE_WIDTH);
	glDisable(GL_MULTISAMPLE);
	glDisable(GL_DEPTH_TEST);
	glDisable(GL_LINE_SMOOTH);
	
	resources_init(&self->resources);
	dialog_init(&self->dialog, self->window);
	clipboard_init(&self->clipboard, &self->anm2);
	snapshots_init(&self->snapshots, &self->anm2, &self->reference, &self->preview);
	preview_init(&self->preview, &self->anm2, &self->reference, &self->resources, &self->settings);
	generate_preview_init(&self->generatePreview, &self->anm2, &self->reference, &self->resources, &self->settings);
	editor_init(&self->editor, &self->anm2, &self->reference, &self->resources, &self->settings);
	
	imgui_init
	(
		&self->imgui,
		&self->dialog,
		&self->resources,
		&self->anm2,
		&self->reference,
		&self->editor,
		&self->preview,
		&self->generatePreview,
		&self->settings,
		&self->snapshots,
		&self->clipboard,
		self->window,
		&self->glContext
	);

	if (!self->argument.empty())
	{
		anm2_deserialize(&self->anm2, &self->resources, self->argument);
		window_title_from_path_set(self->window, self->argument);
	}
	else
		anm2_new(&self->anm2);
}

void loop(State* self)
{
	self->tick = SDL_GetTicks();
	self->update = self->tick;

	while (self->tick > self->lastTick + TICK_DELAY)
	{
		self->tick = SDL_GetTicks();
		
		if (self->tick - self->lastTick < TICK_DELAY)
        	SDL_Delay(TICK_DELAY - (self->tick - self->lastTick));

		_tick(self);

		self->lastTick = self->tick;
	}

	if (self->settings.isVsync)
	{
		_update(self);
		_draw(self);
	}
	else
	{
		while (self->update > self->lastUpdate + UPDATE_DELAY)
		{
			self->update = SDL_GetTicks();
			
			if (self->update - self->lastUpdate < UPDATE_DELAY)
				SDL_Delay(UPDATE_DELAY - (self->update - self->lastUpdate));

			_update(self);
			_draw(self);

			self->lastUpdate = self->update;
		}

		SDL_Delay(STATE_DELAY_MIN);
	}
}

void quit(State* self)
{
	imgui_free();
	generate_preview_free(&self->generatePreview);
	preview_free(&self->preview);
	editor_free(&self->editor);
	resources_free(&self->resources);

	/*
    Mix_CloseAudio();
    Mix_Quit();
	*/

	SDL_GL_DestroyContext(self->glContext);
	SDL_Quit();

	settings_save(&self->settings);
	log_info(STATE_QUIT_INFO);
}