source: GTP/trunk/Lib/Vis/Preprocessing/src/main.cpp @ 2176

#define USE_THREADS 0

#ifdef UNICODE
#undef UNICODE
#endif

#define NOMINMAX
#include <windows.h>
#include <stdio.h>
#include <crtdbg.h>

#include "PreprocessorFactory.h"
#include "Parser.h"
#include "Environment.h"
#include "MeshKdTree.h"
#include "Preprocessor.h"
#include "common.h"
#include "PreprocessorThread.h"
#include "ObjExporter.h"
#include "SceneGraph.h"
#include "GlobalLinesRenderer.h"


#include "ViewCellsManager.h"

#ifdef USE_QT  
        #include "QtPreprocessorThread.h"
        #include "QtGlViewer.h"
        #include "QtGlRenderer.h"
#else
        #if USE_THREADS
                #include "BoostPreprocessorThread.h"
        #endif
#endif

#include "ResourceManager.h"
#include "GlRenderer.h"


#define USE_EXE_PATH false


using namespace GtpVisibilityPreprocessor;

//Preprocessor *preprocessor = NULL;
GlRendererWidget *rendererWidget = NULL;
//GlobalLinesRenderer *globalLinesRenderer = NULL;


// DLL function signature
typedef GlRendererWidget *(*importFunction)(Preprocessor *);



void Cleanup()
{
        DEL_PTR(rendererWidget);
        DEL_PTR(preprocessor);

        Environment::DelSingleton();
        MeshManager::DelSingleton();
        MaterialManager::DelSingleton();
}


static string ReplaceSuffix(string filename, string a, string b)
{
        string result = filename;

        int pos = (int)filename.rfind(a, (int)filename.size() - 1);
        if (pos == filename.size() - a.size()) {
                result.replace(pos, a.size(), b);
        }
        return result;
}


static int SplitFilenames(const string str, vector<string> &filenames)
{
        int pos = 0;

        while(1) {
                int npos = (int)str.find(';', pos);
                
                if (npos < 0 || npos - pos < 1)
                        break;
                filenames.push_back(string(str, pos, npos - pos));
                pos = npos + 1;
        }
        
        filenames.push_back(string(str, pos, str.size() - pos));
        return (int)filenames.size();
}


static string GetInternFilename(const string &filename, const string newSuffix)
{
        vector<string> filenames;
        const int files = SplitFilenames(filename, filenames);

        vector<string>::const_iterator sit, sit_end = filenames.end();
        string kdFilename;

        int i = 0;
        for (sit = filenames.begin(); sit != sit_end; ++ sit, ++ i)
        {
                string currentFile = *sit;
                string strippedFilename;

                if (i == 0)
                {       
                        strippedFilename = currentFile;
                }
                else
                {
                        char *str = StripPath(currentFile.c_str());
                        strippedFilename = string(str);

                        delete [] str;
                }
                
                string suffix("_");

                if (i == (int)filenames.size() - 1)
                {
                        suffix = newSuffix;
                }

                if (strstr(strippedFilename.c_str(), ".x3d")) 
                {
                        kdFilename += ReplaceSuffix(strippedFilename, ".x3d", suffix);
                }
        else if (strstr(strippedFilename.c_str(), ".dat"))
                {
                        kdFilename += ReplaceSuffix(strippedFilename, ".dat", suffix);
                } 
                else if (strstr(strippedFilename.c_str(), ".obj"))
                {
                        kdFilename += ReplaceSuffix(strippedFilename, ".obj", suffix);
                }
                else
                {
                        cerr << "Error: Currently unsupported format for kd, filename " << currentFile << endl;
                }
        }

        //cout << "kdfilename: " << kdFilename << endl;
        return kdFilename;
}


