/cygdrive/d/src/svn/vrut/trunk/core/src/base_modules/iostl/stlwriter.cpp

Go to the documentation of this file.

/*
 * $Id: stlwriter.cpp,v 1.56 2008-06-25 22:50:29 kybav1 Exp $
 *
 * Description  : One line description of file.
 * Author       : Antonin Misek <antonin.misek@skoda-auto.cz>
 * 
 * Purpose      :
 *     Long description of what the file is for.
 */


#include "stlwriter.h"
#include "../../textwriter.h"

using namespace VRUT;


STLWriter::STLWriter(wxBufferedOutputStream * _outputStream, const Scene * _scene) : outputStream(_outputStream), scene(_scene)
{
}

STLWriter::~STLWriter()
{
}

void STLWriter::writeASCII()
{
       TextWriter textWriter((wxBufferedOutputStream *)outputStream);
       //write header
       textWriter.Write("solid ")->Write(scene->GetName())->Write("\n");

       //write facets
       for (std::vector<STLBinary>::const_iterator facetIT = facets.begin(); facetIT != facets.end(); facetIT++)
       {
              textWriter.Write("   facet normal ")->Write(facetIT->norm, 3, " ")->Write("\n      outer loop\n         vertex ");
              textWriter.Write(facetIT->vert1, 3, " ")->Write("\n         vertex ")->Write(facetIT->vert2, 3, " ");
              textWriter.Write("\n         vertex ")->Write(facetIT->vert3, 3, " ")->Write("\n      endloop\n   endfacet\n");
       }
       
       //write tail
       textWriter.Write("endsolid ")->Write(scene->GetName())->Write("\n");
}

void STLWriter::writeBinary()
{
       //write header
       char header[80];
       strncpy(header, scene->GetName().mb_str(wxConvISO8859_1), 80);
       outputStream->Write(header, 80);
       unsigned long count = (unsigned long)facets.size();
       outputStream->Write(&count, 4);

       //write facets
       wxString facet;
       for (std::vector<STLBinary>::const_iterator facetIT = facets.begin(); facetIT != facets.end(); facetIT++)
       {
              outputStream->Write(&(*facetIT), 50);
       }
}

void STLWriter::Write(bool saveBinary)
{
       //build facets list
       std::vector<NODE_ID> nodes;
       scene->GetNodeIDs(&nodes);

       for (std::vector<NODE_ID>::const_iterator nodeIt = nodes.begin();
              nodeIt != nodes.end(); nodeIt++)
       {
              const SceneNode * node = scene->GetNode(*nodeIt);
              if (!node || !node->IsOfType(SceneNode::GEOMETRY))
                     continue;
              scene->UpdateTransformation(*nodeIt);

              const Geometry * geom = scene->GetGeometry(((const GeometryNode *)node)->GetGeometryID());
              if (geom && geom->type == Geometry::GEOMETRY_TRIANGLE)
              {
                     std::vector<Triangle> triangles = geom->Triangulate();
                     MATRIX transf(*node->GetWorldTransMatrix());
                     for (size_t i=0; i<triangles.size(); i++)
                     {
                            STLBinary facet;
                            memcpy(facet.vert1, transf.TransformCoord(triangles.at(i).v1)._v, sizeof(float)*3);
                            memcpy(facet.vert2, transf.TransformCoord(triangles.at(i).v2)._v, sizeof(float)*3);
                            memcpy(facet.vert3, transf.TransformCoord(triangles.at(i).v3)._v, sizeof(float)*3);
                            memcpy(facet.norm, (VECTOR3(facet.vert2) - VECTOR3(facet.vert1)).Cross(VECTOR3(facet.vert3) - VECTOR3(facet.vert1)).Normalize()._v, sizeof(float)*3);
                            if (_isnan(facet.norm[0]) || _isnan(facet.norm[1]) || _isnan(facet.norm[2]))
                                   continue;//skip degenerated triangles
                            facet.attr[0]=0;
                            facet.attr[1]=0;
                            facets.push_back(facet);
                     }
              }
       }
       if (saveBinary)
              writeBinary();
       else
              writeASCII();
}

---