Update feature count on layer modification (Fix #8445)

qgis · Aug 19, 2013 · 5e4a42e · 5e4a42e
1 parent 768cc1a
commit 5e4a42e
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/src/app/legend/qgslegendlayer.cpp b/src/app/legend/qgslegendlayer.cpp
@@ -496,9 +496,9 @@ QString QgsLegendLayer::label()
 {
   QString name = mLyr.layer()->name();
   QgsVectorLayer *vlayer = dynamic_cast<QgsVectorLayer *>( mLyr.layer() );
-  if ( mShowFeatureCount && vlayer && vlayer->featureCount() >= 0 )
+  if ( mShowFeatureCount && vlayer && vlayer->pendingFeatureCount() >= 0 )
   {
-    name += QString( " [%1]" ).arg( vlayer->featureCount() );
+    name += QString( " [%1]" ).arg( vlayer->pendingFeatureCount() );
   }
   return name;
 }

diff --git a/src/core/qgsvectorlayer.cpp b/src/core/qgsvectorlayer.cpp
@@ -1498,6 +1498,7 @@ bool QgsVectorLayer::startEditing()
 
   mEditBuffer = new QgsVectorLayerEditBuffer( this );
   // forward signals
+  connect( mEditBuffer, SIGNAL( layerModified() ), this, SLOT( invalidateSymbolCountedFlag() ) );
   connect( mEditBuffer, SIGNAL( layerModified() ), this, SIGNAL( layerModified() ) ); // TODO[MD]: necessary?
   //connect( mEditBuffer, SIGNAL( layerModified() ), this, SLOT( triggerRepaint() ) ); // TODO[MD]: works well?
   connect( mEditBuffer, SIGNAL( featureAdded( QgsFeatureId ) ), this, SIGNAL( featureAdded( QgsFeatureId ) ) );
@@ -3771,6 +3772,11 @@ void QgsVectorLayer::onCacheImageDelete()
     mCurrentRendererContext->setRenderingStopped( true );
 }
 
+void QgsVectorLayer::invalidateSymbolCountedFlag()
+{
+  mSymbolFeatureCounted = false;
+}
+
 QgsVectorLayer::ValueRelationData &QgsVectorLayer::valueRelation( int idx )
 {
   const QgsFields &fields = pendingFields();

diff --git a/src/core/qgsvectorlayer.h b/src/core/qgsvectorlayer.h
@@ -1305,6 +1305,9 @@ class CORE_EXPORT QgsVectorLayer : public QgsMapLayer
      */
     virtual void onCacheImageDelete();
 
+  protected slots:
+    void invalidateSymbolCountedFlag();
+
   signals:
 
     /**