Add script to convert frame profile JSON file to CSV.
Also slightly refactor profile_pretty.py. (cherry picked from commit d60b1f92213ace6a8ab6a4a60cb01a43f45d3955)master
parent
277ee6830f
commit
ee1b0061c3
|
|
@ -0,0 +1,72 @@
|
|||
#!/usr/bin/env python3
|
||||
"""\
|
||||
@file profile_csv.py
|
||||
@author Nat Goodspeed
|
||||
@date 2024-09-12
|
||||
@brief Convert a JSON file from Develop -> Render Tests -> Frame Profile to CSV
|
||||
|
||||
$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 convert(path, totals=True, unused=True, file=sys.stdout):
|
||||
with open(path) as inf:
|
||||
data = json.load(inf)
|
||||
print('"name", "file1", "file2", "time", "binds", "samples", "triangles"', file=file)
|
||||
|
||||
if totals:
|
||||
t = data['totals']
|
||||
print(f'"totals", "", "", {t["time"]}, {t["binds"]}, {t["samples"]}, {t["triangles"]}',
|
||||
file=file)
|
||||
|
||||
for sh in data['shaders']:
|
||||
print(f'"{sh["name"]}", "{sh["files"][0]}", "{sh["files"][1]}", '
|
||||
f'{sh["time"]}, {sh["binds"]}, {sh["samples"]}, {sh["triangles"]}', file=file)
|
||||
|
||||
if unused:
|
||||
for u in data['unused']:
|
||||
print(f'"{u}", "", "", 0, 0, 0, 0', file=file)
|
||||
|
||||
def main(*raw_args):
|
||||
from argparse import ArgumentParser
|
||||
parser = ArgumentParser(description="""
|
||||
%(prog)s converts a JSON file from Develop -> Render Tests -> Frame Profile to
|
||||
a more-or-less equivalent CSV file. It expands the totals stats and unused
|
||||
shaders list to full shaders lines.
|
||||
""")
|
||||
parser.add_argument('-t', '--totals', action='store_false', default=True,
|
||||
help="""omit totals from CSV file""")
|
||||
parser.add_argument('-u', '--unused', action='store_false', default=True,
|
||||
help="""omit unused shaders from CSV file""")
|
||||
parser.add_argument('path', nargs='?',
|
||||
help="""profile filename to convert (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)
|
||||
|
||||
convert(args.path, totals=args.totals, unused=args.unused)
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
sys.exit(main(*sys.argv[1:]))
|
||||
except (Error, OSError, json.JSONDecodeError) as err:
|
||||
sys.exit(str(err))
|
||||
|
|
@ -18,19 +18,7 @@ import sys
|
|||
class Error(Exception):
|
||||
pass
|
||||
|
||||
def pretty(path=None):
|
||||
if not 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:
|
||||
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(path, file=sys.stderr)
|
||||
|
||||
def pretty(path):
|
||||
with open(path) as inf:
|
||||
data = json.load(inf)
|
||||
json.dump(data, sys.stdout, indent=4)
|
||||
|
|
@ -45,6 +33,18 @@ The file produced by the viewer is a single dense line of JSON.
|
|||
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__":
|
||||
|
|
|
|||
Loading…
Reference in New Issue