import gzip | |
import json | |
import os | |
import sys | |
trace, trace_dir = sys.argv[1:] | |
with open(os.path.join(trace_dir, trace + '.json'), 'rb') as f: | |
trace_json = json.loads(f.read()) | |
# Concatenate trace cpp source into a single gz | |
with open('gen/tracegz_' + sys.argv[1] + '.gz', 'wb') as f: | |
with gzip.GzipFile(fileobj=f, mode='wb', compresslevel=9, mtime=0) as fgz: | |
for fn in trace_json['TraceFiles']: | |
if fn.endswith('.cpp'): | |
with open(os.path.join(trace_dir, fn), 'rb') as fcpp: | |
fgz.write(fcpp.read()) |