#!/usr/bin/env python
"""
Machineshop Model

An example showing inrterrupts and priority queuing.

Scenario:
A workshop has n identical machines. A stream of jobs (enough to keep the
machines busy) arrives. Each machine breaks down periodically. Repairs are carried out
by one repairman. The repairman has other, less important tasks to perform, too. Once he starts one of those,
he completes it before starting with the machine repair. The workshop works continously.
"""


from __future__ import generators
from SimPy.Simulation import *

class Machine(Process):
    def __init__(self,name):
        Process.__init__(self,name)
        myBreaker=Breakdown(self)
        activate(myBreaker,myBreaker.breakmachine())
        self.partsMade=0

    def working(self):
        while True:
            yield hold,self,timePerPart()
            if self.interrupted():
                # broken down
                parttimeleft=self.interruptLeft
                yield request,self,repairman,1
                yield hold,self,repairtime
                #repaired
                yield release,self,repairman
                yield hold,self,parttimeleft
                # part completed
                self.partsMade += 1
            else:
                #part made
                self.partsMade += 1

class Breakdown(Process):
    def __init__(self,myMachine):
        Process.__init__(self)
        self.myMachine=myMachine

    def breakmachine(self):
        while True:
            yield hold,self,timeToFailure()
            self.interrupt(self.myMachine)

class OtherJobs(Process):
    def __init__(self):
        Process.__init__(self)  

    def doingJobs(self):
        while True:
            yield request,self,repairman,0
            #starts working on jobs
            yield hold,self,jobDuration
            yield release,self,repairman
            
## Model data
repairtime=30.0           # minutes
import random
random.seed()
processingTimeMean=10.0   # minutes
processingTimeSigma=2.0
timeToFailureMean = 300.0 # minutes

def timePerPart():
    return random.normalvariate(processingTimeMean,processingTimeSigma)

mean=1/timeToFailureMean
def timeToFailure():
    return random.expovariate(mean)
    
jobDuration=30           # minutes
nrMachines=10

initialize()
repairman=Resource(capacity=1,qType=PriorityQ)

m={}
for i in range(nrMachines):
    m[i+1]=Machine(name="Machine %s" %(i+1))
    activate(m[i+1],m[i+1].working())
oj=OtherJobs()
activate(oj,oj.doingJobs())
weeks = 4 # weeks
simTime = weeks*24*60*7 ## minutes
simulate(until=simTime) ## minutes

print "Machineshop results after %s weeks"%weeks
for i in range(nrMachines):
    print "Machine %s: %s" %(i+1,m[i+1].partsMade)