geopro/src/app/DatasetCategory.cpp

17 lines
512 B
C++

#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