int
main(int argc, char **argv)
{
        //Now just call this function at the start of your program and if you're
        //compiling in debug mode (F5), any leaks will be displayed in the Output
        //window when the program shuts down. If you're not in debug mode this will
        //be ignored. Use it as you will!
        //note: from GDNet Direct [3.8.04 - 3.14.04] void detectMemoryLeaks() {
#if 1
  _CrtSetDbgFlag(_CRTDBG_LEAK_CHECK_DF|_CRTDBG_ALLOC_MEM_DF);
  _CrtSetReportMode(_CRT_ASSERT,_CRTDBG_MODE_FILE);
  _CrtSetReportFile(_CRT_ASSERT,_CRTDBG_FILE_STDERR); 
#endif

        int returnCode = 0;

        InitTiming();
        Debug.open("debug.log");

#if 0
        cout<<"Allocating 1.5GB..."<<endl;

        
        for (int i=0; i < 1000; i++) {
          void *p = malloc(1250000000);
          if (p != NULL) {
                cout<<"success"<<(i+1)*1000<<"MB"<<endl;
          }
          else {
                cout<<"fail"<<endl;
                exit(1);
          }
        }
#endif
        
        Environment::GetSingleton()->Parse(argc, argv, USE_EXE_PATH);
        MeshKdTree::ParseEnvironment();

        char buff[128];
        Environment::GetSingleton()->GetStringValue("Preprocessor.type", buff);
        string preprocessorType(buff);

        if (!(preprocessor = PreprocessorFactory::CreatePreprocessor(preprocessorType)))
        {
                Environment::DelSingleton();
                cerr << "Unknown preprocessor type" << endl;
                exit(1);
        }


        Environment::GetSingleton()->GetStringValue("Scene.filename", buff);
        string filename(buff);
        
        const string externKdTree = ReplaceSuffix(filename, ".obj", ".kdf");
        const string internKdTree = GetInternFilename(filename, preprocessor->mLoadMeshes ?
                                                                                                  ".kdm" : ".kdt");

        if (preprocessor->InitRayCast(externKdTree, internKdTree))
        {
                cout << "ray casting initialized!" << endl;
        }
        else
        {
                cout << "ray casting initialization failed!" << endl;
                Cleanup();
                exit(1);
        }

        //Debug << "using pvs type " << PVS_TYPE << endl;

        /////////////
        //-- load scene 

        if (!preprocessor->LoadScene(filename))
        {
                cout << "loading file " << filename << " failed" << endl;
                Cleanup();
                exit(1);
        }
        
        ////////////
        //-- initialize external ray caster

        if (preprocessor->LoadInternKdTree(internKdTree))
        {
                cout << "intern kd tree loaded!" << endl;
        }
        else
        {
                cout << "loading intern kd tree failed!" << endl;
                Cleanup();
                exit(1);
        }

        // export objects as obj
        if (preprocessor->mExportObj)
        {
                if (strstr(filename.c_str(), ".obj"))
                {
                        cerr << "already in obj format" << endl;
                        if (0)  preprocessor->ExportObj("test.obj", preprocessor->mObjects);
                }
                else
                {
                  
                        const string objname = GetInternFilename(filename, ".obj");

                        cout << "exporting scene to " << objname << endl;
                        bool success = preprocessor->ExportObj(objname, preprocessor->mObjects);

                        
                        if (success)
                        {
                                cout << "finished exporting obj" << endl;
                        }
                        else
                        {
                                cerr << "error exporting " << objname << endl;
                        }
                }
        }
        
        // parse view cells related options
        if (!preprocessor->PrepareViewCells())
        {
                cerr << "error: view cells could not be loaded" << endl;

                Cleanup();
                exit(1);
        }

        string viewCellPointsFile;

        if (strstr(filename.c_str(), ".obj"))
                viewCellPointsFile = ReplaceSuffix(filename, ".obj", ".vc");
        else if (strstr(filename.c_str(), ".dat"))
                viewCellPointsFile = ReplaceSuffix(filename, ".dat", ".vc");
        else if (strstr(filename.c_str(), ".x3d"))
                viewCellPointsFile = ReplaceSuffix(filename, ".x3d", ".vc");

        bool importRandomViewCells;
        Environment::GetSingleton()->GetBoolValue("ViewCells.importRandomViewCells",
                                                                                           importRandomViewCells);

        if (importRandomViewCells)
        {
                cout << "importing random view cells" << endl;
                preprocessor->mViewCellsManager->ImportRandomViewCells(viewCellPointsFile);
                cout << "finished" << endl;
        }

        bool useHwGlobalLines;
        Environment::GetSingleton()->GetBoolValue("Preprocessor.useHwGlobalLines",
                                                                                           useHwGlobalLines);

        if (useHwGlobalLines)
                preprocessor->PrepareHwGlobalLines();

        // create a preprocessor thread (note: capsulates calls to boost fuctions!)
        PreprocessorThread *pt = NULL;

#ifdef TRY_GLOBAL_LINES
        
        preprocessor->PrepareHwGlobalLines();
        globalLinesRenderer->Run();

#else

#ifdef USE_QT
        
        // create a qt application first (must be created before any opengl widget ...)
        QApplication *app = new QApplication(argc, NULL);

        pt = new QtPreprocessorThread(preprocessor);

#else

        //#if USE_THREADS
        //      pt = new BoostPreprocessorThread(preprocessor);
        //#else
                // use a dummy thread
                pt = new DummyPreprocessorThread(preprocessor);
        //#endif
#endif

        preprocessor->SetThread(pt);

        bool guiSupported = false;
        bool useRendererBuffer = true;

        int frames;
        Environment::GetSingleton()->GetIntValue("Preprocessor.pvsRenderErrorSamples", frames);
        
        
#ifdef USE_QT

        if (importRandomViewCells || frames)
          {
                QGLFormat f;
                f.setStencil(true);
                QGLFormat::setDefaultFormat(f);

                // NOTE: render texture should be power of 2 and square
                // renderer must be initialised
                // $$matt
                preprocessor->renderer = 
                  new QtGlRendererBuffer(1024, 1024,
                                                                 preprocessor->mSceneGraph,
                                                                 preprocessor->mViewCellsManager,
                                                                 preprocessor->mKdTree);
          }
#endif

        if (preprocessor->mUseGlRenderer || preprocessor->mUseGlDebugger)
          {
#ifdef USE_QT
                cout << "using gl widget" << endl;
                // create and run the preprocessor application in a parallel thread
                
                pt->InitThread();
                pt->RunThread();

                // display the render widget
                if (!rendererWidget) 
                {

                  if (!QGLFormat::hasOpenGL() || !QGLPixelBuffer::hasOpenGLPbuffers()) {
                        QMessageBox::information(0, "OpenGL pbuffers", 
                                                                           "This system does not support OpenGL/pbuffers.",
                                                                           QMessageBox::Ok);
                          return NULL;
                        }

                  rendererWidget = 
                        new QtGlRendererWidget(preprocessor->mSceneGraph,
                                                                   preprocessor->mViewCellsManager,
                                                                   preprocessor->mKdTree);
                  
                  rendererWidget->Show();
                  
                  QtGlViewer *viewer = 
                        new QtGlViewer(NULL, (QtGlRendererWidget *)rendererWidget);
                  
                  viewer->show();
                  guiSupported = true; 
                }
                
                
                bool exportRandomViewCells;
                Environment::GetSingleton()->GetBoolValue("ViewCells.exportRandomViewCells",
                                                                                                  exportRandomViewCells);

                if (exportRandomViewCells)
                {
                        cout << "exporting random view cells" << endl;
                        preprocessor->mViewCellsManager->ExportRandomViewCells(viewCellPointsFile);
                        cout << "finished" << endl;
                }

                /*bool evaluatePixelError;
                Environment::GetSingleton()->GetBoolValue("Preprocessor.evaluatePixelError",  evaluatePixelError);

                if (evaluatePixelError)
                {
                        cout << "evaluating pixel error" << endl;
                        preprocessor->ComputeRenderError();
                }*/
                
                qApp->exec();
#endif
        }

        // no gui available
        if (!guiSupported) 
        {
                if (preprocessor->mUseGlRenderer || preprocessor->mUseGlDebugger)
                        cout << "warning: gui not supported!" << endl;
                
                preprocessor->mUseGlRenderer = false;
                preprocessor->mUseGlDebugger = false;
        }

        if (!(preprocessor->mUseGlRenderer || preprocessor->mUseGlDebugger)) 
        {
                cout << "executing main thread" << endl;
                // just call the mail method -> will be executed in the main thread
                pt->Main();
        }       
#endif

        // release memory
        Cleanup();
        
        delete pt;

        return returnCode;
}