261 lines
8.4 KiB
C++
261 lines
8.4 KiB
C++
/*****************************************************************************
|
|
*
|
|
* This file is part of Mapnik (c++ mapping toolkit)
|
|
*
|
|
* Copyright (C) 2021 Artem Pavlenko
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*
|
|
*****************************************************************************/
|
|
|
|
#include "gdal_datasource.hpp"
|
|
#include "gdal_featureset.hpp"
|
|
|
|
// mapnik
|
|
#include <mapnik/debug.hpp>
|
|
#include <mapnik/boolean.hpp>
|
|
#include <mapnik/geom_util.hpp>
|
|
#include <mapnik/timer.hpp>
|
|
#include <mapnik/value/types.hpp>
|
|
|
|
#include <gdal_version.h>
|
|
|
|
using mapnik::datasource;
|
|
using mapnik::parameters;
|
|
|
|
using mapnik::box2d;
|
|
using mapnik::coord2d;
|
|
using mapnik::datasource_exception;
|
|
using mapnik::featureset_ptr;
|
|
using mapnik::layer_descriptor;
|
|
using mapnik::query;
|
|
|
|
DATASOURCE_PLUGIN_IMPL(gdal_datasource_plugin, gdal_datasource);
|
|
DATASOURCE_PLUGIN_EXPORT(gdal_datasource_plugin);
|
|
void gdal_datasource_plugin::after_load() const
|
|
{
|
|
GDALAllRegister();
|
|
}
|
|
|
|
void gdal_datasource_plugin::before_unload() const
|
|
{
|
|
GDALDestroyDriverManager();
|
|
}
|
|
|
|
gdal_datasource::gdal_datasource(parameters const& params)
|
|
: datasource(params)
|
|
, dataset_(nullptr, &GDALClose)
|
|
, desc_(gdal_datasource::name(), "utf-8")
|
|
, nodata_value_(params.get<double>("nodata"))
|
|
, nodata_tolerance_(*params.get<double>("nodata_tolerance", 1e-12))
|
|
{
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_datasource: Initializing...";
|
|
|
|
#ifdef MAPNIK_STATS
|
|
mapnik::progress_timer __stats__(std::clog, "gdal_datasource::init");
|
|
#endif
|
|
|
|
auto file = params.get<std::string>("file");
|
|
if (!file.has_value())
|
|
throw datasource_exception("missing <file> parameter");
|
|
|
|
auto base = params.get<std::string>("base");
|
|
if (base.has_value())
|
|
{
|
|
dataset_name_ = *base + "/" + *file;
|
|
}
|
|
else
|
|
{
|
|
dataset_name_ = *file;
|
|
}
|
|
|
|
shared_dataset_ = *params.get<mapnik::boolean_type>("shared", false);
|
|
band_ = *params.get<mapnik::value_integer>("band", -1);
|
|
|
|
// Maximum memory limitation for image will be simply based on the maximum
|
|
// area we allow for an image. The true memory footprint therefore will vary based
|
|
// on the type of imagery that exists. This is not the maximum size of an image
|
|
// on disk but rather the maximum size we will load into mapnik from GDAL.
|
|
// max_im_area based on 50 mb limit for RGBA
|
|
max_image_area_ = *params.get<mapnik::value_integer>("max_image_area", (50 * 1024 * 1024) / 4);
|
|
|
|
#if GDAL_VERSION_NUM >= 1600
|
|
if (shared_dataset_)
|
|
{
|
|
auto ds = GDALOpenShared(dataset_name_.c_str(), GA_ReadOnly);
|
|
dataset_.reset(static_cast<GDALDataset*>(ds));
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
auto ds = GDALOpen(dataset_name_.c_str(), GA_ReadOnly);
|
|
dataset_.reset(static_cast<GDALDataset*>(ds));
|
|
}
|
|
|
|
if (!dataset_)
|
|
{
|
|
throw datasource_exception(CPLGetLastErrorMsg());
|
|
}
|
|
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_featureset: opened Dataset=" << dataset_.get();
|
|
|
|
nbands_ = dataset_->GetRasterCount();
|
|
width_ = dataset_->GetRasterXSize();
|
|
height_ = dataset_->GetRasterYSize();
|
|
desc_.add_descriptor(mapnik::attribute_descriptor("nodata", mapnik::Double));
|
|
|
|
double tr[6];
|
|
bool bbox_override = false;
|
|
const auto bbox_s = params.get<std::string>("extent");
|
|
if (bbox_s.has_value())
|
|
{
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_datasource: BBox Parameter=" << *bbox_s;
|
|
|
|
bbox_override = extent_.from_string(*bbox_s);
|
|
if (!bbox_override)
|
|
{
|
|
throw datasource_exception("GDAL Plugin: bbox parameter '" + *bbox_s + "' invalid");
|
|
}
|
|
}
|
|
|
|
if (bbox_override)
|
|
{
|
|
tr[0] = extent_.minx();
|
|
tr[1] = extent_.width() / (double)width_;
|
|
tr[2] = 0;
|
|
tr[3] = extent_.maxy();
|
|
tr[4] = 0;
|
|
tr[5] = -extent_.height() / (double)height_;
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_datasource extent override gives Geotransform=" << tr[0] << "," << tr[1] << ","
|
|
<< tr[2] << "," << tr[3] << "," << tr[4] << "," << tr[5];
|
|
}
|
|
else
|
|
{
|
|
if (dataset_->GetGeoTransform(tr) != CPLE_None)
|
|
{
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_datasource GetGeotransform failure gives=" << tr[0] << "," << tr[1] << ","
|
|
<< tr[2] << "," << tr[3] << "," << tr[4] << "," << tr[5];
|
|
}
|
|
else
|
|
{
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_datasource Geotransform=" << tr[0] << "," << tr[1] << "," << tr[2] << ","
|
|
<< tr[3] << "," << tr[4] << "," << tr[5];
|
|
}
|
|
}
|
|
|
|
// TODO - We should throw for true non-north up images, but the check
|
|
// below is clearly too restrictive.
|
|
// https://github.com/mapnik/mapnik/issues/970
|
|
/*
|
|
if (tr[2] != 0 || tr[4] != 0)
|
|
{
|
|
throw datasource_exception("GDAL Plugin: only 'north up' images are supported");
|
|
}
|
|
*/
|
|
|
|
dx_ = tr[1];
|
|
dy_ = tr[5];
|
|
|
|
if (!bbox_override)
|
|
{
|
|
double x0 = tr[0];
|
|
double y0 = tr[3];
|
|
double x1 = tr[0] + width_ * dx_ + height_ * tr[2];
|
|
double y1 = tr[3] + width_ * tr[4] + height_ * dy_;
|
|
|
|
/*
|
|
double x0 = tr[0] + (height_) * tr[2]; // minx
|
|
double y0 = tr[3] + (height_) * tr[5]; // miny
|
|
|
|
double x1 = tr[0] + (width_) * tr[1]; // maxx
|
|
double y1 = tr[3] + (width_) * tr[4]; // maxy
|
|
*/
|
|
|
|
extent_.init(x0, y0, x1, y1);
|
|
}
|
|
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_datasource: Raster Size=" << width_ << "," << height_;
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_datasource: Raster Extent=" << extent_;
|
|
}
|
|
|
|
gdal_datasource::~gdal_datasource()
|
|
{
|
|
MAPNIK_LOG_DEBUG(gdal) << "gdal_featureset: Closing Dataset=" << dataset_.get();
|
|
}
|
|
|
|
datasource::datasource_t gdal_datasource::type() const
|
|
{
|
|
return datasource::Raster;
|
|
}
|
|
|
|
const char* gdal_datasource::name()
|
|
{
|
|
return "gdal";
|
|
}
|
|
|
|
box2d<double> gdal_datasource::envelope() const
|
|
{
|
|
return extent_;
|
|
}
|
|
|
|
std::optional<mapnik::datasource_geometry_t> gdal_datasource::get_geometry_type() const
|
|
{
|
|
return std::nullopt;
|
|
}
|
|
|
|
layer_descriptor gdal_datasource::get_descriptor() const
|
|
{
|
|
return desc_;
|
|
}
|
|
|
|
featureset_ptr gdal_datasource::features(query const& q) const
|
|
{
|
|
#ifdef MAPNIK_STATS
|
|
mapnik::progress_timer __stats__(std::clog, "gdal_datasource::features");
|
|
#endif
|
|
|
|
return std::make_shared<gdal_featureset>(*dataset_,
|
|
band_,
|
|
gdal_query(q),
|
|
extent_,
|
|
width_,
|
|
height_,
|
|
nbands_,
|
|
dx_,
|
|
dy_,
|
|
nodata_value_,
|
|
nodata_tolerance_,
|
|
max_image_area_);
|
|
}
|
|
|
|
featureset_ptr gdal_datasource::features_at_point(coord2d const& pt, double tol) const
|
|
{
|
|
#ifdef MAPNIK_STATS
|
|
mapnik::progress_timer __stats__(std::clog, "gdal_datasource::features_at_point");
|
|
#endif
|
|
|
|
return std::make_shared<gdal_featureset>(*dataset_,
|
|
band_,
|
|
gdal_query(pt),
|
|
extent_,
|
|
width_,
|
|
height_,
|
|
nbands_,
|
|
dx_,
|
|
dy_,
|
|
nodata_value_,
|
|
nodata_tolerance_,
|
|
max_image_area_);
|
|
}
|