int SHPAPI_CALL SHPWriteTree(SHPTree *tree, const char *filename )
{
SAHooks sHooks;
SASetupDefaultHooks( &sHooks );
return SHPWriteTreeLL(tree, filename, &sHooks);
}
SHPTreeDiskHandle SHPOpenDiskTree( const char* pszQIXFilename,
SAHooks *psHooks )
{
SHPTreeDiskHandle hDiskTree;
hDiskTree = (SHPTreeDiskHandle) calloc(sizeof(struct SHPDiskTreeInfo),1);
if (psHooks == NULL)
SASetupDefaultHooks( &(hDiskTree->sHooks) );
else
memcpy( &(hDiskTree->sHooks), psHooks, sizeof(SAHooks) );
hDiskTree->fpQIX = hDiskTree->sHooks.FOpen(pszQIXFilename, "rb");
if (hDiskTree->fpQIX == NULL)
{
free(hDiskTree);
return NULL;
}
return hDiskTree;
}
SBNSearchHandle SBNOpenDiskTree( const char* pszSBNFilename,
SAHooks *psHooks )
{
int i;
SBNSearchHandle hSBN;
uchar abyHeader[108];
int nShapeCount;
int nMaxDepth;
int nMaxNodes;
int nNodeDescSize;
int nNodeDescCount;
uchar* pabyData = NULL;
SBNNodeDescriptor* pasNodeDescriptor = NULL;
uchar abyBinHeader[8];
int nCurNode;
int nNextNonEmptyNode;
int nExpectedBinId;
int bBigEndian;
/* -------------------------------------------------------------------- */
/* Establish the byte order on this machine. */
/* -------------------------------------------------------------------- */
i = 1;
if( *((unsigned char *) &i) == 1 )
bBigEndian = FALSE;
else
bBigEndian = TRUE;
/* -------------------------------------------------------------------- */
/* Initialize the handle structure. */
/* -------------------------------------------------------------------- */
hSBN = (SBNSearchHandle)
calloc(sizeof(struct SBNSearchInfo),1);
if (psHooks == NULL)
SASetupDefaultHooks( &(hSBN->sHooks) );
else
memcpy( &(hSBN->sHooks), psHooks, sizeof(SAHooks) );
hSBN->fpSBN = hSBN->sHooks.FOpen(pszSBNFilename, "rb");
if (hSBN->fpSBN == NULL)
{
free(hSBN);
return NULL;
}
/* -------------------------------------------------------------------- */
/* Check file header signature. */
/* -------------------------------------------------------------------- */
if (hSBN->sHooks.FRead(abyHeader, 108, 1, hSBN->fpSBN) != 1 ||
abyHeader[0] != 0 ||
abyHeader[1] != 0 ||
abyHeader[2] != 0x27 ||
(abyHeader[3] != 0x0A && abyHeader[3] != 0x0D) ||
abyHeader[4] != 0xFF ||
abyHeader[5] != 0xFF ||
abyHeader[6] != 0xFE ||
abyHeader[7] != 0x70)
{
hSBN->sHooks.Error( ".sbn file is unreadable, or corrupt." );
SBNCloseDiskTree(hSBN);
return NULL;
}
/* -------------------------------------------------------------------- */
/* Read shapes bounding box. */
/* -------------------------------------------------------------------- */
memcpy(&hSBN->dfMinX, abyHeader + 32, 8);
memcpy(&hSBN->dfMinY, abyHeader + 40, 8);
memcpy(&hSBN->dfMaxX, abyHeader + 48, 8);
memcpy(&hSBN->dfMaxY, abyHeader + 56, 8);
if( !bBigEndian )
{
SwapWord(8, &hSBN->dfMinX);
SwapWord(8, &hSBN->dfMinY);
SwapWord(8, &hSBN->dfMaxX);
SwapWord(8, &hSBN->dfMaxY);
}
if( hSBN->dfMinX > hSBN->dfMaxX ||
hSBN->dfMinY > hSBN->dfMaxY )
{
hSBN->sHooks.Error( "Invalid extent in .sbn file." );
SBNCloseDiskTree(hSBN);
return NULL;
}
/* -------------------------------------------------------------------- */
/* Read and check number of shapes. */
/* -------------------------------------------------------------------- */
nShapeCount = READ_MSB_INT(abyHeader + 28);
hSBN->nShapeCount = nShapeCount;
if (nShapeCount < 0 || nShapeCount > 256000000 )
{
char szErrorMsg[64];
sprintf(szErrorMsg, "Invalid shape count in .sbn : %d", nShapeCount );
hSBN->sHooks.Error( szErrorMsg );
SBNCloseDiskTree(hSBN);
return NULL;
}
/* -------------------------------------------------------------------- */
/* Compute tree depth. */
/* It is computed such as in average there are not more than 8 */
/* shapes per node. With a minimum depth of 2, and a maximum of 15 */
/* -------------------------------------------------------------------- */
nMaxDepth = 2;
while( nMaxDepth < 15 && nShapeCount > ((1 << nMaxDepth) - 1) * 8 )
nMaxDepth ++;
hSBN->nMaxDepth = nMaxDepth;
nMaxNodes = (1 << nMaxDepth) - 1;
/* -------------------------------------------------------------------- */
/* Check that the first bin id is 1. */
/* -------------------------------------------------------------------- */
if( READ_MSB_INT(abyHeader + 100) != 1 )
{
hSBN->sHooks.Error( "Unexpected bin id" );
SBNCloseDiskTree(hSBN);
return NULL;
}
/* -------------------------------------------------------------------- */
/* Read and check number of node descriptors to be read. */
/* There are at most (2^nMaxDepth) - 1, but all are not necessary */
/* described. Non described nodes are empty. */
/* -------------------------------------------------------------------- */
nNodeDescSize = READ_MSB_INT(abyHeader + 104);
nNodeDescSize *= 2; /* 16-bit words */
/* each bin descriptor is made of 2 ints */
nNodeDescCount = nNodeDescSize / 8;
if ((nNodeDescSize % 8) != 0 ||
nNodeDescCount < 0 || nNodeDescCount > nMaxNodes )
{
char szErrorMsg[64];
sprintf(szErrorMsg,
"Invalid node descriptor size in .sbn : %d", nNodeDescSize );
hSBN->sHooks.Error( szErrorMsg );
SBNCloseDiskTree(hSBN);
return NULL;
}
pabyData = (uchar*) malloc( nNodeDescSize );
pasNodeDescriptor = (SBNNodeDescriptor*)
calloc ( nMaxNodes, sizeof(SBNNodeDescriptor) );
if (pabyData == NULL || pasNodeDescriptor == NULL)
{
free(pabyData);
free(pasNodeDescriptor);
hSBN->sHooks.Error( "Out of memory error" );
SBNCloseDiskTree(hSBN);
return NULL;
}
/* -------------------------------------------------------------------- */
/* Read node descriptors. */
/* -------------------------------------------------------------------- */
if (hSBN->sHooks.FRead(pabyData, nNodeDescSize, 1,
hSBN->fpSBN) != 1)
{
free(pabyData);
free(pasNodeDescriptor);
hSBN->sHooks.Error( "Cannot read node descriptors" );
SBNCloseDiskTree(hSBN);
return NULL;
}
hSBN->pasNodeDescriptor = pasNodeDescriptor;
for(i = 0; i < nNodeDescCount; i++)
{
/* -------------------------------------------------------------------- */
/* Each node descriptor contains the index of the first bin that */
/* described it, and the number of shapes in this first bin and */
/* the following bins (in the relevant case). */
/* -------------------------------------------------------------------- */
int nBinStart = READ_MSB_INT(pabyData + 8 * i);
int nNodeShapeCount = READ_MSB_INT(pabyData + 8 * i + 4);
pasNodeDescriptor[i].nBinStart = nBinStart > 0 ? nBinStart : 0;
pasNodeDescriptor[i].nShapeCount = nNodeShapeCount;
if ((nBinStart > 0 && nNodeShapeCount == 0) ||
nNodeShapeCount < 0 || nNodeShapeCount > nShapeCount)
{
hSBN->sHooks.Error( "Inconsistant shape count in bin" );
SBNCloseDiskTree(hSBN);
return NULL;
}
}
free(pabyData);
pabyData = NULL;
/* Locate first non-empty node */
nCurNode = 0;
while(nCurNode < nMaxNodes && pasNodeDescriptor[nCurNode].nBinStart <= 0)
nCurNode ++;
if( nCurNode >= nMaxNodes)
{
hSBN->sHooks.Error( "All nodes are empty" );
SBNCloseDiskTree(hSBN);
return NULL;
}
pasNodeDescriptor[nCurNode].nBinOffset =
hSBN->sHooks.FTell(hSBN->fpSBN);
/* Compute the index of the next non empty node. */
nNextNonEmptyNode = nCurNode + 1;
while(nNextNonEmptyNode < nMaxNodes &&
pasNodeDescriptor[nNextNonEmptyNode].nBinStart <= 0)
nNextNonEmptyNode ++;
nExpectedBinId = 1;
/* -------------------------------------------------------------------- */
/* Traverse bins to compute the offset of the first bin of each */
/* node. */
/* Note: we could use the .sbx file to compute the offsets instead.*/
/* -------------------------------------------------------------------- */
while( hSBN->sHooks.FRead(abyBinHeader, 8, 1,
hSBN->fpSBN) == 1 )
{
int nBinId;
int nBinSize;
nExpectedBinId ++;
nBinId = READ_MSB_INT(abyBinHeader);
nBinSize = READ_MSB_INT(abyBinHeader + 4);
nBinSize *= 2; /* 16-bit words */
if( nBinId != nExpectedBinId )
{
hSBN->sHooks.Error( "Unexpected bin id" );
SBNCloseDiskTree(hSBN);
return NULL;
}
/* Bins are always limited to 100 features */
/* If there are more, then they are located in continuous bins */
if( (nBinSize % 8) != 0 || nBinSize <= 0 || nBinSize > 100 * 8)
{
hSBN->sHooks.Error( "Unexpected bin size" );
SBNCloseDiskTree(hSBN);
return NULL;
}
if( nNextNonEmptyNode < nMaxNodes &&
nBinId == pasNodeDescriptor[nNextNonEmptyNode].nBinStart )
{
nCurNode = nNextNonEmptyNode;
pasNodeDescriptor[nCurNode].nBinOffset =
hSBN->sHooks.FTell(hSBN->fpSBN) - 8;
/* Compute the index of the next non empty node. */
nNextNonEmptyNode = nCurNode + 1;
while(nNextNonEmptyNode < nMaxNodes &&
pasNodeDescriptor[nNextNonEmptyNode].nBinStart <= 0)
nNextNonEmptyNode ++;
}
pasNodeDescriptor[nCurNode].nBinCount ++;
/* Skip shape description */
hSBN->sHooks.FSeek(hSBN->fpSBN, nBinSize, SEEK_CUR);
}
return hSBN;
}
int SHPWriteTreeLL(SHPTree *tree, const char *filename, SAHooks* psHooks )
{
char signature[4] = "SQT";
int i;
char abyBuf[32];
SAFile fp;
SAHooks sHooks;
if (psHooks == NULL)
{
SASetupDefaultHooks( &sHooks );
psHooks = &sHooks;
}
/* -------------------------------------------------------------------- */
/* Open the output file. */
/* -------------------------------------------------------------------- */
fp = psHooks->FOpen(filename, "wb");
if( fp == NULL )
{
return FALSE;
}
/* -------------------------------------------------------------------- */
/* Establish the byte order on this machine. */
/* -------------------------------------------------------------------- */
i = 1;
if( *((unsigned char *) &i) == 1 )
bBigEndian = FALSE;
else
bBigEndian = TRUE;
/* -------------------------------------------------------------------- */
/* Write the header. */
/* -------------------------------------------------------------------- */
memcpy( abyBuf+0, signature, 3 );
if( bBigEndian )
abyBuf[3] = 2; /* New MSB */
else
abyBuf[3] = 1; /* New LSB */
abyBuf[4] = 1; /* version */
abyBuf[5] = 0; /* next 3 reserved */
abyBuf[6] = 0;
abyBuf[7] = 0;
psHooks->FWrite( abyBuf, 8, 1, fp );
psHooks->FWrite( &(tree->nTotalCount), 4, 1, fp );
/* write maxdepth */
psHooks->FWrite( &(tree->nMaxDepth), 4, 1, fp );
/* -------------------------------------------------------------------- */
/* Write all the nodes "in order". */
/* -------------------------------------------------------------------- */
SHPWriteTreeNode( fp, tree->psRoot, psHooks );
psHooks->FClose( fp );
return TRUE;
}
SHPHandle
OpenShapeFile( const char *fn )
{
SHPHandle file;
char *fs = NULL, *dn = NULL;
#ifdef HAVE_SAHOOKS
SAHooks sHooks;
SASetupDefaultHooks( &sHooks );
sHooks.Error = CustomErrors;
#else
// Using ancient version of shapelib without SAHooks or SHPOpenLL.
// For this case live with the misleading "Unable to open" error
// messages.
// int sHooks;
#define SHPOpenLL( a, b, c ) SHPOpen( a, b )
#endif
//*** search build tree ***
if ( plInBuildTree() == 1 )
{
plGetName( SOURCE_DIR, "data", fn, &fs );
if ( ( file = SHPOpenLL( fs, "rb", &sHooks ) ) != NULL )
goto done;
}
//*** search PLPLOT_LIB_ENV = $(PLPLOT_LIB) ***
#if defined ( PLPLOT_LIB_ENV )
if ( ( dn = getenv( PLPLOT_LIB_ENV ) ) != NULL )
{
plGetName( dn, "", fn, &fs );
if ( ( file = SHPOpenLL( fs, "rb", &sHooks ) ) != NULL )
goto done;
fprintf( stderr, PLPLOT_LIB_ENV "=\"%s\"\n", dn ); // what IS set?
}
#endif // PLPLOT_LIB_ENV
//*** search current directory ***
if ( ( file = SHPOpenLL( fn, "rb", &sHooks ) ) != NULL )
{
pldebug( "OpenShapeFile", "Found file %s in current directory.\n", fn );
free_mem( fs );
return ( file );
}
//*** search PLPLOT_HOME_ENV/lib = $(PLPLOT_HOME)/lib ***
#if defined ( PLPLOT_HOME_ENV )
if ( ( dn = getenv( PLPLOT_HOME_ENV ) ) != NULL )
{
plGetName( dn, "lib", fn, &fs );
if ( ( file = SHPOpenLL( fs, "rb", &sHooks ) ) != NULL )
goto done;
fprintf( stderr, PLPLOT_HOME_ENV "=\"%s\"\n", dn ); // what IS set?
}
#endif // PLPLOT_HOME_ENV/lib
//*** search installed location ***
#if defined ( DATA_DIR )
plGetName( DATA_DIR, "", fn, &fs );
if ( ( file = SHPOpenLL( fs, "rb", &sHooks ) ) != NULL )
goto done;
#endif // DATA_DIR
//*** search hardwired location ***
#ifdef PLLIBDEV
plGetName( PLLIBDEV, "", fn, &fs );
if ( ( file = SHPOpenLL( fs, "rb", &sHooks ) ) != NULL )
goto done;
#endif // PLLIBDEV
//*** not found, give up ***
pldebug( "OpenShapeFile", "File %s not found.\n", fn );
free_mem( fs );
return NULL;
done:
pldebug( "OpenShapeFile", "SHPOpen successfully opened two files with basename %s\n", fs );
free_mem( fs );
return ( file );
}
