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

Revision 2699, 10.6 KB checked in by bittner, 16 years ago (diff)

havran ray caster fix
Line 
1#define USE_THREADS 0
2
3#ifdef UNICODE
4#undef UNICODE
5#endif
6
7#define NOMINMAX
8#ifdef _WIN32
9#include <windows.h>
10#ifdef _CRT_SET
11#include <crtdbg.h>
12#endif // _CRT_SET
13#endif
14#include <cstdio>
15
16#include "Camera.h"
17#include "PreprocessorFactory.h"
18#include "Parser.h"
19#include "Environment.h"
20#include "MeshKdTree.h"
21#include "Preprocessor.h"
22#include "common.h"
23#include "PreprocessorThread.h"
24#include "ObjExporter.h"
25#include "SceneGraph.h"
26#include "GlobalLinesRenderer.h"
27#include "RayCaster.h"
28#include "ViewCellsManager.h"
29
30
31#ifdef USE_QT 
32        #include "QtPreprocessorThread.h"
33        #include "QtGlViewer.h"
34        #include "QtGlRenderer.h"
35        #include <QGLWidget>
36
37#else
38        #if USE_THREADS
39                #include "BoostPreprocessorThread.h"
40        #endif
41#endif
42
43#include "ResourceManager.h"
44#include "GlRenderer.h"
45
46
47#define USE_EXE_PATH false
48
49
50using namespace GtpVisibilityPreprocessor;
51
52//Preprocessor *preprocessor = NULL;
53GlRendererWidget *rendererWidget = NULL;
54//GlobalLinesRenderer *globalLinesRenderer = NULL;
55
56
57// DLL function signature
58typedef GlRendererWidget *(*importFunction)(Preprocessor *);
59
60
61static string GetViewPointsListName(const string &filename)
62{
63        string viewCellPointsFile;
64
65        char buff[200];
66        Environment::GetSingleton()->GetStringValue("ViewCells.randomViewPointsList", buff);
67
68        string vpList(buff);
69
70        if (vpList == "")
71        {
72                // if not specified, take file that has same prefix as scene file
73                if (strstr(filename.c_str(), ".obj"))
74                        viewCellPointsFile = ReplaceSuffix(filename, ".obj", ".vc");
75                else if (strstr(filename.c_str(), ".dat"))
76                        viewCellPointsFile = ReplaceSuffix(filename, ".dat", ".vc");
77                else if (strstr(filename.c_str(), ".x3d"))
78                        viewCellPointsFile = ReplaceSuffix(filename, ".x3d", ".vc");
79        }
80        else
81        {
82                viewCellPointsFile = vpList;
83        }
84
85        return viewCellPointsFile;
86}
87
88
89void Cleanup()
90{
91        DEL_PTR(rendererWidget);
92        DEL_PTR(preprocessor);
93
94        Environment::DelSingleton();
95        MeshManager::DelSingleton();
96        MaterialManager::DelSingleton();
97}
98
99
100
101
102static string GetInternFilename(const string &filename, const string newSuffix)
103{
104        vector<string> filenames;
105        const int files = SplitFilenames(filename, filenames);
106
107        vector<string>::const_iterator sit, sit_end = filenames.end();
108        string kdFilename;
109
110        int i = 0;
111        for (sit = filenames.begin(); sit != sit_end; ++ sit, ++ i)
112        {
113                string currentFile = *sit;
114                string strippedFilename;
115
116                if (i == 0)
117                {       
118                        strippedFilename = currentFile;
119                }
120                else
121                {
122                        char *str = StripPath(currentFile.c_str());
123                        strippedFilename = string(str);
124
125                        delete [] str;
126                }
127               
128                string suffix("_");
129
130                if (i == (int)filenames.size() - 1)
131                {
132                        suffix = newSuffix;
133                }
134
135                if (strstr(strippedFilename.c_str(), ".x3d"))
136                {
137                        kdFilename += ReplaceSuffix(strippedFilename, ".x3d", suffix);
138                }
139        else if (strstr(strippedFilename.c_str(), ".dat"))
140                {
141                        kdFilename += ReplaceSuffix(strippedFilename, ".dat", suffix);
142                }
143                else if (strstr(strippedFilename.c_str(), ".obj"))
144                {
145                        kdFilename += ReplaceSuffix(strippedFilename, ".obj", suffix);
146                }
147                else
148                {
149                        cerr << "Error: Currently unsupported format for kd, filename " << currentFile << endl;
150                }
151        }
152
153        //cout << "kdfilename: " << kdFilename << endl;
154        return kdFilename;
155}
156
157int
158main(int argc, char **argv)
159{
160        int returnCode = 0;
161
162#ifdef _CRT_SET
163
164        //Now just call this function at the start of your program and if you're
165        //compiling in debug mode (F5), any leaks will be displayed in the Output
166        //window when the program shuts down. If you're not in debug mode this will
167        //be ignored. Use it as you will!
168        //note: from GDNet Direct [3.8.04 - 3.14.04] void detectMemoryLeaks() {
169
170        _CrtSetDbgFlag(_CRTDBG_LEAK_CHECK_DF|_CRTDBG_ALLOC_MEM_DF);
171        _CrtSetReportMode(_CRT_ASSERT,_CRTDBG_MODE_FILE);
172        _CrtSetReportFile(_CRT_ASSERT,_CRTDBG_FILE_STDERR);
173#endif
174
175        InitTiming();
176        Debug.open("debug.log");
177
178        Environment::GetSingleton()->Parse(argc, argv, USE_EXE_PATH);
179        MeshKdTree::ParseEnvironment();
180
181        char buff[128];
182        Environment::GetSingleton()->GetStringValue("Preprocessor.type", buff);
183        string preprocessorType(buff);
184
185        if (!(preprocessor = PreprocessorFactory::CreatePreprocessor(preprocessorType)))
186        {
187                Environment::DelSingleton();
188                cerr << "Unknown preprocessor type" << endl;
189                exit(1);
190        }
191
192
193        Environment::GetSingleton()->GetStringValue("Scene.filename", buff);
194        string filename(buff);
195
196        const string externKdTree = ReplaceSuffix(filename, ".obj", ".kdf");
197        const string internKdTree = GetInternFilename(filename, preprocessor->mLoadMeshes ?
198                ".kdm" : ".kdt");
199
200        if (preprocessor->InitRayCast(externKdTree, internKdTree))
201        {
202                cout << "ray casting initialized!" << endl;
203        }
204        else
205        {
206                cout << "ray casting initialization failed!" << endl;
207                Cleanup();
208                exit(1);
209        }
210
211        /////////////
212        //-- load scene
213
214        if (!preprocessor->LoadScene(filename))
215        {
216                cout << "loading file " << filename << " failed" << endl;
217                Cleanup();
218                exit(1);
219        }
220
221
222        ////////////
223        //-- initialize external ray caster
224        if (preprocessor->LoadInternKdTree(internKdTree))
225        {
226                cout << "intern kd tree loaded!" << endl;
227        }
228        else
229        {
230                cout << "loading intern kd tree failed!" << endl;
231                Cleanup();
232                exit(1);
233        }
234
235       
236       
237        // export objects as obj
238        if (preprocessor->mExportObj)
239        {
240                if (strstr(filename.c_str(), ".obj"))
241                {
242                        cerr << "already in obj format" << endl;
243                        if (0)  preprocessor->ExportObj("test.obj", preprocessor->mObjects);
244                }
245                else
246                {
247
248                        const string objname = GetInternFilename(filename, ".obj");
249
250                        cout << "exporting scene to " << objname << endl;
251                        bool success = preprocessor->ExportObj(objname, preprocessor->mObjects);
252
253                        if (success)
254                                cout << "finished exporting obj" << endl;
255                        else
256                                cerr << "error exporting " << objname << endl;
257                }
258        }
259
260        // parse view cells related options
261        if (!preprocessor->PrepareViewCells())
262        {
263                cerr << "error: view cells could not be loaded" << endl;
264                Cleanup();
265                exit(1);
266        }
267        else
268                cout << "view cells successfully loaded" << endl;
269
270
271        const string viewCellPointsFile = GetViewPointsListName(filename);
272
273        bool importRandomViewCells;
274        Environment::GetSingleton()->GetBoolValue("ViewCells.importRandomViewCells", importRandomViewCells);
275
276        if (importRandomViewCells)
277        {
278                cout << "importing random view points" << endl;
279
280                if (preprocessor->mViewCellsManager->ImportViewCellsList(viewCellPointsFile))
281                        cout << "successfully loaded " << viewCellPointsFile << endl;
282                else
283                        cerr << "error: file << " << viewCellPointsFile << " not found" << endl;
284        }
285
286        // create meshes of view cells
287        preprocessor->mViewCellsManager->FinalizeViewCells(true);
288
289        cout << "view space box: " << preprocessor->mViewCellsManager->GetViewSpaceBox() << endl;
290        bool useHwGlobalLines;
291        Environment::GetSingleton()->GetBoolValue("Preprocessor.useHwGlobalLines",
292                useHwGlobalLines);
293
294        if (useHwGlobalLines)
295                preprocessor->PrepareHwGlobalLines();
296
297        // create a preprocessor thread (note: capsulates calls to boost fuctions!)
298        PreprocessorThread *pt = NULL;
299
300        //preprocessor->PrepareHwGlobalLines();
301        //globalLinesRenderer->Run();
302
303
304        bool guiSupported = false;
305        const bool useRendererBuffer = true;
306
307        int frames;
308        Environment::GetSingleton()->GetIntValue("Preprocessor.pvsRenderErrorSamples", frames);
309
310        bool evalPixelError;
311        Environment::GetSingleton()->GetBoolValue("Preprocessor.evaluatePixelError", evalPixelError);
312
313
314#ifdef USE_QT
315
316        // create a qt application first (must be created before any opengl widget ...)
317        QApplication *app = new QApplication(argc, NULL);
318
319        pt = new QtPreprocessorThread(preprocessor);   
320
321        if (evalPixelError && (importRandomViewCells || frames))
322        {
323                if (!QGLFormat::hasOpenGL() || !QGLPixelBuffer::hasOpenGLPbuffers())
324                {
325                        cerr << "this system does not support OpenGL/pbuffers" << endl;
326
327                        QMessageBox::information(0, "OpenGL pbuffers",
328                                "This system does not support OpenGL/pbuffers.",
329                                QMessageBox::Ok);
330
331                        return NULL;
332                }
333
334                QGLFormat f;
335                f.setStencil(true);
336                QGLFormat::setDefaultFormat(f);
337
338                // NOTE: render texture should be power of 2 and square
339                // renderer must be initialised
340                // $$matt
341                QtGlRendererBuffer *glbuf =
342                        new QtGlRendererBuffer(1024, 1024,
343                                                                   preprocessor->mSceneGraph,
344                                                                   preprocessor->mViewCellsManager,
345                                                                   preprocessor->mKdTree);
346
347                preprocessor->renderer = glbuf;
348        }
349
350        bool exportRandomViewCells;
351        Environment::GetSingleton()->GetBoolValue("ViewCells.exportRandomViewCells",
352                exportRandomViewCells);
353
354        if (exportRandomViewCells)
355        {
356                cout << "exporting random view cells" << endl;
357                preprocessor->mViewCellsManager->ExportRandomViewCells(viewCellPointsFile);
358                cout << "finished" << endl;
359        }
360
361        if (preprocessor->mUseGlRenderer || preprocessor->mUseGlDebugger)
362        {
363                ////////
364                //-- create and run the preprocessor application in a parallel thread
365
366                cout << "using gl widget" << endl;
367
368                pt->InitThread();
369                pt->RunThread();
370
371                // display the render widget
372                if (!rendererWidget)
373                {
374                        if (!QGLFormat::hasOpenGL() || !QGLPixelBuffer::hasOpenGLPbuffers())
375                        {
376                                cerr << "this system does not support OpenGL/pbuffers" << endl;
377
378                                QMessageBox::information(0, "OpenGL pbuffers",
379                                        "This system does not support OpenGL/pbuffers.",
380                                        QMessageBox::Ok);
381
382                                return NULL;
383                        }
384
385                        rendererWidget =
386                                new QtGlRendererWidget(preprocessor->mSceneGraph,
387                                preprocessor->mViewCellsManager,
388                                preprocessor->mKdTree);
389
390                        rendererWidget->Show();
391
392                        // hack matt
393                        preprocessor->mRendererWidget = rendererWidget;
394                        //rendererWidget->SetWindowTitle("Global Visualization");
395
396                        if (1 && !rendererWidget->GetUseVbos()) // viewer not working with vbo
397                        {
398                                cout << "starting the qt viewer" << endl;
399                                QtGlViewer *viewer =
400                                        new QtGlViewer(NULL, (QtGlRendererWidget *)rendererWidget);
401                                viewer->show();
402                        }
403
404                        guiSupported = true;
405                }
406
407                qApp->exec();
408        }
409#else // USE_QT
410
411        /*#if USE_THREADS
412        pt = new BoostPreprocessorThread(preprocessor);
413        #else
414        // use a dummy thread
415        pt = new DummyPreprocessorThread(preprocessor);
416        #endif
417        */
418#endif
419
420        preprocessor->SetThread(pt);
421
422        // no gui available
423        if (!guiSupported)
424        {
425                if (preprocessor->mUseGlRenderer || preprocessor->mUseGlDebugger)
426                        cout << "warning: gui not supported!" << endl;
427
428                preprocessor->mUseGlRenderer = false;
429                preprocessor->mUseGlDebugger = false;
430        }
431
432        if (!(preprocessor->mUseGlRenderer || preprocessor->mUseGlDebugger))
433        {
434                cout << "executing main thread" << endl;
435                // just call the mail method -> will be executed in the main thread
436                pt->Main();
437        }       
438
439        // release memory
440        Cleanup();
441        delete pt;
442
443        return returnCode;
444}
445
Note: See TracBrowser for help on using the repository browser.