LutzSegmentation.cpp

Go to the documentation of this file.

 
 
#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/ProcessedImage.h"
#include "SEFramework/Source/SourceWithOnDemandProperties.h"
 
#include "SEImplementation/Measurement/MultithreadedMeasurement.h"
#include "SEImplementation/Property/PixelCoordinateList.h"
#include "SEImplementation/Property/SourceId.h"
#include "SEImplementation/Grouping/LineSelectionCriteria.h"
#include "SEImplementation/Segmentation/Lutz.h"
 
#include "SEImplementation/Segmentation/LutzSegmentation.h"
 
namespace SourceXtractor {
 
namespace {
 
class LutzLabellingListener : public Lutz::LutzListener {
public:
  LutzLabellingListener(Segmentation::LabellingListener& listener, std::shared_ptr<SourceFactory> source_factory,
      int window_size) :
    m_listener(listener),
    m_source_factory(source_factory),
    m_window_size(window_size) {}
 
  virtual ~LutzLabellingListener() = default;
 
  void publishGroup(Lutz::PixelGroup& pixel_group) override {
    auto source = m_source_factory->createSource();
    source->setProperty<PixelCoordinateList>(pixel_group.pixel_list);
    source->setProperty<SourceId>();
    m_listener.publishSource(std::move(source));
  }
 
  void notifyProgress(int line, int total) override {
    m_listener.notifyProgress(line, total);
 
    if (m_window_size > 0 && line > m_window_size) {
      m_listener.requestProcessing(
        ProcessSourcesEvent(std::make_shared<LineSelectionCriteria>(line - m_window_size))
      );
    }
  }
 
private:
  Segmentation::LabellingListener& m_listener;
  std::shared_ptr<SourceFactory> m_source_factory;
  int m_window_size;
};
 
}
 
//
// class LutzSegmentation
//
 
void LutzSegmentation::labelImage(Segmentation::LabellingListener& listener, std::shared_ptr<const DetectionImageFrame> frame) {
  Lutz lutz;
  LutzLabellingListener lutz_listener(listener, m_source_factory, m_window_size);
  lutz.labelImage(lutz_listener, *frame->getThresholdedImage());
}
 
} // Segmentation namespace