summaryrefslogtreecommitdiffstats
path: root/tools/performance-log-expand.py
blob: bc04d83a533469b5885c001d4a63ea5ec348d2f0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#!/usr/bin/env python3

"""
performance-log-expand.py -- Delta-decodes GIMP performance logs
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-expand.py < infile > outfile
"""

from xml.etree import ElementTree
import sys

empty_element = ElementTree.Element ("")

# Read performance log from STDIN
log = ElementTree.fromstring (sys.stdin.buffer.read ())

try:
    has_backtrace = bool (int (log.find ("params").find ("backtrace").text))
except:
    has_backtrace = False

def expand_simple (element, last_values):
    last_values.update ({value.tag: value.text for value in element})

    for value in list (element):
        element.remove (value)

    for tag, text in last_values.items ():
        value = ElementTree.SubElement (element, tag)
        value.text = text
        value.tail = "\n"

# Expand samples
last_vars      = {}
last_backtrace = {}

for sample in (log.find ("samples") or empty_element).iterfind ("sample"):
    # Expand variables
    vars = sample.find ("vars") or \
           ElementTree.SubElement (sample, "vars")

    expand_simple (vars, last_vars)

    # Expand backtrace
    if has_backtrace:
        backtrace = sample.find ("backtrace") or \
                    ElementTree.SubElement (sample, "backtrace")

        for thread in backtrace:
            id     = thread.get ("id")
            head   = thread.get ("head")
            tail   = thread.get ("tail")
            attrib = dict (thread.attrib)
            frames = list (thread)

            last_thread = last_backtrace.setdefault (id, [{}, []])
            last_frames = last_thread[1]

            if head:
                frames = last_frames[:int (head)] + frames

                del attrib["head"]

            if tail:
                frames = frames + last_frames[-int (tail):]

                del attrib["tail"]

            last_thread[0] = attrib
            last_thread[1] = frames

            if not frames and thread.text is None:
                del last_backtrace[id]

        for thread in list (backtrace):
            backtrace.remove (thread)

        for id, (attrib, frames) in last_backtrace.items ():
            thread = ElementTree.SubElement (backtrace, "thread", attrib)
            thread.text = "\n"
            thread.tail = "\n"

            thread.extend (frames)

# Expand address map
last_address = {}

for address in (log.find ("address-map") or empty_element).iterfind ("address"):
    expand_simple (address, last_address)

# Write performance log to STDOUT
sys.stdout.buffer.write (ElementTree.tostring (log))