flappy-es/game.py
Mahdi Dibaiee 7d6a23625b fix: increase space between bird and wall
fix: update README statistics
fix: a better load.npy and long.npy
fix: round off score to 2 decimal points
2017-04-03 11:54:56 +04:30

131 lines
3.4 KiB
Python

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(250, 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