AlignmentMarkerGenerator.hpp

Go to the documentation of this file.

#pragma once
 
#include <vector>
 
#include "vc/core/types/TexturedMesh.hpp"
 
namespace volcart::texturing
{
class AlignmentMarkerGenerator
{
public:
    struct LineSegment {
        LineSegment() = default;
        LineSegment(const cv::Vec3d& a, const cv::Vec3d& b, cv::Scalar c = 0);
 
        cv::Vec3d a;
        cv::Vec3d b;
        cv::Scalar color;
    };
 
    void setInputMeshes(std::vector<TexturedMesh> m);
 
    void setLineSegments(std::vector<LineSegment> r);
 
    void setMarkerRadius(int r);
 
    void setMarkerUseRandomColor(bool b);
 
    std::vector<cv::Mat> compute();
 
    std::vector<cv::Mat> getMarkedImages() const;
 
private:
    std::vector<volcart::TexturedMesh> input_;
    std::vector<LineSegment> lineSegments_;
    int markerRadius_{5};
    bool markerRandomColor_{true};
    std::vector<cv::Mat> output_;
};
 
}  // namespace volcart::texturing