#include <imap.hpp>

#include <vector>
#include <iostream>

int main() {
    // applies the function on its way through the sequence
    std::cout << "mapping i*i over [1, 7): ";
    std::vector<int> vec1 = {1, 2, 3, 4, 5, 6};
    for (auto i : iter::imap([] (int i) {return i * i; }, vec1)) {
        std::cout << i << ' ';
    }
    std::cout << '\n';

    // multiple sequences can be used, each iterator will be advanced
    // f(a[0], b[0]) then f(a[1], b[1]) etc.
    std::cout << "vector sum of <1, 2, ..., 6> and <10, 20, ..., 60>: ";
    std::vector<int> vec2 = {10, 20, 30, 40, 50, 60};
    for (auto i : iter::imap([] (int x, int y) { return x + y; }, vec1, vec2)) {
        std::cout << i << ' ';
    }
    std::cout << '\n';

    // it will terminate on the shortest sequence
    std::cout << "vector sum of <1, 2, ..., 6>, <10, 20, ..., 60>, and "
        "<100, 200, 300>: ";
    std::vector<int> vec3 = {100, 200, 300};
    for (auto i : iter::imap([] (int a, int b, int c) { return a + b + c; }, 
                vec1, vec2, vec3)) {
        std::cout << i << ' ';
    }
    std::cout << '\n';
}