import numpy as np import random GRAVITY = 2 FRICTION = 0.9 class Game(): def __init__(self, width, height): self.width = width self.height = height self.bird = Bird(width / 2, height / 2) self.wall = self.create_wall() self.lost = False self.score = 0; def update(self): self.bird.update() self.wall.update() # 10 score for passing a wall if (self.wall.x + self.wall.width) < self.width / 2: self.wall = self.create_wall() self.score += 10 print("\033[32m+\033[0m", end='') if self.intercept(self.bird, self.wall): self.lost = True if self.bird.y < 0 or self.bird.y > self.height: self.lost = True # a constant score for each movement, this way # our birds will try to stay alive longer and # our evolution strategy won't start with zero reward self.score += 0.1 # create a wall, the wall is between the 15%-65% of the screen def create_wall(self): return Wall(self.width - WALL_WIDTH, self.height * (0.15 + np.random.random() * 0.5) ) def intercept(self, bird, wall): return ((bird.x + bird.width) > wall.x and ((bird.y + bird.height) > (wall.gate.y + wall.gate.height) or (bird.y) < wall.gate.y)) JUMP_STEPS = 2 JUMP_SPEED = 7 class Bird(): def __init__(self, x, y): self.x = x self.y = y self.width = 24 self.height = 18 self.velocity = np.array([0, 0], dtype=np.float64) self.acceleration = np.array([0, 0], dtype=np.float64) def jump(self): self.velocity[1] = -JUMP_SPEED self.acceleration[1] = 0 def update(self): self.x += self.velocity[0] self.y += self.velocity[1] self.velocity += self.acceleration self.velocity *= FRICTION self.acceleration[1] = GRAVITY WALL_WIDTH = 30 GATE_HEIGHT = 60 class Wall(): def __init__(self, x, y): self.x = x self.gate = dotdict({ 'y': y, 'height': GATE_HEIGHT }) self.width = WALL_WIDTH def update(self): self.x -= 2 class dotdict(dict): """dot.notation access to dictionary attributes""" __getattr__ = dict.get __setattr__ = dict.__setitem__ __delattr__ = dict.__delitem__ # limit the game to 1000 frames while training, sometimes a game might take # too long to finish after a while of training MAX_FRAMES = 10000 def play(fn, step=None): game = Game(200, 200) frame = 0 # while showing to user, we want to update the GTK frontend # the `step` function is responsible for doing just that, see index.py if step: return step(game, lambda: show_update(fn, game)) while not game.lost and frame < MAX_FRAMES: frame += 1 # input of the model: bird x, bird y, distance to next wall, height of wall's entrance data = np.array([[game.bird.x, game.bird.y, game.bird.x - game.wall.x, game.wall.gate.y]]) jump = fn(data)[0][0] if jump > 0.5: game.bird.jump() game.update() return game.score def show_update(fn, game): if not game.lost: data = np.array([[game.bird.x, game.bird.y, game.bird.x - game.wall.x, game.wall.gate.y]]) jump = fn(data)[0][0] if jump > 0.5: game.bird.jump() game.update() return True else: return False