Changeset 1194 for GTP/trunk/Lib/Vis/Preprocessing/src

Timestamp:

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

Author:

mattausch

Message:

 

Location:

GTP/trunk/Lib/Vis/Preprocessing/src

Files:

• KdTree.cpp (modified) (1 diff)
• KdTree.h (modified) (2 diffs)
• Preprocessor.cpp (modified) (1 diff)
• SceneGraph.cpp (modified) (1 diff)
• ViewCellsParser.cpp (modified) (1 diff)
• main.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;
+}
+
+
+}

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

@@ -391 +391 @@
   CollectLeafPvs();
 
+  bool ExportBinTree(const string &filename);
+  bool LoadBinTree(const string &filename);
+
 protected:
 
@@ -553 +556 @@
         */
         void ProcessMultipleRefs(KdLeaf *leaf) const;
+
+        void ExportBinLeaf(ofstream &stream, KdLeaf *leaf);
+        void ExportBinInterior(ofstream &stream, KdInterior *interior);
+void ImportBinLeaf(ifstream &stream, KdLeaf *leaf);
+        void ImportBinInterior(ifstream &stream, KdInterior *interior);
 
   int mTermMaxNodes;

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

@@ -609 +609 @@
 #endif
         samplesOut.close();
+
         return true;
 }

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

@@ -155 +155 @@
 void SceneGraph::ExportScene(const string filename)
 {
-        // export binary version of mesh
+        
 }
+
 
 void SceneGraph::LoadScene(const string filename)

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

@@ -289 +289 @@
 
 
-void ViewCellsParseHandlers::StartViewCell(ViewCell *viewCell, AttributeList&  attributes)
+void ViewCellsParseHandlers::StartViewCell(ViewCell *viewCell, 
+                                                                                   AttributeList&  attributes)
 {
         int len = attributes.getLength();

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

@@ -161 +161 @@
         preprocessor->BuildKdTree();
         preprocessor->KdTreeStatistics(cout);
-
+        preprocessor->mKdTree->ExportBinTree("kd.bin");
         // parse view cells related options
         preprocessor->PrepareViewCells();