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', '#')
