source: GTP/trunk/Lib/Vis/Preprocessing/src/SceneGraph.cpp @ 2610

//#include <boost/algorithm/string.hpp>

#include <stack>
#include "SceneGraph.h"
#include "X3dExporter.h"
#include "Intersectable.h"


using namespace std;

namespace GtpVisibilityPreprocessor {


bool
SceneGraph::Export( const string filename )
{
  if (strstr(filename.c_str(), ".x3d")) {
    X3dExporter exporter(filename);
    exporter.ExportScene(mRoot);
    return true;
  } else {
    cerr<<"Error: Currently unsuported export format, filename "<<filename<<endl;
    
  }
  
  
    return false;
  
}


SceneGraphNode::~SceneGraphNode()
{
}


SceneGraphLeaf::~SceneGraphLeaf()
{
    CLEAR_CONTAINER(mGeometry);
}


SceneGraphInterior::~SceneGraphInterior()
{
        // recursivly delete all children
        CLEAR_CONTAINER(mChildren);
}



/************************************************************/
/*                 SceneGraph implementation                */
/************************************************************/


SceneGraph::SceneGraph():
mRoot(NULL)
{
}


SceneGraph::~SceneGraph()
{
        DEL_PTR(mRoot);
}


SceneGraphInterior *SceneGraph::GetRoot()
{
        return mRoot;
}


void SceneGraph::SetRoot(SceneGraphInterior *root)
{
        mRoot = root;
}


int SceneGraph::CollectObjects(ObjectContainer &instances)
{
        instances.clear();
        int number = 0;

        stack<SceneGraphNode *> nodeStack;
        nodeStack.push(mRoot);

        while (!nodeStack.empty()) 
        {
                SceneGraphNode *node = nodeStack.top();
                nodeStack.pop();

                if (node->IsLeaf())
                {
                        SceneGraphLeaf *leaf = static_cast<SceneGraphLeaf *>(node);

                        ObjectContainer::const_iterator mi = leaf->mGeometry.begin();

                        for (; mi != leaf->mGeometry.end(); mi++)
                        {
                                instances.push_back(*mi);
                        }
                }
                else
                {
                        SceneGraphInterior *interior = static_cast<SceneGraphInterior *>(node);
                        SceneGraphNodeContainer::iterator ni = interior->mChildren.begin();

                        for (; ni != interior->mChildren.end(); ++ ni) 
                        {
                                nodeStack.push(*ni);    
                                number++;
                        }
                }
        }

        return number;
}


int
SceneGraph::AssignObjectIds()
{
        // matt: rather start with id zero
        int id = 0;
        stack<SceneGraphNode *> nodeStack;

        nodeStack.push(mRoot);

        while (!nodeStack.empty()) 
        {
                SceneGraphNode *node = nodeStack.top();
                nodeStack.pop();

                if (node->IsLeaf())
                {
                        SceneGraphLeaf *leaf = static_cast<SceneGraphLeaf *>(node);
                        ObjectContainer::iterator mi = leaf->mGeometry.begin();

                        for (; mi != leaf->mGeometry.end(); mi ++) 
                        {
                                (*mi)->SetId(id ++);
                        }
                }
                else
                {
                        SceneGraphInterior *interior = static_cast<SceneGraphInterior *>(node);

                        SceneGraphNodeContainer::iterator ni = interior->mChildren.begin();
                        
                        for (; ni != interior->mChildren.end(); ni ++) 
                        {
                                nodeStack.push(*ni);
                        }
                }
        }

        // return max id
        return id;
}

void
SceneGraph::GetStatistics(int &intersectables, int &faces) const
{
        stack<SceneGraphNode *> nodeStack;

        nodeStack.push(mRoot);
        faces = 0;
        intersectables = 0;

        while (!nodeStack.empty()) 
        {
                SceneGraphNode *node = nodeStack.top();
                nodeStack.pop();

                if (node->IsLeaf())
                {
                        SceneGraphLeaf *leaf = static_cast<SceneGraphLeaf *>(node);

                        ObjectContainer::const_iterator mi = leaf->mGeometry.begin();
                        for (; mi != leaf->mGeometry.end(); mi++) 
                        {
                                intersectables++;
                                faces += (*mi)->NumberOfFaces();
                        }
                }
                else
                {
                        SceneGraphInterior *interior = static_cast<SceneGraphInterior *>(node);

                        SceneGraphNodeContainer::iterator ni = interior->mChildren.begin();
                        for (; ni != interior->mChildren.end(); ni++) 
                        {
                                nodeStack.push(*ni);
                        }
                }
        }
}


void SceneGraphLeaf::UpdateBox()
{
        AxisAlignedBox3 box;
        box.Initialize();

        ObjectContainer::const_iterator mi = mGeometry.begin();
        for (; mi != mGeometry.end(); mi++)
                box.Include((*mi)->GetBox());

        mBox = box;
}


void SceneGraphInterior::UpdateBox()
{
        AxisAlignedBox3 box;

        box.Initialize();

        SceneGraphNodeContainer::iterator ni = mChildren.begin();
        
        for (; ni != mChildren.end(); ++ ni) 
        {
                (*ni)->UpdateBox();
                box.Include((*ni)->mBox);
        }

        mBox = box;
}



void SceneGraph::ExportScene(const string filename)
{
        
}


void SceneGraph::LoadScene(const string filename)
{
        //  load binary version of mesh
}


int SceneGraph::GetSize() const
{
        stack<SceneGraphNode *> nodeStack;

        nodeStack.push(mRoot);
        int size = 0;

        while (!nodeStack.empty()) 
        {
                SceneGraphNode *node = nodeStack.top();
                nodeStack.pop();

                if (node->IsLeaf())
                {
                        SceneGraphLeaf *leaf = static_cast<SceneGraphLeaf *>(node);

                        size += leaf->mGeometry.size();
                }
                else
                {
                        SceneGraphInterior *interior = static_cast<SceneGraphInterior *>(node);

                        SceneGraphNodeContainer::iterator ni = interior->mChildren.begin();
                        for (; ni != interior->mChildren.end(); ni++) 
                        {
                                nodeStack.push(*ni);
                        }
                }
        }

        return size;
}


void SceneGraphLeaf::Transform(const Matrix4x4 &trafo)
{
        mTrafo *= trafo;
}

}