121 lines
4.0 KiB
C
121 lines
4.0 KiB
C
|
// Copyright (c) 2005 INRIA Sophia-Antipolis (France).
|
||
|
// All rights reserved.
|
||
|
//
|
||
|
// This file is part of CGAL (www.cgal.org)
|
||
|
//
|
||
|
// $URL: https://github.com/CGAL/cgal/blob/v5.1/Principal_component_analysis_LGPL/include/CGAL/barycenter.h $
|
||
|
// $Id: barycenter.h 0779373 2020-03-26T13:31:46+01:00 Sébastien Loriot
|
||
|
// SPDX-License-Identifier: LGPL-3.0-or-later OR LicenseRef-Commercial
|
||
|
//
|
||
|
//
|
||
|
// Author(s) : Sylvain Pion
|
||
|
|
||
|
#ifndef CGAL_BARYCENTER_H
|
||
|
#define CGAL_BARYCENTER_H
|
||
|
|
||
|
#include <CGAL/basic.h>
|
||
|
#include <CGAL/Origin.h>
|
||
|
#include <iterator>
|
||
|
#include <CGAL/Kernel_traits.h>
|
||
|
#include <CGAL/Kernel/Dimension_utils.h>
|
||
|
#include <CGAL/Dimension.h>
|
||
|
|
||
|
// Functions to compute the point given its barycentric coordinates.
|
||
|
// Works in 2D and 3D (and dD ?).
|
||
|
// Special case for the centroid.
|
||
|
|
||
|
// TODO : Note : more numerically stable variants could be implemented as well.
|
||
|
// TODO : Specify a traits class concept ? Maybe not for these computations.
|
||
|
// TODO : Grep for "barycenter" and "centroid" in CGAL to check existing usages.
|
||
|
// TODO : Add barycentric_coordinates() (to the kernel, this time).
|
||
|
|
||
|
namespace CGAL {
|
||
|
|
||
|
// This one takes an iterator range over std::pair<K::Point_[23], K::FT>
|
||
|
template < typename InputIterator, typename K >
|
||
|
typename std::iterator_traits<InputIterator>::value_type::first_type
|
||
|
barycenter(InputIterator begin, InputIterator end, const K & )
|
||
|
{
|
||
|
typedef typename std::iterator_traits<InputIterator>::value_type pair;
|
||
|
typedef typename pair::second_type FT;
|
||
|
typedef typename pair::first_type Point;
|
||
|
typedef typename Access::Vector<K, typename Ambient_dimension<Point, K>::type>::type Vector;
|
||
|
|
||
|
CGAL_precondition(begin != end);
|
||
|
|
||
|
Vector v = NULL_VECTOR;
|
||
|
FT norm = 0;
|
||
|
|
||
|
while (begin != end) {
|
||
|
pair p = *begin++;
|
||
|
v = v + p.second * (p.first - ORIGIN);
|
||
|
norm += p.second;
|
||
|
}
|
||
|
|
||
|
CGAL_assertion( norm != 0 );
|
||
|
|
||
|
return ORIGIN + v / norm;
|
||
|
}
|
||
|
|
||
|
// This one takes an iterator range over K::Point_[23],
|
||
|
// and an iterator over K::FT.
|
||
|
template < typename PointInputIterator, typename WeightInputIterator,
|
||
|
typename K >
|
||
|
typename std::iterator_traits<PointInputIterator>::value_type
|
||
|
barycenter(PointInputIterator begin, PointInputIterator end,
|
||
|
WeightInputIterator wbegin, const K & )
|
||
|
{
|
||
|
typedef typename std::iterator_traits<PointInputIterator>::value_type Point;
|
||
|
typedef typename std::iterator_traits<WeightInputIterator>::value_type FT;
|
||
|
typedef typename Access::Vector<K, typename Ambient_dimension<Point, K>::type>::type Vector;
|
||
|
|
||
|
CGAL_precondition(begin != end);
|
||
|
|
||
|
Vector v = NULL_VECTOR;
|
||
|
FT norm = 0;
|
||
|
|
||
|
while (begin != end) {
|
||
|
FT weight = *wbegin++;
|
||
|
v = v + weight * (*begin++ - ORIGIN);
|
||
|
norm += weight;
|
||
|
}
|
||
|
|
||
|
CGAL_assertion( norm != 0 );
|
||
|
|
||
|
return ORIGIN + v / norm;
|
||
|
}
|
||
|
|
||
|
// This one takes an iterator range over std::pair<K::Point_[23], K::FT>
|
||
|
// And it uses Kernel_traits<> to find out its kernel.
|
||
|
template < typename InputIterator >
|
||
|
inline
|
||
|
typename std::iterator_traits<InputIterator>::value_type::first_type
|
||
|
barycenter(InputIterator begin, InputIterator end)
|
||
|
{
|
||
|
typedef typename std::iterator_traits<InputIterator>::value_type pair;
|
||
|
typedef typename pair::first_type Point;
|
||
|
typedef typename Kernel_traits<Point>::Kernel K;
|
||
|
|
||
|
return CGAL::barycenter(begin, end, K());
|
||
|
}
|
||
|
|
||
|
// This one takes an iterator range over K::Point_[23],
|
||
|
// and an iterator over K::FT.
|
||
|
// And it uses Kernel_traits<> to find out its kernel.
|
||
|
// To differentiate it from the others, it takes an "int" as K parameter
|
||
|
template < typename PointInputIterator, typename WeightInputIterator >
|
||
|
inline
|
||
|
typename std::iterator_traits<PointInputIterator>::value_type
|
||
|
barycenter(PointInputIterator begin, PointInputIterator end,
|
||
|
WeightInputIterator wbegin, int)
|
||
|
{
|
||
|
typedef typename std::iterator_traits<PointInputIterator>::value_type Point;
|
||
|
typedef typename Kernel_traits<Point>::Kernel K;
|
||
|
|
||
|
return CGAL::barycenter(begin, end, wbegin, K());
|
||
|
}
|
||
|
|
||
|
} //namespace CGAL
|
||
|
|
||
|
#endif
|