Changeset 8408 for OpenSceneGraph/trunk/examples/osgkdtree/osgkdtree.cpp

Timestamp:

06/04/08 19:59:39 (5 years ago)

Author:

robert

Message:

Further work on fleshing out basic classes

Files:

• OpenSceneGraph/trunk/examples/osgkdtree/osgkdtree.cpp (modified) (8 diffs)

OpenSceneGraph/trunk/examples/osgkdtree/osgkdtree.cpp

@@ -23 +23 @@
 #include <osg/Notify>
 #include <osg/io_utils>
+#include <osg/Geometry>
 
 #include <osgDB/ReadFile>
@@ -34 +35 @@
 
 #include <iostream>
+
+namespace osg
+{
 
 class KDNode
@@ -61 +65 @@
 };
 
-class KDTree : public osg::Referenced
+class KDTree : public osg::Shape
 {
     public:
+    
+    
+        KDTree() {}
+        
+        KDTree(const KDTree& rhs, const CopyOp& copyop=CopyOp::SHALLOW_COPY):
+            Shape(rhs,copyop) {}
+
+        META_Shape(osg, KDTree)
     
         typedef std::vector< unsigned int > AxisStack;
@@ -69 +81 @@
         typedef std::vector< osg::ref_ptr<KDLeaf> > KDLeafList;
 
-        osg::BoundingBox    _bb;
+        osg::observer_ptr<osg::Geometry> _geometry;
 
-        AxisStack           _axisStack;
-        KDNodeList          _kdNodes;
-        KDLeafList          _kdLeaves; 
+        osg::BoundingBox                _bb;
+
+        AxisStack                       _axisStack;
+        KDNodeList                      _kdNodes;
+        KDLeafList                      _kdLeaves; 
 };
 
@@ -80 +94 @@
     public:
     
-        void traverse(KDTree& tree, KDNode::value_type nodeIndex)
+        void traverse(KDTree& tree, KDNode::value_type nodeIndex, unsigned int level)
         {
+            for(unsigned int i=0; i<level; ++i)
+            {
+                osg::notify(osg::NOTICE)<<"  ";
+            }
+            osg::notify(osg::NOTICE)<<"traverse("<<nodeIndex<<", "<< level<<") { "<<std::endl;
+            
             if (nodeIndex>=0)
             {        
                 KDNode& node = tree._kdNodes[nodeIndex];
-                traverse(tree,node._leftChild);
-                traverse(tree,node._rightChild);
+                traverse(tree,node._leftChild,level+1);
+                traverse(tree,node._rightChild,level+1);
             }
             else 
@@ -93 +113 @@
                 KDLeaf& leaf = *(tree._kdLeaves[leafIndex]);
             }
+
+            for(unsigned int i=0; i<level; ++i)
+            {
+                osg::notify(osg::NOTICE)<<"  ";
+            }
+            osg::notify(osg::NOTICE)<<"}"<<std::endl;;
         }
 
@@ -98 +124 @@
         void traverse(KDTree& tree)
         {
-            if (!tree._kdNodes.empty()) traverse(tree,0);
+            osg::notify(osg::NOTICE)<<"traverse(tree)"<<std::endl;
+            if (!tree._kdNodes.empty()) traverse(tree,0,0);
         }
     
 };
+
+
+class KDTreeBuilder : public osg::NodeVisitor
+{
+    public:
+    
+        KDTreeBuilder():
+            osg::NodeVisitor(osg::NodeVisitor::TRAVERSE_ALL_CHILDREN) {}
+    
+    
+        void apply(osg::Geode& geode)
+        {
+            for(unsigned int i=0; i<geode.getNumDrawables(); ++i)
+            {
+                osg::Geometry* geom = geode.getDrawable(i)->asGeometry();
+                if (geom)
+                {
+                    geom->setShape(createKDTree(geom));
+                }   
+            }
+        }
+    
+        KDTree* createKDTree(osg::Geometry* geometry);
+};
+
+
+KDTree* KDTreeBuilder::createKDTree(osg::Geometry* geometry)
+{
+    KDTree* kdTree = new KDTree;
+    kdTree->_geometry = geometry;
+    kdTree->_bb = kdTree->_geometry->getBound();
+    
+    osg::notify(osg::NOTICE)<<"osg::KDTreeBuilder::createKDTree()"<<std::endl;
+    
+    return kdTree;
+}    
+
+}
 
 int main(int argc, char **argv)
@@ -116 +181 @@
     }
     
+    osg::KDTreeBuilder builder;
+    scene->accept(builder);
+    
+    
     
     return 0;