workshop.py

from Probability import DistribTriang
import config

class Workshop:
    def __init__(self, name, position, tcycle = 0):
        self.name = name
        self.ina = []
        self.oua = []
        self.tcycle = tcycle
        self.topLeftCorner = position
        self.dailyQty = -1
        self._num = 0
        self.loadQty = -1
        if(name=="Mine"):
            self.distribution = DistribTriang(config._mdt[0],config._mdt[1],config._mdt[2])
            self.dailyQty = self.distribution.mcInv()
        elif (name == "Unloading Dock"):
            self.loadQty = 0

    def setBehavior(self, bt):
        self.behavior = bt

    def addInEdge(self, transit):
        self.ina.append(transit)

    def addOutEdge(self, transit):
        self.oua.append(transit)

    def render(self,g,w,h):
        bbox = (self.topLeftCorner["x"], self.topLeftCorner["y"], self.topLeftCorner["x"] + w,
                self.topLeftCorner["y"] + h)
        g.create_rectangle(bbox, width=1, outline="black", fill="gray70")
        g.create_text((bbox[0] + w/2, bbox[1] + 20), text=str(self.name), font=('times', 12, 'bold'), fill='black')
        if(self.dailyQty >= 0):
            g.create_text((bbox[0] + w/2, bbox[1] + 40), text=str(self.dailyQty), font=('times', 12), fill='black')
        elif(self._num >0):
            g.create_text((bbox[0] + w/2, bbox[1] + 40), text="Transport: "+str(self._num), font=('times', 12), fill='dark green')
        if(self.loadQty >= 0):
            g.create_text((bbox[0] + w/2, bbox[1] + 60), text=str(self.loadQty), font=('times', 12), fill='black')


    def update(self):
        self.behavior.update()

    def computeDailyQty(self):
        if self.distribution :
            self.dailyQty += self.distribution.mcInv()