1 files changed, 41 insertions, 0 deletions

diff --git a/tools/perf/scripts/python/exported-sql-viewer.py b/tools/perf/scripts/python/exported-sql-viewer.py
index baa2b220238a..affd80ebcae0 100755
--- a/tools/perf/scripts/python/exported-sql-viewer.py
+++ b/tools/perf/scripts/python/exported-sql-viewer.py
@@ -887,6 +887,8 @@ class TreeWindowBase(QMdiSubWindow):
                 self.view.setSelectionMode(QAbstractItemView.ContiguousSelection)
                 self.view.CopyCellsToClipboard = CopyTreeCellsToClipboard
 
+                self.context_menu = TreeContextMenu(self.view)
+
         def DisplayFound(self, ids):
                 if not len(ids):
                         return False
@@ -1660,6 +1662,8 @@ class BranchWindow(QMdiSubWindow):
 
                 self.ResizeColumnsToContents()
 
+                self.context_menu = TreeContextMenu(self.view)
+
                 self.find_bar = FindBar(self, self, True)
 
                 self.finder = ChildDataItemFinder(self.model.root)
@@ -2469,6 +2473,39 @@ def CopyCellsToClipboardHdr(view):
 def CopyCellsToClipboardCSV(view):
         CopyCellsToClipboard(view, True, True)
 
+# Context menu
+
+class ContextMenu(object):
+
+        def __init__(self, view):
+                self.view = view
+                self.view.setContextMenuPolicy(Qt.CustomContextMenu)
+                self.view.customContextMenuRequested.connect(self.ShowContextMenu)
+
+        def ShowContextMenu(self, pos):
+                menu = QMenu(self.view)
+                self.AddActions(menu)
+                menu.exec_(self.view.mapToGlobal(pos))
+
+        def AddCopy(self, menu):
+                menu.addAction(CreateAction("&Copy selection", "Copy to clipboard", lambda: CopyCellsToClipboardHdr(self.view), self.view))
+                menu.addAction(CreateAction("Copy selection as CS&V", "Copy to clipboard as CSV", lambda: CopyCellsToClipboardCSV(self.view), self.view))
+
+        def AddActions(self, menu):
+                self.AddCopy(menu)
+
+class TreeContextMenu(ContextMenu):
+
+        def __init__(self, view):
+                super(TreeContextMenu, self).__init__(view)
+
+        def AddActions(self, menu):
+                i = self.view.currentIndex()
+                text = str(i.data()).strip()
+                if len(text):
+                        menu.addAction(CreateAction('Copy "' + text + '"', "Copy to clipboard", lambda: QApplication.clipboard().setText(text), self.view))
+                self.AddCopy(menu)
+
 # Table window
 
 class TableWindow(QMdiSubWindow, ResizeColumnsToContentsBase):
@@ -2492,6 +2529,8 @@ class TableWindow(QMdiSubWindow, ResizeColumnsToContentsBase):
 
                 self.ResizeColumnsToContents()
 
+                self.context_menu = ContextMenu(self.view)
+
                 self.find_bar = FindBar(self, self, True)
 
                 self.finder = ChildDataItemFinder(self.data_model)
@@ -2608,6 +2647,8 @@ class TopCallsWindow(QMdiSubWindow, ResizeColumnsToContentsBase):
                 self.view.setSelectionMode(QAbstractItemView.ContiguousSelection)
                 self.view.CopyCellsToClipboard = CopyTableCellsToClipboard
 
+                self.context_menu = ContextMenu(self.view)
+
                 self.ResizeColumnsToContents()
 
                 self.find_bar = FindBar(self, self, True)