prjxray/utils/sdfmerge.py

59 lines
1.6 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
import argparse
import json
from sdf_timing import sdfparse
def merge(timings_list, site):
merged_timings = dict()
for timings in timings_list:
divider = '/'
if 'divider' in timings['header']:
divider = timings['header']['divider']
for cell in timings['cells']:
for cell_instance in timings['cells'][cell]:
if site in cell_instance.split(divider):
if 'cells' not in merged_timings:
merged_timings['cells'] = dict()
if cell in merged_timings['cells']:
assert merged_timings['cells'][cell] == \
timings['cells'][cell], \
"Attempting to merge differing cells"
merged_timings['cells'][cell] = timings['cells'][cell]
return merged_timings
def main():
parser = argparse.ArgumentParser()
parser.add_argument(
'--sdfs', nargs='+', type=str, help="List of sdf files to merge")
parser.add_argument('--site', type=str, help="Site we want to merge")
parser.add_argument('--json', type=str, help="Debug JSON")
args = parser.parse_args()
timings_list = list()
for sdf in args.sdfs:
with open(sdf, 'r') as fp:
timing = sdfparse.parse(fp.read())
timings_list.append(timing)
merged_sdf = merge(timings_list, args.site)
if args.json is not None:
with open(args.json, 'w') as fp:
json.dump(merged_sdf, fp, indent=4, sort_keys=True)
if __name__ == '__main__':
main()