写一个关于小球运动的子程序 完整代码

当涉及到编写一个关于小球运动的子程序，我们通常会创建一个模拟类，这里是一个简单的Python示例，假设我们正在制作一个基本的二维空间中小球自由落体并反弹的程序：

import pygame
import math

class Ball:
    def __init__(self, x, y, radius, speed):
        self.x = x
        self.y = y
        self.radius = radius
        self.speed = speed
        self.color = (255, 0, 0)  # 红色

    def update(self, gravity=9.8):
        # 更新位置
        self.y += self.speed + gravity * 0.5
        if self.y > self.radius and not self.on_ground():
            self.y = self.radius
            self.speed *= -1  # 反弹

    def draw(self, screen):
        pygame.draw.circle(screen, self.color, (int(self.x), int(self.y)), self.radius)

    def on_ground(self):
        # 判断是否触地，这里简单假设屏幕高度减去球半径即为地面
        return self.y >= pygame.display.get_surface().get_height() - self.radius

# 主程序部分
def main():
    pygame.init()
    screen = pygame.display.set_mode((800, 600))
    ball = Ball(400, 500, 20, 0)
    clock = pygame.time.Clock()

    while True:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                return

        screen.fill((0, 0, 0))
        ball.update()
        ball.draw(screen)
        pygame.display.flip()
        clock.tick(60)  # 控制帧率

if __name__ == "__main__":
    main()

这个程序定义了一个`Ball`类，它包含了小球的位置、速度、颜色等属性以及更新位置和绘制的功能。主函数`main()`负责游戏循环，不断检测事件并更新画布。