UVMap.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <map>
#include <memory>
 
#include <opencv2/core.hpp>
#include <opencv2/imgproc.hpp>
 
#include "vc/core/types/Color.hpp"
#include "vc/core/types/ITKMesh.hpp"
 
namespace volcart
{
const static cv::Vec2d NULL_MAPPING{-1, -1};
 
class UVMap
{
public:
    using Pointer = std::shared_ptr<UVMap>;
 
    enum class Origin { TopLeft = 0, TopRight, BottomLeft, BottomRight };
 
    struct Ratio {
        double width{1}, height{1}, aspect{1};
    };
 
    UVMap() = default;
 
    explicit UVMap(Origin o);
 
    template <typename... Args>
    static auto New(Args... args) -> Pointer
    {
        return std::make_shared<UVMap>(std::forward<Args>(args)...);
    }
    [[nodiscard]] auto size() const -> std::size_t;
 
    [[nodiscard]] auto empty() const -> bool;
    void setOrigin(const Origin& o);
 
    [[nodiscard]] auto origin() const -> Origin;
    void set(std::size_t id, const cv::Vec2d& uv, const Origin& o);
 
    void set(std::size_t id, const cv::Vec2d& uv);
 
    [[nodiscard]] auto get(std::size_t id, const Origin& o) const -> cv::Vec2d;
 
    [[nodiscard]] auto get(std::size_t id) const -> cv::Vec2d;
 
    [[nodiscard]] auto contains(std::size_t id) const -> bool;
 
    [[nodiscard]] auto as_map() const -> std::map<std::size_t, cv::Vec2d>;
    [[nodiscard]] auto ratio() const -> Ratio;
 
    void ratio(double a);
 
    void ratio(double w, double h);
    enum class Rotation { CW90 = 0, CW180, CCW90 };
 
    enum class FlipAxis { Vertical = 0, Horizontal, Both };
 
    enum class AlignmentAxis { None = 0, ZPos, ZNeg, YPos, YNeg, XPos, XNeg };
 
    static auto Plot(const UVMap& uv, const Color& color = color::GREEN)
        -> cv::Mat;
 
    static auto Plot(
        const UVMap& uv,
        const ITKMesh::Pointer& mesh2D,
        int width = -1,
        int height = -1,
        const Color& color = color::LIGHT_GRAY) -> cv::Mat;
 
    static void AlignToAxis(
        UVMap& uv, const ITKMesh::Pointer& mesh, AlignmentAxis axis);
 
    static void Rotate(UVMap& uv, Rotation rotation);
 
    static void Rotate(UVMap& uv, Rotation rotation, cv::Mat& texture);
 
    static void Rotate(
        UVMap& uv, double theta, const cv::Vec2d& center = {0.5, 0.5});
 
    static void Rotate(
        UVMap& uv,
        double theta,
        cv::Mat& texture,
        const cv::Vec2d& center = {0.5, 0.5});
 
    static void Flip(UVMap& uv, FlipAxis axis);
private:
    std::map<std::size_t, cv::Vec2d> map_;
    Origin origin_{Origin::TopLeft};
    Ratio ratio_;
};
}  // namespace volcart