summaryrefslogtreecommitdiff
path: root/scripts/perf/profile_pretty.py
blob: 15b6efd94dcc38bfc7bd325e2c4c41dc3921e8d8 (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
#!/usr/bin/env python3
"""\
@file   profile_pretty.py
@author Nat Goodspeed
@date   2024-09-12
@brief  Pretty-print a JSON file from Develop -> Render Tests -> Frame Profile

$LicenseInfo:firstyear=2024&license=viewerlgpl$
Copyright (c) 2024, Linden Research, Inc.
$/LicenseInfo$
"""

import logsdir
import json
from pathlib import Path
import sys

class Error(Exception):
    pass

def pretty(path):
    with open(path) as inf:
        data = json.load(inf)
    json.dump(data, sys.stdout, indent=4)

def main(*raw_args):
    from argparse import ArgumentParser
    parser = ArgumentParser(description="""
%(prog)s pretty-prints a JSON file from Develop -> Render Tests -> Frame Profile.
The file produced by the viewer is a single dense line of JSON.
""")
    parser.add_argument('path', nargs='?',
                        help="""profile filename to pretty-print (default is most recent)""")

    args = parser.parse_args(raw_args)
    if not args.path:
        logs = logsdir.logsdir()
        profiles = Path(logs).glob('profile.*.json')
        sort = [(p.stat().st_mtime, p) for p in profiles]
        sort.sort(reverse=True)
        try:
            args.path = sort[0][1]
        except IndexError:
            raise Error(f'No profile.*.json files in {logs}')
        # print path to sys.stderr in case user is redirecting stdout
        print(args.path, file=sys.stderr)

    pretty(args.path)

if __name__ == "__main__":
    try:
        sys.exit(main(*sys.argv[1:]))
    except (Error, OSError, json.JSONDecodeError) as err:
        sys.exit(str(err))