def visit_DiagramTest(self, test):
# helper class
self.writer.beginClass("InputEvent")
self.writer.beginConstructor()
self.writer.addFormalParameter("name")
self.writer.addFormalParameter("port")
self.writer.addFormalParameter("parameters")
self.writer.addFormalParameter("time_offset")
self.writer.beginMethodBody()
self.writer.addAssignment(GLC.SelfProperty("name"), "name")
self.writer.addAssignment(GLC.SelfProperty("port"), "port")
self.writer.addAssignment(GLC.SelfProperty("parameters"), "parameters")
self.writer.addAssignment(GLC.SelfProperty("time_offset"),
"time_offset")
self.writer.endMethodBody()
self.writer.endConstructor()
self.writer.endClass()
self.writer.beginClass("Test")
if test.input:
test.input.accept(self)
else:
self.writer.addStaticAttribute("input_events",
GLC.ArrayExpression())
if test.expected:
test.expected.accept(self)
else:
self.writer.addStaticAttribute("expected_events",
GLC.ArrayExpression())
self.writer.endClass()
def visit_ExitAction(self, exit_method):
exited_node = exit_method.parent_node
self.writer.beginMethod("exit_" + exited_node.full_name)
self.writer.beginMethodBody()
#If the exited node is composite take care of potential history and the leaving of descendants
if exited_node.is_composite:
#handle history
if exited_node.save_state_on_exit:
self.writer.addAssignment(
GLC.MapIndexedExpression(
GLC.SelfProperty("history_state"),
GLC.SelfProperty(exited_node.full_name)),
GLC.MapIndexedExpression(
GLC.SelfProperty("current_state"),
GLC.SelfProperty(exited_node.full_name)))
#Take care of leaving children
children = exited_node.children
if exited_node.is_parallel_state:
for child in children:
if not child.is_history:
self.writer.add(
GLC.FunctionCall(
GLC.SelfProperty("exit_" + child.full_name)))
else:
for child in children:
if not child.is_history:
self.writer.beginIf(
GLC.ArrayContains(
GLC.MapIndexedExpression(
GLC.SelfProperty("current_state"),
GLC.SelfProperty(exited_node.full_name)),
GLC.SelfProperty(child.full_name)))
self.writer.add(
GLC.FunctionCall(
GLC.SelfProperty("exit_" + child.full_name)))
self.writer.endIf()
# take care of any AFTER events
for transition in exited_node.transitions:
trigger = transition.getTrigger()
if trigger.isAfter():
self.writer.add(
GLC.MapRemoveElement(GLC.SelfProperty("timers"),
str(trigger.getAfterIndex())))
#Execute user-defined exit action if present
if exit_method.action:
exit_method.action.accept(self)
#Adjust state
self.writer.addAssignment(
GLC.MapIndexedExpression(
GLC.SelfProperty("current_state"),
GLC.SelfProperty(exited_node.parent.full_name)),
GLC.ArrayExpression()) # SPECIAL CASE FOR ORTHOGONAL??
self.writer.endMethodBody()
self.writer.endMethod()
def visit_DiagramTestEvent(self, event):
self.writer.add(
GLC.NewExpression("Event", [
GLC.String(event.name),
GLC.String(event.port),
GLC.ArrayExpression(event.parameters)
]))
def writeTransitionCondition(self, transition, index):
trigger = transition.getTrigger()
self.writer.addAssignment(
GLC.LocalVariableDeclaration("enabled_events"),
GLC.FunctionCall(GLC.SelfProperty("getEnabledEvents")))
if not trigger.isUC():
self.writer.beginForLoopIterateArray("enabled_events", "e")
condition = GLC.EqualsExpression(
GLC.Property(GLC.ForLoopCurrentElement("enabled_events", "e"),
"name"), GLC.String(trigger.getEvent()))
if trigger.getPort() != "":
condition = GLC.AndExpression(
condition,
GLC.EqualsExpression(
GLC.Property(
GLC.ForLoopCurrentElement("enabled_events", "e"),
"port"), GLC.String(trigger.getPort())))
self.writer.beginIf(condition)
# evaluate guard
if transition.hasGuard():
# handle parameters for guard evaluation
if not transition.getTrigger().isUC():
self.writer.addAssignment(
GLC.LocalVariableDeclaration("parameters"),
GLC.Property(
GLC.ForLoopCurrentElement("enabled_events", "e"),
"parameters"))
self.writeFormalEventParameters(transition)
self.writer.startRecordingExpression()
transition.getGuard().accept(self) # --> visit_Expression
expr = self.writer.stopRecordingExpression()
self.writer.beginIf(expr)
if trigger.isUC():
params_expr = GLC.ArrayExpression()
else:
params_expr = GLC.Property(
GLC.ForLoopCurrentElement("enabled_events", "e"), "parameters")
self.writer.add(
GLC.FunctionCall(
GLC.Property(GLC.SelfProperty("small_step"), "addCandidate"), [
GLC.SelfProperty("transition_" +
transition.parent_node.full_name + "_" +
str(index)), params_expr
]))
self.writer.add(GLC.ReturnStatement(GLC.TrueExpression()))
if transition.hasGuard():
self.writer.endIf()
if not trigger.isUC():
self.writer.endIf()
self.writer.endForLoopIterateArray()
def visit_ClassDiagram(self, class_diagram):
header = (
"Generated by Statechart compiler by Glenn De Jonghe and Joeri Exelmans\n"
"\n"
"Date: " + time.asctime() + "\n")
if class_diagram.name or class_diagram.author or class_diagram.description:
header += "\n"
if class_diagram.author:
header += "Model author: " + class_diagram.author + "\n"
if class_diagram.name:
header += "Model name: " + class_diagram.name + "\n"
if class_diagram.description.strip():
header += "Model description:\n"
header += class_diagram.description.strip()
self.writer.addMultiLineComment(header)
self.writer.addVSpace()
self.writer.addInclude(
([GLC.RuntimeModuleIdentifier(), "statecharts_core"]))
if class_diagram.top.strip():
self.writer.addRawCode(class_diagram.top)
self.writer.addVSpace()
self.writer.beginPackage(class_diagram.name)
#visit children
for c in class_diagram.classes:
c.accept(self)
self.writer.beginClass("ObjectManager", ["ObjectManagerBase"])
self.writer.beginConstructor()
self.writer.addFormalParameter("controller")
self.writer.beginMethodBody()
self.writer.beginSuperClassConstructorCall("ObjectManagerBase")
self.writer.addActualParameter("controller")
self.writer.endSuperClassConstructorCall()
self.writer.endMethodBody()
self.writer.endConstructor()
self.writer.beginMethod("instantiate")
self.writer.addFormalParameter("class_name")
self.writer.addFormalParameter("construct_params")
self.writer.beginMethodBody()
for index, c in enumerate(class_diagram.classes):
self.writer.beginElseIf(
GLC.EqualsExpression("class_name", GLC.String(c.name)))
if c.isAbstract():
# cannot instantiate abstract class
self.writer.add(
GLC.ThrowExceptionStatement(
GLC.String("Cannot instantiate abstract class \"" +
c.name +
"\" with unimplemented methods \"" +
"\", \"".join(c.abstract_method_names) +
"\".")))
else:
new_expr = GLC.NewExpression(c.name,
[GLC.SelfProperty("controller")])
param_count = 0
for p in c.constructors[0].parameters:
new_expr.getActualParameters().add(
GLC.ArrayIndexedExpression("construct_params",
str(param_count)))
param_count += 1
self.writer.addAssignment(
GLC.LocalVariableDeclaration("instance"), new_expr)
self.writer.addAssignment(
GLC.Property("instance", "associations"),
GLC.MapExpression())
for a in c.associations:
a.accept(self)
self.writer.endElseIf()
self.writer.add(GLC.ReturnStatement("instance"))
self.writer.endMethodBody()
self.writer.endMethod()
self.writer.endClass() # ObjectManager
if self.platform == Platforms.Threads:
controller_sub_class = "ThreadsControllerBase"
if self.platform == Platforms.EventLoop:
controller_sub_class = "EventLoopControllerBase"
elif self.platform == Platforms.GameLoop:
controller_sub_class = "GameLoopControllerBase"
self.writer.beginClass("Controller", [controller_sub_class])
self.writer.beginConstructor()
for p in class_diagram.default_class.constructors[0].parameters:
p.accept(self)
if self.platform == Platforms.EventLoop:
self.writer.addFormalParameter("event_loop_callbacks")
self.writer.addFormalParameter("finished_callback",
GLC.NoneExpression())
elif self.platform == Platforms.Threads:
self.writer.addFormalParameter("keep_running",
GLC.TrueExpression())
self.writer.beginMethodBody()
self.writer.beginSuperClassConstructorCall(controller_sub_class)
self.writer.addActualParameter(
GLC.NewExpression("ObjectManager", [GLC.SelfExpression()]))
if self.platform == Platforms.EventLoop:
self.writer.addActualParameter("event_loop_callbacks")
self.writer.addActualParameter("finished_callback")
elif self.platform == Platforms.Threads:
self.writer.addActualParameter("keep_running")
self.writer.endSuperClassConstructorCall()
for i in class_diagram.inports:
self.writer.add(
GLC.FunctionCall(GLC.SelfProperty("addInputPort"),
[GLC.String(i)]))
for o in class_diagram.outports:
self.writer.add(
GLC.FunctionCall(GLC.SelfProperty("addOutputPort"),
[GLC.String(o)]))
actual_parameters = [
p.getIdent()
for p in class_diagram.default_class.constructors[0].parameters
]
self.writer.add(
GLC.FunctionCall(
GLC.Property(GLC.SelfProperty("object_manager"),
"createInstance"), [
GLC.String(class_diagram.default_class.name),
GLC.ArrayExpression(actual_parameters)
]))
self.writer.endMethodBody()
self.writer.endConstructor()
self.writer.endClass() # Controller
# Visit test node if there is one
if class_diagram.test:
class_diagram.test.accept(self)
self.writer.endPackage()
def visit_Class(self, class_node):
"""
Generate code for Class construct
"""
super_classes = []
if not class_node.super_class_objs:
# if none of the class' super classes is defined in the diagram,
# we have to inherit RuntimeClassBase
if class_node.statechart:
# only inherit RuntimeClassBase if class has a statechart
super_classes.append("RuntimeClassBase")
if class_node.super_classes:
for super_class in class_node.super_classes:
super_classes.append(super_class)
self.writer.beginClass(class_node.name, super_classes)
#visit constructor
for i in class_node.constructors:
i.accept(self)
self.writer.beginMethod("user_defined_constructor")
for p in class_node.constructors[0].getParams():
p.accept(self)
self.writer.beginMethodBody()
for super_class in class_node.super_classes:
# begin call
if super_class in class_node.super_class_objs:
self.writer.beginSuperClassMethodCall(
super_class, "user_defined_constructor")
else:
self.writer.beginSuperClassConstructorCall(super_class)
# write actual parameters
if super_class in class_node.constructors[
0].super_class_parameters:
for p in class_node.constructors[0].super_class_parameters[
super_class]:
self.writer.addActualParameter(p)
# end call
if super_class in class_node.super_class_objs:
self.writer.endSuperClassMethodCall()
else:
self.writer.endSuperClassConstructorCall()
self.writer.addRawCode(class_node.constructors[0].body)
self.writer.endMethodBody()
self.writer.endMethod()
#visit children
for i in class_node.destructors:
i.accept(self)
for i in class_node.methods:
i.accept(self)
if class_node.statechart:
self.writer.beginMethod("initializeStatechart")
self.writer.beginMethodBody()
for c in class_node.statechart.composites:
self.writer.addAssignment(
GLC.MapIndexedExpression(GLC.SelfProperty("current_state"),
GLC.SelfProperty(c.full_name)),
GLC.ArrayExpression())
if class_node.statechart.histories:
self.writer.addVSpace()
for node in class_node.statechart.combined_history_parents:
self.writer.addAssignment(
GLC.MapIndexedExpression(
GLC.SelfProperty("history_state"),
GLC.SelfProperty(node.full_name)),
GLC.ArrayExpression())
self.writer.addVSpace()
self.writer.addComment("Enter default state")
for default_node in class_node.statechart.root.defaults:
if default_node.is_composite:
self.writer.add(
GLC.FunctionCall(
GLC.SelfProperty("enterDefault_" +
default_node.full_name)))
elif default_node.is_basic:
self.writer.add(
GLC.FunctionCall(
GLC.SelfProperty("enter_" +
default_node.full_name)))
self.writer.endMethodBody()
self.writer.endMethod()
class_node.statechart.accept(self)
self.writer.endClass()
def visit_RaiseEvent(self, raise_event):
self.writer.startRecordingExpression()
self.writer.begin(GLC.NewExpression("Event"))
self.writer.addActualParameter(GLC.String(raise_event.getEventName()))
if raise_event.isOutput():
self.writer.addActualParameter(GLC.String(raise_event.getPort()))
else:
self.writer.addActualParameter(GLC.NoneExpression())
self.writer.end()
new_event_expr = self.writer.stopRecordingExpression()
self.writer.startRecordingExpression()
self.writer.beginArray()
if raise_event.isCD():
self.writer.add(GLC.SelfExpression())
for param in raise_event.getParameters():
param.accept(
self
) # -> visit_Expression will cause expressions to be added to array
self.writer.endArray()
parameters_array_expr = self.writer.stopRecordingExpression()
new_event_expr.getActualParameters().add(parameters_array_expr)
if raise_event.isNarrow():
self.writer.add(
GLC.FunctionCall(
GLC.Property(GLC.SelfProperty("big_step"),
"outputEventOM"),
[
GLC.NewExpression("Event", [
GLC.String("narrow_cast"),
GLC.NoneExpression(),
GLC.ArrayExpression([
GLC.SelfExpression(),
raise_event.getTarget(), new_event_expr
])
])
]))
elif raise_event.isLocal():
self.writer.add(
GLC.FunctionCall(GLC.SelfProperty("raiseInternalEvent"),
[new_event_expr]))
elif raise_event.isOutput():
self.writer.add(
GLC.FunctionCall(
GLC.Property(GLC.SelfProperty("big_step"), "outputEvent"),
[new_event_expr]))
elif raise_event.isCD():
self.writer.add(
GLC.FunctionCall(
GLC.Property(GLC.SelfProperty("big_step"),
"outputEventOM"), [new_event_expr]))
elif raise_event.isBroad():
self.writer.add(
GLC.FunctionCall(
GLC.Property(GLC.SelfProperty("big_step"),
"outputEventOM"), [
GLC.NewExpression("Event", [
GLC.String("broad_cast"),
GLC.NoneExpression(),
GLC.ArrayExpression([new_event_expr])
])
]))