#include "FrameBuffer.hpp"
#include "GraphicsApi.hpp"

#include <stdio.h>

void glerr(GLenum type) {
	printf("GL ERROR - %i\n", type);
}

#define AssertGL(x)                         \
	{                                         \
		x;                                      \
		GLenum __gle = glGetError();            \
		if (__gle != GL_NO_ERROR) glerr(__gle); \
	}

using namespace tp;

RenderBuffer::RenderBuffer(const Vec2F& size) :
	mSize(size), mSamples(0)
{

	mDrawBuffers[0] = { GL_COLOR_ATTACHMENT0 };

	// --------- texture ---------
	AssertGL(glGenTextures(1, &mTextureId));
	AssertGL(glBindTexture(GL_TEXTURE_2D, mTextureId));
	// Give an empty image to OpenGL ( the last "0" )
	AssertGL(glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, (GLsizei) size.x, (GLsizei) size.y, 0, GL_RGBA, GL_UNSIGNED_BYTE, 0)
	);
	// Poor filtering. Needed
	AssertGL(glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR));
	AssertGL(glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR));

	// --------- depth ---------
	AssertGL(glGenRenderbuffers(1, &mDepthBufferID));
	AssertGL(glBindRenderbuffer(GL_RENDERBUFFER, mDepthBufferID));
	AssertGL(glRenderbufferStorage(GL_RENDERBUFFER, GLW_CONTEXT_DEPTH_COMPONENT, (GLsizei) size.x, (GLsizei) size.y));

	// ------------ framebuffer ------------
	AssertGL(glGenFramebuffers(1, &mFrameBufferID));
	AssertGL(glBindFramebuffer(GL_FRAMEBUFFER, mFrameBufferID));
	AssertGL(glFramebufferRenderbuffer(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT, GL_RENDERBUFFER, mDepthBufferID));
	// Set "renderedTexture" as our colour attachement #0
	AssertGL(glFramebufferTexture(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, mTextureId, 0));
	// Set the list of draw buffers.
	AssertGL(glDrawBuffers(1, mDrawBuffers)); // "1" is the size of DrawBuffers
	ASSERT(glCheckFramebufferStatus(GL_FRAMEBUFFER) == GL_FRAMEBUFFER_COMPLETE);
	glBindFramebuffer(GL_FRAMEBUFFER, 0);
}

RenderBuffer::RenderBuffer(const Vec2F& size, tp::uint1 samples) :
	mSize(size), mSamples(samples)
{

	mDrawBuffers[0] = { GL_COLOR_ATTACHMENT0 };

	// ------- texture ---------
	AssertGL(glGenTextures(1, &mTextureId));
	AssertGL(glBindTexture(GL_TEXTURE_2D_MULTISAMPLE, mTextureId));
#ifdef ENV_OS_ANDROID
	AssertGL(
		glTexStorage2DMultisample(GL_TEXTURE_2D_MULTISAMPLE, samples, GL_RGBA, (GLsizei) size.x, (GLsizei) size.y, GL_TRUE)
	);
#else
	AssertGL(
		glTexImage2DMultisample(GL_TEXTURE_2D_MULTISAMPLE, samples, GL_RGBA, (GLsizei) size.x, (GLsizei) size.y, GL_TRUE)
	);
#endif
	// !?
	// AssertGL(glTexParameteri(GL_TEXTURE_2D_MULTISAMPLE, GL_TEXTURE_MAG_FILTER, GL_LINEAR));
	// AssertGL(glTexParameteri(GL_TEXTURE_2D_MULTISAMPLE, GL_TEXTURE_MIN_FILTER, GL_LINEAR));

	// ------- depth -------
	AssertGL(glGenRenderbuffers(1, &mDepthBufferID));
	AssertGL(glBindRenderbuffer(GL_RENDERBUFFER, mDepthBufferID));
	AssertGL(glRenderbufferStorageMultisample(
		GL_RENDERBUFFER, samples, GLW_CONTEXT_DEPTH_COMPONENT, (GLsizei) size.x, (GLsizei) size.y
	));

	// ------- fbuff -------
	AssertGL(glGenFramebuffers(1, &mFrameBufferID));
	AssertGL(glBindFramebuffer(GL_FRAMEBUFFER, mFrameBufferID));
	AssertGL(glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D_MULTISAMPLE, mTextureId, 0));
	AssertGL(glFramebufferRenderbuffer(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT, GL_RENDERBUFFER, mDepthBufferID));
	AssertGL(glDrawBuffers(1, mDrawBuffers));
	ASSERT(glCheckFramebufferStatus(GL_FRAMEBUFFER) == GL_FRAMEBUFFER_COMPLETE);
	glBindFramebuffer(GL_FRAMEBUFFER, 0);
}

