mapnik/plugins/input/shape/shape_io.hpp
2024-07-22 10:20:47 +01:00

109 lines
3.4 KiB
C++

/*****************************************************************************
*
* This file is part of Mapnik (c++ mapping toolkit)
*
* Copyright (C) 2024 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
*
*****************************************************************************/
#ifndef SHAPE_IO_HPP
#define SHAPE_IO_HPP
// stl
#include <memory>
#include <ios>
// mapnik
#include <mapnik/geometry/box2d.hpp>
#include <mapnik/util/noncopyable.hpp>
#include <mapnik/util/spatial_index.hpp>
// boost
#include <boost/optional.hpp>
//
#include "dbfile.hpp"
#include "shapefile.hpp"
struct shape_io : mapnik::util::noncopyable
{
public:
enum shapeType {
shape_null = 0,
shape_point = 1,
shape_polyline = 3,
shape_polygon = 5,
shape_multipoint = 8,
shape_pointz = 11,
shape_polylinez = 13,
shape_polygonz = 15,
shape_multipointz = 18,
shape_pointm = 21,
shape_polylinem = 23,
shape_polygonm = 25,
shape_multipointm = 28,
shape_multipatch = 31
};
shape_io(std::string const& shape_name, bool open_index = true);
~shape_io();
shape_file& shp();
shape_file& shx();
dbf_file& dbf();
inline boost::optional<shape_file&> index()
{
if (index_)
return boost::optional<shape_file&>(*index_);
return boost::optional<shape_file&>();
}
inline bool has_index() const
{
if (index_ && index_->is_open())
{
bool status = mapnik::util::check_spatial_index(index_->file());
index_->seek(0); // rewind
return status;
}
return false;
}
inline int id() const { return id_; }
void move_to(std::streampos pos);
static void read_bbox(shape_file::record_type& record, mapnik::box2d<double>& bbox);
static mapnik::geometry::geometry<double> read_polyline(shape_file::record_type& record);
static mapnik::geometry::geometry<double> read_polygon(shape_file::record_type& record);
static mapnik::geometry::geometry<double> read_polyline_parts(shape_file::record_type& record,
std::vector<std::pair<int, int>> const& parts);
static mapnik::geometry::geometry<double> read_polygon_parts(shape_file::record_type& record,
std::vector<std::pair<int, int>> const& parts);
shapeType type_;
shape_file shp_;
shape_file shx_;
dbf_file dbf_;
std::unique_ptr<shape_file> index_;
int reclength_;
int id_;
box2d<double> cur_extent_;
static const std::string SHP;
static const std::string SHX;
static const std::string DBF;
static const std::string INDEX;
};
#endif // SHAPE_IO_HPP