mapnik/plugins/input/shape/shape_io.cpp

247 lines
6 KiB
C++
Raw Normal View History

2006-03-31 10:32:02 +00:00
/*****************************************************************************
*
2006-03-31 10:32:02 +00:00
* This file is part of Mapnik (c++ mapping toolkit)
2005-06-14 15:06:59 +00:00
*
* Copyright (C) 2011 Artem Pavlenko
2005-06-14 15:06:59 +00:00
*
2006-03-31 10:32:02 +00:00
* 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,
2005-06-14 15:06:59 +00:00
* but WITHOUT ANY WARRANTY; without even the implied warranty of
2006-03-31 10:32:02 +00:00
* 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
2005-06-14 15:06:59 +00:00
*
2006-03-31 10:32:02 +00:00
*****************************************************************************/
2005-06-14 15:06:59 +00:00
#include "shape_io.hpp"
2011-05-10 15:03:09 +00:00
// mapnik
2011-05-10 15:03:09 +00:00
#include <mapnik/datasource.hpp>
2011-05-10 15:03:09 +00:00
// boost
#include <boost/filesystem/operations.hpp>
#include <boost/make_shared.hpp>
2005-06-14 15:06:59 +00:00
using mapnik::datasource_exception;
using mapnik::geometry_type;
2005-06-14 15:06:59 +00:00
const std::string shape_io::SHP = ".shp";
const std::string shape_io::DBF = ".dbf";
const std::string shape_io::INDEX = ".index";
2005-06-14 15:06:59 +00:00
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)
2005-06-14 15:06:59 +00:00
{
2010-09-06 19:20:59 +00:00
bool ok = (shp_.is_open() && dbf_.is_open());
if (! ok)
{
throw datasource_exception("Shape Plugin: cannot read shape file '" + shape_name + "'");
2010-09-06 19:20:59 +00:00
}
if (open_index)
2010-09-06 19:20:59 +00:00
{
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
}
2010-09-06 19:20:59 +00:00
}
2005-06-14 15:06:59 +00:00
}
shape_io::~shape_io() {}
2005-06-14 15:06:59 +00:00
void shape_io::move_to(int pos)
2005-06-14 15:06:59 +00:00
{
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_);
}
2005-06-14 15:06:59 +00:00
}
int shape_io::type() const
{
return type_;
2005-06-14 15:06:59 +00:00
}
2009-12-16 20:02:06 +00:00
const box2d<double>& shape_io::current_extent() const
2005-06-14 15:06:59 +00:00
{
return cur_extent_;
2005-06-14 15:06:59 +00:00
}
shape_file& shape_io::shp()
{
return shp_;
2005-06-14 15:06:59 +00:00
}
dbf_file& shape_io::dbf()
{
return dbf_;
2005-06-14 15:06:59 +00:00
}
void shape_io::read_polyline(mapnik::geometry_container & geom)
{
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();
if (num_parts == 1)
{
geometry_type* line = new geometry_type(mapnik::LineString);
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);
}
geom.push_back(line);
}
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)
{
geometry_type* line = new geometry_type(mapnik::LineString);
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);
}
geom.push_back(line);
}
}
// 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();
//}
2005-06-14 15:06:59 +00:00
}
void shape_io::read_polygon(mapnik::geometry_container & geom)
2005-06-14 15:06:59 +00:00
{
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);
for (int i = 0; i < num_parts; ++i)
{
parts[i] = record.read_ndr_integer();
}
geometry_type* poly = new geometry_type(mapnik::Polygon);
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();
if (k > 0 && !poly->hit_test(x,y,0))
{
geom.push_back(poly);
poly = new geometry_type(mapnik::Polygon);
}
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);
}
}
geom.push_back(poly);
// 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();
//}
2005-06-14 15:06:59 +00:00
}