libostd/examples/range_pipe.cc

97 lines
2.6 KiB
C++

/** @example range_pipe.cc
*
* Simple examples of usage of the range system using the pipe syntax.
*/
#include <array>
#include <ctime>
#include <ostd/range.hh>
#include <ostd/io.hh>
#include <ostd/algorithm.hh>
using namespace ostd;
int main() {
/* algorithm: map - prints 0.5 to 9.5 each on new line */
writeln("range map test");
for (float f: range(10) | map([](int v) { return v + 0.5f; })) {
writeln(f);
}
/* algorithm: filter - prints 10, 15, 8 each on new line */
writeln("range filter test");
auto il = { 5, 5, 5, 5, 5, 10, 15, 4, 8, 2 };
for (int i: iter(il) | filter([](int v) { return v > 5; })) {
writeln(i);
}
/* prints ABCDEF (ASCII 65, 66, 67, 68, 69, 70) */
writeln("string gen test");
auto mr = range(6) | map([](int v) { return char(v + 65); });
std::string s{mr.iter_begin(), mr.iter_end()};
writeln(s);
/* join a few ranges together - prints 11, 22, 33 ... 99 each on new line */
writeln("range join test");
auto x = { 11, 22, 33 };
auto y = { 44, 55, 66 };
auto z = { 77, 88, 99 };
for (auto i: iter(x) | join(iter(y), iter(z))) {
writeln(i);
}
/* chunk a range into subranges - prints
* {11, 22, 33}
* {44, 55, 66}
* {77, 88, 99}
*/
writeln("range chunk test");
auto cr = { 11, 22, 33, 44, 55, 66, 77, 88, 99 };
for (auto r: iter(cr) | chunks(3)) {
writeln(r);
}
/* take test, prints only first 4 */
writeln("range take test");
for (auto r: iter(cr) | take(4)) {
writeln(r);
}
/* {11, 44, 77}, {22, 55, 88}, {33, 66, 99} */
writeln("range zip test");
for (auto v: iter(x) | zip(iter(y), iter(z))) {
writeln(v);
}
/* 2-tuple zip uses std::pair */
writeln("2-tuple range zip");
for (auto v: iter({ 5, 10, 15, 20 }) | zip(iter({ 6, 11, 16, 21 }))) {
writeln(v.first, ", ", v.second);
}
/* more complex pipe */
writeln("several piped algorithms");
srand(static_cast<unsigned int>(std::time(nullptr)));
std::array<int, 100> arr;
generate(iter(arr), []() { return rand() % 128; });
auto r = iter(arr)
| sort ()
| filter([](auto v) { return v >= 65 && v <= 90; })
| map ([](auto v) { return char(v); });
writeln(std::string{r.iter_begin(), r.iter_end()});
/* "list comprehensions" */
writeln("list initialization");
auto vr = range(20)
| filter([](int v) { return v % 2 == 0; })
| map ([](int v) { return v * 2; });
std::vector<int> test{vr.iter_begin(), vr.iter_end()};
writeln(test);
}