create_clock_tree_instance.py

#!/usr/bin/env python3

from graph.core import *

from graph.load_xml import load_graph_types_and_instances
from graph.save_xml_stream import save_graph
import sys
import os
import math



import os
appBase=os.path.dirname(os.path.realpath(__file__))

src=appBase+"/clock_tree_graph_type.xml"
(graphTypes,graphInstances)=load_graph_types_and_instances(src,src)

d=4
b=2
maxTicks=100
if len(sys.argv)>1:
    d=int(sys.argv[1])
if len(sys.argv)>2:
    b=int(sys.argv[2])
if len(sys.argv)>3:
    maxTicks=int(sys.argv[3])

graphType=graphTypes["clock_tree"]
rootType=graphType.device_types["root"]
branchType=graphType.device_types["branch"]
leafType=graphType.device_types["leaf"]
#listenerType=graphType.device_types["output_listener"]

instName="clock_{}_{}".format(d,b)

properties={"max_ticks":maxTicks}
res=GraphInstance(instName, graphType, properties)

nodes={}

def create(prefix, parent,depth):
    node=None
    if depth==0:
        node=DeviceInstance(res, "{}_leaf".format(prefix), leafType, None, None)
        res.add_device_instance(node)
        res.add_edge_instance(EdgeInstance(res,node,"tick_in",parent,"tick_out", None))
        res.add_edge_instance(EdgeInstance(res,parent,"ack_in",node,"ack_out",None))
    else:
        if depth==d:
            node=DeviceInstance(res, prefix, rootType, {"fanout":b}, None)
        else:
            node=DeviceInstance(res, prefix, branchType, {"fanout":b}, None)
        res.add_device_instance(node)
        for i in range(b):
            child=create("{}_{}".format(prefix,i), node, depth-1)

        if depth!=d:
            res.add_edge_instance(EdgeInstance(res,node,"tick_in", parent, "tick_out", None))
            res.add_edge_instance(EdgeInstance(res,parent,"ack_in", node,"ack_out", None))

    return node


root=create("root",None,d)

#listener=DeviceInstance(res, "listener", listenerType)
#res.add_device_instance(listener)
#res.add_edge_instance(EdgeInstance(res,listener,"on_root_output",root,"output"))

save_graph(res,sys.stdout)