/****************************************************************************** * $Id: ogrshape.h 4b7a9ef829d5432b1271c1c3de613c4db80c3205 2019-03-30 11:16:26 +0100 Even Rouault $ * * Project: OpenGIS Simple Features Reference Implementation * Purpose: Private definitions within the Shapefile driver to implement * integration with OGR. * Author: Frank Warmerdam, warmerdam@pobox.com * ****************************************************************************** * Copyright (c) 1999, Les Technologies SoftMap Inc. * Copyright (c) 2008-2013, Even Rouault * * 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. ****************************************************************************/ #ifndef OGRSHAPE_H_INCLUDED #define OGRSHAPE_H_INCLUDED #ifdef RENAME_INTERNAL_SHAPELIB_SYMBOLS #include "gdal_shapelib_symbol_rename.h" #endif #include "ogrsf_frmts.h" #include "shapefil.h" #include "shp_vsi.h" #include "ogrlayerpool.h" #include /* Was limited to 255 until OGR 1.10, but 254 seems to be a more */ /* conventional limit (http://en.wikipedia.org/wiki/Shapefile, */ /* http://www.clicketyclick.dk/databases/xbase/format/data_types.html, */ /* #5052 ) */ #define OGR_DBF_MAX_FIELD_WIDTH 254 /* ==================================================================== */ /* Functions from Shape2ogr.cpp. */ /* ==================================================================== */ OGRFeature *SHPReadOGRFeature( SHPHandle hSHP, DBFHandle hDBF, OGRFeatureDefn * poDefn, int iShape, SHPObject *psShape, const char *pszSHPEncoding ); OGRGeometry *SHPReadOGRObject( SHPHandle hSHP, int iShape, SHPObject *psShape ); OGRFeatureDefn *SHPReadOGRFeatureDefn( const char * pszName, SHPHandle hSHP, DBFHandle hDBF, const char *pszSHPEncoding, int bAdjustType ); OGRErr SHPWriteOGRFeature( SHPHandle hSHP, DBFHandle hDBF, OGRFeatureDefn *poFeatureDefn, OGRFeature *poFeature, const char *pszSHPEncoding, bool* pbTruncationWarningEmitted, bool bRewind ); /************************************************************************/ /* OGRShapeGeomFieldDefn */ /************************************************************************/ class OGRShapeGeomFieldDefn final: public OGRGeomFieldDefn { CPL_DISALLOW_COPY_ASSIGN(OGRShapeGeomFieldDefn) char* pszFullName = nullptr; mutable bool bSRSSet = false; mutable CPLString osPrjFile{}; public: OGRShapeGeomFieldDefn( const char* pszFullNameIn, OGRwkbGeometryType eType, int bSRSSetIn, OGRSpatialReference *poSRSIn) : OGRGeomFieldDefn("", eType), pszFullName(CPLStrdup(pszFullNameIn)), bSRSSet(CPL_TO_BOOL(bSRSSetIn)) { poSRS = poSRSIn; } virtual ~OGRShapeGeomFieldDefn() { CPLFree(pszFullName); } virtual OGRSpatialReference* GetSpatialRef() const override; const CPLString& GetPrjFilename() const { return osPrjFile; } }; /************************************************************************/ /* OGRShapeLayer */ /************************************************************************/ class OGRShapeDataSource; class OGRShapeLayer final: public OGRAbstractProxiedLayer { CPL_DISALLOW_COPY_ASSIGN(OGRShapeLayer) OGRShapeDataSource *poDS; OGRFeatureDefn *poFeatureDefn; int iNextShapeId; int nTotalShapeCount; char *pszFullName; SHPHandle hSHP; DBFHandle hDBF; bool bUpdateAccess; OGRwkbGeometryType eRequestedGeomType; int ResetGeomType( int nNewType ); bool ScanIndices(); GIntBig *panMatchingFIDs; int iMatchingFID; void ClearMatchingFIDs(); OGRGeometry *m_poFilterGeomLastValid; int nSpatialFIDCount; int *panSpatialFIDs; void ClearSpatialFIDs(); bool bHeaderDirty; bool bSHPNeedsRepack; bool bCheckedForQIX; SHPTreeDiskHandle hQIX; bool CheckForQIX(); bool bCheckedForSBN; SBNSearchHandle hSBN; bool CheckForSBN(); bool bSbnSbxDeleted; CPLString ConvertCodePage( const char * ); CPLString osEncoding{}; bool bTruncationWarningEmitted; bool bHSHPWasNonNULL; // Must try to reopen a .shp? bool bHDBFWasNonNULL; // Must try to reopen a .dbf // Current state of opening of file descriptor to .shp and .dbf. typedef enum { FD_OPENED, FD_CLOSED, FD_CANNOT_REOPEN } FileDescriptorState; FileDescriptorState eFileDescriptorsState; bool TouchLayer(); bool ReopenFileDescriptors(); bool bResizeAtClose; void TruncateDBF(); bool bCreateSpatialIndexAtClose; bool bRewindOnWrite; bool m_bAutoRepack; typedef enum { YES, NO, MAYBE } NormandyState; /* French joke. "Peut'et' ben que oui, peut'et' ben que non." Sorry :-) */ NormandyState m_eNeedRepack; protected: virtual void CloseUnderlyingLayer() override; // WARNING: Each of the below public methods should start with a call to // TouchLayer() and test its return value, so as to make sure that // the layer is properly re-opened if necessary. public: OGRErr CreateSpatialIndex( int nMaxDepth ); OGRErr DropSpatialIndex(); OGRErr Repack(); OGRErr RecomputeExtent(); OGRErr ResizeDBF(); void SetResizeAtClose( bool bFlag ) { bResizeAtClose = bFlag; } const char *GetFullName() { return pszFullName; } OGRFeature * FetchShape( int iShapeId ); int GetFeatureCountWithSpatialFilterOnly(); public: OGRShapeLayer( OGRShapeDataSource* poDSIn, const char * pszName, SHPHandle hSHP, DBFHandle hDBF, OGRSpatialReference *poSRS, bool bSRSSet, bool bUpdate, OGRwkbGeometryType eReqType, char ** papszCreateOptions = nullptr); virtual ~OGRShapeLayer(); void ResetReading() override; OGRFeature * GetNextFeature() override; virtual OGRErr SetNextByIndex( GIntBig nIndex ) override; OGRFeature *GetFeature( GIntBig nFeatureId ) override; OGRErr ISetFeature( OGRFeature *poFeature ) override; OGRErr DeleteFeature( GIntBig nFID ) override; OGRErr ICreateFeature( OGRFeature *poFeature ) override; OGRErr SyncToDisk() override; OGRFeatureDefn * GetLayerDefn() override { return poFeatureDefn; } GIntBig GetFeatureCount( int ) override; OGRErr GetExtent( OGREnvelope *psExtent, int bForce ) override; virtual OGRErr GetExtent( int iGeomField, OGREnvelope *psExtent, int bForce ) override { return OGRLayer::GetExtent(iGeomField, psExtent, bForce); } virtual OGRErr CreateField( OGRFieldDefn *poField, int bApproxOK = TRUE ) override; virtual OGRErr DeleteField( int iField ) override; virtual OGRErr ReorderFields( int* panMap ) override; virtual OGRErr AlterFieldDefn( int iField, OGRFieldDefn* poNewFieldDefn, int nFlags ) override; virtual int TestCapability( const char * ) override; virtual void SetSpatialFilter( OGRGeometry * ) override; virtual void SetSpatialFilter( int iGeomField, OGRGeometry *poGeom ) override { OGRLayer::SetSpatialFilter(iGeomField, poGeom); } virtual OGRErr SetAttributeFilter( const char * ) override; void AddToFileList( CPLStringList& oFileList ); void CreateSpatialIndexAtClose( int bFlag ) { bCreateSpatialIndexAtClose = CPL_TO_BOOL(bFlag); } void SetModificationDate( const char* pszStr ); void SetAutoRepack(bool b) { m_bAutoRepack = b; } void SetWriteDBFEOFChar(bool b); }; /************************************************************************/ /* OGRShapeDataSource */ /************************************************************************/ class OGRShapeDataSource final: public OGRDataSource { OGRShapeLayer **papoLayers; int nLayers; char *pszName; bool bDSUpdate; bool bSingleFileDataSource; OGRLayerPool *poPool; void AddLayer( OGRShapeLayer* poLayer ); std::vector oVectorLayerName{}; bool b2GBLimit; char **papszOpenOptions; CPL_DISALLOW_COPY_ASSIGN(OGRShapeDataSource) public: OGRShapeDataSource(); virtual ~OGRShapeDataSource(); OGRLayerPool *GetPool() { return poPool; } bool Open( GDALOpenInfo* poOpenInfo, bool bTestOpen, bool bForceSingleFileDataSource = false ); bool OpenFile( const char *, bool bUpdate ); virtual const char *GetName() override { return pszName; } virtual int GetLayerCount() override; virtual OGRLayer *GetLayer( int ) override; virtual OGRLayer *GetLayerByName( const char * ) override; virtual OGRLayer *ICreateLayer( const char *, OGRSpatialReference * = nullptr, OGRwkbGeometryType = wkbUnknown, char ** = nullptr ) override; virtual OGRLayer *ExecuteSQL( const char *pszStatement, OGRGeometry *poSpatialFilter, const char *pszDialect ) override; virtual int TestCapability( const char * ) override; virtual OGRErr DeleteLayer( int iLayer ) override; virtual char **GetFileList() override; void SetLastUsedLayer( OGRShapeLayer* poLayer ); void UnchainLayer( OGRShapeLayer* poLayer ); SHPHandle DS_SHPOpen( const char * pszShapeFile, const char * pszAccess ); DBFHandle DS_DBFOpen( const char * pszDBFFile, const char * pszAccess ); char **GetOpenOptions() { return papszOpenOptions; } static const char* const* GetExtensionsForDeletion(); }; #endif /* ndef OGRSHAPE_H_INCLUDED */