AttractorsPartitionStep.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_PARTITION_ATTRACTORSPARTITIONSTEP_H
#define _SEIMPLEMENTATION_PARTITION_ATTRACTORSPARTITIONSTEP_H
 
#include <unordered_map>
#include <vector>
 
#include "SEUtils/PixelCoordinate.h"
#include "SEFramework/Image/Image.h"
#include "SEFramework/Source/SourceFactory.h"
#include "SEFramework/Source/SourceWithOnDemandProperties.h"
#include "SEFramework/Pipeline/Partition.h"
 
namespace SourceXtractor {

class AttractorsPartitionStep : public PartitionStep {
public:

  virtual ~AttractorsPartitionStep() = default;
 
  explicit AttractorsPartitionStep(std::shared_ptr<SourceFactory> source_factory) :
        m_source_factory(source_factory) {}
 
  std::vector<std::unique_ptr<SourceInterface>> partition(std::unique_ptr<SourceInterface> source) const override;
 
private:
  std::shared_ptr<SourceFactory> m_source_factory;
 
  void attractPixels(
      const std::vector<std::pair<PixelCoordinate, PixelCoordinate>>& pixels_with_origin,
      std::unordered_map<PixelCoordinate, std::vector<PixelCoordinate>>& attractors,
      std::function<DetectionImage::PixelType (PixelCoordinate)> value_function) const;
 
  std::vector<std::vector<PixelCoordinate>> mergeAttractors(
      const std::unordered_map<PixelCoordinate, std::vector<PixelCoordinate>>& attractors) const;
 
 
}; /* End of AttractorsPartitionStep class */
 
} /* namespace SourceXtractor */
 
 
#endif