如何用PyCATIA提取平面与几何体的交线并导出点坐标至CSV

需求概述 #

提取平面与几何体相交形成的闭合、无自相交的单一零件曲线的点坐标，并保存为CSV文件。原方法通过查找极值点分割曲线、按百分比创建点，过程中产生大量错误，需更高效稳定的实现方式。

原代码问题分析 #

原代码频繁创建/删除几何对象、多次调用part.update()，易触发CATIA交互错误；按百分比生成点在闭合曲线上会重复起点；极值点处理流程繁琐且易出错。

优化实现代码 #

from pycatia import catia
import numpy as np
import csv

def get_point_coords(part, spa_workbench, geometry):
    """获取几何对象的坐标"""
    reference = part.create_reference_from_object(geometry)
    measurable = spa_workbench.get_measurable(reference)
    return measurable.get_point()

def get_curve_total_length(part, spa_workbench, curve):
    """获取曲线总长度"""
    reference = part.create_reference_from_object(curve)
    measurable = spa_workbench.get_measurable(reference)
    return measurable.get_length()

# 配置参数
step_distance = 0.0005  # 点间隔（单位：米），对应原step=5e-4
scale = 1000  # 坐标转换为毫米的缩放系数
output_csv = "curve_points.csv"

# 初始化CATIA连接
caa = catia()
document = caa.active_document
part = document.part
spa_workbench = document.spa_workbench()
hsf = part.hybrid_shape_factory

coords = []

# 遍历混合体查找目标曲线
for geom_set in part.hybrid_bodies:
    for shape in geom_set.hybrid_shapes:
        if shape.name == "NAME":  # 替换为你的目标曲线名称
            # 获取曲线总长度
            total_length = get_curve_total_length(part, spa_workbench, shape)
            # 生成距离序列（从0到总长度，含终点确保闭合）
            distances = np.arange(0, total_length, step_distance)
            distances = np.append(distances, total_length)
            
            # 获取X方向最小点作为基准（可根据需求移除此段）
            dir_x = hsf.add_new_direction_by_coord(1, 0, 0)
            extremum = hsf.add_new_extremum(shape, dir_x, 0)
            geom_set.append_hybrid_shape(extremum)
            part.update()
            x_le, y_le, z_le = get_point_coords(part, spa_workbench, extremum)
            hsf.delete_object_for_datum(extremum)
            part.update()
            
            # 生成曲线上的点并记录坐标
            for dist in distances:
                # 创建临时点（基于距离）
                point = hsf.add_new_point_on_curve_from_distance(shape, dist, True)
                # 直接获取点坐标，无需添加到模型树更新
                x, y, z = point.get_coordinates()
                # 计算偏移后坐标并转换单位
                coords.append([
                    (x - x_le) / scale,
                    (z - z_le) / scale,
                    (y - y_le) / scale
                ])
                # 清理临时点
                hsf.delete_object_for_datum(point)
            
            # 找到目标曲线后跳出循环
            break
    else:
        continue
    break

# 将坐标写入CSV文件
with open(output_csv, "w", newline="") as f:
    writer = csv.writer(f)
    writer.writerow(["X", "Z", "Y"])  # 写入表头
    writer.writerows(coords)

print(f"坐标数据已成功保存到 {output_csv}")

核心优化点 #

• 减少CATIA交互开销：创建临时点后直接调用get_coordinates()获取坐标，无需添加到模型树和反复更新，避免交互错误
• 按距离生成点：基于曲线总长度生成等间隔点，避免闭合曲线的起点重复问题，点分布更均匀
• 简化临时对象处理：统一清理临时几何对象，避免模型树冗余
• 高效坐标获取：部分场景下直接使用点对象的内置方法获取坐标，比通过SPA工作台测量更快速稳定

注意事项 #

• 替换代码中的"NAME"为实际目标曲线的名称
• 调整step_distance和scale参数以匹配你的单位和精度需求
• 若不需要基于X最小点的坐标偏移，可删除极值点相关的代码段
• 运行前确保CATIA已打开目标零件文档

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