muMLE/examples/semantics/translational/renderer.py

from api.od import ODAPI
from concrete_syntax.graphviz.renderer import render_object_diagram, make_graphviz_id
from concrete_syntax.graphviz.make_url import show_graphviz
from examples.petrinet.renderer import render_petri_net_to_dot
from examples.semantics.operational.port.renderer import render_port_to_dot

# COLORS
PLACE_BG = "#DAE8FC" # fill color
PLACE_FG = "#6C8EBF" # font, line, arrow
BERTH_BG = "#FFF2CC"
BERTH_FG = "#D6B656"
CAPACITY_BG = "#F5F5F5"
CAPACITY_FG = "#666666"
WORKER_BG = "#D5E8D4"
WORKER_FG = "#82B366"
GENERATOR_BG = "#FFE6CC"
GENERATOR_FG = "#D79B00"
CLOCK_BG = "black"
CLOCK_FG = "white"

def graphviz_style_fg_bg(fg, bg):
    return f"style=filled,fillcolor=\"{bg}\",color=\"{fg}\",fontcolor=\"{fg}\""

def render_port(state, m, mm):
    dot = render_object_diagram(state, m, mm,
        reify=True,
        only_render=[
            # Only render these types
            "Place", "Berth", "CapacityConstraint", "WorkerSet", "Generator", "Clock",
            "connection", "capacityOf", "canOperate", "generic_link",
            # Petri Net types not included (they are already rendered by other function)
            # Port-State-types not included to avoid cluttering the diagram, but if you need them, feel free to add them.
        ],
        type_to_style={
            "Place": graphviz_style_fg_bg(PLACE_FG, PLACE_BG),
            "Berth": graphviz_style_fg_bg(BERTH_FG, BERTH_BG),
            "CapacityConstraint": graphviz_style_fg_bg(CAPACITY_FG, CAPACITY_BG),
            "WorkerSet": "shape=oval,"+graphviz_style_fg_bg(WORKER_FG, WORKER_BG),
            "Generator": "shape=parallelogram,"+graphviz_style_fg_bg(GENERATOR_FG, GENERATOR_BG),
            "Clock": graphviz_style_fg_bg(CLOCK_FG, CLOCK_BG),

            # same blue as Place, thick line:
            "connection": f"color=\"{PLACE_FG}\",fontcolor=\"{PLACE_FG}\",penwidth=2.0",

            # same grey as CapacityConstraint
            "capacityOf": f"color=\"{CAPACITY_FG}\",fontcolor=\"{CAPACITY_FG}\"",

            # same green as WorkerSet
            "canOperate": f"color=\"{WORKER_FG}\",fontcolor=\"{WORKER_FG}\"",

            # purple line
            "generic_link": "color=purple,fontcolor=purple,arrowhead=onormal",
        }
    )
    return dot

def render_port_and_petri_net(state, m, mm):
    od = ODAPI(state, m, mm)
    dot = ""
    dot += "// petri net:\n"
    dot += render_petri_net_to_dot(od)
    dot += "\n// the rest:\n"
    dot += render_port(state, m, mm)
    return dot


def show_port_and_petri_net(state, m, mm, engine="dot"):
    show_graphviz(render_port_and_petri_net(state, m, mm), engine=engine)