import pygame
import random
import sys

# 初始化 pygame
pygame.init()

# 屏幕设置
SCREEN_WIDTH = 400
SCREEN_HEIGHT = 600
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
pygame.display.set_caption("驾驶小游戏")

# 颜色定义
WHITE = (255, 255, 255)
BLACK = (0, 0, 0)
RED = (255, 0, 0)
GREEN = (0, 255, 0)
GRAY = (100, 100, 100)
BLUE = (0, 0, 255)

# 玩家车辆设置
player_width = 50
player_height = 100
player_x = SCREEN_WIDTH // 2 - player_width // 2
player_y = SCREEN_HEIGHT - player_height - 20
player_speed = 5

# 敌车设置
enemy_width = 50
enemy_height = 100
enemy_speed = 5
enemies = []
spawn_timer = 0
spawn_delay = 30  # 每30帧生成一辆新敌车

# 游戏变量
clock = pygame.time.Clock()
score = 0
font = pygame.font.SysFont(None, 36)
game_over = False

# 主循环
running = True
while running:
    screen.fill(GRAY)  # 背景为道路灰色

    # 绘制道路中线（虚线）
    for y in range(0, SCREEN_HEIGHT, 40):
        pygame.draw.rect(screen, WHITE, (SCREEN_WIDTH // 2 - 5, y, 10, 20))

    # 事件处理
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        if game_over and event.type == pygame.KEYDOWN:
            if event.key == pygame.K_r:
                # 重置游戏
                player_x = SCREEN_WIDTH // 2 - player_width // 2
                enemies = []
                score = 0
                game_over = False

    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 < SCREEN_WIDTH - player_width:
            player_x += player_speed

        # 生成敌车
        spawn_timer += 1
        if spawn_timer >= spawn_delay:
            enemy_x = random.randint(0, SCREEN_WIDTH - enemy_width)
            enemies.append([enemy_x, -enemy_height])  # [x, y]
            spawn_timer = 0
            # 随机增加难度（可选）
            spawn_delay = max(15, spawn_delay - 0.1)

        # 更新敌车位置
        for enemy in enemies[:]:
            enemy[1] += enemy_speed
            # 移除超出屏幕的敌车
            if enemy[1] > SCREEN_HEIGHT:
                enemies.remove(enemy)
                score += 1

        # 碰撞检测
        player_rect = pygame.Rect(player_x, player_y, player_width, player_height)
        for enemy in enemies:
            enemy_rect = pygame.Rect(enemy[0], enemy[1], enemy_width, enemy_height)
            if player_rect.colliderect(enemy_rect):
                game_over = True

    # 绘制玩家车辆
    pygame.draw.rect(screen, BLUE, (player_x, player_y, player_width, player_height))

    # 绘制敌车
    for enemy in enemies:
        pygame.draw.rect(screen, RED, (enemy[0], enemy[1], enemy_width, enemy_height))

    # 显示分数
    score_text = font.render(f"Score: {score}", True, WHITE)
    screen.blit(score_text, (10, 10))

    # 游戏结束提示
    if game_over:
        game_over_text = font.render("Game Over! Press R to Restart", True, WHITE)
        screen.blit(game_over_text, (SCREEN_WIDTH // 2 - game_over_text.get_width() // 2, SCREEN_HEIGHT // 2))

    # 刷新屏幕
    pygame.display.flip()
    clock.tick(60)

pygame.quit()
sys.exit()