source: GTP/trunk/Lib/Vis/Preprocessing/src/SamplingPreprocessor.cpp @ 1563

Revision 1563, 5.4 KB checked in by mattausch, 18 years ago (diff)

fixed bug with view space box

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#include "ViewCellsManager.h"
10#include "RenderSimulator.h"
11
12
13
14namespace GtpVisibilityPreprocessor {
15
16
17SamplingPreprocessor::SamplingPreprocessor(): Preprocessor(), mPass(0)
18{
19  // this should increase coherence of the samples
20  Environment::GetSingleton()->GetIntValue("SamplingPreprocessor.samplesPerPass", mSamplesPerPass);
21  Environment::GetSingleton()->GetIntValue("SamplingPreprocessor.totalSamples", mTotalSamples);
22 
23  mStats.open("stats.log");
24}
25
26SamplingPreprocessor::~SamplingPreprocessor()
27{
28  CLEAR_CONTAINER(mSampleRays);
29  CLEAR_CONTAINER(mVssSampleRays);
30}
31
32Intersectable *
33SamplingPreprocessor::CastRay(
34                                                          const Vector3 &origin,
35                                                          const Vector3 &direction
36                                                          )
37{
38  AxisAlignedBox3 box = mViewCellsManager->GetViewSpaceBox();
39 
40  AxisAlignedBox3 sbox = box;
41  sbox.Enlarge(Vector3(-Limits::Small));
42  if (!sbox.IsInside(origin))
43        return 0;
44 
45  Vector3 point, normal;
46  Intersectable *object;
47
48  // cast ray to KD tree to find intersection with other objects
49#if 0
50  object = CastSimpleRay(origin,
51                                                 direction,
52                                                 mKdTree->GetBox(),
53                                                 point,
54                                                 normal
55                                                 );
56#endif
57
58  if (mDetectEmptyViewSpace && DotProd(normal, direction) >= 0) {
59        object = NULL;
60  }
61 
62  return object;
63}
64
65void
66SamplingPreprocessor::HoleSamplingPass()
67{
68  vector<KdLeaf *> leaves;
69  mKdTree->CollectLeaves(leaves);
70 
71  // go through all the leaves and evaluate their passing contribution
72  for (int i=0 ; i < leaves.size(); i++) {
73    KdLeaf *leaf = leaves[i];
74    cout<<leaf->mPassingRays<<endl;
75  }
76}
77
78
79void
80SamplingPreprocessor::VerifyVisibility(Intersectable *object)
81{
82        // mail all nodes from the pvs
83  Intersectable::NewMail();
84  KdPvsMap::iterator i = object->mKdPvs.mEntries.begin();
85  for (; i != object->mKdPvs.mEntries.end(); i++) {
86        KdNode *node = (*i).first;
87        node->Mail();
88  }
89  Debug << "Get all neighbours from PVS" << endl;
90  vector<KdNode *> invisibleNeighbors;
91  // get all neighbors of all PVS nodes
92  i = object->mKdPvs.mEntries.begin();
93  for (; i != object->mKdPvs.mEntries.end(); i++) {
94        KdNode *node = (*i).first;
95        mKdTree->FindNeighbors(node, invisibleNeighbors, true);
96        AxisAlignedBox3 box = object->GetBox();
97        for (int j=0; j < invisibleNeighbors.size(); j++) {
98          int visibility = ComputeBoxVisibility(mSceneGraph,
99                                                                                        mKdTree,
100                                                                                        box,
101                                                                                        mKdTree->GetBox(invisibleNeighbors[j]),
102                                                                                        1e-6f);
103          //          exit(0);
104        }
105        // now rank all the neighbors according to probability that a new
106        // sample creates some contribution
107  }
108}
109
110bool
111SamplingPreprocessor::ComputeVisibility()
112{
113 
114  Debug << "type: sampling" << endl;
115 
116  cout<<"Sampling Preprocessor started\n"<<flush;
117  //  cout<<"Memory/ray "<<sizeof(VssRay)+sizeof(RssTreeNode::RayInfo)<<endl;
118
119  Randomize(0);
120  const long startTime = GetTime();
121  int totalSamples = 0;
122
123  // if not already loaded, construct view cells from file
124  if (!mLoadViewCells)
125  {
126          ConstructViewCells();
127  }
128   
129  int samples = 0;
130  int i=0;
131  while (samples < mTotalSamples) {
132        for (i=0; i < mSamplesPerPass; i++, samples++) {
133         
134          if (i%10000 == 0)
135                cout<<"+";
136         
137          Vector3 origin, direction;
138          mViewCellsManager->GetViewPoint(origin);
139          direction = UniformRandomVector();
140
141         
142          ViewCell *viewcell = mViewCellsManager->GetViewCell(origin);
143         
144          if (viewcell && viewcell->GetValid()) {
145                // cast rays in both directions to make the number of samples comparable
146                // with the global sampling method which also casts a "double" ray per sample
147                for (int j=0; j < 2; j++) {
148                  Intersectable *object = CastRay(origin, direction);
149                  if (object) {
150                        // if ray not outside of view space
151                        float contribution;
152                        int pvsContribution = 0;
153                        float relativePvsContribution = 0;
154                        if (object) {
155                          float pdf = 1.0f;
156                          if (viewcell->GetPvs().GetSampleContribution(object,
157                                                                                                                   pdf,
158                                                                                                                   contribution))
159                                ++pvsContribution;
160                          relativePvsContribution += contribution;
161                          viewcell->GetPvs().AddSample(object, pdf);
162                        }
163                  }
164                  direction = -direction;
165                }
166          }
167         
168          if (samples > mTotalSamples)
169                break;
170        }
171       
172        //      mVssRays.PrintStatistics(mStats);
173        mStats <<
174          "#Time\n" << TimeDiff(startTime, GetTime())*1e-3<<endl<<
175          "#TotalSamples\n" <<samples<<endl;
176
177        mViewCellsManager->PrintPvsStatistics(mStats);
178        // ComputeRenderError();
179  }
180 
181  //  HoleSamplingPass();
182        if (0) {
183          Exporter *exporter = Exporter::GetExporter("ray-density.x3d");
184          exporter->SetExportRayDensity(true);
185          exporter->ExportKdTree(*mKdTree);
186          delete exporter;
187        }
188       
189       
190        // $$JB temporary removed
191        //      mViewCellsManager->PostProcess(objects, mSampleRays);
192       
193        //-- several visualizations and statistics
194        Debug << "view cells after post processing: " << endl;
195        mViewCellsManager->PrintStatistics(Debug);
196       
197        //-- render simulation after merge
198        cout << "\nevaluating bsp view cells render time after merge ... ";
199       
200        mRenderSimulator->RenderScene();
201        SimulationStatistics ss;
202        mRenderSimulator->GetStatistics(ss);
203       
204        cout << " finished" << endl;
205        cout << ss << endl;
206        Debug << ss << endl;
207       
208        // $$JB temporary removed
209        //mViewCellsManager->Visualize(objects, mSampleRays);   
210       
211        return true;
212}
213
214
215}
Note: See TracBrowser for help on using the repository browser.