STLIO.cpp

Go to the documentation of this file.

/*
 * STLIO.cpp
 *
 *  Created on: Dec 9, 2016
 *      Author: robot
 */


#include "lvr2/io/STLIO.hpp"
#include "lvr2/io/Timestamp.hpp"
#include "lvr2/geometry/BaseVector.hpp"
#include "lvr2/geometry/Normal.hpp"

#include <iostream>
#include <cstdio>
#include <cstring>
#include <stdint.h>
#include <fstream>

using std::cout;
using std::endl;

namespace lvr2
{

STLIO::STLIO() {
        // TODO Auto-generated constructor stub

}

STLIO::~STLIO() {
        // TODO Auto-generated destructor stub
}

ModelPtr STLIO::read(string filename)
{
        return ModelPtr(new Model);
}

void STLIO::save( string filename )
{
        save(this->m_model, filename);
}

void STLIO::save( ModelPtr model, string filename )
{

        MeshBufferPtr mesh = model->m_mesh;
        size_t n_vert = mesh->numVertices();
        size_t n_faces = mesh->numFaces();
        floatArr vertices = mesh->getVertices();
        indexArray indices = mesh->getFaceIndices();

        std::string header_info = "Created by LVR";
        char head[80];
        std::strncpy(head,header_info.c_str(),sizeof(head)-1);
        char attribute[2] = "0";

        std::ofstream myfile(filename.c_str());

        myfile.write(head,sizeof(head));
        myfile.write((char*)&n_faces,4);

        if(myfile.good())
        {
                for(int i = 0; i < n_faces; i++)
                {
                        int a = (int)indices[3 * i];
                        int b = (int)indices[3 * i + 1];
                        int c = (int)indices[3 * i + 2];

            using Vec = BaseVector<float>;
                        Vec v1;
                        Vec v2;
                        Vec v3;

                        v1.x = vertices[3 * a];
                        v1.y = vertices[3 * a + 1];
                        v1.z = vertices[3 * a + 2];

                        v2.x = vertices[3 * b];
                        v2.y = vertices[3 * b + 1];
                        v2.z = vertices[3 * b + 2];

                        v3.x = vertices[3 * c];
                        v3.y = vertices[3 * c + 1];
                        v3.z = vertices[3 * c + 2];

                        Normal<float> normal( (v1 - v2).cross(v1 - v3));

                        myfile.write( (char*)&normal.x, 4);
                        myfile.write( (char*)&normal.y, 4);
                        myfile.write( (char*)&normal.z, 4);

                        myfile.write( (char*)&v1.x, 4);
                        myfile.write( (char*)&v1.y, 4);
                        myfile.write( (char*)&v1.z, 4);

                        myfile.write( (char*)&v2.x, 4);
                        myfile.write( (char*)&v2.y, 4);
                        myfile.write( (char*)&v2.z, 4);

                        myfile.write( (char*)&v3.x, 4);
                        myfile.write( (char*)&v3.y, 4);
                        myfile.write( (char*)&v3.z, 4);

                        uint16_t u = 0;
                        myfile.write( (char*)attribute, 2 );
                }
        }
        else
        {
                cout << timestamp << "Could not open file " << filename << " for writing." << endl;
        }
}

} /* namespace lvr2 */