source: GTP/trunk/Lib/Vis/Preprocessing/src/VrmlExporter.cpp @ 1344

#include <stack>
#include "common.h"
#include "SceneGraph.h"
#include "VrmlExporter.h"
#include "Mesh.h"
#include "KdTree.h"
#include "ViewCellBsp.h"
#include "ViewCell.h"
#include "Polygon3.h"
#include "VssRay.h"
#include "VssTree.h"
#include "VspBspTree.h"
#include "HierarchyManager.h"
#include "RssTree.h"
#include "Beam.h"
#include "IntersectableWrapper.h"
#include "Triangle3.h"


namespace GtpVisibilityPreprocessor {


VrmlExporter::VrmlExporter(const string filename): Exporter(filename)
{
        stream.open(mFilename.c_str());
        stream<<"#VRML V2.0 utf8"<<endl;
}


VrmlExporter::~VrmlExporter()
{
        stream.close();
}


bool VrmlExporter::ExportRays(const RayContainer &rays,
                                                          const float length,
                                                          const RgbColor &color)
{
        if (rays.empty())
                return false;
  RayContainer::const_iterator ri = rays.begin();

  stream<<"Shape {"<<endl;
  stream<<"appearance Appearance {"<<endl;
  stream << "material Material { diffuseColor=\"" << color.r << " "     <<      color.g <<      " "     
                 << color.b     << "}" << endl;
  stream<<"}"<<endl;
  
  stream<<"geometry IndexedLineSet { coordIndex [" << endl;

  int index = 0;
  for (; ri != rays.end(); ri++) 
  {
          stream << index << " " << index + 1 << " -1\n";
          index += 2;
  }
  
  stream << "]"<<endl;
  
  stream << "coord Coordinate { point [" << endl;
  
  ri = rays.begin();
  for (; ri != rays.end(); ri++) 
  {
          Vector3 a = (*ri)->GetLoc();
          
          Vector3 b;
         
          if (length < 0)
          {
                  b = (*ri)->GetLoc() - length*(*ri)->GetDir();
          }
          else
          {
                  if ((*ri)->intersections.empty())
                          b = (*ri)->GetLoc() + length*(*ri)->GetDir();
                  else
                          b = (*ri)->Extrap((*ri)->intersections[0].mT);
          }
    
          stream << a.x << " " << a.y << " " << a.z << " ,";
          stream << b.x << " " << b.y << " " << b.z << " ,\n";
  }
  stream << "]" << endl;
  stream << "}" << endl;
  stream << "}" << endl;
  stream << "}" << endl;
  
  return true;
}


bool VrmlExporter::ExportRays(const VssRayContainer &rays,
                                                          const RgbColor &color)
{

        if (rays.empty())
                return false;

  stream << "Shape {" << endl;
  stream << "appearance Appearance {" << endl;
  stream << "material Material {" << endl;
  stream << "diffuseColor " << color.r << " " << color.g << " " << color.b << endl;
  stream << "}" << endl; // end material
  stream << "}" << endl; // end appearance
 
  stream << "geometry IndexedLineSet { coordIndex [" << endl;

  int index = 0;
  VssRayContainer::const_iterator ri = rays.begin();

  for (; ri != rays.end(); ++ ri) 
  {
          stream << index << " " << index + 1 << " -1\n";
          index += 2 ;
  }
  
  stream << "]" << endl;
  
  stream << "coord Coordinate { point [" << endl;
  
  for (ri = rays.begin(); ri != rays.end(); ++ ri) 
  {
          const Vector3 a = (*ri)->GetOrigin();
          const Vector3 b = (*ri)->mTerminationObject ? (*ri)->GetTermination() : a + 1000 * Normalize((*ri)->GetDir());
          //const Vector3 b = (*ri)->GetTermination(); // matt: change back!!

          stream << a.x << " " << a.y << " " << a.z << " ,";
          stream << b.x << " " << b.y << " " << b.z << " ,\n";
  }

  stream << "]" << endl;
  stream << "}" << endl;
  stream << "}" << endl;
  stream << "}" << endl;
  //stream << "}" << endl;
        
  return true;
}


void VrmlExporter::ExportSceneNode(SceneGraphNode *node)
{
        stream << "Group {" << endl;

        SceneGraphNodeContainer::iterator ni = node->mChildren.begin();

        for (; ni != node->mChildren.end(); ni++)
        {
                ExportSceneNode(*ni);
        }
    ObjectContainer::const_iterator mi = node->mGeometry.begin();
  
        for (; mi != node->mGeometry.end(); mi++) 
        {
                // export the transform...
                ExportIntersectable(*mi);
        }
  
        stream<<"</Group>"<<endl;
}


void VrmlExporter::ExportBspLeaves(const BspTree &tree, const int maxPvs)
{
        stack<pair<BspNode *, BspNodeGeometry *> > tStack;
        ViewCell::NewMail();

        BspNodeGeometry *geom = new BspNodeGeometry();
        tree.ConstructGeometry(tree.GetRoot(), *geom);

        tStack.push(pair<BspNode *, BspNodeGeometry *>(tree.GetRoot(), geom));

        // color code pvs
        if (maxPvs)
        {
                mUseForcedMaterial = true;
        }

        while (!tStack.empty()) 
        {
                BspNode *node = tStack.top().first;
                BspNodeGeometry *cell = tStack.top().second;
                tStack.pop();

                if (!node->IsLeaf()) 
                {
                        BspInterior *interior = dynamic_cast<BspInterior *>(node);
                                                                
                        BspNodeGeometry *front = new BspNodeGeometry();
                        BspNodeGeometry *back = new BspNodeGeometry();

                        cell->SplitGeometry(*front, 
                                                                *back, 
                                                                interior->GetPlane(), 
                                                                tree.GetBoundingBox(), 
                                                                tree.GetEpsilon());

                        tStack.push(pair<BspNode *, BspNodeGeometry *>(interior->GetFront(), front));
                        tStack.push(pair<BspNode *, BspNodeGeometry *>(interior->GetBack(), back));
                }
                else
                {
                        if (maxPvs)
                        {
                                BspLeaf *leaf = dynamic_cast<BspLeaf *>(node);

                                mForcedMaterial.mDiffuseColor.b = 1.0f;
                                const float importance = (float)leaf->GetViewCell()->GetPvs().CountObjectsInPvs() / (float)maxPvs;

                                mForcedMaterial.mDiffuseColor.r = importance;
                                mForcedMaterial.mDiffuseColor.g = 1.0f - mForcedMaterial.mDiffuseColor.r;
                        }

                        ExportPolygons(cell->GetPolys());
                }
                
                DEL_PTR(cell);
        }
}


void VrmlExporter::ExportViewCell(ViewCell *viewCell)
{
        if (viewCell->GetMesh())
        {
                ExportMesh(viewCell->GetMesh());
        }
}


void VrmlExporter::ExportMesh(Mesh *mesh)
{
        stream << "Shape {" << endl;
        stream << "appearance Appearance {" << endl;
  
        // $$ tmp -> random material
    
        float r, g, b;

        if (mUseForcedMaterial) 
        {
                r = mForcedMaterial.mDiffuseColor.r;
                g = mForcedMaterial.mDiffuseColor.g;
                b = mForcedMaterial.mDiffuseColor.b;
    
        } 
        else if (mesh->mMaterial) 
        {
                r = mesh->mMaterial->mDiffuseColor.r;
                g = mesh->mMaterial->mDiffuseColor.g;
                b = mesh->mMaterial->mDiffuseColor.b;
        } 
        else 
        {
                r = RandomValue(0.5, 1.0);
                g = RandomValue(0.5, 1.0);
                b = RandomValue(0.5, 1.0);
    }

        stream << "material Material {" << endl;
        stream << "diffuseColor " << r << " " << g << " " << b << endl;
        stream << "specularColor 0.0 0.0 0.0" << endl;

        stream << "}" << endl; // end material
        stream << "}" << endl; // end apperance
        

        // wireframe modes => set indexed lines
        if (mWireframe)
        {
                stream << "geometry IndexedLineSet {" << endl;
        }
        else
        {
                //stream<<"<IndexedFaceSet ccw=\"TRUE\" coordIndex=\""<<endl;
                stream << "geometry IndexedFaceSet {" << endl;
        }


        stream << "coordIndex [" << endl;

        FaceContainer::const_iterator fi = mesh->mFaces.begin();

        int index = 0;
  
        for (; fi != mesh->mFaces.end(); fi++) 
        {
                Face *face = *fi;
         
                VertexIndexContainer::const_iterator vi = face->mVertexIndices.begin();
         
                for (; vi != face->mVertexIndices.end(); ++ vi)
                {
                        stream << *vi << " ";
                }
          
                if (mWireframe) // final line to finish polygon
                {
                        stream << (*face->mVertexIndices.begin()) << " ";
                }
                
                stream << "-1" << endl;
        }

        stream << "]" << endl; // end coordIndex


        stream << "coord Coordinate {" << endl;
        stream << "point [" << endl;
  
        VertexContainer::const_iterator vi = mesh->mVertices.begin();

        for (; vi != mesh->mVertices.end(); vi++) 
        {
                stream << (*vi).x << " " << (*vi).y << " " << (*vi).z;
                stream << "," << endl;
        }
  
        stream << "]" << endl; // end point
        stream << "}" << endl; // end coord
        stream << "}" << endl; // end geometry

        stream << "}" << endl; // end shape
}


void VrmlExporter::ExportPolygon(Polygon3 *poly)
{
        stream << "Shape {" << endl;
        stream << "appearance Appearance {" << endl;
  
        // $$ tmp -> random material
        float r, g, b;

        if (mUseForcedMaterial) 
        {
                r = mForcedMaterial.mDiffuseColor.r;
                g = mForcedMaterial.mDiffuseColor.g;
                b = mForcedMaterial.mDiffuseColor.b;
        } 
        else if (poly->mMaterial) 
        {
                r = poly->mMaterial->mDiffuseColor.r;
                g = poly->mMaterial->mDiffuseColor.g;
                b = poly->mMaterial->mDiffuseColor.b;
        } 
        else 
        {
                r = RandomValue(0.5, 1.0);
                g = RandomValue(0.5, 1.0);
                b = RandomValue(0.5, 1.0);
        }

        stream << "material Material {" << endl;
        stream << "diffuseColor " << r << " " << g << " " << b << endl;
        stream << "specularColor 0.0 0.0 0.0" << endl;

        stream << "}" << endl; // end material
        stream << "}" << endl; // end apperance
        

        // wireframe modes => set indexed lines
        if (mWireframe)
        {
                stream << "geometry IndexedLineSet {" << endl;
        }
        else
        {
                //stream<<"<IndexedFaceSet ccw=\"TRUE\" coordIndex=\""<<endl;
                stream << "geometry IndexedFaceSet {" << endl;
        }


        stream << "coordIndex [" << endl;


        int index = 0;
        
        VertexContainer::const_iterator vi;  
        
        for (index = 0; index < (int)poly->mVertices.size(); ++ index) 
        {
                stream << index << " ";
        }

        if (mWireframe) // final line to finish polygon
        {
                stream << "0 ";
        }

        stream << "-1" << endl;
    
        stream << "]" << endl; // end coordIndex
        

        stream << "coord Coordinate {" << endl;
        stream << "point [" << endl;

        for (vi = poly->mVertices.begin(); vi != poly->mVertices.end(); ++vi) 
        {
                stream << (*vi).x << " " << (*vi).y << " " << (*vi).z;
                stream << "," << endl;
        }
  
        stream << "]" << endl; // end point
        stream << "}" << endl; // end coord
        stream << "}" << endl; // end geometry

        stream << "}" << endl; // end shape
}


void VrmlExporter::ExportPolygons(const PolygonContainer &polys)
{
        stream << "Shape {" << endl;
        stream << "appearance Appearance {" << endl;
  
        // $$ tmp -> random material
  
        float r, g, b;

        if (mUseForcedMaterial) 
        {
                r = mForcedMaterial.mDiffuseColor.r;
                g = mForcedMaterial.mDiffuseColor.g;
                b = mForcedMaterial.mDiffuseColor.b;
        } 
        else 
        {
                r = RandomValue(0.5, 1.0);
                g = RandomValue(0.5, 1.0);
                b = RandomValue(0.5, 1.0);
        }

        stream << "material Material {" << endl;
        stream << "diffuseColor " << r << " " << g << " " << b << endl;
        stream << "specularColor 0.0 0.0 0.0" << endl;

        stream << "}" << endl; // end material
        stream << "}" << endl; // end apperance

        // wireframe modes => set indexed lines
        if (mWireframe)
        {
                stream << "geometry IndexedLineSet {" << endl;
        }
        else
        {
                //stream<<"<IndexedFaceSet ccw=\"TRUE\" coordIndex=\""<<endl;
                stream << "geometry IndexedFaceSet {" << endl;
        }

        stream << "coordIndex [" << endl;

        int index = 0;
        
        PolygonContainer::const_iterator pit;

    VertexContainer::const_iterator vi;  
        
        for (pit = polys.begin(); pit != polys.end(); ++ pit)
        {
                Polygon3 *poly = *pit;
                int startIdx = index;

                for (vi = poly->mVertices.begin(); vi != poly->mVertices.end(); ++vi) 
                {
                        stream << index ++ << " ";
                }

                stream << startIdx << " ";// finish line
                stream << "-1" << endl;
        }

        stream << "]" << endl; // end coordIndex        

        stream << "coord Coordinate {" << endl;
        stream << "point [" << endl;
  

        for (pit = polys.begin(); pit != polys.end(); ++ pit)
        {
                Polygon3 *poly = *pit;
        for (vi = poly->mVertices.begin(); vi != poly->mVertices.end(); ++vi) 
                {
                        stream << (*vi).x << " " << (*vi).y << " " << (*vi).z;
                        stream << "," << endl;
                }
        }

        stream << "]" << endl; // end point
        stream << "}" << endl; // end coord
        stream << "}" << endl; // end geometry

        stream << "}" << endl; // end shape
}


bool VrmlExporter::ExportBox(const AxisAlignedBox3 &box)
{
        Mesh *mesh = new Mesh;

    // add 8 vertices of the box
        int index = (int)mesh->mVertices.size();
        
        for (int i=0; i < 8; i++) 
        {
                Vector3 v;
                box.GetVertex(i, v);
                mesh->mVertices.push_back(v);
        }
  
        mesh->AddFace(new Face(index + 0, index + 1, index + 3, index + 2) );
        mesh->AddFace(new Face(index + 0, index + 2, index + 6, index + 4) );
        mesh->AddFace(new Face(index + 4, index + 6, index + 7, index + 5) );
  
        mesh->AddFace(new Face(index + 3, index + 1, index + 5, index + 7) );
        mesh->AddFace(new Face(index + 0, index + 4, index + 5, index + 1) );
        mesh->AddFace(new Face(index + 2, index + 3, index + 7, index + 6) );
  
        ExportMesh(mesh);
        
        delete mesh;
        
        return true;
}

bool VrmlExporter::ExportBspTree(const BspTree &tree)
{
        if (mExportRayDensity) 
        {
                return ExportBspTreeRayDensity(tree);
        }
  
        bool savedWireframe = mWireframe;

        SetWireframe();
        
        ExportBox(tree.GetBoundingBox());
        
        if (!savedWireframe)
                SetFilled();

        // export view cells
        ExportBspLeaves(tree);  

        return true;
}


bool VrmlExporter::ExportKdTree(const KdTree &tree, const bool exportGeometry)
{
         if (mExportRayDensity) 
         {
                 return ExportKdTreeRayDensity(tree);
         }
  
         stack<KdNode *> tStack;

         tStack.push(tree.GetRoot());

         Mesh *mesh = new Mesh;
  
         while (!tStack.empty()) 
         {
                 KdNode *node = tStack.top();

                 tStack.pop();

                 AxisAlignedBox3 box = tree.GetBox(node);
                 // add 6 vertices of the box
                 int index = (int)mesh->mVertices.size();

                 for (int i=0; i < 8; i++) 
                 {
                         Vector3 v;
                         box.GetVertex(i, v);
                         mesh->mVertices.push_back(v);
                 }

                 mesh->AddFace(new Face(index + 0, index + 1, index + 3, index + 2) );
                 mesh->AddFace(new Face(index + 0, index + 2, index + 6, index + 4) );
                 mesh->AddFace(new Face(index + 4, index + 6, index + 7, index + 5) );

                 mesh->AddFace(new Face(index + 3, index + 1, index + 5, index + 7) );
                 mesh->AddFace(new Face(index + 0, index + 4, index + 5, index + 1) );
                 mesh->AddFace(new Face(index + 2, index + 3, index + 7, index + 6) );

                 if (!node->IsLeaf()) 
                 {
                         KdInterior *interior = (KdInterior *)node;
                         tStack.push(interior->mFront);
                         tStack.push(interior->mBack);
                 }
         }

         ExportMesh(mesh);

         delete mesh;
         return true;
}


bool VrmlExporter::ExportVssTree(const VssTree &tree)
{
        stack<VssTreeNode *> tStack;
        
        tStack.push(tree.GetRoot());
        
        Mesh *mesh = new Mesh;
        VssRayContainer rays;
        
        while (!tStack.empty()) 
        {
        VssTreeNode *node = tStack.top();
                
                tStack.pop();

                if (!node->IsLeaf()) 
                {
                        VssTreeInterior *interior = (VssTreeInterior *)node;
                        
                        tStack.push(interior->front);
                        tStack.push(interior->back);
                } 
                else 
                {
                        VssTreeLeaf *leaf = (VssTreeLeaf *)node;
                        AxisAlignedBox3 box;
                        box = tree.GetBBox(leaf);
                        IncludeBoxInMesh(box, *mesh);

                        if (tree.ValidLeaf(leaf)) 
                        {
                                Vector3 origin = box.Center();
                                box = tree.GetDirBBox(leaf);
                                VssRay *ray;
                                
                                const int indices[][2] = {{0,0}, {0,1}, {1,1}, {1,0}}; 
                                MeshInstance dummy(mesh);
                                for (int i=0; i < 4; i++) 
                                {
                                        // Vector3 v = box.GetVertex(indices[i][0], indices[i][1], 0);
                                        Vector3 v = box.Center();
                                        
                                        Vector3 direction = VssRay::GetDirection(v.x, v.y);
                                        if (Magnitude(direction) > Limits::Small)
                                        {
                                                direction.Normalize();
                                        }
                                        else
                                        {
                                                direction = Vector3(0, 1, 0);
                                        }

                                        const float k = 100.0f * leaf->GetImportance();


                                        // get 4 corners of the ray directions
                                        ray = new VssRay(origin, origin + (direction * k), NULL, &dummy);
                                        rays.push_back(ray);
                                }
                        }
                }
        }

        ExportMesh(mesh);
        ExportRays(rays);
        CLEAR_CONTAINER(rays);

        delete mesh;
        return true;
}


bool VrmlExporter::ExportVssTree2(const VssTree &tree,
                                                                  const Vector3 direction)
{
        stack<VssTreeNode *> tStack;
        
        mUseForcedMaterial = true;

        Vector3 dirParam;

        dirParam.x = VssRay::GetDirParam(0, Normalize(direction));
        dirParam.y = VssRay::GetDirParam(1, Normalize(direction));

        float maxImportance = 0.0f;
        
        tStack.push(tree.GetRoot());
        
        while (!tStack.empty()) 
        {
        VssTreeNode *node = tStack.top();
                tStack.pop();

                if (!node->IsLeaf()) 
                {
                        VssTreeInterior *interior = (VssTreeInterior *)node;
                        
                        if (interior->axis < 3) 
                        {
                                tStack.push(interior->front);
                                tStack.push(interior->back);
                        } 
                        else 
                        {
                                if (dirParam[interior->axis-3] < interior->position)
                                        tStack.push(interior->back);
                                else
                                        tStack.push(interior->front);
                        }
                } 
                else 
                {
                        VssTreeLeaf *leaf = (VssTreeLeaf *)node;
                        
                        if (tree.ValidLeaf(leaf)) 
                        {
                                float i = leaf->GetImportance();

                                if (i > maxImportance)
                                        maxImportance = i;
                        }
                }
        }

        tStack.push(tree.GetRoot());
        
        while (!tStack.empty()) 
        {
                VssTreeNode *node = tStack.top();
                tStack.pop();
        
                if (!node->IsLeaf()) 
                {
                        VssTreeInterior *interior = (VssTreeInterior *)node;
            
                        if (interior->axis < 3) 
                        {
                                tStack.push(interior->front);
                                tStack.push(interior->back);
                        } 
                        else 
                        {
                                if (dirParam[interior->axis-3] < interior->position)
                                        tStack.push(interior->back);
                                else
                                        tStack.push(interior->front);
                        }
                } 
                else 
                {
                        VssTreeLeaf *leaf = (VssTreeLeaf *)node;
                        
                        if (tree.ValidLeaf(leaf)) 
                        {
                                AxisAlignedBox3 box;
                                box = tree.GetBBox(leaf);
                                Mesh *mesh = new Mesh;
                                IncludeBoxInMesh(box, *mesh);

                                // get 4 corners of the ray directions
                
                                mForcedMaterial.mDiffuseColor.b = 1.0f;
                                mForcedMaterial.mDiffuseColor.r = leaf->GetImportance()/maxImportance;
                                mForcedMaterial.mDiffuseColor.g = 1.0f - mForcedMaterial.mDiffuseColor.r;
                
                                ExportMesh(mesh);
                                delete mesh;
                        }
                }
        }

        mUseForcedMaterial = false;

        return true;
}


bool VrmlExporter::ExportBspTreeRayDensity(const BspTree &tree)
{
        stack<BspNode *> tStack;

        tStack.push(tree.GetRoot());

        bool fm = mUseForcedMaterial;
        
        mUseForcedMaterial = true;
        
        mForcedMaterial.mDiffuseColor.g = 1.0f;
        mForcedMaterial.mDiffuseColor.b = 1.0f;
  
        while (!tStack.empty()) 
        {
                BspNode *node = tStack.top();
                tStack.pop();

                if (node->IsLeaf()) 
                {
                        ViewCell *vc = dynamic_cast<BspLeaf *>(node)->GetViewCell();
#if 0     
                        // set the mesh material according to the ray density
                        if (vc->mPassingRays.mRays) 
                        {
                                float importance = 
                                  vc->mPassingRays.mContributions / (float)vc->mPassingRays.mRays;
                                
                                mForcedMaterial.mDiffuseColor.r = importance;
                                mForcedMaterial.mDiffuseColor.g = 1.0f - mForcedMaterial.mDiffuseColor.r;
                                ExportViewCell(vc);

                        }
#endif
                } 
                else 
                {
                        BspInterior *interior = (BspInterior *)node;
                        tStack.push(interior->GetFront());
                        tStack.push(interior->GetBack());
                }
        }
  
        // restore the state of forced material
        mUseForcedMaterial = fm;

        return true;
}


bool VrmlExporter::ExportKdTreeRayDensity(const KdTree &tree)
{
        stack<KdNode *> tStack;

        tStack.push(tree.GetRoot());

        bool fm = mUseForcedMaterial;
        
        mUseForcedMaterial = true;
    
        mForcedMaterial.mDiffuseColor.g = 1.0f;
        mForcedMaterial.mDiffuseColor.b = 1.0f;


        while (!tStack.empty()) 
        {
                KdNode *node = tStack.top();
                tStack.pop();

                if (node->IsLeaf()) 
                {
                        AxisAlignedBox3 box = tree.GetBox(node);
                        Mesh *mesh = new Mesh;
      
                        // add 6 vertices of the box
                        int index = (int)mesh->mVertices.size();
                        for (int i=0; i < 8; i++) 
                        {
                                Vector3 v;
                                box.GetVertex(i, v);
                                mesh->mVertices.push_back(v);
                        }
                        
                        mesh->AddFace(new Face(index + 0, index + 1, index + 3, index + 2) );
                        mesh->AddFace(new Face(index + 0, index + 2, index + 6, index + 4) );
                        mesh->AddFace(new Face(index + 4, index + 6, index + 7, index + 5) );
            
                        mesh->AddFace(new Face(index + 3, index + 1, index + 5, index + 7) );
                mesh->AddFace(new Face(index + 0, index + 4, index + 5, index + 1) );
                mesh->AddFace(new Face(index + 2, index + 3, index + 7, index + 6) );

                        // set the mesh material according to the ray density
                        KdLeaf *leaf = (KdLeaf *) node;
                        
                        if (leaf->mPassingRays.mRays) 
                        {
                                const float importance = 
                                        leaf->mPassingRays.mContributions/(float)leaf->mPassingRays.mRays;
                                //      float importance = leaf->mPassingRays.mContributions/1000.0f;
                                //      float importance = leaf->mPassingRays.mRays/1000.0f;
                                ///(float)leaf->mPassingRays.mRays;
                                // mForcedMaterial.mDiffuseColor.r = log10(leaf->mPassingRays.mRays)/3.0f;
                                
                                mForcedMaterial.mDiffuseColor.r = importance;
                                mForcedMaterial.mDiffuseColor.g = 1.0f - mForcedMaterial.mDiffuseColor.r;
                                
                                ExportMesh(mesh);
                        }
                        
                        delete mesh;
                } 
                else 
                {
                        KdInterior *interior = (KdInterior *)node;
                        tStack.push(interior->mFront);
                        tStack.push(interior->mBack);
                }
        }
        
        // restore the state of forced material
        mUseForcedMaterial = fm;
        
        return true;
}


struct BspSplitData
{
        /// the current node
        BspNode *mNode;

