求Line与Ray交点函数实现求助：已有Point等类及辅助函数

我已经实现了Point、Ray、Line等几何类，现在需要编写一个函数，接收Line对象和Ray对象作为参数，返回二者的交点。但因为数学基础薄弱，多次生成的函数都无法正常工作——要么返回错误交点，要么误判直线与射线平行。以下是已实现的类及辅助函数：

已实现的类 #

Point类 #

class Point:
    def __init__(self, x: float, y: float):
        self.x = x
        self.y = y

    def __getitem__(self, index):
        if index == 0:
            return self.x
        elif index == 1:
            return self.y

    def __iter__(self):
        yield self.x
        yield self.y

    def __str__(self):
        return f"x = {self.x}; y = {self.y}"

Ray类 #

class Ray:
    def __init__(self, start: Point, end: Point):
        self.start = start
        self.end = end
        
    def get_direction(self):
        x_diff = self.end[0] - self.start[0]
        y_diff = self.end[1] - self.start[1]
        return (x_diff, y_diff)

Line类 #

class Line:
    def __init__(self, point1: Point, point2: Point):
        self.point1 = point1
        self.point2 = point2

    def get_direction(self) -> tuple:
        x_diff = abs(self.point2[0] - self.point1[0])
        y_diff = abs(self.point2[1] - self.point1[1])
        return (x_diff, y_diff)
        
    def get_params(self) -> tuple:
        x1, y1 = self.point1
        x2, y2 = self.point2
        
        a = y2 - y1
        b = x1 - x2
        c = x2 * y1 - x1 * y2
        
        return a, b, c
    
    def get_normal(self):
        a, b, c = self.get_params()
        return -b, a, c

辅助函数 #

判断点是否在直线上 #

def point_on_line(point: Point, line: Line):
    x1, y1 = line.point1
    x2, y2 = line.point2
    x, y = point

    return (y - y1) * (x2 - x1) == (y2 - y1) * (x - x1)

判断点是否在射线上（原函数存在缺陷，已修正） #

def point_on_ray(point: Point, ray: Ray) -> bool:
    start = ray.start
    end = ray.end
    direction = ray.get_direction()

    x_diff = point[0] - start[0]
    y_diff = point[1] - start[1]

    if x_diff * direction[1] == y_diff * direction[0]:
        if x_diff >= 0 and y_diff >= 0:
            return True

    return False

正确的直线与射线交点求解方案 #

修正后的点在射线上判断函数 #

原函数错误假设射线方向分量均为非负，实际射线可向任意方向延伸，修正后通过叉积判断共线、点积判断方向一致性，同时加入浮点数精度容错：

def point_on_ray(point: Point, ray: Ray) -> bool:
    start_x, start_y = ray.start
    dir_x, dir_y = ray.get_direction()
    point_x, point_y = point

    # 计算点相对于起点的向量
    rel_x = point_x - start_x
    rel_y = point_y - start_y

    # 叉积为0表示共线（加入精度容错）
    cross = rel_x * dir_y - rel_y * dir_x
    if abs(cross) > 1e-9:
        return False

    # 点积>=0表示向量同向（加入精度容错）
    dot = rel_x * dir_x + rel_y * dir_y
    return dot >= -1e-9

直线与射线交点求解函数 #

通过直线一般式与射线参数方程联立，处理平行、重合、相交等所有情况，加入浮点数精度处理：

def line_ray_intersection(line: Line, ray: Ray) -> Point | None:
    # 获取直线一般式参数 Ax + By + C = 0
    A, B, C = line.get_params()
    # 射线起点与方向向量
    ray_start_x, ray_start_y = ray.start
    dir_x, dir_y = ray.get_direction()

    # 计算分母：直线法向量与射线方向的点积
    denominator = A * dir_x + B * dir_y

    # 情况1：直线与射线平行
    if abs(denominator) < 1e-9:
        # 检查射线起点是否在直线上
        if abs(A * ray_start_x + B * ray_start_y + C) < 1e-9:
            # 射线与直线重合，无唯一交点，返回None
            return None
        else:
            # 平行且不重合，无交点
            return None

    # 情况2：不平行，计算射线参数t（t>=0时在射线上）
    t = -(A * ray_start_x + B * ray_start_y + C) / denominator

    # 参数t为负，交点不在射线上
    if t < -1e-9:
        return None

    # 计算交点坐标
    intersect_x = ray_start_x + t * dir_x
    intersect_y = ray_start_y + t * dir_y

    return Point(intersect_x, intersect_y)

方案说明 #

• 引入1e-9精度阈值，避免浮点数计算误差导致的判断错误
• 修正射线点判断逻辑，适配任意方向的射线
• 完整覆盖平行、重合、相交等所有几何情况，确保结果准确

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