#出租車(chē)隊(duì)運(yùn)營(yíng)仿真
import collections

#優(yōu)先級(jí)列表，用來(lái)維護(hù)事件列表畦贸，用于實(shí)現(xiàn)將所有事件按照發(fā)生先后順序執(zhí)行
class PriorityList:
    def __init__(self):
        self.list=[]

    #獲取事件隊(duì)列中最早發(fā)生的事件
    def get_first(self):
        #將事件列表中所有發(fā)生事件按照發(fā)生事件排序后輸出排序后的第一個(gè)值
        self.list.sort(key=lambda elem:elem.time)
        return self.list[0]

    #向列表中添加事件
    def add_list(self,event):
        self.list.append(event)

    #從事件隊(duì)列中刪除事件
    def del_list(self,event):
        self.list.remove(event)

    #事件列表是否為空
    def is_empty(self):
        return self.list==[]

#基本單位
#time 事件發(fā)生事件  proc 所屬出租車(chē)協(xié)程  action 出租車(chē)活動(dòng)事件
Event=collections.namedtuple('Event','time proc action')

#出租車(chē)協(xié)程楞捂，模擬出租車(chē)的各種活動(dòng)
def taxi_process(ident,trips,start_time=0):
    """
    每次改變狀態(tài)時(shí)創(chuàng)建事件，將控制權(quán)讓給仿真器
    """
    time=yield Event(start_time,ident,'level garage')
    for i in range(trips):
        time=yield Event(time,ident,'pick up passenger')
        time=yield Event(time,ident,'drop off passenger')

    yield Event(time,ident,'going home')

#測(cè)試taxi_process,模擬出租車(chē)運(yùn)行
def main_01():
    taxi=taxi_process(ident=13,trips=2,start_time=0)

    current_state=taxi.send(None) #預(yù)激出租車(chē)協(xié)程
    print(current_state)  #level garge 開(kāi)出車(chē)庫(kù)

    #第1次接客(發(fā)生在離開(kāi)車(chē)庫(kù)7分鐘后),載客行駛事件23分鐘
    current_state=taxi.send(current_state.time+7)
    print(current_state) #pick up passenger 接到乘客

    current_state=taxi.send(current_state.time+23)
    print(current_state) #drop off passenger 卸載乘客

    #第2次接客(發(fā)生在離開(kāi)車(chē)庫(kù)5分鐘后),載客行駛時(shí)間48分鐘
    current_state=taxi.send(current_state.time+5)
    print(current_state) #pick up passenger 接到乘客

    current_state=taxi.send(current_state.time+48)
    print(current_state) #drop off passenger 卸載乘客

    #返回車(chē)庫(kù)(發(fā)生在最后一次卸載乘客10分鐘后)
    current_state=taxi.send(current_state.time+10)
    print(current_state) #going home 返回車(chē)庫(kù)

    try:
        current_state=taxi.send(current_state.time+1)
        print(current_state)
    except StopIteration as e:
        print('此時(shí)出租車(chē)協(xié)程執(zhí)行完畢胶坠，拋出StopInteration')

"""
執(zhí)行:
    main_01()
    
輸出結(jié)果如下:

    Event(time=0, proc=13, action='level garage')
    Event(time=7, proc=13, action='pick up passenger')
    Event(time=30, proc=13, action='drop off passenger')
    Event(time=35, proc=13, action='pick up passenger')
    Event(time=83, proc=13, action='drop off passenger')
    Event(time=93, proc=13, action='going home')
    此時(shí)出租車(chē)協(xié)程執(zhí)行完畢沈善，拋出StopInteration
"""

#出租車(chē)模擬類(lèi)
import random

