source: GTP/trunk/App/Demos/Vis/CHC_revisited/RenderTraverser.cpp @ 2761

#include "RenderTraverser.h"
#include "glInterface.h"
#include "Timers.h"


namespace CHCDemo
{

RenderTraverser::RenderTraverser(): 
mFrameID(1), 
mVisibilityThreshold(0),
mBvh(NULL), 
mIsQueryMode(false),
mUseOptimization(true),
mCurrentQueryIdx(0)
{
}


RenderTraverser::~RenderTraverser()
{
        //DelQueries();
}


/*void RenderTraverser::Render(int mode)
{
        mDistanceQueue.push(mHierarchyRoot);

        long startTime = getTime();

        Preprocess();

        switch(mode)
        {
                case RENDER_CULL_FRUSTUM:
                        RenderCullFrustum();
            break;
                case RENDER_STOP_AND_WAIT:
                        RenderStopAndWait();
                        break;
                case RENDER_COHERENT:
                        RenderCoherentWithQueue();
                        break;
                default:
                        RenderCullFrustum();
                        break;
        }

        mFrameID ++;
        

        long endTime = getTime();
        mRenderTime = endTime - startTime;

        finishTiming();
}*/

void RenderTraverser::EnqueueNode(BvhNode *node)
{
        mBvh->CalcDistance(node);
        mDistanceQueue.push(node);
}


void RenderTraverser::TraverseNode(BvhNode *node)
{
        if (node->IsVirtualLeaf()||node->IsLeaf())
        {
                //mNumRenderedGeometry +=
                mBvh->Render(node, mRenderState);
        }
        else 
        {
                // for non leafs this renders only the bounding volume (if the flag is set)
                //node->Render();
                BvhInterior *interior = static_cast<BvhInterior *>(node);

                EnqueueNode(interior->GetFront());
                EnqueueNode(interior->GetBack());
        }
}


void RenderTraverser::SetHierarchy(Bvh *bvh)
{
        mBvh = bvh;
        //DelQueries();
}


void RenderTraverser::SetRenderState(RenderState *state)
{
        mRenderState = state;
        //DelQueries();
}


#if 0

Bvh *RenderTraverser::GetHierarchy()
{
        return mBvh;
}


void RenderTraverser::RenderVisualization()
{
        mDistanceQueue.push(mHierarchyRoot);

        while(! mDistanceQueue.empty())
        {
                HierarchyNode *node = mDistanceQueue.top();
                mDistanceQueue.pop();

                // identify previously visible nodes
                bool wasVisible = node->Visible() && (node->LastVisited() == mFrameID - 1);

                if(wasVisible)
                        TraverseNode(node);
                else
                {
                        // also render culled nodes
                        glColor3f(1.0,0.0,0.0);
                        node->RenderBoundingVolumeForVisualization();           
                }
        }
}


void RenderTraverser::PullUpVisibility(HierarchyNode *node)
{
        while(node && !node->Visible())
        {
                node->SetVisible(true);
                node = node->GetParent();
        }
}

bool RenderTraverser::ResultAvailable(HierarchyNode *node) const
{
        unsigned int result;
        if (mUseArbQueries)
        {
                glGetQueryObjectuivARB(node->GetOcclusionQuery(),
                                                           GL_QUERY_RESULT_AVAILABLE_ARB, &result);
        }
        else
        {
                
                glGetOcclusionQueryuivNV(node->GetOcclusionQuery(), 
                                                                 GL_PIXEL_COUNT_AVAILABLE_NV, &result);

        }
        
        return (result == GL_TRUE);
}


unsigned int RenderTraverser::GetOcclusionQueryResult(HierarchyNode *node) const
{
        unsigned int result;

        if (mUseArbQueries)
        {
                glGetQueryObjectuivARB(node->GetOcclusionQuery(), GL_QUERY_RESULT_ARB, &result);
        }
        else
        {
                glGetOcclusionQueryuivNV(node->GetOcclusionQuery(), GL_PIXEL_COUNT_NV, &result);
        }

        return result;
}


void RenderTraverser::Switch2GLQueryState()
{       
        // boolean used to avoid unnecessary state changes
        if(!mIsQueryMode)
        {
                glColorMask(GL_FALSE, GL_FALSE, GL_FALSE, GL_FALSE);
                glDepthMask(GL_FALSE);
                glDisable(GL_LIGHTING);
                mIsQueryMode = true;
        }
}


void RenderTraverser::Switch2GLRenderState()
{
        // boolean used to avoid unnecessary state changes
        if(mIsQueryMode)
        {
                // switch back to rendermode            
                glColorMask(GL_TRUE, GL_TRUE, GL_TRUE, GL_TRUE);
                glDepthMask(GL_TRUE);
                glEnable(GL_LIGHTING);
                mIsQueryMode = false;
        }
}

void RenderTraverser::IssueOcclusionQuery(HierarchyNode *node, const bool wasVisible)
{
        // get next available test id
        unsigned int occlusionQuery = mOcclusionQueries[mCurrentQueryIdx ++];
        
        node->SetOcclusionQuery(occlusionQuery);

        // do the actual occlusion query for this node
        if (mUseArbQueries)
        {
                glBeginQueryARB(GL_SAMPLES_PASSED_ARB, occlusionQuery);
        }
        else
        {
                glBeginOcclusionQueryNV(occlusionQuery);
        }
        
        // if leaf and was visible => will be rendered anyway, thus we
        // can also test with the real geometry 
        if(node->IsLeaf() && wasVisible && mUseOptimization)
        {
                mNumRenderedGeometry += node->Render();
        }
        else
        {
                // change state so the bounding box gets not actually rendered on the screen
                Switch2GLQueryState();
                node->RenderBoundingVolume();
                Switch2GLRenderState();
        }

        if (mUseArbQueries)
        {
                glEndQueryARB(GL_SAMPLES_PASSED_ARB);
        }
        else
        {
                glEndOcclusionQueryNV();
        }
}

void RenderTraverser::Preprocess()
{
        if (!mOcclusionQueries)
        {
                mOcclusionQueries = new unsigned int[mHierarchyRoot->GetNumHierarchyNodes()];

                // generate ids for occlusion test
                if (mUseArbQueries)
                {
                        glGenQueriesARB(mHierarchyRoot->GetNumHierarchyNodes(), mOcclusionQueries);
                }
                else
                {
                        glGenOcclusionQueriesNV(mHierarchyRoot->GetNumHierarchyNodes(), mOcclusionQueries);
                }
        }

        // view frustum planes for view frustum culling
        calcViewFrustumPlanes(&mClipPlanes, mProjViewMatrix);
        calcAABNPVertexIndices(mNPVertexIndices, mClipPlanes);

        mCurrentQueryIdx = 0;

        // reset statistics
        mNumTraversedNodes = 0;
        mNumQueryCulledNodes = 0;
        mNumFrustumCulledNodes = 0;
        mNumRenderedGeometry = 0;
}


void RenderTraverser::DelQueries()
{
        if (!mOcclusionQueries)
                return;

        // tell the driver that the occlusion queries won't be needed any more
        if (mUseArbQueries)
        {
                glDeleteQueriesARB(mHierarchyRoot->GetNumHierarchyNodes(), mOcclusionQueries);
        }
        else
        {
                glDeleteOcclusionQueriesNV(mHierarchyRoot->GetNumHierarchyNodes(), mOcclusionQueries);
        }

        delete [] mOcclusionQueries;

        mOcclusionQueries = NULL;
}


void RenderTraverser::SetViewpoint(Vector3 const &viewpoint)
{
        copyVector3(mViewpoint, viewpoint);
}
        

void RenderTraverser::SetProjViewMatrix(Matrix4x4 const &projViewMatrix)
{
        copyMatrix(mProjViewMatrix, projViewMatrix);
}


bool RenderTraverser::InsideViewFrustum(HierarchyNode *node, bool &intersects)
{
        Vector3x8 vertices;
        
        bool intersect = false;

        calcAABoxPoints(vertices, node->GetBoundingVolume());

        // simply test all 6 vertices
        for (int i = 0; i < 6; i++)
        {               
                // test the n-vertex
                // note: the calcAABNearestVertexId should be preprocessed
                if(!pointBeforePlane(mClipPlanes.plane[i], vertices[mNPVertexIndices[i * 2]]))
                {
                        // outside
                        return false;
                }
        }

        // test if bounding box is intersected by nearplane (using the p-vertex)
        intersects = (!pointBeforePlane(mClipPlanes.plane[5], vertices[mNPVertexIndices[11]]));

        // -- get vector from viewpoint to center of bounding volume
        Vector3 vec;
        calcAABoxCenter(vec, node->GetBoundingVolume());
        diffVector3(vec, vec, mViewpoint);

        // compute distance from nearest point to viewpoint
        diffVector3(vec, vertices[calcAABNearestVertexIdx(vec)], mViewpoint);
        node->SetDistance(squaredLength(vec));
        
        return true;
}


void RenderTraverser::SetVisibilityThreshold(int threshold)
{
        mVisibilityThreshold = threshold;
}

void RenderTraverser::SetUseArbQueries(const bool useArbQueries)
{
        DelQueries();
        mUseArbQueries = useArbQueries;
}

bool RenderTraverser::GetUseArbQueries() const
{
        return mUseArbQueries;
}

long RenderTraverser::GetRenderTime()
{
        return mRenderTime;
}

int RenderTraverser::GetNumTraversedNodes()
{
        return mNumTraversedNodes;
}

int RenderTraverser::GetNumQueryCulledNodes()
{
        return mNumQueryCulledNodes;
}

int RenderTraverser::GetNumFrustumCulledNodes()
{
        return mNumFrustumCulledNodes;
}


int RenderTraverser::GetNumRenderedGeometry()
{
        return mNumRenderedGeometry;
}


#endif


void RenderTraverser::SetUseOptimization(bool useOptimization)
{
        mUseOptimization = useOptimization;
        printf("using opt %d\n", mUseOptimization);
}

}