Initial commit

models/dist_statesaving/model.py

@@ -0,0 +1,121 @@
+import sys
+sys.path.append("../../src/")
+
+from infinity import INFINITY
+from simulator import Simulator
+from DEVS import *
+import random
+
+class Event(object):
+    def __init__(self, eventSize):
+        self.eventSize = eventSize
+
+    def copy(self):
+        return Event(self.eventSize)
+
+class ProcessorState(object):
+    def __init__(self):
+        self.event1_counter = INFINITY
+        self.event1 = None
+        self.queue = []
+
+    def copy(self):
+        a = ProcessorState()
+        a.event1_counter = self.event1_counter
+        a.event1 = None if self.event1 is None else self.event1.copy()
+        a.queue = [i.copy() for i in self.queue]
+        return a
+
+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 GeneratorState(object):
+    def __init__(self):
+        pass
+
+    def copy(self):
+        return GeneratorState()
+
+class Generator(AtomicDEVS):
+    def __init__(self):
+        AtomicDEVS.__init__(self, "Generator")
+        self.state = GeneratorState()
+        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.generator1 = self.addSubModel(Generator(), 0)
+        self.generator2 = self.addSubModel(Generator(), 1)
+        self.generator3 = self.addSubModel(Generator(), 2)
+        self.recurse1 = self.addSubModel(CoupledRecursion(width, depth, randomta), 0)
+        self.recurse2 = self.addSubModel(CoupledRecursion(width, depth, randomta), 1)
+        self.recurse3 = self.addSubModel(CoupledRecursion(width, depth, randomta), 2)
+
+        self.connectPorts(self.generator1.send_event1, self.recurse1.recv_event1)
+        self.connectPorts(self.generator2.send_event1, self.recurse2.recv_event1)
+        self.connectPorts(self.generator3.send_event1, self.recurse3.recv_event1)