Mercurial Hosting > traffic-intelligence
diff trajectorymanagement/src/CanberraMetric.h @ 1159:e1e7acef8eab
moved trajectory management library into Traffic Intelligence
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Mon, 22 Feb 2021 22:09:35 -0500 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/trajectorymanagement/src/CanberraMetric.h Mon Feb 22 22:09:35 2021 -0500 @@ -0,0 +1,86 @@ +#ifndef CANBERRAMETRIC_H_ +#define CANBERRAMETRIC_H_ + +#include "Metric.h" + +#include "TrajectoryExceptions.h" + +/** + * CanberraMetric class. + * + * The Canberra metric measures the similarity between two trajectories. + */ +template<typename Tr> +class CanberraMetric: public Metric<Tr, double> +{ +public: + /** + * Compute distance between two trajectories. + * + * @param[in] a input trajectory + * @param[in] b input trajectory + * @param[out] result distance between two trajectories + */ + void distance(const Trajectory<Tr> *a, const Trajectory<Tr> *b, double &result, unsigned int nbOfPoints = std::numeric_limits<unsigned int>::max()) + { + result = double(0); + + if (nbOfPoints == std::numeric_limits<unsigned int>::max()) + checkTrajectoryLength(a->size(), b->size()); + + double diff = double(0); + for (unsigned int i = 0; i < a->size() and i < nbOfPoints; ++i) + { + cv::Point3_<typeof(static_cast<Tr>(a->getPoint(i)).x)> A(a->getPoint(i)); + cv::Point3_<typeof(static_cast<Tr>(a->getPoint(i)).x)> B(b->getPoint(i)); + diff += calculate(A.x, B.x); + diff += calculate(A.y, B.y); + diff += calculate(A.z, B.z); + } + + result = diff; + } + +private: + /** + * Check the length of the trajectories. + * + * @param[in] a the size of the trajectory + * @param[in] b the size of the trajectory + */ + void checkTrajectoryLength(size_t a, size_t b) const + { + if (a != b || a == 0) + { + throw TrajectoryLengthErrorException(); + } + } + + /** + * Compute similarity between two values. + * + * @param a first value + * @param b second value + * @return result + */ + double calculate(double a, double b) + { + double result = std::abs(a - b); + + a = std::fabs(a); + b = std::fabs(b); + + a += b; + + if (a == 0) + { + return 0; + } + + result /= a; + + return result; + } +}; + +#endif /* CANBERRAMETRIC_H_ */