Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 70d831e8 authored by Adrian Hunter's avatar Adrian Hunter Committed by Arnaldo Carvalho de Melo
Browse files

perf scripts python: call-graph-from-sql.py: Factor out CallGraphModel from TreeModel 



Factor out CallGraphModel from TreeModel, which paves the way to reuse
TreeModel in future reports.

Signed-off-by: default avatarAdrian 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-10-adrian.hunter@intel.com


Signed-off-by: default avatarArnaldo Carvalho de Melo <acme@redhat.com>
parent e99ef814

tools/perf/scripts/python/call-graph-from-sql.py

+61 −29
Original line number Diff line number Diff line
@@ -201,42 +201,47 @@ class TreeItem(): 
			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 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



class TreeModel(QAbstractItemModel):



	def __init__(self, db, parent=None):

	def __init__(self, root, parent=None):

		super(TreeModel, self).__init__(parent)

		self.db = db

		self.root = TreeItem(db, 0, None)

		self.root = root

		self.last_row_read = 0



	def columnCount(self, parent):

		return self.root.columnCount()



	def rowCount(self, parent):

	def Item(self, parent):

		if parent.isValid():

			parent_item = parent.internalPointer()

			return parent.internalPointer()

		else:

			parent_item = self.root

		return parent_item.childCount()

			return self.root



	def rowCount(self, parent):

		result = self.Item(parent).childCount()

		if result < 0:

			result = 0

			self.dataChanged.emit(parent, parent)

		return result



	def hasChildren(self, parent):

		return self.Item(parent).hasChildren()



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

		if role == Qt.TextAlignmentRole:

			if section > 1:

				return Qt.AlignRight

			return self.columnAlignment(section)

		if role != Qt.DisplayRole:

			return None

		if orientation != Qt.Horizontal:

			return None

		return self.root.columnHeader(section)

		return self.columnHeader(section)



	def parent(self, child):

		child_item = child.internalPointer()
@@ -246,21 +251,48 @@ class TreeModel(QAbstractItemModel): 
		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)

		child_item = self.Item(parent).getChildItem(row)

		return self.createIndex(row, column, child_item)



	def DisplayData(self, item, index):

		return item.getData(index.column())



	def columnAlignment(self, column):

		return Qt.AlignLeft



	def columnFont(self, column):

		return None



	def data(self, index, role):

		if role == Qt.TextAlignmentRole:

			if index.column() > 1:

				return Qt.AlignRight

			return self.columnAlignment(index.column())

		if role == Qt.FontRole:

			return self.columnFont(index.column())

		if role != Qt.DisplayRole:

			return None

		index_item = index.internalPointer()

		return index_item.getData(index.column())

		item = index.internalPointer()

		return self.DisplayData(item, index)



# Context-sensitive call graph data model



class CallGraphModel(TreeModel):



	def __init__(self, glb, parent=None):

		super(CallGraphModel, self).__init__(TreeItem(glb.db, 0, None), parent)

		self.glb = glb



	def columnCount(self, parent=None):

		return 7



	def columnHeader(self, column):

		headers = ["Call Path", "Object", "Count ", "Time (ns) ", "Time (%) ", "Branch Count ", "Branch Count (%) "]

		return headers[column]



	def columnAlignment(self, column):

		alignment = [ Qt.AlignLeft, Qt.AlignLeft, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight, Qt.AlignRight ]

		return alignment[column]



# Main window



class MainWindow(QMainWindow):
@@ -275,7 +307,7 @@ class MainWindow(QMainWindow): 
		self.setWindowIcon(self.style().standardIcon(QStyle.SP_ComputerIcon))

		self.setMinimumSize(200, 100)



		self.model = TreeModel(glb.db)

		self.model = CallGraphModel(glb)



		self.view = QTreeView()

		self.view.setModel(self.model)