QgsOgrFeatureIterator::QgsOgrFeatureIterator( QgsOgrFeatureSource* source, bool ownSource, const QgsFeatureRequest& request )
: QgsAbstractFeatureIteratorFromSource( source, ownSource, request )
, ogrDataSource( 0 )
, ogrLayer( 0 )
, mSubsetStringSet( false )
, mGeometrySimplifier( NULL )
{
mFeatureFetched = false;
ogrDataSource = OGROpen( TO8F( mSource->mFilePath ), false, NULL );
if ( mSource->mLayerName.isNull() )
{
ogrLayer = OGR_DS_GetLayer( ogrDataSource, mSource->mLayerIndex );
}
else
{
ogrLayer = OGR_DS_GetLayerByName( ogrDataSource, TO8( mSource->mLayerName ) );
}
if ( !mSource->mSubsetString.isEmpty() )
{
ogrLayer = QgsOgrUtils::setSubsetString( ogrLayer, ogrDataSource, mSource->mEncoding, mSource->mSubsetString );
mSubsetStringSet = true;
}
// make sure we fetch just relevant fields
mFetchGeometry = ( mRequest.filterType() == QgsFeatureRequest::FilterRect ) || !( mRequest.flags() & QgsFeatureRequest::NoGeometry );
QgsAttributeList attrs = ( mRequest.flags() & QgsFeatureRequest::SubsetOfAttributes ) ? mRequest.subsetOfAttributes() : mSource->mFields.allAttributesList();
QgsOgrUtils::setRelevantFields( ogrLayer, mSource->mFields.count(), mFetchGeometry, attrs );
// spatial query to select features
if ( mRequest.filterType() == QgsFeatureRequest::FilterRect )
{
OGRGeometryH filter = 0;
QString wktExtent = QString( "POLYGON((%1))" ).arg( mRequest.filterRect().asPolygon() );
QByteArray ba = wktExtent.toAscii();
const char *wktText = ba;
OGR_G_CreateFromWkt(( char ** )&wktText, NULL, &filter );
QgsDebugMsg( "Setting spatial filter using " + wktExtent );
OGR_L_SetSpatialFilter( ogrLayer, filter );
OGR_G_DestroyGeometry( filter );
}
else
{
OGR_L_SetSpatialFilter( ogrLayer, 0 );
}
//start with first feature
rewind();
}
void object::test<9>()
{
char* wkt1 = "LINESTRING(0 0, 10 10)";
err_ = OGR_G_CreateFromWkt(&wkt1, NULL, &g1_);
ensure_equals("Can't import geometry from WKT", OGRERR_NONE, err_);
ensure("Can't create geometry", NULL != g1_);
char* wkt2 = "LINESTRING(0 0, 0 10)";
err_ = OGR_G_CreateFromWkt(&wkt2, NULL, &g2_);
ensure_equals("Can't import geometry from WKT", OGRERR_NONE, err_);
ensure("Can't create geometry", NULL != g2_);
ensure_equals("OGR_G_Touches() failed with FALSE",
OGR_G_Touches(g1_, g2_), TRUE);
char* wkt3 = "POLYGON((20 20, 20 30, 30 20, 20 20))";
err_ = OGR_G_CreateFromWkt(&wkt3, NULL, &g3_);
ensure_equals("Can't import geometry from WKT", OGRERR_NONE, err_);
ensure("Can't create geometry", NULL != g3_);
ensure_equals("OGR_G_Touches() failed with TRUE",
OGR_G_Touches(g1_, g3_), FALSE);
}
QgsOgrFeatureIterator::QgsOgrFeatureIterator( QgsOgrProvider* p, const QgsFeatureRequest& request )
: QgsAbstractFeatureIterator( request )
, P( p )
, ogrDataSource( 0 )
, ogrLayer( 0 )
, mSubsetStringSet( false )
, mGeometrySimplifier( NULL )
{
mFeatureFetched = false;
ogrDataSource = OGROpen( TO8F( P->filePath() ), false, NULL );
if ( P->layerName().isNull() )
{
ogrLayer = OGR_DS_GetLayer( ogrDataSource, P->layerIndex() );
}
else
{
ogrLayer = OGR_DS_GetLayerByName( ogrDataSource, TO8( p->layerName() ) );
}
if ( !P->subsetString().isEmpty() )
{
ogrLayer = P->setSubsetString( ogrLayer, ogrDataSource );
mSubsetStringSet = true;
}
ensureRelevantFields();
// spatial query to select features
if ( mRequest.filterType() == QgsFeatureRequest::FilterRect )
{
OGRGeometryH filter = 0;
QString wktExtent = QString( "POLYGON((%1))" ).arg( mRequest.filterRect().asPolygon() );
QByteArray ba = wktExtent.toAscii();
const char *wktText = ba;
OGR_G_CreateFromWkt(( char ** )&wktText, NULL, &filter );
QgsDebugMsg( "Setting spatial filter using " + wktExtent );
OGR_L_SetSpatialFilter( ogrLayer, filter );
OGR_G_DestroyGeometry( filter );
}
else
{
OGR_L_SetSpatialFilter( ogrLayer, 0 );
}
//start with first feature
rewind();
}
bool Writer::IsValidGeometryWKT(std::string const& input) const
{
#ifdef PDAL_HAVE_GDAL
OGRGeometryH g;
char* wkt = const_cast<char*>(input.c_str());
OGRErr e = OGR_G_CreateFromWkt(&wkt, NULL, &g);
OGR_G_DestroyGeometry(g);
if (e != 0) return false;
return true;
#else
throw pdal_error("GDAL support not available for WKT validation");
#endif
}
Geometry(const std::string& wkt, const SpatialRef& srs)
{
OGRGeometryH geom;
char *p_wkt = const_cast<char *>(wkt.data());
OGRSpatialReferenceH ref = srs.get();
if (srs.empty())
{
ref = NULL;
}
bool isJson = wkt.find("{") != wkt.npos ||
wkt.find("}") != wkt.npos;
if (!isJson)
{
OGRErr err = OGR_G_CreateFromWkt(&p_wkt, ref, &geom);
if (err != OGRERR_NONE)
{
std::cout << "wkt: " << wkt << std::endl;
std::ostringstream oss;
oss << "unable to construct OGR Geometry";
oss << " '" << CPLGetLastErrorMsg() << "'";
throw pdal::pdal_error(oss.str());
}
}
else
{
// Assume it is GeoJSON and try constructing from that
geom = OGR_G_CreateGeometryFromJson(p_wkt);
if (!geom)
throw pdal_error("Unable to create geometry from "
"input GeoJSON");
OGR_G_AssignSpatialReference(geom, ref);
}
newRef(geom);
}
void object::test<3>()
{
ensure("SRS UTM handle is NULL", NULL != srs_utm_);
ensure("SRS LL handle is NULL", NULL != srs_ll_);
err_ = OSRSetUTM(srs_utm_, 11, TRUE);
ensure_equals("Can't set UTM zone", err_, OGRERR_NONE);
err_ = OSRSetWellKnownGeogCS(srs_utm_, "WGS84");
ensure_equals("Can't set GeogCS", err_, OGRERR_NONE);
err_ = OSRSetWellKnownGeogCS(srs_ll_, "WGS84");
ensure_equals("Can't set GeogCS", err_, OGRERR_NONE);
ct_ = OCTNewCoordinateTransformation(srs_ll_, srs_utm_);
ensure("PROJ.4 missing, transforms not available", NULL != ct_);
const char* wkt = "POINT(-117.5 32.0)";
OGRGeometryH geom = NULL;
err_ = OGR_G_CreateFromWkt((char**) &wkt, NULL, &geom);
ensure_equals("Can't import geometry from WKT", OGRERR_NONE, err_);
ensure("Can't create geometry", NULL != geom);
err_ = OGR_G_Transform(geom, ct_);
ensure_equals("OGR_G_Transform() failed", err_, OGRERR_NONE);
OGRSpatialReferenceH srs = NULL;
srs = OGR_G_GetSpatialReference(geom);
char* wktSrs = NULL;
err_ = OSRExportToPrettyWkt(srs, &wktSrs, FALSE);
ensure("Exported SRS to WKT is NULL", NULL != wktSrs);
std::string pretty(wktSrs);
ensure_equals("SRS output is incorrect", pretty.substr(0, 6), std::string("PROJCS"));
OGRFree(wktSrs);
OGR_G_DestroyGeometry(geom);
}
void object::test<10>()
{
OGRErr err = OGRERR_NONE;
// Read feature without geometry
std::string tmp(data_tmp_);
tmp += SEP;
tmp += "tpoly.shp";
OGRDataSourceH ds = OGR_Dr_Open(drv_, tmp.c_str(), false);
ensure("Can't open layer", NULL != ds);
OGRLayerH lyr = OGR_DS_GetLayer(ds, 0);
ensure("Can't get layer", NULL != lyr);
// Set empty filter for attributes
err = OGR_L_SetAttributeFilter(lyr, NULL);
ensure_equals("Can't set attribute filter", OGRERR_NONE, err);
// Set spatial filter
const char* wkt = "LINESTRING(479505 4763195,480526 4762819)";
OGRGeometryH filterGeom = NULL;
err = OGR_G_CreateFromWkt((char**) &wkt, NULL, &filterGeom);
ensure_equals("Can't create geometry from WKT", OGRERR_NONE, err);
OGR_L_SetSpatialFilter(lyr, filterGeom);
// Prepare tester collection
std::vector<int> list;
list.push_back(158);
list.push_back(0);
// Test attributes
ensure_equal_attributes(lyr, "eas_id", list);
OGR_G_DestroyGeometry(filterGeom);
OGR_DS_Destroy(ds);
}
void object::test<9>()
{
// Open directory as a datasource
OGRDataSourceH ds = OGR_Dr_Open(drv_, data_tmp_ .c_str(), false);
ensure("Can't open datasource", NULL != ds);
std::string sql("select * from tpoly where prfedea = '35043413'");
OGRLayerH lyr = OGR_DS_ExecuteSQL(ds, sql.c_str(), NULL, NULL);
ensure("Can't create layer from query", NULL != lyr);
// Prepare tester collection
std::vector<std::string> list;
list.push_back("35043413");
// Test attributes
ensure_equal_attributes(lyr, "prfedea", list);
// Test geometry
const char* wkt = "POLYGON ((479750.688 4764702.000,479658.594 4764670.000,"
"479640.094 4764721.000,479735.906 4764752.000,"
"479750.688 4764702.000))";
OGRGeometryH testGeom = NULL;
OGRErr err = OGR_G_CreateFromWkt((char**) &wkt, NULL, &testGeom);
ensure_equals("Can't create geometry from WKT", OGRERR_NONE, err);
OGR_L_ResetReading(lyr);
OGRFeatureH feat = OGR_L_GetNextFeature(lyr);
ensure("Can't featch feature", NULL != feat);
ensure_equal_geometries(OGR_F_GetGeometryRef(feat), testGeom, 0.001);
OGR_F_Destroy(feat);
OGR_G_DestroyGeometry(testGeom);
OGR_DS_ReleaseResultSet(ds, lyr);
OGR_DS_Destroy(ds);
}
QgsOgrFeatureIterator::QgsOgrFeatureIterator( QgsOgrProvider* p, const QgsFeatureRequest& request )
: QgsAbstractFeatureIterator( request ), P( p )
{
// make sure that only one iterator is active
if ( P->mActiveIterator )
{
QgsMessageLog::logMessage( QObject::tr( "Already active iterator on this provider was closed." ), QObject::tr( "OGR" ) );
P->mActiveIterator->close();
}
P->mActiveIterator = this;
mFeatureFetched = false;
ensureRelevantFields();
// spatial query to select features
if ( mRequest.filterType() == QgsFeatureRequest::FilterRect )
{
OGRGeometryH filter = 0;
QString wktExtent = QString( "POLYGON((%1))" ).arg( mRequest.filterRect().asPolygon() );
QByteArray ba = wktExtent.toAscii();
const char *wktText = ba;
OGR_G_CreateFromWkt(( char ** )&wktText, NULL, &filter );
QgsDebugMsg( "Setting spatial filter using " + wktExtent );
OGR_L_SetSpatialFilter( P->ogrLayer, filter );
OGR_G_DestroyGeometry( filter );
}
else
{
OGR_L_SetSpatialFilter( P->ogrLayer, 0 );
}
//start with first feature
rewind();
}
