/***************************************************************************** * * This file is part of Mapnik (c++ mapping toolkit) * * Copyright (C) 2007 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 * *****************************************************************************/ //$Id$ #include #include #include #include #include #include #include #include // ogr #include "ogr_featureset.hpp" //#define MAPNIK_OGR_USE_WKB_CONVERSION using std::clog; using std::endl; using mapnik::query; using mapnik::Envelope; using mapnik::CoordTransform; using mapnik::Feature; using mapnik::feature_ptr; using mapnik::point_impl; using mapnik::line_string_impl; using mapnik::polygon_impl; using mapnik::geometry2d; using mapnik::geometry_utils; using mapnik::transcoder; ogr_featureset::ogr_featureset(OGRDataSource & dataset, OGRLayer & layer, std::string const& encoding, bool multiple_geometries) : dataset_(dataset), layer_(layer), tr_(new transcoder(encoding)), fidcolumn_(layer_.GetFIDColumn ()), multiple_geometries_(multiple_geometries) { layer_.ResetReading (); } ogr_featureset::~ogr_featureset() {} feature_ptr ogr_featureset::next() { OGRFeature* feat = layer_.GetNextFeature(); if (feat != NULL) { OGRGeometry* geom=feat->GetGeometryRef(); if (geom != NULL) { feature_ptr feature(new Feature(feat->GetFID())); convert_geometry (geom, feature); OGRFeatureDefn* def = layer_.GetLayerDefn(); int fld_count = def->GetFieldCount(); for (int i = 0; i < fld_count; i++) { OGRFieldDefn* fld = def->GetFieldDefn (i); OGRFieldType type_oid = fld->GetType (); std_string name = fld->GetNameRef (); switch (type_oid) { case OFTInteger: { boost::put(*feature,name,feat->GetFieldAsInteger (i)); break; } case OFTReal: { boost::put(*feature,name,feat->GetFieldAsDouble (i)); break; } case OFTString: case OFTWideString: // deprecated ! case OFTWideStringList: // deprecated ! { UnicodeString ustr = tr_->transcode(feat->GetFieldAsString (i)); boost::put(*feature,name,ustr); break; } case OFTIntegerList: case OFTRealList: case OFTStringList: { #ifdef MAPNIK_DEBUG clog << "unhandled type_oid=" << type_oid << endl; #endif break; } case OFTBinary: { #ifdef MAPNIK_DEBUG clog << "unhandled type_oid=" << type_oid << endl; #endif //boost::put(*feature,name,feat->GetFieldAsBinary (i, size)); break; } case OFTDate: case OFTTime: case OFTDateTime: // unhandled ! { #ifdef MAPNIK_DEBUG clog << "unhandled type_oid=" << type_oid << endl; #endif break; } default: // unknown { #ifdef MAPNIK_DEBUG clog << "unknown type_oid=" << type_oid << endl; #endif break; } } } OGRFeature::DestroyFeature (feat); return feature; } OGRFeature::DestroyFeature (feat); } return feature_ptr(); } int ogr_featureset::endian() { const int t = 1; return (*(char*)&t == 0) ? wkbXDR : wkbNDR; } void ogr_featureset::convert_geometry (OGRGeometry* geom, feature_ptr feature) { switch (wkbFlatten (geom->getGeometryType())) { case wkbPoint: convert_point (static_cast(geom), feature); break; case wkbLineString: convert_linestring (static_cast(geom), feature); break; case wkbPolygon: convert_polygon (static_cast(geom), feature); break; case wkbMultiPoint: if (multiple_geometries_) convert_multipoint_2 (static_cast(geom), feature); else convert_multipoint (static_cast(geom), feature); break; case wkbMultiLineString: if (multiple_geometries_) convert_multilinestring_2 (static_cast(geom), feature); else convert_multilinestring (static_cast(geom), feature); break; case wkbMultiPolygon: if (multiple_geometries_) convert_multipolygon_2 (static_cast(geom), feature); else convert_multipolygon (static_cast(geom), feature); break; case wkbGeometryCollection: convert_collection (static_cast(geom), feature); break; case wkbLinearRing: convert_linestring (static_cast(geom), feature); break; case wkbNone: case wkbUnknown: default: #ifdef MAPNIK_DEBUG clog << "unknown geometry_type=" << wkbFlatten (geom->getGeometryType()) << endl; #endif break; } } void ogr_featureset::convert_point (OGRPoint* geom, feature_ptr feature) { geometry2d* point = new point_impl; point->move_to (geom->getX(), geom->getY()); feature->add_geometry (point); } void ogr_featureset::convert_linestring (OGRLineString* geom, feature_ptr feature) { int num_points = geom->getNumPoints (); geometry2d * line = new line_string_impl; line->set_capacity (num_points); line->move_to (geom->getX (0), geom->getY (0)); for (int i=1;iline_to (geom->getX (i), geom->getY (i)); } feature->add_geometry (line); } void ogr_featureset::convert_polygon (OGRPolygon* geom, feature_ptr feature) { OGRLinearRing* exterior = geom->getExteriorRing (); int num_points = exterior->getNumPoints (); int num_interior = geom->getNumInteriorRings (); int capacity = 0; for (int r=0;rgetInteriorRing (r); capacity += interior->getNumPoints (); } geometry2d * poly = new polygon_impl; poly->set_capacity (num_points + capacity); poly->move_to (exterior->getX (0), exterior->getY (0)); for (int i=1;iline_to (exterior->getX (i), exterior->getY (i)); } for (int r=0;rgetInteriorRing (r); num_points = interior->getNumPoints (); poly->move_to(interior->getX (0), interior->getY (0)); for (int i=1;iline_to(interior->getX (i), interior->getY (i)); } } feature->add_geometry (poly); } void ogr_featureset::convert_multipoint (OGRMultiPoint* geom, feature_ptr feature) { int num_geometries = geom->getNumGeometries (); geometry2d* point = new point_impl; for (int i=0;i(geom->getGeometryRef (i)); point->move_to (ogrpoint->getX(), ogrpoint->getY()); } } void ogr_featureset::convert_multipoint_2 (OGRMultiPoint* geom, feature_ptr feature) { int num_geometries = geom->getNumGeometries (); for (int i=0;i(geom->getGeometryRef (i)), feature); } } void ogr_featureset::convert_multilinestring (OGRMultiLineString* geom, feature_ptr feature) { int num_geometries = geom->getNumGeometries (); int num_points = 0; for (int i=0;i(geom->getGeometryRef (i)); num_points += ls->getNumPoints (); } geometry2d * line = new line_string_impl; line->set_capacity (num_points); for (int i=0;i(geom->getGeometryRef (i)); num_points = ls->getNumPoints (); line->move_to (ls->getX (0), ls->getY (0)); for (int i=1;iline_to (ls->getX (i), ls->getY (i)); } } feature->add_geometry (line); } void ogr_featureset::convert_multilinestring_2 (OGRMultiLineString* geom, feature_ptr feature) { int num_geometries = geom->getNumGeometries (); for (int i=0;i(geom->getGeometryRef (i)), feature); } } void ogr_featureset::convert_multipolygon (OGRMultiPolygon* geom, feature_ptr feature) { int num_geometries = geom->getNumGeometries (); int capacity = 0; for (int i=0;i(geom->getGeometryRef (i)); OGRLinearRing* exterior = p->getExteriorRing (); capacity += exterior->getNumPoints (); for (int r=0;rgetNumInteriorRings ();r++) { OGRLinearRing* interior = p->getInteriorRing (r); capacity += interior->getNumPoints (); } } geometry2d * poly = new polygon_impl; poly->set_capacity (capacity); for (int i=0;i(geom->getGeometryRef (i)); OGRLinearRing* exterior = p->getExteriorRing (); int num_points = exterior->getNumPoints (); int num_interior = p->getNumInteriorRings (); poly->move_to (exterior->getX (0), exterior->getY (0)); for (int i=1;iline_to (exterior->getX (i), exterior->getY (i)); } for (int r=0;rgetInteriorRing (r); num_points = interior->getNumPoints (); poly->move_to(interior->getX (0), interior->getY (0)); for (int i=1;iline_to(interior->getX (i), interior->getY (i)); } } } feature->add_geometry (poly); } void ogr_featureset::convert_multipolygon_2 (OGRMultiPolygon* geom, feature_ptr feature) { int num_geometries = geom->getNumGeometries (); for (int i=0;i(geom->getGeometryRef (i)), feature); } } void ogr_featureset::convert_collection (OGRGeometryCollection* geom, feature_ptr feature) { int num_geometries = geom->getNumGeometries (); for (int i=0;igetGeometryRef (i); if (g != NULL) { convert_geometry (g, feature); } } } /* int size=geom->WkbSize (); char data[size]; geom->exportToWkb ((OGRwkbByteOrder) endian(), reinterpret_cast(&data[0])); geometry_utils::from_wkb (*feature, &data[0], size, multiple_geometries_); */