35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import codecs, httplib, json, os, urllib, shutil, subprocess, sys, argparse
|
||
|
|
|
||
|
|
upstream_git = 'https://github.com/catapult-project/catapult.git'
|
||
|
|
|
||
|
|
script_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
|
||
|
|
catapult_src_dir = os.path.join(script_dir, 'catapult-upstream')
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('trace_file_or_dir',
|
||
|
|
help='Path to trace file or directory of trace files.')
|
||
|
|
parser.add_argument('--output_file', dest='outfile', default=os.path.join(os.getcwd(), 'mapper_output.json'),
|
||
|
|
help='Path to output file to store results.')
|
||
|
|
parser.add_argument('--mapper_func', dest='func', default='AvgDrawFrame',
|
||
|
|
help='Name of javascript mapper function in systrace_parser.html.')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
# Update the source if needed.
|
||
|
|
if not os.path.exists(catapult_src_dir):
|
||
|
|
# Pull the latest source from the upstream git.
|
||
|
|
git_args = ['git', 'clone', upstream_git, catapult_src_dir]
|
||
|
|
p = subprocess.Popen(git_args, stdout=subprocess.PIPE, cwd=script_dir)
|
||
|
|
p.communicate()
|
||
|
|
if p.wait() != 0:
|
||
|
|
print 'Failed to checkout source from upstream git.'
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
mapper_func_file = os.path.join(script_dir, 'systrace_parser.html')
|
||
|
|
path_to_process_traces = os.path.join(catapult_src_dir, 'trace_processor/bin/process_traces')
|
||
|
|
run_command = path_to_process_traces + " --mapper_handle " + mapper_func_file + ":" + args.func + " --output_file " + args.outfile + " " + args.trace_file_or_dir
|
||
|
|
print run_command
|
||
|
|
sys.exit(os.system(run_command))
|
||
|
|
|