#include <boost/foreach.hpp>
#include "Ms.h"
#include "Renderer.h"
#include "Camera.h"
#include "Ez.h"
#include "RenderableNode.h"


//======================================================================================================================
// Constructor                                                                                                         =
//======================================================================================================================
Ms::Ms(Renderer& r_):
	RenderingPass(r_),
	ez(r_)
{}


//======================================================================================================================
// init                                                                                                                =
//======================================================================================================================
void Ms::init(const RendererInitializer& initializer)
{
	try
	{
		// create FBO
		fbo.create();
		fbo.bind();

		// inform in what buffers we draw
		fbo.setNumOfColorAttachements(3);

		// create the FAIs
		normalFai.createEmpty2D(r.getWidth(), r.getHeight(), GL_RG16F, GL_RG, GL_FLOAT);
		diffuseFai.createEmpty2D(r.getWidth(), r.getHeight(), GL_RGB16F, GL_RGB, GL_FLOAT);
		specularFai.createEmpty2D(r.getWidth(), r.getHeight(), GL_RGBA16F, GL_RGBA, GL_FLOAT);
		depthFai.createEmpty2D(r.getWidth(), r.getHeight(), GL_DEPTH24_STENCIL8, GL_DEPTH_STENCIL, GL_UNSIGNED_INT_24_8);

		normalFai.setRepeat(false);
		diffuseFai.setRepeat(false);
		specularFai.setRepeat(false);
		depthFai.setRepeat(false);

		// attach the buffers to the FBO
		glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, normalFai.getGlId(), 0);
		glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT1, GL_TEXTURE_2D, diffuseFai.getGlId(), 0);
		glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT2, GL_TEXTURE_2D, specularFai.getGlId(), 0);

		//glFramebufferTexture2D(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT, GL_TEXTURE_2D, depthFai.getGlId(), 0);
		//glFramebufferTexture2D(GL_FRAMEBUFFER, GL_STENCIL_ATTACHMENT, GL_TEXTURE_2D, depthFai.getGlId(), 0);
		glFramebufferTexture2D(GL_FRAMEBUFFER, GL_DEPTH_STENCIL_ATTACHMENT, GL_TEXTURE_2D, depthFai.getGlId(), 0);

		// test if success
		fbo.checkIfGood();

		// unbind
		fbo.unbind();
	}
	catch(std::exception& e)
	{
		throw EXCEPTION("Cannot create deferred shading material stage FBO: " + e.what());
	}

	ez.init(initializer);
}


//======================================================================================================================
// run                                                                                                                 =
//======================================================================================================================
void Ms::run()
{
	if(ez.isEnabled())
	{
		ez.run();
	}

	fbo.bind();

	if(!ez.isEnabled())
	{
		glClear(GL_DEPTH_BUFFER_BIT);
	}

	Renderer::setViewport(0, 0, r.getWidth(), r.getHeight());

	//glEnable(GL_DEPTH_TEST);
	//app->getScene().skybox.Render(cam.getViewMatrix().getRotationPart());
	//glDepthFunc(GL_LEQUAL);

	// if ez then change the default depth test and disable depth writing
	if(ez.isEnabled())
	{
		glDepthMask(false);
		glDepthFunc(GL_EQUAL);
	}

	glClear(GL_COLOR_BUFFER_BIT);

	// render all
	BOOST_FOREACH(const RenderableNode* node, r.getCamera().getVisibleMsRenderableNodes())
	{
		r.getSceneDrawer().renderRenderableNode(*node, r.getCamera(), SceneDrawer::RPT_COLOR);
	}

	// restore depth
	if(ez.isEnabled())
	{
		glDepthMask(true);
		glDepthFunc(GL_LESS);
	}

	fbo.unbind();
	ON_GL_FAIL_THROW_EXCEPTION();
}