def __init__(self, hz=200, sim_speed=5, spawn_rate=60):
        self.intersection_state = 0

        self.state = np.array((self.intersection_state, \
                env.northward_cars_waiting, \
                env.eastward_cars_waiting, \
                env.southward_cars_waiting, \
                env.westward_cars_waiting))

        self.reward = 0
        self.done = False

        self.intersection = World.Intersection()
        self.traffic_control = Controller.TrafficController()

        self.img = pygame.image.load("intersection.png").convert()
Esempio n. 2
0
    def __init__(self, hz=200, sim_speed=1):
        self.intersection_state = 0
        self.hz = hz
        self.sim_speed = sim_speed

        self.state = np.array((self.intersection_state, \
                env.north_cars_wait_time, \
                env.east_cars_wait_time, \
                env.south_cars_wait_time, \
                env.west_cars_wait_time))

        self.reward = 0
        self.done = False
        self.prev_total_wait_time = 0

        self.intersection = World.Intersection()
        self.traffic_control = Controller.TrafficController()

        self.img = pygame.image.load("intersection.png").convert()
        self.time = 0