如何使用PyVista计算两条直线交点并实现可视化？

问题重现 #

尝试用PyVista计算两条直线的交点并绘制，但运行代码后得到包含3个空PolyData的元组，无法绘制交点：

import pyvista as pv

line1 = pv.Line(pointa=(-1,0,0), pointb=(2,0,0))
line2 = pv.Line(pointa=(0,-1,0), pointb=(0,2,0))
intersection = line1.intersection(line2)

plotter = pv.Plotter()
plotter.add_mesh(line1)
plotter.add_mesh(line2)
plotter.add_mesh(intersection)

plotter.show()

返回结果：

(PolyData (0x2634b15bee0)
   N Cells:    0
   N Points:   0
   N Strips:   0
   X Bounds:   1.000e+299, -1.000e+299
   Y Bounds:   1.000e+299, -1.000e+299
   Z Bounds:   1.000e+299, -1.000e+299
   N Arrays:   0,
 PolyData (0x2634b15bac0)
   N Cells:    0
   N Points:   0
   N Strips:   0
   X Bounds:   1.000e+299, -1.000e+299
   Y Bounds:   1.000e+299, -1.000e+299
   Z Bounds:   1.000e+299, -1.000e+299
   N Arrays:   0,
 PolyData (0x2634b14a100)
   N Cells:    0
   N Points:   0
   N Strips:   0
   X Bounds:   1.000e+299, -1.000e+299
   Y Bounds:   1.000e+299, -1.000e+299
   Z Bounds:   1.000e+299, -1.000e+299
   N Arrays:   0)

问题原因 #

PyVista的intersection方法是为**多边形网格（如面、体）**设计的，并不适用于线段/直线的相交检测。它返回的元组(intersection, left, right)仅在网格存在面级相交时才会生成有效数据，对于线段这种一维单元，无法通过该方法得到交点。

解决方案 #

方法1：数学计算直线/线段交点 #

直接通过线性代数计算两条直线的交点，若需要限制为线段交点，可额外判断交点是否在线段范围内：

import pyvista as pv
import numpy as np

def line_intersection(line1_start, line1_end, line2_start, line2_end):
    # 转换为向量形式
    line1_vec = np.array(line1_end) - np.array(line1_start)
    line2_vec = np.array(line2_end) - np.array(line2_start)
    cross = np.cross(line1_vec, line2_vec)
    
    # 判断直线是否平行
    if np.allclose(cross, 0):
        return None
    
    # 计算参数t
    t_numerator = np.cross(np.array(line2_start) - np.array(line1_start), line2_vec)
    t = np.dot(t_numerator, cross) / np.dot(cross, cross)
    intersection_point = np.array(line1_start) + t * line1_vec
    
    # 若要限制为线段交点，取消注释以下代码
    # s_numerator = np.cross(np.array(line2_start) - np.array(line1_start), line1_vec)
    # s = np.dot(s_numerator, cross) / np.dot(cross, cross)
    # if 0 <= t <= 1 and 0 <= s <= 1:
    #     return intersection_point
    # else:
    #     return None
    
    return intersection_point

# 定义两条直线的端点
line1_start = (-1, 0, 0)
line1_end = (2, 0, 0)
line2_start = (0, -1, 0)
line2_end = (0, 2, 0)

# 计算交点
intersection_point = line_intersection(line1_start, line1_end, line2_start, line2_end)

# 绘制结果
plotter = pv.Plotter()
plotter.add_mesh(pv.Line(line1_start, line1_end), color='blue', label='Line 1')
plotter.add_mesh(pv.Line(line2_start, line2_end), color='red', label='Line 2')
# 用球体标记交点
if intersection_point is not None:
    plotter.add_mesh(pv.Sphere(center=intersection_point, radius=0.05), color='green', label='Intersection')

plotter.add_legend()
plotter.show()

方法2：使用VTK的线段相交工具 #

利用PyVista底层的VTK库，直接调用vtkLineLineIntersection类进行精确的线段/直线相交检测：

import pyvista as pv
import vtk

def vtk_line_intersection(line1_start, line1_end, line2_start, line2_end):
    intersector = vtk.vtkLineLineIntersection()
    intersector.SetLine1(line1_start[0], line1_start[1], line1_start[2],
                         line1_end[0], line1_end[1], line1_end[2])
    intersector.SetLine2(line2_start[0], line2_start[1], line2_start[2],
                         line2_end[0], line2_end[1], line2_end[2])
    intersector.Compute()
    
    # 获取相交结果：0=不相交，1=相交于一点，2=重合
    status = intersector.GetIntersectionStatus()
    if status == 1:
        return intersector.GetIntersectionPoint()
    elif status == 2:
        # 处理重合情况，此处返回线段端点示例
        return line1_start
    else:
        return None

# 定义直线端点
line1_start = (-1, 0, 0)
line1_end = (2, 0, 0)
line2_start = (0, -1, 0)
line2_end = (0, 2, 0)

# 计算交点
intersection_point = vtk_line_intersection(line1_start, line1_end, line2_start, line2_end)

# 绘制
plotter = pv.Plotter()
plotter.add_mesh(pv.Line(line1_start, line1_end), color='blue')
plotter.add_mesh(pv.Line(line2_start, line2_end), color='red')
if intersection_point:
    plotter.add_mesh(pv.Sphere(center=intersection_point, radius=0.05), color='green')
plotter.show()

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