1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
#pragma once
#include "cartesian_product.h"
#include "concatenate.h"
#include "enumerate.h"
#include "filtering.h"
#include "mapped.h"
#include "zip.h"
#include <util/generic/adaptor.h>
#include <util/generic/xrange.h>
#include <tuple>
#include <algorithm>
namespace NFuncTools {
using ::Enumerate;
using ::Reversed;
using ::Zip;
using ::Concatenate;
using ::CartesianProduct;
template <typename TValue>
auto Range(TValue from, TValue to, TValue step) {
return xrange(from, to, step);
}
template <typename TValue>
auto Range(TValue from, TValue to) {
return xrange(from, to);
}
template <typename TValue>
auto Range(TValue to) {
return xrange(to);
}
//! Usage: for (i32 x : Map([](i32 x) { return x * x; }, a)) {...}
template <typename TMapper, typename TContainerOrRef>
auto Map(TMapper&& mapper, TContainerOrRef&& container) {
return ::MakeMappedRange(std::forward<TContainerOrRef>(container), std::forward<TMapper>(mapper));
}
//! Usage: for (auto i : Map<int>(floats)) {...}
template <typename TMapResult, typename TContainerOrRef>
auto Map(TContainerOrRef&& container) {
return Map([](const auto& x) { return TMapResult(x); }, std::forward<TContainerOrRef>(container));
}
//! Usage: for (i32 x : Filter(predicate, container)) {...}
template <typename TPredicate, typename TContainerOrRef>
auto Filter(TPredicate&& predicate, TContainerOrRef&& container) {
return ::MakeFilteringRange(std::forward<TContainerOrRef>(container), std::forward<TPredicate>(predicate));
}
}
|
