SegmentationConfig.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_SEGMENTATIONCONFIG_H
#define _SEIMPLEMENTATION_SEGMENTATIONCONFIG_H
 
#include <memory>
#include "Configuration/Configuration.h"
 
namespace SourceXtractor {
 
class SegmentationConfig : public Euclid::Configuration::Configuration {
public:
 
  enum class Algorithm {
    UNKNOWN,
    LUTZ,
    BFS,
    ML
  };
 
  virtual ~SegmentationConfig() = default;
 
  explicit SegmentationConfig(long manager_id);
 
  std::map<std::string, Configuration::OptionDescriptionList> getProgramOptions() override;
  void preInitialize(const UserValues& args) override;
 
  Algorithm getAlgorithmOption() const {
    return m_selected_algorithm;
  }
 
  int getLutzWindowSize() const {
    return m_lutz_window_size;
  }
 
  int getBfsMaxDelta() const {
    return m_bfs_max_delta;
  }
 
  bool isFilteringEnabled() const {
    return m_filter != nullptr;
  }
 
  std::shared_ptr<DetectionImageFrame::ImageFilter> getFilter() const {
    return m_filter;
  }
 
  std::string getOnnxModelPath() const {
    return m_onnx_model_path;
  }
 
  double getMLThreashold() const {
    return m_ml_threshold;
  }
 
 
private:
  std::shared_ptr<DetectionImageFrame::ImageFilter> getDefaultFilter() const;
  std::shared_ptr<DetectionImageFrame::ImageFilter> loadFilter(const std::string& filename) const;
  std::shared_ptr<DetectionImageFrame::ImageFilter> loadFITSFilter(const std::string& filename) const;
  std::shared_ptr<DetectionImageFrame::ImageFilter> loadASCIIFilter(const std::string& filename) const;
 
  Algorithm m_selected_algorithm;
  std::shared_ptr<DetectionImageFrame::ImageFilter> m_filter;
 
  int m_lutz_window_size;
  int m_bfs_max_delta;
  std::string m_onnx_model_path;
  double m_ml_threshold;
}; /* End of SegmentationConfig class */
 
} /* namespace SourceXtractor */
 
 
#endif