anki-3d-engine/src/Renderer/Drawers/SceneDrawer.cpp

#include "SceneDrawer.h"
#include "Math.h"
#include "Material.h"
#include "RenderableNode.h"
#include "Camera.h"
#include "Renderer.h"
#include "App.h"
#include "Scene.h"
#include "MaterialRuntime.h"
#include "GlStateMachine.h"


//======================================================================================================================
// Constructor                                                                                                         =
//======================================================================================================================
SceneDrawer::UsrDefVarVisitor::UsrDefVarVisitor(const MaterialRuntimeUserDefinedVar& udvr_,
                                                const Renderer& r_, uint& texUnit_):
	udvr(udvr_),
	r(r_),
	texUnit(texUnit_)
{}


//======================================================================================================================
// Visitor functors                                                                                                    =
//======================================================================================================================


template<typename Type>
void SceneDrawer::UsrDefVarVisitor::operator()(const Type& x) const
{
	udvr.getUniVar().set(&x);
}


void SceneDrawer::UsrDefVarVisitor::operator()(const RsrcPtr<Texture>* x) const
{
	const RsrcPtr<Texture>& texPtr = *x;
	texPtr->setRepeat(true);
	udvr.getUniVar().set(*texPtr, texUnit);
	++texUnit;
}


template<>
void SceneDrawer::UsrDefVarVisitor::operator()(const MtlUserDefinedVar::Fai& x) const
{
	switch(x)
	{
		case MtlUserDefinedVar::MS_DEPTH_FAI:
			udvr.getUniVar().set(r.getMs().getDepthFai(), texUnit);
			break;
		case MtlUserDefinedVar::IS_FAI:
			udvr.getUniVar().set(r.getIs().getFai(), texUnit);
			break;
		case MtlUserDefinedVar::PPS_PRE_PASS_FAI:
			udvr.getUniVar().set(r.getPps().getPrePassFai(), texUnit);
			break;
		case MtlUserDefinedVar::PPS_POST_PASS_FAI:
			udvr.getUniVar().set(r.getPps().getPostPassFai(), texUnit);
			break;
		default:
			ASSERT(0);
	}
	++texUnit;
}



