CoreThresholdPartitionTask.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_PLUGIN_CORETHRESHOLDTASK_H_
#define _SEIMPLEMENTATION_PLUGIN_CORETHRESHOLDTASK_H_
 
#include "NCorePixel.h"
#include "SEImplementation/Property/PixelCoordinateList.h"
 
#include "SEFramework/Task/SourceTask.h"
#include "SEImplementation/Plugin/DetectionFramePixelValues/DetectionFramePixelValues.h"
#include "SEImplementation/Plugin/DetectionFrameImages/DetectionFrameImages.h"
 
#include "SEImplementation/Measurement/MultithreadedMeasurement.h"
 
namespace SourceXtractor {
 
class CoreThresholdPartitionTask : public SourceTask {
 
public:
  explicit CoreThresholdPartitionTask(double snr_level): m_snr_level(snr_level) {};
 
  virtual ~CoreThresholdPartitionTask() = default;
 
  virtual void computeProperties(SourceInterface& source) const {
    long int n_snr_level=0;
 
    // get the detection frame and the SNR image
    const auto& detection_frame_images = source.getProperty<DetectionFrameImages>();
    const auto& snr_image = detection_frame_images.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;
    }
 
    // set the property
    source.setProperty<NCorePixel>(n_snr_level);
};
 
private:
  double m_snr_level;
 
}; // End of CoreThresholdTask class
} // namespace SExtractor
 
#endif /* _SEIMPLEMENTATION_PLUGIN_SNRLEVELSOURCETASK_H_ */