MultiThresholdPartitionStep.h

Go to the documentation of this file.

/*
 * MultiThresholdPartitionStep.h
 *
 *  Created on: Jan 17, 2017
 *      Author: mschefer
 */
 
#ifndef _SEIMPLEMENTATION_PARTITION_MULTITHRESHOLDPARTITIONSTEP_H_
#define _SEIMPLEMENTATION_PARTITION_MULTITHRESHOLDPARTITIONSTEP_H_
 
#include <boost/random.hpp>
 
#include "SEUtils/Types.h"
 
#include "SEImplementation/Property/PixelCoordinateList.h"
#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/VectorImage.h"
 
#include "SEFramework/Pipeline/Partition.h"
#include "SEFramework/Source/SourceFactory.h"
 
 
namespace SourceXtractor {
 
class MultiThresholdNode;

class MultiThresholdPartitionStep : public PartitionStep {
 
public:
 
  MultiThresholdPartitionStep(std::shared_ptr<SourceFactory> source_factory, SeFloat contrast,
      unsigned int thresholds_nb, unsigned int min_deblend_area, unsigned int seed);
 
  virtual ~MultiThresholdPartitionStep() = default;
 
  virtual std::vector<std::unique_ptr<SourceInterface>> partition(std::unique_ptr<SourceInterface> source) const;
 
private:
  std::vector<std::unique_ptr<SourceInterface>> reassignPixels(
      const std::vector<std::unique_ptr<SourceInterface>>& sources,
      const std::vector<PixelCoordinate>& pixel_coords,
      std::shared_ptr<VectorImage<DetectionImage::PixelType>> image,
      const std::vector<std::shared_ptr<MultiThresholdNode>>& source_nodes,
      const PixelCoordinate& offset
      ) const;
 
  std::shared_ptr<SourceFactory> m_source_factory;
  SeFloat m_contrast;
  unsigned int m_thresholds_nb;
  unsigned int m_min_deblend_area;
  unsigned int m_seed;
  boost::random::mt19937 m_rng;
};
 
 
}
 
 
 
 
#endif /* _SEIMPLEMENTATION_PARTITION_MULTITHRESHOLDPARTITIONSTEP_H_ */