示例#1
0
def main():
    messageController = MessageController()

    simSettings = SimulatorSettings()
    sim = Simulator(simSettings, messageController)

    con = Controller(simSettings, sim.hospitalsLocations(), messageController)
    con.printState()

    for i in range(1000):
        sim.simulatorMianLoop()
        con.controllerMainLoop()
        in_data = input()
        if in_data == 'y':  #write y to show the current state
            con.printState()
示例#2
0
class MainWindow(QtWidgets.QMainWindow):
    def __init__(self, onj=None, *args, **kwargs):
        super(MainWindow, self).__init__(*args, **kwargs)
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        self.messageController = MessageController()
        self.simSettings = SimulatorSettings()
        self.sim = Simulator(self.simSettings, self.messageController)
        self.con = Controller(self.simSettings, self.sim.hospitalsLocations(),
                              self.sim.ambulancesDistribution(),
                              self.messageController)
        self.ui.Mplwidget
        self.ui.Mplwidget.draw_hospital_on_map(self.con.Hospitals)
        self.ui.hospwidget
        self.ui.hospwidget.draw_all_hosp_inf(self.con.Hospitals)
        self.tablica_komunikatow = []
        self.control_events = []
        self.observ_from_message = []
        self.ui.start.clicked.connect(self.start_sim)
        self.ui.pauza.clicked.connect(self.stop_sim)
        self.ui.reset.clicked.connect(self.reset_sim)
        self.ui.delay_time.setValue(50)
        self.ui.delay_time.valueChanged.connect(self.change_interval)
        self.timer = QtCore.QTimer(self,
                                   interval=self.ui.delay_time.value(),
                                   timeout=self.simulation)
        self.printedCar = []
        self.printedVirus = []
        self.iterationsNumber = 0
        #self.con.printState()

    @QtCore.pyqtSlot()
    def start_sim(self):
        QtCore.QTimer.singleShot(0, self.simulation)
        self.timer.start()

    @QtCore.pyqtSlot()
    def stop_sim(self):
        self.timer.stop()

    def reset_sim(self):
        del self.messageController
        del self.simSettings
        del self.sim
        del self.con
        self.ui.Mplwidget.clean_hospital_on_map()
        self.ui.hospwidget.clean_all_hosp_inf()
        self.clear_widget_scrol()
        self.messageController = MessageController()
        self.simSettings = SimulatorSettings()
        self.sim = Simulator(self.simSettings, self.messageController)
        self.con = Controller(self.simSettings, self.sim.hospitalsLocations(),
                              self.sim.ambulancesDistribution(),
                              self.messageController)
        self.ui.Mplwidget.draw_hospital_on_map(self.con.Hospitals)
        self.ui.hospwidget.draw_all_hosp_inf(self.con.Hospitals)
        self.printedCar.clear()
        self.printedVirus.clear()
        self.iterationsNumber = 0
        #self.con.printState()

    def change_interval(self):
        self.timer.setInterval(self.ui.delay_time.value())

    def simulation(self):
        self.sim.simulatorMianLoop(self.tablica_komunikatow,
                                   self.control_events)
        if self.messageController.readAllObservableEventsForSimulation():
            self.observ_from_message.append(
                self.messageController.readAllObservableEventsForSimulation())

        if len(self.observ_from_message) > 0:
            #print(self.observ_from_message)
            #if len(self.observ_from_message[0])>1:
            #    self.timer.stop()
            for event in self.observ_from_message[0]:
                #print(event)
                if event[0] == 'E1o':
                    self.ui.Mplwidget.draw_accident_on_map(tuple(event[1][0]))
                    self.printedVirus.append(tuple(event[1][0]))
                elif event[
                        0] == 'E4o':  #Karetka dojechała na miejsce zgłoszenia - usuń ze szpitala
                    self.ui.Mplwidget.remove_car_from_map(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                    self.printedCar.remove(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                    self.ui.Mplwidget.draw_car_on_map(tuple(event[1][2]))
                    self.printedCar.append(tuple(event[1][2]))
                elif event[
                        0] == 'E9o':  #Karetka wróciła do szpitala bez chorego
                    self.ui.Mplwidget.remove_car_from_map(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                    self.printedCar.remove(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                elif event[
                        0] == 'E6o':  #Karetka wróciła do szpitala z chorym - usuń ze szpitala
                    self.ui.Mplwidget.remove_car_from_map(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                    self.printedCar.remove(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                elif event[
                        0] == 'E10o':  #Karetka wraca do szpitala bez chorego
                    self.ui.Mplwidget.remove_accident_from_map(
                        tuple(event[1][2]))
                    self.printedCar.remove(tuple(event[1][2]))
                    self.ui.Mplwidget.remove_car_from_map(tuple(event[1][2]))
                    self.printedVirus.remove(tuple(event[1][2]))
                    self.ui.Mplwidget.draw_car_on_map(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                    self.printedCar.append(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
            self.observ_from_message.clear()

        if len(self.control_events) > 0:
            #print(self.control_events)
            for event in self.control_events[0]:
                #print('control', event)
                if event[0] == 'E1c':  #Karetka wyjeżdża ze szpitala
                    self.ui.Mplwidget.draw_car_on_map(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                    self.printedCar.append(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                elif event[0] == 'E2c':  #Karetka wraca do szpitala z chorym
                    self.ui.Mplwidget.remove_accident_from_map(
                        tuple(event[1][1]))
                    self.ui.Mplwidget.remove_car_from_map(tuple(event[1][1]))
                    self.printedCar.remove(tuple(event[1][1]))
                    self.printedVirus.remove(tuple(event[1][1]))
                    self.ui.Mplwidget.draw_car_on_map(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                    self.printedCar.append(
                        (self.con.Hospitals[event[1][0] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][0] - 1].location[1] + 1))
                else:
                    self.ui.Mplwidget.draw_car_on_map(
                        (self.con.Hospitals[event[1][1] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][1] - 1].location[1] + 1))
                    self.printedCar.append(
                        (self.con.Hospitals[event[1][1] - 1].location[0] + 1,
                         self.con.Hospitals[event[1][1] - 1].location[1] + 1))
            self.control_events.clear()

        self.update_hospital_state(self.con.Hospitals)
        self.con.controllerMainLoop(self.tablica_komunikatow)
        #print(0.4*self.simSettings.numberOfAmbulances)
        if len(self.tablica_komunikatow) > 0:
            count_ambulances = 0
            for hospitals in self.con.Hospitals:
                count_ambulances += np.in1d(hospitals.ambulances, 6).sum()
            if self.tablica_komunikatow[0][0] == '!' and (
                    not self.sim.stability() or count_ambulances <
                    0.4 * self.simSettings.numberOfAmbulances):
                self.timer.stop()
                self.con.printState()
                print("System przestał być wydolny po ", self.iterationsNumber,
                      " iteracjach")
        self.update_widgets_scrol()
        self.iterationsNumber += 1
        #self.con.printState()
        #print("Karetki: ",self.printedCar)
        #print("Zgłosze: ",self.printedVirus)

    def update_widgets_scrol(self):
        for elem in self.tablica_komunikatow:
            object = QLabel(elem)
            self.ui.verticalLayout_2.addWidget(object)
        self.tablica_komunikatow = []

    def clear_widget_scrol(self):
        while self.ui.verticalLayout_2.count():
            item = self.ui.verticalLayout_2.takeAt(0)
            widget = item.widget()
            if widget is not None:
                widget.setParent(None)

    def update_hospital_state(self, hospitals):
        index = self.ui.hospwidget.glayout.count()
        for i in range(len(hospitals)):
            count_ambulances = np.in1d(hospitals[i].ambulances, 6).sum()
            self.ui.hospwidget.glayout.children()[i].itemAt(
                1).widget().setText(str(count_ambulances))
            self.ui.hospwidget.glayout.children()[i].itemAt(
                2).widget().changecolor(hospitals[i].personnel)
            self.ui.hospwidget.glayout.children()[i].itemAt(
                3).widget().changecolor(hospitals[i].volume)