Liang-Barskey算法的 python3-pygame实现-计算机图形学

初代效果

后续更新放在代码仓库

Code

#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
'''
@File    :   L_B_LineClip.py
@Time    :   2020/11/26 14:29:16
@Author  :   Kearney
@Version :   0.0
@Contact :   191615342@qq.com
@License :   GPL 3.0
@Desc    :   Liang-Barskey算法的 python3-pygame实现
             1、画矩形确定显示区域
             2、画红色线后自动裁剪，显示区域内的线段会显示为黄色
'''
import pygame
import sys

width = 920     # 画板宽度
height = 640    # 画版高度

XL = XR = YB = YT = -1
x1 = y1 = x2 = y2 = -1
iswin = False   # 显示区域是否确定

pygame.init()
screen = pygame.display.set_mode((width, height))
pygame.display.set_caption("Liang-Barskey_LineClip_Kearney")


def clipTest(p, q, u1, u2):
    retVal = True
    if (p < 0):
        r = q / p
        if (r > u2):
            retVal = False
        elif (r > u1):
            u1 = r
    elif (p > 0):
        r = q / p
        if (r < u1):
            retVal = False
        elif (r < u2):
            u2 = r
    elif q < 0:
        retVal = False
    return (u1, u2, retVal)


# 对在显示区域外的部分线段进行裁剪
def L_B_LineClip(x1, y1, x2, y2, sur):
    # print('(', x1, ',', y1, ')', '  (', x2, ',', y2, ')')
    u1 = 0
    u2 = 1
    dx = x2 - x1
    dy = y2 - y1
    (u1, u2, bool) = clipTest(-dx, x1 - XL, u1, u2)    # 计算左边界交点参数，更新u
    if bool:
        (u1, u2, bool) = clipTest(dx, XR - x1, u1, u2)    # 计算右边界交点参数，更新u
        if bool:
            (u1, u2, bool) = clipTest(-dy, y1 - YB, u1, u2)    # 计算下边界交点参数，更新u
            if bool:
                (u1, u2, bool) = clipTest(dy, YT - y1, u1, u2)  # 计算上边界交点参数，更新u
                if bool:
                    if u2 < 1:  # 计算终点坐标
                        x2 = (int)(x1 + u2 * dx)
                        y2 = (int)(y1 + u2 * dy)
                    if u1 > 0:  # 计算起点坐标
                        x1 = (int)(x1 + u1 * dx)
                        y1 = (int)(y1 + u1 * dy)
                    pygame.draw.line(screen, (255, 255, 0), (x1, y1), (x2, y2))


while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        elif event.type == pygame.MOUSEBUTTONDOWN:
            if x1 < 0:
                x1, y1 = event.pos
            else:
                x2, y2 = event.pos
            if iswin and x2 > 0:    # 显示区域完成后
                pygame.draw.line(screen, (255, 0, 0), (x1, y1),
                                 (x2, y2))  # 调用内置函数划线画红线
                L_B_LineClip(x1, y1, x2, y2, screen)
                x1 = y1 = x2 = y2 = -1
            if (pygame.mouse.get_pressed()[2]):  # 鼠标右键按下，清除线条
                x1 = y1 = x2 = y2 = -1
                screen.fill((0, 0, 0))
                if XL > 0:  # 保留显示区域
                    pygame.draw.rect(screen, (0, 255, 0),
                                     [XL, YB, XR - XL, YT - YB], 1)
                print("Reset Line done")
        elif event.type == pygame.MOUSEMOTION:  # 设置显示区域
            x, y = pygame.mouse.get_pos()
            if not iswin:
                if x1 > 0 and x2 < 0:
                    screen.fill((0, 0, 0))
                    pygame.draw.rect(screen, (0, 255, 0),
                                     [x1, y1, x - x1, y - y1], 1)
                elif x1 > 0 and x2 > 0:
                    screen.fill((0, 0, 0))
                    # print('window: (', x1, ',', y1, ')', ' (', x2, ',', y2,
                    #       ')')
                    XL = x1
                    XR = x2
                    YB = y1
                    YT = y2
                    pygame.draw.rect(screen, (0, 255, 0),
                                     [XL, YB, XR - XL, YT - YB], 1)
                    x1 = y1 = x2 = y2 = -1  # 重置坐标，开始划线
                    iswin = True
    pygame.display.update()
sys.exit()

References

• liang-Barskey算法