import sys
from bedit_core.serialization import load, save
from bedit_core.bondgraph import causality_inference
from bedit_core.modelica import compose_modelica_model

def main(path: str):
    doc = load(path)
    for id, root in doc.root.items():
        doc.root[id] = causality_inference(doc.root[id])
        composition = compose_modelica_model(doc.root[id])
        print(composition.modelica)
    save(doc, path)

if __name__ == "__main__":
    main(sys.argv[1])