/****************************************************************************** * $Id: ogrpgeotablelayer.cpp 10646 2007-01-18 02:38:10Z warmerdam $ * * Project:  OpenGIS Simple Features Reference Implementation * Purpose:  Implements OGRPGeoTableLayer class, access to an existing table. * Author:   Frank Warmerdam, warmerdam@pobox.com * ****************************************************************************** * Copyright (c) 2005, Frank Warmerdam * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included * in all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER * DEALINGS IN THE SOFTWARE. ****************************************************************************/#include "cpl_conv.h"#include "ogr_pgeo.h"CPL_CVSID("$Id: ogrpgeotablelayer.cpp 10646 2007-01-18 02:38:10Z warmerdam $");/************************************************************************//*                          OGRPGeoTableLayer()                         *//************************************************************************/OGRPGeoTableLayer::OGRPGeoTableLayer( OGRPGeoDataSource *poDSIn ){    poDS = poDSIn;    pszQuery = NULL;    bUpdateAccess = TRUE;    iNextShapeId = 0;    nSRSId = -1;    poFeatureDefn = NULL;    memset( &sExtent, 0, sizeof(sExtent) );}/************************************************************************//*                          ~OGRPGeoTableLayer()                          *//************************************************************************/OGRPGeoTableLayer::~OGRPGeoTableLayer(){    CPLFree( pszQuery );    ClearStatement();}/************************************************************************//*                             Initialize()                             *//************************************************************************/CPLErr OGRPGeoTableLayer::Initialize( const char *pszTableName,                                       const char *pszGeomCol,                                      int nShapeType,                                       double dfExtentLeft,                                      double dfExtentRight,                                      double dfExtentBottom,                                      double dfExtentTop,                                      int nSRID,                                      int bHasZ ){    CPLODBCSession *poSession = poDS->GetSession();    CPLFree( pszGeomColumn );    if( pszGeomCol == NULL )        pszGeomColumn = NULL;    else        pszGeomColumn = CPLStrdup( pszGeomCol );    CPLFree( pszFIDColumn );    pszFIDColumn = NULL;    sExtent.MinX = dfExtentLeft;    sExtent.MaxX = dfExtentRight;    sExtent.MinY = dfExtentBottom;    sExtent.MaxY = dfExtentTop;    LookupSRID( nSRID );/* -------------------------------------------------------------------- *//*      Setup geometry type.                                            *//* -------------------------------------------------------------------- */    OGRwkbGeometryType  eOGRType;    switch( nShapeType )    {        case SHPT_POINT:        case SHPT_POINTM:        case SHPT_POINTZ:            eOGRType = wkbPoint;            break;        case SHPT_ARC:        case SHPT_ARCZ:        case SHPT_ARCM:            eOGRType = wkbLineString;            break;                    case SHPT_MULTIPOINT:        case SHPT_MULTIPOINTZ:        case SHPT_MULTIPOINTM:            eOGRType = wkbMultiPoint;            break;        default:            eOGRType = wkbUnknown;            break;    }    if( eOGRType != wkbUnknown && bHasZ )        eOGRType = (OGRwkbGeometryType)(((int) eOGRType) | wkb25DBit);/* -------------------------------------------------------------------- *//*      Do we have a simple primary key?                                *//* -------------------------------------------------------------------- */    CPLODBCStatement oGetKey( poSession );        if( oGetKey.GetPrimaryKeys( pszTableName ) && oGetKey.Fetch() )    {        pszFIDColumn = CPLStrdup(oGetKey.GetColData( 3 ));                if( oGetKey.Fetch() ) // more than one field in key!         {            CPLFree( pszFIDColumn );            pszFIDColumn = NULL;            CPLDebug( "PGeo", "%s: Compound primary key, ignoring.",                      pszTableName );        }        else            CPLDebug( "PGeo",                       "%s: Got primary key %s.",                      pszTableName, pszFIDColumn );    }    else        CPLDebug( "PGeo", "%s: no primary key", pszTableName );/* -------------------------------------------------------------------- *//*      Get the column definitions for this table.                      *//* -------------------------------------------------------------------- */    CPLODBCStatement oGetCol( poSession );    CPLErr eErr;    if( !oGetCol.GetColumns( pszTableName ) )    {        CPLError( CE_Failure, CPLE_AppDefined,                   "GetColumns() failed on %s.\n%s",                  pszTableName, poSession->GetLastError() );        return CE_Failure;    }    eErr = BuildFeatureDefn( pszTableName, &oGetCol );    if( eErr != CE_None )        return eErr;    if( poFeatureDefn->GetFieldCount() == 0 )    {        CPLError( CE_Failure, CPLE_AppDefined,                   "No column definitions found for table '%s', layer not usable.",                   pszTableName );        return CE_Failure;    }/* -------------------------------------------------------------------- *//*      Set geometry type.                                              *//*                                                                      *//*      ... per reports from Craig Miller, it seems we cannot really    *//*      trust the ShapeType value.  At the very least "line" tables     *//*      sometimes have multilinestrings.  So for now we just always     *//*      return wkbUnknown.                                              *//* -------------------------------------------------------------------- */#ifdef notdef    poFeatureDefn->SetGeomType( eOGRType );#endif    return CE_None;}/************************************************************************//*                           ClearStatement()                           *//************************************************************************/void OGRPGeoTableLayer::ClearStatement(){    if( poStmt != NULL )    {        delete poStmt;        poStmt = NULL;    }}/************************************************************************//*                            GetStatement()                            *//************************************************************************/CPLODBCStatement *OGRPGeoTableLayer::GetStatement(){    if( poStmt == NULL )        ResetStatement();    return poStmt;}/************************************************************************//*                           ResetStatement()                           *//************************************************************************/OGRErr OGRPGeoTableLayer::ResetStatement(){    ClearStatement();    iNextShapeId = 0;    poStmt = new CPLODBCStatement( poDS->GetSession() );    poStmt->Append( "SELECT * FROM " );    poStmt->Append( poFeatureDefn->GetName() );    if( pszQuery != NULL )        poStmt->Appendf( " WHERE %s", pszQuery );    if( poStmt->ExecuteSQL() )        return OGRERR_NONE;    else    {        delete poStmt;        poStmt = NULL;        return OGRERR_FAILURE;    }}/************************************************************************//*                            ResetReading()                            *//************************************************************************/void OGRPGeoTableLayer::ResetReading(){    ClearStatement();    OGRPGeoLayer::ResetReading();}/************************************************************************//*                             GetFeature()                             *//************************************************************************/OGRFeature *OGRPGeoTableLayer::GetFeature( long nFeatureId ){    if( pszFIDColumn == NULL )        return OGRPGeoLayer::GetFeature( nFeatureId );    ClearStatement();    iNextShapeId = nFeatureId;    poStmt = new CPLODBCStatement( poDS->GetSession() );    poStmt->Append( "SELECT * FROM " );    poStmt->Append( poFeatureDefn->GetName() );    poStmt->Appendf( " WHERE %s = %d", pszFIDColumn, nFeatureId );    if( !poStmt->ExecuteSQL() )    {        delete poStmt;        poStmt = NULL;        return NULL;    }    return GetNextRawFeature();}/************************************************************************//*                         SetAttributeFilter()                         *//************************************************************************/OGRErr OGRPGeoTableLayer::SetAttributeFilter( const char *pszQuery ){    if( (pszQuery == NULL && this->pszQuery == NULL)        || (pszQuery != NULL && this->pszQuery != NULL             && EQUAL(pszQuery,this->pszQuery)) )        return OGRERR_NONE;    CPLFree( this->pszQuery );    this->pszQuery = CPLStrdup( pszQuery );    ClearStatement();    return OGRERR_NONE;}/************************************************************************//*                           TestCapability()                           *//************************************************************************/int OGRPGeoTableLayer::TestCapability( const char * pszCap ){    if( EQUAL(pszCap,OLCSequentialWrite)              || EQUAL(pszCap,OLCRandomWrite) )        return bUpdateAccess;    else if( EQUAL(pszCap,OLCCreateField) )        return bUpdateAccess;    else         return OGRPGeoLayer::TestCapability( pszCap );}/************************************************************************//*                          GetFeatureCount()                           *//*                                                                      *//*      If a spatial filter is in effect, we turn control over to       *//*      the generic counter.  Otherwise we return the total count.      *//*      Eventually we should consider implementing a more efficient     *//*      way of counting features matching a spatial query.              *//************************************************************************/int OGRPGeoTableLayer::GetFeatureCount( int bForce ){    if( m_poFilterGeom != NULL )        return OGRPGeoLayer::GetFeatureCount( bForce );    CPLODBCStatement oStmt( poDS->GetSession() );    oStmt.Append( "SELECT COUNT(*) FROM " );    oStmt.Append( poFeatureDefn->GetName() );    if( pszQuery != NULL )        oStmt.Appendf( " WHERE %s", pszQuery );    if( !oStmt.ExecuteSQL() || !oStmt.Fetch() )    {        CPLError( CE_Failure, CPLE_AppDefined,                   "GetFeatureCount() failed on query %s.\n%s",                  oStmt.GetCommand(), poDS->GetSession()->GetLastError() );        return -1;    }    return atoi(oStmt.GetColData(0));}/************************************************************************//*                             GetExtent()                              *//************************************************************************/OGRErr OGRPGeoTableLayer::GetExtent( OGREnvelope *psExtent, int bForce ){    *psExtent = sExtent;    return OGRERR_NONE;}