def start_simulation(self, widget): N = [] t_zero = time.time() for i in range(0, len(self.Nodes)): N.append([[0, 0], 0]) M = Map(self.Nodes, self.Connectors) S = Simulation(10, 2, self.Nodes, self.Connectors) print "Nodes: " + str(self.Nodes) print "Connectors: " + str(self.Connectors) self.Vehicles = utils.generate_vehicles(self.Connectors) print "Vehicles: " + str(self.Vehicles) M.disp_map() while True: time.sleep(0.05) [N, self.Vehicles] = S.simulate_vehicles(self.Vehicles, N) M.disp_vehicles(self.Vehicles, N) #M.disp_infections(self.Vehicles, N) for event in pygame.event.get(): if event.type == pygame.QUIT: M.close_window() #DB.save(x) self.info.set_text("Runtime: " + str(time.time() - t_zero) + "s") print "Runtime: " + str(time.time() - t_zero) + "s" return 0
def get_connector_spec(self, widget): NCM = Map(self.Nodes, self.Connectors) #NCM.Connectors.append(self.newConnector) while True: NCM.Conn.append(self.newConnector) NCM.disp_map()#, self.newConnector]) NCM.Conn.pop(-1) pos = NCM.get_pos() if not NCM.Open: del(NCM) return 0 for node in self.Nodes: if utils.dist(pos, node[2]) < NCM.hitbox: if len(self.newConnector[1]) < 1: self.newConnector[1].append(self.Nodes.index(node)) elif pos != self.newConnector[1][-1]: self.newConnector[1].append(self.Nodes.index(node))
def get_connector_spec(self, widget): NCM = Map(self.Nodes, self.Connectors) #NCM.Connectors.append(self.newConnector) while True: NCM.Conn.append(self.newConnector) NCM.disp_map() #, self.newConnector]) NCM.Conn.pop(-1) pos = NCM.get_pos() if not NCM.Open: del (NCM) return 0 for node in self.Nodes: if utils.dist(pos, node[2]) < NCM.hitbox: if len(self.newConnector[1]) < 1: self.newConnector[1].append(self.Nodes.index(node)) elif pos != self.newConnector[1][-1]: self.newConnector[1].append(self.Nodes.index(node))
def get_node_pos(self, widget): NNM = Map(self.Nodes, self.Connectors) NNM.disp_map() self.newNode[2] = NNM.get_pos() NNM.close_window()