GroupIdCheckImage.cpp

Go to the documentation of this file.

/*
 * GroupIdCheckImage.cpp
 *
 *  Created on: 2019 M01 30
 *      Author: mschefer
 */

#include "SEFramework/Source/SourceGroupInterface.h"

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

#include "SEImplementation/CheckImages/GroupIdCheckImage.h"

namespace SourceXtractor {

void GroupIdCheckImage::handleMessage(const SourceGroupInterface& group) {
  auto hdu_index = group.cbegin()->getProperty<DetectionFrameInfo>().getHduIndex();
  auto check_image = CheckImages::getInstance().getGroupImage(hdu_index);
  if (check_image) {
    // get the ID of the group
    auto group_id = group.getProperty<GroupInfo>().getGroupId();

    for (auto& source : group) {
      auto& coordinates = source.getProperty<PixelCoordinateList>();

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

}