anki-3d-engine/src/Renderer/Ssao.cpp

/**
 * @file
 *
 * Post-processing stage screen space ambient occlusion pass
 */

#include <boost/lexical_cast.hpp>
#include "Renderer.h"
#include "Camera.h"
#include "RsrcMngr.h"


//======================================================================================================================
// initBlurFbos                                                                                                        =
//======================================================================================================================
void Renderer::Pps::Ssao::initBlurFbo(Fbo& fbo, Texture& fai)
{
	// create FBO
	fbo.create();
	fbo.bind();

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

	// create the texes
	fai.createEmpty2D(bwidth, bheight, GL_ALPHA8, GL_ALPHA, GL_FLOAT, false);
	fai.setTexParameter(GL_TEXTURE_MAG_FILTER, GL_NEAREST);
	fai.setTexParameter(GL_TEXTURE_MIN_FILTER, GL_NEAREST);

	// attach
	glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, fai.getGlId(), 0);

	// test if success
	if(!fbo.isGood())
		FATAL("Cannot create deferred shading post-processing stage SSAO blur FBO");

	// unbind
	fbo.unbind();
}


//======================================================================================================================
// init                                                                                                                =
//======================================================================================================================
void Renderer::Pps::Ssao::init()
{
	width = renderingQuality * r.width;
	height = renderingQuality * r.height;
	bwidth = height * bluringQuality;
	bheight = height * bluringQuality;


	//
	// init FBOs
	//

	// create FBO
	pass0Fbo.create();
	pass0Fbo.bind();

	// inform in what buffers we draw
	pass0Fbo.setNumOfColorAttachements(1);

	// create the FAI
	pass0Fai.createEmpty2D(width, height, GL_ALPHA8, GL_ALPHA, GL_FLOAT, false);
	pass0Fai.setTexParameter(GL_TEXTURE_MAG_FILTER, GL_NEAREST);
	pass0Fai.setTexParameter(GL_TEXTURE_MIN_FILTER, GL_NEAREST);

	// attach
	glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, pass0Fai.getGlId(), 0);

	// test if success
	if(!pass0Fbo.isGood())
		FATAL("Cannot create deferred shading post-processing stage SSAO pass FBO");

	// unbind
	pass0Fbo.unbind();

	initBlurFbo(pass1Fbo, pass1Fai);
	initBlurFbo(pass2Fbo, fai);


	//
	// Shaders
	//

	ssaoSProg = RsrcMngr::shaders.load("shaders/PpsSsao.glsl");

	string pps = "#define _PPS_SSAO_PASS_0_\n#define PASS0_FAI_WIDTH " + lexical_cast<string>(static_cast<float>(width)) +
	             "\n";
	string prefix = "Pass0Width" + lexical_cast<string>(width);
	blurSProg = RsrcMngr::shaders.load(ShaderProg::createSrcCodeToCache("shaders/PpsSsaoBlur.glsl", pps.c_str(),
	                                                                     prefix.c_str()).c_str());


	pps = "#define _PPS_SSAO_PASS_1_\n#define PASS1_FAI_HEIGHT " + lexical_cast<string>(static_cast<float>(bheight)) +
	      "\n";
	prefix = "Pass1Height" + lexical_cast<string>(bheight);
	blurSProg2 = RsrcMngr::shaders.load(ShaderProg::createSrcCodeToCache("shaders/PpsSsaoBlur.glsl", pps.c_str(),
	                                                                      prefix.c_str()).c_str());

	camerarangeUniVar = ssaoSProg->findUniVar("camerarange");
	msDepthFaiUniVar = ssaoSProg->findUniVar("msDepthFai");
	noiseMapUniVar = ssaoSProg->findUniVar("noiseMap");
	msNormalFaiUniVar = ssaoSProg->findUniVar("msNormalFai");
	blurSProgFaiUniVar = blurSProg->findUniVar("tex"); /// @todo rename the tex in the shader
	blurSProg2FaiUniVar = blurSProg2->findUniVar("tex"); /// @todo rename the tex in the shader


	//
	// noise map
	//

	/// @todo fix this crap
	// load noise map and disable temporally the texture compression and enable mipmapping
	bool texCompr = Texture::compressionEnabled;
	bool mipmaping = Texture::mipmappingEnabled;
	Texture::compressionEnabled = false;
	Texture::mipmappingEnabled = true;
	noiseMap = RsrcMngr::textures.load("gfx/noise3.tga");
	noiseMap->setTexParameter(GL_TEXTURE_WRAP_S, GL_REPEAT);
	noiseMap->setTexParameter(GL_TEXTURE_WRAP_T, GL_REPEAT);
	//noise_map->setTexParameter(GL_TEXTURE_MAG_FILTER, GL_NEAREST);
	//noise_map->setTexParameter(GL_TEXTURE_MIN_FILTER, GL_NEAREST);
	Texture::compressionEnabled = texCompr;
	Texture::mipmappingEnabled = mipmaping;

}


//======================================================================================================================
// run                                                                                                                 =
//======================================================================================================================
void Renderer::Pps::Ssao::run()
{
	const Camera& cam = *r.cam;

	glDisable(GL_BLEND);
	glDisable(GL_DEPTH_TEST);


	// 1st pass
	Renderer::setViewport(0, 0, width, height);
	pass0Fbo.bind();
	ssaoSProg->bind();
	Vec2 camRange(cam.getZNear(), cam.getZFar());
	camerarangeUniVar->setVec2(&camRange);
	msDepthFaiUniVar->setTexture(r.ms.depthFai, 0);
	noiseMapUniVar->setTexture(*noiseMap, 1);
	msNormalFaiUniVar->setTexture(r.ms.normalFai, 2);
	Renderer::drawQuad(0);

	// for 2nd and 3rd passes
	Renderer::setViewport(0, 0, bwidth, bheight);

	// 2nd pass
	pass1Fbo.bind();
	blurSProg->bind();
	blurSProgFaiUniVar->setTexture(pass0Fai, 0);
	Renderer::drawQuad(0);

	// 3rd pass
	pass2Fbo.bind();
	blurSProg2->bind();
	blurSProg2FaiUniVar->setTexture(pass1Fai, 0);
	Renderer::drawQuad(0);

	// end
	Fbo::unbind();
}