def SetItems(parentItem, fictiveObjects, level): level += 1 for object in fictiveObjects: type = object.type if not type in showTypes: continue # Construct text if type=='cell': type = '##' elif type=='attribute': type = 'attr' # if type == 'import': text = "%s (%s)" % (object.name, object.text) elif type=='todo': text = object.name else: text = "%s %s" % (type, object.name) # Create item thisItem = QtGui.QTreeWidgetItem(parentItem, [text]) color = QtGui.QColor(colours[object.type]) thisItem.setForeground(0, QtGui.QBrush(color)) font = thisItem.font(0) font.setBold(True) thisItem.setFont(0, font) thisItem.linenr = object.linenr # Is this the current item? if ln and object.linenr <= ln and object.linenr2 > ln: selectedItem[0] = thisItem # Any children that we should display? if object.children: SetItems(thisItem, object.children, level) # Set visibility thisItem.setExpanded( bool(level < showLevel) )
def updateStructure(self): """ Updates the tree. """ # Get editor editor = iep.editors.getCurrentEditor() if not editor: return # Something to show result = iep.parser._getResult() if result is None: return # Do the ids match? id0, id1, id2 = self._currentEditorId, id(editor), result.editorId if id0 != id1 or id0 != id2: return # Get current line number and the structure ln = editor.textCursor().blockNumber() ln += 1 # is ln as in line number area # Define colours colours = {'cell':'#007F00', 'class':'#0000FF', 'def':'#007F7F', 'attribute':'#444444', 'import':'#8800BB', 'todo':'#FF3333'} # Define what to show showTypes = self._config.showTypes # Define to what level to show (now is also a good time to save) showLevel = int( self._slider.value() ) self._config.level = showLevel # Define function to set items selectedItem = [None] def SetItems(parentItem, fictiveObjects, level): level += 1 for object in fictiveObjects: type = object.type if not type in showTypes: continue # Construct text if type=='cell': type = '##' elif type=='attribute': type = 'attr' # if type == 'import': text = "%s (%s)" % (object.name, object.text) elif type=='todo': text = object.name else: text = "%s %s" % (type, object.name) # Create item thisItem = QtGui.QTreeWidgetItem(parentItem, [text]) color = QtGui.QColor(colours[object.type]) thisItem.setForeground(0, QtGui.QBrush(color)) font = thisItem.font(0) font.setBold(True) thisItem.setFont(0, font) thisItem.linenr = object.linenr # Is this the current item? if ln and object.linenr <= ln and object.linenr2 > ln: selectedItem[0] = thisItem # Any children that we should display? if object.children: SetItems(thisItem, object.children, level) # Set visibility thisItem.setExpanded( bool(level < showLevel) ) # Go self._tree.setUpdatesEnabled(False) self._tree.clear() SetItems(self._tree, result.rootItem.children, 0) self._tree.setUpdatesEnabled(True) # Handle selected item selectedItem = selectedItem[0] if selectedItem: selectedItem.setBackground(0, QtGui.QBrush(QtGui.QColor('#CCC'))) self._tree.scrollToItem(selectedItem) # ensure visible