OGRErr OGRPGDumpLayer::CreateFeatureViaCopy( OGRFeature *poFeature )
{
int i;
CPLString osCommand;
/* First process geometry */
for( i = 0; i < poFeature->GetGeomFieldCount(); i++ )
{
OGRGeometry *poGeometry = poFeature->GetGeomFieldRef(i);
char *pszGeom = NULL;
if ( NULL != poGeometry /* && (bHasWkb || bHasPostGISGeometry || bHasPostGISGeography) */)
{
OGRPGDumpGeomFieldDefn* poGFldDefn =
(OGRPGDumpGeomFieldDefn*) poFeature->GetGeomFieldDefnRef(i);
poGeometry->closeRings();
poGeometry->setCoordinateDimension( poGFldDefn->nCoordDimension );
//CheckGeomTypeCompatibility(poGeometry);
/*if (bHasWkb)
pszGeom = GeometryToBYTEA( poGeometry );
else*/
pszGeom = OGRGeometryToHexEWKB( poGeometry, poGFldDefn->nSRSId );
}
if (osCommand.size() > 0)
osCommand += "\t";
if ( pszGeom )
{
osCommand += pszGeom,
CPLFree( pszGeom );
}
else
{
osCommand += "\\N";
}
}
/* Next process the field id column */
int nFIDIndex = -1;
if( bFIDColumnInCopyFields )
{
if (osCommand.size() > 0)
osCommand += "\t";
nFIDIndex = poFeatureDefn->GetFieldIndex( pszFIDColumn );
/* Set the FID */
if( poFeature->GetFID() != OGRNullFID )
{
osCommand += CPLString().Printf("%ld ", poFeature->GetFID());
}
else
{
osCommand += "\\N" ;
}
}
/* Now process the remaining fields */
int nFieldCount = poFeatureDefn->GetFieldCount();
int bAddTab = osCommand.size() > 0;
for( i = 0; i < nFieldCount; i++ )
{
if (i == nFIDIndex)
continue;
const char *pszStrValue = poFeature->GetFieldAsString(i);
char *pszNeedToFree = NULL;
if (bAddTab)
osCommand += "\t";
bAddTab = TRUE;
if( !poFeature->IsFieldSet( i ) )
{
osCommand += "\\N" ;
continue;
}
int nOGRFieldType = poFeatureDefn->GetFieldDefn(i)->GetType();
// We need special formatting for integer list values.
if( nOGRFieldType == OFTIntegerList )
{
int nCount, nOff = 0, j;
const int *panItems = poFeature->GetFieldAsIntegerList(i,&nCount);
pszNeedToFree = (char *) CPLMalloc(nCount * 13 + 10);
strcpy( pszNeedToFree, "{" );
for( j = 0; j < nCount; j++ )
{
if( j != 0 )
strcat( pszNeedToFree+nOff, "," );
nOff += strlen(pszNeedToFree+nOff);
sprintf( pszNeedToFree+nOff, "%d", panItems[j] );
}
strcat( pszNeedToFree+nOff, "}" );
pszStrValue = pszNeedToFree;
}
// We need special formatting for real list values.
else if( nOGRFieldType == OFTRealList )
{
int nCount, nOff = 0, j;
const double *padfItems =poFeature->GetFieldAsDoubleList(i,&nCount);
pszNeedToFree = (char *) CPLMalloc(nCount * 40 + 10);
strcpy( pszNeedToFree, "{" );
for( j = 0; j < nCount; j++ )
{
if( j != 0 )
strcat( pszNeedToFree+nOff, "," );
nOff += strlen(pszNeedToFree+nOff);
//Check for special values. They need to be quoted.
if( CPLIsNan(padfItems[j]) )
sprintf( pszNeedToFree+nOff, "NaN" );
else if( CPLIsInf(padfItems[j]) )
sprintf( pszNeedToFree+nOff, (padfItems[j] > 0) ? "Infinity" : "-Infinity" );
else
sprintf( pszNeedToFree+nOff, "%.16g", padfItems[j] );
}
strcat( pszNeedToFree+nOff, "}" );
pszStrValue = pszNeedToFree;
}
// We need special formatting for string list values.
else if( nOGRFieldType == OFTStringList )
{
CPLString osStr;
char **papszItems = poFeature->GetFieldAsStringList(i);
pszStrValue = pszNeedToFree = CPLStrdup(OGRPGDumpEscapeStringList(papszItems, FALSE));
}
// Binary formatting
else if( nOGRFieldType == OFTBinary )
{
int nLen = 0;
GByte* pabyData = poFeature->GetFieldAsBinary( i, &nLen );
char* pszBytea = GByteArrayToBYTEA( pabyData, nLen);
pszStrValue = pszNeedToFree = pszBytea;
}
else if( nOGRFieldType == OFTReal )
{
char* pszComma = strchr((char*)pszStrValue, ',');
if (pszComma)
*pszComma = '.';
//Check for special values. They need to be quoted.
double dfVal = poFeature->GetFieldAsDouble(i);
if( CPLIsNan(dfVal) )
pszStrValue = "NaN";
else if( CPLIsInf(dfVal) )
pszStrValue = (dfVal > 0) ? "Infinity" : "-Infinity";
}
if( nOGRFieldType != OFTIntegerList &&
nOGRFieldType != OFTRealList &&
nOGRFieldType != OFTInteger &&
nOGRFieldType != OFTReal &&
nOGRFieldType != OFTBinary )
{
int iChar;
int iUTFChar = 0;
int nMaxWidth = poFeatureDefn->GetFieldDefn(i)->GetWidth();
for( iChar = 0; pszStrValue[iChar] != '\0'; iChar++ )
{
//count of utf chars
if ((pszStrValue[iChar] & 0xc0) != 0x80)
{
if( nMaxWidth > 0 && iUTFChar == nMaxWidth )
{
CPLDebug( "PG",
"Truncated %s field value, it was too long.",
poFeatureDefn->GetFieldDefn(i)->GetNameRef() );
break;
}
iUTFChar++;
}
/* Escape embedded \, \t, \n, \r since they will cause COPY
to misinterpret a line of text and thus abort */
if( pszStrValue[iChar] == '\\' ||
pszStrValue[iChar] == '\t' ||
pszStrValue[iChar] == '\r' ||
pszStrValue[iChar] == '\n' )
{
osCommand += '\\';
}
osCommand += pszStrValue[iChar];
}
}
else
{
osCommand += pszStrValue;
}
if( pszNeedToFree )
CPLFree( pszNeedToFree );
}
/* Add end of line marker */
//osCommand += "\n";
/* ------------------------------------------------------------ */
/* Execute the copy. */
/* ------------------------------------------------------------ */
OGRErr result = OGRERR_NONE;
poDS->Log(osCommand, FALSE);
return result;
}
OGRErr OGRPGDumpLayer::CreateFeatureViaInsert( OGRFeature *poFeature )
{
CPLString osCommand;
int i = 0;
int bNeedComma = FALSE;
OGRErr eErr = OGRERR_FAILURE;
int bEmptyInsert = FALSE;
if( NULL == poFeature )
{
CPLError( CE_Failure, CPLE_AppDefined,
"NULL pointer to OGRFeature passed to CreateFeatureViaInsert()." );
return eErr;
}
/* -------------------------------------------------------------------- */
/* Form the INSERT command. */
/* -------------------------------------------------------------------- */
osCommand.Printf( "INSERT INTO %s (", pszSqlTableName );
for(i = 0; i < poFeatureDefn->GetGeomFieldCount(); i++ )
{
OGRGeometry *poGeom = poFeature->GetGeomFieldRef(i);
if( poGeom != NULL )
{
if( bNeedComma )
osCommand += ", ";
OGRGeomFieldDefn* poGFldDefn = poFeature->GetGeomFieldDefnRef(i);
osCommand = osCommand + OGRPGDumpEscapeColumnName(poGFldDefn->GetNameRef()) + " ";
bNeedComma = TRUE;
}
}
if( poFeature->GetFID() != OGRNullFID && pszFIDColumn != NULL )
{
if( bNeedComma )
osCommand += ", ";
osCommand = osCommand + OGRPGDumpEscapeColumnName(pszFIDColumn) + " ";
bNeedComma = TRUE;
}
for( i = 0; i < poFeatureDefn->GetFieldCount(); i++ )
{
if( !poFeature->IsFieldSet( i ) )
continue;
if( !bNeedComma )
bNeedComma = TRUE;
else
osCommand += ", ";
osCommand = osCommand
+ OGRPGDumpEscapeColumnName(poFeatureDefn->GetFieldDefn(i)->GetNameRef());
}
if (!bNeedComma)
bEmptyInsert = TRUE;
osCommand += ") VALUES (";
/* Set the geometry */
bNeedComma = FALSE;
for(i = 0; i < poFeatureDefn->GetGeomFieldCount(); i++ )
{
OGRGeometry *poGeom = poFeature->GetGeomFieldRef(i);
if( poGeom != NULL )
{
char *pszWKT = NULL;
OGRPGDumpGeomFieldDefn* poGFldDefn =
(OGRPGDumpGeomFieldDefn*) poFeature->GetGeomFieldDefnRef(i);
poGeom->closeRings();
poGeom->setCoordinateDimension( poGFldDefn->nCoordDimension );
if( bNeedComma )
osCommand += ", ";
if( bWriteAsHex )
{
char* pszHex = OGRGeometryToHexEWKB( poGeom, poGFldDefn->nSRSId );
osCommand += "'";
if (pszHex)
osCommand += pszHex;
osCommand += "'";
CPLFree(pszHex);
}
else
{
poGeom->exportToWkt( &pszWKT );
if( pszWKT != NULL )
{
osCommand +=
CPLString().Printf(
"GeomFromEWKT('SRID=%d;%s'::TEXT) ", poGFldDefn->nSRSId, pszWKT );
OGRFree( pszWKT );
}
else
osCommand += "''";
}
bNeedComma = TRUE;
}
}
/* Set the FID */
if( poFeature->GetFID() != OGRNullFID && pszFIDColumn != NULL )
{
if( bNeedComma )
osCommand += ", ";
osCommand += CPLString().Printf( "%ld ", poFeature->GetFID() );
bNeedComma = TRUE;
}
for( i = 0; i < poFeatureDefn->GetFieldCount(); i++ )
{
if( !poFeature->IsFieldSet( i ) )
continue;
if( bNeedComma )
osCommand += ", ";
else
bNeedComma = TRUE;
AppendFieldValue(osCommand, poFeature, i);
}
osCommand += ")";
if (bEmptyInsert)
osCommand.Printf( "INSERT INTO %s DEFAULT VALUES", pszSqlTableName );
/* -------------------------------------------------------------------- */
/* Execute the insert. */
/* -------------------------------------------------------------------- */
poDS->Log(osCommand);
return OGRERR_NONE;
}
