OBJReader.hpp

Go to the documentation of this file.

#pragma once
 
#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 OBJReader
{
public:
    void setPath(const filesystem::path& p);
 
    auto read() -> ITKMesh::Pointer;
 
    auto getMesh() -> ITKMesh::Pointer;
 
    auto getUVMap() -> UVMap::Pointer;
 
    auto getTextureMat() -> cv::Mat;
 
private:
    using VertexRefs = cv::Vec3i;
    using Face = std::vector<VertexRefs>;
 
    enum class RefType {
        Invalid,
        Vertex,
        VertexWithTexture,
        VertexWithNormal,
        VertexWithTextureAndNormal
    };
 
    void reset_();
 
    void parse_();
    void parse_vertex_(const std::vector<std::string>& strs);
    void parse_normal_(const std::vector<std::string>& strs);
    void parse_tcoord_(const std::vector<std::string>& strs);
    void parse_face_(const std::vector<std::string>& strs);
    void parse_mtllib_(const std::vector<std::string>& strs);
    auto classify_vertref_(const std::string& ref) -> RefType;
 
    void build_mesh_();
 
    filesystem::path path_;
    filesystem::path texturePath_;
    ITKMesh::Pointer mesh_;
    UVMap::Pointer uvMap_;
    cv::Mat textureMat_;
 
    std::vector<cv::Vec3d> vertices_;
    std::vector<cv::Vec3d> normals_;
    std::vector<cv::Vec2d> uvs_;
    std::vector<OBJReader::Face> faces_;
};
 
}  // namespace volcart::io