SDTSAttrReader *SDTSTransfer::GetLayerAttrReader( int iEntry )
{
SDTSAttrReader *poAttrReader;
if( iEntry < 0
|| iEntry >= nLayers
|| oCATD.GetEntryType( panLayerCATDEntry[iEntry] ) != SLTAttr )
{
return NULL;
}
poAttrReader = new SDTSAttrReader( &oIREF );
if( !poAttrReader->Open(
oCATD.GetEntryFilePath( panLayerCATDEntry[iEntry] ) ) )
{
delete poAttrReader;
return NULL;
}
else
{
return poAttrReader;
}
}
SDTSAttrReader *SDTSTransfer::GetLayerAttrReader( int iEntry )
{
if( iEntry < 0
|| iEntry >= nLayers
|| oCATD.GetEntryType( panLayerCATDEntry[iEntry] ) != SLTAttr )
{
return nullptr;
}
SDTSAttrReader *poAttrReader = new SDTSAttrReader();
if( !poAttrReader->Open(
oCATD.GetEntryFilePath( panLayerCATDEntry[iEntry] ) ) )
{
panLayerCATDEntry[iEntry] = SLTUnknown; // to prevent further attempt
delete poAttrReader;
return nullptr;
}
return poAttrReader;
}
static void
AddPrimaryAttrToDBFSchema( DBFHandle hDBF, SDTSTransfer *poTransfer,
char ** papszModuleList )
{
for( int iModule = 0;
papszModuleList != NULL && papszModuleList[iModule] != NULL;
iModule++ )
{
SDTSAttrReader *poAttrReader;
/* -------------------------------------------------------------------- */
/* Get a reader on the desired module. */
/* -------------------------------------------------------------------- */
poAttrReader = (SDTSAttrReader *)
poTransfer->GetLayerIndexedReader(
poTransfer->FindLayer( papszModuleList[iModule] ) );
if( poAttrReader == NULL )
{
printf( "Unable to open attribute module %s, skipping.\n" ,
papszModuleList[iModule] );
continue;
}
poAttrReader->Rewind();
/* -------------------------------------------------------------------- */
/* Read the first record so we can clone schema information off */
/* of it. */
/* -------------------------------------------------------------------- */
SDTSAttrRecord *poAttrFeature;
poAttrFeature = (SDTSAttrRecord *) poAttrReader->GetNextFeature();
if( poAttrFeature == NULL )
{
fprintf( stderr,
"Didn't find any meaningful attribute records in %s.\n",
papszModuleList[iModule] );
continue;
}
/* -------------------------------------------------------------------- */
/* Clone schema off the first record. Eventually we need to */
/* get the information out of the DDR record, but it isn't */
/* clear to me how to accomplish that with the SDTS++ API. */
/* */
/* The following approach may fail (dramatically) if some */
/* records do not include all subfields. Furthermore, no */
/* effort is made to make DBF field names unique. The SDTS */
/* attributes often have names much beyond the 14 character dbf */
/* limit which may result in non-unique attributes. */
/* -------------------------------------------------------------------- */
DDFFieldDefn *poFDefn = poAttrFeature->poATTR->GetFieldDefn();
int iSF;
DDFField *poSR = poAttrFeature->poATTR;
for( iSF=0; iSF < poFDefn->GetSubfieldCount(); iSF++ )
{
DDFSubfieldDefn *poSFDefn = poFDefn->GetSubfield( iSF );
int nWidth = poSFDefn->GetWidth();
switch( poSFDefn->GetType() )
{
case DDFString:
if( nWidth == 0 )
{
int nMaxBytes;
const char * pachData = poSR->GetSubfieldData(poSFDefn,
&nMaxBytes);
nWidth = strlen(poSFDefn->ExtractStringData(pachData,
nMaxBytes, NULL ));
}
DBFAddField( hDBF, poSFDefn->GetName(), FTString, nWidth, 0 );
break;
case DDFInt:
if( nWidth == 0 )
nWidth = 9;
DBFAddField( hDBF, poSFDefn->GetName(), FTInteger, nWidth, 0 );
break;
case DDFFloat:
DBFAddField( hDBF, poSFDefn->GetName(), FTDouble, 18, 6 );
break;
default:
fprintf( stderr,
"Dropping attribute `%s' of module `%s'. "
"Type unsupported\n",
poSFDefn->GetName(),
papszModuleList[iModule] );
break;
}
}
if( !poAttrReader->IsIndexed() )
delete poAttrFeature;
} /* next module */
}
static void WriteAttributeDBF( const char * pszShapefile,
SDTSTransfer * poTransfer,
const char * pszMODN )
{
SDTSAttrReader *poAttrReader;
/* -------------------------------------------------------------------- */
/* Fetch a reference to the indexed Pointgon reader. */
/* -------------------------------------------------------------------- */
poAttrReader = (SDTSAttrReader *)
poTransfer->GetLayerIndexedReader( poTransfer->FindLayer( pszMODN ) );
if( poAttrReader == NULL )
{
fprintf( stderr, "Failed to open %s.\n",
poTransfer->GetCATD()->GetModuleFilePath( pszMODN ) );
return;
}
poAttrReader->Rewind();
/* -------------------------------------------------------------------- */
/* Create the database file, and our basic set of attributes. */
/* -------------------------------------------------------------------- */
DBFHandle hDBF;
char szDBFFilename[1024];
sprintf( szDBFFilename, "%s.dbf", pszShapefile );
hDBF = DBFCreate( szDBFFilename );
if( hDBF == NULL )
{
fprintf( stderr, "Unable to create shapefile .dbf for `%s'\n",
pszShapefile );
return;
}
DBFAddField( hDBF, "SDTSRecId", FTInteger, 8, 0 );
/* -------------------------------------------------------------------- */
/* Prepare the schema. */
/* -------------------------------------------------------------------- */
char **papszMODNList = CSLAddString( NULL, pszMODN );
AddPrimaryAttrToDBFSchema( hDBF, poTransfer, papszMODNList );
CSLDestroy( papszMODNList );
/* ==================================================================== */
/* Process all the records in the module. */
/* ==================================================================== */
SDTSAttrRecord *poRecord;
int iRecord = 0;
while((poRecord = (SDTSAttrRecord*)poAttrReader->GetNextFeature()) != NULL)
{
DBFWriteIntegerAttribute( hDBF, iRecord, 0,
poRecord->oModId.nRecord );
WriteAttrRecordToDBF( hDBF, iRecord, poTransfer, poRecord->poATTR );
if( !poAttrReader->IsIndexed() )
delete poRecord;
iRecord++;
}
/* -------------------------------------------------------------------- */
/* Close, and cleanup. */
/* -------------------------------------------------------------------- */
DBFClose( hDBF );
}
OGRSDTSLayer::OGRSDTSLayer( SDTSTransfer * poTransferIn, int iLayerIn,
OGRSDTSDataSource * poDSIn ) :
bPolygonsBuilt(FALSE)
{
poDS = poDSIn;
poTransfer = poTransferIn;
iLayer = iLayerIn;
poReader = poTransfer->GetLayerIndexedReader( iLayer );
/* -------------------------------------------------------------------- */
/* Define the feature. */
/* -------------------------------------------------------------------- */
int iCATDEntry = poTransfer->GetLayerCATDEntry( iLayer );
poFeatureDefn =
new OGRFeatureDefn(poTransfer->GetCATD()->GetEntryModule(iCATDEntry));
SetDescription( poFeatureDefn->GetName() );
poFeatureDefn->Reference();
poFeatureDefn->GetGeomFieldDefn(0)->SetSpatialRef(poDS->GetSpatialRef());
OGRFieldDefn oRecId( "RCID", OFTInteger );
poFeatureDefn->AddFieldDefn( &oRecId );
if( poTransfer->GetLayerType(iLayer) == SLTPoint )
{
poFeatureDefn->SetGeomType( wkbPoint );
}
else if( poTransfer->GetLayerType(iLayer) == SLTLine )
{
poFeatureDefn->SetGeomType( wkbLineString );
oRecId.SetName( "SNID" );
poFeatureDefn->AddFieldDefn( &oRecId );
oRecId.SetName( "ENID" );
poFeatureDefn->AddFieldDefn( &oRecId );
}
else if( poTransfer->GetLayerType(iLayer) == SLTPoly )
{
poFeatureDefn->SetGeomType( wkbPolygon );
}
else if( poTransfer->GetLayerType(iLayer) == SLTAttr )
{
poFeatureDefn->SetGeomType( wkbNone );
}
/* -------------------------------------------------------------------- */
/* Add schema from referenced attribute records. */
/* -------------------------------------------------------------------- */
char **papszATIDRefs = NULL;
if( poTransfer->GetLayerType(iLayer) != SLTAttr )
papszATIDRefs = poReader->ScanModuleReferences();
else
papszATIDRefs = CSLAddString( papszATIDRefs,
poTransfer->GetCATD()->GetEntryModule(iCATDEntry) );
for( int iTable = 0;
papszATIDRefs != NULL && papszATIDRefs[iTable] != NULL;
iTable++ )
{
SDTSAttrReader *poAttrReader;
DDFFieldDefn *poFDefn;
/* -------------------------------------------------------------------- */
/* Get the attribute table reader, and the associated user */
/* attribute field. */
/* -------------------------------------------------------------------- */
int nLayerIdx = poTransfer->FindLayer( papszATIDRefs[iTable] );
if( nLayerIdx < 0 )
continue;
poAttrReader = (SDTSAttrReader *)
poTransfer->GetLayerIndexedReader(nLayerIdx);
if( poAttrReader == NULL )
continue;
poFDefn = poAttrReader->GetModule()->FindFieldDefn( "ATTP" );
if( poFDefn == NULL )
poFDefn = poAttrReader->GetModule()->FindFieldDefn( "ATTS" );
if( poFDefn == NULL )
continue;
/* -------------------------------------------------------------------- */
/* Process each user subfield on the attribute table into an */
/* OGR field definition. */
/* -------------------------------------------------------------------- */
for( int iSF=0; iSF < poFDefn->GetSubfieldCount(); iSF++ )
{
DDFSubfieldDefn *poSFDefn = poFDefn->GetSubfield( iSF );
int nWidth = poSFDefn->GetWidth();
char *pszFieldName;
if( poFeatureDefn->GetFieldIndex( poSFDefn->GetName() ) != -1 )
pszFieldName = CPLStrdup( CPLSPrintf( "%s_%s",
papszATIDRefs[iTable],
poSFDefn->GetName() ) );
else
pszFieldName = CPLStrdup( poSFDefn->GetName() );
switch( poSFDefn->GetType() )
{
case DDFString:
{
OGRFieldDefn oStrField( pszFieldName, OFTString );
if( nWidth != 0 )
oStrField.SetWidth( nWidth );
poFeatureDefn->AddFieldDefn( &oStrField );
}
break;
case DDFInt:
{
OGRFieldDefn oIntField( pszFieldName, OFTInteger );
if( nWidth != 0 )
oIntField.SetWidth( nWidth );
poFeatureDefn->AddFieldDefn( &oIntField );
}
break;
case DDFFloat:
{
OGRFieldDefn oRealField( pszFieldName, OFTReal );
// We don't have a precision in DDF files, so we never even
// use the width. Otherwise with a precision of zero the
// result would look like an integer.
poFeatureDefn->AddFieldDefn( &oRealField );
}
break;
default:
break;
}
CPLFree( pszFieldName );
} /* next iSF (subfield) */
} /* next iTable */
CSLDestroy( papszATIDRefs );
}
