Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
postgres provider update:
- lazy determine extent and invalidate extent on subst string change (fixes #2752 for postgres)
- use estimate extent when "use estimate metadata" is set (reverts r13499)
- remove rest of unused threading


git-svn-id: http://svn.osgeo.org/qgis/trunk/qgis@13584 c8812cc2-4d05-0410-92ff-de0c093fc19c
  • Loading branch information
jef committed May 28, 2010
1 parent 4cefe6a commit b86f859
Show file tree
Hide file tree
Showing 2 changed files with 61 additions and 130 deletions.
183 changes: 61 additions & 122 deletions src/providers/postgres/qgspostgresprovider.cpp
Expand Up @@ -134,7 +134,7 @@ QgsPostgresProvider::QgsPostgresProvider( QString const & uri )
}

deduceEndian();
calculateExtents();
layerExtent.setMinimal();
featuresCounted = -1;

// set the primary key
Expand Down Expand Up @@ -717,12 +717,6 @@ void QgsPostgresProvider::setExtent( QgsRectangle& newExtent )
layerExtent.setYMinimum( newExtent.yMinimum() );
}

// TODO - make this function return the real extent_
QgsRectangle QgsPostgresProvider::extent()
{
return layerExtent; //extent_->MinX, extent_->MinY, extent_->MaxX, extent_->MaxY);
}

/**
* Return the feature type
*/
Expand Down Expand Up @@ -2740,9 +2734,8 @@ bool QgsPostgresProvider::setSubsetString( QString theSQL )
// uri? Perhaps this needs some rationalisation.....
setDataSourceUri( mUri.uri() );

// need to recalculate the number of features...
featuresCounted = -1;
calculateExtents();
layerExtent.setMinimal();

return true;
}
Expand All @@ -2752,16 +2745,13 @@ bool QgsPostgresProvider::setSubsetString( QString theSQL )
*/
long QgsPostgresProvider::featureCount() const
{
if( featuresCounted >= 0 )
if ( featuresCounted >= 0 )
return featuresCounted;

// get total number of features

// First get an approximate count; then delegate to
// a thread the task of getting the full count.
QString sql;

if ( !isQuery && mUseEstimatedMetadata && sqlWhereClause.isEmpty() )
if ( !isQuery && mUseEstimatedMetadata )
{
sql = QString( "select reltuples::int from pg_catalog.pg_class where oid=regclass(%1)::oid" ).arg( quotedValue( mQuery ) );
}
Expand All @@ -2787,130 +2777,79 @@ long QgsPostgresProvider::featureCount() const
return featuresCounted;
}

void QgsPostgresProvider::calculateExtents()
QgsRectangle QgsPostgresProvider::extent()
{
QString sql;
Result result;
QString ext;

// get the extents
if ( !isQuery && ( mUseEstimatedMetadata || sqlWhereClause.isEmpty() ) )
if ( layerExtent.isEmpty() )
{
// do stats exists?
sql = QString( "SELECT COUNT(*) FROM pg_stats WHERE schemaname=%1 AND tablename=%2 AND attname=%3" )
.arg( quotedValue( mSchemaName ) )
.arg( quotedValue( mTableName ) )
.arg( quotedValue( geometryColumn ) );
result = connectionRO->PQexec( sql );
if ( PQresultStatus( result ) == PGRES_TUPLES_OK && PQntuples( result ) == 1 )
QString sql;
Result result;
QString ext;

// get the extents
if ( !isQuery && ( mUseEstimatedMetadata || sqlWhereClause.isEmpty() ) )
{
if ( QString::fromUtf8( PQgetvalue( result, 0, 0 ) ).toInt() > 0 )
// do stats exists?
sql = QString( "SELECT COUNT(*) FROM pg_stats WHERE schemaname=%1 AND tablename=%2 AND attname=%3" )
.arg( quotedValue( mSchemaName ) )
.arg( quotedValue( mTableName ) )
.arg( quotedValue( geometryColumn ) );
result = connectionRO->PQexec( sql );
if ( PQresultStatus( result ) == PGRES_TUPLES_OK && PQntuples( result ) == 1 )
{
sql = QString( "select estimated_extent(%1,%2,%3)" )
.arg( quotedValue( mSchemaName ) )
.arg( quotedValue( mTableName ) )
.arg( quotedValue( geometryColumn ) );
result = connectionRO->PQexec( sql );
if ( PQresultStatus( result ) == PGRES_TUPLES_OK && PQntuples( result ) == 1 )
ext = PQgetvalue( result, 0, 0 );
if ( QString::fromUtf8( PQgetvalue( result, 0, 0 ) ).toInt() > 0 )
{
sql = QString( "select estimated_extent(%1,%2,%3)" )
.arg( quotedValue( mSchemaName ) )
.arg( quotedValue( mTableName ) )
.arg( quotedValue( geometryColumn ) );
result = connectionRO->PQexec( sql );
if ( PQresultStatus( result ) == PGRES_TUPLES_OK && PQntuples( result ) == 1 )
ext = PQgetvalue( result, 0, 0 );
}
}
}
else
{
QgsDebugMsg( QString( "no column statistics for %1.%2.%3" ).arg( mSchemaName ).arg( mTableName ).arg( geometryColumn ) );
}
}

if ( ext.isEmpty() )
{
sql = QString( "select extent(%1) from %2" )
.arg( quotedIdentifier( geometryColumn ) )
.arg( mQuery );

if ( !sqlWhereClause.isEmpty() )
sql += QString( " where %1" ).arg( sqlWhereClause );

result = connectionRO->PQexec( sql );
if ( PQresultStatus( result ) != PGRES_TUPLES_OK )
connectionRO->PQexecNR( "ROLLBACK" );
else if ( PQntuples( result ) == 1 )
ext = PQgetvalue( result, 0, 0 );
}

QgsDebugMsg( "Got extents using: " + sql );

QRegExp rx( "\\((.+) (.+),(.+) (.+)\\)" );
if ( ext.contains( rx ) )
{
QStringList ex = rx.capturedTexts();

layerExtent.setXMinimum( ex[1].toDouble() );
layerExtent.setYMinimum( ex[2].toDouble() );
layerExtent.setXMaximum( ex[3].toDouble() );
layerExtent.setYMaximum( ex[4].toDouble() );
}
else
{
QgsDebugMsg( "extents query failed" );
}

QgsDebugMsg( "Set extents to: " + layerExtent.toString() );
}

