source: GTP/trunk/App/Demos/Vis/FriendlyCulling/src/Visualization.cpp @ 2790

#include "Visualization.h"
#include "glInterface.h"
#include "Camera.h"
#include "SceneEntity.h"
#include "Bvh.h"
#include "RenderState.h"

#include <stack>


using namespace std;


namespace CHCDemoEngine
{

GLUquadric *mSphere;


static void RenderBoxForViz(const AxisAlignedBox3 &box)
{
        glBegin(GL_LINE_LOOP);
        glVertex3d(box.Min().x, box.Max().y, box.Min().z);
        glVertex3d(box.Max().x, box.Max().y, box.Min().z);
        glVertex3d(box.Max().x, box.Min().y, box.Min().z);
        glVertex3d(box.Min().x, box.Min().y, box.Min().z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(box.Min().x, box.Min().y, box.Max().z);
        glVertex3d(box.Max().x, box.Min().y, box.Max().z);
        glVertex3d(box.Max().x, box.Max().y, box.Max().z);
        glVertex3d(box.Min().x, box.Max().y, box.Max().z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(box.Max().x, box.Min().y, box.Min().z);
        glVertex3d(box.Max().x, box.Min().y, box.Max().z);
        glVertex3d(box.Max().x, box.Max().y, box.Max().z);
        glVertex3d(box.Max().x, box.Max().y, box.Min().z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(box.Min().x, box.Min().y, box.Min().z);
        glVertex3d(box.Min().x, box.Min().y, box.Max().z);
        glVertex3d(box.Min().x, box.Max().y, box.Max().z);
        glVertex3d(box.Min().x, box.Max().y, box.Min().z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(box.Min().x, box.Min().y, box.Min().z);
        glVertex3d(box.Max().x, box.Min().y, box.Min().z);
        glVertex3d(box.Max().x, box.Min().y, box.Max().z);
        glVertex3d(box.Min().x, box.Min().y, box.Max().z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(box.Min().x, box.Max().y, box.Min().z);
        glVertex3d(box.Max().x, box.Max().y, box.Min().z);
        glVertex3d(box.Max().x, box.Max().y, box.Max().z);
        glVertex3d(box.Min().x, box.Max().y, box.Max().z);

        glEnd();
}


/******************************************************/
/*           Vizualization implementation           */
/******************************************************/


Visualization::Visualization(Bvh *bvh, 
                                                         Camera *camera, 
                                                         Camera *vizCamera, 
                                                         RenderState *renderState): 
mBvh(bvh), 
mCamera(camera),
mVizCamera(vizCamera),
mRenderState(renderState),
mFrameId(0)
{
        mSphere = (GLUquadric *)gluNewQuadric();
}


void Visualization::SetHierarchy(Bvh *bvh)
{
        mBvh = bvh;
}


void Visualization::SetRenderState(RenderState *state)
{
        mRenderState = state;
}


void Visualization::SetFrameId(int frameId)
{
        mFrameId = frameId;
}


void Visualization::Render()
{
        stack<BvhNode *> tStack;
        tStack.push(mBvh->GetRoot());

        glDisable(GL_LIGHTING);
        
        RenderViewPoint();
        RenderFrustum();
        RenderBoxForViz(mBvh->GetBox());

        glEnable(GL_LIGHTING);

        glEnableClientState(GL_VERTEX_ARRAY);
        glEnableClientState(GL_NORMAL_ARRAY);

        while (!tStack.empty())
        {
                BvhNode *node = tStack.top();
                tStack.pop();

                if (!node->IsVirtualLeaf())
                {
                        BvhInterior *interior = static_cast<BvhInterior *>(node);

                        tStack.push(interior->GetFront());
                        tStack.push(interior->GetBack());
                }
                else
                {
                        if (node->GetLastRenderedFrame() == mFrameId)
                        {
                                int geometrySize;
                                SceneEntity **entities = mBvh->GetGeometry(node, geometrySize);

                                for (int i = 0; i < geometrySize; ++ i)
                                {
                                        SceneEntity *ent = entities[i];
                                        ent->Render(mRenderState);
                                }               
                        }
                        //leaves.push_back(static_cast<BvhLeaf *>(node));
                }
        }

        mRenderState->Reset();

        glDisableClientState(GL_VERTEX_ARRAY);
        glDisableClientState(GL_NORMAL_ARRAY);
}


void Visualization::RenderViewPoint()
{
        glPushMatrix();
        Vector3 pos = mCamera->GetPosition();
        pos.z += 100;
        glTranslatef(pos.x, pos.y, pos.z);
        
        glScalef(5.0f, 5.0f, 5.0f);
        glPushAttrib(GL_CURRENT_BIT);

        glColor3f(1.0f, 0.0f, 0.0f);
        
        gluSphere((::GLUquadric *)mSphere,
                2e-3f * Magnitude(mBvh->GetBox().Size()), 6, 6);

        glPopAttrib();
        glPopMatrix();
}


void Visualization::RenderFrustum()
{
        glColor3f(1.0f, 0.0f, 0.0f);

        Vector3 ftl, ftr, fbl, fbr, ntl, ntr, nbl, nbr;
        mCamera->ComputePoints(ftl, ftr, fbl, fbr, ntl, ntr, nbl, nbr);

        glLineWidth(2);

        glBegin(GL_LINE_LOOP);
        glVertex3d(fbl.x, fbl.y, fbl.z);
        glVertex3d(fbr.x, fbr.y, fbr.z);
        glVertex3d(ftr.x, ftr.y, ftr.z);
        glVertex3d(ftl.x, ftl.y, ftl.z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(nbl.x, nbl.y, nbl.z);
        glVertex3d(nbr.x, nbr.y, nbr.z);
        glVertex3d(ntr.x, ntr.y, ntr.z);
        glVertex3d(ntl.x, ntl.y, ntl.z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(fbl.x, fbl.y, fbl.z);
        glVertex3d(ftl.x, ftl.y, ftl.z);
        glVertex3d(ntl.x, ntl.y, ntl.z);
        glVertex3d(nbl.x, nbl.y, nbl.z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(fbr.x, fbr.y, fbr.z);
        glVertex3d(ftr.x, ftr.y, ftr.z);
        glVertex3d(ntr.x, ntr.y, ntr.z);
        glVertex3d(nbr.x, nbr.y, nbr.z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(fbr.x, fbr.y, fbr.z);
        glVertex3d(fbl.x, fbl.y, fbl.z);
        glVertex3d(nbl.x, nbl.y, nbl.z);
        glVertex3d(nbr.x, nbr.y, nbr.z);
        glEnd();

        glBegin(GL_LINE_LOOP);
        glVertex3d(ftr.x, ftr.y, ftr.z);
        glVertex3d(ftl.x, ftl.y, ftl.z);
        glVertex3d(ntl.x, ntl.y, ntl.z);
        glVertex3d(ntr.x, ntr.y, ntr.z);
        glEnd();
}


}