mapnik/src/memory_datasource.cpp

133 lines
3.4 KiB
C++

/*****************************************************************************
*
* This file is part of Mapnik (c++ mapping toolkit)
*
* Copyright (C) 2011 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
*
*****************************************************************************/
// mapnik
#include <mapnik/debug.hpp>
#include <mapnik/query.hpp>
#include <mapnik/box2d.hpp>
#include <mapnik/memory_datasource.hpp>
#include <mapnik/memory_featureset.hpp>
// boost
// stl
#include <algorithm>
namespace mapnik {
struct accumulate_extent
{
accumulate_extent(box2d<double> & ext)
: ext_(ext),first_(true) {}
void operator() (feature_ptr feat)
{
for (std::size_t i=0;i<feat->num_geometries();++i)
{
geometry_type & geom = feat->get_geometry(i);
if ( first_ )
{
first_ = false;
ext_ = geom.envelope();
}
else
{
ext_.expand_to_include(geom.envelope());
}
}
}
box2d<double> & ext_;
bool first_;
};
memory_datasource::memory_datasource(datasource::datasource_t type, bool bbox_check)
: datasource(parameters()),
desc_("in-memory datasource","utf-8"),
type_(type),
bbox_check_(bbox_check) {}
memory_datasource::~memory_datasource() {}
void memory_datasource::push(feature_ptr feature)
{
// TODO - collect attribute descriptors?
//desc_.add_descriptor(attribute_descriptor(fld_name,mapnik::Integer));
features_.push_back(feature);
}
datasource::datasource_t memory_datasource::type() const
{
return type_;
}
featureset_ptr memory_datasource::features(const query& q) const
{
return std::make_shared<memory_featureset>(q.get_bbox(),*this,bbox_check_);
}
featureset_ptr memory_datasource::features_at_point(coord2d const& pt, double tol) const
{
box2d<double> box = box2d<double>(pt.x, pt.y, pt.x, pt.y);
box.pad(tol);
MAPNIK_LOG_DEBUG(memory_datasource) << "memory_datasource: Box=" << box << ", Point x=" << pt.x << ",y=" << pt.y;
return std::make_shared<memory_featureset>(box,*this);
}
void memory_datasource::set_envelope(box2d<double> const& box)
{
extent_ = box;
}
box2d<double> memory_datasource::envelope() const
{
if (!extent_.valid())
{
accumulate_extent func(extent_);
std::for_each(features_.begin(),features_.end(),func);
}
return extent_;
}
boost::optional<datasource::geometry_t> memory_datasource::get_geometry_type() const
{
// TODO - detect this?
return datasource::Collection;
}
layer_descriptor memory_datasource::get_descriptor() const
{
return desc_;
}
size_t memory_datasource::size() const
{
return features_.size();
}
void memory_datasource::clear()
{
features_.clear();
}
}