Changeset 2808 for GTP/trunk/App/Demos/Vis/FriendlyCulling/src/SceneQuery.cpp

Timestamp:

06/29/08 23:48:17 (16 years ago)

Author:

mattausch

Message:

 

File:

• GTP/trunk/App/Demos/Vis/FriendlyCulling/src/SceneQuery.cpp (modified) (4 diffs)

GTP/trunk/App/Demos/Vis/FriendlyCulling/src/SceneQuery.cpp

@@ -7 +7 @@
 #include <IL/il.h>
 #include <assert.h>
+#include "RenderTexture.h"
+
+
+using namespace std;
+
+int texWidth = 2048;
+int texHeight = 2048;
+
+/*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);
+        }
+}
 
 
@@ -26 +60 @@
 {
 
-using namespace std;
-
-
-//const static int viewport[4] = {0, 0, 512, 512};
-const static int viewport[4] = {0, 0, 1024, 768};
-//const static int viewport[4] = {0, 0, 2048, 2048};
-
-
-
-SceneQuery::SceneQuery(const AxisAlignedBox3 &sceneBox, RenderTraverser *renderer): 
-mSceneBox(sceneBox), mDepth(NULL)
-{
-        Prepare(renderer);
-}
-
-
-bool SceneQuery::CalcIntersection(Vector3 &pt)
-{
-        int px = (pt.x - mSceneBox.Min(0)) * (viewport[2] - 1) / mSceneBox.Size(0);
-        int py = (pt.y - mSceneBox.Min(1)) * (viewport[3] - 1) / mSceneBox.Size(1);
-
-        unsigned char d = mDepth[px + py * viewport[2]];
-
-        const float offs = mSceneBox.Size(2) * 1e-1f;
-
-        static float depth = (float)d;
-
-        if (d > 0)
-        {
-                const float x = 0.1f;
-                depth = depth * x + d * (1.0f - x);
-                pt.z =  mSceneBox.Max().z - mSceneBox.Size().z * depth / 255.0f + offs;
-                //cout << "new depth " << pt.z << " (" << d << ")" << endl;
-
-                return true;
-        }
-        //cout << "invalid depth: " << d << endl;
-
-        return false;
-}
-
-
-void SceneQuery::Prepare(RenderTraverser *renderer)
-{
-        cout << "Preparing scene queries" << endl;
-
-        const float xlen = mSceneBox.Size().x * 0.5f;
-        const float ylen = mSceneBox.Size().y * 0.5f;
-        
-        Camera *orthoCam = new Camera(xlen, ylen);
-        orthoCam->SetOrtho(true);
-
-        orthoCam->SetNear(0.0f);
-        orthoCam->Yaw(M_PI * 0.5f);
-
-        Vector3 pos = Vector3(mSceneBox.Center().x, mSceneBox.Center().y, mSceneBox.Max().z);
-        orthoCam->SetPosition(pos);
-
-        //glPixelStorei(GL_PACK_ROW_LENGTH, viewport[2]);
-        glPixelStorei(GL_PACK_ALIGNMENT, 1);
-        glReadBuffer(GL_BACK);
-
-        // hack: should create offscreen buffer for this
-        glViewport(0, 0, viewport[2], viewport[3]);
-
-        glMatrixMode(GL_PROJECTION);
-        glLoadIdentity();
-        
-        glOrtho(-xlen, xlen, -ylen, ylen, 0.0f, mSceneBox.Size().z); 
-        
-        glMatrixMode(GL_MODELVIEW);
-        
-        orthoCam->SetupCameraView();
-
-        glClear(GL_DEPTH_BUFFER_BIT);
-
-        mDepth = new unsigned char[viewport[2] * viewport[3]];
-        //mDepth = new float[viewport[2] * viewport[3]];
-
-        //renderer->SetCamera(orthoCam);
-
-        renderer->RenderScene();
-
-        glReadPixels(0, 0, viewport[2], viewport[3], GL_DEPTH_COMPONENT, GL_UNSIGNED_BYTE, mDepth);
-        //glReadPixels(0, 0, viewport[2], viewport[3], GL_DEPTH_COMPONENT, GL_FLOAT, mDepth);
-/*
+
+void GrabDepthBuffer(float *data, RenderTexture *rt)
+{
+        rt->BindDepth();
+        rt->EnableTextureTarget();
+
+        const int texFormat = GL_DEPTH_COMPONENT;
+        glGetTexImage(rt->GetTextureTarget(), 0, texFormat, GL_FLOAT, data);
+
+        rt->DisableTextureTarget();
+}
+
+
+void ExportDepthBuffer(float *data)
+{
         startil();
 
-        if (!ilTexImage(viewport[2], viewport[3], 1, 1, IL_LUMINANCE, IL_UNSIGNED_BYTE, mDepth))
+        ILstring filename = ILstring("depth.tga");
+        ilRegisterType(IL_FLOAT);
+
+        const int depth = 1;
+        const int bpp = 1;
+
+        if (!ilTexImage(texWidth, texHeight, depth, bpp, IL_LUMINANCE, IL_FLOAT, data))
         {
                 cerr << "IL error " << ilGetError() << endl;
@@ -121 +90 @@
         }
 
-        ILstring writename = ILstring("out.tga");
-
-        ilSetInteger(IL_TGA_RLE, 1);
-        if (!ilSaveImage(writename))
+        if (!ilSaveImage(filename))
         {
                 cerr << "TGA write error " << ilGetError() << endl;
@@ -130 +96 @@
 
         stopil();
-*/
+
+        cout << "exported depth buffer" << endl;
+}
+
+
+SceneQuery::SceneQuery(const AxisAlignedBox3 &sceneBox, RenderTraverser *renderer): 
+mSceneBox(sceneBox), mDepth(NULL)
+{
+        Prepare(renderer);
+}
+
+
+bool SceneQuery::CalcIntersection(Vector3 &pt)
+{
+        const int px = (pt.x - mSceneBox.Min(0)) * (texWidth - 1) / mSceneBox.Size(0);
+        const int py = (pt.y - mSceneBox.Min(1)) * (texHeight - 1) / mSceneBox.Size(1);
+
+        float d = mDepth[px + py * texHeight];
+
+        static float depth = d;
+
+        if (d > 0)
+        {
+                // temporal smoothing of depth values
+                const float x = 0.5f;
+                depth = d * x + depth * (1.0f - x);
+
+                const float offs = mSceneBox.Size(2) * 5e-2f;
+                pt.z =  mSceneBox.Max().z - mSceneBox.Size().z * depth + offs;
+
+                return true;
+        }
+
+        return false;
+}
+
+
+void SceneQuery::Prepare(RenderTraverser *renderer)
+{
+        cout << "Preparing scene queries" << endl;
+
+        RenderTexture *depthTexture = new RenderTexture(texWidth, texHeight, true, true);
+
+#ifdef ATI
+        depthTexture->Initialize(true, true, false, false, false, 8, 8, 8, 8, RenderTexture::RT_COPY_TO_TEXTURE);
+#else
+        depthTexture->Initialize(true, true, false, false, false, 8, 8, 8, 8);//, RenderTexture::RT_COPY_TO_TEXTURE);
+#endif
+
+        PrintGLerror("Init");
+
+
+        const float xlen = mSceneBox.Size().x * 0.5f;
+        const float ylen = mSceneBox.Size().y * 0.5f;
+        
+        Camera *orthoCam = new Camera(xlen, ylen);
+        orthoCam->SetOrtho(true);
+
+        orthoCam->SetNear(0.0f);
+        orthoCam->Yaw(M_PI * 0.5f);
+
+        Vector3 pos = Vector3(mSceneBox.Center().x, mSceneBox.Center().y, mSceneBox.Max().z);
+        orthoCam->SetPosition(pos);
+
+        depthTexture->BeginCapture();
+
+        glViewport(0, 0, texWidth, texHeight);
+
+        glClearColor(1, 1, 1, 1);
+        glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
+
+        glFrontFace(GL_CCW);
+        glCullFace(GL_BACK);
+
+        glDisable(GL_CULL_FACE);
+        //glEnable(GL_CULL_FACE);
+
+        glShadeModel(GL_FLAT);
+        glEnable(GL_DEPTH_TEST);
+
+        glMatrixMode(GL_PROJECTION);
+        glLoadIdentity();
+        
+        glOrtho(-xlen, xlen, -ylen, ylen, 0.0f, mSceneBox.Size().z); 
+        
+        glMatrixMode(GL_MODELVIEW);
+        
+        orthoCam->SetupCameraView();
+
+        mDepth = new float[texHeight * texWidth];
+
+        //renderer->SetCamera(orthoCam);
+        renderer->RenderScene();
+
+        depthTexture->EndCapture();
+
+        GrabDepthBuffer(mDepth, depthTexture);
+        //ExportDepthBuffer(mDepth);
+        //PrintGLerror("grab");
+
+        DEL_PTR(depthTexture);
         DEL_PTR(orthoCam);
 }
 
-}
+
+
+
+}