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)
