procedural-buildings/PLYWriter.cpp at main · santipaprika/procedural-buildings · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#include "PLYWriter.h"

bool PLYWriter::writeMesh(const std::string &filename, const Mesh &mesh)
{
    std::ofstream fout(filename);
    writeHeader(fout, mesh);
    writeVertices(fout, mesh);
    writeFaces(fout, mesh);
    return true;
}

void PLYWriter::writeHeader(std::ofstream &fout, const Mesh &mesh)
{
    fout << "ply\n";
    fout << "format binary_little_endian 1.0\n";
    fout << "element vertex " << mesh.vertices.size() << "\n";
    fout << "property float x\n";
    fout << "property float y\n";
    fout << "property float z\n";
    fout << "element face " << mesh.faces.size() << "\n";
    fout << "property list uchar int vertex_index\n";
    fout << "end_header\n";
}

void PLYWriter::writeVertices(std::ofstream &fout, const Mesh &mesh)
{
    for (int i = 0; i < mesh.vertices.size(); ++i) {
        glm::vec3 vertex = mesh.vertices[i];
        fout.write(reinterpret_cast<char*>(&vertex.x), sizeof(float));
        fout.write(reinterpret_cast<char*>(&vertex.y), sizeof(float));
        fout.write(reinterpret_cast<char*>(&vertex.z), sizeof(float));
    }
}

void PLYWriter::writeFaces(std::ofstream &fout, const Mesh &mesh)
{
    for (int i = 0; i < mesh.faces.size(); ++i) {
        unsigned char n = 3;
        glm::ivec3 indices(mesh.faces[i]);
        int i0 = indices[0];
        int i1 = indices[1];
        int i2 = indices[2];
        fout.write(reinterpret_cast<char*>(&n), sizeof(unsigned char));
        fout.write(reinterpret_cast<char*>(&i0), sizeof(int));
        fout.write(reinterpret_cast<char*>(&i1), sizeof(int));
        fout.write(reinterpret_cast<char*>(&i2), sizeof(int));
    }
}