CoreThresholdPartitionStep.cpp

Go to the documentation of this file.

 
#include "SEImplementation/Property/PixelCoordinateList.h"
#include "SEImplementation/Plugin/DetectionFrameImages/DetectionFrameImages.h"
 
#include "SEImplementation/Plugin/CoreThresholdPartition/CoreThresholdPartitionStep.h"
 
namespace SourceXtractor {
 
CoreThresholdPartitionStep::CoreThresholdPartitionStep(double snr_level, unsigned int min_pixel_count) :
    m_snr_level(snr_level), m_min_pixel_count(min_pixel_count) {
}
 
std::vector<std::unique_ptr<SourceInterface>> CoreThresholdPartitionStep::partition(std::unique_ptr<SourceInterface> source) const {
  long int n_snr_level(0);
 
  // get the SNR image
  const auto snr_image = source->getProperty<DetectionFrameImages>().getLockedImage(LayerSignalToNoiseMap);
 
  // go over all pixels
  for (auto pixel_coord : source->getProperty<PixelCoordinateList>().getCoordinateList())
    // enhance the counter if the SNR is above the level
    if (snr_image->getValue(pixel_coord.m_x, pixel_coord.m_y) >= m_snr_level)
      n_snr_level += 1;
 
  // check whether the pixel # is above the threshold
  std::vector<std::unique_ptr<SourceInterface>> sources;
  if (n_snr_level >= m_min_pixel_count) {
    sources.emplace_back(std::move(source));
  }
  return sources;
}
 
} // SEImplementation namespace