def setupModelData(self): numberOfNodesByType = nodes.getNumberOfNodesByType() for nodeType in nodes.getAllNodeTypes(): if self.__nodeTypeMatchesFilter(nodeType): nodeTypeItem = TreeItem([nodeType, numberOfNodesByType[nodeType], ""], parent=self.__rootItem) nodeTypeItem.setRepresentsNodeType(True) self.__rootItem.appendChild(nodeTypeItem) for node in self.__nodeDict[nodeType]: if self.__filter.isStringInFilter(node["name"].value()): nodeItem = TreeItem([node["name"].value(), "", ""], parent=nodeTypeItem) nodeItem.setRepresentsNode(True) nodeTypeItem.appendChild(nodeItem)
def __init__(self, parent = None): QtCore.QAbstractTableModel.__init__(self, parent) self.__nodeTypes = nodes.getAllNodeTypes() self.__numberOfNodesByType = nodes.getNumberOfNodesByType() self.__headerNames = ('NodeType', '#')