101 lines
3.4 KiB
Python
101 lines
3.4 KiB
Python
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)
|