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);
      }
    }
  }
}

}