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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
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 xrange(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:])
|
