PLYWriter.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <fstream>
 
#include <opencv2/core.hpp>
 
#include "vc/core/filesystem.hpp"
#include "vc/core/types/ITKMesh.hpp"
#include "vc/core/types/UVMap.hpp"
 
namespace volcart::io
{
class PLYWriter
{
public:
    PLYWriter() = default;
 
    PLYWriter(filesystem::path outputPath, ITKMesh::Pointer mesh);
 
    PLYWriter(
        filesystem::path outputPath, ITKMesh::Pointer mesh, cv::Mat texture);
    void setPath(const filesystem::path& path);
 
    void setMesh(ITKMesh::Pointer mesh);
 
    void setUVMap(UVMap::Pointer uvMap);
 
    void setTexture(cv::Mat texture);
 
    void setVertexColors(const std::vector<std::uint16_t>& c);
    void write();
private:
    filesystem::path outputPath_;
    std::ofstream outputMesh_;
    ITKMesh::Pointer mesh_;
    UVMap::Pointer uvMap_;
    cv::Mat texture_;
    std::vector<std::uint16_t> vcolors_;
 
    void write_header_();
 
    void write_vertices_();
    void write_faces_();
};
}  // namespace volcart::io