class Simulator:
    #計(jì)算指定aciton到下一個(gè)action之間的耗時(shí)
    def compute_duration(self,previous_action):
        if previous_action=='drop off passenger':
            return random.choice([1,2])
        elif previous_action=='pick up passenger':
            return random.choice([7,8])
        elif previous_action=='level garage':
            return random.choice([10,15])
        elif previous_action=='going home':
            return random.choice([20,30])
        else:
            raise Exception('Unknown Previoud_Action')

    def __init__(self,procs_dict):
        self.events=PriorityList()
        self.procs=procs_dict

    #仿真系統(tǒng)主邏輯
    def run(self,end_time):
        #預(yù)激每一個(gè)出租車(chē)協(xié)程
        for _,proc in self.procs.items():
            #每個(gè)出租車(chē)協(xié)程的第一個(gè)事件
            first_event=proc.send(None) #預(yù)激
            #將每個(gè)出租車(chē)協(xié)程的第一個(gè)事件放入事件列表
            self.events.add_list(first_event)


        #這個(gè)仿真系統(tǒng)的事件主循環(huán)
        sim_time=0
        #遍歷并處理每個(gè)事件
        while sim_time<end_time:#判斷運(yùn)行時(shí)間是否超時(shí)
            #事件列表為空,所有事件執(zhí)行完畢
            if self.events.is_empty():
                print('******** end all events ********')
                return

            current_event=self.events.get_first() #獲取事件列表中最早發(fā)生的事件

            #處理當(dāng)前事件（此處用打印來(lái)模擬處理過(guò)程）
            sim_time,proc_id,previous_action=current_event
            print('taxi:',proc_id,proc_id*'    ',current_event) #打印當(dāng)前事件
            #從事件標(biāo)記列表中移除已處理過(guò)的事件
            self.events.del_list(current_event)

            active_proc=self.procs[proc_id] #獲取這個(gè)事件所屬的出租車(chē)協(xié)程
            next_time=sim_time+self.compute_duration(previous_action) #計(jì)算下一個(gè)事件的發(fā)生時(shí)間

            try:
                next_event=active_proc.send(next_time) #向出租車(chē)協(xié)程發(fā)送參數(shù)(參數(shù)表示下次事件發(fā)生的時(shí)間)
            except StopIteration:
                #如果觸發(fā)這個(gè)異常表示出租車(chē)協(xié)程執(zhí)行完畢
                del self.procs[proc_id]
            else:
                #將新事件添加到事件列表
                self.events.add_list(next_event)
        else:
            #執(zhí)行時(shí)間超時(shí)后的處理
            print('---->執(zhí)行超時(shí)<----')


#測(cè)試出租車(chē)協(xié)程能否正常執(zhí)行
def main_02():
    data_dict={
        1:taxi_process(ident=1,trips=3,start_time=0),
        2:taxi_process(ident=2,trips=4,start_time=30),
        3:taxi_process(ident=3,trips=2,start_time=13),
    }

    sim1=Simulator(data_dict)
    sim1.run(100) #時(shí)間足夠執(zhí)行完成所有的出租車(chē)協(xié)程

'''
執(zhí)行:
    main_02()
輸出結(jié)果:
    taxi: 1      Event(time=0, proc=1, action='level garage')
    taxi: 1      Event(time=10, proc=1, action='pick up passenger')
    taxi: 3              Event(time=13, proc=3, action='level garage')
    taxi: 1      Event(time=17, proc=1, action='drop off passenger')
    taxi: 1      Event(time=18, proc=1, action='pick up passenger')
    taxi: 1      Event(time=26, proc=1, action='drop off passenger')
    taxi: 3              Event(time=28, proc=3, action='pick up passenger')
    taxi: 1      Event(time=28, proc=1, action='pick up passenger')
    taxi: 2          Event(time=30, proc=2, action='level garage')
    taxi: 1      Event(time=35, proc=1, action='drop off passenger')
    taxi: 3              Event(time=36, proc=3, action='drop off passenger')
    taxi: 1      Event(time=37, proc=1, action='going home')
    taxi: 3              Event(time=38, proc=3, action='pick up passenger')
    taxi: 2          Event(time=40, proc=2, action='pick up passenger')
    taxi: 3              Event(time=46, proc=3, action='drop off passenger')
    taxi: 2          Event(time=47, proc=2, action='drop off passenger')
    taxi: 3              Event(time=48, proc=3, action='going home')
    taxi: 2          Event(time=48, proc=2, action='pick up passenger')
    taxi: 2          Event(time=56, proc=2, action='drop off passenger')
    taxi: 2          Event(time=57, proc=2, action='pick up passenger')
    taxi: 2          Event(time=64, proc=2, action='drop off passenger')
    taxi: 2          Event(time=66, proc=2, action='pick up passenger')
    taxi: 2          Event(time=74, proc=2, action='drop off passenger')
    taxi: 2          Event(time=75, proc=2, action='going home')
    ******** end all events ********
'''

#測(cè)試設(shè)置超時(shí)時(shí)間是否有效
def main_03():
    data_dict={
        1:taxi_process(ident=1,trips=3,start_time=0),
        2:taxi_process(ident=2,trips=4,start_time=30),
        3:taxi_process(ident=3,trips=2,start_time=13),
    }

    sim1=Simulator(data_dict)
    sim1.run(30) #時(shí)間不足以執(zhí)行完成所有的出租車(chē)協(xié)程
'''
執(zhí)行
    main_03()
輸出結(jié)果:
    taxi: 1      Event(time=0, proc=1, action='level garage')
    taxi: 3              Event(time=13, proc=3, action='level garage')
    taxi: 1      Event(time=15, proc=1, action='pick up passenger')
    taxi: 1      Event(time=22, proc=1, action='drop off passenger')
    taxi: 3              Event(time=23, proc=3, action='pick up passenger')
    taxi: 1      Event(time=23, proc=1, action='pick up passenger')
    taxi: 2          Event(time=30, proc=2, action='level garage')
    taxi: 3              Event(time=30, proc=3, action='drop off passenger')
    taxi: 1      Event(time=30, proc=1, action='drop off passenger')
    ---->執(zhí)行超時(shí)<---- 
'''