File: read_build_trace_gz.py

package info (click to toggle)
android-platform-tools 35.0.2-1~exp6
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 211,716 kB
  • sloc: cpp: 995,749; java: 290,495; ansic: 145,647; xml: 58,531; python: 39,608; sh: 14,500; javascript: 5,198; asm: 4,866; makefile: 3,115; yacc: 769; awk: 368; ruby: 183; sql: 140; perl: 88; lex: 67
file content (72 lines) | stat: -rw-r--r-- 2,235 bytes parent folder | download | duplicates (3)
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
#!/usr/bin/env python3
#
# Copyright (C) 2023 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import gzip
import json
import os
import sys

from collections import defaultdict

READ_DURATION = [
    'soong',
    'kati build',
    'ninja',
    'total',
]

class Trace:
    def __init__(self, trace_file):
        self.duration = dict()
        self._queue = defaultdict(list)
        self.target = os.path.splitext(os.path.basename(trace_file))[0]
        if not os.path.isfile(trace_file):
            return
        self._trace_file = gzip.open(trace_file, 'rt', encoding='utf-8')
        self._trace_data = json.load(self._trace_file)
        for t in self._trace_data:
            if 'ph' not in t:
                continue
            if t['ph'] == 'X':
                self.duration[t['name']] = t['dur']
                continue
            if t['ph'] == 'B':
                self._queue[(t['pid'], t['pid'])].append((t['name'], t['ts']))
                continue
            if t['ph'] == 'E':
                queue = self._queue[(t['pid'], t['pid'])]
                if not queue:
                    raise Exception('pid:{}, tid:{} not started'.format(t['pid'], t['pid']))
                name, ts = queue.pop()
                self.duration[name] = t['ts'] - ts
                continue

    def out_durations(self):
        out_str = self.target
        for name in READ_DURATION:
            if name not in self.duration:
                continue
            out_str = '{}, {}'.format(out_str, self.duration[name])
        out_str += '\n'
        sys.stdout.write(out_str)


def main(argv):
    trace = Trace(argv[1])
    trace.out_durations()

if __name__ == '__main__':
    main(sys.argv)