catalog.hpp

Go to the documentation of this file.

// Andrew Naplavkov
 
#ifndef BOAT_GDAL_CATALOG_HPP
#define BOAT_GDAL_CATALOG_HPP
 
#include <boat/db/catalog.hpp>
#include <boat/db/query.hpp>
#include <boat/gdal/dataset.hpp>
#include <boat/gdal/detail/raster.hpp>
#include <boat/gdal/detail/vector.hpp>
 
namespace boat::gdal {
 
struct catalog : db::catalog {
    dataset_ptr dataset;
 
    std::vector<db::source> sources() override
    {
        return gdal::subdatasets(dataset.get()) |
               std::views::transform(
                   [](auto p) { return db::source(p.first, p.second); }) |
               std::ranges::to<std::vector>();
    }
 
    std::vector<db::layer> layers() override
    {
        auto ret = std::vector<db::layer>{};
        if (GDALGetRasterCount(dataset.get()))
            ret.push_back({"", "_layer", "raster", true});
        ret.append_range(vectors(dataset.get()));
        return ret;
    }
 
    db::table get_table(std::string_view, std::string_view table_name) override
    {
        return gdal::get_table(GDALDatasetGetLayerByName(
            dataset.get(), std::string{table_name}.data()));
    }
 
    db::rowset select(db::table const& tbl, db::page const& rq) override
    {
        auto q = db::query{};
        q << "\n select";
        if (auto col = OGR_L_GetFIDColumn(GDALDatasetGetLayerByName(
                dataset.get(), tbl.table_name.data()));
            col && std::strlen(col))
            q << " FID as \"" << col << "\",";
        q << " * from " << db::id{tbl.table_name};
        for (auto sep{"\n order by "}; auto& key : rq.order_by)
            q << std::exchange(sep, ", ") << db::id{key.column_name}
              << (key.descending ? " desc" : "");
        q << "\n limit " << to_chars(rq.limit) << "\n offset "
          << to_chars(rq.offset);
        if (auto lyr = execute(dataset.get(), q.text('"', {}).data(), "OGRSQL"))
            return gdal::select(
                lyr.get(), fields::make(lyr.get(), rq.select_list), rq.limit);
        return {};
    }
 
    db::rowset select(db::table const& tbl, db::bbox const& rq) override
    {
        auto lyr =
            GDALDatasetGetLayerByName(dataset.get(), tbl.table_name.data());
        auto fd = OGR_L_GetLayerDefn(lyr);
        OGR_L_SetSpatialFilterRectEx(
            lyr,
            std::max<>(0, OGR_FD_GetGeomFieldIndex(fd, rq.layer_column.data())),
            rq.xmin,
            rq.ymin,
            rq.xmax,
            rq.ymax);
        return gdal::select(lyr, fields::make(lyr, rq.select_list), rq.limit);
    }
 
    void insert(  //
        db::table const& tbl,
        db::rowset const& rs,
        std::stop_token tok = {}) override
    {
        auto lyr =
            GDALDatasetGetLayerByName(dataset.get(), tbl.table_name.data());
        gdal::insert(lyr, rs, tok);
    }
 
    db::table create(db::table const& tbl) override
    {
        return gdal::get_table(add_table(dataset.get(), tbl));
    }
 
    void drop(std::string_view, std::string_view table_name) override
    {
        delete_table(dataset.get(), table_name);
    }
 
    db::raster get_raster(db::layer const&) override
    {
        return gdal::get_raster(dataset.get());
    }
 
    std::generator<std::pair<tile, gil::any_image>> read(
        db::raster rast,
        std::vector<tile> ts) override
    {
        for (auto& t : ts)
            co_yield {t, gdal::read(dataset.get(), rast, t)};
    }
 
    void write(  //
        db::raster const& rast,
        db::rect const& rect,
        gil::any_image_view img) override
    {
        gdal::write(dataset.get(), rast, rect, img);
    }
 
    void set_autocommit(bool on) override
    {
        gdal::set_autocommit(dataset.get(), on);
    }
 
    void commit() override { gdal::commit(dataset.get()); }
};
 
}  // namespace boat::gdal
 
#endif  // BOAT_GDAL_CATALOG_HPP