source: GTP/trunk/Lib/Vis/Preprocessing/src/GlobalLinesRenderer.cpp @ 1950

#include "glInterface.h"
#include "GlobalLinesRenderer.h"
#include "common.h"
#include "RenderTexture.h"
#include "Preprocessor.h"
#include "GlRenderer.h"

#include <Cg/cg.h>
#include <Cg/cgGL.h>

//#include <QtOpenGL>

namespace GtpVisibilityPreprocessor {


static CGcontext sCgContext = NULL;
static CGprogram sCgDepthPeelingProgram = NULL;
static CGprogram sCgPassThroughProgram = NULL;

static CGprofile sCgFragmentProfile;
static CGparameter sTextureParam;

GlobalLinesRenderer *globalLinesRenderer = NULL;

static int texWidth = 256;
static int texHeight = 256;

static void cgErrorCallback()
{
        CGerror lastError = cgGetError();

        if(lastError)
        {
                printf("%s\n\n", cgGetErrorString(lastError));
                printf("%s\n", cgGetLastListing(sCgContext));
                printf("Cg error, exiting...\n");

                exit(0);
        }
}

static void PrintGLerror(char *msg)
{
        GLenum errCode;
        const GLubyte *errStr;
        
        if ((errCode = glGetError()) != GL_NO_ERROR) 
        {
                errStr = gluErrorString(errCode);
                fprintf(stderr,"OpenGL ERROR: %s: %s\n", errStr, msg);
        }
}


void Reshape(int w, int h)
{
        if (h == 0) h = 1;

        glViewport(0, 0, w, h);

        glMatrixMode(GL_PROJECTION);
        glLoadIdentity();

        //gluPerspective(60.0, (GLfloat)w/(GLfloat)h, 3, 5000.0);
        //gluPerspective(60.0, (GLfloat)w/(GLfloat)h, 0.5, 10.0);
        glOrtho(-1000, 1000, -1000, 1000, 1, 1000);
}


void Display()
{
        glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
        glMatrixMode(GL_MODELVIEW);
        glPushMatrix();
        
        //globalLinesRenderer->DrawGeometry();
        
        globalLinesRenderer->CastGlobalLines(Beam(), 0);

        PrintGLerror("display");

        glPopMatrix();
        glutSwapBuffers();
}


void Idle()
{
        glutPostRedisplay();
}


void Keyboard(unsigned char key, int x, int y)
{
        switch(key)
        {
        case '2':
                ++ globalLinesRenderer->mMaxDepth;
                cout << "max depth: " << globalLinesRenderer->mMaxDepth << endl;
                return;
        case '1':
                -- globalLinesRenderer->mMaxDepth;
                cout << "max depth: " << globalLinesRenderer->mMaxDepth << endl;
                return;
        default:
                return;
        }
}

/*GlobalLinesRenderer::GlobalLinesRenderer(RenderTexture *buffer1, 
                                                                                 RenderTexture *buffer2, 
                                                                                 Preprocessor *preprocessor,
                                                                                 GlRenderer *renderer)
: mNewBuffer(buffer1), mOldBuffer(buffer2), mPreprocessor(preprocessor), mMaxDepth(100),
mRenderer(renderer)
{
}
*/
GlobalLinesRenderer::GlobalLinesRenderer(Preprocessor *preprocessor, 
                                                                                 GlRenderer *renderer): 
mNewBuffer(NULL), 
mOldBuffer(NULL), 
mMaxDepth(0), 
mRenderer(renderer), 
mPreprocessor(preprocessor)
{
}


GlobalLinesRenderer::~GlobalLinesRenderer()
{
        if (sCgDepthPeelingProgram)
                cgDestroyProgram(sCgDepthPeelingProgram);
        if (sCgContext)
                cgDestroyContext(sCgContext);
}

 
void GlobalLinesRenderer::CastGlobalLines(Beam &beam, const int samples)
{
        // bind pixel shader implementing the front depth buffer functionality
        ApplyDepthPeeling(beam, samples);
}


void GlobalLinesRenderer::RenderObject(Intersectable *obj)
{
        mRenderer->RenderIntersectable(obj);
}


void GlobalLinesRenderer::DrawGeometry()
{ 
        //mRenderer->mUseFalseColors = true;
        ObjectContainer::const_iterator oit, oit_end = mPreprocessor->mObjects.end();

        Intersectable::NewMail();

        for (oit = mPreprocessor->mObjects.begin(); oit != oit_end; ++ oit)
        {
                RenderObject(*oit);
        }
}


void GlobalLinesRenderer::SwitchRenderTextures()
{
        RenderTexture *buffer = mOldBuffer;
        mOldBuffer = mNewBuffer;
        mNewBuffer = buffer;
}


void GlobalLinesRenderer::InitGl()
{
        glutInitDisplayMode(GLUT_RGB | GLUT_DEPTH | GLUT_DOUBLE);
        glutInitWindowPosition(50, 50);
        glutInitWindowSize(512, 512);
        glutCreateWindow("TestRenderDepthTexture");  

        int err = glewInit();
        if (GLEW_OK != err)
        {
                // problem: glewInit failed, something is seriously wrong
                fprintf(stderr, "GLEW Error: %s\n", glewGetErrorString(err));
                exit(-1);
        }  
        
        glutKeyboardFunc(Keyboard);
        glutDisplayFunc(Display);
        glutIdleFunc(Idle);
        glutReshapeFunc(Reshape);

        Reshape(512, 512);
        glMatrixMode(GL_MODELVIEW);
        glLoadIdentity();

        AxisAlignedBox3 bbox = globalLinesRenderer->mPreprocessor->mKdTree->GetBox();
        const Vector3 midPoint = bbox.Center();
        //const Vector3 viewPoint = midPoint + Vector3(-800, 0, 0);
        const Vector3 viewPoint = midPoint + Vector3(2, 0, 0);

        cout << "midPoint: " << midPoint << endl;
        cout << "viewPoint: " << viewPoint << endl;
        cout << "scene: " << bbox << endl;

        /*gluLookAt(viewPoint.x, viewPoint.y, viewPoint.z, 
                          midPoint.x, midPoint.y, midPoint.z, 
                          0, 1, 0);
*/
        gluLookAt(0, 0, 3, 0, 0, 0, 0, 1, 0);

        //glDisable(GL_CULL_FACE);
        glEnable(GL_CULL_FACE);
        glDisable(GL_LIGHTING);
        //glEnable(GL_COLOR_MATERIAL);
        glDisable(GL_COLOR_MATERIAL);
        glEnable(GL_DEPTH_TEST); 
        glClearColor(0.1, 0.2, 0.3, 1);
        
        // A square, mipmapped, anisotropically filtered 8-bit RGBA texture with
        // depth and stencil.
        // Note that RT_COPY_TO_TEXTURE is required for depth textures on ATI hardware
        
        mNewBuffer = new RenderTexture(texWidth, texHeight, true, true);
        //mNewBuffer->Initialize(true, true, false, true, true, 8, 8, 8, 8);//, RenderTexture::RT_COPY_TO_TEXTURE);
        mNewBuffer->Initialize(true, true, false, true, true, 8, 8, 8, 8, RenderTexture::RT_COPY_TO_TEXTURE);
        
        mOldBuffer = new RenderTexture(texWidth, texHeight, true, true);
        //mOldBuffer ->Initialize(true, true, false, true, true, 8, 8, 8, 8);//, RenderTexture::RT_COPY_TO_TEXTURE);
        mOldBuffer ->Initialize(true, true, false, true, true, 8, 8, 8, 8, RenderTexture::RT_COPY_TO_TEXTURE);

        // Setup Cg
        cgSetErrorCallback(cgErrorCallback);

        // Create cgContext.
        sCgContext = cgCreateContext();

        // get the best profile for this hardware
        sCgFragmentProfile = cgGLGetLatestProfile(CG_GL_FRAGMENT);
        
        //assert(sCgFragmentProfile != CG_PROFILE_UNKNOWN);
        cgGLSetOptimalOptions(sCgFragmentProfile);

        sCgDepthPeelingProgram = 
                cgCreateProgramFromFile(sCgContext, 
                                                                CG_SOURCE,
                                                                "../src/depth_peeling.cg", 
                                                                GLEW_ARB_fragment_program ? CG_PROFILE_ARBFP1 : CG_PROFILE_FP30,
                                                                NULL,
                                                                NULL);

        if(sCgDepthPeelingProgram != NULL)
        {
                cgGLLoadProgram(sCgDepthPeelingProgram);
                sTextureParam = cgGetNamedParameter(sCgDepthPeelingProgram, "depthTex");  
        }

        sCgPassThroughProgram = 
                cgCreateProgramFromFile(sCgContext, 
                                                                CG_SOURCE,
                                                                "../src/passthrough.cg", 
                                                                GLEW_ARB_fragment_program ? CG_PROFILE_ARBFP1 : CG_PROFILE_FP30,
                                                                NULL,
                                                                NULL);

        if(sCgPassThroughProgram != NULL)
        {
                cgGLLoadProgram(sCgPassThroughProgram);
        }


    // setup the rendering context for the RenderTexture
        mNewBuffer->BeginCapture();
        {
                Reshape(texWidth, texHeight);
                glClearColor(0.1, 0.7, 0.2, 1);
                glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);

                glFrontFace(GL_CCW);
                glCullFace(GL_BACK);
                glEnable(GL_CULL_FACE);
                glShadeModel(GL_FLAT);
                glEnable(GL_DEPTH_TEST);
                
                glMatrixMode(GL_MODELVIEW);
                glLoadIdentity();
                gluLookAt(viewPoint.x, viewPoint.y, viewPoint.z, 
                                  midPoint.x, midPoint.y, midPoint.z, 
                                  0, 1, 0);

        }
        mNewBuffer->EndCapture();

