- forgot a file in previous commit (closes #472 again)
This commit is contained in:
kunitoki
parent
088561bd5e
commit
2a7d41f1ff
1 changed files with 194 additions and 0 deletions
194
plugins/input/ogr/ogr_layer_ptr.hpp
Normal file
194
plugins/input/ogr/ogr_layer_ptr.hpp
Normal file
|
|
@ -0,0 +1,194 @@
|
|||
/*****************************************************************************
|
||||
*
|
||||
* 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
|
||||
*
|
||||
*****************************************************************************/
|
||||
|
||||
#ifndef OGR_LAYER_PTR_HPP
|
||||
#define OGR_LAYER_PTR_HPP
|
||||
|
||||
// stl
|
||||
#include <iostream>
|
||||
#include <fstream>
|
||||
#include <stdexcept>
|
||||
|
||||
// ogr
|
||||
#include <ogrsf_frmts.h>
|
||||
|
||||
class ogr_layer_ptr
|
||||
{
|
||||
public:
|
||||
ogr_layer_ptr()
|
||||
: datasource_(NULL),
|
||||
layer_(NULL),
|
||||
owns_layer_(false),
|
||||
is_valid_(false)
|
||||
{
|
||||
}
|
||||
|
||||
~ogr_layer_ptr()
|
||||
{
|
||||
free_layer();
|
||||
}
|
||||
|
||||
void free_layer()
|
||||
{
|
||||
if (owns_layer_ && layer_ != NULL && datasource_ != NULL)
|
||||
{
|
||||
datasource_->ReleaseResultSet(layer_);
|
||||
}
|
||||
|
||||
datasource_ = NULL;
|
||||
layer_ = NULL;
|
||||
layer_name_ = "";
|
||||
owns_layer_ = false;
|
||||
is_valid_ = false;
|
||||
}
|
||||
|
||||
void layer_by_name(OGRDataSource* const datasource,
|
||||
const std::string& layer_name)
|
||||
{
|
||||
free_layer();
|
||||
|
||||
datasource_ = datasource;
|
||||
|
||||
OGRLayer* ogr_layer = datasource_->GetLayerByName(layer_name.c_str());
|
||||
if (ogr_layer)
|
||||
{
|
||||
layer_name_ = layer_name;
|
||||
layer_ = ogr_layer;
|
||||
is_valid_ = true;
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
std::clog << "OGR Plugin: layer_from_name layer is " << layer_name_ << std::endl;
|
||||
#endif
|
||||
}
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
debug_print_last_error();
|
||||
#endif
|
||||
}
|
||||
|
||||
void layer_by_index(OGRDataSource* const datasource,
|
||||
int layer_index)
|
||||
{
|
||||
free_layer();
|
||||
|
||||
datasource_ = datasource;
|
||||
|
||||
OGRLayer* ogr_layer = datasource_->GetLayer(layer_index);
|
||||
if (ogr_layer)
|
||||
{
|
||||
OGRFeatureDefn* def = ogr_layer->GetLayerDefn();
|
||||
if (def != 0)
|
||||
{
|
||||
layer_ = ogr_layer;
|
||||
layer_name_ = def->GetName();
|
||||
is_valid_ = true;
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
std::clog << "OGR Plugin: layer_from_index layer is " << layer_name_ << std::endl;
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
debug_print_last_error();
|
||||
#endif
|
||||
}
|
||||
|
||||
void layer_by_sql(OGRDataSource* const datasource,
|
||||
const std::string& layer_sql)
|
||||
{
|
||||
free_layer();
|
||||
|
||||
datasource_ = datasource;
|
||||
owns_layer_ = true;
|
||||
|
||||
// TODO - actually filter fields!
|
||||
// http://trac.osgeo.org/gdal/wiki/rfc29_desired_fields
|
||||
// http://trac.osgeo.org/gdal/wiki/rfc28_sqlfunc
|
||||
|
||||
OGRGeometry* spatial_filter = NULL;
|
||||
const char* sql_dialect = NULL;
|
||||
OGRLayer* ogr_layer = datasource_->ExecuteSQL(layer_sql.c_str(), spatial_filter, sql_dialect);
|
||||
|
||||
if (ogr_layer)
|
||||
{
|
||||
OGRFeatureDefn* def = ogr_layer->GetLayerDefn();
|
||||
if (def != 0)
|
||||
{
|
||||
layer_ = ogr_layer;
|
||||
layer_name_ = def->GetName();
|
||||
is_valid_ = true;
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
std::clog << "OGR Plugin: layer_from_sql layer is " << layer_name_ << std::endl;
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
debug_print_last_error();
|
||||
#endif
|
||||
}
|
||||
|
||||
const std::string& layer_name() const
|
||||
{
|
||||
return layer_name_;
|
||||
}
|
||||
|
||||
OGRLayer* layer() const
|
||||
{
|
||||
return layer_;
|
||||
}
|
||||
|
||||
bool is_valid() const
|
||||
{
|
||||
return is_valid_;
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
#ifdef MAPNIK_DEBUG
|
||||
void debug_print_last_error()
|
||||
{
|
||||
if (! is_valid_)
|
||||
{
|
||||
const std::string err = CPLGetLastErrorMsg();
|
||||
if (err.size() == 0)
|
||||
{
|
||||
std::clog << "OGR Plugin: error getting layer" << std::endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
std::clog << "OGR Plugin: " << err << std::endl;
|
||||
}
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
OGRDataSource* datasource_;
|
||||
OGRLayer* layer_;
|
||||
std::string layer_name_;
|
||||
bool owns_layer_;
|
||||
bool is_valid_;
|
||||
};
|
||||
|
||||
#endif // OGR_LAYER_PTR_HPP
|
||||
Loading…
Reference in a new issue