Flappy Bird环境代码

以下是使用Python和Pygame库编写的Flappy Bird环境代码：

import pygame
import random

WIDTH = 288
HEIGHT = 512
FPS = 60

pygame.init()
pygame.mixer.init()
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Flappy Bird")
clock = pygame.time.Clock()

font_name = pygame.font.match_font('arial')

def draw_text(surf, text, size, x, y):
    font = pygame.font.Font(font_name, size)
    text_surface = font.render(text, True, (255, 255, 255))
    text_rect = text_surface.get_rect()
    text_rect.midtop = (x, y)
    surf.blit(text_surface, text_rect)

def draw_base(base_x):
    screen.blit(base_img, (base_x, 450))
    screen.blit(base_img, (base_x + WIDTH, 450))

def draw_pipes(pipes):
    for pipe in pipes:
        if pipe.bottom >= 400:
            screen.blit(pipe_img, (pipe.x, pipe.bottom))
        else:
            flip_pipe = pygame.transform.flip(pipe_img, False, True)
            screen.blit(flip_pipe, (pipe.x, pipe.top))

def check_collision(player, pipes):
    for pipe in pipes:
        if pipe.top <= player.y <= pipe.bottom:
            if pipe.x - 10 <= player.x <= pipe.x + 52:
                return True
    if player.y > 450 or player.y < 0:
        return True
    return False

class Player(pygame.sprite.Sprite):
    def __init__(self):
        pygame.sprite.Sprite.__init__(self)
        self.image = pygame.image.load('assets/images/bird.png').convert_alpha()
        self.rect = self.image.get_rect()
        self.rect.centerx = 50
        self.rect.centery = HEIGHT/2
        self.y_speed = 0

    def update(self):
        self.y_speed += 1
        self.rect.centery += self.y_speed

class Pipe:
    def __init__(self):
        self.top = random.randint(100, 300)
        self.bottom = self.top + 150
        self.x = WIDTH

    def update(self):
        self.x -= 2

def game():
    player = Player()
    pipes = []
    score = 0
    base_x = 0
    pipe_timer = pygame.time.get_ticks()

    running = True
    while running:
        clock.tick(FPS)

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                running = False
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_SPACE:
                    player.y_speed = -10

        if pygame.time.get_ticks() - pipe_timer > 2000:
            pipes.append(Pipe())
            pipe_timer = pygame.time.get_ticks()

        for pipe in pipes:
            pipe.update()

        if pipes and pipes[0].x < -50:
            pipes.pop(0)

        base_x -= 2
        if base_x < -WIDTH:
            base_x = 0

        if check_collision(player, pipes):
            running = False

        for pipe in pipes:
            if pipe.x == player.rect.centerx:
                score += 1

        screen.blit(background_img, (0, 0))
        draw_pipes(pipes)
        draw_base(base_x)
        draw_text(screen, str(score), 32, WIDTH/2, 30)

        player.update()
        screen.blit(player.image, player.rect)

        pygame.display.update()

    pygame.quit()

background_img = pygame.image.load('assets/images/background.png').convert()
base_img = pygame.image.load('assets/images/base.png').convert()
pipe_img = pygame.image.load('assets/images/pipe.png').convert()

game()

这个代码实现了一个基本的Flappy Bird游戏环境，包括游戏逻辑、碰撞检测、得分计算等功能。你可以使用该环境来训练和测试自己编写的Flappy Bird AI算法。