Program Listing for File dumper.cpp

↰ Return to documentation for file (pymembrane/cppmodule/src/dumper/dumper.cpp)

#include "dumper.hpp"

#include <memory>
#include "../mesh/computegeometry.hpp"
#include "../mesh/halfedges.hpp"
#include "../types/globaltypes.hpp"
#include "../types/pymembvector.hpp"
#include "../system/systemclass.hpp"
#include "../utils/tofile.hpp"
#include "../utils/fromstring.hpp"

void DumperClass::mesh_json(const std::string &file_name, const bool &ffaces, const bool &fvertices)
{
}

void DumperClass::mesh_txt(const std::string &file_name, const bool &ffaces, const bool &fvertices)
{
    if (ffaces)
    {
        std::ofstream file(file_name + "_faces.dat", std::ios::out);
        if (file.good())
        {
            for (const auto& face : _system.faces)
            {
                util::writeToFile(file, face.id, face.v1, face.v2, face.v3, 1, face.type);
                file << "\n";
            }
        }
        else
        {
            py::print("Error no faces file is provided\n");
        }
    }

    if (fvertices)
    {
        std::ofstream file(file_name + "_vertices.dat", std::ios::out);
        if (file.good())
        {
            for (const auto& vertex : _system.vertices)
            {
                util::writeToFile(file,
                                  vertex.id,
                                  util::to_string_with_precision(vertex.r.x, 12),
                                  util::to_string_with_precision(vertex.r.y, 12),
                                  util::to_string_with_precision(vertex.r.z, 12),
                                  vertex.type);
                file << "\n";
            }
        }
        else
        {
            py::print("Error no vertices file is provided\n");
        }
    }
}


void DumperClass::mesh_inp(const std::string &file_name)
{

    auto vertices = _system.get_vertices();
    auto faces = _system.get_faces();

    int Numvertices = vertices.size();
    int Numfaces = faces.size();
    std::ofstream vFile(file_name + "_vertices.inp", std::ios::out);
    if(vFile.is_open())
    {
        for(int i=0; i<Numvertices; i++) {
            vFile << vertices[i].id << " " << vertices[i].r.x << " "
            << vertices[i].r.y << " " << vertices[i].r.z << " "
            << vertices[i].type << std::endl;
        }
    }

    std::ofstream fFile(file_name + "_faces.inp", std::ios::out);
    if(fFile.is_open())
    {
        for(int i=0; i<Numfaces; i++) {
            fFile << faces[i].id << " " << faces[i].v1 << " "
            << faces[i].v2 << " " << faces[i].v3 << " "
            << 1 << " " << faces[i].type << std::endl;
        }
    }
}