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

Revision 375, 24.7 KB checked in by mattausch, 19 years ago (diff)

added bsp view cells statistics

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        //-- post processing of bsp view cells
534        if (mBspTree)
535        {
536                int vcSize = 0;
537                int pvsSize = 0;
538
539                BspViewCellsStatistics stat;
540               
541                Debug << "overall scene size: " << objects.size() << endl;
542                mBspTree->EvaluateViewCellsStats(stat);
543               
544                Debug << "original view cell partition:\n" << stat << endl;
545               
546                cout << "starting post processing using " << mSampleRays.size() << " samples ... ";
547               
548                long startTime = GetTime();
549                int merged = PostprocessViewCells(mSampleRays);
550               
551                cout << "finished" << endl;
552                cout << "merged " << merged << " view cells in "
553                         << TimeDiff(startTime, GetTime()) *1e-3 << " secs" << endl;
554
555                cout << "evaluating render time heuristics ... ";
556
557                //-- recount pvs
558                mBspTree->EvaluateViewCellsStats(stat);
559               
560                Debug << "after post processing:\n" << stat << endl;
561
562                Real rt = SimulateRendering();
563                cout << "finished" << endl;
564
565                Debug << "render time: " << rt * 1e-3 << endl;
566        }
567       
568  //  HoleSamplingPass();
569        if (0) {
570                Exporter *exporter = Exporter::GetExporter("ray-density.x3d");
571                exporter->SetExportRayDensity(true);
572                exporter->ExportKdTree(*mKdTree);
573
574                if (mBspTree && (ViewCell::sHierarchy == ViewCell::BSP))
575                        exporter->ExportBspTree(*mBspTree);
576
577                delete exporter;
578        }
579 
580        bool exportRays = false;
581        if (exportRays) {
582                Exporter *exporter = NULL;
583                exporter = Exporter::GetExporter("sample-rays.x3d");
584                exporter->SetWireframe();
585                exporter->ExportKdTree(*mKdTree);
586                exporter->ExportBspTree(*mBspTree);
587
588                for (i=0; i < pvsOut; i++)
589                exporter->ExportRays(rays[i], 1000, RgbColor(1, 0, 0));
590                exporter->SetFilled();
591                 
592                delete exporter;
593        }
594
595        //-- several visualizations and statistics
596        if (1) {
597                if (mBspTree && (ViewCell::sHierarchy == ViewCell::BSP))
598        {
599                bool exportSplits = false;
600                environment->GetBoolValue("BspTree.exportSplits", exportSplits);
601
602                // export the bsp splits
603                cout << "exporting splits ... ";
604                if (exportSplits)
605                        ExportSplits(objects);
606                cout << "finished" << endl;
607
608                ExportBspPvs(objects);
609        }
610   for (int k=0; k < pvsOut; k++) {
611                 Intersectable *object = objects[k];
612                 char s[64];
613                 sprintf(s, "sample-pvs%04d.x3d", k);
614                 Exporter *exporter = Exporter::GetExporter(s);
615                 exporter->SetWireframe();
616                 
617                 
618                 KdPvsMap::iterator i = object->mKdPvs.mEntries.begin();
619                 Intersectable::NewMail();
620                 
621                 // avoid adding the object to the list
622                 object->Mail();
623                 ObjectContainer visibleObjects;
624                 
625                 for (; i != object->mKdPvs.mEntries.end(); i++)
626                         {
627                                 KdNode *node = (*i).first;
628                                 exporter->ExportBox(mKdTree->GetBox(node));
629                                 mKdTree->CollectObjects(node, visibleObjects);
630                         }
631                 
632                 exporter->ExportRays(rays[k], 1000, RgbColor(0, 1, 0));
633                 exporter->SetFilled();
634                 
635                 for (int j = 0; j < visibleObjects.size(); j++)
636                         exporter->ExportIntersectable(visibleObjects[j]);
637                 
638                 
639                 Material m;
640                 m.mDiffuseColor = RgbColor(1, 0, 0);
641                 exporter->SetForcedMaterial(m);
642                 exporter->ExportIntersectable(object);
643                 
644                 delete exporter;
645         }
646  }
647 
648  return true;
649}
650
651bool SamplingPreprocessor::ProcessBspViewCells(Ray &ray,
652                                                                                           Intersectable *object,
653                                                                                           int faceIndex,
654                                                                                           int &contributingSamples,
655                                                                                           int &sampleContributions)
656{
657        // save rays for bsp tree construction
658        if ((BspTree::sConstructionMethod == BspTree::FROM_RAYS) &&
659                (mSampleRays.size() < mBspConstructionSamples))
660        {
661                // also add origin to sample in order to extract it as input polygons
662                MeshInstance *mi = dynamic_cast<MeshInstance *>(object);
663                ray.sourceObject = Ray::Intersection(0.0, mi, faceIndex);
664                                                       
665        mSampleRays.push_back(new Ray(ray));
666               
667                return false;
668        }
669       
670        if (!mBspTree) // construct BSP tree using the samples
671        {
672        cout << "building bsp tree from " << mSampleRays.size() << " samples " << endl;
673                BuildBspTree();
674               
675                // add contributions of saved samples to PVS
676                contributingSamples += mBspTree->GetStat().contributingSamples;
677                sampleContributions += mBspTree->GetStat().sampleContributions;
678
679                BspTreeStatistics(Debug);       
680
681                if (0) Export("vc_bsptree.x3d", false, false, true);
682
683                // throw away samples because BSP leaves not stored in order
684                // Need ordered rays for post processing => collect new rays
685                CLEAR_CONTAINER(mSampleRays);
686        }
687       
688        if ((int)mSampleRays.size() < mPostProcessSamples)
689        {
690                mSampleRays.push_back(new Ray(ray));
691        }
692
693        return true;
694}
695
696// merge or subdivide view cells
697int SamplingPreprocessor::PostprocessViewCells(const RayContainer &rays)
698{
699        int merged = 0;
700
701        RayContainer::const_iterator rit, rit_end = rays.end();
702        vector<Ray::BspIntersection>::const_iterator iit;
703
704        for (rit = rays.begin(); rit != rays.end(); ++ rit)
705        { 
706                // traverse leaves stored in the rays and compare and merge consecutive
707                // leaves (i.e., the neighbors in the tree)
708                if ((*rit)->bspIntersections.empty())
709                        continue;
710
711                iit = (*rit)->bspIntersections.begin();
712
713                BspLeaf *previousLeaf = (*iit).mLeaf;
714                ++ iit;
715               
716                for (; iit != (*rit)->bspIntersections.end(); ++ iit)
717                {
718                        BspLeaf *leaf = (*iit).mLeaf;
719
720                        if (mBspTree->ShouldMerge(leaf, previousLeaf))
721                        {                       
722                                mBspTree->MergeViewCells(leaf, previousLeaf);
723
724                                ++ merged;
725                        }
726                        previousLeaf = leaf;
727                }
728        }
729
730        return merged;
731}
732
733void SamplingPreprocessor::ExportSplits(const ObjectContainer &objects)
734{
735        Exporter *exporter = Exporter::GetExporter("bsp_splits.x3d");
736
737        if (exporter)
738        {       
739                Material m;
740                m.mDiffuseColor = RgbColor(1, 0, 0);
741                exporter->SetForcedMaterial(m);
742                exporter->SetWireframe();
743                exporter->ExportBspSplits(*mBspTree);
744
745                // take forced material, else big scenes cannot be viewed
746                m.mDiffuseColor = RgbColor(0, 1, 0);
747                exporter->SetForcedMaterial(m);
748                exporter->SetFilled();
749
750                exporter->ResetForcedMaterial();
751               
752                // export rays
753                if (0)
754                {
755                        RayContainer outRays;
756               
757                        for (int i = 0; i < mSampleRays.size(); ++ i)
758                        {
759                                // only rays piercing geometry
760                                if (!mSampleRays[i]->intersections.empty())
761                                        outRays.push_back(mSampleRays[i]);
762                        }
763                        if (BspTree::sConstructionMethod == BspTree::FROM_RAYS)
764                        {
765                                // export rays
766                                exporter->ExportRays(outRays, 1000, RgbColor(1, 1, 0));
767                        }
768                }
769
770                // export scene geometry
771                if (0)
772                {
773                        Material m;//= RandomMaterial();
774                        m.mDiffuseColor = RgbColor(0, 1, 0);
775                        exporter->SetForcedMaterial(m);
776            exporter->SetWireframe();
777
778                        for (int j = 0; j < objects.size(); ++ j)
779                                exporter->ExportIntersectable(objects[j]);
780                }
781
782                delete exporter;
783        }
784}
785
786void SamplingPreprocessor::ExportBspPvs(const ObjectContainer &objects)
787{
788        //-- some random view cells and rays for output
789        const int leafOut = 10;
790
791        vector<Ray> vcRays[leafOut];
792        vector<BspLeaf *> bspLeaves;
793                       
794        for (int i = 0; i < leafOut; ++ i)
795                bspLeaves.push_back(mBspTree->GetRandomLeaf());
796       
797        const int raysOut = min((int)mSampleRays.size(), 20000);
798
799        ViewCell::NewMail();
800
801        for (int i = 0; i < bspLeaves.size(); ++ i)
802        {
803                cout << "creating output for view cell " << i << " ... ";
804                // check whether we can add the current ray to the output rays
805                for (int k = 0; k < raysOut; ++ k)
806                {
807                        Ray *ray = mSampleRays[k];
808
809                        for     (int j = 0; j < (int)ray->bspIntersections.size(); ++ j)
810                        {
811                                BspLeaf *leaf = ray->bspIntersections[j].mLeaf;
812
813                                if (bspLeaves[i]->GetViewCell() == leaf->GetViewCell())
814                                {
815                                        vcRays[i].push_back(*ray);
816                                }
817                        }
818                }
819
820                Intersectable::NewMail();
821
822                ViewCell *vc = bspLeaves[i]->GetViewCell();
823
824                //bspLeaves[j]->Mail();
825                char s[64]; sprintf(s, "bsp-pvs%04d.x3d", i);
826
827                Exporter *exporter = Exporter::GetExporter(s);
828                exporter->SetFilled();
829
830                ViewCellPvsMap::iterator it = vc->GetPvs().mEntries.begin();
831
832                exporter->SetWireframe();
833
834                Material m;//= RandomMaterial();
835                m.mDiffuseColor = RgbColor(0, 1, 0);
836                exporter->SetForcedMaterial(m);
837
838                if (vc->GetMesh())
839                        exporter->ExportViewCell(vc);
840                else
841                {
842            PolygonContainer cell;
843                        // export view cell
844                        mBspTree->ConstructGeometry(bspLeaves[i]->GetViewCell(), cell);
845                        exporter->ExportPolygons(cell);
846                        CLEAR_CONTAINER(cell);
847                }
848
849                Debug << i << ": pvs size=" << (int)vc->GetPvs().GetSize()
850                                << ", piercing rays=" << (int)vcRays[i].size() << endl;
851
852                // export view cells
853                if (0)
854                {
855                        m.mDiffuseColor = RgbColor(1, 0, 1);
856                        exporter->SetForcedMaterial(m);
857                        exporter->ExportViewCells(mViewCells);
858                }
859
860                // export rays piercing this view cell
861                exporter->ExportRays(vcRays[i], 1000, RgbColor(0, 1, 0));
862
863                m.mDiffuseColor = RgbColor(1, 0, 0);
864                exporter->SetForcedMaterial(m);
865
866                // output PVS of view cell
867                for (; it != vc->GetPvs().mEntries.end(); ++ it)
868                {
869                        Intersectable *intersect = (*it).first;
870                        if (!intersect->Mailed())
871                        {
872                                exporter->ExportIntersectable(intersect);
873                                intersect->Mail();
874                        }                       
875                }
876                       
877                // output rest of the objects
878                if (0)
879                {
880                        Material m;//= RandomMaterial();
881                        m.mDiffuseColor = RgbColor(0, 0, 1);
882                        exporter->SetForcedMaterial(m);
883
884                        for (int j = 0; j < objects.size(); ++ j)
885                                if (!objects[j]->Mailed())
886                                {
887                                        exporter->SetForcedMaterial(m);
888                                        exporter->ExportIntersectable(objects[j]);
889                                        objects[j]->Mail();
890                                }
891                }
892                DEL_PTR(exporter);
893                cout << "finished" << endl;
894        }
895
896        if (1) // export view cells and leaves
897        {
898                cout << "exporting view cells and leaves ... ";
899                Exporter *exporter = Exporter::GetExporter("viewCells.x3d");
900                if (exporter)
901                {
902                        exporter->ExportBspViewCellPartition(*mBspTree);
903                        delete exporter;
904                }
905
906                exporter = Exporter::GetExporter("bspLeaves.x3d");
907
908                if (exporter)
909                {
910                        exporter->ExportBspLeaves(*mBspTree);
911                        delete exporter;
912                }
913                cout << "finished" << endl;
914        }
915
916}
917
918
919
920Real SamplingPreprocessor::RenderPvs(ViewCell &viewCell,
921                                                                         const float objRenderTime) const
922{
923        return viewCell.GetPvs().GetSize() * objRenderTime;
924}
925
926Real SamplingPreprocessor::SimulateRendering()
927{
928        Real renderTime = 0;
929
930        // render time for 1 object of PVS
931        const float objRt = 1.0f;
932         // const overhead for crossing a view cell border
933        const float vcOverhead = 0.01f;
934       
935        float totalArea = 0;
936
937        ViewCellContainer viewCells;
938
939        mBspTree->CollectViewCells(viewCells);
940
941        ViewCellContainer::const_iterator it, it_end = viewCells.end();
942        PolygonContainer::const_iterator pit;
943
944        for (it = viewCells.begin(); it != it_end; ++ it)
945        {
946                // surface area substitute for probability
947                PolygonContainer cell;
948                float area = 0;
949
950                mBspTree->ConstructGeometry(dynamic_cast<BspViewCell *>(*it), cell);
951
952                for (pit = cell.begin(); pit != cell.end(); ++ pit)
953                        area += (*pit)->GetArea();
954               
955                renderTime += area * RenderPvs(*(*it), objRt);
956                totalArea += area;
957        }
958
959        renderTime /= totalArea;
960
961        renderTime += (float)viewCells.size() * vcOverhead;
962       
963        return renderTime;
964}
Note: See TracBrowser for help on using the repository browser.