Report correct count of inserted nodes when adding legend nodes

wonder-sk · wonder-sk · commit ee0e90a44b52 · 2015-01-22T13:42:31.000+07:00
diff --git a/src/core/layertree/qgslayertreemodel.cpp b/src/core/layertree/qgslayertreemodel.cpp
@@ -1079,24 +1079,27 @@ void QgsLayerTreeModel::addLegendToLayer( QgsLayerTreeLayer* nodeL )
 
   bool isEmbedded = filteredLstNew.count() == 1 && filteredLstNew[0]->isEmbeddedInParent();
 
-  if ( ! isEmbedded ) beginInsertRows( node2index( nodeL ), 0, filteredLstNew.count() - 1 );
-
   foreach ( QgsLayerTreeModelLegendNode* n, lstNew )
   {
     n->setParent( this );
     connect( n, SIGNAL( dataChanged() ), this, SLOT( legendNodeDataChanged() ) );
   }
 
-  LayerLegendData& data = mLegend[nodeL];
+  LayerLegendData data;
   data.originalNodes = lstNew;
   data.activeNodes = filteredLstNew;
-
   data.tree = 0;
 
   // maybe the legend nodes form a tree - try to create a tree structure from the list
   if ( testFlag( ShowLegendAsTree ) )
     tryBuildLegendTree( data );
 
+  int count = data.tree ? data.tree->children[0].count() : filteredLstNew.count();
+
+  if ( ! isEmbedded ) beginInsertRows( node2index( nodeL ), 0, count - 1 );
+
+  mLegend[nodeL] = data;
+
   if ( ! isEmbedded ) endInsertRows();
 }
 