/**
* Event sink for events from threads
*/
void QgsPostgresProvider::customEvent( QEvent * e )
{
QgsDebugMsg( "received a custom event " + QString::number( e->type() ) );

switch (( int ) e->type() )
{
case QGis::ProviderExtentCalcEvent:

QgsDebugMsg( "extent has been calculated" );

// Collect the new extent from the event and set this layer's
// extent with it.

else
{
QgsRectangle* r = (( QgsProviderExtentCalcEvent* ) e )->layerExtent();
setExtent( *r );
QgsDebugMsg( QString( "no column statistics for %1.%2.%3" ).arg( mSchemaName ).arg( mTableName ).arg( geometryColumn ) );
}
}

QgsDebugMsg( "new extent has been saved" );

QgsDebugMsg( "Set extent to: " + layerExtent.toString() );

QgsDebugMsg( "emitting fullExtentCalculated()" );

emit fullExtentCalculated();

// TODO: Only uncomment this when the overview map canvas has been subclassed
// from the QgsMapCanvas

// QgsDebugMsg("emitting repaintRequested()");
// emit repaintRequested();

break;
if ( ext.isEmpty() )
{
sql = QString( "select extent(%1) from %2" )
.arg( quotedIdentifier( geometryColumn ) )
.arg( mQuery );

case QGis::ProviderCountCalcEvent:
if ( !sqlWhereClause.isEmpty() )
sql += QString( " where %1" ).arg( sqlWhereClause );

QgsDebugMsg( "count has been calculated" );
result = connectionRO->PQexec( sql );
if ( PQresultStatus( result ) != PGRES_TUPLES_OK )
connectionRO->PQexecNR( "ROLLBACK" );
else if ( PQntuples( result ) == 1 )
ext = PQgetvalue( result, 0, 0 );
}

featuresCounted = (( QgsProviderCountCalcEvent* ) e )->featuresCounted();
QgsDebugMsg( "Got extents using: " + sql );

QgsDebugMsg( "count is " + QString::number( featuresCounted ) );
QRegExp rx( "\\((.+) (.+),(.+) (.+)\\)" );
if ( ext.contains( rx ) )
{
QStringList ex = rx.capturedTexts();

break;
layerExtent.setXMinimum( ex[1].toDouble() );
layerExtent.setYMinimum( ex[2].toDouble() );
layerExtent.setXMaximum( ex[3].toDouble() );
layerExtent.setYMaximum( ex[4].toDouble() );
}
else
{
QgsDebugMsg( "extents query failed" );
}

default:
// do nothing
break;
QgsDebugMsg( "Set extents to: " + layerExtent.toString() );
}

QgsDebugMsg( "Finished processing custom event " + QString::number( e->type() ) );

return layerExtent;
}


Expand Down
8 changes: 0 additions & 8 deletions src/providers/postgres/qgspostgresprovider.h
Expand Up @@ -567,14 +567,6 @@ class QgsPostgresProvider : public QgsVectorDataProvider

int enabledCapabilities;

//! Calculate the extents of the layer
void calculateExtents();

/**
* Event sink for events from threads
*/
void customEvent( QEvent *e );

void appendGeomString( QgsGeometry *geom, QString &geomParam ) const;
QString paramValue( QString fieldvalue, const QString &defaultValue ) const;

Expand Down

0 comments on commit b86f859

Please sign in to comment.