provider.hpp

Go to the documentation of this file.

// Andrew Naplavkov
 
#ifndef BOAT_GUI_PROVIDER_HPP
#define BOAT_GUI_PROVIDER_HPP
 
#include <boat/db/catalog.hpp>
#include <boat/gui/caches/cache.hpp>
#include <boat/gui/detail/geometry.hpp>
#include <boat/gui/detail/tile.hpp>
#include <boat/gui/variant.hpp>
#include <random>
 
namespace boat::gui {
 
struct provider {
    std::move_only_function<db::catalog&()> catalog;
    db::layer layer;
    std::shared_ptr<caches::cache> cache;
    size_t cache_key;
    geometry::geographic::grid filter;
 
    std::generator<variant> renderables()
    {
        if (layer.raster)
            co_yield std::ranges::elements_of(rasters());
        else
            co_yield std::ranges::elements_of(vectors());
    }
 
private:
    std::generator<geometry::geographic::geometry_collection> vectors()
    {
        namespace bgi = boost::geometry::index;
        auto tbl = get_or_invoke(cache.get(), cache_key, [&] {
            return catalog().get_table(layer.schema_name, layer.table_name);
        });
        auto& col = layer.column_name;
        auto it = std::ranges::find(tbl.columns, col, &db::column::column_name);
        check(it != tbl.columns.end(), col);
        auto crs = geometry::to_srs_variant(*it);
        auto voids = bgi::rtree<geometry::cartesian::box, bgi::rstar<4>>{};
        auto gen = std::mt19937{std::random_device()()};
        for (auto& box : boxes(filter, crs)) {
            if (bgi::qbegin(voids, bgi::contains(box)) != bgi::qend(voids))
                continue;
            auto a = box.min_corner(), b = box.max_corner();
            auto geoms = get_or_invoke(
                cache.get(),
                std::tuple{cache_key, a.x(), a.y(), b.x(), b.y()},
                [&] {
                    auto rs = catalog().select(
                        tbl,
                        db::bbox{{col}, col, a.x(), a.y(), b.x(), b.y(), 1024});
                    auto wkb = std::vector<blob>(128);
                    auto end = std::ranges::sample(
                        rs | db::view<blob>, wkb.begin(), wkb.size(), gen);
                    wkb.erase(end, wkb.end());
                    auto inv = geometry::transform(
                        geometry::srs_inverse(geometry::transformation(crs)));
                    auto ret = geometry::geographic::geometry_collection{};
                    for (blob_view v : wkb)
                        if (auto g = inv(get<geometry::geographic::variant>(v)))
                            ret.push_back(std::move(*g));
                    return ret;
                });
            if (geoms.empty())
                voids.insert(box);
            else
                co_yield std::move(geoms);
        }
    }
 
    std::generator<raster> rasters()
    {
        auto r = get_or_invoke(cache.get(), cache_key, [&] {
            return catalog().get_raster(layer);
        });
        auto affine = geometry::matrix{{
            {r.xscale, r.xskew, r.xorig},
            {r.yskew, r.yscale, r.yorig},
            {0., 0., 1.},
        }};
        auto crs = geometry::to_srs_variant(r);
        auto uncached = std::vector<tile>{};
        for (auto& t : tiles(filter, r.width, r.height, affine, crs)) {
            auto any =
                cache ? cache->get(std::tuple{cache_key, t}) : std::any{};
            if (!any.has_value()) {
                uncached.push_back(t);
                continue;
            }
            auto rgba =
                std::any_cast<boost::gil::rgba8_image_t>(std::move(any));
            co_yield {
                std::move(rgba), affine * t.affine(r.width, r.height), crs};
        }
        if (uncached.empty())
            co_return;
        for (auto [t, img] : catalog().read(r, std::move(uncached))) {
            auto rgba = gil::to<boost::gil::rgba8_image_t>(const_view(img));
            if (cache)
                cache->put(std::tuple{cache_key, t}, rgba);
            co_yield {
                std::move(rgba), affine * t.affine(r.width, r.height), crs};
        }
    }
};
 
}  // namespace boat::gui
 
#endif  // BOAT_GUI_PROVIDER_HPP