#include "Editor.hpp"

#include "GraphicsApi.hpp"

using namespace tp;

Editor::Editor() {

	{ // create path tracer gpu texture
		glGenTextures(1, &mPathRenderTexture);
		glBindTexture(GL_TEXTURE_2D, mPathRenderTexture);

		glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_REPEAT);
		glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_REPEAT);
		glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
		glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);

		glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA32F, 100, 100, 0, GL_RGBA, GL_FLOAT, nullptr);
		glBindTexture(GL_TEXTURE_2D, 0);
	}
}

void Editor::renderViewport() {
	switch (mRenderType) {
		case RenderType::RASTER:
			mRasterRenderer.render(mScene, mScene.mRenderSettings.size);
			break;

		case RenderType::PATH_TRACER:
			{
				// TODO : include in render viewport method
				// renderPathFrame();
				break;
			}

		default:
			break;
	}
}

uint4 Editor::getViewportTexID() {
	switch (mRenderType) {
		case RenderType::PATH_TRACER:
			return mPathRenderTexture;

		case RenderType::RASTER:
			return mRasterRenderer.getRenderBufferID();

		default:
			return 0;
	}
}

void Editor::loadDefaults() {
	mScene.load("rsc/scene/script.lua");
}

void Editor::setViewportSize(const Vec2F& size) {
	// TODO remove
	// mScene.mCamera.rotate(0.01f, 0.0);

	mScene.mRenderSettings.size = size;

	mRasterRenderer.getRenderBuffer()->resize(size);
	mScene.mCamera.setRatio(size.y / size.x);
}

Editor::~Editor() {
	glDeleteTextures(1, &mPathRenderTexture);
}

void Editor::renderPathFrame() {
	mPathRenderer.render(mScene, mPathTracerBuffers, mScene.mRenderSettings);
	// mPathTracerBuffers.color.flipY();

	{
		glBindTexture(GL_TEXTURE_2D, mPathRenderTexture);
		const auto size = mPathTracerBuffers.color.size();
		glTexImage2D(
			GL_TEXTURE_2D,
			0,
			GL_RGBA32F,
			(GLsizei) size.x,
			(GLsizei) size.y,
			0,
			GL_RGBA,
			GL_FLOAT,
			mPathTracerBuffers.color.getBuff()
		);
		glBindTexture(GL_TEXTURE_2D, 0);

	}
}

void Editor::setRenderType(Editor::RenderType type) {
	mRenderType = type;
}