MeasurementFrameRectangleTaskNoDetect.cpp

Go to the documentation of this file.

 
#include <iostream>
 
#include <SEImplementation/Plugin/MeasurementFrameCoordinates/MeasurementFrameCoordinates.h>
#include <SEImplementation/Plugin/MeasurementFrameInfo/MeasurementFrameInfo.h>
#include <SEImplementation/Plugin/PixelBoundaries/PixelBoundaries.h>
#include "SEImplementation/Plugin/DetectionFrameCoordinates/DetectionFrameCoordinates.h"
#include <SEImplementation/Plugin/WorldCentroid/WorldCentroid.h>
#include <SEImplementation/Plugin/AssocMode/AssocMode.h>
 
#include <SEImplementation/Plugin/MeasurementFrameRectangle/MeasurementFrameRectangle.h>
#include <SEImplementation/Plugin/MeasurementFrameRectangle/MeasurementFrameRectangleTaskNoDetect.h>
 
namespace SourceXtractor {
 
void MeasurementFrameRectangleTaskNoDetect::computeProperties(SourceInterface& source) const {
  auto measurement_frame_coordinates = source.getProperty<MeasurementFrameCoordinates>(m_instance).getCoordinateSystem();
  const auto& measurement_frame_info = source.getProperty<MeasurementFrameInfo>(m_instance);
  const auto& world_centroid = source.getProperty<WorldCentroid>();
  const auto& assoc_mode = source.getProperty<AssocMode>();
 
  auto coord = world_centroid.getCentroid();
 
  bool bad_coordinates = false;
  ImageCoordinate coord1, coord2, coord3, coord4;
  try {
    int w = assoc_mode.getRefFramePixelWidth();
    int h = assoc_mode.getRefFramePixelHeight();
 
    auto c = measurement_frame_coordinates->worldToImage(coord);
    coord1 = ImageCoordinate(c.m_x - w, c.m_y - h);
    coord2 = ImageCoordinate(c.m_x + w, c.m_y - h);
    coord3 = ImageCoordinate(c.m_x - w, c.m_y + h);
    coord4 = ImageCoordinate(c.m_x + w, c.m_y + h);
  }
  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
  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()) {
    source.setIndexedProperty<MeasurementFrameRectangle>(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);
 
    source.setIndexedProperty<MeasurementFrameRectangle>(m_instance, min_coord, max_coord);
  }
}
 
} // SEImplementation namespace