source: GTP/trunk/Lib/Vis/Preprocessing/src/Preprocessor.cpp @ 1570

Revision 1570, 24.0 KB checked in by mattausch, 18 years ago (diff)
RevLine 
[372]1#include "SceneGraph.h"
2#include "Exporter.h"
3#include "UnigraphicsParser.h"
4#include "X3dParser.h"
5#include "Preprocessor.h"
6#include "ViewCell.h"
7#include "Environment.h"
[439]8#include "ViewCellsManager.h"
[445]9#include "ViewCellBsp.h"
10#include "VspBspTree.h"
[469]11#include "RenderSimulator.h"
[496]12#include "GlRenderer.h"
[749]13#include "PlyParser.h"
[1020]14#include "SamplingStrategy.h"
[1233]15#include "VspTree.h"
16#include "OspTree.h"
[1221]17#include "ObjParser.h"
[1264]18#include "BvHierarchy.h"
[1279]19#include "HierarchyManager.h"
[1287]20#include "VssRay.h"
[1520]21#include "IntelRayCaster.h"
22#include "InternalRayCaster.h"
[1264]23
[1287]24
[372]25
[1292]26#define DEBUG_RAYCAST 0
27
28
[863]29namespace GtpVisibilityPreprocessor {
[860]30
[1020]31const static bool ADDITIONAL_GEOMETRY_HACK = false;
[860]32
[492]33
[1001]34// HACK: Artificially modify scene to watch rendercost changes
[750]35static void AddGeometry(SceneGraph *scene)
36{
[1328]37        scene->GetRoot()->UpdateBox();
[752]38
[750]39        AxisAlignedBox3 sceneBox = scene->GetBox();
40
41        int n = 200;
42
[1291]43        if (0)
[750]44        {
[1291]45                // form grid of boxes
46                for (int i = 0; i < n; ++ i)
[750]47                {
[1291]48                        for (int j = 0; j < n; ++ j)
49                        {
50                                const Vector3 scale2((float)j * 0.8f / n + 0.1f,  0.05f, (float)i * 0.8f  / (float)n + 0.1f);
[750]51
[1291]52                                const Vector3 pt2 = sceneBox.Min() + scale2 * (sceneBox.Max() - sceneBox.Min());
53
54                                const Vector3 boxSize = sceneBox.Size() * Vector3(0.0025f, 0.01f, 0.0025f);
55                                AxisAlignedBox3 box(pt2, pt2 + boxSize);
56                                Mesh *mesh = CreateMeshFromBox(box);
57
58                                mesh->Preprocess();
59
60                                MeshInstance *mi = new MeshInstance(mesh);
[1328]61                                scene->GetRoot()->mGeometry.push_back(mi);
[1291]62                        }
[750]63                }
64
[1291]65                for (int i = 0; i < n; ++ i)
[750]66                {
[1291]67                        for (int j = 0; j < n; ++ j)
68                        {
69                                const Vector3 scale2(0.15f, (float)j * 0.8f / n + 0.1f, (float)i * 0.8f  / (float)n + 0.1f);
70
71                                Vector3 pt2 = sceneBox.Min() + scale2 * (sceneBox.Max() - sceneBox.Min());
72
73                                Vector3 boxSize = sceneBox.Size() * Vector3(0.0025f, 0.01f, 0.0025f);
74                                AxisAlignedBox3 box(pt2, pt2 + boxSize);
75                                Mesh *mesh = CreateMeshFromBox(box);
76
77                                mesh->Preprocess();
78
79                                MeshInstance *mi = new MeshInstance(mesh);
[1328]80                                scene->GetRoot()->mGeometry.push_back(mi);
[1291]81                        }
82                }
83
84                for (int i = 0; i < n; ++ i)
85                {
86                        const Vector3 scale2(2, 0.2f, (float)i * 0.8f  / (float)n + 0.1f);
87
[750]88                        Vector3 pt2 = sceneBox.Min() + scale2 * (sceneBox.Max() - sceneBox.Min());
[1291]89
90                        //Vector3 boxSize = sceneBox.Size() * Vector3(0.0025, 0.01, 0.0025);
91                        Vector3 boxSize = sceneBox.Size() * Vector3(0.005f, 0.02f, 0.005f);
92
93                        AxisAlignedBox3 box(pt2 + 0.1f, pt2 + boxSize);
[991]94                        Mesh *mesh = CreateMeshFromBox(box);
[750]95
96                        mesh->Preprocess();
[1291]97
[750]98                        MeshInstance *mi = new MeshInstance(mesh);
[1328]99                        scene->GetRoot()->mGeometry.push_back(mi);
[750]100                }
[1291]101
[1328]102                scene->GetRoot()->UpdateBox();
[750]103        }
104
[840]105        if (1)
106        {
[1221]107                // plane separating view space regions
[1135]108                const Vector3 scale(1.0f, 0.0, 0);
[750]109
[840]110                Vector3 pt = sceneBox.Min() + scale * (sceneBox.Max() - sceneBox.Min());
[750]111
[840]112                Plane3 cuttingPlane(Vector3(1, 0, 0), pt);
113                Mesh *planeMesh = new Mesh();
[1291]114
[840]115                Polygon3 *poly = sceneBox.CrossSection(cuttingPlane);
116                IncludePolyInMesh(*poly, *planeMesh);
[1291]117
[840]118                planeMesh->Preprocess();
[1291]119
[840]120                MeshInstance *planeMi = new MeshInstance(planeMesh);
[1328]121                scene->GetRoot()->mGeometry.push_back(planeMi);
[840]122        }       
[750]123}
124
125
[372]126Preprocessor::Preprocessor():
127mKdTree(NULL),
[409]128mBspTree(NULL),
[445]129mVspBspTree(NULL),
[1279]130mHierarchyManager(NULL),
[1002]131mViewCellsManager(NULL),
[1251]132mRenderSimulator(NULL),
[1279]133mPass(0),
[1520]134mSceneGraph(NULL),
135mRayCaster(NULL)
[308]136{
[1004]137        Environment::GetSingleton()->GetBoolValue("Preprocessor.useGlRenderer", mUseGlRenderer);
[538]138 
[840]139        // renderer will be constructed when the scene graph and viewcell manager will be known
140        renderer = NULL;
[496]141 
[1004]142        Environment::GetSingleton()->GetBoolValue("Preprocessor.useGlDebugger", mUseGlDebugger);
[1379]143        Environment::GetSingleton()->GetBoolValue("Preprocessor.loadMeshes", mLoadMeshes);
[1004]144        Environment::GetSingleton()->GetBoolValue("Preprocessor.quitOnFinish", mQuitOnFinish);
145        Environment::GetSingleton()->GetBoolValue("Preprocessor.computeVisibility", mComputeVisibility);
146        Environment::GetSingleton()->GetBoolValue("Preprocessor.detectEmptyViewSpace", mDetectEmptyViewSpace);
147        Environment::GetSingleton()->GetBoolValue("Preprocessor.exportVisibility", mExportVisibility );
[1520]148       
[871]149        char buffer[256];
[1004]150        Environment::GetSingleton()->GetStringValue("Preprocessor.visibilityFile",  buffer);
[1415]151   
[871]152        mVisibilityFileName = buffer;
[1004]153        Environment::GetSingleton()->GetBoolValue("Preprocessor.applyVisibilityFilter", mApplyVisibilityFilter );
154        Environment::GetSingleton()->GetBoolValue("Preprocessor.applyVisibilitySpatialFilter",
[904]155                                                          mApplyVisibilitySpatialFilter );
[1004]156        Environment::GetSingleton()->GetFloatValue("Preprocessor.visibilityFilterWidth", mVisibilityFilterWidth);
[878]157
[840]158        Debug << "detect empty view space=" << mDetectEmptyViewSpace << endl;
[1379]159        Debug << "load meshes: " << mLoadMeshes << endl;
[372]160}
161
162
163Preprocessor::~Preprocessor()
164{
[1291]165        cout << "cleaning up" << endl;
[752]166
[1291]167        cout << "Deleting view cells manager ... \n";
168        DEL_PTR(mViewCellsManager);
169        cout << "done.\n";
[752]170
[1291]171        cout << "Deleting bsp tree ... \n";
172        DEL_PTR(mBspTree);
173        cout << "done.\n";
[752]174
[1291]175        cout << "Deleting kd tree...\n";
176        DEL_PTR(mKdTree);
177        cout << "done.\n";
[752]178
[1291]179        cout << "Deleting hierarchy manager...\n";
180        DEL_PTR(mHierarchyManager);
181        cout << "done.\n";
[1002]182
[1291]183        cout << "Deleting vspbsp tree...\n";
184        DEL_PTR(mVspBspTree);
185        cout << "done.\n";
[1002]186
[1291]187        cout << "Deleting scene graph...\n";
188        DEL_PTR(mSceneGraph);
189        cout << "done.\n";
190
191        DEL_PTR(mRenderSimulator);
192        DEL_PTR(renderer);
[1523]193        DEL_PTR(mRayCaster);
[372]194}
195
[1521]196
197GlRendererBuffer *Preprocessor::GetRenderer()
198{
199        return renderer;
200}
201
202
203static int SplitFilenames(const string str, vector<string> &filenames)
[387]204{
205        int pos = 0;
206
207        while(1) {
[469]208                int npos = (int)str.find(';', pos);
[387]209               
210                if (npos < 0 || npos - pos < 1)
211                        break;
212                filenames.push_back(string(str, pos, npos - pos));
213                pos = npos + 1;
214        }
215       
216        filenames.push_back(string(str, pos, str.size() - pos));
[440]217        return (int)filenames.size();
[387]218}
219
[750]220
[372]221bool
222Preprocessor::LoadScene(const string filename)
223{
[925]224        // use leaf nodes of the original spatial hierarchy as occludees
[508]225        mSceneGraph = new SceneGraph;
[372]226 
[508]227        Parser *parser;
[387]228        vector<string> filenames;
[1404]229        const int files = SplitFilenames(filename, filenames);
[712]230        cout << "number of input files: " << files << endl;
[387]231        bool result = false;
[1344]232
233        // root for different files
234        mSceneGraph->SetRoot(new SceneGraphNode());
235
[1404]236        // intel ray caster can only trace triangles
[1520]237        int rayCastMethod;
238        Environment::GetSingleton()->GetIntValue("Preprocessor.rayCastMethod", rayCastMethod);
239        vector<FaceParentInfo> *fi = (rayCastMethod == RayCaster::INTEL_RAYCASTER) ?
[1404]240                &mFaceParents : NULL;
241
[387]242        if (files == 1) {
243               
244                if (strstr(filename.c_str(), ".x3d"))
[749]245                  parser = new X3dParser;
[387]246                else
[811]247                  if (strstr(filename.c_str(), ".ply") || strstr(filename.c_str(), ".plb"))
[749]248                        parser = new PlyParser;
[1221]249                  else if (strstr(filename.c_str(), ".obj"))
250                          parser = new ObjParser;
251                  else
252                          parser = new UnigraphicsParser;
[372]253
[1328]254                cout << filename << endl;
[1404]255               
256                result = parser->ParseFile(
[1328]257                                filename,
[1344]258                                mSceneGraph->GetRoot(),
[1379]259                                mLoadMeshes,
[1404]260                                fi);
261                       
[387]262                delete parser;
[372]263
[1404]264        }
265        else {
266                vector<string>::const_iterator fit, fit_end = filenames.end();
[1344]267               
[1404]268                for (fit = filenames.begin(); fit != fit_end; ++ fit)
269                {
270                        const string filename = *fit;
[1328]271
[1404]272                        cout << "parsing file " << filename.c_str() << endl;
273                        if (strstr(filename.c_str(), ".x3d"))
274                                parser = new X3dParser;
275                        else
276                                parser = new UnigraphicsParser;
277
278                        SceneGraphNode *node = new SceneGraphNode();
279                        const bool success = parser->ParseFile(
280                                filename,
281                                node,
282                                mLoadMeshes,
283                                fi);
284
285                        if (success)
286                        {
287                                mSceneGraph->GetRoot()->mChildren.push_back(node);
288                                result = true; // at least one file parsed
289                        }
290
291                        delete parser;
[387]292                }
293        }
[1344]294       
[752]295        if (result)
[1344]296        {
[752]297                // HACK
[1020]298                if (ADDITIONAL_GEOMETRY_HACK)
299                        AddGeometry(mSceneGraph);
[1328]300
[1020]301                mSceneGraph->AssignObjectIds();
[1344]302 
[1020]303                int intersectables, faces;
304                mSceneGraph->GetStatistics(intersectables, faces);
[1344]305 
[1020]306                cout<<filename<<" parsed successfully."<<endl;
307                cout<<"#NUM_OBJECTS (Total numner of objects)\n"<<intersectables<<endl;
308                cout<<"#NUM_FACES (Total numner of faces)\n"<<faces<<endl;
309                mSceneGraph->CollectObjects(&mObjects);
[1328]310                mSceneGraph->GetRoot()->UpdateBox();
[1344]311 
[1020]312                if (0)
313                {
314                        Exporter *exporter = Exporter::GetExporter("testload.x3d");
315                        if (exporter)
316                        {
317                                exporter->ExportGeometry(mObjects);
318                                delete exporter;
319                        }
320                }
[387]321        }
[1328]322
[492]323        return result;
[372]324}
325
326bool
327Preprocessor::ExportPreprocessedData(const string filename)
328{
[1486]329        mViewCellsManager->ExportViewCells(filename, true, mObjects);
330        return true;
[372]331}
332
[1486]333
[372]334bool
[871]335Preprocessor::PostProcessVisibility()
336{
337 
[904]338  if (mApplyVisibilityFilter || mApplyVisibilitySpatialFilter) {
[997]339        cout<<"Applying visibility filter ...";
[1002]340        cout<<"filter width = " << mVisibilityFilterWidth << endl;
[904]341       
[1002]342        if (!mViewCellsManager)
[1199]343          return false;
344       
[871]345        mViewCellsManager->ApplyFilter(mKdTree,
[904]346                                                                   mApplyVisibilityFilter ? mVisibilityFilterWidth : -1.0f,
347                                                                   mApplyVisibilitySpatialFilter ? mVisibilityFilterWidth : -1.0f);
[997]348        cout << "done." << endl;
[871]349  }
350 
351  // export the preprocessed information to a file
352  if (mExportVisibility)
[1486]353  {
354          ExportPreprocessedData(mVisibilityFileName);
355  }
356
[871]357  return true;
358}
359
360
361bool
[372]362Preprocessor::BuildKdTree()
363{
364  mKdTree = new KdTree;
[1344]365
[372]366  // add mesh instances of the scene graph to the root of the tree
367  KdLeaf *root = (KdLeaf *)mKdTree->GetRoot();
[1344]368       
[372]369  mSceneGraph->CollectObjects(&root->mObjects);
[1344]370 
[1415]371  const long startTime = GetTime();
[1201]372  cout << "building kd tree ... " << endl;
[1344]373
[372]374  mKdTree->Construct();
[1344]375
[1415]376  cout << "finished kd tree construction in " << TimeDiff(startTime, GetTime()) * 1e-3
377           << " secs " << endl;
[1344]378
[372]379  return true;
380}
381
[1415]382
[372]383void
384Preprocessor::KdTreeStatistics(ostream &s)
385{
386  s<<mKdTree->GetStatistics();
387}
388
389void
390Preprocessor::BspTreeStatistics(ostream &s)
391{
392        s << mBspTree->GetStatistics();
393}
394
395bool
396Preprocessor::Export( const string filename,
[492]397                                          const bool scene,
[1545]398                                          const bool kdtree
[492]399                                          )
[372]400{
401  Exporter *exporter = Exporter::GetExporter(filename);
402       
403  if (exporter) {
[1545]404    if (2 && scene)
[1328]405      exporter->ExportScene(mSceneGraph->GetRoot());
[372]406
[1545]407    if (1 && kdtree) {
[372]408      exporter->SetWireframe();
409      exporter->ExportKdTree(*mKdTree);
410    }
411
412    delete exporter;
413    return true;
414  }
415
416  return false;
417}
[429]418
[508]419
[463]420bool Preprocessor::PrepareViewCells()
421{
[1523]422        ///////
[577]423        //-- parse view cells construction method
[1563]424
[1004]425        Environment::GetSingleton()->GetBoolValue("ViewCells.loadFromFile", mLoadViewCells);
[577]426        char buf[100];
427       
428        if (mLoadViewCells)
[997]429        {       
[1004]430                Environment::GetSingleton()->GetStringValue("ViewCells.filename", buf);
[1264]431                cout << "loading view cells from " << buf << endl;
[1563]432               
[1004]433                mViewCellsManager = ViewCellsManager::LoadViewCells(buf, &mObjects, true);
[1563]434
[1522]435                if (!mViewCellsManager)
436                        return false;
[577]437        }
438        else
439        {
[1288]440                // parse type of view cell container
[1004]441                Environment::GetSingleton()->GetStringValue("ViewCells.type", buf);             
[1002]442            mViewCellsManager = CreateViewCellsManager(buf);
[1112]443
444                // default view space is the extent of the scene
[1563]445                AxisAlignedBox3 box = mSceneGraph->GetBox();
446                                       
447                if (0)
448                {
449                        // use a small box outside of the scene
450                        box.Scale(Vector3(0.1f,0.5f,0.5f));
451                        box.Translate(Vector3(Magnitude(mSceneGraph->GetBox().Size())*0.5f, 0, 0));
452                }
[1545]453
[1563]454                mViewCellsManager->SetViewSpaceBox(box);
455         
[1545]456                bool loadVcGeometry;
457                Environment::GetSingleton()->GetBoolValue("ViewCells.loadGeometry", loadVcGeometry);
458
459                bool extrudeBaseTriangles;
460                Environment::GetSingleton()->GetBoolValue("ViewCells.useBaseTrianglesAsGeometry", extrudeBaseTriangles);
461
462                char vcGeomFilename[100];
463                Environment::GetSingleton()->GetStringValue("ViewCells.geometryFilename", vcGeomFilename);
464               
465                if (mViewCellsManager->GetType() == ViewCellsManager::BSP)
466                {
467                        if (!mViewCellsManager->LoadViewCellsGeometry(vcGeomFilename, extrudeBaseTriangles))
468                        {
469                                cerr << "loading view cells geometry failed" << endl;
470                        }
471                }
472                else
473                {
474                        cerr << "loading view cells geometry is not implemented for this manager" << endl;
475                }
[577]476        }
[1112]477       
[1486]478        ////////
[1523]479        //-- evaluation of render cost heuristics
[473]480        float objRenderCost = 0, vcOverhead = 0, moveSpeed = 0;
481
[1004]482        Environment::GetSingleton()->GetFloatValue("Simulation.objRenderCost",objRenderCost);
483        Environment::GetSingleton()->GetFloatValue("Simulation.vcOverhead", vcOverhead);
484        Environment::GetSingleton()->GetFloatValue("Simulation.moveSpeed", moveSpeed);
[694]485       
[473]486        mRenderSimulator =
487                new RenderSimulator(mViewCellsManager, objRenderCost, vcOverhead, moveSpeed);
[440]488
[480]489        mViewCellsManager->SetRenderer(mRenderSimulator);
[440]490
[538]491        if (mUseGlRenderer || mUseGlDebugger)
[540]492        {
493                // NOTE: render texture should be power of 2 and square
494                // renderer must be initialised
[1145]495                // $$matt
496//              renderer = new GlRendererBuffer(1024, 768, mSceneGraph, mViewCellsManager, mKdTree);
[556]497                //              renderer->makeCurrent();
[746]498               
[540]499        }
[496]500       
[463]501        return true;
[490]502}
503
[1292]504 
[1563]505bool Preprocessor::ConstructViewCells()
[1292]506{
507        // construct view cells using it's own set of samples
508        mViewCellsManager->Construct(this);
[1486]509
510        // visualizations and statistics
511        Debug << "finished view cells:" << endl;
[1292]512        mViewCellsManager->PrintStatistics(Debug);
[1486]513           
514        return true;
[1292]515}
[490]516
[1294]517
[1288]518HierarchyManager *Preprocessor::CreateHierarchyManager(const char *name)
519{
520        HierarchyManager *hierarchyManager;
521
522        if (strcmp(name, "osp") == 0)
523        {
[1291]524                Debug << "hierarchy manager: osp" << endl;
[1288]525                // HACK for testing if per kd evaluation works!!
[1293]526                const bool ishack = false;
[1288]527                if (ishack)
[1421]528                        hierarchyManager = new HierarchyManager(mKdTree);
[1288]529                else
[1421]530                        hierarchyManager = new HierarchyManager(HierarchyManager::KD_BASED_OBJ_SUBDIV);
[1288]531        }
532        else if (strcmp(name, "bvh") == 0)
533        {
[1291]534                Debug << "hierarchy manager: bvh" << endl;
[1421]535                hierarchyManager = new HierarchyManager(HierarchyManager::BV_BASED_OBJ_SUBDIV);
[1288]536        }
[1291]537        else // only view space partition
538        {
539                Debug << "hierarchy manager: obj" << endl;
[1421]540                hierarchyManager = new HierarchyManager(HierarchyManager::NO_OBJ_SUBDIV);
[1291]541        }
[1288]542
543        return hierarchyManager;
544}
545
546
[575]547ViewCellsManager *Preprocessor::CreateViewCellsManager(const char *name)
548{
[1264]549        ViewCellsTree *vcTree = new ViewCellsTree;
550
[575]551        if (strcmp(name, "kdTree") == 0)
552        {
[1264]553                mViewCellsManager = new KdViewCellsManager(vcTree, mKdTree);
[575]554        }
555        else if (strcmp(name, "bspTree") == 0)
556        {
557                Debug << "view cell type: Bsp" << endl;
558
[577]559                mBspTree = new BspTree();
[1264]560                mViewCellsManager = new BspViewCellsManager(vcTree, mBspTree);
[575]561        }
562        else if (strcmp(name, "vspBspTree") == 0)
563        {
564                Debug << "view cell type: VspBsp" << endl;
565
[1004]566                mVspBspTree = new VspBspTree();
[1264]567                mViewCellsManager = new VspBspViewCellsManager(vcTree, mVspBspTree);
[575]568        }
[1006]569        else if (strcmp(name, "vspOspTree") == 0)
[575]570        {
[1421]571                Debug << "view cell type: VspOsp" << endl;
[1288]572                char buf[100];         
573                Environment::GetSingleton()->GetStringValue("Hierarchy.type", buf);     
[1022]574
[1288]575                HierarchyManager *mHierarchyManager = CreateHierarchyManager(buf);
[1279]576                mViewCellsManager = new VspOspViewCellsManager(vcTree, mHierarchyManager);
[575]577        }
[1421]578        else if (strcmp(name, "sceneDependent") == 0) //TODO
[575]579        {
[1143]580                Debug << "view cell type: Bsp" << endl;
[1421]581               
[575]582                mBspTree = new BspTree();
[1264]583                mViewCellsManager = new BspViewCellsManager(vcTree, mBspTree);
[575]584        }
585        else
586        {
[664]587                cerr << "Wrong view cells type " << name << "!!!" << endl;
[575]588                exit(1);
589        }
590
591        return mViewCellsManager;
592}
593
594
[491]595// use ascii format to store rays
596#define USE_ASCII 0
597
598
[1145]599static inline bool ilt(Intersectable *obj1, Intersectable *obj2)
[490]600{
601        return obj1->mId < obj2->mId;
602}
603
604
[1414]605bool Preprocessor::LoadKdTree(const string filename)
[1197]606{
[1414]607        mKdTree = new KdTree();
608        return mKdTree->LoadBinTree(filename.c_str(), mObjects);
[1197]609}
610
[1414]611
612bool Preprocessor::ExportKdTree(const string filename)
[1197]613{
[1414]614        return mKdTree->ExportBinTree(filename.c_str());
[1197]615}
616
617
[490]618bool Preprocessor::LoadSamples(VssRayContainer &samples,
619                                                           ObjectContainer &objects) const
620{
621        std::stable_sort(objects.begin(), objects.end(), ilt);
622        char fileName[100];
[1004]623        Environment::GetSingleton()->GetStringValue("Preprocessor.samplesFilename", fileName);
[490]624       
[491]625    Vector3 origin, termination;
626        // HACK: needed only for lower_bound algorithm to find the
627        // intersected objects
628        MeshInstance sObj(NULL);
629        MeshInstance tObj(NULL);
[490]630
[491]631#if USE_ASCII
[656]632        ifstream samplesIn(fileName);
[490]633        if (!samplesIn.is_open())
634                return false;
635
636        string buf;
637        while (!(getline(samplesIn, buf)).eof())
638        {
[491]639                sscanf(buf.c_str(), "%f %f %f %f %f %f %d %d",
[490]640                           &origin.x, &origin.y, &origin.z,
[491]641                           &termination.x, &termination.y, &termination.z,
642                           &(sObj.mId), &(tObj.mId));
[490]643               
[491]644                Intersectable *sourceObj = NULL;
645                Intersectable *termObj = NULL;
646               
647                if (sObj.mId >= 0)
[490]648                {
649                        ObjectContainer::iterator oit =
[491]650                                lower_bound(objects.begin(), objects.end(), &sObj, ilt);
651                        sourceObj = *oit;
652                }
653               
654                if (tObj.mId >= 0)
655                {
656                        ObjectContainer::iterator oit =
657                                lower_bound(objects.begin(), objects.end(), &tObj, ilt);
658                        termObj = *oit;
659                }
[490]660
[491]661                samples.push_back(new VssRay(origin, termination, sourceObj, termObj));
662        }
663#else
664        ifstream samplesIn(fileName, ios::binary);
665        if (!samplesIn.is_open())
666                return false;
667
668        while (1)
669        {
670                 samplesIn.read(reinterpret_cast<char *>(&origin), sizeof(Vector3));
671                 samplesIn.read(reinterpret_cast<char *>(&termination), sizeof(Vector3));
672                 samplesIn.read(reinterpret_cast<char *>(&(sObj.mId)), sizeof(int));
673                 samplesIn.read(reinterpret_cast<char *>(&(tObj.mId)), sizeof(int));
674               
675                 if (samplesIn.eof())
676                        break;
677
678                Intersectable *sourceObj = NULL;
679                Intersectable *termObj = NULL;
680               
681                if (sObj.mId >= 0)
682                {
683                        ObjectContainer::iterator oit =
684                                lower_bound(objects.begin(), objects.end(), &sObj, ilt);
685                        sourceObj = *oit;
[490]686                }
[491]687               
688                if (tObj.mId >= 0)
[490]689                {
[491]690                        ObjectContainer::iterator oit =
691                                lower_bound(objects.begin(), objects.end(), &tObj, ilt);
692                        termObj = *oit;
[490]693                }
[491]694
695                samples.push_back(new VssRay(origin, termination, sourceObj, termObj));
[490]696        }
[491]697
698#endif
[490]699        samplesIn.close();
700
701        return true;
702}
703
[508]704
705bool Preprocessor::ExportSamples(const VssRayContainer &samples) const
[490]706{
[491]707        char fileName[100];
[1004]708        Environment::GetSingleton()->GetStringValue("Preprocessor.samplesFilename", fileName);
[491]709       
[490]710
711        VssRayContainer::const_iterator it, it_end = samples.end();
712       
[491]713#if USE_ASCII
714        ofstream samplesOut(fileName);
[490]715        if (!samplesOut.is_open())
716                return false;
717
718        for (it = samples.begin(); it != it_end; ++ it)
719        {
720                VssRay *ray = *it;
[491]721                int sourceid = ray->mOriginObject ? ray->mOriginObject->mId : -1;               
722                int termid = ray->mTerminationObject ? ray->mTerminationObject->mId : -1;       
723
[490]724                samplesOut << ray->GetOrigin().x << " " << ray->GetOrigin().y << " " << ray->GetOrigin().z << " "
725                                   << ray->GetTermination().x << " " << ray->GetTermination().y << " " << ray->GetTermination().z << " "
[491]726                                   << sourceid << " " << termid << "\n";
[490]727        }
[491]728#else
729        ofstream samplesOut(fileName, ios::binary);
730        if (!samplesOut.is_open())
731                return false;
732
733        for (it = samples.begin(); it != it_end; ++ it)
734        {       
735                VssRay *ray = *it;
736                Vector3 origin(ray->GetOrigin());
737                Vector3 termination(ray->GetTermination());
738               
739                int sourceid = ray->mOriginObject ? ray->mOriginObject->mId : -1;               
740                int termid = ray->mTerminationObject ? ray->mTerminationObject->mId : -1;               
741
742                samplesOut.write(reinterpret_cast<char *>(&origin), sizeof(Vector3));
743                samplesOut.write(reinterpret_cast<char *>(&termination), sizeof(Vector3));
744                samplesOut.write(reinterpret_cast<char *>(&sourceid), sizeof(int));
745                samplesOut.write(reinterpret_cast<char *>(&termid), sizeof(int));
746    }
747#endif
[490]748        samplesOut.close();
[1194]749
[490]750        return true;
751}
[563]752
[1020]753bool Preprocessor::GenerateRays(const int number,
754                                                                const int sampleType,
755                                                                SimpleRayContainer &rays)
756{
757        const int startSize = (int)rays.size();
758        SamplingStrategy *strategy = GenerateSamplingStrategy(sampleType);
[860]759
[1020]760        if (!strategy)
[1381]761        {
[1020]762                return false;
[1381]763        }
[1020]764
765        for (int i=0; (int)rays.size() - startSize < number; ++ i)
766        {
767                SimpleRay newRay;
768
[1381]769                if (strategy->GenerateSample(newRay))
770                {
771#if 1
[1020]772                        rays.AddRay(newRay);
[1381]773#else
774                        GenerateRayBundle(rays, newRay, 16, 0);
775#endif
776                }       
[1020]777        }
778
779        delete strategy;
780    return true;
[878]781}
[1020]782
783
784SamplingStrategy *Preprocessor::GenerateSamplingStrategy(const int strategyId) const
785{
786        switch (strategyId)
787        {
[1520]788        case SamplingStrategy::OBJECT_BASED_DISTRIBUTION:
[1020]789                return new ObjectBasedDistribution(*this);
[1520]790        case SamplingStrategy::OBJECT_DIRECTION_BASED_DISTRIBUTION:
[1020]791                return new ObjectDirectionBasedDistribution(*this);
[1520]792        case SamplingStrategy::DIRECTION_BASED_DISTRIBUTION:
[1020]793                return new DirectionBasedDistribution(*this);
[1520]794        case SamplingStrategy::DIRECTION_BOX_BASED_DISTRIBUTION:
[1020]795                return new DirectionBoxBasedDistribution(*this);
[1520]796        case SamplingStrategy::SPATIAL_BOX_BASED_DISTRIBUTION:
[1020]797                return new SpatialBoxBasedDistribution(*this);
[1021]798        //case OBJECTS_INTERIOR_DISTRIBUTION:
799        //      return new ObjectsInteriorDistribution(*this);
[1020]800        default: // no valid strategy
[1279]801                Debug << "warning: no valid sampling strategy" << endl;
[1020]802                return NULL;
803        }
[1221]804
[1570]805        return NULL; // should never come here
[1020]806}
807
808
[1418]809bool Preprocessor::InitRayCast(const string externKdTree, const string internkdtree)
[1221]810{
[1418]811        bool loadKdTree;
[1415]812        Environment::GetSingleton()->GetBoolValue("Preprocessor.loadKdTree", loadKdTree);
813       
814        if (!loadKdTree)
[1418]815        {       
[1415]816                //-- build new kd tree from scene geometry
817                BuildKdTree();
818                KdTreeStatistics(cout);
819        }
820        else
821        {
822                const long startTime = GetTime();
[1418]823                cout << "loading kd tree file " << internkdtree << " ... ";
[1415]824
[1418]825                if (!LoadKdTree(internkdtree))
[1415]826                {
[1418]827                        cout << "error loading kd tree with filename " << internkdtree << ", rebuilding it instead ..." << endl;
828
829                        BuildKdTree();
830                        KdTreeStatistics(cout);
[1415]831                }
[1416]832
[1415]833                cout << "finished in " << TimeDiff(startTime, GetTime()) * 1e-3 << " secs" << endl;
834
835                if (0)
836                {
837                        Exporter *exporter = Exporter::GetExporter("dummykd.x3d");
838                       
839                        if (exporter)
840                        {
841                                exporter->ExportKdTree(*mKdTree, true);
842                                delete exporter;
843                        }
844                }
845        }
846
[1520]847        int rayCastMethod;
848        Environment::GetSingleton()->GetIntValue("Preprocessor.rayCastMethod", rayCastMethod);
849
850        if (rayCastMethod == 0)
[1221]851        {
[1520]852                mRayCaster = new InternalRayCaster(*this, mKdTree);
853                cout << "ray cast method: internal" << endl;
[1221]854        }
[1251]855        else
[1520]856        {
[1524]857#ifdef GTP_INTERNAL
[1520]858                mRayCaster = new IntelRayCaster(*this, externKdTree);
859                cout << "ray cast method: intel" << endl;
[1524]860#endif
[1251]861        }
862
[1520]863        return true;
[1281]864}
865
[1292]866
[1281]867void
868Preprocessor::CastRays(
869                                           SimpleRayContainer &rays,
[1520]870                                           VssRayContainer &vssRays,
[1528]871                                           const bool castDoubleRays,
872                                           const bool pruneInvalidRays
[1281]873                                           )
874{
[1524]875        const long t1 = GetTime();
[1344]876
[1524]877        for (int i = 0; i < (int)rays.size();)
[1292]878        {
[1524]879                if (i + 16 < (int)rays.size())
[1520]880                {
[1524]881                        mRayCaster->CastRays16(
882                                i,
883                                rays,                           
884                                vssRays,
885                                mViewCellsManager->GetViewSpaceBox(),
[1528]886                                castDoubleRays,
887                                pruneInvalidRays);
[1524]888                        i += 16;
[1292]889                }
[1524]890                else
891                {
892                        mRayCaster->CastRay(
[1528]893                                rays[i],
[1524]894                                vssRays,
895                                mViewCellsManager->GetViewSpaceBox(),
[1528]896                                castDoubleRays,
897                                pruneInvalidRays);
[1524]898                        i ++;
899                }
900        }
[1294]901
[1524]902#if DEBUB_RAYCAST
903        long t2 = GetTime();
904        if (i % 10000 == 0)
905                cout<<".";
[1286]906
[1524]907        if (castDoubleRays)
908                cout << 2 * rays.size() / (1e3f * TimeDiff(t1, t2)) << "M rays/s" << endl;
909        else
910                cout << rays.size() / (1e3f * TimeDiff(t1, t2)) << "M rays/s" << endl;
911#endif 
[1251]912}
913
914
[1381]915bool Preprocessor::GenerateRayBundle(SimpleRayContainer &rayBundle,                                                                     
916                                                                         const SimpleRay &mainRay,
917                                                                         const int number,
918                                                                         const int pertubType) const
919{
920        rayBundle.push_back(mainRay);
921
[1564]922        const float pertubOrigin = 0.0f;
923        const float pertubDir = 0.2f;
[1381]924
925        for (int i = 0; i < number - 1; ++ i)
926        {
927                Vector3 pertub;
928
929                pertub.x = RandomValue(0.0f, pertubDir);
930                pertub.y = RandomValue(0.0f, pertubDir);
931                pertub.z = RandomValue(0.0f, pertubDir);
932                const Vector3 newDir = mainRay.mDirection + pertub;
933                //const Vector3 newDir = mainRay.mDirection;
934
935                pertub.x = RandomValue(0.0f, pertubOrigin);
936                pertub.y = RandomValue(0.0f, pertubOrigin);
937                pertub.z = RandomValue(0.0f, pertubOrigin);
938                const Vector3 newOrigin = mainRay.mOrigin + pertub;
939                //const Vector3 newOrigin = mainRay.mOrigin;
940
941                rayBundle.push_back(SimpleRay(newOrigin, newDir, 0));
942        }
943
944        return true;
945}
946
947
[1251]948void Preprocessor::SetupRay(Ray &ray,
949                                                        const Vector3 &point,
950                                                        const Vector3 &direction
[1520]951                                                        ) const
[1251]952{
953        ray.Clear();
954        // do not store anything else then intersections at the ray
955        ray.Init(point, direction, Ray::LOCAL_RAY);
956}
957
[1520]958
[1251]959}
Note: See TracBrowser for help on using the repository browser.