make SmartRange sum and avg compatible with Eigen vectors

This commit is contained in:
Max Lyon
2019-11-28 11:50:03 +01:00
parent 04e798920d
commit 8fa5f36d13

View File

@@ -78,13 +78,13 @@ struct SmartRangeT
* @param f Functor that is applied to all elements before computing the sum
*/
template <typename Functor>
auto sum(Functor&& f) -> decltype (f(std::declval<HandleT>())+f(std::declval<HandleT>()))
auto sum(Functor&& f) -> decltype (f(std::declval<HandleT>()))
{
auto range = static_cast<const RangeT*>(this);
auto begin = range->begin();
auto end = range->end();
assert(begin != end);
decltype (f(*begin) + f(*begin)) sum = f(*begin);
decltype (f(*begin)) sum = f(*begin);
auto it = begin;
++it;
for (; it != end; ++it)
@@ -99,13 +99,13 @@ struct SmartRangeT
* @param f Functor that is applied to all elements before computing the average.
*/
template <typename Functor>
auto avg(Functor&& f) -> decltype (1.0 * (f(std::declval<HandleT>())+f(std::declval<HandleT>())))
auto avg(Functor&& f) -> decltype (f(std::declval<HandleT>()))
{
auto range = static_cast<const RangeT*>(this);
auto begin = range->begin();
auto end = range->end();
assert(begin != end);
decltype (f(*begin) + f(*begin)) sum = f(*begin);
decltype (f(*begin)) sum = f(*begin);
auto it = begin;
++it;
int n_elements = 1;