PixelCentroidTask.cpp

Go to the documentation of this file.

 
#include "SEImplementation/Property/PixelCoordinateList.h"
#include "SEImplementation/Plugin/DetectionFramePixelValues/DetectionFramePixelValues.h"
#include "SEImplementation/Plugin/PixelCentroid/PixelCentroid.h"
#include "SEImplementation/Plugin/PixelBoundaries/PixelBoundaries.h"
 
#include "SEImplementation/Plugin/PixelCentroid/PixelCentroidTask.h"
 
namespace SourceXtractor {
 
void PixelCentroidTask::computeProperties(SourceInterface& source) const {
  const auto& pixel_values = source.getProperty<DetectionFramePixelValues>().getFilteredValues();
  const auto& min_coord = source.getProperty<PixelBoundaries>().getMin();
 
  double centroid_x = 0.0;
  double centroid_y = 0.0;
  double total_value = 0.0;
 
  auto i = pixel_values.begin();
  for (auto pixel_coord : source.getProperty<PixelCoordinateList>().getCoordinateList()) {
    pixel_coord -= min_coord;
    SeFloat value = *i++;
 
    total_value += value;
    centroid_x += pixel_coord.m_x * value;
    centroid_y += pixel_coord.m_y * value;
  }
 
  centroid_x /= total_value;
  centroid_y /= total_value;
 
  source.setProperty<PixelCentroid>(centroid_x + min_coord.m_x, centroid_y + min_coord.m_y);
}
 
}