#include "DatasetCategory.hpp"
#include "repo/CategoryDescriptor.hpp"

namespace geopro::app {

CategoryBuckets splitByCategory(const std::vector<geopro::data::DsRow>& rows) {
    const auto& cat = geopro::data::categoryCatalog();
    CategoryBuckets b;
    b.segments.resize(cat.size());
    for (const auto& r : rows)
        for (std::size_t i = 0; i < cat.size(); ++i)
            if (cat[i].classify && cat[i].classify(r)) { b.segments[i].push_back(r); break; }
    return b;
}

}  // namespace geopro::app