Initial commit

models/seq_devstone/model.py

@@ -0,0 +1,101 @@
+import sys
+sys.path.append("../../src/")
+
+from infinity import INFINITY
+from simulator import Simulator
+from DEVS import *
+import random
+
+class Event:
+    def __init__(self, eventSize):
+        self.eventSize = eventSize
+
+    def copy(self):
+        return Event(self.eventSize)
+
+class ProcessorState:
+    def __init__(self):
+        self.event1_counter = INFINITY
+        self.event1 = None
+        self.queue = []
+
+class Processor(AtomicDEVS):
+    def __init__(self, name, randomta):
+        AtomicDEVS.__init__(self, name)
+        self.recv_event1 = self.addInPort("in_event1")
+        self.send_event1 = self.addOutPort("out_event1")
+        self.state = ProcessorState()
+        self.randomta = randomta
+                
+    def timeAdvance(self):
+        return self.state.event1_counter
+
+    def intTransition(self):
+        self.state.event1_counter -= self.timeAdvance()
+        if self.state.event1_counter == 0 and self.state.queue == []:
+            self.state.event1_counter = INFINITY
+            self.state.event1 = None
+        else:
+            self.state.event1 = self.state.queue.pop()
+            self.state.event1_counter = round(random.uniform(0.75, 1.25), 4) if self.randomta else 1.0
+        return self.state
+
+    def extTransition(self, inputs):
+        self.state.event1_counter -= self.elapsed
+        #Only one element, so exploit this
+        ev1 = inputs[self.recv_event1][0]
+        if self.state.event1 is not None:
+            self.state.queue.append(ev1)
+        else:
+            self.state.event1 = ev1
+            self.state.event1_counter = round(random.uniform(0.75, 1.25), 4) if self.randomta else 1.0
+        return self.state
+
+    def outputFnc(self):
+        return {self.send_event1: [self.state.event1]}
+
+class Generator(AtomicDEVS):
+    def __init__(self):
+        AtomicDEVS.__init__(self, "Generator")
+        self.state = "gen_event1"
+        self.send_event1 = self.addOutPort("out_event1")
+                
+    def timeAdvance(self):
+        return 1
+
+    def intTransition(self):
+        return self.state
+
+    def outputFnc(self):
+        return {self.send_event1: [Event(1)]}
+
+class CoupledRecursion(CoupledDEVS):
+    def __init__(self, width, depth, randomta):
+        CoupledDEVS.__init__(self, "Coupled" + str(depth))
+        self.recv_event1 = self.addInPort("in_event1")
+        self.send_event1 = self.addOutPort("out_event1")
+
+        if depth > 1:
+            self.recurse = self.addSubModel(CoupledRecursion(width, depth-1, randomta))
+            self.connectPorts(self.recv_event1, self.recurse.recv_event1)
+
+        for i in range(width):
+            processor = self.addSubModel(Processor("Processor%s_%s" % (depth, i), randomta))
+            if i == 0:
+                if depth > 1:
+                    self.connectPorts(self.recurse.send_event1, processor.recv_event1)
+                else:
+                    self.connectPorts(self.recv_event1, processor.recv_event1)
+            else:
+                self.connectPorts(prev.send_event1, processor.recv_event1)
+            prev = processor
+        self.connectPorts(prev.send_event1, self.send_event1)
+
+class DEVStone(CoupledDEVS):
+    def __init__(self, width, depth, randomta):
+        random.seed(1)
+        CoupledDEVS.__init__(self, "DEVStone")
+        self.generator = self.addSubModel(Generator())
+        self.recurse = self.addSubModel(CoupledRecursion(width, depth, randomta))
+
+        self.connectPorts(self.generator.send_event1, self.recurse.recv_event1)