ScaleMarkerGenerator.hpp

Go to the documentation of this file.

#pragma once
 
#include <opencv2/core.hpp>
 
#include "vc/core/types/Color.hpp"
 
namespace volcart::texturing
{
class ScaleMarkerGenerator
{
public:
    enum class Type { Metric = 0, Imperial, ReferenceImage };
 
    ScaleMarkerGenerator() = default;
 
    void setScaleType(Type t) { type_ = t; }
 
    void setInputImage(const cv::Mat& i) { inputImg_ = i; }
 
    void setInputImagePixelSize(double s) { inputImgPixSize_ = s; }
 
    void setScaleBarColor(const Color& c) { scaleBarColor_ = c; }
 
    void setReferenceImage(const cv::Mat& i) { refImg_ = i; }
 
    void setReferenceImagePixelSize(double s) { refImgPixSize_ = s; }
 
    cv::Mat compute();
 
    cv::Mat getOutputImg() { return outputImg_; }
 
    cv::Mat getOutputScaleImage() { return scaleImg_; }
 
private:
    Type type_{Type::Metric};
 
    cv::Mat inputImg_;
    double inputImgPixSize_{1.0};
 
    cv::Mat scaleImg_;
    cv::Mat outputImg_;
 
    Color scaleBarColor_{volcart::color::WHITE};
 
    cv::Mat refImg_;
    double refImgPixSize_{1.0};
 
    double scaleFactor_{1.0};
    std::string scaleUnit_{"cm"};
 
    cv::Mat resize_ref_image_();
 
    cv::Mat generate_scale_bar_();
 
    void draw_tick_label_(
        cv::Mat img, const std::string& value, const cv::Point& position);
};
}  // namespace volcart::texturing