mask_seed_setup.h

Go to the documentation of this file.

 
#ifndef STEPSABC_MASK_SEED_SETUP_H
#define STEPSABC_MASK_SEED_SETUP_H
 
#include "parameters.h"
#include <opencv2/opencv.hpp>
#include <string>
#include <vector>
 
namespace cppxdic {
namespace stepsABC {
 
struct MaskSeedResult {
    cv::Mat mask;                   // uint8 ROI mask (0 outside, 255 inside)
    std::vector<cv::Point> polygon; // polygon vertices defining the mask (pixel coords)
    std::vector<SeedPoint> seeds;   // one or more seed points (pixel world coords)
    bool valid = false;             // true when mask + at least one seed are set
};
 
class MaskSeedSetup {
public:
    static MaskSeedResult runGui(const cv::Mat& reference_image, int num_seeds,
                                 std::string& error_message);
 
    static MaskSeedResult loadFromFiles(const std::string& polygon_or_mask_path,
                                        const std::string& seed_path, const cv::Size& image_size,
                                        std::string& error_message);
 
    static bool save(const MaskSeedResult& result, const std::string& out_prefix,
                     std::string& error_message);
 
    static cv::Mat polygonToMask(const std::vector<cv::Point>& polygon, const cv::Size& image_size);
 
    static std::vector<cv::Point> readPolygonFile(const std::string& path,
                                                  std::string& error_message);
 
    static std::vector<SeedPoint> readSeedFile(const std::string& path, std::string& error_message);
 
    static bool guiAvailable();
};
 
} // namespace stepsABC
} // namespace cppxdic
 
#endif // STEPSABC_MASK_SEED_SETUP_H