TetraMeshWriter.h

Go to the documentation of this file.

/*
This Module is designed to output surface mesh file of TetrahedronSet;
the output file format: obj
*/
 
#pragma once
#include "Module/OutputModule.h"
#include "Module/TopologyModule.h"
 
#include "Topology/TetrahedronSet.h"
 
#include <string>
#include <memory>
 
namespace dyno
{
 
    template <typename TDataType> class TriangleSet;
    template <typename TDataType> class TetrahedronSet;
 
    template<typename TDataType>
    class TetraMeshWriter : public OutputModule
    {
        DECLARE_TCLASS(TetraMeshWriter, TDataType)
 
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
        typedef typename TopologyModule::Triangle Triangle;
        typedef typename TopologyModule::Tetrahedron Tetrahedron;
        typedef typename TopologyModule::Tri2Tet Tri2Tet;
 
        TetraMeshWriter();
        ~TetraMeshWriter();
 
        void setNamePrefix(std::string prefix);
        void setOutputPath(std::string path);
 
        void setTetrahedronSetPtr(std::shared_ptr<TetrahedronSet<TDataType>> ptr_tets) { this->ptr_TetrahedronSet = ptr_tets;  this->updatePtr(); }
        bool updatePtr();
 
        bool outputSurfaceMesh();
        void output()override;
 
    protected:
        //void updateImpl() override;
 
    public:
 
 
    protected:
        int max_output_files = 150;
        int idle_frame_num = 0;       //output one file of [num]+1 frames
        int current_idle_frame = 0;
        std::string file_postfix = ".obj";
 
        DArray<Triangle>* ptr_triangles;
        DArray<Tri2Tet>* ptr_tri2tet;
        DArray<Coord>* ptr_vertices;
        DArray<Tetrahedron>* ptr_tets;
        std::shared_ptr<TetrahedronSet<TDataType>> ptr_TetrahedronSet;
        clock_t last, now;
    };
}