/***************************************************************************** * * 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; }