Changeset 1279 for GTP/trunk/Lib/Vis/Preprocessing/src/HierarchyManager.cpp

Timestamp:

08/24/06 18:05:53 (18 years ago)

Author:

mattausch

Message:

 

File:

• GTP/trunk/Lib/Vis/Preprocessing/src/HierarchyManager.cpp (modified) (3 diffs)

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

@@ -34 +34 @@
 
 
-HierarchyManager::HierarchyManager(VspTree &vspTree, OspTree &ospTree)
-//:mVspTree(vspTree), mOspTree(ospTree)
-{
+HierarchyManager::HierarchyManager(VspTree *vspTree, 
+                                                                   const int objectSpaceSubdivisionType):
+mObjectSpaceSubdivisonType(objectSpaceSubdivisionType),
+mVspTree(vspTree), 
+mOspTree(NULL), 
+mBvHierarchy(NULL)
+{
+        switch(mObjectSpaceSubdivisonType)
+        {
+        case KD_BASED_OBJ_SUBDIV:
+                mOspTree = new OspTree();
+                mOspTree->mVspTree = mVspTree;
+                //mOspTree->mHierarchyManager = this;
+                Debug << "creating osp tree" << endl;
+                break;
+        case BV_BASED_OBJ_SUBDIV:
+        mBvHierarchy = new BvHierarchy();
+                mBvHierarchy->mVspTree = mVspTree;
+                //mBvHierarchy->mHierarchyManager = this;
+                Debug << "creating bv hierachy" << endl;
+                break;
+        default:
+                break;
+        }
+
+        if (mVspTree)
+                mVspTree->mHierarchyManager = this;
+
         char subdivisionStatsLog[100];
         Environment::GetSingleton()->GetStringValue("Hierarchy.subdivisionStats", 
                 subdivisionStatsLog);
         mSubdivisionStats.open(subdivisionStatsLog);
+}
+
+
+HierarchyManager::HierarchyManager(VspTree *vspTree, KdTree *kdTree): 
+mObjectSpaceSubdivisonType(KD_BASED_OBJ_SUBDIV),
+mVspTree(vspTree), 
+mBvHierarchy(NULL)
+{
+        mOspTree = new OspTree(*kdTree);
+        mOspTree->mVspTree = mVspTree;
+
+        //mOspTree->mHierarchyManager = this;
+        Debug << "creating osp tree" << endl;
+
+        if (mVspTree)
+                mVspTree->mHierarchyManager = this;
+
+        char subdivisionStatsLog[100];
+        Environment::GetSingleton()->GetStringValue("Hierarchy.subdivisionStats", 
+                subdivisionStatsLog);
+        mSubdivisionStats.open(subdivisionStatsLog);
+}
+
+
+HierarchyManager::~HierarchyManager()
+{
+        DEL_PTR(mOspTree);
+        //DEL_PTR(mVspTree);
+        DEL_PTR(mBvHierarchy);
+}
+
+
+void HierarchyManager::SetViewCellsManager(ViewCellsManager *vcm)
+{
+        mVspTree->SetViewCellsManager(vcm);
+
+        if (mOspTree)
+                mOspTree->SetViewCellsManager(vcm);
+        if (mBvHierarchy)
+                mBvHierarchy->SetViewCellsManager(vcm);
+}
+
+
+void HierarchyManager::SetViewCellsTree(ViewCellsTree *vcTree)
+{
+        mVspTree->SetViewCellsTree(vcTree);
 }
 
@@ -211 +282 @@
         mVspTree.mStoreObjectPvs = false;
 #endif
-
         SubdivisionCandidate *vsc = 
                 mVspTree->PrepareConstruction(sampleRays, forcedViewSpace, *viewSpaceRays);
-
+        
         // add to queue
         mTQueue.Push(vsc);
@@ -409 +479 @@
 
 
-}
+void HierarchyManager::ExportObjectSpaceHierarchy(OUT_STREAM &stream)
+{
+        // the type of the view cells hierarchy
+        switch (mObjectSpaceSubdivisonType)
+        {
+        case KD_BASED_OBJ_SUBDIV:
+                stream << "<ObjectSpaceHierarchy type=\"osp\">" << endl;
+                mOspTree->Export(stream);
+                stream << endl << "</ObjectSpaceHierarchy>" << endl;
+                break;
+                
+        case BV_BASED_OBJ_SUBDIV:
+                stream << "<ObjectSpaceHierarchy type=\"bvh\">" << endl;
+                mBvHierarchy->Export(stream);
+                stream << endl << "</ObjectSpaceHierarchy>" << endl;
+                break;
+        }
+}
+
+
+bool HierarchyManager::AddSampleToPvs(Intersectable *obj, 
+                                                                          const Vector3 &hitPoint,
+                                                                          ViewCell *vc,
+                                                                          const float pdf, 
+                                                                          float &contribution) const
+{
+        if (!obj) return false;
+
+        switch (mObjectSpaceSubdivisonType)
+        {
+        case NO_OBJ_SUBDIV:
+                // potentially visible objects
+                return vc->AddPvsSample(obj, pdf, contribution);
+
+        case KD_BASED_OBJ_SUBDIV:
+                {
+                        // potentially visible kd cells
+                        KdLeaf *leaf = mOspTree->GetLeaf(hitPoint/*ray->mOriginNode*/);
+                        return mOspTree->AddLeafToPvs(leaf, vc, pdf, contribution);
+                }
+        case BV_BASED_OBJ_SUBDIV:
+                {
+                        BvhLeaf *leaf = mBvHierarchy->GetLeaf(obj);
+                        return mBvHierarchy->AddLeafToPvs(leaf, vc, pdf, contribution);
+                }
+        default:
+                return false;
+        }
+}
+
+
+void HierarchyManager::PrintObjectSpaceHierarchyStatistics(ofstream &stream) const
+{
+        switch (mObjectSpaceSubdivisonType)
+        {
+        case KD_BASED_OBJ_SUBDIV:
+                {
+                        stream << mOspTree->GetStatistics();
+                        break;
+                }
+        case BV_BASED_OBJ_SUBDIV:
+                {
+                        stream << mBvHierarchy->GetStatistics();
+                        break;
+                }
+        default:
+                break;
+        }
+}
+
+
+void HierarchyManager::ExportObjectSpaceHierarchyForViz(const ObjectContainer &objects) const
+{
+        if (mOspTree && mOspTree->GetRoot()) 
+        {       
+                //-- export final object partition
+                Exporter *exporter = Exporter::GetExporter("final_object_partition.wrl");
+                
+                if (exporter)
+                {
+                        cout << "exporting object space partition ... ";
+
+                        if (1)
+                        {
+                                exporter->ExportGeometry(objects);
+                        }
+                        
+#if 0           
+                        // export rays
+                        exporter->ExportRays(visRays, RgbColor(0, 1, 0));
+#endif
+
+                        exporter->SetWireframe();
+        
+                        const int colorCode = 0;
+                        const int maxPvs = 200;//mOspTree.GetStatistics().maxPvs;
+
+                        exporter->ExportOspTree(*mOspTree, colorCode == 0 ? 0 : maxPvs);
+
+                        delete exporter;
+
+                        cout << "finished" << endl;
+                }
+        }
+}
+
+}