forked from OctaForge/libostd
59 lines
1.6 KiB
C++
59 lines
1.6 KiB
C++
/* OctaSTD extensions for std::vector.
|
|
*
|
|
* This file is part of OctaSTD. See COPYING.md for futher information.
|
|
*/
|
|
|
|
#ifndef OSTD_VECTOR_HH
|
|
#define OSTD_VECTOR_HH
|
|
|
|
#include <vector>
|
|
#include <memory>
|
|
#include <type_traits>
|
|
|
|
#include "ostd/range.hh"
|
|
|
|
namespace ostd {
|
|
|
|
template<typename T, typename A>
|
|
struct ranged_traits<std::vector<T, A>> {
|
|
static PointerRange<T> iter(std::vector<T, A> &v) {
|
|
return PointerRange<T>{v.data(), v.data() + v.size()};
|
|
}
|
|
};
|
|
|
|
template<typename T, typename A>
|
|
struct ranged_traits<std::vector<T, A> const> {
|
|
static PointerRange<T const> iter(std::vector<T, A> const &v) {
|
|
return PointerRange<T const>{v.data(), v.data() + v.size()};
|
|
}
|
|
};
|
|
|
|
template<typename T, typename A = std::allocator<T>, typename R>
|
|
inline std::vector<T, A> make_vector(R range, A const &alloc = A{}) {
|
|
std::vector<T, A> ret{alloc};
|
|
using C = RangeCategory<R>;
|
|
if constexpr(std::is_convertible_v<C, FiniteRandomAccessRangeTag>) {
|
|
/* finite random access or contiguous */
|
|
auto h = range.half();
|
|
ret.reserve(range.size());
|
|
ret.insert(ret.end(), h, h + range.size());
|
|
} else {
|
|
/* infinite random access and below */
|
|
for (; !range.empty(); range.pop_front()) {
|
|
ret.push_back(range.front());
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
template<typename R, typename A = std::allocator<RangeValue<R>>>
|
|
inline std::vector<RangeValue<R>, A> make_vector(
|
|
R &&range, A const &alloc = A{}
|
|
) {
|
|
return make_vector<RangeValue<R>, A>(std::forward<R>(range), alloc);
|
|
}
|
|
|
|
} /* namespace ostd */
|
|
|
|
#endif
|