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

Timestamp:

08/10/06 09:23:38 (18 years ago)

Author:

mattausch

Message:

 

File:

• GTP/trunk/Lib/Vis/Preprocessing/src/KdTree.cpp (modified) (1 diff)

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

@@ -1125 +1125 @@
 
 
-}
+#define INTERIOR_START_ID -2
+#define LEAF_START_ID -3
+#define LEAF_END_ID -4
+
+
+void KdTree::ExportBinLeaf(ofstream &stream, KdLeaf *leaf)
+{
+        ObjectContainer::const_iterator it, it_end = leaf->mObjects.end();
+        
+        int leafStartId = LEAF_START_ID;
+        stream.write(reinterpret_cast<char *>(&leafStartId), sizeof(int));
+        for (it = leaf->mObjects.begin(); it != it_end; ++ it)
+        {       
+                Intersectable *obj = *it;
+                                
+                int id = obj->mId;              
+                
+                //stream.write(reinterpret_cast<char *>(&origin), sizeof(Vector3));
+                stream.write(reinterpret_cast<char *>(&id), sizeof(int));
+    }
+        
+        // end leaf
+        int leafEndId = LEAF_END_ID;
+        stream.write(reinterpret_cast<char *>(&leafEndId), sizeof(int));
+}
+
+
+void KdTree::ImportBinLeaf(ifstream &stream, KdLeaf *leaf)
+{
+        ObjectContainer::const_iterator it, it_end = leaf->mObjects.end();
+        
+        int leafId = LEAF_START_ID;
+        int objId = leafId;
+        
+        while (objId != LEAF_END_ID)
+        {       
+                stream.read(reinterpret_cast<char *>(&objId), sizeof(int));
+                 //if (samplesIn.eof()) break;
+    }
+}
+
+
+void KdTree::ExportBinInterior(ofstream &stream, KdInterior *interior)
+{
+        int interiorid = INTERIOR_START_ID;
+        stream.write(reinterpret_cast<char *>(&interiorid), sizeof(int));
+
+        int axis = interior->mAxis;
+        Vector3 pos = interior->mPosition;
+
+        stream.write(reinterpret_cast<char *>(&axis), sizeof(int));
+        stream.write(reinterpret_cast<char *>(&pos), sizeof(Vector3));
+}
+
+
+void KdTree::ImportBinInterior(ifstream &stream, KdInterior *interior)
+{
+        int interiorid = -2;
+        stream.read(reinterpret_cast<char *>(&interiorid), sizeof(int));
+
+        int axis = interior->mAxis;
+        Vector3 pos = interior->mPosition;
+
+        stream.read(reinterpret_cast<char *>(&axis), sizeof(int));
+        stream.read(reinterpret_cast<char *>(&pos), sizeof(Vector3));
+}
+
+
+bool KdTree::ExportBinTree(const string &filename)
+{
+        ofstream stream(filename.c_str(), ios::binary);
+        
+        if (!stream.is_open())
+                return false;
+
+        // export binary version of mesh
+        stack<KdNode *> tStack;
+
+        tStack.push(mRoot);
+
+        while(!tStack.empty())
+        {
+                KdNode *node = tStack.top();
+
+                if (node->IsLeaf())
+                {
+                        ExportBinLeaf(stream, dynamic_cast<KdLeaf *>(node));
+                }
+                else
+                {
+                        KdInterior *interior = dynamic_cast<KdInterior *>(node);
+
+                        ExportBinInterior(stream, interior);
+                        
+                        tStack.push(interior->mFront);
+                        tStack.push(interior->mBack);
+                }
+        }
+
+        return true;
+}
+
+
+bool KdTree::LoadBinTree(const string &filename)
+{
+        ofstream stream(filename.c_str(), ios::binary);
+        
+        if (!stream.is_open())
+                return false;
+
+        // export binary version of mesh
+        stack<KdNode *> tStack;
+
+        tStack.push(mRoot);
+
+        while(!tStack.empty())
+        {
+                KdNode *node = tStack.top();
+
+                if (node->IsLeaf())
+                {
+                        ExportBinLeaf(stream, dynamic_cast<KdLeaf *>(node));
+                }
+                else
+                {
+                        KdInterior *interior = dynamic_cast<KdInterior *>(node);
+
+                        ExportBinInterior(stream, interior);
+                        
+                        tStack.push(interior->mFront);
+                        tStack.push(interior->mBack);
+                }
+        }
+
+        return true;
+}
+
+
+}