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 std::shared_ptr<SourceInterface>& source) {
  auto hdu_index = source->getProperty<DetectionFrameInfo>().getHduIndex();
  auto check_image = CheckImages::getInstance().getSegmentationImage(hdu_index);
  if (check_image != nullptr) {
    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);
    }
  }
}
 
}