diff options
author | Arnur Nigmetov <nigmetov@tugraz.at> | 2019-09-16 17:26:42 +0200 |
---|---|---|
committer | Arnur Nigmetov <nigmetov@tugraz.at> | 2019-09-16 17:26:42 +0200 |
commit | 7f6b96187423ba374b697f6f411c4a70d24ee297 (patch) | |
tree | 215ae5e4e422a3ce8b0b2055b794581886681b0b /geom_matching/wasserstein/include/wasserstein.h | |
parent | 498e457b221e9e70131ad73d9b881285d8d9572e (diff) |
Add traits for float in Wasserstein.
Diffstat (limited to 'geom_matching/wasserstein/include/wasserstein.h')
-rw-r--r-- | geom_matching/wasserstein/include/wasserstein.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/geom_matching/wasserstein/include/wasserstein.h b/geom_matching/wasserstein/include/wasserstein.h index 35d0bf6..17bb211 100644 --- a/geom_matching/wasserstein/include/wasserstein.h +++ b/geom_matching/wasserstein/include/wasserstein.h @@ -67,6 +67,17 @@ struct DiagramTraits<PairContainer_, std::pair<double, double>> static RealType get_y(const PointType& p) { return p.second; } }; +template<class PairContainer_> +struct DiagramTraits<PairContainer_, std::pair<float, float>> +{ + using PointType = std::pair<float, float>; + using RealType = float; + using Container = std::vector<PointType>; + + static RealType get_x(const PointType& p) { return p.first; } + static RealType get_y(const PointType& p) { return p.second; } +}; + namespace ws { |