Changeset 507 for trunk/VUT/GtpVisibilityPreprocessor/src/KdTree.cpp

Timestamp:

01/07/06 10:50:29 (18 years ago)

Author:

bittner

Message:

Preparation for hw based sampling

File:

• trunk/VUT/GtpVisibilityPreprocessor/src/KdTree.cpp (modified) (2 diffs)

trunk/VUT/GtpVisibilityPreprocessor/src/KdTree.cpp

@@ -941 +941 @@
 KdTree::GetRandomLeaf(const bool onlyUnmailed)
 {
-        stack<KdNode *> nodeStack;
+  stack<KdNode *> nodeStack;
   nodeStack.push(mRoot);
-
-        int mask = rand();
+  
+  int mask = rand();
         
   while (!nodeStack.empty()) {
@@ -967 +967 @@
 
 int
-CastBeam(
-                 Beam &beam
-                 )
-{
-  
-
-  return 0;
-}
+KdTree::CastBeam(
+                                 Beam &beam
+                                 )
+{
+  stack<KdNode *> nodeStack;
+  nodeStack.push(mRoot);
+  
+  while (!nodeStack.empty()) {
+    KdNode *node = nodeStack.top();
+    nodeStack.pop();
+        
+        int side = beam.ComputeIntersection(GetBox(node));
+        switch (side) {
+        case -1:
+          beam.mKdNodes.push_back(node);
+          break;
+        case 0:
+          if (node->IsLeaf())
+                beam.mKdNodes.push_back(node);
+          else {
+                KdInterior *interior = (KdInterior *)node;
+                KdNode *first = interior->mBack;
+                KdNode *second = interior->mFront;
+                
+                if (interior->mAxis < 3) {
+                  // spatial split -> decide on the order of the nodes
+                  if (beam.mPlanes[0].mNormal[interior->mAxis] > 0)
+                        swap(first, second);
+                }
+
+                nodeStack.push(first);
+                nodeStack.push(second);
+          }
+          break;
+          // default: cull
+        }
+  }
+
+  return beam.mKdNodes.size();
+}