source: GTP/trunk/App/Demos/Vis/FriendlyCulling/src/ObjExporter.cpp @ 3274

#include "ObjExporter.h"
#include "SceneEntity.h"
#include "Shape.h"
#include "Geometry.h"
#include "Transform3.h"


using namespace std;

// MAGIC of all bin exports
#ifndef MAGIC 
#define MAGIC 0x827923
#endif

#define BVH_VERSION 2.1

namespace CHCDemoEngine
{

static int currentIdx = 0;


int ObjExporter::WriteGeometry(Geometry *geom, ofstream &ostream, Transform3 *trafo)
{
        int numVertices;
        Vector3 *vertices = geom->GetVertices(numVertices);

        char str[100];

        //cout << "vtx: " << numVertices << endl;
        for (int i = 0; i < numVertices; ++ i)
        {
                Vector3 v = trafo->GetMatrix() * vertices[i];

                sprintf(str, "v %f %f %f\n", v.x, v.z, -v.y);
                ostream << str;
        }

        return numVertices;
}


bool ObjExporter::Export(const string &filename, Bvh *bvh)
{
         cout << "Exporting obj to file '" << filename.c_str() << "'" << endl;
  
         ofstream ostream(filename.c_str());

         if (!ostream.is_open()) 
         {
                 cerr << "Error: Cannot open file " << filename.c_str() << " for writing" << endl;
                 return false;
         }

         int numVertices = 0;

         BvhNodeContainer nodes;
         bvh->CollectVirtualLeaves(bvh->GetStaticRoot(), nodes);

         cout << "writing " << nodes.size() << " objects ..." << endl; 

         BvhNodeContainer::const_iterator it, it_end = nodes.end();

         int j = 0;
         const int t = 1000;

         for (it = nodes.begin(); it != it_end; ++ it, ++ j)
         {
                 if  ((j % t) == (t - 1))
                         cout << j + 1 << endl;

                 BvhNode *node = *it;

                 int geometrySize;
                 SceneEntity **entities = bvh->GetGeometry(node, geometrySize);

                 for (int i = 0; i < geometrySize; ++ i)
                 {
                         Transform3 *trafo = entities[i]->GetTransform();

                         ShapeContainer::iterator sit, send;
                         entities[i]->GetLODLevel(0, sit, send);

                         for (; sit != send; ++ sit)
                         {
                                 Geometry *geom = (*sit)->GetGeometry();
                                 
                                 int dummy;
                                 if (!geom->GetVertices(dummy))
                                 {
                                         cerr << "error: geometry was deleted on loading, exporting obj failed" << endl;
                                         return false;
                                 }

                                 numVertices += WriteGeometry(geom, ostream, trafo);
                         }
                 }
         }

         cout << "finished" << endl;
         cout << "writing " << numVertices << " indices ..." << endl;

         for (int i = 0; i < numVertices; i += 3)
         {
                 ostream << "f " << i + 1 << " " << i + 2 << " " << i + 3 << endl;
         }

         cout << "successfully exported obj" << endl;
        
         return true;
}


}