//======================================================================================================================
// setupShaderProg                                                                                                     =
//======================================================================================================================
void SceneDrawer::setupShaderProg(const MaterialRuntime& mtlr, const Transform& nodeWorldTransform, const Camera& cam,
                                  const Renderer& r, float blurring)
{
	uint textureUnit = 0;

	mtlr.getMaterial().getShaderProg().bind();

	//
	// FFP stuff
	//
	GlStateMachineSingleton::getInstance().enable(GL_BLEND, mtlr.isBlendingEnabled());
	if(mtlr.isBlendingEnabled())
	{
		glBlendFunc(mtlr.getBlendingSfactor(), mtlr.getBlendingDfactor());
	}

	GlStateMachineSingleton::getInstance().enable(GL_DEPTH_TEST, mtlr.isDepthTestingEnabled());

	if(mtlr.isWireframeEnabled())
	{
		glPolygonMode(GL_FRONT_AND_BACK, GL_LINE);
	}
	else
	{
		glPolygonMode(GL_FRONT_AND_BACK, GL_FILL);
	}


	//
	// calc needed matrices
	//
	Mat4 modelMat(nodeWorldTransform);
	const Mat4& projectionMat = cam.getProjectionMatrix();
	const Mat4& viewMat = cam.getViewMatrix();
	Mat4 modelViewMat;
	Mat3 normalMat;
	Mat4 modelViewProjectionMat;

	// should I calculate the modelViewMat ?
	if(mtlr.getStdUniVar(Material::SUV_MODELVIEW_MAT) ||
	   mtlr.getStdUniVar(Material::SUV_MODELVIEWPROJECTION_MAT) ||
	   mtlr.getStdUniVar(Material::SUV_NORMAL_MAT))
	{
		// Optimization
		if(modelMat == Mat4::getIdentity())
		{
			modelViewMat = viewMat;
		}
		else
		{
			modelViewMat = Mat4::combineTransformations(viewMat, modelMat);
		}
	}

	// set matrices
	if(mtlr.getStdUniVar(Material::SUV_MODEL_MAT))
	{
		mtlr.getStdUniVar(Material::SUV_MODEL_MAT)->set(&modelMat);
	}

	if(mtlr.getStdUniVar(Material::SUV_VIEW_MAT))
	{
		mtlr.getStdUniVar(Material::SUV_VIEW_MAT)->set(&viewMat);
	}

	if(mtlr.getStdUniVar(Material::SUV_PROJECTION_MAT))
	{
		mtlr.getStdUniVar(Material::SUV_PROJECTION_MAT)->set(&projectionMat);
	}

	if(mtlr.getStdUniVar(Material::SUV_MODELVIEW_MAT))
	{
		mtlr.getStdUniVar(Material::SUV_MODELVIEW_MAT)->set(&modelViewMat);
	}

	if(mtlr.getStdUniVar(Material::SUV_VIEWPROJECTION_MAT))
	{
		mtlr.getStdUniVar(Material::SUV_VIEWPROJECTION_MAT)->set(&r.getViewProjectionMat());
	}

	if(mtlr.getStdUniVar(Material::SUV_NORMAL_MAT))
	{
		normalMat = modelViewMat.getRotationPart();
		mtlr.getStdUniVar(Material::SUV_NORMAL_MAT)->set(&normalMat);
	}

	if(mtlr.getStdUniVar(Material::SUV_MODELVIEWPROJECTION_MAT))
	{
		modelViewProjectionMat = projectionMat * modelViewMat;
		mtlr.getStdUniVar(Material::SUV_MODELVIEWPROJECTION_MAT)->set(&modelViewProjectionMat);
	}


	//
	// FAis
	//
	if(mtlr.getStdUniVar(Material::SUV_MS_NORMAL_FAI))
	{
		mtlr.getStdUniVar(Material::SUV_MS_NORMAL_FAI)->set(r.getMs().getNormalFai(), textureUnit++);
	}

	if(mtlr.getStdUniVar(Material::SUV_MS_DIFFUSE_FAI))
	{
		mtlr.getStdUniVar(Material::SUV_MS_DIFFUSE_FAI)->set(r.getMs().getDiffuseFai(), textureUnit++);
	}

	if(mtlr.getStdUniVar(Material::SUV_MS_SPECULAR_FAI))
	{
		mtlr.getStdUniVar(Material::SUV_MS_SPECULAR_FAI)->set(r.getMs().getSpecularFai(), textureUnit++);
	}

	if(mtlr.getStdUniVar(Material::SUV_MS_DEPTH_FAI))
	{
		mtlr.getStdUniVar(Material::SUV_MS_DEPTH_FAI)->set(r.getMs().getDepthFai(), textureUnit++);
	}

	if(mtlr.getStdUniVar(Material::SUV_IS_FAI))
	{
		mtlr.getStdUniVar(Material::SUV_IS_FAI)->set(r.getIs().getFai(), textureUnit++);
	}

	if(mtlr.getStdUniVar(Material::SUV_PPS_PRE_PASS_FAI))
	{
		mtlr.getStdUniVar(Material::SUV_PPS_PRE_PASS_FAI)->set(r.getPps().getPrePassFai(), textureUnit++);
	}

	if(mtlr.getStdUniVar(Material::SUV_PPS_POST_PASS_FAI))
	{
		mtlr.getStdUniVar(Material::SUV_PPS_POST_PASS_FAI)->set(r.getPps().getPostPassFai(), textureUnit++);
	}


	//
	// Other
	//
	if(mtlr.getStdUniVar(Material::SUV_RENDERER_SIZE))
	{
		Vec2 v(r.getWidth(), r.getHeight());
		mtlr.getStdUniVar(Material::SUV_RENDERER_SIZE)->set(&v);
	}

	if(mtlr.getStdUniVar(Material::SUV_SCENE_AMBIENT_COLOR))
	{
		Vec3 col(SceneSingleton::getInstance().getAmbientCol());
		mtlr.getStdUniVar(Material::SUV_SCENE_AMBIENT_COLOR)->set(&col);
	}

	if(mtlr.getStdUniVar(Material::SUV_BLURRING))
	{
		/*blurring *= 10.0;
		INFO(blurring);*/
		float b = blurring;
		mtlr.getStdUniVar(Material::SUV_BLURRING)->set(&b);
	}


	//
	// set user defined vars
	//
	BOOST_FOREACH(const MaterialRuntimeUserDefinedVar& udvr, mtlr.getUserDefinedVars())
	{
		boost::apply_visitor(UsrDefVarVisitor(udvr, r, textureUnit), udvr.getDataVariant());
	}

	ON_GL_FAIL_THROW_EXCEPTION();
}


//======================================================================================================================
// renderRenderableNode                                                                                                =
//======================================================================================================================
void SceneDrawer::renderRenderableNode(const RenderableNode& renderable, const Camera& cam,
                                       RenderingPassType rtype) const
{
	const MaterialRuntime* mtlr;
	const Vao* vao;
	float blurring = 0.0;

	switch(rtype)
	{
		case RPT_COLOR:
			mtlr = &renderable.getCpMtlRun();
			vao = &renderable.getCpVao();

			blurring = (renderable.getWorldTransform().getOrigin() -
			            renderable.getPrevWorldTransform().getOrigin()).getLength();

			break;

		case RPT_DEPTH:
			mtlr = &renderable.getDpMtlRun();
			vao = &renderable.getDpVao();
			break;
	}

	setupShaderProg(*mtlr, renderable.getWorldTransform(), cam, r, blurring);

	vao->bind();
	glDrawElements(GL_TRIANGLES, renderable.getVertIdsNum(), GL_UNSIGNED_SHORT, 0);
	vao->unbind();
}