1import gzip 2import json 3import os 4import sys 5 6trace, trace_dir = sys.argv[1:] 7 8with open(os.path.join(trace_dir, trace + '.json'), 'rb') as f: 9 trace_json = json.loads(f.read()) 10 11# Concatenate trace cpp source into a single gz 12with open('gen/tracegz_' + sys.argv[1] + '.gz', 'wb') as f: 13 with gzip.GzipFile(fileobj=f, mode='wb', compresslevel=9, mtime=0) as fgz: 14 for fn in trace_json['TraceFiles']: 15 if fn.endswith('.cpp'): 16 with open(os.path.join(trace_dir, fn), 'rb') as fcpp: 17 fgz.write(fcpp.read()) 18