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

Revision 1713, 7.8 KB checked in by mattausch, 18 years ago (diff)
Line 
1#ifdef GTP_INTERNAL
2#define USE_QT 1
3#else
4#define USE_QT 0
5#endif
6
7#define USE_THREADS 1
8
9#ifdef UNICODE
10#undef UNICODE
11#endif
12
13#include <windows.h>
14#include <stdio.h>
15#include <crtdbg.h>
16
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
24#include "PreprocessorThread.h"
25#include "ObjExporter.h"
26#include "SceneGraph.h"
27
28
29
30#if !USE_QT && USE_THREADS
31#include "BoostPreprocessorThread.h"
32#endif
33
34#include "ResourceManager.h"
35#include "GlRenderer.h"
36
37#if USE_QT
38#include "QtPreprocessorThread.h"
39#include "QtGlRenderer.h"
40#endif
41
42#define USE_EXE_PATH false
43
44
45using namespace GtpVisibilityPreprocessor;
46
47Preprocessor *preprocessor = NULL;
48GlRendererWidget *rendererWidget = NULL;
49
50// DLL function signature
51typedef GlRendererWidget *(*importFunction)(Preprocessor *);
52
53
54
55void Cleanup()
56{
57        DEL_PTR(rendererWidget);
58        DEL_PTR(preprocessor);
59
60        Environment::DelSingleton();
61        MeshManager::DelSingleton();
62        MaterialManager::DelSingleton();
63}
64
65
66static string ReplaceSuffix(string filename, string a, string b)
67{
68        string result = filename;
69
70        int pos = (int)filename.rfind(a, (int)filename.size() - 1);
71        if (pos == filename.size() - a.size()) {
72                result.replace(pos, a.size(), b);
73        }
74        return result;
75}
76
77
78static int SplitFilenames(const string str, vector<string> &filenames)
79{
80        int pos = 0;
81
82        while(1) {
83                int npos = (int)str.find(';', pos);
84               
85                if (npos < 0 || npos - pos < 1)
86                        break;
87                filenames.push_back(string(str, pos, npos - pos));
88                pos = npos + 1;
89        }
90       
91        filenames.push_back(string(str, pos, str.size() - pos));
92        return (int)filenames.size();
93}
94
95
96static string GetInternFilename(const string &filename, const string newSuffix)
97{
98        vector<string> filenames;
99        const int files = SplitFilenames(filename, filenames);
100
101        vector<string>::const_iterator sit, sit_end = filenames.end();
102        string kdFilename;
103
104        int i = 0;
105        for (sit = filenames.begin(); sit != sit_end; ++ sit, ++ i)
106        {
107                string currentFile = *sit;
108                string strippedFilename;
109
110                if (i == 0)
111                {       
112                        strippedFilename = currentFile;
113                }
114                else
115                {
116                        char *str = StripPath(currentFile.c_str());
117                        strippedFilename = string(str);
118
119                        delete [] str;
120                }
121               
122                string suffix("_");
123
124                if (i == (int)filenames.size() - 1)
125                {
126                        suffix = newSuffix;
127                }
128
129                if (strstr(strippedFilename.c_str(), ".x3d"))
130                {
131                        kdFilename += ReplaceSuffix(strippedFilename, ".x3d", suffix);
132                }
133        else if (strstr(strippedFilename.c_str(), ".dat"))
134                {
135                        kdFilename += ReplaceSuffix(strippedFilename, ".dat", suffix);
136                }
137                else if (strstr(strippedFilename.c_str(), ".obj"))
138                {
139                        kdFilename += ReplaceSuffix(strippedFilename, ".obj", suffix);
140                }
141                else
142                {
143                        cerr << "Error: Currently unsupported format for kd, filename " << currentFile << endl;
144                }
145        }
146
147        //cout << "kdfilename: " << kdFilename << endl;
148        return kdFilename;
149}
150
151
152int
153main(int argc, char **argv)
154{
155
156        //Now just call this function at the start of your program and if you're
157        //compiling in debug mode (F5), any leaks will be displayed in the Output
158        //window when the program shuts down. If you're not in debug mode this will
159        //be ignored. Use it as you will!
160        //note: from GDNet Direct [3.8.04 - 3.14.04] void detectMemoryLeaks() {
161#if 1
162  _CrtSetDbgFlag(_CRTDBG_LEAK_CHECK_DF|_CRTDBG_ALLOC_MEM_DF);
163  _CrtSetReportMode(_CRT_ASSERT,_CRTDBG_MODE_FILE);
164  _CrtSetReportFile(_CRT_ASSERT,_CRTDBG_FILE_STDERR);
165#endif
166
167        int returnCode = 0;
168       
169        InitTiming();
170
171        Debug.open("debug.log");
172 
173        Environment::GetSingleton()->Parse(argc, argv, USE_EXE_PATH);
174        MeshKdTree::ParseEnvironment();
175
176       
177        char buff[128];
178        Environment::GetSingleton()->GetStringValue("Preprocessor.type", buff);
179        string preprocessorType(buff);
180
181        if (!(preprocessor = PreprocessorFactory::CreatePreprocessor(preprocessorType)))
182        {
183                Environment::DelSingleton();
184                cerr << "Unknown preprocessor type" << endl;
185                exit(1);
186        }
187
188
189        /////////////
190        //-- load scene
191
192        Environment::GetSingleton()->GetStringValue("Scene.filename", buff);
193        string filename(buff);
194       
195
196        if (!preprocessor->LoadScene(filename))
197        {
198                cout << "loading file " << filename << " failed" << endl;
199                Cleanup();
200                exit(1);
201        }
202
203        const string externKdTree = ReplaceSuffix(filename, ".obj", ".kdf");
204        const string internKdTree = GetInternFilename(filename, preprocessor->mLoadMeshes ? ".kdm" : ".kdt");
205
206        ////////////
207        //-- initialize external ray caster
208       
209        if (preprocessor->InitRayCast(externKdTree, internKdTree))
210        {
211                cout << "ray casting initialized!" << endl;
212        }
213        else
214        {
215                cout << "ray casting initialization failed" << endl;
216                Cleanup();
217                exit(1);
218        }
219
220       
221        // export objects as obj
222        if (preprocessor->mExportObj)
223        {
224                if (strstr(filename.c_str(), ".obj"))
225                {
226                        cerr << "already in obj format" << endl;
227                        if (0)  preprocessor->ExportObj("test.obj", preprocessor->mObjects);
228                }
229                else
230                {
231                        const string objname = GetInternFilename(filename, ".obj");
232
233                        cout << "exporting scene to " << objname << endl;
234                        bool success = preprocessor->ExportObj(objname, preprocessor->mObjects);
235
236                        if (success)
237                        {
238                                cout << "finished exporting obj" << endl;
239                        }
240                        else
241                        {
242                                cerr << "error exporting " << objname << endl;
243                        }
244                }
245        }
246        if (0){
247        Exporter *exporter= Exporter::GetExporter("test1.wrl");
248        AxisAlignedBox3 bbox = preprocessor->mKdTree->GetBox();
249        bbox.Scale(Vector3(0.5, 1, 0.5));
250    exporter->SetWireframe();
251        exporter->ExportBox(preprocessor->mKdTree->GetBox());
252        exporter->ExportBox(bbox);
253        exporter->SetFilled();
254        exporter->ExportGeometry(preprocessor->mObjects, true, &bbox);
255        delete exporter;
256        }
257       
258       
259        // parse view cells related options
260        if (!preprocessor->PrepareViewCells())
261        {
262                Cleanup();
263                exit(1);
264        }
265
266        if (0)
267        {
268                preprocessor->Export(filename + "-out.x3d", true, false);
269                preprocessor->Export(filename + "-kdtree.x3d", false, true);   
270        }
271
272        // create a preprocessor thread (note: capsulates calls to boost fuctions!)
273        //PreprocessorThread *pt = PreprocessorThreadFactory::Create(preprocessor);
274        PreprocessorThread *pt;
275
276#if USE_QT
277        pt = new QtPreprocessorThread(preprocessor);
278#else
279#if USE_THREADS
280        pt = new BoostPreprocessorThread(preprocessor);
281#else
282        pt = new PreprocessorThread(preprocessor);
283#endif
284#endif
285
286        bool guiSupported = false;
287
288        if (preprocessor->mUseGlRenderer || preprocessor->mUseGlDebugger)
289          {
290                cout << "using gl widget" << endl;
291                // create and run the preprocessor application in a parallel thread
292#if USE_QT             
293#if USE_THREADS
294                pt->InitThread();
295                if (!preprocessor->mDelayVisibilityComputation)
296                  pt->RunThread();
297#endif
298               
299                // display the render widget
300                if (!rendererWidget)
301                {
302                  // create a qt application first (must be created before any opengl widget ...)
303                  QApplication *app = new QApplication(argc, NULL);
304                 
305                  if (!QGLFormat::hasOpenGL() || !QGLPixelBuffer::hasOpenGLPbuffers()) {
306                        QMessageBox::information(0, "OpenGL pbuffers",
307                                                                         "This system does not support OpenGL/pbuffers.",
308                                                                         QMessageBox::Ok);
309                        return NULL;
310                  }
311                 
312                 
313                  rendererWidget = new QtGlRendererWidget(preprocessor->mSceneGraph,
314                                                                                                  preprocessor->mViewCellsManager,
315                                                                                                  preprocessor->mKdTree);
316
317                  ((QtGlRendererWidget *)rendererWidget)->SetThread((QtPreprocessorThread *)pt);
318                  rendererWidget->Show();
319                  guiSupported = true;
320                }
321
322                qApp->exec();   
323#endif
324          }
325       
326        if (!guiSupported) {
327          preprocessor->mUseGlRenderer = false;
328          preprocessor->mUseGlDebugger = false;
329        }
330
331        if (!(preprocessor->mUseGlRenderer || preprocessor->mUseGlDebugger)) {
332                // just call the mail method -> will be executed in the main thread
333                pt->Main();
334        }       
335
336        // release memory
337        Cleanup();
338       
339        return returnCode;
340}
341
Note: See TracBrowser for help on using the repository browser.