Batching.hpp

#pragma once
 
#include "util/Assert.hpp"
 
#include <cstddef>
#include <functional>
#include <iterator>
 
namespace util {

void
forEachBatch(std::ranges::forward_range auto&& container, std::size_t batchSize, auto&& fn)
{
    ASSERT(batchSize > 0, "Batch size must be greater than 0");
 
    auto to = std::begin(container);
    auto end = std::end(container);
 
    while (to != end) {
        auto from = to;
 
        auto cnt = batchSize;
        while (to != end and cnt > 0) {
            ++to;
            --cnt;
        }
 
        std::invoke(fn, from, to);
    }
}
 
}  // namespace util