Python turtle编写简单的球类小游戏

import turtle
import random
import math
'''
初始化游戏界面大小
'''
# 游戏区域的宽度
game_wid = 600  
# 游戏区域的高度
game_hei = 400  
# 砖块的大小，以及每一个小球初始大小
cell = 20
# 红球初始大小
red_size = cell
# 红色小球
red_ball = None
# 存储绿色小球的列表
green_balls = []
# 存储蓝色小球的列表
blue_balls = []
# 红色小球的方向 当前方向 0 向右，90 向上 180 向左 -90 向下
dir = 0

'''
随机位置计算函数
'''
def rand_pos():
    # 水平有 30 个单元格，垂直有 20 个单元格
    x = random.randint(-14, 14)
    y = random.randint(-9, 9)
    return x * cell, y * cell

'''
绘制一个指定填充颜色的正方形
填充颜色可以不指定
'''
def draw_square(color):
    if color is not None:
        # 的颜色就填充
        turtle.fillcolor(color)
        turtle.begin_fill()
    for i in range(4):
        turtle.fd(cell)
        turtle.left(90)
    if color is not None:
        turtle.end_fill()

'''
自定义画笔形状
name:画笔名称
color:可选项
'''
def custom_shape(name, size):
    turtle.begin_poly()
    turtle.penup()
    turtle.circle(size)
    turtle.end_poly()
    cs = turtle.get_poly()
    turtle.register_shape(name, cs)

'''
移到某点
'''
def move_pos(pen, pos):
    pen.penup()
    pen.Goto(pos)
    pen.pendown()

'''
改变红色小球 4 方向的函数，
这些函数只有当使用者触发按键后方可调用，故这些函数也称为回调函数。
'''
def dir_right():
    global dir
    dir = 0
def dir_left():
    global dir
    dir = 180
def dir_up():
    global dir
    dir = 90
def dir_down():
    global dir
    dir = -90
   
'''
注册键盘响应事件，用来改变红球的方向
'''
def register_event():
    for key, f in {"Up": dir_up, "Down": dir_down, "Left": dir_left, "Right": dir_right}.items():
        turtle.onkey(f, key)
    turtle.listen()
'''
当红色小球遇到墙体后，也要修改方向
'''    
def is_meet_Qt():
    global dir
    if red_ball.xcor() < -220:
        dir = 0
    if red_ball.xcor() > 240:
        dir = 180
    if red_ball.ycor() > 140:
        dir = -90
    if red_ball.ycor() < -120:
        dir = 90

'''
绘制四面的墙体
'''
def draw_blocks():
    # 隐藏画笔
    turtle.hideturtle()
    # 上下各30个单元格，左右各 20 个单元格
    for j in [30, 20, 30, 20]:
        for i in range(j):
            # 调用前面绘制正方形的函数
            draw_square('gray')
            turtle.fd(cell)
        turtle.right(90)
        turtle.fd(-cell)
    # 回到原点
    move_pos(turtle, (0, 0))

def init_ball(pos, color, shape):
    #  由主画笔创建新画笔
    ball = turtle.Turtle()
    ball.color(color)
    # 指定新画笔的形状，如果不指定，则为默认形状
    ball.shape(shape)
    # 移到随机位置
    move_pos(ball, pos)
    # 移动过程要不显示任何轨迹
    ball.penup()
    return ball

def ran_gb_ball(balls, color):
    # 随机创建蓝色、绿色小球的频率，
    # 也就是说，不是调用此函数就一定会创建小球，概率大概是调用 5 次其中会有一次创建
    ran = random.randint(1, 5)
    # 随机一个角度
    a = random.randint(0, 360)
    # 1/5 的概率
    if ran == 5:
        turtle.tracer(False)
        # 每一个小球就是一只画笔
        ball = init_ball(rand_pos(), color, 'ball')
        ball.seth(a)
        # 添加到列表中
        balls.append(ball)
        turtle.tracer(True)

def gb_ball_m(balls):
    s = 20
    a = random.randint(0, 360)
    r = random.randint(0, 10)
    for b in balls:
        b.fd(s)
        if b.xcor() < -220 or b.xcor() > 240 or b.ycor() > 140 or b.ycor() < -120:
            b.goto(rand_pos())

'''
红球是否碰到绿、蓝球
'''
def r_g_b_meet():
    global red_size
    # 红色小球的坐标
    s_x, s_y = red_ball.pos()
    # 迭代绿色小球，蓝色小球列表
    for bs in [green_balls, blue_balls]:
        for b in bs:
            # 计算蓝色或绿色小球坐标
            f_x, f_y = b.pos()
            # 计算和红色球之间的距离
            x_ = math.fabs(s_x - f_x)
            y_ = math.fabs(s_y - f_y)
            # 碰撞距离：两个球的半径之和
            h = cell + red_size
            if 0 <= x_ <= h and y_ >= 0 and y_ <= h:
                if b in green_balls:
                    # 遇到绿色球红球变大
                    red_size += 2
                if b in blue_balls:
                    # 遇到蓝色球红球变大
                    red_size -= 2
                # 关键代码    
                custom_shape('red', red_size)
                return True
    return False

def ball_move():
    red_ball.seth(dir)
    red_ball.fd(40)
    # 检查红球是否碰到墙体
    is_meet_qt()
    # 随机创建绿色小球
    ran_gb_ball(green_balls, 'green')
    # 随机创建蓝色小球
    ran_gb_ball(blue_balls, 'blue')
    # 让绿色小球移动
    gb_ball_m(green_balls)
    # 让蓝色小球移动
    gb_ball_m(blue_balls)
    # 检查红球是否碰到蓝色、绿色小球
    r_g_b_meet()
    # 定时器
    turtle.ontimer(ball_move, 100)

if __name__ == "__main__":
    # 关闭动画效果
    turtle.tracer(False)
    # 注册事件
    register_event()
    # 定制 2 种画笔形状
    for name in ['red', 'ball']:
        custom_shape(name, cell)
    # 主画笔移动墙体的左上角
    move_pos(turtle, (-300, 200))
    # 绘制墙体
    draw_blocks()
    red_ball = init_ball(rand_pos(), 'red', 'red')
    turtle.tracer(True)
    # 让红球移动起来
    ball_move()
    #
    turtle.done()