SourceIdCheckImage.cpp

Go to the documentation of this file.

/*
 * SegmentationCheckImage.h
 *
 *  Created on: Dec 11, 2017
 *      Author: mschefer
 */
 
#include "SEImplementation/Plugin/DetectionFrameInfo/DetectionFrameInfo.h"
#include "SEImplementation/Property/PixelCoordinateList.h"
#include "SEImplementation/CheckImages/CheckImages.h"
#include "SEImplementation/Plugin/SourceIDs/SourceID.h"
 
#include "SEImplementation/CheckImages/SourceIdCheckImage.h"
 
namespace SourceXtractor {
 
void SourceIdCheckImage::handleMessage(const SourceGroupInterface& group) {
  auto hdu_index = group.cbegin()->getProperty<DetectionFrameInfo>().getHduIndex();
  auto check_image = CheckImages::getInstance().getPartitionImage(hdu_index);
  if (check_image != nullptr) {
    for (auto& source : group) {
      const auto& coordinates = source.getProperty<PixelCoordinateList>();
 
      // get the ID for each (multithresholded) source
      const auto& source_id = source.getProperty<SourceID>().getId();
 
      // iterate over the pixels and set the source-id value
      for (auto& coord : coordinates.getCoordinateList()) {
        check_image->setValue(coord.m_x, coord.m_y, source_id);
      }
    }
  }
}
 
}