source: trunk/VUT/GtpVisibilityPreprocessor/src/SamplingPreprocessor.cpp @ 374

Revision 374, 24.4 KB checked in by bittner, 19 years ago (diff)
Line 
1#include "SceneGraph.h"
2#include "KdTree.h"
3#include "SamplingPreprocessor.h"
4#include "X3dExporter.h"
5#include "Environment.h"
6#include "MutualVisibility.h"
7#include "Polygon3.h"
8#include "ViewCell.h"
9
10SamplingPreprocessor::SamplingPreprocessor(): mPass(0), mSampleRays(NULL)
11{
12  // this should increase coherence of the samples
13  environment->GetIntValue("Sampling.samplesPerPass", mSamplesPerPass);
14  environment->GetIntValue("Sampling.totalSamples", mTotalSamples);
15  environment->GetIntValue("BspTree.Construction.samples", mBspConstructionSamples);
16  environment->GetIntValue("BspTree.PostProcessing.samples", mPostProcessSamples);
17  mKdPvsDepth = 100;
18  mStats.open("stats.log");
19
20}
21
22SamplingPreprocessor::~SamplingPreprocessor()
23{
24        CLEAR_CONTAINER(mSampleRays);
25}
26
27void
28SamplingPreprocessor::SetupRay(Ray &ray,
29                                                                                                                         const Vector3 &point,
30                                                                                                                         const Vector3 &direction,
31                                                                                                                         const int type)
32{
33  ray.intersections.clear();
34  ray.kdLeaves.clear();
35  ray.testedObjects.clear();
36  ray.bspIntersections.clear();
37        ray.mFlags |= Ray::STORE_KDLEAVES | Ray::STORE_BSP_INTERSECTIONS;
38  //  cout<<point<<" "<<direction<<endl;
39  ray.Init(point, direction, type);
40}
41
42KdNode *
43SamplingPreprocessor::GetNodeForPvs(KdLeaf *leaf)
44{
45  KdNode *node = leaf;
46  while (node->mParent && node->mDepth > mKdPvsDepth)
47    node = node->mParent;
48  return node;
49}
50
51bool
52SamplingPreprocessor::BuildBspTree()
53{
54        // delete old tree
55        DEL_PTR(mBspTree);
56        mBspTree = new BspTree(&mUnbounded);
57        ObjectContainer objects;
58       
59        switch (BspTree::sConstructionMethod)
60        {
61        case BspTree::FROM_INPUT_VIEW_CELLS:
62                mBspTree->SetGenerateViewCells(false);
63                mBspTree->Construct(mViewCells);
64                break;
65        case BspTree::FROM_SCENE_GEOMETRY:
66                DeleteViewCells(); // we generate new view cells
67                mBspTree->SetGenerateViewCells(true);
68                mSceneGraph->CollectObjects(&objects);
69                mBspTree->Construct(objects);
70                break;
71        case BspTree::FROM_RAYS:
72                DeleteViewCells(); // we generate new view cells
73                mBspTree->SetGenerateViewCells(true);
74                mBspTree->Construct(mSampleRays);
75                break;
76        default:
77                Debug << "Error: Method not available\n";
78                break;
79        }
80       
81       
82        return true;
83}
84
85int
86SamplingPreprocessor::AddNodeSamples(const Ray &ray,
87                                                                                                                                                 Intersectable *sObject,
88                                                                                                                                                 Intersectable *tObject
89                                                                                                                                                 )
90{
91  int contributingSamples = 0;
92  int j;
93        int objects = 0;
94        if (sObject)
95                objects++;
96        if (tObject)
97                objects++;
98
99        if (objects) {
100                for (j=0; j < ray.kdLeaves.size(); j++) {
101                        KdNode *node = GetNodeForPvs( ray.kdLeaves[j] );
102                        if (sObject)
103                                contributingSamples += sObject->mKdPvs.AddSample(node);
104                        if (tObject)
105                                contributingSamples += tObject->mKdPvs.AddSample(node);
106                }
107        }
108       
109        for (j=1; j < ((int)ray.kdLeaves.size() - 1); j++) {
110                ray.kdLeaves[j]->AddPassingRay2(ray,
111                                                                                                                                                objects,
112                                                                                                                                                ray.kdLeaves.size()
113                                                                                                                                                );
114  }
115 
116  return contributingSamples;
117}
118
119
120int SamplingPreprocessor::AddObjectSamples(Intersectable *obj, const Ray &ray)
121{
122        int contributingSamples = 0;
123        int j;
124 
125        // object can be seen from the view cell => add to view cell pvs
126        for (j=0; j < ray.bspIntersections.size(); ++ j)
127        {       
128                BspLeaf *leaf = ray.bspIntersections[j].mLeaf;
129                // if ray not in unbounded space
130                if (leaf->GetViewCell() != &mUnbounded)
131                        contributingSamples +=
132                                leaf->GetViewCell()->GetPvs().AddSample(obj);
133        }
134 
135        // rays passing through this viewcell
136        if (mPass > 1)
137                for (j=1; j < ((int)ray.bspIntersections.size() - 1); ++ j)
138                {
139                        BspLeaf *leaf = ray.bspIntersections[j].mLeaf;
140
141                        if (leaf->GetViewCell() != &mUnbounded)
142                                leaf->GetViewCell()->
143                                        AddPassingRay(ray, contributingSamples ? 1 : 0);
144                }
145 
146        return contributingSamples;
147}
148
149
150void
151SamplingPreprocessor::HoleSamplingPass()
152{
153  vector<KdLeaf *> leaves;
154  mKdTree->CollectLeaves(leaves);
155 
156  // go through all the leaves and evaluate their passing contribution
157  for (int i=0 ; i < leaves.size(); i++) {
158    KdLeaf *leaf = leaves[i];
159    cout<<leaf->mPassingRays<<endl;
160  }
161}
162
163
164int
165SamplingPreprocessor::CastRay(Intersectable *object,
166                                                                                                                        Ray &ray
167                                                                                                                        )
168{
169        int sampleContributions = 0;
170
171        long t1 = GetRealTime();
172        // cast ray to KD tree to find intersection with other objects
173        mKdTree->CastRay(ray);
174        long t2 = GetRealTime();
175
176        if (0 && object && object->GetId() > 2197) {
177                object->Describe(cout)<<endl;
178                cout<<ray<<endl;
179        }
180
181        if (ViewCell::sHierarchy == ViewCell::BSP)
182        {
183                // cast ray to BSP tree to get intersection with view cells
184                if (mBspTree)
185                {
186                        mBspTree->CastRay(ray);
187                       
188                        if (object)
189                                sampleContributions += AddObjectSamples(object, ray);
190                       
191                        if (!ray.intersections.empty()) // second intersection found
192                                {
193                                        sampleContributions +=
194                                                AddObjectSamples(ray.intersections[0].mObject, ray);
195                                }
196                }
197        }
198        else
199                {
200                        if (ray.kdLeaves.size()) {
201                                Intersectable *terminator =
202                                        ray.intersections.size() ? ray.intersections[0].mObject: NULL;
203
204                                sampleContributions += AddNodeSamples(ray,
205                                                                                                                                                                                        object,
206                                                                                                                                                                                        terminator);
207                        }
208                }
209       
210        return sampleContributions;
211}
212
213//  void
214//  SamplingPreprocessor::AvsGenerateRandomRay(Ray &ray)
215//  {
216//    int objId = RandomValue(0, mObjects.size());
217//    Intersectable *object = objects[objId];
218//    object->GetRandomSurfacePoint(point, normal);
219//    direction = UniformRandomVector(normal);
220//    SetupRay(ray, point, direction);
221//  }
222
223//  void
224//  SamplingPreprocessor::AvsHandleRay(Ray &ray)
225//  {
226//    int sampleContributions = 0;
227
228//    mKdTree->CastRay(ray);
229 
230//    if (ray.leaves.size()) {
231//      sampleContributions += AddNodeSamples(object, ray, pass);
232   
233//      if (ray.intersections.size()) {
234//        sampleContributions += AddNodeSamples(ray.intersections[0].mObject, ray, pass);
235//      }
236//    }
237//  }
238
239//  void
240//  SamplingPreprocessor::AvsBorderSampling(Ray &ray)
241//  {
242 
243
244//  }
245
246//  void
247//  SamplingPreprocessor::AvsPass()
248//  {
249//    Ray ray;
250//    while (1) {
251//      AvsGenerateRay(ray);
252//      HandleRay(ray);
253//      while ( !mRayQueue.empty() ) {
254//        Ray ray = mRayQueue.pop();
255//        mRayQueue.pop();
256//        AdaptiveBorderSampling(ray);
257//      }
258//    }
259 
260 
261
262//  }
263
264
265int
266SamplingPreprocessor::CastEdgeSamples(
267                                                                                                                                                        Intersectable *object,
268                                                                                                                                                        const Vector3 &point,
269                                                                                                                                                        MeshInstance *mi,
270                                                                                                                                                        const int samples
271                                                                                                                                                        )
272{
273        Ray ray;
274        int maxTries = samples*10;
275        int i;
276        int rays = 0;
277        int edgeSamplesContributions = 0;
278        for (i=0; i < maxTries && rays < samples; i++) {
279                // pickup a random face of each mesh
280                Mesh *mesh = mi->GetMesh();
281                int face = RandomValue(0, mesh->mFaces.size()-1);
282               
283                Polygon3 poly(mesh->mFaces[face], mesh);
284                poly.Scale(1.001);
285                // now extend a random edge of the face
286                int edge = RandomValue(0, poly.mVertices.size()-1);
287                float t = RandomValue(0.0f,1.0f);
288                Vector3 target = t*poly.mVertices[edge] + (1.0f-t)*poly.mVertices[(edge + 1)%
289                                                                                                                                                                                                                                                                                 poly.mVertices.size()];
290                SetupRay(ray, point, target - point, Ray::LOCAL_RAY);
291                if (!mesh->CastRay(ray, mi)) {
292                        // the rays which intersect the mesh have been discarded since they are not tangent
293                        // to the mesh
294                        rays++;
295                        edgeSamplesContributions += CastRay(object, ray);
296                }
297        }
298        return edgeSamplesContributions;
299}
300
301KdNode *
302SamplingPreprocessor::GetNodeToSample(Intersectable *object)
303{
304        int pvsSize = object->mKdPvs.GetSize();
305        KdNode *nodeToSample = NULL;
306       
307        bool samplePvsBoundary = false;
308        if (pvsSize && samplePvsBoundary) {
309                // this samples the nodes from the boundary of the current PVS
310                // mail all nodes from the pvs
311                Intersectable::NewMail();
312                KdPvsMap::iterator i = object->mKdPvs.mEntries.begin();
313               
314                for (; i != object->mKdPvs.mEntries.end(); i++) {
315                        KdNode *node = (*i).first;
316                        node->Mail();
317                }
318               
319                int maxTries = 2*pvsSize;
320                Debug << "Finding random neighbour" << endl;   
321                for (int tries = 0; tries < 10; tries++) {
322                        int index = RandomValue(0, pvsSize - 1);
323                        KdPvsData data;
324                        KdNode *node;
325                        object->mKdPvs.GetData(index, node, data);
326                        nodeToSample = mKdTree->FindRandomNeighbor(node, true);
327                        if (nodeToSample)
328                                break;
329                }
330        } else {
331                // just pickup a random node
332                //              nodeToSample = mKdTree->GetRandomLeaf(Plane3(normal, point));
333                nodeToSample = mKdTree->GetRandomLeaf();
334        }
335        return nodeToSample;
336}
337
338void
339SamplingPreprocessor::VerifyVisibility(Intersectable *object)
340{
341        // mail all nodes from the pvs
342        Intersectable::NewMail();
343        KdPvsMap::iterator i = object->mKdPvs.mEntries.begin();
344        for (; i != object->mKdPvs.mEntries.end(); i++) {
345                KdNode *node = (*i).first;
346                node->Mail();
347        }
348        Debug << "Get all neighbours from PVS" << endl;
349        vector<KdNode *> invisibleNeighbors;
350        // get all neighbors of all PVS nodes
351        i = object->mKdPvs.mEntries.begin();
352        for (; i != object->mKdPvs.mEntries.end(); i++) {
353                KdNode *node = (*i).first;
354                mKdTree->FindNeighbors(node, invisibleNeighbors, true);
355                AxisAlignedBox3 box = object->GetBox();
356                for (int j=0; j < invisibleNeighbors.size(); j++) {
357                        int visibility = ComputeBoxVisibility(mSceneGraph,
358                                                              mKdTree,
359                                                              box,
360                                                              mKdTree->GetBox(invisibleNeighbors[j]),
361                                                              1e-6f);
362                        //            exit(0);
363                }
364                // now rank all the neighbors according to probability that a new
365                // sample creates some contribution
366        }
367}
368
369bool
370SamplingPreprocessor::ComputeVisibility()
371{
372 
373  // pickup an object
374  ObjectContainer objects;
375 
376  mSceneGraph->CollectObjects(&objects);
377
378  Vector3 point, normal, direction;
379  Ray ray;
380
381  long startTime = GetTime();
382 
383  int i;
384  int totalSamples = 0;
385
386  int pvsOut = Min((int)objects.size(), 10);
387
388  vector<Ray> rays[10];
389
390  while (totalSamples < mTotalSamples) {
391                int passContributingSamples = 0;
392                int passSampleContributions = 0;
393                int passSamples = 0;
394                int index = 0;
395               
396                int reverseSamples = 0;
397               
398                       
399                //cout << "totalSamples: "  << totalSamples << endl;
400
401                for (i = 0; i < objects.size(); i++) {
402                                               
403                        KdNode *nodeToSample = NULL;
404                        Intersectable *object = objects[i];
405               
406                        int pvsSize = 0;
407                        if (ViewCell::sHierarchy == ViewCell::KD)
408                                pvsSize = object->mKdPvs.GetSize();
409                                               
410                       
411                        if (0 && pvsSize && mPass == 1000 ) {
412                                VerifyVisibility(object);
413                        }
414                       
415                        int faceIndex = object->GetRandomSurfacePoint(point, normal);
416                       
417                        bool viewcellSample = true;
418                        int sampleContributions;
419                        bool debug = false; //(object->GetId() >= 2199);
420                        if (viewcellSample) {
421                                //mKdTree->GetRandomLeaf(Plane3(normal, point));
422
423                                nodeToSample = GetNodeToSample(object);
424
425                                for (int k=0; k < mSamplesPerPass; k++) {
426                                        bool reverseSample = false;
427                                       
428
429                                        if (nodeToSample) {
430                                                AxisAlignedBox3 box = mKdTree->GetBox(nodeToSample);
431                                                Vector3 pointToSample = box.GetRandomPoint();
432                                                //                                              pointToSample.y = 0.9*box.Min().y + 0.1*box.Max().y;
433                                                if (object->GetRandomVisibleSurfacePoint( point, normal, pointToSample, 3 )) {
434                                                        direction = pointToSample - point;
435                                                } else {
436                                                        reverseSamples++;
437                                                        reverseSample = true;
438                                                        direction = point - pointToSample;
439                                                        point = pointToSample;
440                                                }
441                                        }
442                                        else {
443                                                direction = UniformRandomVector(normal);
444                                        }
445                                       
446                                        // construct a ray
447                                        SetupRay(ray, point, direction, Ray::LOCAL_RAY);
448                                       
449                                        sampleContributions = CastRay(reverseSample ? NULL : object, ray);
450                                       
451                                        //-- CORR matt: put block inside loop
452                                        if (sampleContributions) {
453                                                passContributingSamples ++;
454                                                passSampleContributions += sampleContributions;
455                                        }
456
457                                        if ( i < pvsOut )
458                                                rays[i].push_back(ray);
459               
460                                        if (!ray.intersections.empty()) {
461                                                // check whether we can add this to the rays
462                                                for (int j = 0; j < pvsOut; j++) {
463                                                        if (objects[j] == ray.intersections[0].mObject) {
464                                                                rays[j].push_back(ray);
465                                                        }
466                                                }
467                                        }
468                                        //-------------------
469                                        if (ViewCell::sHierarchy == ViewCell::BSP)
470                                        {
471                                                ProcessBspViewCells(ray,
472                                                                                        object,
473                                                                                        faceIndex,
474                                                                                        passContributingSamples,
475                                                                                        passSampleContributions);
476                                        }
477                                }
478                        } else {
479                                // edge samples
480                                // get random visible mesh
481                                //                              object->GetRandomVisibleMesh(Plane3(normal, point));
482                        }
483       
484                        // CORR matt: must add all samples
485                        passSamples += mSamplesPerPass;
486                }
487       
488                totalSamples += passSamples;
489               
490                //    if (pass>10)
491                //      HoleSamplingPass();
492   
493                mPass++;
494
495                int pvsSize = 0;
496       
497                if (ViewCell::sHierarchy == ViewCell::BSP) {
498                        for (i=0; i < mViewCells.size(); i++) {
499                                ViewCell *vc = mViewCells[i];
500                                pvsSize += vc->GetPvs().GetSize();
501                        }
502                } else  {
503                        for (i=0; i < objects.size(); i++) {
504                                Intersectable *object = objects[i];
505                                pvsSize += object->mKdPvs.GetSize();
506                        }
507                }
508
509                float avgRayContrib = (passContributingSamples > 0) ?
510                        passSampleContributions/(float)passContributingSamples : 0;
511
512                cout << "#Pass " << mPass << " : t = " << TimeDiff(startTime, GetTime())*1e-3 << "s" << endl;
513                cout << "#TotalSamples=" << totalSamples/1000
514                                 << "k   #SampleContributions=" << passSampleContributions << " ("
515                                 << 100*passContributingSamples/(float)passSamples<<"%)" << " avgPVS="
516                                 << pvsSize/(float)objects.size() << endl
517                                 << "avg ray contrib=" << avgRayContrib << endl
518                                 << "reverse samples [%]" << reverseSamples/(float)passSamples*100.0f << endl;
519
520                mStats <<
521                        "#Pass\n" <<mPass<<endl<<
522                        "#Time\n" << TimeDiff(startTime, GetTime())*1e-3 << endl<<
523                        "#TotalSamples\n" << totalSamples<< endl<<
524                        "#SampleContributions\n" << passSampleContributions << endl <<
525                        "#PContributingSamples\n"<<100*passContributingSamples/(float)passSamples<<endl <<
526                        "#AvgPVS\n"<< pvsSize/(float)objects.size() << endl <<
527                        "#AvgRayContrib\n" << avgRayContrib << endl;
528        }
529       
530        if (ViewCell::sHierarchy == ViewCell::KD)       
531                cout << "#totalPvsSize=" << mKdTree->CollectLeafPvs() << endl;
532 
533        if (mBspTree)
534        {
535                cout << "starting post processing using " << mSampleRays.size() << " samples ... ";
536
537                Debug << "original pvs size: " << mBspTree->CountViewCellPvs() << endl;
538
539                // merge or subdivide view cells
540                long startTime = GetTime();
541                int merged = PostprocessViewCells(mSampleRays);
542               
543                cout << "finished" << endl;
544                cout << "merged " << merged << " view cells in "
545                         << TimeDiff(startTime, GetTime()) *1e-3 << " secs" << endl;
546
547                //-- recount pvs
548                Debug << "merged pvs size: " << mBspTree->CountViewCellPvs() << endl;
549
550
551                cout << "evaluating render time heuristics ... ";
552                Real rt = SimulateRendering();
553                cout << "finished" << endl;
554
555                Debug << "render time: " << rt * 1e-3 << endl;
556        }
557       
558  //  HoleSamplingPass();
559        if (0) {
560                Exporter *exporter = Exporter::GetExporter("ray-density.x3d");
561                exporter->SetExportRayDensity(true);
562                exporter->ExportKdTree(*mKdTree);
563
564                if (mBspTree && (ViewCell::sHierarchy == ViewCell::BSP))
565                        exporter->ExportBspTree(*mBspTree);
566
567                delete exporter;
568        }
569 
570        bool exportRays = false;
571        if (exportRays) {
572                Exporter *exporter = NULL;
573                exporter = Exporter::GetExporter("sample-rays.x3d");
574                exporter->SetWireframe();
575                exporter->ExportKdTree(*mKdTree);
576                exporter->ExportBspTree(*mBspTree);
577
578                for (i=0; i < pvsOut; i++)
579                exporter->ExportRays(rays[i], 1000, RgbColor(1, 0, 0));
580                exporter->SetFilled();
581                 
582                delete exporter;
583        }
584
585        //-- several visualizations and statistics
586        if (1) {
587                if (mBspTree && (ViewCell::sHierarchy == ViewCell::BSP))
588        {
589                bool exportSplits = false;
590                environment->GetBoolValue("BspTree.exportSplits", exportSplits);
591
592                // export the bsp splits
593                if (exportSplits)
594                        ExportSplits(objects);
595               
596                ExportBspPvs(objects);
597        }
598   for (int k=0; k < pvsOut; k++) {
599                 Intersectable *object = objects[k];
600                 char s[64];
601                 sprintf(s, "sample-pvs%04d.x3d", k);
602                 Exporter *exporter = Exporter::GetExporter(s);
603                 exporter->SetWireframe();
604                 
605                 
606                 KdPvsMap::iterator i = object->mKdPvs.mEntries.begin();
607                 Intersectable::NewMail();
608                 
609                 // avoid adding the object to the list
610                 object->Mail();
611                 ObjectContainer visibleObjects;
612                 
613                 for (; i != object->mKdPvs.mEntries.end(); i++)
614                         {
615                                 KdNode *node = (*i).first;
616                                 exporter->ExportBox(mKdTree->GetBox(node));
617                                 mKdTree->CollectObjects(node, visibleObjects);
618                         }
619                 
620                 exporter->ExportRays(rays[k], 1000, RgbColor(0, 1, 0));
621                 exporter->SetFilled();
622                 
623                 for (int j = 0; j < visibleObjects.size(); j++)
624                         exporter->ExportIntersectable(visibleObjects[j]);
625                 
626                 
627                 Material m;
628                 m.mDiffuseColor = RgbColor(1, 0, 0);
629                 exporter->SetForcedMaterial(m);
630                 exporter->ExportIntersectable(object);
631                 
632                 delete exporter;
633         }
634  }
635 
636  return true;
637}
638
639bool SamplingPreprocessor::ProcessBspViewCells(Ray &ray,
640                                                                                           Intersectable *object,
641                                                                                           int faceIndex,
642                                                                                           int &contributingSamples,
643                                                                                           int &sampleContributions)
644{
645        // save rays for bsp tree construction
646        if ((BspTree::sConstructionMethod == BspTree::FROM_RAYS) &&
647                (mSampleRays.size() < mBspConstructionSamples))
648        {
649                // also add origin to sample in order to extract it as input polygons
650                MeshInstance *mi = dynamic_cast<MeshInstance *>(object);
651                ray.sourceObject = Ray::Intersection(0.0, mi, faceIndex);
652                                                       
653        mSampleRays.push_back(new Ray(ray));
654               
655                return false;
656        }
657       
658        if (!mBspTree) // construct BSP tree using the samples
659        {
660        cout << "building bsp tree from " << mSampleRays.size() << " samples " << endl;
661                BuildBspTree();
662               
663                // add contributions of saved samples to PVS
664                contributingSamples += mBspTree->GetStat().contributingSamples;
665                sampleContributions += mBspTree->GetStat().sampleContributions;
666
667                BspTreeStatistics(Debug);       
668
669                if (0) Export("vc_bsptree.x3d", false, false, true);
670
671                // throw away samples because BSP leaves not stored in order
672                // Need ordered rays for post processing => collect new rays
673                CLEAR_CONTAINER(mSampleRays);
674        }
675       
676        if ((int)mSampleRays.size() < mPostProcessSamples)
677        {
678                mSampleRays.push_back(new Ray(ray));
679        }
680
681        return true;
682}
683
684// merge or subdivide view cells
685int SamplingPreprocessor::PostprocessViewCells(const RayContainer &rays)
686{
687        int merged = 0;
688        RayContainer::const_iterator rit, rit_end = rays.end();
689        vector<Ray::BspIntersection>::const_iterator iit;
690
691        for (rit = rays.begin(); rit != rays.end(); ++ rit)
692        { 
693                // traverse leaves stored in the rays and compare and merge consecutive
694                // leaves (i.e., the neighbors in the tree)
695                if ((*rit)->bspIntersections.empty())
696                        continue;
697
698                iit = (*rit)->bspIntersections.begin();
699
700                BspLeaf *previousLeaf = (*iit).mLeaf;
701                ++ iit;
702               
703                for (; iit != (*rit)->bspIntersections.end(); ++ iit)
704                {
705                        BspLeaf *leaf = (*iit).mLeaf;
706
707                        if (mBspTree->ShouldMerge(leaf, previousLeaf))
708                        {                       
709                                mBspTree->MergeViewCells(leaf, previousLeaf);
710                                ++ merged;
711                        }
712                        previousLeaf = leaf;
713                }
714        }
715
716        return merged;
717}
718
719void SamplingPreprocessor::ExportSplits(const ObjectContainer &objects)
720{
721        Exporter *exporter = Exporter::GetExporter("bsp_splits.x3d");
722
723        if (exporter)
724        {       
725                cout << "exporting splits ... ";
726
727                Material m;
728                m.mDiffuseColor = RgbColor(1, 0, 0);
729                exporter->SetForcedMaterial(m);
730                exporter->SetWireframe();
731                exporter->ExportBspSplits(*mBspTree);
732
733                // take forced material, else big scenes cannot be viewed
734                m.mDiffuseColor = RgbColor(0, 1, 0);
735                exporter->SetForcedMaterial(m);
736                exporter->SetFilled();
737
738                exporter->ResetForcedMaterial();
739               
740                // export rays
741                if (0)
742                {
743                        RayContainer outRays;
744               
745                        for (int i = 0; i < mSampleRays.size(); ++ i)
746                        {
747                                // only rays piercing geometry
748                                if (!mSampleRays[i]->intersections.empty())
749                                        outRays.push_back(mSampleRays[i]);
750                        }
751                        if (BspTree::sConstructionMethod == BspTree::FROM_RAYS)
752                        {
753                                // export rays
754                                exporter->ExportRays(outRays, 1000, RgbColor(1, 1, 0));
755                        }
756                }
757
758                // export scene geometry
759                if (0)
760                {
761                        Material m;//= RandomMaterial();
762                        m.mDiffuseColor = RgbColor(0, 1, 0);
763                        exporter->SetForcedMaterial(m);
764            exporter->SetWireframe();
765
766                        for (int j = 0; j < objects.size(); ++ j)
767                                exporter->ExportIntersectable(objects[j]);
768                }
769
770                delete exporter;
771
772                cout << "finished" << endl;
773        }
774}
775
776void SamplingPreprocessor::ExportBspPvs(const ObjectContainer &objects)
777{
778        //-- some random view cells and rays for output
779        const int leafOut = 10;
780
781        vector<Ray> vcRays[leafOut];
782        vector<BspLeaf *> bspLeaves;
783                       
784        for (int i = 0; i < leafOut; ++ i)
785                bspLeaves.push_back(mBspTree->GetRandomLeaf());
786       
787        const int raysOut = min((int)mSampleRays.size(), 20000);
788
789        // check whether we can add the current ray to the output rays
790        for (int k = 0; k < raysOut; ++ k)
791        {
792                Ray *ray = mSampleRays[k];
793
794                for (int j = 0; j < (int)ray->bspIntersections.size(); ++ j)
795                {
796                        BspLeaf *leaf = ray->bspIntersections[j].mLeaf;
797
798                        for (int i = 0; i < (int)bspLeaves.size(); ++ i)
799                        {
800                                if (bspLeaves[i]->GetViewCell() == leaf->GetViewCell())
801                                {
802                                        vcRays[i].push_back(*ray);
803                                }
804                        }
805                }
806        }
807
808        ViewCell::NewMail();
809
810        for (int i = 0; i < bspLeaves.size(); ++ i)
811        {
812                Intersectable::NewMail();
813
814                ViewCell *vc = bspLeaves[i]->GetViewCell();
815
816                //bspLeaves[j]->Mail();
817                char s[64]; sprintf(s, "bsp-pvs%04d.x3d", i);
818
819                Exporter *exporter = Exporter::GetExporter(s);
820                exporter->SetFilled();
821
822                ViewCellPvsMap::iterator it = vc->GetPvs().mEntries.begin();
823
824                exporter->SetWireframe();
825
826                Material m;//= RandomMaterial();
827                m.mDiffuseColor = RgbColor(0, 1, 0);
828                exporter->SetForcedMaterial(m);
829
830                if (vc->GetMesh())
831                        exporter->ExportViewCell(vc);
832                else
833                {
834            PolygonContainer cell;
835                        // export view cell
836                        mBspTree->ConstructGeometry(bspLeaves[i]->GetViewCell(), cell);
837                        exporter->ExportPolygons(cell);
838                        CLEAR_CONTAINER(cell);
839                }
840
841                Debug << i << ": pvs size=" << (int)vc->GetPvs().GetSize()
842                                << ", piercing rays=" << (int)vcRays[i].size() << endl;
843
844                // export view cells
845                if (0)
846                {
847                        m.mDiffuseColor = RgbColor(1, 0, 1);
848                        exporter->SetForcedMaterial(m);
849                        exporter->ExportViewCells(mViewCells);
850                }
851
852                // export rays piercing this view cell
853                exporter->ExportRays(vcRays[i], 1000, RgbColor(0, 1, 0));
854
855                m.mDiffuseColor = RgbColor(1, 0, 0);
856                exporter->SetForcedMaterial(m);
857
858                // output PVS of view cell
859                for (; it != vc->GetPvs().mEntries.end(); ++ it)
860                {
861                        Intersectable *intersect = (*it).first;
862                        if (!intersect->Mailed())
863                        {
864                                exporter->ExportIntersectable(intersect);
865                                intersect->Mail();
866                        }                       
867                }
868                       
869                // output rest of the objects
870                if (0)
871                {
872                        Material m;//= RandomMaterial();
873                        m.mDiffuseColor = RgbColor(0, 0, 1);
874                        exporter->SetForcedMaterial(m);
875
876                        for (int j = 0; j < objects.size(); ++ j)
877                                if (!objects[j]->Mailed())
878                                {
879                                        exporter->SetForcedMaterial(m);
880                                        exporter->ExportIntersectable(objects[j]);
881                                        objects[j]->Mail();
882                                }
883                }
884                DEL_PTR(exporter);
885        }
886
887        if (1) // export view cells and leaves
888        {
889                Exporter *exporter = Exporter::GetExporter("viewCells.x3d");
890                if (exporter)
891                {
892                        exporter->ExportBspViewCellPartition(*mBspTree);
893                        delete exporter;
894                }
895
896                exporter = Exporter::GetExporter("bspLeaves.x3d");
897
898                if (exporter)
899                {
900                        exporter->ExportBspLeaves(*mBspTree);
901                        delete exporter;
902                }
903        }
904
905}
906
907
908
909Real SamplingPreprocessor::RenderPvs(ViewCell &viewCell,
910                                                                         const float objRenderTime) const
911{
912        return viewCell.GetPvs().GetSize() * objRenderTime;
913}
914
915Real SamplingPreprocessor::SimulateRendering()
916{
917        Real renderTime = 0;
918
919        // render time for 1 object of PVS
920        const float objRt = 1.0f;
921         // const overhead for crossing a view cell border
922        const float vcOverhead = 0.01f;
923       
924        float totalArea = 0;
925
926        ViewCellContainer viewCells;
927
928        mBspTree->CollectViewCells(viewCells);
929
930        ViewCellContainer::const_iterator it, it_end = viewCells.end();
931        PolygonContainer::const_iterator pit;
932
933        for (it = viewCells.begin(); it != it_end; ++ it)
934        {
935                // surface area substitute for probability
936                PolygonContainer cell;
937                float area = 0;
938
939                mBspTree->ConstructGeometry(dynamic_cast<BspViewCell *>(*it), cell);
940
941                for (pit = cell.begin(); pit != cell.end(); ++ pit)
942                        area += (*pit)->GetArea();
943               
944                renderTime += area * RenderPvs(*(*it), objRt);
945                totalArea += area;
946        }
947
948        renderTime /= totalArea;
949
950        renderTime += (float)viewCells.size() * vcOverhead;
951       
952        return renderTime;
953}
Note: See TracBrowser for help on using the repository browser.