        // setup the rendering context for the RenderTexture
        mOldBuffer->BeginCapture();
        {
                Reshape(texWidth, texHeight);
                glMatrixMode(GL_MODELVIEW);
                glLoadIdentity();
                
                glFrontFace(GL_CCW);
                glCullFace(GL_BACK);
                glEnable(GL_CULL_FACE);
                glShadeModel(GL_FLAT);
                glEnable(GL_DEPTH_TEST);
                
                gluLookAt(viewPoint.x, viewPoint.y, viewPoint.z, 
                                  midPoint.x, midPoint.y, midPoint.z, 
                                  0, 1, 0);

                glClearColor(0.6, 0.2, 0.2, 1);
        }
        mOldBuffer->EndCapture();

        PrintGLerror("init");
}


void GlobalLinesRenderer::ProcessDepthBuffer()
{
}


void GlobalLinesRenderer::Run()
{
        glutMainLoop();
}


void GlobalLinesRenderer::ApplyDepthPeeling(Beam &beam, const int samples)
{
        //if (mNewBuffer->IsDoubleBuffered()) 
        //      glDrawBuffer(GL_BACK);

        mNewBuffer->BeginCapture();
        {
                //cgGLBindProgram(sCgPassThroughProgram);
                //cgGLEnableProfile(sCgFragmentProfile);
                glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
                glMatrixMode(GL_MODELVIEW);
                glPushMatrix();
                { 
                        //glutSolidTorus(0.25, 1, 32, 64);
                        DrawGeometry();
                }
                glPopMatrix();
        }
        mNewBuffer->EndCapture();
        
        //mNewBuffer->BindDepth();
        float *data = new float[texWidth * texHeight];
        //const int texFormat = WGL_TEXTURE_DEPTH_COMPONENT_NV;
        const int texFormat = GL_LUMINANCE;
        glGetTexImage(mNewBuffer->GetTextureTarget(), 0, texFormat, GL_FLOAT, data);
        
        delete [] data;

        PrintGLerror("get data");

        if (mNewBuffer->IsRectangleTexture()) cout << "rect" << endl;
//else cout << "haga"<<endl;

        for(int i = 0; i < mMaxDepth; ++ i) 
        {
                // Peel another layer
                SwitchRenderTextures(); // switch pointer between rendertextures

                mNewBuffer->BeginCapture();
                {
                        //if (mNewBuffer->IsDoubleBuffered()) 
                        //      glDrawBuffer(GL_BACK);

                        glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);         

                        cgGLBindProgram(sCgDepthPeelingProgram);
                //      cgGLBindProgram(sCgPassThroughProgram);
                        cgGLEnableProfile(sCgFragmentProfile);
                        cgGLSetTextureParameter(sTextureParam, mOldBuffer->GetDepthTextureID());
                        cgGLEnableTextureParameter(sTextureParam);

                        glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
                        glMatrixMode(GL_MODELVIEW);
                        glPushMatrix();
                        { 
                                //glutSolidTorus(0.25, 1, 32, 64);
                                DrawGeometry();
                        }
                        glPopMatrix();

                        cgGLDisableTextureParameter(sTextureParam);
                        cgGLDisableProfile(sCgFragmentProfile);
                }
                mNewBuffer->EndCapture();
        }

        PrintGLerror("display 1");

        //mNewBuffer->Bind();
        mNewBuffer->BindDepth();
        mNewBuffer->EnableTextureTarget();


        PrintGLerror("display 2");
        glColor3f(1,1,1);
        glBegin(GL_QUADS);
    glTexCoord2f(0, 0); glVertex3f(-1000, -1000, -0.5f);
    glTexCoord2f(1, 0); glVertex3f( 1000, -1000, -0.5f);
    glTexCoord2f(1, 1); glVertex3f( 1000,  1000, -0.5f);
    glTexCoord2f(0, 1); glVertex3f(-1000,  1000, -0.5f);
    glEnd();

    if (mNewBuffer->IsFloatTexture())
        {
                // cgGLDisableTextureParameter(textureParam);
                // cgGLDisableProfile(theProfile);
        }
        else 
        {
                mNewBuffer->DisableTextureTarget();
        }
}


}