1 void Photontracer :: Reflect ( Intsct* intsct , Photon photon , int dep , Photonmap*
photonmap , bool refracted )
2 {
3 Object * obj = intsct -> GetObj();
4 photon .dir = ReflDir(intsct ->I, intsct ->N);
5 Color basecolor = obj ->GetColor(intsct ->P) * obj -> GetMaterial () ->crefl ;
6 float power = photon.color.Power();
7 photon . color = photon.color * basecolor;
8 PhotonTrace(photon , dep + 1, photonmap , refracted);
9 }
10
11 void Photontracer :: Refract ( Intsct* intsct , Photon photon , int dep , Photonmap*
photonmap , bool refracted )
12 {
13 Object * obj = intsct -> GetObj();
14 double n = obj ->GetMaterial()->N;
15 bool valid;
16 n = refracted ? 1.0 / n : n;
17 photon .dir = RefrDir(intsct ->I, intsct ->N, n, valid);
18 if(! valid)return;
19 Color basecolor = obj ->GetMaterial ()->incolor * obj -> GetMaterial () ->crefc ;
20 if( refracted)
21 basecolor = basecolor * (obj ->GetMaterial ()->absorb * -intsct ->dep).Exp ()
;
22 float power = photon.color.Power();
23 photon . color = photon.color * basecolor;
24 PhotonTrace(photon , dep + 1, photonmap , refracted);
25 }
处理光子的反射和折射。
1 void Photonmap:: Balance ()
2 {
3 std:: cout << "Stored␣Photons␣=␣" << photons.size() << std ::endl ;
4 Photon ** p = new Photon*[photons.size () ];
5 for(int i = 0; i < photons.size (); i++)
6 p[i] = & photons[i];
7 head = SegBalance(p, 0, photons . size());
8 }
9 KDT* Photonmap :: SegBalance(Photon ** p, int l, int r)
10 {
11 if(l >= r)return NULL ;
12 int mid = (l + r) / 2;
13 KDT* node = new KDT();
14 int axis = 1;
15 if( Box_max.y - Box_min.y > Box_max.x - Box_min.x && Box_max .y - Box_min .y >
Box_max.z - Box_min .z) axis = 2;
16 if( Box_max.z - Box_min.z > Box_max.x - Box_min.x && Box_max .z - Box_min .z >
Box_max.y - Box_min .y) axis = 3;
17 Axis = axis;
18 std:: nth_element(p + l, p + mid , p + r, cmp );
19
3
评论0