Changeset 2705 for GTP/trunk/Lib/Vis/Preprocessing/src/KdTree.cpp

Timestamp:

05/24/08 11:59:35 (16 years ago)

Author:

mattausch

Message:

enabled view cell visualization

File:

• GTP/trunk/Lib/Vis/Preprocessing/src/KdTree.cpp (modified) (6 diffs)

GTP/trunk/Lib/Vis/Preprocessing/src/KdTree.cpp

@@ -34 +34 @@
 
 KdNode::KdNode(KdInterior *parent):
-mParent(parent), mMailbox(0), mIntersectable(NULL), mMailbox2(0)
+mParent(parent), 
+mMailbox(0), 
+mIntersectable(NULL), 
+mMailbox2(0),
+mNumObjects(0)
 {
   if (parent)
@@ -136 +140 @@
   SetPvsTerminationNodes(area);
   
+  // sets the number of objects in the subtree
+  KdNode::NewMail();
+  SetNumObjects(mRoot);
+
   return true;
 }
@@ -1057 +1065 @@
 void
 KdTree::CollectKdObjects(const AxisAlignedBox3 &box,
-                         ObjectContainer &objects
-                        )
+                                                 ObjectContainer &objects
+                                                 )
 {
   stack<KdNode *> nodeStack;
@@ -1124 +1132 @@
 void
 KdTree::CollectObjects(const AxisAlignedBox3 &box,
-                       ObjectContainer &objects)
+                                           ObjectContainer &objects)
 {
   stack<KdNode *> nodeStack;
@@ -1152 +1160 @@
     }
   }
+}
+
+
+int KdTree::GetNumObjects(KdNode *node) const
+{
+        return node->mNumObjects;
+}
+
+
+void KdTree::SetNumObjects(KdNode *node)
+{
+        node->mNumObjects = 0;
+
+        if (node->IsLeaf()) 
+        {
+                KdLeaf *leaf = (KdLeaf *)node;
+
+                for (int j=0; j < leaf->mObjects.size(); ++ j) 
+                {
+                        Intersectable *object = leaf->mObjects[j];
+
+                        if (!object->Mailed()) 
+                        {
+                                object->Mail();
+                                ++ node->mNumObjects;
+                        }
+                }
+        }
+        else 
+        {
+                KdInterior *interior = (KdInterior *)node;
+
+                SetNumObjects(interior->mFront);
+                SetNumObjects(interior->mBack);
+
+                node->mNumObjects += interior->mFront->mNumObjects;
+                node->mNumObjects += interior->mBack->mNumObjects;
+        }
 }
 
@@ -1670 +1716 @@
         SetPvsTerminationNodes(area);
 
+        // sets the number of objects in the subtree
+        KdNode::NewMail();
+        SetNumObjects(mRoot);
+
         Debug << mStat << endl;