OGRErr GPkgEnvelopeToOGR(GByte *pabyGpkg, size_t szGpkg, OGREnvelope *poEnv)
{
CPLAssert( poEnv != NULL );
CPLAssert( pabyGpkg != NULL );
GPkgHeader oHeader;
/* Read header */
OGRErr err = GPkgHeaderFromWKB(pabyGpkg, &oHeader);
if ( err != OGRERR_NONE )
return err;
if ( oHeader.bEmpty || oHeader.iDims == 0 )
{
return OGRERR_FAILURE;
}
poEnv->MinX = oHeader.MinX;
poEnv->MaxX = oHeader.MaxX;
poEnv->MinY = oHeader.MinY;
poEnv->MaxY = oHeader.MaxY;
// if ( oHeader.iDims == 3 )
// {
// poEnv->MinZ = oHeader.MinZ;
// poEnv->MaxZ = oHeader.MaxZ;
// }
return OGRERR_NONE;
}
OGRGeometry* GPkgGeometryToOGR(const GByte *pabyGpkg, size_t szGpkg, OGRSpatialReference *poSrs)
{
CPLAssert( pabyGpkg != NULL );
GPkgHeader oHeader;
OGRGeometry *poGeom;
/* Read header */
OGRErr err = GPkgHeaderFromWKB(pabyGpkg, &oHeader);
if ( err != OGRERR_NONE )
return NULL;
/* WKB pointer */
const GByte *pabyWkb = pabyGpkg + oHeader.szHeader;
size_t szWkb = szGpkg - oHeader.szHeader;
/* Parse WKB */
err = OGRGeometryFactory::createFromWkb((GByte*)pabyWkb, poSrs, &poGeom, szWkb);
if ( err != OGRERR_NONE )
return NULL;
return poGeom;
}
void OGRGeoPackageLayer::BuildFeatureDefn( const char *pszLayerName,
sqlite3_stmt *hStmt )
{
m_poFeatureDefn = new OGRSQLiteFeatureDefn( pszLayerName );
SetDescription( m_poFeatureDefn->GetName() );
m_poFeatureDefn->SetGeomType(wkbNone);
m_poFeatureDefn->Reference();
int nRawColumns = sqlite3_column_count( hStmt );
panFieldOrdinals = (int *) CPLMalloc( sizeof(int) * nRawColumns );
int iCol;
for( iCol = 0; iCol < nRawColumns; iCol++ )
{
OGRFieldDefn oField( OGRSQLiteParamsUnquote(sqlite3_column_name( hStmt, iCol )),
OFTString );
// In some cases, particularly when there is a real name for
// the primary key/_rowid_ column we will end up getting the
// primary key column appearing twice. Ignore any repeated names.
if( m_poFeatureDefn->GetFieldIndex( oField.GetNameRef() ) != -1 )
continue;
if( EQUAL(oField.GetNameRef(), "FID") )
{
CPLFree(m_pszFidColumn);
m_pszFidColumn = CPLStrdup(oField.GetNameRef());
iFIDCol = iCol;
}
if( m_pszFidColumn != NULL && EQUAL(m_pszFidColumn, oField.GetNameRef()))
continue;
// The rowid is for internal use, not a real column.
if( EQUAL(oField.GetNameRef(),"_rowid_") )
continue;
// this will avoid the old geom field to appear when running something
// like "select st_buffer(geom,5) as geom, * from my_layer"
if( m_poFeatureDefn->GetGeomFieldCount() &&
EQUAL(oField.GetNameRef(), m_poFeatureDefn->GetGeomFieldDefn(0)->GetNameRef()) )
continue;
int nColType = sqlite3_column_type( hStmt, iCol );
const char * pszDeclType = sqlite3_column_decltype(hStmt, iCol);
// Recognize a geometry column from trying to build the geometry
// Useful for OGRSQLiteSelectLayer
if( nColType == SQLITE_BLOB && m_poFeatureDefn->GetGeomFieldCount() == 0 )
{
const int nBytes = sqlite3_column_bytes( hStmt, iCol );
if( nBytes > 4 )
{
int iGpkgSize = sqlite3_column_bytes(hStmt, iCol);
const GByte* pabyGpkg = (const GByte*)sqlite3_column_blob( hStmt, iCol );
GPkgHeader oHeader;
OGRGeometry* poGeom = NULL;
int nSRID;
if( GPkgHeaderFromWKB(pabyGpkg, &oHeader) == OGRERR_NONE )
{
OGRGeomFieldDefn oGeomField(oField.GetNameRef(), wkbUnknown);
/* Read the SRS */
OGRSpatialReference *poSRS = m_poDS->GetSpatialRef(oHeader.iSrsId);
if ( poSRS )
{
oGeomField.SetSpatialRef(poSRS);
poSRS->Dereference();
}
OGRwkbGeometryType eGeomType = wkbUnknown;
if( pszDeclType != NULL )
{
eGeomType = GPkgGeometryTypeToWKB(pszDeclType, (oHeader.iDims == 3));
if( eGeomType != wkbNone )
oGeomField.SetType( eGeomType );
}
#ifdef SQLITE_HAS_COLUMN_METADATA
const char* pszTableName = sqlite3_column_table_name( hStmt, iCol );
if( oGeomField.GetType() == wkbUnknown && pszTableName != NULL )
{
OGRGeoPackageLayer* poLayer = (OGRGeoPackageLayer*)
m_poDS->GetLayerByName(pszTableName);
if( poLayer != NULL && poLayer->GetLayerDefn()->GetGeomFieldCount() > 0)
{
oGeomField.SetType( poLayer->GetLayerDefn()->GetGeomFieldDefn(0)->GetType() );
}
}
#endif
m_poFeatureDefn->AddGeomFieldDefn(&oGeomField);
iGeomCol = iCol;
continue;
}
// Try also spatialite geometry blobs
else if( OGRSQLiteLayer::ImportSpatiaLiteGeometry( pabyGpkg, iGpkgSize,
&poGeom, &nSRID ) == OGRERR_NONE )
{
OGRGeomFieldDefn oGeomField(oField.GetNameRef(), wkbUnknown);
/* Read the SRS */
OGRSpatialReference *poSRS = m_poDS->GetSpatialRef(nSRID);
if ( poSRS )
{
oGeomField.SetSpatialRef(poSRS);
poSRS->Dereference();
}
delete poGeom;
m_poFeatureDefn->AddGeomFieldDefn(&oGeomField);
iGeomCol = iCol;
continue;
}
}
}
switch( nColType )
{
case SQLITE_INTEGER:
if( CSLTestBoolean(CPLGetConfigOption("OGR_PROMOTE_TO_INTEGER64", "FALSE")) )
oField.SetType( OFTInteger64 );
else
{
GIntBig nVal = sqlite3_column_int64(hStmt, iCol);
if( (GIntBig)(int)nVal == nVal )
oField.SetType( OFTInteger );
else
oField.SetType( OFTInteger64 );
}
break;
case SQLITE_FLOAT:
oField.SetType( OFTReal );
break;
case SQLITE_BLOB:
oField.SetType( OFTBinary );
break;
default:
/* leave it as OFTString */;
}
if (pszDeclType != NULL)
{
OGRFieldSubType eSubType;
int nMaxWidth;
OGRFieldType eFieldType = GPkgFieldToOGR(pszDeclType, eSubType, nMaxWidth);
if( (int)eFieldType <= OFTMaxType )
{
oField.SetType(eFieldType);
oField.SetSubType(eSubType);
oField.SetWidth(nMaxWidth);
}
}
m_poFeatureDefn->AddFieldDefn( &oField );
panFieldOrdinals[m_poFeatureDefn->GetFieldCount() - 1] = iCol;
}
}
void OGRGeoPackageLayer::BuildFeatureDefn( const char *pszLayerName,
sqlite3_stmt *hStmt )
{
m_poFeatureDefn = new OGRSQLiteFeatureDefn( pszLayerName );
SetDescription( m_poFeatureDefn->GetName() );
m_poFeatureDefn->SetGeomType(wkbNone);
m_poFeatureDefn->Reference();
int nRawColumns = sqlite3_column_count( hStmt );
panFieldOrdinals = (int *) CPLMalloc( sizeof(int) * nRawColumns );
int iCol;
for( iCol = 0; iCol < nRawColumns; iCol++ )
{
OGRFieldDefn oField( OGRSQLiteParamsUnquote(sqlite3_column_name( hStmt, iCol )),
OFTString );
// In some cases, particularly when there is a real name for
// the primary key/_rowid_ column we will end up getting the
// primary key column appearing twice. Ignore any repeated names.
if( m_poFeatureDefn->GetFieldIndex( oField.GetNameRef() ) != -1 )
continue;
if( EQUAL(oField.GetNameRef(), "FID") )
{
CPLFree(m_pszFidColumn);
m_pszFidColumn = CPLStrdup(oField.GetNameRef());
iFIDCol = iCol;
}
if( m_pszFidColumn != NULL && EQUAL(m_pszFidColumn, oField.GetNameRef()))
continue;
// The rowid is for internal use, not a real column.
if( EQUAL(oField.GetNameRef(),"_rowid_") )
continue;
int nColType = sqlite3_column_type( hStmt, iCol );
const char * pszDeclType = sqlite3_column_decltype(hStmt, iCol);
// Recognize a geometry column from trying to build the geometry
// Usefull for OGRSQLiteSelectLayer
if( nColType == SQLITE_BLOB && m_poFeatureDefn->GetGeomFieldCount() == 0 )
{
const int nBytes = sqlite3_column_bytes( hStmt, iCol );
if( nBytes > 4 )
{
const GByte* pabyGpkg = (const GByte*)sqlite3_column_blob( hStmt, iCol );
GPkgHeader oHeader;
if( GPkgHeaderFromWKB(pabyGpkg, &oHeader) == OGRERR_NONE )
{
OGRGeomFieldDefn oGeomField(oField.GetNameRef(), wkbUnknown);
/* Read the SRS */
OGRSpatialReference *poSRS = m_poDS->GetSpatialRef(oHeader.iSrsId);
if ( poSRS )
{
oGeomField.SetSpatialRef(poSRS);
poSRS->Dereference();
}
OGRwkbGeometryType eGeomType = wkbUnknown;
if( pszDeclType != NULL )
{
eGeomType = GPkgGeometryTypeToWKB(pszDeclType, (oHeader.iDims == 3));
if( eGeomType != wkbNone )
oGeomField.SetType( eGeomType );
}
#ifdef SQLITE_HAS_COLUMN_METADATA
const char* pszTableName = sqlite3_column_table_name( hStmt, iCol );
if( oGeomField.GetType() == wkbUnknown && pszTableName != NULL )
{
OGRGeoPackageLayer* poLayer = (OGRGeoPackageLayer*)
m_poDS->GetLayerByName(pszTableName);
if( poLayer != NULL && poLayer->GetLayerDefn()->GetGeomFieldCount() > 0)
{
oGeomField.SetType( poLayer->GetLayerDefn()->GetGeomFieldDefn(0)->GetType() );
}
}
#endif
m_poFeatureDefn->AddGeomFieldDefn(&oGeomField);
iGeomCol = iCol;
continue;
}
}
}
switch( nColType )
{
case SQLITE_INTEGER:
oField.SetType( OFTInteger );
break;
case SQLITE_FLOAT:
oField.SetType( OFTReal );
break;
case SQLITE_BLOB:
oField.SetType( OFTBinary );
break;
default:
/* leave it as OFTString */;
}
if (pszDeclType != NULL)
{
OGRFieldType eFieldType = GPkgFieldToOGR(pszDeclType);
if( (int)eFieldType <= OFTMaxType )
oField.SetType(eFieldType);
}
m_poFeatureDefn->AddFieldDefn( &oField );
panFieldOrdinals[m_poFeatureDefn->GetFieldCount() - 1] = iCol;
}
}