import pygame
import random
import sys

# 初始化pygame
pygame.init()

# 窗口设置
WIDTH, HEIGHT = 480, 700
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("简易飞行模拟器")

# 颜色定义
WHITE = (255, 255, 255)
BLACK = (0, 0, 0)
RED = (255, 0, 0)
BLUE = (0, 120, 255)
YELLOW = (255, 255, 0)

# 玩家飞机
player_size = 40
player_x = WIDTH // 2 - player_size // 2
player_y = HEIGHT - 100
player_speed = 7

# 敌机
enemies = []
enemy_size = 35
enemy_speed_base = 4
spawn_timer = 0

# 子弹设置
bullets = []
bullet_width = 6
bullet_height = 16
bullet_speed = 10

score = 0

# 字体容错修复，防止SysFont内部BUG
try:
    font = pygame.font.SysFont("simhei", 32)
except:
    try:
        font = pygame.font.SysFont("msyh", 32)
    except:
        font = pygame.font.Font(None, 32)

clock = pygame.time.Clock()
FPS = 60

def draw_plane(x, y, size, color):
    pygame.draw.polygon(screen, color, [
        (x + size//2, y),
        (x, y + size),
        (x + size, y + size)
    ])

def draw_enemy(x, y, size):
    pygame.draw.rect(screen, RED, (x, y, size, size))

running = True
game_over = False

while running:
    clock.tick(FPS)
    screen.fill(BLACK)

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        if event.type == pygame.KEYDOWN:
            if game_over:
                if event.key == pygame.K_SPACE:
                    # 重置游戏
                    player_x = WIDTH // 2 - player_size // 2
                    player_y = HEIGHT - 100
                    enemies.clear()
                    bullets.clear()
                    score = 0
                    game_over = False
            else:
                # 发射子弹
                if event.key == pygame.K_SPACE:
                    bullet_x = player_x + player_size//2 - bullet_width//2
                    bullet_y = player_y - bullet_height
                    bullets.append([bullet_x, bullet_y])

    if not game_over:
        # 飞机移动
        keys = pygame.key.get_pressed()
        if keys[pygame.K_LEFT] and player_x > 0:
            player_x -= player_speed
        if keys[pygame.K_RIGHT] and player_x < WIDTH - player_size:
            player_x += player_speed
        if keys[pygame.K_UP] and player_y > 0:
            player_y -= player_speed
        if keys[pygame.K_DOWN] and player_y < HEIGHT - player_size:
            player_y += player_speed

        # 生成敌机
        spawn_timer += 1
        if spawn_timer > 40:
            ex = random.randint(0, WIDTH - enemy_size)
            enemies.append([ex, -enemy_size])
            spawn_timer = 0

        # 更新敌机
        for enemy in enemies[:]:
            enemy[1] += enemy_speed_base
            if enemy[1] > HEIGHT:
                enemies.remove(enemy)

        # 更新子弹
        for bullet in bullets[:]:
            bullet[1] -= bullet_speed
            if bullet[1] < -bullet_height:
                bullets.remove(bullet)

        # 子弹击中敌机检测
        for bullet in bullets[:]:
            bx, by = bullet
            bullet_rect = pygame.Rect(bx, by, bullet_width, bullet_height)
            for enemy in enemies[:]:
                ex, ey = enemy
                enemy_rect = pygame.Rect(ex, ey, enemy_size, enemy_size)
                if bullet_rect.colliderect(enemy_rect):
                    if bullet in bullets:
                        bullets.remove(bullet)
                    if enemy in enemies:
                        enemies.remove(enemy)
                        score += 10
                    break

        # 玩家碰撞敌机
        player_rect = pygame.Rect(player_x, player_y, player_size, player_size)
        for enemy in enemies:
            ex, ey = enemy
            enemy_rect = pygame.Rect(ex, ey, enemy_size, enemy_size)
            if player_rect.colliderect(enemy_rect):
                game_over = True
                break

        # 绘制飞机
        draw_plane(player_x, player_y, player_size, BLUE)
        # 绘制敌机
        for e in enemies:
            draw_enemy(e[0], e[1], enemy_size)
        # 绘制子弹
        for b in bullets:
            pygame.draw.rect(screen, YELLOW, (b[0], b[1], bullet_width, bullet_height))

        # 分数文字
        score_text = font.render(f"分数：{score}", True, WHITE)
        screen.blit(score_text, (10, 10))

    else:
        over_text = font.render("游戏结束！按空格重新开始", True, WHITE)
        text_rect = over_text.get_rect(center=(WIDTH//2, HEIGHT//2))
        screen.blit(over_text, text_rect)

    pygame.display.update()

pygame.quit()
sys.exit()