mirror of
https://gitlab.gnome.org/GNOME/gimp.git
synced 2025-07-04 09:53:25 +00:00
3f630378b0
In the performance-log viewer, add an option to filter which threads, and which states of each thread, are included in the profile. By default, all threads in the RUNNING state are included.
2598 lines
80 KiB
Python
Executable file
2598 lines
80 KiB
Python
Executable file
#!/usr/bin/env python3
|
||
|
||
"""
|
||
performance-log-viewer.py -- GIMP performance log viewer
|
||
Copyright (C) 2018 Ell
|
||
|
||
This program is free software: you can redistribute it and/or modify
|
||
it under the terms of the GNU General Public License as published by
|
||
the Free Software Foundation; either version 3 of the License, or
|
||
(at your option) any later version.
|
||
|
||
This program is distributed in the hope that 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.
|
||
|
||
You should have received a copy of the GNU General Public License
|
||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
|
||
Usage: performance-log-viewer.py < infile
|
||
"""
|
||
|
||
import builtins, sys, math, statistics, functools, enum, re
|
||
|
||
from collections import namedtuple
|
||
from xml.etree import ElementTree
|
||
|
||
import gi
|
||
gi.require_version ("Gdk", "3.0")
|
||
gi.require_version ("Gtk", "3.0")
|
||
from gi.repository import GLib, GObject, Gio, Gdk, Gtk, Pango
|
||
|
||
def compose (head = None, *tail):
|
||
return (
|
||
lambda *args, **kwargs: head (compose (*tail) (*args, **kwargs))
|
||
) if tail else head or (lambda x: x)
|
||
|
||
def div (x, y):
|
||
return x / y if y else \
|
||
+math.inf if x > 0 else \
|
||
-math.inf if x < 0 else \
|
||
None
|
||
|
||
def format_float (x):
|
||
return "%g" % (round (100 * x) / 100)
|
||
|
||
def format_percentage (x, digits = 0):
|
||
scale = 10 ** digits
|
||
|
||
return "%g%%" % (round (100 * scale * x) / scale)
|
||
|
||
def format_size (size):
|
||
return GLib.format_size_full (size, GLib.FormatSizeFlags.IEC_UNITS)
|
||
|
||
def format_duration (t):
|
||
return "%02d:%02d:%02d.%02d" % (int (t / 3600),
|
||
int (t / 60) % 60,
|
||
int (t % 60),
|
||
round (100 * t) % 100)
|
||
|
||
def format_color (color):
|
||
return "#%02x%02x%02x" % tuple (
|
||
map (lambda x: min (max (round (255 * x), 0), 255), color)
|
||
)
|
||
|
||
def is_bright_color (color):
|
||
return max (tuple (color)[0:3]) > 0.5
|
||
|
||
def get_basename (path):
|
||
match = re.fullmatch (".*[\\\\/](.+?)[\\\\/]?", path)
|
||
|
||
return match[1] if match else path
|
||
|
||
VariableType = namedtuple ("VariableType",
|
||
("parse", "format", "format_numeric"))
|
||
|
||
var_types = {
|
||
"boolean": VariableType (
|
||
parse = int,
|
||
format = compose (str, bool),
|
||
format_numeric = format_float
|
||
),
|
||
|
||
"integer": VariableType (
|
||
parse = int,
|
||
format = format_float,
|
||
format_numeric = None
|
||
),
|
||
|
||
"size": VariableType (
|
||
parse = int,
|
||
format = format_size,
|
||
format_numeric = None
|
||
),
|
||
|
||
"size-ratio": VariableType (
|
||
parse = lambda x: div (*map (int, x.split ("/"))),
|
||
format = format_percentage,
|
||
format_numeric = None
|
||
),
|
||
|
||
"int-ratio": VariableType (
|
||
parse = lambda x: div (*map (int, x.split (":"))),
|
||
format = lambda x: "%g:%g" % (
|
||
(1, 0) if math.isinf (x) else
|
||
(0, 1) if x == 0 else
|
||
(round (100 * x) / 100, 1) if x > 1 else
|
||
(1, round (100 / x) / 100)
|
||
),
|
||
format_numeric = None
|
||
),
|
||
|
||
"percentage": VariableType (
|
||
parse = float,
|
||
format = format_percentage,
|
||
format_numeric = None
|
||
),
|
||
|
||
"duration": VariableType (
|
||
parse = float,
|
||
format = format_duration,
|
||
format_numeric = None
|
||
),
|
||
|
||
"rate-of-change": VariableType (
|
||
parse = float,
|
||
format = lambda x: "%s/s" % format_size (x),
|
||
format_numeric = None
|
||
)
|
||
}
|
||
var_types = {
|
||
type: VariableType (
|
||
parse = parse,
|
||
format = lambda x, f = format: \
|
||
f (x) if x is not None else "N/A",
|
||
format_numeric = lambda x, f = format_numeric or format:
|
||
f (x) if x is not None else "N/A"
|
||
)
|
||
for type, (parse, format, format_numeric) in var_types.items ()
|
||
}
|
||
|
||
# Read performance log from STDIN
|
||
log = ElementTree.fromstring (sys.stdin.buffer.read ())
|
||
|
||
Variable = namedtuple ("Variable", ("type", "color"))
|
||
Value = namedtuple ("Value", ("value", "raw"))
|
||
|
||
var_colors = [
|
||
(0.8, 0.4, 0.4),
|
||
(0.8, 0.6, 0.4),
|
||
(0.4, 0.8, 0.4),
|
||
(0.8, 0.8, 0.4),
|
||
(0.4, 0.4, 0.8),
|
||
(0.4, 0.8, 0.8),
|
||
(0.8, 0.4, 0.8),
|
||
(0.8, 0.8, 0.8)
|
||
]
|
||
|
||
var_defs = {}
|
||
|
||
for var in log.find ("var-defs"):
|
||
color = var_colors[len (var_defs) % len (var_colors)]
|
||
|
||
var_defs[var.get ("name")] = Variable (var.get ("type"), color)
|
||
|
||
AddressInfo = namedtuple ("AddressInfo", ("id",
|
||
"name",
|
||
"object",
|
||
"symbol",
|
||
"offset",
|
||
"source",
|
||
"line"))
|
||
|
||
address_map = {}
|
||
|
||
if log.find ("address-map"):
|
||
for address in log.find ("address-map").iterfind ("address"):
|
||
value = int (address.get ("value"), 0)
|
||
|
||
object = address.find ("object").text
|
||
symbol = address.find ("symbol").text
|
||
base = address.find ("base").text
|
||
source = address.find ("source").text
|
||
line = address.find ("line").text
|
||
|
||
address_map[value] = AddressInfo (
|
||
id = int (base, 0) if base else value,
|
||
name = symbol or base or hex (value),
|
||
object = object,
|
||
symbol = symbol,
|
||
offset = value - int (base, 0) if base else None,
|
||
source = source,
|
||
line = int (line) if line else None
|
||
)
|
||
|
||
class ThreadState (enum.Enum):
|
||
SUSPENDED = enum.auto ()
|
||
RUNNING = enum.auto ()
|
||
|
||
def __str__ (self):
|
||
return {
|
||
ThreadState.SUSPENDED: "S",
|
||
ThreadState.RUNNING: "R"
|
||
}[self]
|
||
|
||
Thread = namedtuple ("Thread", ("id", "name", "state", "frames"))
|
||
Frame = namedtuple ("Frame", ("id", "address", "info"))
|
||
|
||
Sample = namedtuple ("Sample", ("t", "vars", "markers", "backtrace"))
|
||
Marker = namedtuple ("Marker", ("id", "t", "description"))
|
||
|
||
samples = []
|
||
markers = []
|
||
|
||
for element in log.find ("samples"):
|
||
if element.tag == "sample":
|
||
sample = Sample (
|
||
t = int (element.get ("t")),
|
||
vars = {},
|
||
markers = markers,
|
||
backtrace = []
|
||
)
|
||
|
||
for var in element.find ("vars"):
|
||
sample.vars[var.tag] = Value (
|
||
value = var_types[var_defs[var.tag].type].parse (var.text) \
|
||
if var.text else None,
|
||
raw = var.text.strip () if var.text else None
|
||
)
|
||
|
||
if element.find ("backtrace"):
|
||
for thread in element.find ("backtrace").iterfind ("thread"):
|
||
id = thread.get ("id")
|
||
name = thread.get ("name")
|
||
running = thread.get ("running")
|
||
|
||
t = Thread (
|
||
id = int (id),
|
||
name = name,
|
||
state = ThreadState.RUNNING if running and int (running) \
|
||
else ThreadState.SUSPENDED,
|
||
frames = []
|
||
)
|
||
|
||
for frame in thread.iterfind ("frame"):
|
||
address = int (frame.get ("address"), 0)
|
||
|
||
info = address_map.get (address, None)
|
||
|
||
if not info:
|
||
info = AddressInfo (
|
||
id = address,
|
||
name = hex (address),
|
||
object = None,
|
||
symbol = None,
|
||
offset = None,
|
||
source = None,
|
||
line = None
|
||
)
|
||
|
||
t.frames.append (Frame (
|
||
id = len (t.frames),
|
||
address = address,
|
||
info = info
|
||
))
|
||
|
||
sample.backtrace.append (t)
|
||
|
||
samples.append (sample)
|
||
|
||
markers = []
|
||
elif element.tag == "marker":
|
||
marker = Marker (
|
||
id = int (element.get ("id")),
|
||
t = int (element.get ("t")),
|
||
description = element.text.strip () if element.text else None
|
||
)
|
||
|
||
markers.append (marker)
|
||
|
||
if samples and markers:
|
||
samples[-1].markers += markers
|
||
|
||
markers = None
|
||
|
||
class SelectionOp (enum.Enum):
|
||
REPLACE = enum.auto ()
|
||
ADD = enum.auto ()
|
||
SUBTRACT = enum.auto ()
|
||
INTERSECT = enum.auto ()
|
||
XOR = enum.auto ()
|
||
|
||
class Selection (GObject.GObject):
|
||
__gsignals__ = {
|
||
"changed": (GObject.SIGNAL_RUN_FIRST, None, ()),
|
||
"change-complete": (GObject.SIGNAL_RUN_FIRST, None, ()),
|
||
"highlight-changed": (GObject.SIGNAL_RUN_FIRST, None, ())
|
||
}
|
||
|
||
def __init__ (self, iter = ()):
|
||
GObject.GObject.__init__ (self)
|
||
|
||
self.selection = set (iter)
|
||
self.highlight = None
|
||
self.cursor = None
|
||
self.cursor_dir = 0
|
||
|
||
self.pending_change_completion = False
|
||
|
||
def __str__ (self):
|
||
n_sel = len (self.selection)
|
||
|
||
if n_sel == 0 or n_sel == len (samples):
|
||
return "All Samples"
|
||
elif n_sel == 1:
|
||
i, = self.selection
|
||
|
||
return "Sample %d" % i
|
||
else:
|
||
sel = list (self.selection)
|
||
|
||
sel.sort ()
|
||
|
||
if all (sel[i] + 1 == sel[i + 1] for i in range (n_sel - 1)):
|
||
return "Samples %d–%d" % (sel[0], sel[-1])
|
||
else:
|
||
return "%d Samples" % n_sel
|
||
|
||
def copy (self):
|
||
selection = Selection ()
|
||
|
||
selection.highlight = self.highlight
|
||
selection.cursor = self.cursor
|
||
selection.cursor_dir = self.cursor_dir
|
||
selection.selection = self.selection.copy ()
|
||
|
||
return selection
|
||
|
||
def get_effective_selection (self):
|
||
if self.selection:
|
||
return self.selection
|
||
else:
|
||
return set (range (len (samples)))
|
||
|
||
def select (self, selection, op = SelectionOp.REPLACE):
|
||
if op == SelectionOp.REPLACE:
|
||
self.selection = selection.copy ()
|
||
elif op == SelectionOp.ADD:
|
||
self.selection |= selection
|
||
elif op == SelectionOp.SUBTRACT:
|
||
self.selection -= selection
|
||
elif op == SelectionOp.INTERSECT:
|
||
self.selection &= selection
|
||
elif op == SelectionOp.XOR:
|
||
self.selection.symmetric_difference_update (selection)
|
||
|
||
if len (self.selection) == 1:
|
||
(self.cursor,) = self.selection
|
||
else:
|
||
self.cursor = None
|
||
|
||
self.cursor_dir = 0
|
||
|
||
self.pending_change_completion = True
|
||
|
||
self.emit ("changed")
|
||
|
||
def select_range (self, first, last, op = SelectionOp.REPLACE):
|
||
if first > last:
|
||
temp = first
|
||
first = last
|
||
last = temp
|
||
|
||
first = max (first, 0)
|
||
last = min (last, len (samples) - 1)
|
||
|
||
if first <= last:
|
||
self.select (set (range (first, last + 1)), op)
|
||
else:
|
||
self.select (set (), op)
|
||
|
||
def change_complete (self):
|
||
if self.pending_change_completion:
|
||
self.pending_change_completion = False
|
||
|
||
self.emit ("change-complete")
|
||
|
||
def set_highlight (self, highlight):
|
||
self.highlight = highlight
|
||
|
||
self.emit ("highlight-changed")
|
||
|
||
selection = Selection ()
|
||
|
||
class FindSamplesPopover (Gtk.Popover):
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.Popover.__init__ (self, *args, **kwargs)
|
||
|
||
vbox = Gtk.Box (orientation = Gtk.Orientation.VERTICAL,
|
||
border_width = 20,
|
||
spacing = 8)
|
||
self.add (vbox)
|
||
vbox.show ()
|
||
|
||
entry = Gtk.Entry (width_chars = 40,
|
||
placeholder_text = "Python expression")
|
||
self.entry = entry
|
||
vbox.pack_start (entry, False, False, 0)
|
||
entry.show ()
|
||
|
||
entry.connect ("activate", self.find_samples)
|
||
|
||
entry.get_buffer ().connect (
|
||
"notify::text",
|
||
lambda *args: self.entry.get_style_context ().remove_class ("error")
|
||
)
|
||
|
||
frame = Gtk.Frame (label = "Selection",
|
||
shadow_type = Gtk.ShadowType.NONE)
|
||
vbox.pack_start (frame, False, False, 8)
|
||
frame.get_label_widget ().get_style_context ().add_class ("dim-label")
|
||
frame.show ()
|
||
|
||
vbox2 = Gtk.Box (orientation = Gtk.Orientation.VERTICAL,
|
||
border_width = 8,
|
||
spacing = 8)
|
||
frame.add (vbox2)
|
||
vbox2.show ()
|
||
|
||
self.radios = []
|
||
|
||
radio = Gtk.RadioButton.new_with_mnemonic (None, "_Replace")
|
||
self.radios.append ((radio, SelectionOp.REPLACE))
|
||
vbox2.pack_start (radio, False, False, 0)
|
||
radio.show ()
|
||
|
||
radio = Gtk.RadioButton.new_with_mnemonic_from_widget (radio, "_Add")
|
||
self.radios.append ((radio, SelectionOp.ADD))
|
||
vbox2.pack_start (radio, False, False, 0)
|
||
radio.show ()
|
||
|
||
radio = Gtk.RadioButton.new_with_mnemonic_from_widget (radio, "_Subtract")
|
||
self.radios.append ((radio, SelectionOp.SUBTRACT))
|
||
vbox2.pack_start (radio, False, False, 0)
|
||
radio.show ()
|
||
|
||
radio = Gtk.RadioButton.new_with_mnemonic_from_widget (radio, "_Intersect")
|
||
self.radios.append ((radio, SelectionOp.INTERSECT))
|
||
vbox2.pack_start (radio, False, False, 0)
|
||
radio.show ()
|
||
|
||
button = Gtk.Button.new_with_mnemonic ("_Find")
|
||
vbox.pack_start (button, False, False, 0)
|
||
button.set_halign (Gtk.Align.CENTER)
|
||
button.show ()
|
||
|
||
button.connect ("clicked", self.find_samples)
|
||
|
||
def do_hide (self):
|
||
self.entry.set_text ("")
|
||
self.entry.get_style_context ().remove_class ("error")
|
||
|
||
Gtk.Popover.do_hide (self)
|
||
|
||
def find_samples (self, *args):
|
||
def var_name (var):
|
||
return var.replace ("-", "_")
|
||
|
||
try:
|
||
f = eval ("lambda thread, function, %s: %s" % (
|
||
", ".join (map (var_name, var_defs)),
|
||
self.entry.get_text ()))
|
||
except:
|
||
self.entry.get_style_context ().add_class ("error")
|
||
|
||
return
|
||
|
||
sel = set ()
|
||
|
||
for i in range (len (samples)):
|
||
try:
|
||
def thread (id, state = None):
|
||
for thread in samples[i].backtrace or []:
|
||
if (type (id) == int and id == thread.id) or \
|
||
(type (id) == str and thread.name and \
|
||
re.fullmatch (id, thread.name)):
|
||
if state is None or \
|
||
re.fullmatch (state, str (thread.state)):
|
||
return True
|
||
|
||
return False
|
||
|
||
def function (name, exclusive = False):
|
||
for thread in samples[i].backtrace or []:
|
||
if exclusive:
|
||
if re.fullmatch (name, thread.frames[0].info.name):
|
||
return True
|
||
else:
|
||
for frame in thread.frames:
|
||
if re.fullmatch (name, frame.info.name):
|
||
return True
|
||
|
||
return False
|
||
|
||
if f (thread, function, **{
|
||
var_name (var): value.value
|
||
for var, value in samples[i].vars.items ()
|
||
}):
|
||
sel.add (i)
|
||
except:
|
||
pass
|
||
|
||
op = [op for radio, op in self.radios if radio.get_active ()][0]
|
||
|
||
selection.select (sel, op)
|
||
|
||
selection.change_complete ()
|
||
|
||
self.hide ()
|
||
|
||
class CellRendererColorToggle (Gtk.CellRendererToggle):
|
||
padding = 3
|
||
|
||
color = GObject.Property (type = Gdk.RGBA, default = Gdk.RGBA (0, 0, 0))
|
||
|
||
def do_render (self, cr, widget, background_area, cell_area, flags):
|
||
state = widget.get_state ()
|
||
style = widget.get_style_context ()
|
||
bg_color = style.get_background_color (state)
|
||
active = self.get_property ("active")
|
||
size = max (min (cell_area.width, cell_area.height) -
|
||
2 * self.padding,
|
||
0)
|
||
|
||
(r, g, b, a) = self.color
|
||
|
||
if is_bright_color (bg_color):
|
||
bg = (r, g, b)
|
||
fg = (0.75 * r, 0.75 * g, 0.75 * b)
|
||
else:
|
||
bg = (0.75 * r, 0.75 * g, 0.75 * b)
|
||
fg = (r, g, b)
|
||
|
||
x = cell_area.x + (cell_area.width - size) // 2
|
||
y = cell_area.y + (cell_area.height - size) // 2
|
||
|
||
cr.rectangle (x, y, size, size)
|
||
|
||
if active:
|
||
cr.set_source_rgba (*bg)
|
||
else:
|
||
Gdk.cairo_set_source_rgba (cr, bg_color)
|
||
|
||
cr.fill_preserve ()
|
||
|
||
cr.set_source_rgb (*fg)
|
||
cr.set_line_width (2)
|
||
cr.stroke ()
|
||
|
||
class VariableSet (Gtk.TreeView):
|
||
class Store (Gtk.ListStore):
|
||
NAME = 0
|
||
ACTIVE = 1
|
||
COLOR = 2
|
||
|
||
def __init__ (self):
|
||
Gtk.ListStore.__init__ (self, str, bool, Gdk.RGBA)
|
||
|
||
for var in var_defs:
|
||
i = self.append ((var, False, Gdk.RGBA (*var_defs[var].color)))
|
||
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.TreeView.__init__ (self, *args, headers_visible = False, **kwargs)
|
||
|
||
store = self.Store ()
|
||
self.store = store
|
||
self.set_model (store)
|
||
|
||
col = Gtk.TreeViewColumn ()
|
||
self.append_column (col)
|
||
|
||
cell = CellRendererColorToggle ()
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "active", store.ACTIVE)
|
||
col.add_attribute (cell, "color", store.COLOR)
|
||
|
||
cell.connect ("toggled", self.var_toggled)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, True)
|
||
col.add_attribute (cell, "text", store.NAME)
|
||
|
||
def var_toggled (self, cell, path):
|
||
self.store[path][self.store.ACTIVE] = not cell.get_property ("active")
|
||
|
||
class SampleGraph (Gtk.DrawingArea):
|
||
def __init__ (self, model = None, *args, **kwargs):
|
||
Gtk.DrawingArea.__init__ (self, *args, can_focus = True, **kwargs)
|
||
|
||
self.style_widget = Gtk.Entry ()
|
||
|
||
self.model = model
|
||
|
||
if model:
|
||
model.connect ("row-changed", lambda *args: self.update ())
|
||
|
||
self.update ()
|
||
|
||
self.selection = None
|
||
self.sel = None
|
||
|
||
selection.connect ("changed", self.selection_changed)
|
||
selection.connect ("highlight-changed",
|
||
lambda selection: self.queue_draw ())
|
||
|
||
self.add_events (Gdk.EventMask.BUTTON_PRESS_MASK |
|
||
Gdk.EventMask.BUTTON_RELEASE_MASK |
|
||
Gdk.EventMask.KEY_PRESS_MASK |
|
||
Gdk.EventMask.KEY_RELEASE_MASK)
|
||
|
||
self.selection_changed (selection)
|
||
|
||
def sample_to_x (self, i):
|
||
if not samples:
|
||
return None
|
||
|
||
width = self.get_allocated_width ()
|
||
n_samples = max (len (samples), 2)
|
||
|
||
return 1 + (width - 3) * i / (n_samples - 1)
|
||
|
||
def sample_to_range (self, i):
|
||
if not samples:
|
||
return None
|
||
|
||
width = self.get_allocated_width ()
|
||
n_samples = max (len (samples), 2)
|
||
|
||
return (1 + math.floor ((width - 3) * (i - 0.5) / (n_samples - 1)),
|
||
1 + math.ceil ((width - 3) * (i + 0.5) / (n_samples - 1)))
|
||
|
||
def x_to_sample (self, x):
|
||
if not samples:
|
||
return None
|
||
|
||
width = max (self.get_allocated_width (), 4)
|
||
n_samples = len (samples)
|
||
|
||
return round ((n_samples - 1) * (x - 1) / (width - 3))
|
||
|
||
def update (self):
|
||
if not samples or not self.model:
|
||
return
|
||
|
||
self.max_value = 1
|
||
|
||
for row in self.model:
|
||
var_name = row[self.model.NAME]
|
||
var_active = row[self.model.ACTIVE]
|
||
|
||
if var_active:
|
||
self.max_value = max (
|
||
self.max_value,
|
||
max (sample.vars[var_name].value or 0 for sample in samples)
|
||
)
|
||
|
||
self.queue_draw ()
|
||
|
||
def selection_changed (self, selection):
|
||
if selection.selection:
|
||
self.sel = list (selection.selection)
|
||
self.sel.sort ()
|
||
else:
|
||
self.sel = None
|
||
|
||
self.queue_draw ()
|
||
|
||
def do_get_preferred_width (self):
|
||
return (300, 300)
|
||
|
||
def do_get_preferred_height (self):
|
||
if self.model:
|
||
return (32, 256)
|
||
else:
|
||
return (16, 16)
|
||
|
||
def update_selection (self):
|
||
sel = self.selection.copy ()
|
||
|
||
i0 = self.selection_i0
|
||
i1 = self.selection_i1
|
||
|
||
if self.selection_range:
|
||
swap = i0 > i1
|
||
|
||
if swap:
|
||
temp = i0
|
||
i0 = i1
|
||
i1 = temp
|
||
|
||
n_samples = len (samples)
|
||
|
||
while i0 > 0 and not samples[i0 - 1].markers: i0 -= 1
|
||
while i1 < n_samples - 1 and not samples[i1 + 1].markers: i1 += 1
|
||
|
||
if swap:
|
||
temp = i0
|
||
i0 = i1
|
||
i1 = temp
|
||
|
||
sel.select_range (i0, i1, self.selection_op)
|
||
|
||
selection.select (sel.selection)
|
||
|
||
selection.cursor = i1
|
||
selection.cursor_dir = i1 - i0
|
||
|
||
def do_button_press_event (self, event):
|
||
self.grab_focus ()
|
||
|
||
if event.button != 1:
|
||
return False
|
||
|
||
i = self.x_to_sample (event.x)
|
||
|
||
if i is None:
|
||
return False
|
||
|
||
self.selection = selection.copy ()
|
||
self.selection_i0 = i
|
||
self.selection_i1 = i
|
||
self.selection_op = SelectionOp.REPLACE
|
||
self.selection_range = event.type != Gdk.EventType.BUTTON_PRESS
|
||
|
||
state = event.state & Gdk.ModifierType.MODIFIER_MASK
|
||
|
||
if state == Gdk.ModifierType.SHIFT_MASK:
|
||
self.selection_op = SelectionOp.ADD
|
||
elif state == Gdk.ModifierType.CONTROL_MASK:
|
||
self.selection_op = SelectionOp.SUBTRACT
|
||
elif state == (Gdk.ModifierType.SHIFT_MASK |
|
||
Gdk.ModifierType.CONTROL_MASK):
|
||
self.selection_op = SelectionOp.INTERSECT
|
||
|
||
self.update_selection ()
|
||
|
||
self.grab_add ()
|
||
|
||
return True
|
||
|
||
def do_button_release_event (self, event):
|
||
if event.button != 1:
|
||
return False
|
||
|
||
if self.selection is None:
|
||
return False
|
||
|
||
self.selection = None
|
||
|
||
self.grab_remove ()
|
||
|
||
selection.change_complete ()
|
||
|
||
return True
|
||
|
||
def do_motion_notify_event (self, event):
|
||
i = self.x_to_sample (event.x)
|
||
|
||
selection.set_highlight (i)
|
||
|
||
if self.selection and i is not None:
|
||
self.selection_i1 = i
|
||
|
||
self.update_selection ()
|
||
|
||
return True
|
||
|
||
return False
|
||
|
||
def do_leave_notify_event (self, evnet):
|
||
selection.set_highlight (None)
|
||
|
||
return False
|
||
|
||
def do_key_press_event (self, event):
|
||
if event.keyval == Gdk.KEY_Left or \
|
||
event.keyval == Gdk.KEY_Right or \
|
||
event.keyval == Gdk.KEY_Home or \
|
||
event.keyval == Gdk.KEY_KP_Home or \
|
||
event.keyval == Gdk.KEY_End or \
|
||
event.keyval == Gdk.KEY_KP_End:
|
||
n_samples = len (samples)
|
||
|
||
state = event.state & Gdk.ModifierType.MODIFIER_MASK
|
||
|
||
op = SelectionOp.REPLACE
|
||
|
||
if state == Gdk.ModifierType.SHIFT_MASK:
|
||
op = SelectionOp.XOR
|
||
|
||
cursor = selection.cursor
|
||
cursor_dir = selection.cursor_dir
|
||
|
||
if event.keyval == Gdk.KEY_Left or \
|
||
event.keyval == Gdk.KEY_Home or \
|
||
event.keyval == Gdk.KEY_KP_Home:
|
||
if selection.cursor is not None:
|
||
if cursor_dir <= 0 or op == SelectionOp.REPLACE:
|
||
cursor -= 1
|
||
else:
|
||
cursor = n_samples - 1
|
||
|
||
cursor_dir = -1
|
||
elif event.keyval == Gdk.KEY_Right or \
|
||
event.keyval == Gdk.KEY_End or \
|
||
event.keyval == Gdk.KEY_KP_End:
|
||
if cursor is not None:
|
||
if cursor_dir >= 0 or op == SelectionOp.REPLACE:
|
||
cursor += 1
|
||
else:
|
||
cursor = 0
|
||
|
||
cursor_dir = +1
|
||
|
||
if cursor < 0 or cursor >= n_samples:
|
||
cursor = min (max (cursor, 0), n_samples - 1)
|
||
|
||
selection.cursor = cursor
|
||
selection.cursor_dir = cursor_dir
|
||
|
||
if op != SelectionOp.REPLACE:
|
||
return True
|
||
|
||
i0 = cursor
|
||
|
||
if event.keyval == Gdk.KEY_Home or \
|
||
event.keyval == Gdk.KEY_KP_Home:
|
||
cursor = 0
|
||
elif event.keyval == Gdk.KEY_End or \
|
||
event.keyval == Gdk.KEY_KP_End:
|
||
cursor = n_samples - 1
|
||
|
||
if op == SelectionOp.REPLACE:
|
||
i0 = cursor
|
||
|
||
selection.select_range (i0, cursor, op)
|
||
|
||
selection.cursor = cursor
|
||
selection.cursor_dir = cursor_dir
|
||
|
||
return True
|
||
elif event.keyval == Gdk.KEY_Escape:
|
||
selection.select (set ())
|
||
|
||
return True
|
||
|
||
return False
|
||
|
||
def do_key_release_event (self, event):
|
||
selection.change_complete ()
|
||
|
||
return False
|
||
|
||
def do_draw (self, cr):
|
||
state = self.get_state ()
|
||
style = (self.style_widget if self.model else
|
||
self).get_style_context ()
|
||
(width, height) = (self.get_allocated_width (),
|
||
self.get_allocated_height ())
|
||
|
||
fg_color = tuple (style.get_color (state))
|
||
grid_color = (*fg_color[:3], 0.25 * fg_color[3])
|
||
highlight_color = grid_color
|
||
selection_color = (*fg_color[:3], 0.15 * fg_color[3])
|
||
|
||
Gtk.render_background (style, cr, 0, 0, width, height)
|
||
|
||
if self.model:
|
||
vscale = (height - 4) / self.max_value
|
||
|
||
cr.save ()
|
||
|
||
cr.translate (0, height - 2)
|
||
cr.scale (1, -1)
|
||
|
||
first_sample = True
|
||
|
||
for row in self.model:
|
||
var_name = row[self.model.NAME]
|
||
var_active = row[self.model.ACTIVE]
|
||
|
||
if var_active:
|
||
is_boolean = var_defs[var_name].type == "boolean"
|
||
is_continuous = not is_boolean
|
||
|
||
for i in range (len (samples)):
|
||
value = samples[i].vars[var_name].value
|
||
|
||
if value is not None:
|
||
if is_boolean:
|
||
value *= self.max_value
|
||
|
||
y = value * vscale
|
||
|
||
if is_continuous:
|
||
x = self.sample_to_x (i)
|
||
|
||
if first_sample:
|
||
cr.move_to (x, y)
|
||
else:
|
||
cr.line_to (x, y)
|
||
else:
|
||
(x0, x1) = self.sample_to_range (i)
|
||
|
||
if first_sample:
|
||
cr.move_to (x0, y)
|
||
else:
|
||
cr.line_to (x0, y)
|
||
|
||
cr.line_to (x1, y)
|
||
|
||
first_sample = False
|
||
else:
|
||
first_sample = True
|
||
|
||
(r, g, b) = var_defs[var_name].color
|
||
|
||
cr.set_source_rgb (r, g, b)
|
||
cr.set_line_width (2)
|
||
cr.stroke ()
|
||
|
||
cr.restore ()
|
||
|
||
cr.set_line_width (1)
|
||
|
||
cr.set_source_rgba (*grid_color)
|
||
|
||
n_hgrid_lines = 4
|
||
n_vgrid_lines = 1
|
||
|
||
for i in range (n_hgrid_lines + 1):
|
||
cr.move_to (0, round (i * (height - 1) / n_hgrid_lines) + 0.5)
|
||
cr.rel_line_to (width, 0)
|
||
|
||
cr.stroke ()
|
||
|
||
for i in range (n_vgrid_lines + 1):
|
||
cr.move_to (round (i * (width - 1) / n_vgrid_lines) + 0.5, 0)
|
||
cr.rel_line_to (0, height)
|
||
|
||
cr.stroke ()
|
||
else:
|
||
for i in range (len (samples)):
|
||
if samples[i].markers:
|
||
(x0, x1) = self.sample_to_range (i)
|
||
|
||
cr.rectangle (x0, 0, x1 - x0, height)
|
||
|
||
cr.set_source_rgba (*fg_color)
|
||
cr.fill ()
|
||
|
||
if selection.highlight is not None:
|
||
(x0, x1) = self.sample_to_range (selection.highlight)
|
||
|
||
cr.rectangle (x0, 0, x1 - x0, height)
|
||
|
||
(r, g, b, a) = style.get_color (state)
|
||
|
||
cr.set_source_rgba (*highlight_color)
|
||
cr.fill ()
|
||
|
||
if self.sel:
|
||
def draw_selection ():
|
||
x0 = self.sample_to_range (i0)[0]
|
||
x1 = self.sample_to_range (i1)[1]
|
||
|
||
cr.rectangle (x0, 0, x1 - x0, height)
|
||
|
||
(r, g, b, a) = style.get_color (state)
|
||
|
||
cr.set_source_rgba (*selection_color)
|
||
cr.fill ()
|
||
|
||
i0 = None
|
||
|
||
for i in self.sel:
|
||
if i0 is None:
|
||
i0 = i
|
||
i1 = i
|
||
elif i == i1 + 1:
|
||
i1 = i
|
||
else:
|
||
draw_selection ()
|
||
|
||
i0 = i
|
||
i1 = i
|
||
|
||
if i0 is not None:
|
||
draw_selection ()
|
||
|
||
class SampleGraphList (Gtk.Box):
|
||
Item = namedtuple (
|
||
"SampleGraphListGraph", ("widget",
|
||
"model",
|
||
"remove_button",
|
||
"move_up_button",
|
||
"move_down_button")
|
||
)
|
||
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.Box.__init__ (self,
|
||
*args,
|
||
orientation = Gtk.Orientation.VERTICAL,
|
||
**kwargs)
|
||
|
||
self.items = []
|
||
|
||
self.vset_size_group = Gtk.SizeGroup (
|
||
mode = Gtk.SizeGroupMode.HORIZONTAL
|
||
)
|
||
|
||
hbox = Gtk.Box (orientation = Gtk.Orientation.HORIZONTAL)
|
||
self.pack_start (hbox, False, False, 0)
|
||
hbox.show ()
|
||
|
||
empty = Gtk.DrawingArea ()
|
||
hbox.pack_start (empty, False, True, 0)
|
||
self.vset_size_group.add_widget (empty)
|
||
empty.show ()
|
||
|
||
graph = SampleGraph (has_tooltip = True)
|
||
hbox.pack_start (graph, True, True, 0)
|
||
graph.show ()
|
||
|
||
graph.connect ("query-tooltip", self.graph_query_tooltip)
|
||
|
||
separator = Gtk.Separator (orientation = Gtk.Orientation.HORIZONTAL)
|
||
self.pack_start (separator, False, False, 0)
|
||
separator.show ()
|
||
|
||
vbox = Gtk.Box (orientation = Gtk.Orientation.VERTICAL)
|
||
self.items_vbox = vbox
|
||
self.pack_start (vbox, False, False, 0)
|
||
vbox.show ()
|
||
|
||
self.add_item (0)
|
||
|
||
def update_items (self):
|
||
for widget in self.items_vbox.get_children ():
|
||
self.items_vbox.remove (widget)
|
||
|
||
i = 0
|
||
|
||
for item in self.items:
|
||
if i > 0:
|
||
separator = Gtk.Separator (
|
||
orientation = Gtk.Orientation.HORIZONTAL
|
||
)
|
||
self.items_vbox.pack_start (separator, False, False, 0)
|
||
separator.show ()
|
||
|
||
self.items_vbox.pack_start (item.widget, False, False, 0)
|
||
|
||
item.remove_button.set_sensitive (len (self.items) > 1)
|
||
item.move_up_button.set_sensitive (i > 0)
|
||
item.move_down_button.set_sensitive (i < len (self.items) - 1)
|
||
|
||
i += 1
|
||
|
||
def add_item (self, i):
|
||
hbox = Gtk.Box (orientation = Gtk.Orientation.HORIZONTAL)
|
||
hbox.show ()
|
||
|
||
vbox = Gtk.Box (orientation = Gtk.Orientation.VERTICAL)
|
||
hbox.pack_start (vbox, False, True, 0)
|
||
self.vset_size_group.add_widget (vbox)
|
||
vbox.show ()
|
||
|
||
scroll = Gtk.ScrolledWindow (
|
||
hscrollbar_policy = Gtk.PolicyType.NEVER,
|
||
vscrollbar_policy = Gtk.PolicyType.AUTOMATIC
|
||
)
|
||
vbox.pack_start (scroll, True, True, 0)
|
||
scroll.show ()
|
||
|
||
vset = VariableSet ()
|
||
scroll.add (vset)
|
||
vset.show ()
|
||
|
||
buttons = Gtk.ButtonBox (orientation = Gtk.Orientation.HORIZONTAL)
|
||
vbox.pack_start (buttons, False, False, 0)
|
||
buttons.set_layout (Gtk.ButtonBoxStyle.EXPAND)
|
||
buttons.show ()
|
||
|
||
button = Gtk.Button.new_from_icon_name ("list-add-symbolic",
|
||
Gtk.IconSize.BUTTON)
|
||
add_button = button
|
||
buttons.add (button)
|
||
button.show ()
|
||
|
||
button = Gtk.Button.new_from_icon_name ("list-remove-symbolic",
|
||
Gtk.IconSize.BUTTON)
|
||
remove_button = button
|
||
buttons.add (button)
|
||
button.show ()
|
||
|
||
button = Gtk.Button.new_from_icon_name ("go-up-symbolic",
|
||
Gtk.IconSize.BUTTON)
|
||
move_up_button = button
|
||
buttons.add (button)
|
||
button.show ()
|
||
|
||
button = Gtk.Button.new_from_icon_name ("go-down-symbolic",
|
||
Gtk.IconSize.BUTTON)
|
||
move_down_button = button
|
||
buttons.add (button)
|
||
button.show ()
|
||
|
||
graph = SampleGraph (vset.get_model (), has_tooltip = True)
|
||
hbox.pack_start (graph, True, True, 0)
|
||
graph.show ()
|
||
|
||
graph.connect ("query-tooltip", self.graph_query_tooltip)
|
||
|
||
item = self.Item (
|
||
widget = hbox,
|
||
model = vset.get_model (),
|
||
remove_button = remove_button,
|
||
move_up_button = move_up_button,
|
||
move_down_button = move_down_button
|
||
)
|
||
self.items.insert (i, item)
|
||
|
||
add_button.connect ("clicked",
|
||
lambda *args: self.add_item (
|
||
self.items.index (item) + 1
|
||
))
|
||
remove_button.connect ("clicked",
|
||
lambda *args: self.remove_item (
|
||
self.items.index (item)
|
||
))
|
||
move_up_button.connect ("clicked",
|
||
lambda *args: self.move_item (
|
||
self.items.index (item),
|
||
-1
|
||
))
|
||
move_down_button.connect ("clicked",
|
||
lambda *args: self.move_item (
|
||
self.items.index (item),
|
||
+1
|
||
))
|
||
|
||
self.update_items ()
|
||
|
||
def remove_item (self, i):
|
||
del self.items[i]
|
||
|
||
self.update_items ()
|
||
|
||
def move_item (self, i, offset):
|
||
item = self.items[i]
|
||
del self.items[i]
|
||
self.items.insert (i + offset, item)
|
||
|
||
self.update_items ()
|
||
|
||
def graph_query_tooltip (self, graph, x, y, keyboard_mode, tooltip):
|
||
if keyboard_mode:
|
||
return False
|
||
|
||
i = graph.x_to_sample (x)
|
||
|
||
if i is None or i < 0 or i >= len (samples):
|
||
return False
|
||
|
||
grid = Gtk.Grid (column_spacing = 4)
|
||
tooltip.set_custom (grid)
|
||
grid.show ()
|
||
|
||
row = 0
|
||
|
||
label = Gtk.Label ()
|
||
grid.attach (label, 0, row, 2, 1)
|
||
label.set_markup ("<b>Sample %d</b>" % i)
|
||
label.show ()
|
||
|
||
row += 1
|
||
|
||
label = Gtk.Label ()
|
||
grid.attach (label, 0, row, 2, 1)
|
||
label.set_markup ("<sub>%s</sub>" %
|
||
format_duration (samples[i].t / 1000000))
|
||
label.get_style_context ().add_class ("dim-label")
|
||
label.show ()
|
||
|
||
row += 1
|
||
|
||
for item in self.items:
|
||
model = item.model
|
||
|
||
vars = tuple (var[model.NAME] for var in model if var[model.ACTIVE])
|
||
|
||
if not vars:
|
||
continue
|
||
|
||
separator = Gtk.Separator (orientation = Gtk.Orientation.HORIZONTAL)
|
||
grid.attach (separator, 0, row, 2, 1)
|
||
separator.show ()
|
||
|
||
row += 1
|
||
|
||
for var in vars:
|
||
color = format_color (var_defs[var].color)
|
||
|
||
label = Gtk.Label (halign = Gtk.Align.START)
|
||
grid.attach (label, 0, row, 1, 1)
|
||
label.set_markup (
|
||
"<span color=\"%s\"><b>%s</b></span>" % (color, var)
|
||
)
|
||
label.show ()
|
||
|
||
value = samples[i].vars[var].value
|
||
text = var_types[var_defs[var].type].format (value) \
|
||
if value is not None else "N/A"
|
||
|
||
label = Gtk.Label (text, halign = Gtk.Align.END)
|
||
grid.attach (label, 1, row, 1, 1)
|
||
label.show ()
|
||
|
||
row += 1
|
||
|
||
markers = samples[i].markers
|
||
|
||
if markers:
|
||
separator = Gtk.Separator (orientation = Gtk.Orientation.HORIZONTAL)
|
||
grid.attach (separator, 0, row, 2, 1)
|
||
separator.show ()
|
||
|
||
row += 1
|
||
|
||
for marker in markers:
|
||
label = Gtk.Label (halign = Gtk.Align.START)
|
||
grid.attach (label, 0, row, 1, 1)
|
||
label.set_markup ("<b>Marker %d</b>" % (marker.id))
|
||
label.show ()
|
||
|
||
if marker.description:
|
||
label = Gtk.Label (marker.description,
|
||
halign = Gtk.Align.END)
|
||
grid.attach (label, 1, row, 1, 1)
|
||
label.show ()
|
||
|
||
row += 1
|
||
|
||
return True
|
||
|
||
class InformationViewer (Gtk.ScrolledWindow):
|
||
class Store (Gtk.ListStore):
|
||
NAME = 0
|
||
VALUE = 1
|
||
|
||
def __init__ (self):
|
||
Gtk.ListStore.__init__ (self, str, str)
|
||
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.ScrolledWindow.__init__ (
|
||
self,
|
||
*args,
|
||
hscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
vscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
**kwargs
|
||
)
|
||
|
||
vbox = Gtk.Box (orientation = Gtk.Orientation.VERTICAL,
|
||
border_width = 32,
|
||
margin_left = 64,
|
||
margin_right = 64,
|
||
spacing = 32)
|
||
self.add (vbox)
|
||
vbox.show ()
|
||
|
||
def add_element (element):
|
||
name = {
|
||
"params": "Log Parameters",
|
||
"gimp-version": "GIMP Version",
|
||
"env": "Environment",
|
||
"gegl-config": "GEGL Config"
|
||
}.get (element.tag, element.tag)
|
||
text = element.text.strip ()
|
||
n_items = len (element)
|
||
|
||
if not text and n_items == 0:
|
||
return
|
||
|
||
vbox2 = Gtk.Box (orientation = Gtk.Orientation.VERTICAL,
|
||
spacing = 16)
|
||
vbox.pack_start (vbox2, False, False, 0)
|
||
vbox2.show ()
|
||
|
||
label = Gtk.Label (xalign = 0)
|
||
vbox2.pack_start (label, False, False, 0)
|
||
label.set_markup ("<b>%s</b>" % name)
|
||
label.show ()
|
||
|
||
frame = Gtk.Frame (shadow_type = Gtk.ShadowType.IN)
|
||
vbox2.pack_start (frame, False, False, 0)
|
||
frame.show ()
|
||
|
||
if text:
|
||
scrolled = Gtk.ScrolledWindow (
|
||
hscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
vscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
height_request = 400
|
||
)
|
||
frame.add (scrolled)
|
||
scrolled.show ()
|
||
|
||
text = Gtk.TextView (editable = False,
|
||
monospace = True,
|
||
wrap_mode = Gtk.WrapMode.WORD,
|
||
left_margin = 16,
|
||
right_margin = 16,
|
||
top_margin = 16,
|
||
bottom_margin = 16)
|
||
scrolled.add (text)
|
||
text.get_buffer ().set_text (element.text.strip (), -1)
|
||
text.show ()
|
||
else:
|
||
scrolled = Gtk.ScrolledWindow (
|
||
hscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
vscrollbar_policy = Gtk.PolicyType.NEVER
|
||
)
|
||
frame.add (scrolled)
|
||
scrolled.show ()
|
||
|
||
store = self.Store ()
|
||
|
||
for item in element:
|
||
store.append ((item.tag, item.text.strip ()))
|
||
|
||
tree = Gtk.TreeView (model = store)
|
||
scrolled.add (tree)
|
||
tree.show ()
|
||
|
||
col = Gtk.TreeViewColumn (title = "Name")
|
||
tree.append_column (col)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", store.NAME)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Value")
|
||
tree.append_column (col)
|
||
col.set_alignment (0.5)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", store.VALUE)
|
||
|
||
params = log.find ("params")
|
||
|
||
if params:
|
||
add_element (params)
|
||
|
||
info = log.find ("info")
|
||
|
||
if info:
|
||
for element in info:
|
||
add_element (element)
|
||
|
||
class VariablesViewer (Gtk.ScrolledWindow):
|
||
class Store (Gtk.ListStore):
|
||
NAME = 0
|
||
COLOR = 1
|
||
VALUE = 2
|
||
RAW = 3
|
||
MIN = 4
|
||
MAX = 5
|
||
MEDIAN = 6
|
||
MEAN = 7
|
||
STDEV = 8
|
||
LAST_COLUMN = 9
|
||
|
||
def __init__ (self):
|
||
n_stats = self.LAST_COLUMN - self.COLOR
|
||
|
||
Gtk.ListStore.__init__ (self,
|
||
*((str, Gdk.RGBA) + n_stats * (str,)))
|
||
enum.Enum.__init__ (self)
|
||
|
||
for var in var_defs:
|
||
i = self.append (((var,
|
||
Gdk.RGBA (*var_defs[var].color)) +
|
||
n_stats * ("",)))
|
||
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.Box.__init__ (self,
|
||
*args,
|
||
hscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
vscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
**kwargs)
|
||
|
||
self.needs_update = True
|
||
|
||
store = self.Store ()
|
||
self.store = store
|
||
|
||
tree = Gtk.TreeView (model = store)
|
||
self.add (tree)
|
||
tree.show ()
|
||
|
||
self.single_sample_cols = []
|
||
self.multi_sample_cols = []
|
||
|
||
col = Gtk.TreeViewColumn (title = "Variable")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = CellRendererColorToggle (active = True)
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "color", store.COLOR)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", store.NAME)
|
||
|
||
def add_value_column (title, column, single_sample):
|
||
col = Gtk.TreeViewColumn (title = title)
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
col.set_alignment (0.5)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", column)
|
||
|
||
if single_sample:
|
||
self.single_sample_cols.append (col)
|
||
else:
|
||
self.multi_sample_cols.append (col)
|
||
|
||
add_value_column ("Value", store.VALUE, True)
|
||
add_value_column ("Raw", store.RAW, True)
|
||
add_value_column ("Min", store.MIN, False)
|
||
add_value_column ("Max", store.MAX, False)
|
||
add_value_column ("Median", store.MEDIAN, False)
|
||
add_value_column ("Mean", store.MEAN, False)
|
||
add_value_column ("Std. Dev.", store.STDEV, False)
|
||
|
||
col = Gtk.TreeViewColumn ()
|
||
tree.append_column (col)
|
||
|
||
selection.connect ("change-complete", self.selection_change_complete)
|
||
|
||
def update (self):
|
||
if not self.needs_update:
|
||
return
|
||
|
||
self.needs_update = False
|
||
|
||
sel = selection.get_effective_selection ()
|
||
n_sel = len (sel)
|
||
|
||
if n_sel == 1:
|
||
i, = sel
|
||
|
||
for row in self.store:
|
||
var_name = row[self.store.NAME]
|
||
|
||
var = samples[i].vars[var_name]
|
||
var_type = var_types[var_defs[var_name].type]
|
||
|
||
row[self.store.VALUE] = var_type.format (var.value)
|
||
row[self.store.RAW] = var.raw if var.raw is not None \
|
||
else "N/A"
|
||
else:
|
||
for row in self.store:
|
||
var_name = row[self.store.NAME]
|
||
|
||
var_type = var_types[var_defs[var_name].type]
|
||
|
||
vals = (samples[i].vars[var_name].value for i in sel)
|
||
vals = tuple (val for val in vals if val is not None)
|
||
|
||
if vals:
|
||
min_val = min (vals)
|
||
max_val = max (vals)
|
||
median = statistics.median (vals)
|
||
mean = statistics.mean (vals)
|
||
stdev = statistics.pstdev (vals, mean)
|
||
|
||
row[self.store.MIN] = var_type.format (min_val)
|
||
row[self.store.MAX] = var_type.format (max_val)
|
||
row[self.store.MEDIAN] = var_type.format (median)
|
||
row[self.store.MEAN] = var_type.format_numeric (mean)
|
||
row[self.store.STDEV] = var_type.format_numeric (stdev)
|
||
else:
|
||
row[self.store.MIN] = \
|
||
row[self.store.MAX] = \
|
||
row[self.store.MEDIAN] = \
|
||
row[self.store.MEAN] = \
|
||
row[self.store.STDEV] = var_type.format (None)
|
||
|
||
for col in self.single_sample_cols: col.set_visible (n_sel == 1)
|
||
for col in self.multi_sample_cols: col.set_visible (n_sel > 1)
|
||
|
||
def do_map (self):
|
||
self.update ()
|
||
|
||
Gtk.ScrolledWindow.do_map (self)
|
||
|
||
def selection_change_complete (self, selection):
|
||
self.needs_update = True
|
||
|
||
if self.get_mapped ():
|
||
self.update ()
|
||
|
||
class BacktraceViewer (Gtk.Box):
|
||
class ThreadStore (Gtk.ListStore):
|
||
INDEX = 0
|
||
ID = 1
|
||
NAME = 2
|
||
STATE = 3
|
||
|
||
def __init__ (self):
|
||
Gtk.ListStore.__init__ (self, int, int, str, str)
|
||
|
||
class FrameStore (Gtk.ListStore):
|
||
ID = 0
|
||
ADDRESS = 1
|
||
OBJECT = 2
|
||
FUNCTION = 3
|
||
OFFSET = 4
|
||
SOURCE = 5
|
||
LINE = 6
|
||
|
||
def __init__ (self):
|
||
Gtk.ListStore.__init__ (self, int, str, str, str, str, str, str)
|
||
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.Box.__init__ (self,
|
||
*args,
|
||
orientation = Gtk.Orientation.HORIZONTAL,
|
||
**kwargs)
|
||
|
||
self.needs_update = True
|
||
|
||
vbox = Gtk.Box (orientation = Gtk.Orientation.VERTICAL)
|
||
self.pack_start (vbox, False, False, 0)
|
||
vbox.show ()
|
||
|
||
header = Gtk.HeaderBar (title = "Threads", has_subtitle = False)
|
||
vbox.pack_start (header, False, False, 0)
|
||
header.show ()
|
||
|
||
scrolled = Gtk.ScrolledWindow (
|
||
hscrollbar_policy = Gtk.PolicyType.NEVER,
|
||
vscrollbar_policy = Gtk.PolicyType.AUTOMATIC
|
||
)
|
||
vbox.pack_start (scrolled, True, True, 0)
|
||
scrolled.show ()
|
||
|
||
store = self.ThreadStore ()
|
||
self.thread_store = store
|
||
|
||
tree = Gtk.TreeView (model = store)
|
||
self.thread_tree = tree
|
||
scrolled.add (tree)
|
||
tree.set_search_column (store.NAME)
|
||
tree.show ()
|
||
|
||
tree.connect ("row-activated", self.threads_row_activated)
|
||
|
||
tree.get_selection ().connect ("changed",
|
||
self.threads_selection_changed)
|
||
|
||
col = Gtk.TreeViewColumn (title = "ID")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", self.ThreadStore.ID)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Name")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", self.ThreadStore.NAME)
|
||
|
||
col = Gtk.TreeViewColumn (title = "State")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", self.ThreadStore.STATE)
|
||
|
||
separator = Gtk.Separator (orientation = Gtk.Orientation.VERTICAL)
|
||
self.pack_start (separator, False, False, 0)
|
||
separator.show ()
|
||
|
||
vbox = Gtk.Box (orientation = Gtk.Orientation.VERTICAL)
|
||
self.pack_start (vbox, True, True, 0)
|
||
vbox.show ()
|
||
|
||
header = Gtk.HeaderBar (title = "Stack", has_subtitle = False)
|
||
vbox.pack_start (header, False, False, 0)
|
||
header.show ()
|
||
|
||
scrolled = Gtk.ScrolledWindow (
|
||
hscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
vscrollbar_policy = Gtk.PolicyType.AUTOMATIC
|
||
)
|
||
vbox.pack_start (scrolled, True, True, 0)
|
||
scrolled.show ()
|
||
|
||
store = self.FrameStore ()
|
||
self.frame_store = store
|
||
|
||
tree = Gtk.TreeView (model = store, has_tooltip = True)
|
||
scrolled.add (tree)
|
||
tree.set_search_column (store.FUNCTION)
|
||
tree.show ()
|
||
|
||
tree.connect ("row-activated", self.frames_row_activated)
|
||
tree.connect ("query-tooltip", self.frames_query_tooltip)
|
||
|
||
def format_filename_col (tree_col, cell, model, iter, col):
|
||
object = model[iter][col]
|
||
|
||
cell.set_property ("text", get_basename (object) if object else "")
|
||
|
||
self.tooltip_columns = {}
|
||
|
||
col = Gtk.TreeViewColumn (title = "#")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", self.FrameStore.ID)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Address")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", self.FrameStore.ADDRESS)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Object")
|
||
self.tooltip_columns[col] = store.OBJECT
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.set_cell_data_func (cell, format_filename_col, store.OBJECT)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Function")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", self.FrameStore.FUNCTION)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Offset")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", self.FrameStore.OFFSET)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Source")
|
||
self.tooltip_columns[col] = store.SOURCE
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.set_cell_data_func (cell, format_filename_col, store.SOURCE)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Line")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", self.FrameStore.LINE)
|
||
|
||
col = Gtk.TreeViewColumn ()
|
||
tree.append_column (col)
|
||
|
||
selection.connect ("change-complete", self.selection_change_complete)
|
||
|
||
@GObject.Property (type = bool, default = False)
|
||
def available (self):
|
||
sel = selection.get_effective_selection ()
|
||
|
||
if len (sel) == 1:
|
||
i, = sel
|
||
|
||
return bool (samples[i].backtrace)
|
||
|
||
return False
|
||
|
||
def update (self):
|
||
if not self.needs_update or not self.available:
|
||
return
|
||
|
||
self.needs_update = False
|
||
|
||
tid = None
|
||
|
||
sel_rows = self.thread_tree.get_selection ().get_selected_rows ()[1]
|
||
|
||
if sel_rows:
|
||
tid = self.thread_store[sel_rows[0]][self.ThreadStore.ID]
|
||
|
||
i, = selection.get_effective_selection ()
|
||
|
||
self.thread_store.clear ()
|
||
|
||
for t in range (len (samples[i].backtrace)):
|
||
thread = samples[i].backtrace[t]
|
||
|
||
iter = self.thread_store.append (
|
||
(t, thread.id, thread.name, str (thread.state))
|
||
)
|
||
|
||
if thread.id == tid:
|
||
self.thread_tree.get_selection ().select_iter (iter)
|
||
|
||
def do_map (self):
|
||
self.update ()
|
||
|
||
Gtk.Box.do_map (self)
|
||
|
||
def selection_change_complete (self, selection):
|
||
self.needs_update = True
|
||
|
||
if self.get_mapped ():
|
||
self.update ()
|
||
|
||
self.notify ("available")
|
||
|
||
def threads_row_activated (self, tree, path, col):
|
||
iter = self.thread_store.get_iter (path)
|
||
|
||
tid = self.thread_store[iter][self.ThreadStore.ID]
|
||
|
||
sel = set ()
|
||
|
||
for i in range (len (samples)):
|
||
threads = filter (lambda thread:
|
||
thread.id == tid and
|
||
thread.state == ThreadState.RUNNING,
|
||
samples[i].backtrace or [])
|
||
|
||
if list (threads):
|
||
sel.add (i)
|
||
|
||
selection.select (sel)
|
||
|
||
selection.change_complete ()
|
||
|
||
def threads_selection_changed (self, tree_sel):
|
||
self.frame_store.clear ()
|
||
|
||
(store, rows) = tree_sel.get_selected_rows ()
|
||
|
||
if not rows:
|
||
return
|
||
|
||
i, = selection.get_effective_selection ()
|
||
|
||
try:
|
||
frames = samples[i].backtrace[store[rows[0]][store.INDEX]].frames
|
||
|
||
for frame in frames:
|
||
info = frame.info
|
||
|
||
self.frame_store.append ((
|
||
frame.id, hex (frame.address), info.object, info.symbol,
|
||
hex (info.offset) if info.offset is not None else None,
|
||
info.source, str (info.line) if info.line else None
|
||
))
|
||
except:
|
||
pass
|
||
|
||
def frames_row_activated (self, tree, path, col):
|
||
iter = self.frame_store.get_iter (path)
|
||
|
||
address = int (self.frame_store[iter][self.FrameStore.ADDRESS], 0)
|
||
info = address_map.get (address, None)
|
||
|
||
if not info:
|
||
return
|
||
|
||
id = info.id
|
||
|
||
if not id:
|
||
return
|
||
|
||
sel = set ()
|
||
|
||
def has_frame (sample, id):
|
||
for thread in sample.backtrace or []:
|
||
for frame in thread.frames:
|
||
if frame.info.id == id:
|
||
return True
|
||
|
||
return False
|
||
|
||
for i in range (len (samples)):
|
||
if has_frame (samples[i], id):
|
||
sel.add (i)
|
||
|
||
selection.select (sel)
|
||
|
||
selection.change_complete ()
|
||
|
||
def frames_query_tooltip (self, tree, x, y, keyboard_mode, tooltip):
|
||
hit, x, y, model, path, iter = tree.get_tooltip_context (x, y,
|
||
keyboard_mode)
|
||
|
||
if hit:
|
||
column = -1
|
||
|
||
if keyboard_mode:
|
||
cursor_path, cursor_col = tree.get_cursor ()
|
||
|
||
if path.compare (cursor_path) == 0:
|
||
column = self.tooltip_columns[cursor_col]
|
||
else:
|
||
for col in self.tooltip_columns:
|
||
area = tree.get_cell_area (path, col)
|
||
|
||
if x >= area.x and x < area.x + area.width and \
|
||
y >= area.y and y < area.y + area.height:
|
||
column = self.tooltip_columns[col]
|
||
|
||
break
|
||
|
||
if column >= 0:
|
||
value = model[iter][column]
|
||
|
||
if value:
|
||
tooltip.set_text (str (value))
|
||
|
||
return True
|
||
|
||
return False
|
||
|
||
class ProfileViewer (Gtk.ScrolledWindow):
|
||
class ThreadFilter (Gtk.TreeView):
|
||
class Store (Gtk.ListStore):
|
||
VISIBLE = 0
|
||
ID = 1
|
||
NAME = 2
|
||
STATE = {list (ThreadState)[i]: 3 + i
|
||
for i in range (len (ThreadState))}
|
||
|
||
def __init__ (self):
|
||
Gtk.ListStore.__init__ (self,
|
||
bool, int, str,
|
||
*(len (self.STATE) * (bool,)))
|
||
|
||
threads = list ({thread.id
|
||
for sample in samples
|
||
for thread in sample.backtrace or ()})
|
||
threads.sort ()
|
||
|
||
states = [state == ThreadState.RUNNING for state in self.STATE]
|
||
|
||
for id in threads:
|
||
self.append ((False, id, None, *states))
|
||
|
||
def get_filter (self):
|
||
return {row[self.ID]: {state
|
||
for state, column in self.STATE.items ()
|
||
if row[column]}
|
||
for row in self}
|
||
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.TreeView.__init__ (self, *args, **kwargs)
|
||
|
||
self.needs_update = True
|
||
|
||
store = self.Store ()
|
||
self.store = store
|
||
|
||
filter = Gtk.TreeModelFilter (child_model = store)
|
||
filter.set_visible_column (store.VISIBLE)
|
||
self.set_model (filter)
|
||
|
||
col = Gtk.TreeViewColumn (title = "ID")
|
||
self.append_column (col)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", store.ID)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Name")
|
||
self.append_column (col)
|
||
|
||
cell = Gtk.CellRendererText ()
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "text", store.NAME)
|
||
|
||
for state in store.STATE:
|
||
col = Gtk.TreeViewColumn (title = str (state))
|
||
col.column = store.STATE[state]
|
||
self.append_column (col)
|
||
col.set_alignment (0.5)
|
||
col.set_clickable (True)
|
||
|
||
def col_clicked (col):
|
||
active = not all (row[col.column] for row in filter)
|
||
|
||
for row in filter:
|
||
row[col.column] = active
|
||
|
||
col.connect ("clicked", col_clicked)
|
||
|
||
cell = Gtk.CellRendererToggle ()
|
||
cell.column = store.STATE[state]
|
||
col.pack_start (cell, False)
|
||
col.add_attribute (cell, "active", store.STATE[state])
|
||
|
||
def cell_toggled (cell, path):
|
||
store[path][cell.column] = not cell.get_property ("active")
|
||
|
||
cell.connect ("toggled", cell_toggled)
|
||
|
||
selection.connect ("change-complete",
|
||
self.selection_change_complete)
|
||
|
||
def update (self):
|
||
if not self.needs_update:
|
||
return
|
||
|
||
self.needs_update = False
|
||
|
||
sel = selection.get_effective_selection ()
|
||
|
||
threads = {thread.id: thread.name
|
||
for i in sel
|
||
for thread in samples[i].backtrace or ()}
|
||
|
||
for row in self.store:
|
||
id = row[self.store.ID]
|
||
|
||
if id in threads:
|
||
row[self.store.VISIBLE] = True
|
||
row[self.store.NAME] = threads[id]
|
||
else:
|
||
row[self.store.VISIBLE] = False
|
||
row[self.store.NAME] = None
|
||
|
||
def do_map (self):
|
||
self.update ()
|
||
|
||
Gtk.TreeView.do_map (self)
|
||
|
||
def selection_change_complete (self, selection):
|
||
self.needs_update = True
|
||
|
||
if self.get_mapped ():
|
||
self.update ()
|
||
|
||
class ThreadPopover (Gtk.Popover):
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.Popover.__init__ (self, *args, border_width = 4, **kwargs)
|
||
|
||
frame = Gtk.Frame (shadow_type = Gtk.ShadowType.IN)
|
||
self.add (frame)
|
||
frame.show ()
|
||
|
||
scrolled = Gtk.ScrolledWindow (
|
||
hscrollbar_policy = Gtk.PolicyType.NEVER,
|
||
vscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
propagate_natural_height = True,
|
||
max_content_height = 400
|
||
)
|
||
frame.add (scrolled)
|
||
scrolled.show ()
|
||
|
||
thread_filter = ProfileViewer.ThreadFilter ()
|
||
self.thread_filter = thread_filter
|
||
scrolled.add (thread_filter)
|
||
thread_filter.show ()
|
||
|
||
class Profile (Gtk.Box):
|
||
ProfileFrame = namedtuple ("ProfileFrame", ("sample", "stack", "i"))
|
||
|
||
class Direction (enum.Enum):
|
||
CALLEES = enum.auto ()
|
||
CALLERS = enum.auto ()
|
||
|
||
class Store (Gtk.ListStore):
|
||
ID = 0
|
||
FUNCTION = 1
|
||
EXCLUSIVE = 2
|
||
INCLUSIVE = 3
|
||
|
||
def __init__ (self):
|
||
Gtk.ListStore.__init__ (self,
|
||
GObject.TYPE_UINT64, str, float, float)
|
||
|
||
__gsignals__ = {
|
||
"subprofile-added": (GObject.SIGNAL_RUN_FIRST,
|
||
None, (Gtk.Widget,)),
|
||
"subprofile-removed": (GObject.SIGNAL_RUN_FIRST,
|
||
None, (Gtk.Widget,))
|
||
}
|
||
|
||
def __init__ (self,
|
||
root = None,
|
||
id = None,
|
||
title = None,
|
||
frames = None,
|
||
direction = Direction.CALLEES,
|
||
sort = (Store.INCLUSIVE, Gtk.SortType.DESCENDING),
|
||
*args,
|
||
**kwargs):
|
||
Gtk.Box.__init__ (self,
|
||
*args,
|
||
orientation = Gtk.Orientation.HORIZONTAL,
|
||
**kwargs)
|
||
|
||
self.root = root or self
|
||
self.id = id
|
||
self.frames = frames
|
||
self.direction = direction
|
||
|
||
self.subprofile = None
|
||
|
||
vbox = Gtk.Box (orientation = Gtk.Orientation.VERTICAL)
|
||
self.pack_start (vbox, False, False, 0)
|
||
vbox.show ()
|
||
|
||
header = Gtk.HeaderBar (title = title or "All Functions")
|
||
self.header = header
|
||
vbox.pack_start (header, False, False, 0)
|
||
header.show ()
|
||
|
||
if not id:
|
||
popover = ProfileViewer.ThreadPopover ()
|
||
|
||
thread_filter_store = popover.thread_filter.store
|
||
|
||
self.thread_filter_store = thread_filter_store
|
||
self.thread_filter = thread_filter_store.get_filter ()
|
||
|
||
button = Gtk.MenuButton (popover = popover)
|
||
header.pack_end (button)
|
||
button.show ()
|
||
|
||
button.connect ("toggled", self.thread_filter_button_toggled)
|
||
|
||
hbox = Gtk.Box (orientation = Gtk.Orientation.HORIZONTAL,
|
||
spacing = 4)
|
||
button.add (hbox)
|
||
hbox.show ()
|
||
|
||
label = Gtk.Label ("Threads")
|
||
hbox.pack_start (label, False, False, 0)
|
||
label.show ()
|
||
|
||
image = Gtk.Image.new_from_icon_name ("pan-down-symbolic",
|
||
Gtk.IconSize.BUTTON)
|
||
hbox.pack_start (image, False, False, 0)
|
||
image.show ()
|
||
|
||
button = Gtk.Button (tooltip_text = "Call-graph direction")
|
||
header.pack_end (button)
|
||
button.show ()
|
||
|
||
button.connect ("clicked", self.direction_button_clicked)
|
||
|
||
image = Gtk.Image ()
|
||
self.direction_image = image
|
||
button.add (image)
|
||
image.show ()
|
||
else:
|
||
button = Gtk.Button.new_from_icon_name (
|
||
"edit-select-all-symbolic",
|
||
Gtk.IconSize.BUTTON
|
||
)
|
||
header.pack_end (button)
|
||
button.set_tooltip_text (
|
||
str (Selection (frame.sample for frame in frames))
|
||
)
|
||
button.show ()
|
||
|
||
button.connect ("clicked", self.select_samples_clicked)
|
||
|
||
scrolled = Gtk.ScrolledWindow (
|
||
hscrollbar_policy = Gtk.PolicyType.NEVER,
|
||
vscrollbar_policy = Gtk.PolicyType.AUTOMATIC
|
||
)
|
||
vbox.pack_start (scrolled, True, True, 0)
|
||
scrolled.show ()
|
||
|
||
store = self.Store ()
|
||
self.store = store
|
||
store.set_sort_column_id (*sort)
|
||
|
||
tree = Gtk.TreeView (model = store)
|
||
self.tree = tree
|
||
scrolled.add (tree)
|
||
tree.set_search_column (store.FUNCTION)
|
||
tree.show ()
|
||
|
||
tree.get_selection ().connect ("changed",
|
||
self.tree_selection_changed)
|
||
|
||
tree.connect ("row-activated", self.tree_row_activated)
|
||
tree.connect ("key-press-event", self.tree_key_press_event)
|
||
|
||
col = Gtk.TreeViewColumn (title = "Function")
|
||
tree.append_column (col)
|
||
col.set_resizable (True)
|
||
|
||
cell = Gtk.CellRendererText (ellipsize = Pango.EllipsizeMode.END)
|
||
col.pack_start (cell, True)
|
||
col.add_attribute (cell, "text", store.FUNCTION)
|
||
cell.set_property ("width-chars", 40)
|
||
|
||
def format_percentage_col (tree_col, cell, model, iter, col):
|
||
cell.set_property ("text",
|
||
format_percentage (model[iter][col], 2))
|
||
|
||
col = Gtk.TreeViewColumn (title = "Self")
|
||
tree.append_column (col)
|
||
col.set_alignment (0.5)
|
||
col.set_sort_column_id (store.EXCLUSIVE)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.set_cell_data_func (cell,
|
||
format_percentage_col, store.EXCLUSIVE)
|
||
|
||
col = Gtk.TreeViewColumn (title = "All")
|
||
tree.append_column (col)
|
||
col.set_alignment (0.5)
|
||
col.set_sort_column_id (store.INCLUSIVE)
|
||
|
||
cell = Gtk.CellRendererText (xalign = 1)
|
||
col.pack_start (cell, False)
|
||
col.set_cell_data_func (cell,
|
||
format_percentage_col, store.INCLUSIVE)
|
||
|
||
if id:
|
||
self.update ()
|
||
|
||
def update (self):
|
||
self.remove_subprofile ()
|
||
|
||
sel_id = None
|
||
|
||
sel_rows = self.tree.get_selection ().get_selected_rows ()[1]
|
||
|
||
if sel_rows:
|
||
sel_id = self.store[sel_rows[0]][self.store.ID]
|
||
|
||
if not self.id:
|
||
self.update_frames ()
|
||
|
||
self.update_store ()
|
||
|
||
self.select (sel_id)
|
||
|
||
self.update_ui ()
|
||
|
||
def update_frames (self):
|
||
self.frames = []
|
||
|
||
for i in selection.get_effective_selection ():
|
||
for thread in samples[i].backtrace or []:
|
||
if thread.state in self.thread_filter[thread.id]:
|
||
thread_frames = thread.frames
|
||
|
||
if self.direction == self.Direction.CALLERS:
|
||
thread_frames = reversed (thread_frames)
|
||
|
||
stack = []
|
||
prev_id = 0
|
||
|
||
for frame in thread_frames:
|
||
id = frame.info.id
|
||
|
||
if id == prev_id:
|
||
continue
|
||
|
||
self.frames.append (self.ProfileFrame (
|
||
sample = i,
|
||
stack = stack,
|
||
i = len (stack)
|
||
))
|
||
|
||
stack.append (frame)
|
||
|
||
prev_id = id
|
||
|
||
def update_store (self):
|
||
stacks = {}
|
||
symbols = {}
|
||
|
||
sort = self.store.get_sort_column_id ()
|
||
|
||
self.store = self.Store ()
|
||
|
||
for frame in self.frames:
|
||
info = frame.stack[frame.i].info
|
||
symbol_id = info.id
|
||
stack_id = builtins.id (frame.stack)
|
||
|
||
symbol = symbols.get (symbol_id, None)
|
||
|
||
if not symbol:
|
||
symbol = [info, 0, 0]
|
||
symbols[symbol_id] = symbol
|
||
|
||
stack = stacks.get (stack_id, None)
|
||
|
||
if not stack:
|
||
stack = set ()
|
||
stacks[stack_id] = stack
|
||
|
||
if frame.i == 0:
|
||
symbol[1] += 1
|
||
|
||
if symbol_id not in stack:
|
||
stack.add (symbol_id)
|
||
|
||
symbol[2] += 1
|
||
|
||
n_stacks = len (stacks)
|
||
|
||
for symbol in symbols.values ():
|
||
id = symbol[0].id
|
||
name = symbol[0].name if id != self.id else "[Self]"
|
||
|
||
self.store.append ((id,
|
||
name,
|
||
symbol[1] / n_stacks,
|
||
symbol[2] / n_stacks))
|
||
|
||
self.store.set_sort_column_id (*sort)
|
||
|
||
self.tree.set_model (self.store)
|
||
self.tree.set_search_column (self.store.FUNCTION)
|
||
|
||
def update_ui (self):
|
||
if not self.id:
|
||
if self.direction == self.Direction.CALLEES:
|
||
icon_name = "format-indent-more-symbolic"
|
||
else:
|
||
icon_name = "format-indent-less-symbolic"
|
||
|
||
self.direction_image.set_from_icon_name (icon_name,
|
||
Gtk.IconSize.BUTTON)
|
||
else:
|
||
if self.direction == self.Direction.CALLEES:
|
||
subtitle = "Callees"
|
||
else:
|
||
subtitle = "Callers"
|
||
|
||
self.header.set_subtitle (subtitle)
|
||
|
||
def select (self, id):
|
||
if id is not None:
|
||
for row in self.store:
|
||
if row[self.store.ID] == id:
|
||
iter = row.iter
|
||
path = self.store.get_path (iter)
|
||
|
||
self.tree.get_selection ().select_iter (iter)
|
||
|
||
self.tree.scroll_to_cell (path, None, True, 0.5, 0)
|
||
|
||
break
|
||
else:
|
||
self.tree.get_selection ().unselect_all ()
|
||
|
||
def add_subprofile (self, subprofile):
|
||
self.remove_subprofile ()
|
||
|
||
box = Gtk.Box (orientation = Gtk.Orientation.HORIZONTAL)
|
||
self.subprofile = box
|
||
self.pack_start (box, True, True, 0)
|
||
box.show ()
|
||
|
||
separator = Gtk.Separator (orientation = Gtk.Orientation.VERTICAL)
|
||
box.pack_start (separator, False, False, 0)
|
||
separator.show ()
|
||
|
||
box.pack_start (subprofile, True, True, 0)
|
||
subprofile.show ()
|
||
|
||
subprofile.connect ("subprofile-added",
|
||
lambda profile, subprofile:
|
||
self.emit ("subprofile-added",
|
||
subprofile))
|
||
subprofile.connect ("subprofile-removed",
|
||
lambda profile, subprofile:
|
||
self.emit ("subprofile-removed",
|
||
subprofile))
|
||
|
||
self.emit ("subprofile-added", subprofile)
|
||
|
||
def remove_subprofile (self):
|
||
if self.subprofile:
|
||
subprofile = self.subprofile
|
||
|
||
self.remove (subprofile)
|
||
|
||
self.subprofile = None
|
||
|
||
self.emit ("subprofile-removed", subprofile)
|
||
|
||
def thread_filter_button_toggled (self, button):
|
||
if not button.get_active ():
|
||
thread_filter = self.thread_filter_store.get_filter ()
|
||
|
||
if thread_filter != self.thread_filter:
|
||
self.thread_filter = thread_filter
|
||
|
||
self.update ()
|
||
|
||
def direction_button_clicked (self, button):
|
||
if self.direction == self.Direction.CALLEES:
|
||
self.direction = self.Direction.CALLERS
|
||
else:
|
||
self.direction = self.Direction.CALLEES
|
||
|
||
self.update ()
|
||
|
||
def select_samples_clicked (self, button):
|
||
self.root.select (self.id)
|
||
|
||
sel = set ()
|
||
|
||
for frame in self.frames:
|
||
sel.add (frame.sample)
|
||
|
||
selection.select (sel)
|
||
|
||
selection.change_complete ()
|
||
|
||
def tree_selection_changed (self, tree_sel):
|
||
self.remove_subprofile ()
|
||
|
||
sel_rows = tree_sel.get_selected_rows ()[1]
|
||
|
||
if not sel_rows:
|
||
return
|
||
|
||
id = self.store[sel_rows[0]][self.store.ID]
|
||
title = self.store[sel_rows[0]][self.store.FUNCTION]
|
||
|
||
if id == self.id:
|
||
return
|
||
|
||
frames = []
|
||
|
||
for frame in self.frames:
|
||
if frame.stack[frame.i].info.id == id:
|
||
frames.append (frame)
|
||
|
||
if frame.i > 0:
|
||
frames.append (self.ProfileFrame (sample = frame.sample,
|
||
stack = frame.stack,
|
||
i = frame.i - 1))
|
||
|
||
self.add_subprofile (ProfileViewer.Profile (
|
||
self.root,
|
||
id,
|
||
title,
|
||
frames,
|
||
self.direction,
|
||
self.store.get_sort_column_id ()
|
||
))
|
||
|
||
def tree_row_activated (self, tree, path, col):
|
||
if self.root != self:
|
||
self.root.select (self.store[path][self.store.ID])
|
||
|
||
def tree_key_press_event (self, tree, event):
|
||
if event.keyval == Gdk.KEY_Escape:
|
||
self.select (None)
|
||
|
||
if self.root is not self:
|
||
self.get_parent ().get_ancestor (
|
||
ProfileViewer.Profile
|
||
).tree.grab_focus ()
|
||
|
||
return True
|
||
|
||
return False
|
||
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.ScrolledWindow.__init__ (
|
||
self,
|
||
*args,
|
||
hscrollbar_policy = Gtk.PolicyType.AUTOMATIC,
|
||
vscrollbar_policy = Gtk.PolicyType.NEVER,
|
||
**kwargs
|
||
)
|
||
|
||
self.adjustment_changed_handler = None
|
||
self.needs_update = True
|
||
|
||
profile = self.Profile ()
|
||
self.root_profile = profile
|
||
self.add (profile)
|
||
profile.show ()
|
||
|
||
selection.connect ("change-complete", self.selection_change_complete)
|
||
|
||
profile.connect ("subprofile-added", self.subprofile_added)
|
||
|
||
@GObject.Property (type = bool, default = False)
|
||
def available (self):
|
||
sel = selection.get_effective_selection ()
|
||
|
||
if len (sel) > 1:
|
||
return any (samples[i].backtrace for i in sel)
|
||
|
||
return False
|
||
|
||
def update (self):
|
||
if not self.needs_update or not self.available:
|
||
return
|
||
|
||
self.needs_update = False
|
||
|
||
self.root_profile.update ()
|
||
|
||
def do_map (self):
|
||
self.update ()
|
||
|
||
Gtk.ScrolledWindow.do_map (self)
|
||
|
||
def selection_change_complete (self, selection):
|
||
self.needs_update = True
|
||
|
||
if self.get_mapped ():
|
||
self.update ()
|
||
|
||
self.notify ("available")
|
||
|
||
def subprofile_added (self, profile, subprofile):
|
||
if not self.adjustment_changed_handler:
|
||
adjustment = self.get_hadjustment ()
|
||
|
||
def adjustment_changed (adjustment):
|
||
GObject.signal_handler_disconnect (
|
||
adjustment,
|
||
self.adjustment_changed_handler
|
||
)
|
||
self.adjustment_changed_handler = None
|
||
|
||
adjustment.set_value (adjustment.get_upper ())
|
||
|
||
self.adjustment_changed_handler = adjustment.connect (
|
||
"changed",
|
||
adjustment_changed
|
||
)
|
||
|
||
class LogViewer (Gtk.Window):
|
||
def __init__ (self, *args, **kwargs):
|
||
Gtk.Window.__init__ (
|
||
self,
|
||
*args,
|
||
default_width = 1024,
|
||
default_height = 768,
|
||
window_position = Gtk.WindowPosition.CENTER,
|
||
**kwargs)
|
||
|
||
header = Gtk.HeaderBar (
|
||
title = "GIMP Performance Log Viewer",
|
||
show_close_button = True
|
||
)
|
||
self.header = header
|
||
self.set_titlebar (header)
|
||
header.show ()
|
||
|
||
button = Gtk.MenuButton ()
|
||
header.pack_end (button)
|
||
button.set_tooltip_text ("Find samples")
|
||
button.show ()
|
||
|
||
image = Gtk.Image.new_from_icon_name ("edit-find-symbolic",
|
||
Gtk.IconSize.BUTTON)
|
||
button.add (image)
|
||
image.show ()
|
||
|
||
popover = FindSamplesPopover (relative_to = button)
|
||
self.find_popover = popover
|
||
button.set_popover (popover)
|
||
|
||
paned = Gtk.Paned (orientation = Gtk.Orientation.VERTICAL)
|
||
self.paned = paned
|
||
self.add (paned)
|
||
paned.set_position (144)
|
||
paned.show ()
|
||
|
||
graphs = SampleGraphList ()
|
||
paned.add1 (graphs)
|
||
paned.child_set (graphs, shrink = False)
|
||
graphs.show ()
|
||
|
||
hbox = Gtk.Box (orientation = Gtk.Orientation.HORIZONTAL)
|
||
paned.add2 (hbox)
|
||
hbox.show ()
|
||
|
||
sidebar = Gtk.StackSidebar ()
|
||
hbox.pack_start (sidebar, False, False, 0)
|
||
sidebar.show ()
|
||
|
||
stack = Gtk.Stack (transition_type = Gtk.StackTransitionType.CROSSFADE)
|
||
self.stack = stack
|
||
hbox.pack_start (stack, True, True, 0)
|
||
stack.show ()
|
||
|
||
sidebar.set_stack (stack)
|
||
|
||
info_viewer = InformationViewer ()
|
||
stack.add_titled (info_viewer, "information", "Information")
|
||
info_viewer.show ()
|
||
|
||
vars_viewer = VariablesViewer ()
|
||
stack.add_titled (vars_viewer, "variables", "Variables")
|
||
vars_viewer.show ()
|
||
|
||
box = Gtk.Box (orientation = Gtk.Orientation.VERTICAL)
|
||
self.cflow_box = box
|
||
stack.add_named (box, "cflow")
|
||
|
||
backtrace_viewer = BacktraceViewer ()
|
||
self.backtrace_viewer = backtrace_viewer
|
||
box.pack_start (backtrace_viewer, True, True, 0)
|
||
|
||
backtrace_viewer.bind_property ("available",
|
||
backtrace_viewer, "visible",
|
||
GObject.BindingFlags.SYNC_CREATE)
|
||
|
||
backtrace_viewer.connect ("notify::available",
|
||
self.cflow_notify_available)
|
||
|
||
profile_viewer = ProfileViewer ()
|
||
self.profile_viewer = profile_viewer
|
||
box.pack_start (profile_viewer, True, True, 0)
|
||
|
||
profile_viewer.bind_property ("available",
|
||
profile_viewer, "visible",
|
||
GObject.BindingFlags.SYNC_CREATE)
|
||
|
||
profile_viewer.connect ("notify::available",
|
||
self.cflow_notify_available)
|
||
|
||
self.cflow_notify_available (self)
|
||
|
||
selection.connect ("change-complete", self.selection_change_complete)
|
||
|
||
self.selection_change_complete (selection)
|
||
|
||
def selection_change_complete (self, selection):
|
||
self.header.set_subtitle (str (selection))
|
||
|
||
def cflow_notify_available (self, *args):
|
||
if self.backtrace_viewer.available:
|
||
self.stack.child_set (self.cflow_box, title = "Backtrace")
|
||
self.cflow_box.show ()
|
||
elif self.profile_viewer.available:
|
||
self.stack.child_set (self.cflow_box, title = "Profile")
|
||
self.cflow_box.show ()
|
||
else:
|
||
self.cflow_box.hide ()
|
||
|
||
Gtk.Settings.get_default ().set_property ("gtk-application-prefer-dark-theme",
|
||
True)
|
||
|
||
window = LogViewer ()
|
||
window.show ()
|
||
|
||
window.connect ("destroy", Gtk.main_quit)
|
||
|
||
Gtk.main ()
|