Instead of passing the tree root, get it from a method that can be implemented in any derived class.
Showing the changes to tools/perf/scripts/python (1 of 1 file). Show all
Unified · Split
| @@ -167,9 +167,10 @@ class Thread(QThread): · 166 unmodified lines | ||
| 167 | 167 | |
| 168 | 168 | class TreeModel(QAbstractItemModel): |
| 169 | 169 | |
| 170 | − def __init__(self, root, parent=None): | |
| 170 | + def __init__(self, glb, parent=None): | |
| 171 | 171 | super(TreeModel, self).__init__(parent) |
| 172 | − self.root = root | |
| 172 | + self.glb = glb | |
| 173 | + self.root = self.GetRoot() | |
| 173 | 174 | self.last_row_read = 0 |
| 174 | 175 | |
| 175 | 176 | def Item(self, parent): |
| @@ -562,8 +563,10 @@ class CallGraphRootItem(CallGraphLevelItemBase): · 386 unmodified lines | ||
| 562 | 563 | class CallGraphModel(TreeModel): |
| 563 | 564 | |
| 564 | 565 | def __init__(self, glb, parent=None): |
| 565 | − super(CallGraphModel, self).__init__(CallGraphRootItem(glb), parent) | |
| 566 | − self.glb = glb | |
| 566 | + super(CallGraphModel, self).__init__(glb, parent) | |
| 567 | + | |
| 568 | + def GetRoot(self): | |
| 569 | + return CallGraphRootItem(self.glb) | |
| 567 | 570 | |
| 568 | 571 | def columnCount(self, parent=None): |
| 569 | 572 | return 7 |
| @@ -1339,8 +1342,7 @@ class BranchModel(TreeModel): · 769 unmodified lines | ||
| 1339 | 1342 | progress = Signal(object) |
| 1340 | 1343 | |
| 1341 | 1344 | def __init__(self, glb, event_id, where_clause, parent=None): |
| 1342 | − super(BranchModel, self).__init__(BranchRootItem(), parent) | |
| 1343 | − self.glb = glb | |
| 1345 | + super(BranchModel, self).__init__(glb, parent) | |
| 1344 | 1346 | self.event_id = event_id |
| 1345 | 1347 | self.more = True |
| 1346 | 1348 | self.populated = 0 |
| @@ -1364,6 +1366,9 @@ class BranchModel(TreeModel): · 17 unmodified lines | ||
| 1364 | 1366 | self.fetcher.done.connect(self.Update) |
| 1365 | 1367 | self.fetcher.Fetch(glb_chunk_sz) |
| 1366 | 1368 | |
| 1369 | + def GetRoot(self): | |
| 1370 | + return BranchRootItem() | |
| 1371 | + | |
| 1367 | 1372 | def columnCount(self, parent=None): |
| 1368 | 1373 | return 8 |
| 1369 | 1374 | |