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())