463 lines
11 KiB
C++
463 lines
11 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
|
|
*
|
|
*****************************************************************************/
|
|
|
|
#include "shape_io.hpp"
|
|
|
|
// mapnik
|
|
#include <mapnik/datasource.hpp>
|
|
|
|
// boost
|
|
#include <boost/filesystem/operations.hpp>
|
|
#include <boost/make_shared.hpp>
|
|
|
|
using mapnik::datasource_exception;
|
|
using mapnik::geometry_type;
|
|
|
|
const std::string shape_io::SHP = ".shp";
|
|
const std::string shape_io::DBF = ".dbf";
|
|
const std::string shape_io::INDEX = ".index";
|
|
|
|
shape_io::shape_io(const std::string& shape_name, bool open_index)
|
|
: type_(shape_null),
|
|
shp_(shape_name + SHP),
|
|
dbf_(shape_name + DBF),
|
|
reclength_(0),
|
|
id_(0)
|
|
{
|
|
bool ok = (shp_.is_open() && dbf_.is_open());
|
|
if (! ok)
|
|
{
|
|
throw datasource_exception("Shape Plugin: cannot read shape file '" + shape_name + "'");
|
|
}
|
|
|
|
if (open_index)
|
|
{
|
|
try
|
|
{
|
|
index_= boost::make_shared<shape_file>(shape_name + INDEX);
|
|
}
|
|
catch (...)
|
|
{
|
|
#ifdef MAPNIK_DEBUG
|
|
std::clog << "Shape Plugin: warning - could not open index: '" + shape_name + INDEX + "'" << std::endl;
|
|
#endif
|
|
}
|
|
}
|
|
}
|
|
|
|
shape_io::~shape_io() {}
|
|
|
|
void shape_io::move_to(int pos)
|
|
{
|
|
shp_.seek(pos);
|
|
id_ = shp_.read_xdr_integer();
|
|
reclength_ = shp_.read_xdr_integer();
|
|
type_ = shp_.read_ndr_integer();
|
|
|
|
if (type_ != shape_null && type_ != shape_point && type_ != shape_pointm && type_ != shape_pointz)
|
|
{
|
|
shp_.read_envelope(cur_extent_);
|
|
}
|
|
}
|
|
|
|
int shape_io::type() const
|
|
{
|
|
return type_;
|
|
}
|
|
|
|
const box2d<double>& shape_io::current_extent() const
|
|
{
|
|
return cur_extent_;
|
|
}
|
|
|
|
shape_file& shape_io::shp()
|
|
{
|
|
return shp_;
|
|
}
|
|
|
|
#if 0
|
|
shape_file& shape_io::shx()
|
|
{
|
|
return shx_;
|
|
}
|
|
#endif
|
|
|
|
dbf_file& shape_io::dbf()
|
|
{
|
|
return dbf_;
|
|
}
|
|
|
|
geometry_type* shape_io::read_polyline()
|
|
{
|
|
shape_file::record_type record(reclength_ * 2 - 36);
|
|
shp_.read_record(record);
|
|
|
|
int num_parts = record.read_ndr_integer();
|
|
int num_points = record.read_ndr_integer();
|
|
geometry_type* line = new geometry_type(mapnik::LineString);
|
|
line->set_capacity(num_points + num_parts);
|
|
if (num_parts == 1)
|
|
{
|
|
line->set_capacity(num_points + 1);
|
|
record.skip(4);
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
line->move_to(x, y);
|
|
for (int i = 1; i < num_points; ++i)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
line->line_to(x, y);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::vector<int> parts(num_parts);
|
|
for (int i = 0; i < num_parts; ++i)
|
|
{
|
|
parts[i] = record.read_ndr_integer();
|
|
}
|
|
|
|
int start, end;
|
|
for (int k = 0; k < num_parts; ++k)
|
|
{
|
|
start = parts[k];
|
|
if (k == num_parts - 1)
|
|
{
|
|
end = num_points;
|
|
}
|
|
else
|
|
{
|
|
end = parts[k + 1];
|
|
}
|
|
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
line->move_to(x, y);
|
|
|
|
for (int j = start + 1; j < end; ++j)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
line->line_to(x, y);
|
|
}
|
|
}
|
|
}
|
|
return line;
|
|
}
|
|
|
|
geometry_type* shape_io::read_polylinem()
|
|
{
|
|
shape_file::record_type record(reclength_ * 2 - 36);
|
|
shp_.read_record(record);
|
|
|
|
int num_parts = record.read_ndr_integer();
|
|
int num_points = record.read_ndr_integer();
|
|
geometry_type* line = new geometry_type(mapnik::LineString);
|
|
line->set_capacity(num_points + num_parts);
|
|
if (num_parts == 1)
|
|
{
|
|
record.skip(4);
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
line->move_to(x, y);
|
|
for (int i = 1; i < num_points; ++i)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
line->line_to(x, y);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::vector<int> parts(num_parts);
|
|
for (int i = 0; i < num_parts; ++i)
|
|
{
|
|
parts[i] = record.read_ndr_integer();
|
|
}
|
|
|
|
int start, end;
|
|
for (int k = 0; k < num_parts; ++k)
|
|
{
|
|
start = parts[k];
|
|
if (k == num_parts - 1)
|
|
{
|
|
end = num_points;
|
|
}
|
|
else
|
|
{
|
|
end = parts[k + 1];
|
|
}
|
|
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
line->move_to(x, y);
|
|
|
|
for (int j = start + 1; j < end; ++j)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
line->line_to(x, y);
|
|
}
|
|
}
|
|
}
|
|
|
|
// m-range
|
|
//double m0=record.read_double();
|
|
//double m1=record.read_double();
|
|
|
|
//for (int i=0;i<num_points;++i)
|
|
//{
|
|
// double m=record.read_double();
|
|
//}
|
|
|
|
return line;
|
|
}
|
|
|
|
geometry_type* shape_io::read_polylinez()
|
|
{
|
|
shape_file::record_type record(reclength_ * 2 - 36);
|
|
shp_.read_record(record);
|
|
|
|
int num_parts = record.read_ndr_integer();
|
|
int num_points = record.read_ndr_integer();
|
|
geometry_type* line = new geometry_type(mapnik::LineString);
|
|
line->set_capacity(num_points + num_parts);
|
|
if (num_parts == 1)
|
|
{
|
|
record.skip(4);
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
line->move_to(x, y);
|
|
for (int i = 1; i < num_points; ++i)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
line->line_to(x, y);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::vector<int> parts(num_parts);
|
|
for (int i = 0; i < num_parts; ++i)
|
|
{
|
|
parts[i] = record.read_ndr_integer();
|
|
}
|
|
|
|
int start, end;
|
|
for (int k = 0; k < num_parts; ++k)
|
|
{
|
|
start = parts[k];
|
|
if (k == num_parts - 1)
|
|
{
|
|
end = num_points;
|
|
}
|
|
else
|
|
{
|
|
end = parts[k + 1];
|
|
}
|
|
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
line->move_to(x, y);
|
|
|
|
for (int j = start + 1; j < end; ++j)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
line->line_to(x, y);
|
|
}
|
|
}
|
|
}
|
|
|
|
// z-range
|
|
//double z0=record.read_double();
|
|
//double z1=record.read_double();
|
|
//for (int i=0;i<num_points;++i)
|
|
// {
|
|
// double z=record.read_double();
|
|
// }
|
|
|
|
// m-range
|
|
//double m0=record.read_double();
|
|
//double m1=record.read_double();
|
|
|
|
//for (int i=0;i<num_points;++i)
|
|
//{
|
|
// double m=record.read_double();
|
|
//}
|
|
|
|
return line;
|
|
}
|
|
|
|
geometry_type* shape_io::read_polygon()
|
|
{
|
|
shape_file::record_type record(reclength_ * 2 - 36);
|
|
shp_.read_record(record);
|
|
|
|
int num_parts = record.read_ndr_integer();
|
|
int num_points = record.read_ndr_integer();
|
|
std::vector<int> parts(num_parts);
|
|
geometry_type* poly = new geometry_type(mapnik::Polygon);
|
|
poly->set_capacity(num_points + num_parts);
|
|
for (int i = 0; i < num_parts; ++i)
|
|
{
|
|
parts[i] = record.read_ndr_integer();
|
|
}
|
|
|
|
for (int k = 0; k < num_parts; k++)
|
|
{
|
|
int start = parts[k];
|
|
int end;
|
|
if (k == num_parts - 1)
|
|
{
|
|
end = num_points;
|
|
}
|
|
else
|
|
{
|
|
end = parts[k + 1];
|
|
}
|
|
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
poly->move_to(x, y);
|
|
|
|
for (int j=start+1;j<end;j++)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
poly->line_to(x, y);
|
|
}
|
|
}
|
|
return poly;
|
|
}
|
|
|
|
geometry_type* shape_io::read_polygonm()
|
|
{
|
|
shape_file::record_type record(reclength_ * 2 - 36);
|
|
shp_.read_record(record);
|
|
|
|
int num_parts = record.read_ndr_integer();
|
|
int num_points = record.read_ndr_integer();
|
|
std::vector<int> parts(num_parts);
|
|
geometry_type* poly = new geometry_type(mapnik::Polygon);
|
|
poly->set_capacity(num_points + num_parts);
|
|
for (int i = 0; i < num_parts; ++i)
|
|
{
|
|
parts[i] = record.read_ndr_integer();
|
|
}
|
|
|
|
for (int k = 0; k < num_parts; k++)
|
|
{
|
|
int start = parts[k];
|
|
int end;
|
|
if (k == num_parts - 1)
|
|
{
|
|
end = num_points;
|
|
}
|
|
else
|
|
{
|
|
end = parts[k + 1];
|
|
}
|
|
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
poly->move_to(x, y);
|
|
|
|
for (int j = start + 1; j < end; j++)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
poly->line_to(x, y);
|
|
}
|
|
}
|
|
|
|
// m-range
|
|
//double m0=record.read_double();
|
|
//double m1=record.read_double();
|
|
|
|
//for (int i=0;i<num_points;++i)
|
|
//{
|
|
// double m=record.read_double();
|
|
//}
|
|
return poly;
|
|
}
|
|
|
|
geometry_type* shape_io::read_polygonz()
|
|
{
|
|
shape_file::record_type record(reclength_ * 2 - 36);
|
|
shp_.read_record(record);
|
|
|
|
int num_parts = record.read_ndr_integer();
|
|
int num_points = record.read_ndr_integer();
|
|
std::vector<int> parts(num_parts);
|
|
geometry_type* poly = new geometry_type(mapnik::Polygon);
|
|
poly->set_capacity(num_points + num_parts);
|
|
for (int i = 0; i < num_parts; ++i)
|
|
{
|
|
parts[i] = record.read_ndr_integer();
|
|
}
|
|
|
|
for (int k = 0; k < num_parts; k++)
|
|
{
|
|
int start = parts[k];
|
|
int end;
|
|
if (k == num_parts - 1)
|
|
{
|
|
end = num_points;
|
|
}
|
|
else
|
|
{
|
|
end = parts[k + 1];
|
|
}
|
|
|
|
double x = record.read_double();
|
|
double y = record.read_double();
|
|
poly->move_to(x, y);
|
|
|
|
for (int j = start + 1; j < end; j++)
|
|
{
|
|
x = record.read_double();
|
|
y = record.read_double();
|
|
poly->line_to(x, y);
|
|
}
|
|
}
|
|
|
|
// z-range
|
|
//double z0=record.read_double();
|
|
//double z1=record.read_double();
|
|
//for (int i=0;i<num_points;++i)
|
|
//{
|
|
// double z=record.read_double();
|
|
//}
|
|
|
|
// m-range
|
|
//double m0=record.read_double();
|
|
//double m1=record.read_double();
|
|
|
|
//for (int i=0;i<num_points;++i)
|
|
//{
|
|
// double m=record.read_double();
|
|
//}
|
|
|
|
return poly;
|
|
}
|