#include <groupby.hpp>

#include <vector>
#include <iostream>
#include <string>
#include <functional>


int main() {
    auto len = std::mem_fn(&std::string::length);
    std::vector<std::string> vec = {
        "hi", "ab", "ho",
        "abc", "def",
        "abcde", "efghi"
    };

    std::cout << "strings grouped by their length\n";
    for (auto&& gb : iter::groupby(vec, len)) {
        std::cout << "key(" << gb.first << "): ";
        for (auto&& s : gb.second) {
            std::cout << s << "  ";
        }
        std::cout << '\n';
    }

    // groups may be skipped entirely or partially consumed
    std::cout << "skipping length of 3\n";
    for (auto&& gb : iter::groupby(vec, len)) {
        if (gb.first == 3) {
            continue;
        }
        std::cout << "key(" << gb.first << "): ";
        for (auto&& s : gb.second) {
            std::cout << s << "  ";
        }
        std::cout << '\n';
    }
   

    std::cout << "ints grouped by their value:\n";
    std::vector<int> ivec = {5, 5, 6, 6, 19, 19, 19, 19, 69, 0, 10, 10};
    for (auto gb : iter::groupby(ivec)) {
        std::cout << "key(" << gb.first << "): ";
        for (auto s : gb.second) {
            std::cout << s << "  ";
        }
        std::cout << '\n';
    }
}