perf scripts python: call-graph-from-sql.py: Refactor TreeItem class
class TreeItem represents items at all levels of the call-graph tree. However, not all the levels represent the same data i.e. the top-level is comms, the next level is threads, and subsequent levels are functions. Consequently it is simpler to have separate classes for different levels with commonality in a base class. Refactor TreeItem class accordingly. Signed-off-by: Adrian Hunter <adrian.hunter@intel.com> Cc: Andi Kleen <ak@linux.intel.com> Cc: Jiri Olsa <jolsa@redhat.com> Link: http://lkml.kernel.org/r/20181001062853.28285-12-adrian.hunter@intel.com Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>
This commit is contained in:
Adrian Hunter
Arnaldo Carvalho de Melo
parent
4be9ace7e1
commit
341e73cbd3
|
|
@ -74,145 +74,6 @@ def QueryExec(query, stmt):
|
||||||
if not ret:
|
if not ret:
|
||||||
raise Exception("Query failed: " + query.lastError().text())
|
raise Exception("Query failed: " + query.lastError().text())
|
||||||
|
|
||||||
class TreeItem():
|
|
||||||
|
|
||||||
def __init__(self, db, row, parent_item):
|
|
||||||
self.db = db
|
|
||||||
self.row = row
|
|
||||||
self.parent_item = parent_item
|
|
||||||
self.query_done = False;
|
|
||||||
self.child_count = 0
|
|
||||||
self.child_items = []
|
|
||||||
self.data = ["", "", "", "", "", "", ""]
|
|
||||||
self.comm_id = 0
|
|
||||||
self.thread_id = 0
|
|
||||||
self.call_path_id = 1
|
|
||||||
self.branch_count = 0
|
|
||||||
self.time = 0
|
|
||||||
if not parent_item:
|
|
||||||
self.setUpRoot()
|
|
||||||
|
|
||||||
def setUpRoot(self):
|
|
||||||
self.query_done = True
|
|
||||||
query = QSqlQuery(self.db)
|
|
||||||
QueryExec(query, 'SELECT id, comm FROM comms')
|
|
||||||
while query.next():
|
|
||||||
if not query.value(0):
|
|
||||||
continue
|
|
||||||
child_item = TreeItem(self.db, self.child_count, self)
|
|
||||||
self.child_items.append(child_item)
|
|
||||||
self.child_count += 1
|
|
||||||
child_item.setUpLevel1(query.value(0), query.value(1))
|
|
||||||
|
|
||||||
def setUpLevel1(self, comm_id, comm):
|
|
||||||
self.query_done = True;
|
|
||||||
self.comm_id = comm_id
|
|
||||||
self.data[0] = comm
|
|
||||||
self.child_items = []
|
|
||||||
self.child_count = 0
|
|
||||||
query = QSqlQuery(self.db)
|
|
||||||
QueryExec(query, 'SELECT thread_id, ( SELECT pid FROM threads WHERE id = thread_id ), ( SELECT tid FROM threads WHERE id = thread_id ) FROM comm_threads WHERE comm_id = ' + str(comm_id))
|
|
||||||
while query.next():
|
|
||||||
child_item = TreeItem(self.db, self.child_count, self)
|
|
||||||
self.child_items.append(child_item)
|
|
||||||
self.child_count += 1
|
|
||||||
child_item.setUpLevel2(comm_id, query.value(0), query.value(1), query.value(2))
|
|
||||||
|
|
||||||
def setUpLevel2(self, comm_id, thread_id, pid, tid):
|
|
||||||
self.comm_id = comm_id
|
|
||||||
self.thread_id = thread_id
|
|
||||||
self.data[0] = str(pid) + ":" + str(tid)
|
|
||||||
|
|
||||||
def getChildItem(self, row):
|
|
||||||
return self.child_items[row]
|
|
||||||
|
|
||||||
def getParentItem(self):
|
|
||||||
return self.parent_item
|
|
||||||
|
|
||||||
def getRow(self):
|
|
||||||
return self.row
|
|
||||||
|
|
||||||
def addChild(self, call_path_id, name, dso, count, time, branch_count):
|
|
||||||
child_item = TreeItem(self.db, self.child_count, self)
|
|
||||||
child_item.comm_id = self.comm_id
|
|
||||||
child_item.thread_id = self.thread_id
|
|
||||||
child_item.call_path_id = call_path_id
|
|
||||||
child_item.branch_count = branch_count
|
|
||||||
child_item.time = time
|
|
||||||
child_item.data[0] = name
|
|
||||||
child_item.data[1] = dsoname(dso)
|
|
||||||
child_item.data[2] = str(count)
|
|
||||||
child_item.data[3] = str(time)
|
|
||||||
child_item.data[4] = PercentToOneDP(time, self.time)
|
|
||||||
child_item.data[5] = str(branch_count)
|
|
||||||
child_item.data[6] = PercentToOneDP(branch_count, self.branch_count)
|
|
||||||
self.child_items.append(child_item)
|
|
||||||
self.child_count += 1
|
|
||||||
|
|
||||||
def selectCalls(self):
|
|
||||||
self.query_done = True;
|
|
||||||
query = QSqlQuery(self.db)
|
|
||||||
ret = query.exec_('SELECT id, call_path_id, branch_count, call_time, return_time, '
|
|
||||||
'( SELECT name FROM symbols WHERE id = ( SELECT symbol_id FROM call_paths WHERE id = call_path_id ) ), '
|
|
||||||
'( SELECT short_name FROM dsos WHERE id = ( SELECT dso_id FROM symbols WHERE id = ( SELECT symbol_id FROM call_paths WHERE id = call_path_id ) ) ), '
|
|
||||||
'( SELECT ip FROM call_paths where id = call_path_id ) '
|
|
||||||
'FROM calls WHERE parent_call_path_id = ' + str(self.call_path_id) + ' AND comm_id = ' + str(self.comm_id) + ' AND thread_id = ' + str(self.thread_id) +
|
|
||||||
' ORDER BY call_path_id')
|
|
||||||
if not ret:
|
|
||||||
raise Exception("Query failed: " + query.lastError().text())
|
|
||||||
last_call_path_id = 0
|
|
||||||
name = ""
|
|
||||||
dso = ""
|
|
||||||
count = 0
|
|
||||||
branch_count = 0
|
|
||||||
total_branch_count = 0
|
|
||||||
time = 0
|
|
||||||
total_time = 0
|
|
||||||
while query.next():
|
|
||||||
if query.value(1) == last_call_path_id:
|
|
||||||
count += 1
|
|
||||||
branch_count += query.value(2)
|
|
||||||
time += query.value(4) - query.value(3)
|
|
||||||
else:
|
|
||||||
if count:
|
|
||||||
self.addChild(last_call_path_id, name, dso, count, time, branch_count)
|
|
||||||
last_call_path_id = query.value(1)
|
|
||||||
name = query.value(5)
|
|
||||||
dso = query.value(6)
|
|
||||||
count = 1
|
|
||||||
total_branch_count += branch_count
|
|
||||||
total_time += time
|
|
||||||
branch_count = query.value(2)
|
|
||||||
time = query.value(4) - query.value(3)
|
|
||||||
if count:
|
|
||||||
self.addChild(last_call_path_id, name, dso, count, time, branch_count)
|
|
||||||
total_branch_count += branch_count
|
|
||||||
total_time += time
|
|
||||||
# Top level does not have time or branch count, so fix that here
|
|
||||||
if total_branch_count > self.branch_count:
|
|
||||||
self.branch_count = total_branch_count
|
|
||||||
if self.branch_count:
|
|
||||||
for child_item in self.child_items:
|
|
||||||
child_item.data[6] = PercentToOneDP(child_item.branch_count, self.branch_count)
|
|
||||||
if total_time > self.time:
|
|
||||||
self.time = total_time
|
|
||||||
if self.time:
|
|
||||||
for child_item in self.child_items:
|
|
||||||
child_item.data[4] = PercentToOneDP(child_item.time, self.time)
|
|
||||||
|
|
||||||
def childCount(self):
|
|
||||||
if not self.query_done:
|
|
||||||
self.selectCalls()
|
|
||||||
return self.child_count
|
|
||||||
|
|
||||||
def hasChildren(self):
|
|
||||||
if not self.query_done:
|
|
||||||
return True
|
|
||||||
return self.child_count > 0
|
|
||||||
|
|
||||||
def getData(self, column):
|
|
||||||
return self.data[column]
|
|
||||||
|
|
||||||
# Tree data model
|
# Tree data model
|
||||||
|
|
||||||
class TreeModel(QAbstractItemModel):
|
class TreeModel(QAbstractItemModel):
|
||||||
|
|
@ -277,12 +138,144 @@ class TreeModel(QAbstractItemModel):
|
||||||
item = index.internalPointer()
|
item = index.internalPointer()
|
||||||
return self.DisplayData(item, index)
|
return self.DisplayData(item, index)
|
||||||
|
|
||||||
|
# Context-sensitive call graph data model item base
|
||||||
|
|
||||||
|
class CallGraphLevelItemBase(object):
|
||||||
|
|
||||||
|
def __init__(self, glb, row, parent_item):
|
||||||
|
self.glb = glb
|
||||||
|
self.row = row
|
||||||
|
self.parent_item = parent_item
|
||||||
|
self.query_done = False;
|
||||||
|
self.child_count = 0
|
||||||
|
self.child_items = []
|
||||||
|
|
||||||
|
def getChildItem(self, row):
|
||||||
|
return self.child_items[row]
|
||||||
|
|
||||||
|
def getParentItem(self):
|
||||||
|
return self.parent_item
|
||||||
|
|
||||||
|
def getRow(self):
|
||||||
|
return self.row
|
||||||
|
|
||||||
|
def childCount(self):
|
||||||
|
if not self.query_done:
|
||||||
|
self.Select()
|
||||||
|
if not self.child_count:
|
||||||
|
return -1
|
||||||
|
return self.child_count
|
||||||
|
|
||||||
|
def hasChildren(self):
|
||||||
|
if not self.query_done:
|
||||||
|
return True
|
||||||
|
return self.child_count > 0
|
||||||
|
|
||||||
|
def getData(self, column):
|
||||||
|
return self.data[column]
|
||||||
|
|
||||||
|
# Context-sensitive call graph data model level 2+ item base
|
||||||
|
|
||||||
|
class CallGraphLevelTwoPlusItemBase(CallGraphLevelItemBase):
|
||||||
|
|
||||||
|
def __init__(self, glb, row, comm_id, thread_id, call_path_id, time, branch_count, parent_item):
|
||||||
|
super(CallGraphLevelTwoPlusItemBase, self).__init__(glb, row, parent_item)
|
||||||
|
self.comm_id = comm_id
|
||||||
|
self.thread_id = thread_id
|
||||||
|
self.call_path_id = call_path_id
|
||||||
|
self.branch_count = branch_count
|
||||||
|
self.time = time
|
||||||
|
|
||||||
|
def Select(self):
|
||||||
|
self.query_done = True;
|
||||||
|
query = QSqlQuery(self.glb.db)
|
||||||
|
QueryExec(query, "SELECT call_path_id, name, short_name, COUNT(calls.id), SUM(return_time - call_time), SUM(branch_count)"
|
||||||
|
" FROM calls"
|
||||||
|
" INNER JOIN call_paths ON calls.call_path_id = call_paths.id"
|
||||||
|
" INNER JOIN symbols ON call_paths.symbol_id = symbols.id"
|
||||||
|
" INNER JOIN dsos ON symbols.dso_id = dsos.id"
|
||||||
|
" WHERE parent_call_path_id = " + str(self.call_path_id) +
|
||||||
|
" AND comm_id = " + str(self.comm_id) +
|
||||||
|
" AND thread_id = " + str(self.thread_id) +
|
||||||
|
" GROUP BY call_path_id, name, short_name"
|
||||||
|
" ORDER BY call_path_id")
|
||||||
|
while query.next():
|
||||||
|
child_item = CallGraphLevelThreeItem(self.glb, self.child_count, self.comm_id, self.thread_id, query.value(0), query.value(1), query.value(2), query.value(3), int(query.value(4)), int(query.value(5)), self)
|
||||||
|
self.child_items.append(child_item)
|
||||||
|
self.child_count += 1
|
||||||
|
|
||||||
|
# Context-sensitive call graph data model level three item
|
||||||
|
|
||||||
|
class CallGraphLevelThreeItem(CallGraphLevelTwoPlusItemBase):
|
||||||
|
|
||||||
|
def __init__(self, glb, row, comm_id, thread_id, call_path_id, name, dso, count, time, branch_count, parent_item):
|
||||||
|
super(CallGraphLevelThreeItem, self).__init__(glb, row, comm_id, thread_id, call_path_id, time, branch_count, parent_item)
|
||||||
|
dso = dsoname(dso)
|
||||||
|
self.data = [ name, dso, str(count), str(time), PercentToOneDP(time, parent_item.time), str(branch_count), PercentToOneDP(branch_count, parent_item.branch_count) ]
|
||||||
|
self.dbid = call_path_id
|
||||||
|
|
||||||
|
# Context-sensitive call graph data model level two item
|
||||||
|
|
||||||
|
class CallGraphLevelTwoItem(CallGraphLevelTwoPlusItemBase):
|
||||||
|
|
||||||
|
def __init__(self, glb, row, comm_id, thread_id, pid, tid, parent_item):
|
||||||
|
super(CallGraphLevelTwoItem, self).__init__(glb, row, comm_id, thread_id, 1, 0, 0, parent_item)
|
||||||
|
self.data = [str(pid) + ":" + str(tid), "", "", "", "", "", ""]
|
||||||
|
self.dbid = thread_id
|
||||||
|
|
||||||
|
def Select(self):
|
||||||
|
super(CallGraphLevelTwoItem, self).Select()
|
||||||
|
for child_item in self.child_items:
|
||||||
|
self.time += child_item.time
|
||||||
|
self.branch_count += child_item.branch_count
|
||||||
|
for child_item in self.child_items:
|
||||||
|
child_item.data[4] = PercentToOneDP(child_item.time, self.time)
|
||||||
|
child_item.data[6] = PercentToOneDP(child_item.branch_count, self.branch_count)
|
||||||
|
|
||||||
|
# Context-sensitive call graph data model level one item
|
||||||
|
|
||||||
|
class CallGraphLevelOneItem(CallGraphLevelItemBase):
|
||||||
|
|
||||||
|
def __init__(self, glb, row, comm_id, comm, parent_item):
|
||||||
|
super(CallGraphLevelOneItem, self).__init__(glb, row, parent_item)
|
||||||
|
self.data = [comm, "", "", "", "", "", ""]
|
||||||
|
self.dbid = comm_id
|
||||||
|
|
||||||
|
def Select(self):
|
||||||
|
self.query_done = True;
|
||||||
|
query = QSqlQuery(self.glb.db)
|
||||||
|
QueryExec(query, "SELECT thread_id, pid, tid"
|
||||||
|
" FROM comm_threads"
|
||||||
|
" INNER JOIN threads ON thread_id = threads.id"
|
||||||
|
" WHERE comm_id = " + str(self.dbid))
|
||||||
|
while query.next():
|
||||||
|
child_item = CallGraphLevelTwoItem(self.glb, self.child_count, self.dbid, query.value(0), query.value(1), query.value(2), self)
|
||||||
|
self.child_items.append(child_item)
|
||||||
|
self.child_count += 1
|
||||||
|
|
||||||
|
# Context-sensitive call graph data model root item
|
||||||
|
|
||||||
|
class CallGraphRootItem(CallGraphLevelItemBase):
|
||||||
|
|
||||||
|
def __init__(self, glb):
|
||||||
|
super(CallGraphRootItem, self).__init__(glb, 0, None)
|
||||||
|
self.dbid = 0
|
||||||
|
self.query_done = True;
|
||||||
|
query = QSqlQuery(glb.db)
|
||||||
|
QueryExec(query, "SELECT id, comm FROM comms")
|
||||||
|
while query.next():
|
||||||
|
if not query.value(0):
|
||||||
|
continue
|
||||||
|
child_item = CallGraphLevelOneItem(glb, self.child_count, query.value(0), query.value(1), self)
|
||||||
|
self.child_items.append(child_item)
|
||||||
|
self.child_count += 1
|
||||||
|
|
||||||
# Context-sensitive call graph data model
|
# Context-sensitive call graph data model
|
||||||
|
|
||||||
class CallGraphModel(TreeModel):
|
class CallGraphModel(TreeModel):
|
||||||
|
|
||||||
def __init__(self, glb, parent=None):
|
def __init__(self, glb, parent=None):
|
||||||
super(CallGraphModel, self).__init__(TreeItem(glb.db, 0, None), parent)
|
super(CallGraphModel, self).__init__(CallGraphRootItem(glb), parent)
|
||||||
self.glb = glb
|
self.glb = glb
|
||||||
|
|
||||||
def columnCount(self, parent=None):
|
def columnCount(self, parent=None):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue