diff options
Diffstat (limited to 'android_bench_suite/gen_json.py')
-rwxr-xr-x | android_bench_suite/gen_json.py | 96 |
1 files changed, 96 insertions, 0 deletions
diff --git a/android_bench_suite/gen_json.py b/android_bench_suite/gen_json.py new file mode 100755 index 00000000..e1252933 --- /dev/null +++ b/android_bench_suite/gen_json.py @@ -0,0 +1,96 @@ +#!/usr/bin/env python2 +# +# Copyright 2017 The Chromium OS Authors. All rights reserved. +# Use of this source code is governed by a BSD-style license that can be +# found in the LICENSE file. +# +# pylint: disable=cros-logging-import + +"""Script to help generate json format report from raw data.""" +from __future__ import print_function + +import argparse +import config +import json +import logging +import sys + +# Turn the logging level to INFO before importing other autotest +# code, to avoid having failed import logging messages confuse the +# test_droid user. +logging.basicConfig(level=logging.INFO) + + +def _parse_arguments_internal(argv): + parser = argparse.ArgumentParser(description='Convert result to JSON' + 'format') + + parser.add_argument( + '-b', '--bench', help='Generate JSON format file for which benchmark.') + + parser.add_argument( + '-i', '--input', help='Specify the input result file name.') + + parser.add_argument( + '-o', '--output', help='Specify the output JSON format result file') + + parser.add_argument( + '-p', + '--platform', + help='Indicate the platform(experiment or device) name ' + 'to be shown in JSON') + + parser.add_argument( + '--iterations', + type=int, + help='How many iterations does the result include.') + return parser.parse_args(argv) + +# Collect data and generate JSON {} tuple from benchmark result +def collect_data(infile, bench, it): + result_dict = {} + with open(infile + str(it)) as fin: + if bench not in config.bench_parser_dict: + logging.error('Please input the correct benchmark name.') + raise ValueError('Wrong benchmark name: %s' % bench) + parse = config.bench_parser_dict[bench] + result_dict = parse(bench, fin) + return result_dict + +# If there is no original output file, create a new one and init it. +def create_outfile(outfile, bench): + with open(outfile, 'w') as fout: + obj_null = {'data': {bench.lower(): []}, 'platforms': []} + json.dump(obj_null, fout) + +# Seek the original output file and try to add new result into it. +def get_outfile(outfile, bench): + try: + return open(outfile) + except IOError: + create_outfile(outfile, bench) + return open(outfile) + +def main(argv): + arguments = _parse_arguments_internal(argv) + + bench = arguments.bench + infile = arguments.input + outfile = arguments.output + platform = arguments.platform + iteration = arguments.iterations + + result = [] + for i in range(iteration): + result += collect_data(infile, bench, i) + + with get_outfile(outfile, bench) as fout: + obj = json.load(fout) + obj['platforms'].append(platform) + obj['data'][bench.lower()].append(result) + with open(outfile, 'w') as fout: + json.dump(obj, fout) + + +if __name__ == '__main__': + main(sys.argv[1:]) |