82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
from astvisitors import *
|
|
|
|
MODEL = """
|
|
------------------------------------------------------------------------
|
|
-- Generated using Kripkomat
|
|
------------------------------------------------------------------------
|
|
MODULE {name}
|
|
VAR
|
|
state : {{ {states} }};
|
|
{variables}
|
|
ASSIGN
|
|
init(state) := {initial};
|
|
next(state) :=
|
|
case
|
|
{transitions}
|
|
TRUE : state;
|
|
esac;
|
|
DEFINE
|
|
{properties}
|
|
"""
|
|
|
|
VARIABLE = " {n}:{t};"
|
|
|
|
TRAN = " (state = {n}) & ({cond}) : {m};"
|
|
PROP = """ -- Property "{prop}"
|
|
{alias} := {logic};
|
|
"""
|
|
PROP_LOGIC = "state = {state}"
|
|
|
|
LTL_FORM = """LTLSPEC
|
|
{ltl};
|
|
"""
|
|
|
|
class ModelBuilder:
|
|
def __init__(self, states, initial, transitions, properties, ltls=[], name="main"):
|
|
self._name = name
|
|
self._states = states
|
|
self._initial = initial
|
|
self._tran = transitions
|
|
self._props = properties
|
|
self._ltls = ltls
|
|
|
|
def generate(self):
|
|
cg = c_generator.CGenerator()
|
|
cond_tf = NuSmvConditionTransformer()
|
|
|
|
# build model
|
|
states_decl = ",".join(self._states)
|
|
transitions = []
|
|
|
|
# find variables in the condition
|
|
varextract = NuSmvVariableExtractor()
|
|
|
|
for tran in self._tran:
|
|
cond = cond_tf.transform(tran.condition)
|
|
expr = cg.visit(cond)
|
|
transition = TRAN.format(n=tran.state_from, m=tran.state_to, cond=expr)
|
|
transitions.append(transition)
|
|
varextract.visit(tran.condition)
|
|
|
|
variables = []
|
|
for v,t in varextract.variables.items():
|
|
variables.append(VARIABLE.format(n=v, t=t))
|
|
|
|
properties = []
|
|
#for alias,(states,prop) in self._props.items():
|
|
# logic = " | ".join([PROP_LOGIC.format(state=x) for x in states])
|
|
# prop_str = PROP.format(prop=prop, alias=alias, logic=logic)
|
|
# properties.append(prop_str)
|
|
|
|
out = MODEL.format(name=self._name,
|
|
states=states_decl,
|
|
variables="\n".join(variables),
|
|
initial=self._initial,
|
|
transitions="\n".join(transitions),
|
|
properties="\n".join(properties))
|
|
|
|
# add LTL formulas
|
|
for ltl in self._ltls:
|
|
out += LTL_FORM.format(ltl=ltl)
|
|
return out
|