nuclear@41: #include <stdio.h>
nuclear@41: #include <stdlib.h>
nuclear@41: #include <string.h>
nuclear@41: #include <limits.h>
nuclear@41: #include "opengl.h"
nuclear@41: #include "renderer_deferred.h"
nuclear@41: #include "level.h"
nuclear@41: #include "sdr.h"
nuclear@41: #include "datapath.h"
nuclear@41: 
nuclear@41: static unsigned int load_sdr(const char *vfname, const char *pfname);
nuclear@65: static void attach_color_buffer(unsigned int fbo, int count, const unsigned int *tex);
nuclear@41: 
nuclear@41: DeferredRenderer::DeferredRenderer()
nuclear@41: {
nuclear@72: 	fbo = 0;
nuclear@41: 	for(int i=0; i<MRT_COUNT; i++) {
nuclear@41: 		mrt_tex[i] = 0;
nuclear@41: 	}
nuclear@72: 	mrt_prog = deferred_debug = deferred_omni = 0;
nuclear@41: 
nuclear@41: 	curr_prog = 0;
nuclear@60: 	num_draw_bufs = 1;
nuclear@41: }
nuclear@41: 
nuclear@41: DeferredRenderer::~DeferredRenderer()
nuclear@41: {
nuclear@41: 	if(mrt_prog) {
nuclear@41: 		free_program(mrt_prog);
nuclear@41: 	}
nuclear@41: 	if(deferred_omni) {
nuclear@41: 		free_program(deferred_omni);
nuclear@41: 	}
nuclear@41: 	if(deferred_debug) {
nuclear@41: 		free_program(deferred_debug);
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	if(mrt_tex[0]) {
nuclear@41: 		glDeleteTextures(MRT_COUNT, mrt_tex);
nuclear@41: 	}
nuclear@41: }
nuclear@41: 
nuclear@41: bool DeferredRenderer::init(int xsz, int ysz)
nuclear@41: {
nuclear@72: 	if(!Renderer::init(xsz, ysz)) {
nuclear@72: 		return false;
nuclear@72: 	}
nuclear@41: 
nuclear@41: 	if(!GLEW_ARB_texture_float) {
nuclear@69: 		fprintf(stderr, "%s: error: no floating point texture support\n", __FUNCTION__);
nuclear@41: 		return false;
nuclear@41: 	}
nuclear@41: 	if(!GLEW_ARB_draw_buffers) {
nuclear@69: 		fprintf(stderr, "%s: error: no multiple render target support\n", __FUNCTION__);
nuclear@41: 		return false;
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	int max_draw_buf;
nuclear@41: 	glGetIntegerv(GL_MAX_DRAW_BUFFERS_ARB, &max_draw_buf);
nuclear@41: 	printf("max draw buffers: %d\n", max_draw_buf);
nuclear@41: 	if(max_draw_buf < MRT_COUNT) {
nuclear@69: 		fprintf(stderr, "%s: error: not enough draw buffers (%d), %d required\n", __FUNCTION__, max_draw_buf, MRT_COUNT);
nuclear@41: 		return false;
nuclear@41: 	}
nuclear@60: 	num_draw_bufs = MRT_COUNT;
nuclear@41: 
nuclear@41: 	if(!(mrt_prog = load_sdr("mrt.v.glsl", "mrt.p.glsl"))) {
nuclear@41: 		return false;
nuclear@41: 	}
nuclear@41: 	set_uniform_int(mrt_prog, "tex_dif", 0);
nuclear@41: 	set_uniform_int(mrt_prog, "tex_norm", 1);
nuclear@41: 
nuclear@41: 	if(!(deferred_omni = load_sdr("deferred_omni.v.glsl", "deferred_omni.p.glsl"))) {
nuclear@41: 		return false;
nuclear@41: 	}
nuclear@41: 	for(int i=0; i<MRT_COUNT; i++) {
nuclear@41: 		char uname[32];
nuclear@41: 		sprintf(uname, "mrt%d", i);
nuclear@41: 		set_uniform_int(deferred_omni, uname, i);
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	return true;
nuclear@41: }
nuclear@41: 
nuclear@41: int DeferredRenderer::get_tangent_location() const
nuclear@41: {
nuclear@41: 	return get_attrib_loc(mrt_prog, "attr_tangent");
nuclear@41: }
nuclear@41: 
nuclear@41: unsigned int DeferredRenderer::get_current_program() const
nuclear@41: {
nuclear@41: 	return curr_prog;
nuclear@41: }
nuclear@41: 
nuclear@41: void DeferredRenderer::resize(int xsz, int ysz)
nuclear@41: {
nuclear@72: 	// keep a copy of the previous texture size because Renderer::resize may change it
nuclear@72: 
nuclear@72: 	int prev_tex_xsz = tex_xsz;
nuclear@72: 	int prev_tex_ysz = tex_ysz;
nuclear@72: 
nuclear@72: 	Renderer::resize(xsz, ysz);
nuclear@41: 
nuclear@41: 	// if we need a bigger rendertarget ...
nuclear@72: 	if(xsz > prev_tex_xsz || ysz > prev_tex_ysz) {
nuclear@41: 		// ... resize all the color buffers
nuclear@41: 		for(int i=0; i<MRT_COUNT; i++) {
nuclear@41: 			glBindTexture(GL_TEXTURE_2D, mrt_tex[i]);
nuclear@41: 			glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA32F, tex_xsz, tex_ysz, 0, GL_RGBA, GL_FLOAT, 0);
nuclear@41: 		}
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	// update the texture coordinate scaling factors
nuclear@41: 	float tex_scale_x = (float)width / tex_xsz;
nuclear@41: 	float tex_scale_y = (float)height / tex_ysz;
nuclear@41: 
nuclear@41: 	set_uniform_float2(deferred_omni, "tex_scale", tex_scale_x, tex_scale_y);
nuclear@41: 	set_uniform_float2(deferred_omni, "fb_size", width, height);
nuclear@41: }
nuclear@41: 
nuclear@41: void DeferredRenderer::render(const Level *level) const
nuclear@41: {
nuclear@62: 	render_pre(level);
nuclear@62: 
nuclear@41: 	// render into the MRT buffers
nuclear@41: 	glUseProgram(mrt_prog);
nuclear@41: 	curr_prog = mrt_prog;
nuclear@41: 
nuclear@41: 	glBindFramebufferEXT(GL_FRAMEBUFFER, fbo);
nuclear@65: 	attach_color_buffer(fbo, num_draw_bufs, mrt_tex);
nuclear@41: 
nuclear@41: 	glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@41: 	level->draw();
nuclear@65: 
nuclear@65: 	attach_color_buffer(fbo, 1, &rend_tex);
nuclear@65: 	glClear(GL_COLOR_BUFFER_BIT);
nuclear@41: 
nuclear@41: 	// post-process lighting
nuclear@60: 	light_pass(level);
nuclear@60: 
nuclear@60: 	glUseProgram(0);
nuclear@60: 	curr_prog = 0;
nuclear@62: 
nuclear@62: 	render_post(level);
nuclear@65: 
nuclear@65: 	glBindFramebufferEXT(GL_FRAMEBUFFER, 0);
nuclear@60: }
nuclear@60: 
nuclear@60: void DeferredRenderer::light_pass(const Level *level) const
nuclear@60: {
nuclear@65: 	glPushAttrib(GL_ENABLE_BIT | GL_POLYGON_BIT | GL_TEXTURE_BIT);
nuclear@41: 
nuclear@41: 	glEnable(GL_BLEND);
nuclear@41: 	glBlendFunc(GL_ONE, GL_ONE);
nuclear@41: 
nuclear@41: 	glDisable(GL_LIGHTING);
nuclear@41: 	glDisable(GL_DEPTH_TEST);
nuclear@41: 	glCullFace(GL_FRONT);
nuclear@41: 
nuclear@41: 	glUseProgram(deferred_omni);
nuclear@41: 	curr_prog = deferred_omni;
nuclear@41: 
nuclear@41: 	for(int i=0; i<MRT_COUNT; i++) {
nuclear@41: 		glActiveTexture(GL_TEXTURE0 + i);
nuclear@41: 		glBindTexture(GL_TEXTURE_2D, mrt_tex[i]);
nuclear@41: 		glEnable(GL_TEXTURE_2D);
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	glDepthMask(0);
nuclear@41: 	level->draw_lights();
nuclear@41: 	glDepthMask(1);
nuclear@41: 
nuclear@41: 	for(int i=0; i<MRT_COUNT; i++) {
nuclear@41: 		glActiveTexture(GL_TEXTURE0 + MRT_COUNT - i - 1);
nuclear@41: 		glDisable(GL_TEXTURE_2D);
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	glPopAttrib();
nuclear@41: }
nuclear@41: 
nuclear@41: static const char *fbstname[] = {
nuclear@41: 	"GL_FRAMEBUFFER_COMPLETE",
nuclear@41: 	"GL_FRAMEBUFFER_INCOMPLETE_ATTACHMENT",
nuclear@41: 	"GL_FRAMEBUFFER_INCOMPLETE_MISSING_ATTACHMENT",
nuclear@41: 	"no such fbo error",
nuclear@41: 	"GL_FRAMEBUFFER_INCOMPLETE_DIMENSIONS",
nuclear@41: 	"GL_FRAMEBUFFER_INCOMPLETE_FORMATS",
nuclear@41: 	"GL_FRAMEBUFFER_INCOMPLETE_DRAW_BUFFER",
nuclear@41: 	"GL_FRAMEBUFFER_INCOMPLETE_READ_BUFFER",
nuclear@41: 	"GL_FRAMEBUFFER_UNSUPPORTED"
nuclear@41: };
nuclear@41: 
nuclear@72: bool DeferredRenderer::create_rtarg()
nuclear@41: {
nuclear@72: 	if(!Renderer::create_rtarg()) {
nuclear@72: 		return false;
nuclear@72: 	}
nuclear@72: 
nuclear@41: 	unsigned int clamp = GLEW_ARB_texture_border_clamp ? GL_CLAMP_TO_EDGE : GL_CLAMP;
nuclear@41: 
nuclear@41: 	glBindFramebufferEXT(GL_FRAMEBUFFER, fbo);
nuclear@41: 
nuclear@41: 	glGenTextures(MRT_COUNT, mrt_tex);
nuclear@41: 	for(int i=0; i<MRT_COUNT; i++) {
nuclear@41: 		glBindTexture(GL_TEXTURE_2D, mrt_tex[i]);
nuclear@41: 		glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, clamp);
nuclear@41: 		glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, clamp);
nuclear@41: 		glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
nuclear@41: 		glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
nuclear@41: 		glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA32F, tex_xsz, tex_ysz, 0, GL_RGBA, GL_FLOAT, 0);
nuclear@41: 
nuclear@41: 		// attach to fbo
nuclear@62: 		GLenum color_att = GL_COLOR_ATTACHMENT0 + (num_draw_bufs == MRT_COUNT ? i : 0);
nuclear@62: 		glFramebufferTexture2DEXT(GL_FRAMEBUFFER, color_att, GL_TEXTURE_2D, mrt_tex[i], 0);
nuclear@41: 		CHECKGLERR;
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	int fbst = glCheckFramebufferStatusEXT(GL_FRAMEBUFFER);
nuclear@41: 	if(fbst != GL_FRAMEBUFFER_COMPLETE) {
nuclear@41: 		fprintf(stderr, "incomplete fbo: %u (%s)\n", fbo, fbstname[fbst - GL_FRAMEBUFFER_COMPLETE]);
nuclear@41: 		return false;
nuclear@41: 	}
nuclear@41: 	CHECKGLERR;
nuclear@41: 
nuclear@41: 	glBindFramebufferEXT(GL_FRAMEBUFFER, 0);
nuclear@41: 	return true;
nuclear@41: }
nuclear@41: 
nuclear@41: static unsigned int load_sdr(const char *vfname, const char *pfname)
nuclear@41: {
nuclear@41: 	unsigned int prog;
nuclear@41: 
nuclear@63: 	std::string vsfile = datafile_path(vfname);
nuclear@63: 	std::string psfile = datafile_path(pfname);
nuclear@63: 
nuclear@63: 	const char *vs = vsfile.empty() ? 0 : vsfile.c_str();
nuclear@63: 	const char *ps = psfile.empty() ? 0 : psfile.c_str();
nuclear@63: 
nuclear@63: 	if(!(prog = create_program_load(vs, ps))) {
nuclear@63: 		fprintf(stderr, "failed to load shader program (%s, %s)\n", vs, ps);
nuclear@41: 		return 0;
nuclear@41: 	}
nuclear@41: 	return prog;
nuclear@41: }
nuclear@41: 
nuclear@65: static void attach_color_buffer(unsigned int fbo, int count, const unsigned int *tex)
nuclear@65: {
nuclear@65: 	glBindFramebufferEXT(GL_FRAMEBUFFER, fbo);
nuclear@65: 
nuclear@65: 	for(int i=0; i<count; i++) {
nuclear@65: 		GLenum color_att = GL_COLOR_ATTACHMENT0 + i;
nuclear@65: 		glFramebufferTexture2DEXT(GL_FRAMEBUFFER, color_att, GL_TEXTURE_2D, tex[i], 0);
nuclear@65: 		CHECKGLERR;
nuclear@65: 	}
nuclear@65: 
nuclear@65: 	static GLenum draw_bufs[] = {
nuclear@65: 		GL_COLOR_ATTACHMENT0_EXT, GL_COLOR_ATTACHMENT1_EXT,
nuclear@65: 		GL_COLOR_ATTACHMENT2_EXT, GL_COLOR_ATTACHMENT3_EXT,
nuclear@65: 		GL_COLOR_ATTACHMENT4_EXT, GL_COLOR_ATTACHMENT5_EXT,
nuclear@65: 		GL_COLOR_ATTACHMENT6_EXT, GL_COLOR_ATTACHMENT7_EXT
nuclear@65: 	};
nuclear@65: 	glDrawBuffersARB(count, draw_bufs);
nuclear@65: }
nuclear@65: 
nuclear@41: #ifdef DBG_VIS_MRT
nuclear@41: // visualize the MRT buffers
nuclear@41: static void draw_deferred_debug()
nuclear@41: {
nuclear@41: 	glPushAttrib(GL_ENABLE_BIT | GL_POLYGON_BIT);
nuclear@41: 	glUseProgram(deferred_debug);
nuclear@41: 	glDisable(GL_DEPTH_TEST);
nuclear@41: 
nuclear@41: 	glMatrixMode(GL_PROJECTION);
nuclear@41: 	glPushMatrix();
nuclear@41: 	glLoadIdentity();
nuclear@41: 	glMatrixMode(GL_MODELVIEW);
nuclear@41: 	glPushMatrix();
nuclear@41: 	glLoadIdentity();
nuclear@41: 
nuclear@41: 	for(int i=0; i<MRT_COUNT; i++) {
nuclear@41: 		glActiveTexture(GL_TEXTURE0 + i);
nuclear@41: 		glBindTexture(GL_TEXTURE_2D, mrt_tex[i]);
nuclear@41: 		glEnable(GL_TEXTURE_2D);
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	glBegin(GL_QUADS);
nuclear@41: 	glTexCoord2f(0, 0);
nuclear@41: 	glVertex2f(-1, -1);
nuclear@41: 	glTexCoord2f(1, 0);
nuclear@41: 	glVertex2f(1, -1);
nuclear@41: 	glTexCoord2f(1, 1);
nuclear@41: 	glVertex2f(1, 1);
nuclear@41: 	glTexCoord2f(0, 1);
nuclear@41: 	glVertex2f(-1, 1);
nuclear@41: 	glEnd();
nuclear@41: 
nuclear@41: 	for(int i=0; i<MRT_COUNT; i++) {
nuclear@41: 		glActiveTexture(GL_TEXTURE0 + MRT_COUNT - i - 1);
nuclear@41: 		glDisable(GL_TEXTURE_2D);
nuclear@41: 	}
nuclear@41: 
nuclear@41: 	glMatrixMode(GL_PROJECTION);
nuclear@41: 	glPopMatrix();
nuclear@41: 	glMatrixMode(GL_MODELVIEW);
nuclear@41: 	glPopMatrix();
nuclear@41: 
nuclear@41: 	glUseProgram(0);
nuclear@41: 	glPopAttrib();
nuclear@41: }
nuclear@41: #endif