uint4 RenderBuffer::buffId() const { return mFrameBufferID; }

void RenderBuffer::beginDraw() {
	AssertGL(glBindFramebuffer(GL_FRAMEBUFFER, mFrameBufferID));
	setViewport({ 0.f, 0.f, mSize.x, mSize.y });
}

void RenderBuffer::setViewport(const RectF& viewport) {
	AssertGL(glViewport((GLsizei) viewport.x, (GLsizei) viewport.y, (GLsizei) viewport.z, (GLsizei) viewport.w));
}

void RenderBuffer::clear() {
	AssertGL(glClearColor(mClearCol.r, mClearCol.g, mClearCol.b, mClearCol.a));
	AssertGL(glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT | GL_STENCIL_BUFFER_BIT));
}

void RenderBuffer::endDraw() {
	glBindFramebuffer(GL_FRAMEBUFFER, 0);
	glClearColor(0, 0, 0, 0);
	glUseProgram(0);
}

RenderBuffer::~RenderBuffer() {
	glDeleteFramebuffers(1, &mFrameBufferID);
	glDeleteTextures(1, &mTextureId);
	glDeleteRenderbuffers(1, &mDepthBufferID);
}

uint4 RenderBuffer::texId() const { return mTextureId; }

const Vec2F& RenderBuffer::getSize() const { return mSize; }

void RenderBuffer::resize(const Vec2F& size) {
	if (size == mSize) return;

	// Update size
	mSize = size;

	// Bind framebuffer
	AssertGL(glBindFramebuffer(GL_FRAMEBUFFER, mFrameBufferID));

	// Resize color texture
	if (mSamples) {
		AssertGL(glBindTexture(GL_TEXTURE_2D_MULTISAMPLE, mTextureId));
		AssertGL(glTexImage2DMultisample(
			GL_TEXTURE_2D_MULTISAMPLE, mSamples, GL_RGBA, (GLsizei)size.x, (GLsizei)size.y, GL_TRUE
		));
		AssertGL(glBindTexture(GL_TEXTURE_2D_MULTISAMPLE, 0));
	} else {
		AssertGL(glBindTexture(GL_TEXTURE_2D, mTextureId));
		AssertGL(glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, (GLsizei)size.x, (GLsizei)size.y, 0, GL_RGBA, GL_UNSIGNED_BYTE, 0));
		AssertGL(glBindTexture(GL_TEXTURE_2D, 0));
	}

	// Resize depth renderbuffer
	if (mSamples) {
		AssertGL(glBindRenderbuffer(GL_RENDERBUFFER, mDepthBufferID));
		AssertGL(glRenderbufferStorageMultisample(
			GL_RENDERBUFFER, mSamples, GLW_CONTEXT_DEPTH_COMPONENT, (GLsizei)size.x, (GLsizei)size.y
		));
	} else {
		AssertGL(glBindRenderbuffer(GL_RENDERBUFFER, mDepthBufferID));
		AssertGL(glRenderbufferStorage(GL_RENDERBUFFER, GLW_CONTEXT_DEPTH_COMPONENT, (GLsizei)size.x, (GLsizei)size.y));
	}

	// Unbind framebuffer
	AssertGL(glBindFramebuffer(GL_FRAMEBUFFER, 0));
}