forked from OctaForge/libostd
for MapRange, specialize stored func in a few ways to avoid octa::Function (and thus avoid the significant overhead it brings)
This commit is contained in:
parent
2652fe9242
commit
36d9a35066
|
@ -412,17 +412,81 @@ T foldr(R range, T init, F func) {
|
||||||
return init;
|
return init;
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename T, typename R>
|
namespace detail {
|
||||||
|
template<typename F>
|
||||||
|
struct MapLambdaArgs: MapLambdaArgs<decltype(&F::operator())> {};
|
||||||
|
|
||||||
|
template<typename C, typename R, typename A>
|
||||||
|
struct MapLambdaArgs<R(C::*)(A) const> {
|
||||||
|
typedef R Result;
|
||||||
|
typedef A Arg;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename F>
|
||||||
|
using MapLambdaRet = typename MapLambdaArgs<F>::Result;
|
||||||
|
|
||||||
|
template<typename F>
|
||||||
|
using MapLambdaArg = typename MapLambdaArgs<F>::Arg;
|
||||||
|
|
||||||
|
template<typename T, typename F>
|
||||||
|
struct MapFuncTest {
|
||||||
|
template<typename FF>
|
||||||
|
static char test(MapLambdaRet<FF> (*)(MapLambdaArg<FF>));
|
||||||
|
template<typename FF>
|
||||||
|
static int test(...);
|
||||||
|
|
||||||
|
static constexpr bool value = (sizeof(test<F>(octa::declval<F>())) == 1) &&
|
||||||
|
octa::IsConvertible<
|
||||||
|
octa::RangeReference<T>,
|
||||||
|
decltype(octa::declval<F>()(octa::declval<
|
||||||
|
octa::RangeReference<T>
|
||||||
|
>()))
|
||||||
|
>::value;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T, typename R, typename F,
|
||||||
|
bool = octa::IsDefaultConstructible<F>::value &&
|
||||||
|
octa::IsMoveConstructible<F>::value
|
||||||
|
> struct MapFuncTypeObjBase {
|
||||||
|
typedef octa::Function<R(octa::RangeReference<T>)> Type;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T, typename R, typename F>
|
||||||
|
struct MapFuncTypeObjBase<T, R, F, true> {
|
||||||
|
typedef F Type;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T, typename R, typename F, bool = MapFuncTest<T, F>::value>
|
||||||
|
struct MapFuncTypeObj {
|
||||||
|
typedef typename MapFuncTypeObjBase<T, R, F>::Type Type;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T, typename R, typename F>
|
||||||
|
struct MapFuncTypeObj<T, R, F, true> {
|
||||||
|
typedef MapLambdaRet<F>(*Type)(MapLambdaArg<F>);
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T, typename R, typename F, bool = octa::IsClass<F>::value>
|
||||||
|
struct MapFuncType {
|
||||||
|
typedef typename MapFuncTypeObj<T, R, F>::Type Type;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T, typename R, typename F>
|
||||||
|
struct MapFuncType<T, R, F, false> {
|
||||||
|
typedef F Type;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, typename F, typename R>
|
||||||
struct MapRange: InputRange<
|
struct MapRange: InputRange<
|
||||||
MapRange<T, R>, octa::RangeCategory<T>, R, R, octa::RangeSize<T>
|
MapRange<T, F, R>, octa::RangeCategory<T>, R, R, octa::RangeSize<T>
|
||||||
> {
|
> {
|
||||||
private:
|
private:
|
||||||
T p_range;
|
T p_range;
|
||||||
octa::Function<R(octa::RangeReference<T>)> p_func;
|
typename octa::detail::MapFuncType<T, R, F>::Type p_func;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
MapRange(): p_range(), p_func() {}
|
MapRange(): p_range(), p_func() {}
|
||||||
template<typename F>
|
|
||||||
MapRange(const T &range, const F &func):
|
MapRange(const T &range, const F &func):
|
||||||
p_range(range), p_func(func) {}
|
p_range(range), p_func(func) {}
|
||||||
MapRange(const MapRange &it):
|
MapRange(const MapRange &it):
|
||||||
|
@ -485,9 +549,9 @@ public:
|
||||||
return p_func(p_range[idx]);
|
return p_func(p_range[idx]);
|
||||||
}
|
}
|
||||||
|
|
||||||
MapRange<T, R> slice(octa::RangeSize<T> start,
|
MapRange slice(octa::RangeSize<T> start,
|
||||||
octa::RangeSize<T> end) {
|
octa::RangeSize<T> end) {
|
||||||
return MapRange<T, R>(p_range.slice(start, end), p_func);
|
return MapRange(p_range.slice(start, end), p_func);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -496,10 +560,11 @@ namespace detail {
|
||||||
= decltype(declval<F>()(octa::declval<octa::RangeReference<R>>()));
|
= decltype(declval<F>()(octa::declval<octa::RangeReference<R>>()));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
template<typename R, typename F>
|
template<typename R, typename F>
|
||||||
MapRange<R, octa::detail::MapReturnType<R, F>> map(R range,
|
MapRange<R, F, octa::detail::MapReturnType<R, F>> map(R range,
|
||||||
F func) {
|
F func) {
|
||||||
return octa::MapRange<R, octa::detail::MapReturnType<R, F>>(range,
|
return octa::MapRange<R, F, octa::detail::MapReturnType<R, F>>(range,
|
||||||
func);
|
func);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue