DetectionIdCheckImage.cpp

Go to the documentation of this file.

/*
 * DetectionIdCheckImage.cpp
 *
 *  Created on: Jun 25, 2018
 *      Author: mschefer
 */

#include "SEImplementation/Plugin/DetectionFrameInfo/DetectionFrameInfo.h"
#include "SEImplementation/Property/PixelCoordinateList.h"
#include "SEImplementation/CheckImages/CheckImages.h"
#include "SEImplementation/Property/SourceId.h"

#include "SEImplementation/CheckImages/DetectionIdCheckImage.h"

namespace SourceXtractor {

void DetectionIdCheckImage::handleMessage(const SourceInterface& source) {
  auto hdu_index = source.getProperty<DetectionFrameInfo>().getHduIndex();
  auto check_image = CheckImages::getInstance().getSegmentationImage(hdu_index);
  if (check_image != nullptr) {
    const auto& coordinates = source.getProperty<PixelCoordinateList>();

    // get the ID for each detected source
    const auto& source_id = source.getProperty<SourceId>().getDetectionId();

    // iterate over the pixels and set the detection_id value
    for (auto& coord : coordinates.getCoordinateList()) {
      check_image->setValue(coord.m_x, coord.m_y, source_id);
    }
  }
}

}