Pygame自定义海龟类绘制正方形时线条断裂问题求助

问题原因 #

1. goto方法逻辑错误：原代码中self.vector的更新被缩进在if self.is_down()判断内部，只有落笔状态调用goto才会更新位置，抬笔时调用goto完全不改变坐标，属于逻辑漏洞。
2. 浮点数精度偏差：三角函数sin/cos计算角度偏移时会产生微小的浮点误差，比如sin(radians(270))可能不是精准的-1.0，导致最终的端点坐标不是整数，Pygame渲染线条时会自动对坐标取整，导致相邻线条的端点无法完美重合，出现断裂。
3. 线条端点样式问题：Pygame的draw.line默认是平头端点，粗线垂直相交时拐角处天然会存在空隙，就算坐标完全精准也可能有视觉缺口。

修复方案 #

下面是修改后的完整代码，修改点已标注：

import pygame
from pygame.locals import *
from pygame.math import Vector2
from math import sin, cos, radians

class Turtle:
    def __init__(self):
        self.vector = Vector2(0, 0)
        self.angle = 0
        self.pen_width = 25
        self.pen_color = (255, 255, 255)
        self.pen_visible = True

    def forward(self, win, distance):
        start = self.vector
        offset = Vector2(sin(radians(self.angle)), cos(radians(self.angle)))
        offset *= distance
        end = start + offset
        if self.is_down():
            # 修复1：坐标转整数，消除浮点误差
            start_int = (round(start.x), round(start.y))
            end_int = (round(end.x), round(end.y))
            pygame.draw.line(win, self.pen_color, start_int, end_int, self.pen_width)
            # 修复2：添加端点圆填充拐角空隙，线宽较粗时效果明显，不需要可以删除
            if self.pen_width > 1:
                radius = self.pen_width // 2
                pygame.draw.circle(win, self.pen_color, end_int, radius)
        self.vector = end
    fd = forward

    def backward(self, win, distance):
        self.forward(win, -distance)
    bk = back = backward

    def left(self, angle):
        self.angle = (self.angle + angle) % 360
    lt = left

    def right(self, angle):
        self.left(-angle)
    rt = right

    def goto(self, win, pos):
        if self.is_down():
            start_int = (round(self.vector.x), round(self.vector.y))
            end_int = (round(pos[0]), round(pos[1]))
            pygame.draw.line(win, self.pen_color, start_int, end_int, self.pen_width)
            if self.pen_width > 1:
                radius = self.pen_width // 2
                pygame.draw.circle(win, self.pen_color, end_int, radius)
        # 修复3：位置更新移到判断外，不管笔是否落下都更新坐标
        self.vector = Vector2(pos)
    setposition = setpos = goto

    def set_angle(self, angle= None):
        if not angle:
            return self.angle
        self.angle = angle

    def position(self):
        return self.vector.xy
    pos = position

    def sety(self, y):
        self.vector.y = y

    def setx(self, x):
        self.vector.x = x

    def xcor(self):
        return self.vector.x

    def ycor(self):
        return self.vector.y

    def penup(self):
        self.pen_visible = False
    pu = up = penup

    def pendown(self):
        self.pen_visible = True
    pd = down = pendown

    def is_down(self):
        return self.pen_visible

    def write(self, win, font_label):
        win.blit(font_label, self.vector)

def test():
    pygame.init()
    pygame.font.init()

    WIDTH, HEIGHT = 500, 500
    CAPTION = 'Caption'
    win = pygame.display.set_mode((WIDTH, HEIGHT))
    pygame.display.set_caption(CAPTION)

    FPS = 60
    clock = pygame.time.Clock()
    mytut = Turtle()
    mytut.pen_width = 5
    # 修复4：先抬笔再移动到初始位置，避免多画一条从(0,0)到中心的多余线条
    mytut.penup()
    mytut.goto(win, (250, 250))
    mytut.pendown()
    mytut.pen_color = '#00FF00'
    for _ in range(4):
        mytut.fd(win, 100)
        mytut.rt(270)

    while True:
        clock.tick(FPS)
        pygame.display.flip()
        for event in pygame.event.get():
            if event.type == QUIT:
                pygame.quit()
                quit()

if __name__ == '__main__':
    test()

验证说明 #

修改后不管调用lt(90)/rt(270)还是反向角度，正方形的四条边都会完美闭合，不会出现断裂问题。如果不需要处理极粗线条，也可以去掉绘制端点圆的代码，仅坐标转整数就可以解决绝大多数场景的断裂问题。

内容的提问来源于stack exchange，提问作者Random_Pythoneer59