SegmentationFactory.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_SEGMENTATIONFACTORY_H
#define _SEIMPLEMENTATION_SEGMENTATIONFACTORY_H
 
 
#include "SEFramework/Task/TaskProvider.h"
#include "SEFramework/Configuration/Configurable.h"
#include "SEFramework/Pipeline/Segmentation.h"
 
#include "SEImplementation/Configuration/SegmentationConfig.h"
#include "SEImplementation/Plugin/AssocMode/AssocModeConfig.h"
 
 
namespace SourceXtractor {

class SegmentationFactory : public Configurable {
 
public:

  virtual ~SegmentationFactory() = default;

  explicit SegmentationFactory(std::shared_ptr<TaskProvider> task_provider);
 
  std::shared_ptr<Segmentation> createSegmentation() const;
  
  // Implementation of the Configurable interface
  void configure(Euclid::Configuration::ConfigManager& manager) override;
  void reportConfigDependencies(Euclid::Configuration::ConfigManager& manager) const override;
 
private:
  std::shared_ptr<DetectionImageFrame::ImageFilter> m_filter;
 
  SegmentationConfig::Algorithm m_algorithm;
  std::shared_ptr<TaskProvider> m_task_provider;
 
  int m_lutz_window_size;
  int m_bfs_max_delta;
 
  std::string m_model_path;
  double m_ml_threshold;
 
  std::vector<std::vector<AssocModeConfig::CatalogEntry>> m_catalogs;
 
}; /* End of SegmentationFactory class */
 
} /* namespace SourceXtractor */
 
 
#endif