From 219bc2557b1ad36414e6a70cfcdd878f6d0ea150 Mon Sep 17 00:00:00 2001
From: Mika Fischer <fischer@videmo.de>
Date: Fri, 27 Jan 2012 13:32:53 +0100
Subject: [PATCH] Fix selection in scene view when one model item has multiple
 scene items

---
 sloth/gui/annotationscene.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/sloth/gui/annotationscene.py b/sloth/gui/annotationscene.py
index e78fd22..e9a56bb 100644
--- a/sloth/gui/annotationscene.py
+++ b/sloth/gui/annotationscene.py
@@ -206,12 +206,15 @@ class AnnotationScene(QGraphicsScene):
         self._labeltool.treeview().setSelectedItems(model_items)
         self.editSelectedItems()
 
-    def onSelectionChangedInTreeView(self, items):
+    def onSelectionChangedInTreeView(self, model_items):
         block = self.blockSignals(True)
-        items = [self.itemFromIndex(item.index()) for item in items]
+        selected_items = set()
+        for model_item in model_items:
+            for item in self.itemsFromIndex(model_item.index()):
+                selected_items.add(item)
         for item in self.items():
             item.setSelected(False)
-        for item in items:
+        for item in selected_items:
             if item is not None:
                 item.setSelected(True)
         self.blockSignals(block)