array.hpp

#ifndef OPENEV_CONTAINERS_ARRAY_HPP
#define OPENEV_CONTAINERS_ARRAY_HPP
 
#include "openev/core/types.hpp"
#include <array>
#include <cstddef>
#include <numeric>
 
namespace ev {
template <typename T, std::size_t N>
class Array_ : public std::array<Event_<T>, N> {
  using std::array<Event_<T>, N>::array;
 
public:
  [[nodiscard]] inline double duration() const {
    return std::array<Event_<T>, N>::back().t - std::array<Event_<T>, N>::front().t;
  }

  [[nodiscard]] inline double rate() const {
    return std::array<Event_<T>, N>::size() / duration();
  }

  [[nodiscard]] Eventd mean() const {
    const double x = std::accumulate(std::array<Event_<T>, N>::begin(), std::array<Event_<T>, N>::end(), 0.0, [](double sum, const Event_<T> &e) { return sum + e.x; }) / N;
    const double y = std::accumulate(std::array<Event_<T>, N>::begin(), std::array<Event_<T>, N>::end(), 0.0, [](double sum, const Event_<T> &e) { return sum + e.y; }) / N;
    const double t = std::accumulate(std::array<Event_<T>, N>::begin(), std::array<Event_<T>, N>::end(), 0.0, [](double sum, const Event_<T> &e) { return sum + e.t; }) / N;
    const double p = std::accumulate(std::array<Event_<T>, N>::begin(), std::array<Event_<T>, N>::end(), 0.0, [](double sum, const Event_<T> &e) { return sum + e.p; }) / N;
    return {x, y, t, p > 0.5};
  }

  [[nodiscard]] inline cv::Point2d meanPoint() const {
    const double x = std::accumulate(std::array<Event_<T>, N>::begin(), std::array<Event_<T>, N>::end(), 0.0, [](double sum, const Event_<T> &e) { return sum + e.x; }) / N;
    const double y = std::accumulate(std::array<Event_<T>, N>::begin(), std::array<Event_<T>, N>::end(), 0.0, [](double sum, const Event_<T> &e) { return sum + e.y; }) / N;
    return {x, y};
  }

  [[nodiscard]] inline double meanTime() const {
    return std::accumulate(std::array<Event_<T>, N>::begin(), std::array<Event_<T>, N>::end(), 0.0, [](double sum, const Event_<T> &e) { return sum + e.t; }) / N;
  }

  [[nodiscard]] inline double midTime() const {
    return 0.5 * (std::array<Event_<T>, N>::front().t + std::array<Event_<T>, N>::back().t);
  }
};
template <std::size_t N>
using Arrayi = Array_<int, N>; 
template <std::size_t N>
using Arrayl = Array_<long, N>; 
template <std::size_t N>
using Arrayf = Array_<float, N>; 
template <std::size_t N>
using Arrayd = Array_<double, N>; 
template <std::size_t N>
using Array = Arrayi<N>; 
} // namespace ev
 
#endif // OPENEV_CONTAINERS_ARRAY_HPP