pyffi.qskope.global

52 """General purpose model for QModelIndexed access to data loaded with 53 pyffi.""" 54 # column definitions 55 NUM_COLUMNS = 3 56 COL_TYPE = 0 57 COL_NUMBER = 2 58 COL_NAME = 1 59

60 - class IndexDict(DictMixin):

61 - def __init__(self):

62 self.clear()

63

64 - def __getitem__(self, key):

65 try: 66 return self.data[id(key)] 67 except KeyError: 68 index = self.free_indices[-1] 69 self.data[id(key)] = index 70 if len(self.free_indices) == 1: 71 self.free_indices[0] += 1 72 else: 73 self.free_indices.pop()

74

75 - def __delitem__(self, key):

76 # index becomes available 77 self.free_indices.append(self.data[id(key)]) 78 # remove it 79 del self.data[id(key)]

80

81 - def clear(self):

82 # all indices larger than the first element 83 # are free as well 84 self.free_indices = [0] 85 self.data = {}

86

87 - def __init__(self, parent=None, globalnode=None, edge_filter=EdgeFilter()):

88 """Initialize the model to display the given data.""" 89 QtCore.QAbstractItemModel.__init__(self, parent) 90 # set up the tree 91 self.root_item = GlobalTreeItem( 92 data=GlobalTreeItemData(node=globalnode), 93 edge_filter=edge_filter) 94 # set up the index dictionary 95 self.index_dict = self.IndexDict() 96 self.updateIndexDict(self.root_item)

97

98 - def updateIndexDict(self, item):

99 self.index_dict[item.data.node] 100 for child_item in item.children: 101 self.updateIndexDict(child_item)

102 103

104 - def flags(self, index):

105 """Return flags for the given index: all indices are enabled and 106 selectable.""" 107 # all items are selectable 108 # they are enabled if their edge_type is active 109 if not index.isValid(): 110 return QtCore.Qt.ItemFlags() 111 item = index.internalPointer() 112 if item.edge_type.active: 113 flags = QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable 114 else: 115 flags = QtCore.Qt.ItemIsSelectable 116 return QtCore.Qt.ItemFlags(flags)

117

118 - def data(self, index, role):

119 """Return the data of model index in a particular role.""" 120 # check if the index is valid 121 # check if the role is supported 122 if not index.isValid() or role != QtCore.Qt.DisplayRole: 123 return QtCore.QVariant() 124 # get the data for display 125 data = index.internalPointer().data 126 127 # the type column 128 if index.column() == self.COL_TYPE: 129 return QtCore.QVariant(data.typename) 130 elif index.column() == self.COL_NAME: 131 return QtCore.QVariant(data.display) 132 elif index.column() == self.COL_NUMBER: 133 return QtCore.QVariant(self.index_dict[data.node]) 134 135 # other colums: invalid 136 else: 137 return QtCore.QVariant()

138

139 - def headerData(self, section, orientation, role):

140 """Return header data.""" 141 if (orientation == QtCore.Qt.Horizontal 142 and role == QtCore.Qt.DisplayRole): 143 if section == self.COL_TYPE: 144 return QtCore.QVariant("Type") 145 elif section == self.COL_NAME: 146 return QtCore.QVariant("Name") 147 elif section == self.COL_NUMBER: 148 return QtCore.QVariant("#") 149 return QtCore.QVariant()

150

151 - def rowCount(self, parent = QtCore.QModelIndex()):

152 """Calculate a row count for the given parent index.""" 153 if not parent.isValid(): 154 return 1 155 else: 156 # get the parent child count = number of references 157 return len(parent.internalPointer().children)

158

159 - def columnCount(self, parent = QtCore.QModelIndex()):

160 """Return column count.""" 161 # column count is constant everywhere 162 return self.NUM_COLUMNS

163

164 - def index(self, row, column, parent):

165 """Create an index to item (row, column) of object parent.""" 166 # check if the parent is valid 167 if not parent.isValid(): 168 # parent is not valid, so we need a top-level object 169 # return the index with row'th block as internal pointer 170 item = self.root_item 171 else: 172 # parent is valid, so we need to go get the row'th reference 173 # get the parent pointer 174 item = parent.internalPointer().children[row] 175 return self.createIndex(row, column, item)

176

177 - def parent(self, index):

178 """Calculate parent of a given index.""" 179 # get parent structure 180 parent_item = index.internalPointer().parent 181 # if parent's parent is None, then index must be a top 182 # level object, so return invalid index 183 if parent_item is None: 184 return QtCore.QModelIndex() 185 # if parent's parent is not None, then it must be member of 186 # some deeper nested structure, so calculate the row as usual 187 else: 188 return self.createIndex(parent_item.row, 0, parent_item)

Source Code for Module pyffi.qskope.global_model