add weighted average to smart ranges

This commit is contained in:
Max Lyon
2020-08-24 14:54:45 +02:00
parent f0ee8876d9
commit ac67648317

View File

@@ -124,6 +124,34 @@ struct SmartRangeT
return (1.0 / n_elements) * result;
}
/** @brief Computes the weighted average of elements.
*
* Computes the weighted average of all elements in the range after applying the functor \p f.
*
* @param f Functor that is applied to all elements before computing the average.
* @param w Functor returning element weight.
*/
template <typename Functor, typename WeightFunctor>
auto avg(Functor&& f, WeightFunctor&& w) -> typename std::decay<decltype ((1.0/(w(std::declval<HandleT>())+w(std::declval<HandleT>())))*f(std::declval<HandleT>()))>::type
{
auto range = static_cast<const RangeT*>(this);
auto begin = range->begin();
auto end = range->end();
assert(begin != end);
typename std::decay<decltype (w(*begin))>::type weight = w(*begin);
typename std::decay<decltype (w(*begin)*f(*begin))>::type result = weight * f(*begin);
typename std::decay<decltype (w(*begin)+w(*begin))>::type weight_sum = weight;
auto it = begin;
++it;
for (; it != end; ++it)
{
weight = w(*it);
result += weight*f(*it);
weight_sum += weight;
}
return (1.0 / weight_sum) * result;
}
/** @brief Check if any element fulfils condition.
*
* Checks if functor \p f returns true for any of the elements in the range.