MeasurementFrameGroupRectangleTask.cpp

Go to the documentation of this file.

/*
 * MeasurementFrameGroupRectangleTask.cpp
 *
 *  Created on: Sep 5, 2018
 *      Author: Alejandro Alvarez Ayllon
 */
 
#include "SEImplementation/Plugin/MeasurementFrameCoordinates/MeasurementFrameCoordinates.h"
#include "SEImplementation/Plugin/MeasurementFrameInfo/MeasurementFrameInfo.h"
#include <SEImplementation/Plugin/DetectionFrameGroupStamp/DetectionFrameGroupStamp.h>
#include "SEImplementation/Plugin/DetectionFrameCoordinates/DetectionFrameCoordinates.h"
 
#include <SEImplementation/Plugin/MeasurementFrameGroupRectangle/MeasurementFrameGroupRectangle.h>
#include <SEImplementation/Plugin/MeasurementFrameGroupRectangle/MeasurementFrameGroupRectangleTask.h>
 
namespace SourceXtractor {
 
void MeasurementFrameGroupRectangleTask::computeProperties(SourceGroupInterface& group) const {
  auto& detection_group_stamp = group.getProperty<DetectionFrameGroupStamp>();
  auto detection_frame_coordinates = group.begin()->getProperty<DetectionFrameCoordinates>().getCoordinateSystem();
  auto measurement_frame_coordinates = group.begin()->getProperty<MeasurementFrameCoordinates>(m_instance).getCoordinateSystem();
 
  const auto& measurement_frame_info = group.begin()->getProperty<MeasurementFrameInfo>(m_instance);
 
  // Get the coordinates of the detection frame group stamp
  auto stamp_top_left = detection_group_stamp.getTopLeft();
  auto width = detection_group_stamp.getStamp().getWidth();
  auto height = detection_group_stamp.getStamp().getHeight();
 
  // Transform the 4 corner coordinates from detection image
  ImageCoordinate coord1, coord2, coord3, coord4;
  bool bad_coordinates = false;
 
  try {
    coord1 = measurement_frame_coordinates->worldToImage(
        detection_frame_coordinates->imageToWorld(ImageCoordinate(stamp_top_left.m_x, stamp_top_left.m_y)));
    coord2 = measurement_frame_coordinates->worldToImage(
        detection_frame_coordinates->imageToWorld(ImageCoordinate(stamp_top_left.m_x + width, stamp_top_left.m_y)));
    coord3 = measurement_frame_coordinates->worldToImage(
        detection_frame_coordinates->imageToWorld(ImageCoordinate(stamp_top_left.m_x + width, stamp_top_left.m_y + height)));
    coord4 = measurement_frame_coordinates->worldToImage(
        detection_frame_coordinates->imageToWorld(ImageCoordinate(stamp_top_left.m_x, stamp_top_left.m_y + height)));
  }
  catch (const InvalidCoordinatesException&) {
    bad_coordinates = true;
  }
 
  // Determine the min/max coordinates
  auto min_x = std::min(coord1.m_x, std::min(coord2.m_x, std::min(coord3.m_x, coord4.m_x)));
  auto min_y = std::min(coord1.m_y, std::min(coord2.m_y, std::min(coord3.m_y, coord4.m_y)));
  auto max_x = std::max(coord1.m_x, std::max(coord2.m_x, std::max(coord3.m_x, coord4.m_x)));
  auto max_y = std::max(coord1.m_y, std::max(coord2.m_y, std::max(coord3.m_y, coord4.m_y)));
 
  PixelCoordinate min_coord, max_coord;
  min_coord.m_x = int(min_x);
  min_coord.m_y = int(min_y);
  max_coord.m_x = int(max_x) + 1;
  max_coord.m_y = int(max_y) + 1;
 
  // The full boundaries may lie outside of the frame
  // Or maybe there was a projection issue, so skip the frame for that source
  if (bad_coordinates || max_coord.m_x < 0 || max_coord.m_y < 0 || min_coord.m_x >= measurement_frame_info.getWidth() ||
      min_coord.m_y >= measurement_frame_info.getHeight()) {
    group.setIndexedProperty<MeasurementFrameGroupRectangle>(m_instance, bad_coordinates);
  }
  // Clip the coordinates to fit the available image
  else {
    min_coord.m_x = std::max(0, min_coord.m_x);
    min_coord.m_y = std::max(0, min_coord.m_y);
    max_coord.m_x = std::min(measurement_frame_info.getWidth() - 1, max_coord.m_x);
    max_coord.m_y = std::min(measurement_frame_info.getHeight() - 1, max_coord.m_y);
 
    group.setIndexedProperty<MeasurementFrameGroupRectangle>(m_instance, min_coord, max_coord);
  }
}
 
} // SEImplementation namespace