        vector<Plane3> mPlanes;
        vector<bool> mSides;
        bool mIsFront;
        int mDepth;

        BspSplitData(BspNode *node): 
        mNode(node), mIsFront(false), mDepth(0)
        {};     

        BspSplitData(BspNode *node, 
                                vector<Plane3> planes, 
                                vector<bool> sides, 
                                const bool isFront,
                                const int depth): 
        mNode(node), mPlanes(planes), mSides(sides), 
        mIsFront(isFront), mDepth(depth)
        {};
};


void VrmlExporter::ExportLeavesGeometry(const BspTree &tree, 
                                                                           const vector<BspLeaf *> &leaves)
{
        vector<BspLeaf *>::const_iterator it, it_end = leaves.end();

        for (it = leaves.begin(); it != it_end; ++ it)
        {
                BspNodeGeometry geom;
                tree.ConstructGeometry(*it, geom);
                
                ExportPolygons(geom.GetPolys());
        }
}


void VrmlExporter::ExportBspNodeSplits(BspNode *root,
                                                                          const AxisAlignedBox3 &box, 
                                                                          const bool exportDepth,
                                                                          const bool epsilon)
{
        std::stack<BspSplitData> tStack;

        BspSplitData tData(root);
        tStack.push(tData);
  
        PolygonContainer polys;
        vector <int> depths;

        int maxDepth = 0;

        while (!tStack.empty())
        {
                // filter polygons donw the tree
                BspSplitData tData = tStack.top();
            tStack.pop();       
                
                if (tData.mNode->IsLeaf())
                {
                        if (tData.mDepth > maxDepth)
                                maxDepth = tData.mDepth;
                }
                else
                {
                        BspInterior *interior = dynamic_cast<BspInterior *>(tData.mNode);

                        // add current side of split plane
                        if (tData.mNode != root)
                                tData.mSides.push_back(tData.mIsFront);

                        // bounded plane is added to the polygons
                        Polygon3 *planePoly = 
                                box.CrossSection(interior->GetPlane());
                
                        // do all the splits with the previous planes
                        for (int i = 0; i < (int)tData.mPlanes.size(); ++ i)
                        {                               
                                if (planePoly->ClassifyPlane(tData.mPlanes[i], epsilon) 
                                        == Polygon3::SPLIT)
                                {
                                        Polygon3 *frontPoly = new Polygon3();
                                        Polygon3 *backPoly = new Polygon3();

                                        planePoly->Split(tData.mPlanes[i], 
                                                                         *frontPoly, 
                                                                         *backPoly,
                                                                         epsilon);

                                        DEL_PTR(planePoly);

                                        if(tData.mSides[i] == true)
                                        {
                                                planePoly = frontPoly;
                                                DEL_PTR(backPoly);
                                        }
                                        else
                                        {
                                                planePoly = backPoly;
                                                DEL_PTR(frontPoly);
                                        }
                                }
                        }

                        tData.mPlanes.push_back(interior->GetPlane()); // add plane to split planes

                        if (planePoly->Valid(epsilon))
                        {
                                polys.push_back(planePoly);
                                depths.push_back(tData.mDepth);
                        }
                        else
                                DEL_PTR(planePoly);
                        
                        // push the children on the stack
                        tStack.push(BspSplitData(interior->GetFront(), tData.mPlanes, 
                                                     tData.mSides, true, tData.mDepth + 1));
                        tStack.push(BspSplitData(interior->GetBack(), tData.mPlanes, 
                                                     tData.mSides, false, tData.mDepth + 1));
                }
        }

        if (maxDepth > 0)
        {       
                mUseForcedMaterial = true;
                        
                for (int i = 0; i < (int)polys.size(); ++ i)
                {
                        mForcedMaterial.mDiffuseColor.b = 1.0f;
                        float importance =  (float)depths[i]/ (float)maxDepth;
            
                        mForcedMaterial.mDiffuseColor.r = importance;
                        mForcedMaterial.mDiffuseColor.g = 1.0f - mForcedMaterial.mDiffuseColor.r;

                        ExportPolygon(polys[i]);
                }
        }
        else
        {
                ExportPolygons(polys);
        }

        CLEAR_CONTAINER(polys);
}


void VrmlExporter::ExportBspSplits(const BspTree &tree,
                                                                  const bool exportDepth)
{
        ExportBspNodeSplits(tree.GetRoot(), 
                                                tree.GetBoundingBox(), 
                                                exportDepth, 
                                                tree.GetEpsilon());
}


void VrmlExporter::ExportBspSplits(const VspBspTree &tree,
                                                                  const bool exportDepth)
{
        ExportBspNodeSplits(tree.GetRoot(), 
                                                tree.GetBoundingBox(), 
                                                exportDepth, 
                                                tree.GetEpsilon());
}


void VrmlExporter::ExportBspSplitPlanes(const BspTree &tree)
{
        std::stack<BspNode *> tStack;

        tStack.push(tree.GetRoot());
  
        PolygonContainer polys;

        while (!tStack.empty())
        {
                // filter polygons donw the tree
                BspNode *node = tStack.top();
            tStack.pop();       
                
                if (!node->IsLeaf())
                {
                        BspInterior *interior = dynamic_cast<BspInterior *>(node);

                        // bounded plane is added to the polygons
                        polys.push_back(tree.GetBoundingBox().CrossSection(interior->GetPlane()));
                
                        // push the children on the stack
                        tStack.push(interior->GetBack());
                        tStack.push(interior->GetFront());
                }
        }

        ExportPolygons(polys);
        CLEAR_CONTAINER(polys);
}


void VrmlExporter::ExportGeometry(const ObjectContainer &objects)
{
        ObjectContainer::const_iterator oit, oit_end = objects.end();

        if (1)
        {
                // export using native or random material
                for (oit = objects.begin(); oit != oit_end; ++ oit)
                {
                        if (0)
                        {
                                mUseForcedMaterial = true;
                                SetForcedMaterial(RandomMaterial());
                        }
                        ExportIntersectable(*oit);
                }
                return;
        }

        ///////////////////////////////////
        // method 2 (hacky)
        // all object exported as one mesh
        PolygonContainer polys;

        for (oit = objects.begin(); oit != oit_end; ++ oit)
        {
                MeshInstance *mi = dynamic_cast<MeshInstance *>(*oit);
                polys.push_back(new Polygon3(mi->GetMesh()->mVertices));
        }

        Mesh dummyMesh;
        PolygonContainer::const_iterator pit, pit_end = polys.end();

        for (pit = polys.begin(); pit != pit_end; ++ pit)
        {
                Polygon3 *poly = (*pit);
                IncludePolyInMesh(*poly, dummyMesh);
        }
        
        ExportMesh(&dummyMesh);

        CLEAR_CONTAINER(polys);
}



bool VrmlExporter::ExportRssTree2(const RssTree &tree,
                                                                  const Vector3 direction)
{
  stack<RssTreeNode *> tStack;
  
  
  mUseForcedMaterial = true;

  Vector3 dirParam;

  dirParam.x = VssRay::GetDirParam(0, Normalize(direction));
  dirParam.y = VssRay::GetDirParam(1, Normalize(direction));

  float maxImportance = 0.0f;
  tStack.push(tree.GetRoot());
  while (!tStack.empty()) {
        
        RssTreeNode *node = tStack.top();
    tStack.pop();
        
    if (!node->IsLeaf()) {
      RssTreeInterior *interior = (RssTreeInterior *)node;
          if (interior->axis < 3) {
                tStack.push(interior->front);
                tStack.push(interior->back);
          } else {
                if (dirParam[interior->axis-3] < interior->position)
                  tStack.push(interior->back);
                else
                  tStack.push(interior->front);
          }
    } else {
          RssTreeLeaf *leaf = (RssTreeLeaf *)node;
          if (tree.ValidLeaf(leaf)) {
                float i = leaf->GetImportance();
                if (i > maxImportance)
                  maxImportance = i;
          }
        }
  }

  tStack.push(tree.GetRoot());
  while (!tStack.empty()) {

        RssTreeNode *node = tStack.top();
    tStack.pop();
        
                        
    if (!node->IsLeaf()) {
      RssTreeInterior *interior = (RssTreeInterior *)node;
          if (interior->axis < 3) {
                tStack.push(interior->front);
                tStack.push(interior->back);
          } else {
                if (dirParam[interior->axis-3] < interior->position)
                  tStack.push(interior->back);
                else
                  tStack.push(interior->front);
          }
    } else {
          RssTreeLeaf *leaf = (RssTreeLeaf *)node;
          if (tree.ValidLeaf(leaf)) {
                AxisAlignedBox3 box;
                box = tree.GetShrankedBBox(leaf);
                Mesh *mesh = new Mesh;
                IncludeBoxInMesh(box, *mesh);
                
                // get 4 corners of the ray directions
                
                mForcedMaterial.mDiffuseColor.b = 1.0f;
                mForcedMaterial.mDiffuseColor.r = leaf->GetImportance()/maxImportance;
                mForcedMaterial.mDiffuseColor.g = 1.0f - mForcedMaterial.mDiffuseColor.r;
                
                ExportMesh(mesh);
                delete mesh;
          }
        }
  }

  mUseForcedMaterial = false;

  return true;
}


void
VrmlExporter::ExportViewpoint(const Vector3 &point,
                                                         const Vector3 &direction)
{
        stream << "Viewpoint {" << endl;
  
        stream << "position " << point.x << " " << point.y << " " << point.z << endl;
  //  stream<<"orientation "<<direction.x<<direction.y<<direction.z<<endl;
  
  stream << "}" << endl; // end view point;

}


void VrmlExporter::ExportBeam(const Beam &beam, const AxisAlignedBox3 &box)
{
        if (beam.mMesh)
        {
                ExportMesh(beam.mMesh);
                return;
        }

        PolygonContainer polys;
        //ExportBox(beam.mBox);

        const float zfar = 2.0f * Magnitude(box.Diagonal());

        // box should not never remove part of beam polygons
        Vector3 bmin = beam.mBox.Min() - Vector3(zfar * 2.0f);
        Vector3 bmax = beam.mBox.Max() + Vector3(zfar * 2.0f);

        AxisAlignedBox3 bbox(bmin, bmax);
        Plane3 fplane;
        fplane.mNormal = -beam.mPlanes[0].mNormal;

        fplane.mD = beam.mPlanes[0].mD - zfar - 1.0f; 
        
        vector<Plane3> planes = beam.mPlanes;
        planes.push_back(fplane);

        for (int i = 0; i < planes.size(); ++ i)
        {
                Polygon3 *poly = bbox.CrossSection(planes[i]);
                if (!poly->Valid(Limits::Small))
                        DEL_PTR(poly);
                
                for (int j = 0; (j < planes.size()) && poly; ++ j)
                {
                        if (j != i)
                        {
                                Polygon3 *front = new Polygon3();
                                Polygon3 *back = new Polygon3();
                                
                                poly->Split(planes[j], *front, *back, Limits::Small);
                                DEL_PTR(poly);
                                DEL_PTR(front);

                                if (!back->Valid(Limits::Small))
                                        DEL_PTR(back);
                                poly = back;
                        }
                }
                if (poly)
                        polys.push_back(poly);
        }

        ExportPolygons(polys);
        CLEAR_CONTAINER(polys);
}

}