AssocSegmentation.h

Go to the documentation of this file.

/*
 * AssocSegmentation.h
 *
 *  Created on: Sep 6, 2022
 *      Author: mschefer
 */
 
#ifndef _SEIMPLEMENTATION_SEGMENTATION_ASSOCSEGMENTATION_H_
#define _SEIMPLEMENTATION_SEGMENTATION_ASSOCSEGMENTATION_H_
 
#include "SEUtils/PixelCoordinate.h"
 
#include "SEFramework/Source/SourceFactory.h"
#include "SEFramework/Pipeline/Segmentation.h"
 
#include "SEImplementation/Plugin/AssocMode/AssocModeConfig.h"
 
namespace SourceXtractor {
 

class AssocSegmentation : public Segmentation::Labelling {
public:
 
  virtual ~AssocSegmentation() = default;
 
  AssocSegmentation(std::shared_ptr<SourceFactory> source_factory, std::vector<AssocModeConfig::CatalogEntry> source_list)
      : m_source_factory(source_factory), m_source_list(source_list) {
    assert(source_factory != nullptr);
  }
 
  void labelImage(Segmentation::LabellingListener& listener, std::shared_ptr<const DetectionImageFrame> frame) override;
 
private:
  std::shared_ptr<SourceFactory> m_source_factory;
  std::vector<AssocModeConfig::CatalogEntry> m_source_list;
};
 
}
 
#endif /* _SEIMPLEMENTATION_SEGMENTATION_ASSOCSEGMENTATION_H_ */