summaryrefslogtreecommitdiff
path: root/gen_json.py
diff options
context:
space:
mode:
Diffstat (limited to 'gen_json.py')
-rwxr-xr-xgen_json.py100
1 files changed, 50 insertions, 50 deletions
diff --git a/gen_json.py b/gen_json.py
index ad617ff4..812afd8d 100755
--- a/gen_json.py
+++ b/gen_json.py
@@ -22,75 +22,75 @@ logging.basicConfig(level=logging.INFO)
def _parse_arguments_internal(argv):
- parser = argparse.ArgumentParser(description='Convert result to JSON'
- 'format')
+ 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(
+ '-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(
+ '-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(
+ '-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(
+ '-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)
+ 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
+ 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)
+ 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)
+ try:
+ return open(outfile)
+ except IOError:
+ create_outfile(outfile, bench)
+ return open(outfile)
def main(argv):
- arguments = _parse_arguments_internal(argv)
+ arguments = _parse_arguments_internal(argv)
- bench = arguments.bench
- infile = arguments.input
- outfile = arguments.output
- platform = arguments.platform
- iteration = arguments.iterations
+ 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)
+ 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)
+ 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:])
+ main(sys.argv[1:])