68 lines
2.6 KiB
Python
68 lines
2.6 KiB
Python
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)
|