1 | #include "SamplingStrategy.h"
|
---|
2 | #include "Intersectable.h"
|
---|
3 | #include "DifferenceSampling.h"
|
---|
4 | #include "ViewCell.h"
|
---|
5 | #include "ViewCellsManager.h"
|
---|
6 | #include "Preprocessor.h"
|
---|
7 | #include "Timer/PerfTimer.h"
|
---|
8 |
|
---|
9 |
|
---|
10 | #ifdef GTP_INTERNAL
|
---|
11 | #include "ArchModeler2MLRT.hxx"
|
---|
12 | #endif
|
---|
13 |
|
---|
14 | PerfTimer sFilterTimer;
|
---|
15 |
|
---|
16 |
|
---|
17 | namespace GtpVisibilityPreprocessor {
|
---|
18 |
|
---|
19 |
|
---|
20 | /** Collects the pvs of the objects which are
|
---|
21 | in the first but not in the second pvs.
|
---|
22 | */
|
---|
23 | static void GetPvsDifference(const ObjectPvs &largerPvs,
|
---|
24 | const ObjectPvs &smallerPvs,
|
---|
25 | ObjectContainer &pvsDifference)
|
---|
26 | {
|
---|
27 | Intersectable::NewMail();
|
---|
28 | Intersectable *obj;
|
---|
29 |
|
---|
30 | ObjectPvsIterator bit = smallerPvs.GetIterator();
|
---|
31 |
|
---|
32 | // mail smaller pvs
|
---|
33 | while (bit.HasMoreEntries())
|
---|
34 | {
|
---|
35 | obj = bit.Next();
|
---|
36 | obj->Mail();
|
---|
37 | }
|
---|
38 |
|
---|
39 | ObjectPvsIterator ait = largerPvs.GetIterator();
|
---|
40 |
|
---|
41 | // write differece pvs
|
---|
42 | while (ait.HasMoreEntries())
|
---|
43 | {
|
---|
44 | obj = ait.Next();
|
---|
45 | pvsDifference.push_back(obj);
|
---|
46 |
|
---|
47 | //if (!obj->Mailed())
|
---|
48 | // c.AddSampleDirty(obj, 0);
|
---|
49 | }
|
---|
50 | }
|
---|
51 |
|
---|
52 |
|
---|
53 | DifferenceSampling::DifferenceSampling(Preprocessor &preprocessor):
|
---|
54 | SamplingStrategy(preprocessor), mCurrentViewCell(0), mNumSamples(0)
|
---|
55 | {}
|
---|
56 |
|
---|
57 |
|
---|
58 | void DifferenceSampling::ComputeDifferenceSet(ViewCell *vc,
|
---|
59 | ObjectContainer &differenceSet)
|
---|
60 | {
|
---|
61 | const ObjectPvs &pvs = vc->GetPvs();
|
---|
62 | const float filterSize = 100;
|
---|
63 | ObjectPvs filteredPvs;
|
---|
64 |
|
---|
65 | sFilterTimer.Entry();
|
---|
66 |
|
---|
67 | PvsFilterStatistics pvsStats = mPreprocessor.
|
---|
68 | mViewCellsManager->ApplyFilter2(vc,
|
---|
69 | false,
|
---|
70 | filterSize,
|
---|
71 | filteredPvs);
|
---|
72 |
|
---|
73 | sFilterTimer.Exit();
|
---|
74 |
|
---|
75 | //mDifferencePvs.Clear(false);
|
---|
76 | //mDifferencePvs.Reserve(filteredPvs.GetSize());
|
---|
77 | mPvsDifference.clear();
|
---|
78 | mPvsDifference.reserve(filteredPvs.GetSize());
|
---|
79 |
|
---|
80 | GetPvsDifference(filteredPvs, pvs, mPvsDifference);
|
---|
81 |
|
---|
82 | cerr << "found " << mPvsDifference.size() << " different objects" << endl;
|
---|
83 | }
|
---|
84 |
|
---|
85 |
|
---|
86 | bool DifferenceSampling::GenerateSample(SimpleRay &ray)
|
---|
87 | {
|
---|
88 | if (mNumSamples <= 0)
|
---|
89 | {
|
---|
90 | // choose random view cell
|
---|
91 | ViewCellContainer &viewCells = mPreprocessor.mViewCellsManager->GetViewCells();
|
---|
92 | const int vcIdx = (int)RandomValue(0, (float)viewCells.size() - 0.5f);
|
---|
93 |
|
---|
94 | mCurrentViewCell = viewCells[vcIdx];
|
---|
95 |
|
---|
96 | if (mCurrentViewCell->GetPvs().Empty())
|
---|
97 | return false;
|
---|
98 |
|
---|
99 | cout << "computing new difference set" << endl;
|
---|
100 |
|
---|
101 | ComputeDifferenceSet(mCurrentViewCell, mPvsDifference);
|
---|
102 |
|
---|
103 | // n samples per object
|
---|
104 | const int n = 50;
|
---|
105 | mNumSamples = (int)mPvsDifference.size() * n;
|
---|
106 | }
|
---|
107 |
|
---|
108 | if (mPvsDifference.empty())
|
---|
109 | return false;
|
---|
110 |
|
---|
111 | -- mNumSamples;
|
---|
112 |
|
---|
113 | cout << "x";
|
---|
114 | Vector3 origin, direction;
|
---|
115 | Vector3 point;
|
---|
116 | Vector3 normal, vcnormal;
|
---|
117 |
|
---|
118 | // get a new object from the difference set
|
---|
119 | const int objIdx = (int)RandomValue(0, (float)mPvsDifference.size() - 0.5f);
|
---|
120 | Intersectable *object = mPvsDifference[objIdx];
|
---|
121 |
|
---|
122 | // cast a ray from the view cells border towards the object
|
---|
123 | object->GetRandomSurfacePoint(point, normal);
|
---|
124 | mCurrentViewCell->GetRandomEdgePoint(origin, vcnormal);
|
---|
125 |
|
---|
126 | direction = point - origin;
|
---|
127 |
|
---|
128 | // $$ jb the pdf is yet not correct for all sampling methods!
|
---|
129 | const float c = Magnitude(direction);
|
---|
130 |
|
---|
131 | if ((c <= Limits::Small) /*|| (DotProd(direction, normal) < 0)*/)
|
---|
132 | {
|
---|
133 | return false;
|
---|
134 | }
|
---|
135 |
|
---|
136 | // $$ jb the pdf is yet not correct for all sampling methods!
|
---|
137 | const float pdf = 1.0f;
|
---|
138 | //cout << "p: " << point << " ";
|
---|
139 | direction *= 1.0f / c;
|
---|
140 | ray = SimpleRay(origin, direction, DIFFERENCE_SAMPLING_BASED_DISTRIBUTION, pdf);
|
---|
141 |
|
---|
142 | //cout << "ray: " << ray.mOrigin << " " << ray.mDirection << endl;
|
---|
143 |
|
---|
144 | return true;
|
---|
145 | }
|
---|
146 |
|
---|
147 |
|
---|
148 | } |
---|