Messages in this thread | | | Date | Tue, 26 May 2015 12:43:03 -0300 | From | Arnaldo Carvalho de Melo <> | Subject | Re: [PATCH V5 15/16] perf tools: Add example call-graph script |
| |
Em Fri, May 22, 2015 at 02:54:12PM +0300, Adrian Hunter escreveu: > Signed-off-by: Adrian Hunter <adrian.hunter@intel.com>
Can you please describe what the script does and if possible, in ascii art some approximation of its output?
I applied up to the patch before this one (14/16), nooking at 16/16 now.
- Arnaldo
> --- > .../scripts/python/call-graph-from-postgresql.py | 285 +++++++++++++++++++++ > 1 file changed, 285 insertions(+) > create mode 100644 tools/perf/scripts/python/call-graph-from-postgresql.py > > diff --git a/tools/perf/scripts/python/call-graph-from-postgresql.py b/tools/perf/scripts/python/call-graph-from-postgresql.py > new file mode 100644 > index 0000000..5e8cfdf > --- /dev/null > +++ b/tools/perf/scripts/python/call-graph-from-postgresql.py > @@ -0,0 +1,285 @@ > +#!/usr/bin/python2 > +# call-graph-from-postgresql.py: create call-graph from postgresql database > +# Copyright (c) 2014, Intel Corporation. > +# > +# This program is free software; you can redistribute it and/or modify it > +# under the terms and conditions of the GNU General Public License, > +# version 2, as published by the Free Software Foundation. > +# > +# This program is distributed in the hope it will be useful, but WITHOUT > +# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or > +# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for > +# more details. > + > +import sys > +from PySide.QtCore import * > +from PySide.QtGui import * > +from PySide.QtSql import * > +from decimal import * > + > +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) > + ret = query.exec_('SELECT id, comm FROM comms') > + if not ret: > + raise Exception("Query failed: " + query.lastError().text()) > + 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) > + ret = query.exec_('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)) > + if not ret: > + raise Exception("Query failed: " + query.lastError().text()) > + 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 timePercent(self, b): > + if not self.time: > + return "0.0" > + x = (b * Decimal(100)) / self.time > + return str(x.quantize(Decimal('.1'), rounding=ROUND_HALF_UP)) > + > + def branchPercent(self, b): > + if not self.branch_count: > + return "0.0" > + x = (b * Decimal(100)) / self.branch_count > + return str(x.quantize(Decimal('.1'), rounding=ROUND_HALF_UP)) > + > + 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 > + if dso == "[kernel.kallsyms]": > + dso = "[kernel]" > + child_item.data[1] = dso > + child_item.data[2] = str(count) > + child_item.data[3] = str(time) > + child_item.data[4] = self.timePercent(time) > + child_item.data[5] = str(branch_count) > + child_item.data[6] = self.branchPercent(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] = self.branchPercent(child_item.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] = self.timePercent(child_item.time) > + > + def childCount(self): > + if not self.query_done: > + self.selectCalls() > + return self.child_count > + > + def columnCount(self): > + return 7 > + > + def columnHeader(self, column): > + headers = ["Call Path", "Object", "Count ", "Time (ns) ", "Time (%) ", "Branch Count ", "Branch Count (%) "] > + return headers[column] > + > + def getData(self, column): > + return self.data[column] > + > +class TreeModel(QAbstractItemModel): > + > + def __init__(self, db, parent=None): > + super(TreeModel, self).__init__(parent) > + self.db = db > + self.root = TreeItem(db, 0, None) > + > + def columnCount(self, parent): > + return self.root.columnCount() > + > + def rowCount(self, parent): > + if parent.isValid(): > + parent_item = parent.internalPointer() > + else: > + parent_item = self.root > + return parent_item.childCount() > + > + def headerData(self, section, orientation, role): > + if role == Qt.TextAlignmentRole: > + if section > 1: > + return Qt.AlignRight > + if role != Qt.DisplayRole: > + return None > + if orientation != Qt.Horizontal: > + return None > + return self.root.columnHeader(section) > + > + def parent(self, child): > + child_item = child.internalPointer() > + if child_item is self.root: > + return QModelIndex() > + parent_item = child_item.getParentItem() > + return self.createIndex(parent_item.getRow(), 0, parent_item) > + > + def index(self, row, column, parent): > + if parent.isValid(): > + parent_item = parent.internalPointer() > + else: > + parent_item = self.root > + child_item = parent_item.getChildItem(row) > + return self.createIndex(row, column, child_item) > + > + def data(self, index, role): > + if role == Qt.TextAlignmentRole: > + if index.column() > 1: > + return Qt.AlignRight > + if role != Qt.DisplayRole: > + return None > + index_item = index.internalPointer() > + return index_item.getData(index.column()) > + > +class MainWindow(QMainWindow): > + > + def __init__(self, db, dbname, parent=None): > + super(MainWindow, self).__init__(parent) > + > + self.setObjectName("MainWindow") > + self.setWindowTitle("Call Graph: " + dbname) > + self.move(100, 100) > + self.resize(800, 600) > + style = self.style() > + icon = style.standardIcon(QStyle.SP_MessageBoxInformation) > + self.setWindowIcon(icon); > + > + self.model = TreeModel(db) > + > + self.view = QTreeView() > + self.view.setModel(self.model) > + > + self.setCentralWidget(self.view) > + > +if __name__ == '__main__': > + if (len(sys.argv) < 2): > + print >> sys.stderr, "Usage is: call-graph-from-postgresql.py <database name>" > + raise Exception("Too few arguments") > + > + dbname = sys.argv[1] > + > + db = QSqlDatabase.addDatabase('QPSQL') > + > + opts = dbname.split() > + for opt in opts: > + if '=' in opt: > + opt = opt.split('=') > + if opt[0] == 'hostname': > + db.setHostName(opt[1]) > + elif opt[0] == 'port': > + db.setPort(int(opt[1])) > + elif opt[0] == 'username': > + db.setUserName(opt[1]) > + elif opt[0] == 'password': > + db.setPassword(opt[1]) > + elif opt[0] == 'dbname': > + dbname = opt[1] > + else: > + dbname = opt > + > + db.setDatabaseName(dbname) > + if not db.open(): > + raise Exception("Failed to open database " + dbname + " error: " + db.lastError().text()) > + > + app = QApplication(sys.argv) > + window = MainWindow(db, dbname) > + window.show() > + err = app.exec_() > + db.close() > + sys.exit(err) > -- > 1.9.1
| |