colorize using the select column

diff --git a/sface/screens/mainscreen.py b/sface/screens/mainscreen.py
index ac0dedb..ae18c9d 100644 (file)
--- a/sface/screens/mainscreen.py
+++ b/sface/screens/mainscreen.py
@@ -29,19 +29,16 @@ class NodeNameDelegate(QStyledItemDelegate):
 
     def paint(self, painter, option, index):
         data = "%s" % index.data().toString()
-
-#         model = index.model()
-#         select_index = model.index(index.row(), 2, index.parent())
-#         select_data = select_index.data().toString()
+        model = index.model()
+        select_index = model.index(index.row(), 2, index.parent())
+        select_data = select_index.data().toString()
 #         if select_data == "false":
 #             print select_data
 #             model.setData(index, QString("*%s" % data), Qt.EditRole)
 #             model.setData(select_index, QString("true"), Qt.EditRole)
 
 
-        if data.startswith("*"): # already in the sliver
-            data = " %s " % data[1:]
-
+        if select_data == "true": # already in the sliver
             fm = QFontMetrics(option.font)
             rect = option.rect
             rect.setWidth(fm.width(QString(data)))
@@ -328,7 +325,7 @@ class SliceWidget(QWidget):
             available_nodes = filter(lambda x:x not in sliver_nodes, all_nodes)
 
             for node in sliver_nodes:
-                nodeItem = TreeItem([QString(""), QString("*%s" % node), QString("true")], networkItem)
+                nodeItem = TreeItem([QString(""), QString("%s" % node), QString("true")], networkItem)
                 networkItem.appendChild(nodeItem)
 
             for node in available_nodes: