测试的是：python: 小球在旋转的六边形内部滚动，逼真的模拟小球的弹性，重力以及为六边形内部边框的摩擦力，think in english，show me the code. 
[python_vXazBg9Zc7] 
效果好，速度快！ 
import pygame
import math

# Initialize Pygame
pygame.init()

# Screen dimensions
WIDTH = 800
HEIGHT = 600
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Rotating Hexagon Ball Simulation")

# Colors
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
RED = ...