OBJWriter.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <fstream>
#include <iostream>
 
#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 OBJWriter
{
 
public:
    OBJWriter() = default;
 
    OBJWriter(filesystem::path outputPath, ITKMesh::Pointer mesh);
 
    OBJWriter(
        filesystem::path outputPath,
        ITKMesh::Pointer mesh,
        UVMap::Pointer uvMap,
        cv::Mat uvImg);
    void setPath(const filesystem::path& path);
 
    void setMesh(ITKMesh::Pointer mesh);
 
    void setUVMap(UVMap::Pointer uvMap);
 
    void setTexture(cv::Mat uvImg);
 
    void setTextureFormat(std::string fmt);
    void write();
private:
    filesystem::path outputPath_;
    std::string textureFmt_{"tif"};
    std::ofstream outputMesh_;
    std::ofstream outputMTL_;
 
    std::map<std::uint32_t, cv::Vec3i> pointLinks_;
 
    ITKMesh::Pointer mesh_;
    UVMap::Pointer uvMap_;
    cv::Mat texture_;
 
    void write_obj_();
    void write_mtl_();
    void write_texture_();
 
    void write_header_();
    void write_vertices_();
    void write_texture_coordinates_();
    void write_faces_();
};
 
}  // namespace volcart::io