引言

PyMeshGen是一个基于Python的开源非结构化网格生成工具，专注于CFD/FEA领域的网格生成。在实际工程应用中，几何模型的导入导出是网格生成流程的第一步，也是至关重要的一步。为了支持主流的CAD几何格式，PyMeshGen引入了PythonOCC库，实现了STEP、IGES、STL等格式的几何模型导入导出及渲染功能。

本文将详细介绍PyMeshGen如何集成PythonOCC，以及相关的技术实现细节。

PythonOCC简介

PythonOCC是OpenCASCADE Technology（OCCT）的Python绑定库。OpenCASCADE是一个强大的开源几何建模内核，广泛应用于CAD/CAM/CAE领域。PythonOCC将OpenCASCADE的C++接口封装为Python API，使得开发者可以使用Python进行复杂的几何建模操作。

PythonOCC的主要特点包括：

• 支持STEP、IGES、STL等主流CAD格式
• 提供完整的几何建模功能（点、线、面、体等）
• 支持布尔运算、几何变换等高级操作
• 与VTK等可视化库无缝集成

为什么选择PythonOCC

在PyMeshGen中集成PythonOCC主要基于以下考虑：

1. 格式支持广泛：PythonOCC支持STEP、IGES等参数化几何格式，以及STL等网格格式，覆盖了CFD/FEA领域的主要需求。

2. 几何信息完整：相比直接读取网格文件，通过PythonOCC导入参数化几何可以获取完整的拓扑信息（面、边、顶点等），这对于后续的网格生成至关重要。

3. 可视化友好：PythonOCC可以方便地与VTK集成，将几何模型转换为VTK PolyData进行可视化。

4. Python生态：作为Python库，PythonOCC可以无缝集成到PyMeshGen的Python代码库中。

PythonOCC安装

PyMeshGen项目使用Python 3.12版本进行开发。PythonOCC可以通过Conda轻松安装，无需手动编译。

安装步骤

# first create an environment conda create --name=pyoccenv python=3.12 conda activate pyoccenv conda install -c conda-forge pythonocc-core=7.9.0

验证安装

# 测试PythonOCC是否可用try:from OCC.Core.STEPControl import STEPControl_Readerfrom OCC.Core.TopoDS import TopoDS_Shape    print("PythonOCC安装成功！")    print(f"Python版本: {__import__('sys').version}")    print(f"OCC版本: {__import__('OCC').__version__}")except ImportError as e:    print(f"PythonOCC安装失败: {e}")

技术实现

1. OpenCASCADE DLL预加载

在Windows平台上，OpenCASCADE依赖于多个DLL文件，如果加载顺序不当会导致导入失败。PyMeshGen通过occ_loader.py模块实现了DLL预加载机制：

defensure_occ_loaded():"""确保 OpenCASCADE DLL 已正确加载"""try:# 动态查找 OpenCASCADE 安装路径        site_packages = site.getsitepackages()[0] if site.getsitepackages() else os.path.join(sys.prefix, 'Lib', 'site-packages')        occ_dir = os.path.join(site_packages, 'OCC')# 按顺序加载所有 DLLfor dll_name in dll_list:            dll_loaded = Falsefor dll_path in dll_load_paths:                full_dll_path = os.path.join(dll_path, dll_name)if os.path.exists(full_dll_path):try:                        kernel32.LoadLibraryExW(full_dll_path, None, LOAD_WITH_ALTERED_SEARCH_PATH)                        dll_loaded = Truebreakexcept Exception:passexcept Exception:pass

该函数会在程序启动时自动调用，确保OpenCASCADE的DLL按正确顺序加载，避免依赖问题。

2. 几何文件导入

几何文件导入的核心实现在geometry_io.py模块中。该模块提供了统一的导入接口，自动识别文件格式：

defimport_geometry_file(filename: str) -> TopoDS_Shape:"""导入几何文件，自动识别文件格式"""ifnot os.path.exists(filename):raise FileNotFoundError(f"文件不存在: {filename}")    file_ext = os.path.splitext(filename)[1].lower()if file_ext in ['.step', '.stp']:return import_step_file(filename)elif file_ext in ['.iges', '.igs']:return import_iges_file(filename)elif file_ext in ['.stl']:return import_stl_file(filename)else:raise ValueError(f"不支持的文件格式: {file_ext}")

对于STEP格式，使用STEPControl_Reader进行读取：

defimport_step_file(filename: str) -> TopoDS_Shape:"""导入STEP文件"""try:from OCC.Core.STEPControl import STEPControl_Readerfrom OCC.Core.IFSelect import IFSelect_RetDoneexcept ImportError:raise ImportError("无法导入STEP模块，请确保已安装pythonocc-core")    step_reader = STEPControl_Reader()    status = step_reader.ReadFile(filename)if status != IFSelect_RetDone:raise ValueError(f"无法读取STEP文件: {filename}")    step_reader.TransferRoots()    shape = step_reader.OneShape()return shape

类似地，IGES和STL格式也有对应的导入函数。

3. 几何信息提取

导入几何后，PyMeshGen会提取几何的统计信息，包括顶点数、边数、面数等：

defget_shape_statistics(shape: TopoDS_Shape) -> dict:"""获取几何形状的统计信息"""from OCC.Core.TopExp import TopExp_Explorerfrom OCC.Core.TopAbs import TopAbs_VERTEX, TopAbs_EDGE, TopAbs_FACE    stats = {'vertex_count': 0,'edge_count': 0,'face_count': 0,'solid_count': 0,'shell_count': 0    }    explorer = TopExp_Explorer(shape, TopAbs_VERTEX)while explorer.More():        stats['vertex_count'] += 1        explorer.Next()    explorer.Init(shape, TopAbs_EDGE)while explorer.More():        stats['edge_count'] += 1        explorer.Next()    explorer.Init(shape, TopAbs_FACE)while explorer.More():        stats['face_count'] += 1        explorer.Next()return stats

这些信息会在GUI中显示，帮助用户了解导入的几何模型。

4. 几何可视化

为了在GUI中可视化几何模型，需要将OpenCASCADE的TopoDS_Shape转换为VTK的PolyData。这一功能在occ_to_vtk.py模块中实现：

defshape_to_vtk_polydata(shape: TopoDS_Shape,                          mesh_quality: float = 2.0,                         relative: bool = False) -> vtk.vtkPolyData:"""将TopoDS_Shape转换为VTK PolyData"""# 对形状进行网格化    mesher = BRepMesh_IncrementalMesh(shape, mesh_quality, relative, True, True)    mesher.Perform()# 创建VTK点集和单元    points = vtk.vtkPoints()    triangles = vtk.vtkCellArray()# 处理顶点和三角形数据# ...# 创建PolyData对象    polydata = vtk.vtkPolyData()    polydata.SetPoints(points)    polydata.SetPolys(triangles)return polydata

该函数使用BRepMesh_IncrementalMesh将几何形状离散化为三角网格，然后提取顶点和面片信息，构建VTK PolyData对象。

5. 几何文件导出

除了导入，PyMeshGen也支持几何导出功能。导出接口同样支持多种格式：

defexport_geometry_file(shape: TopoDS_Shape, filename: str) -> bool:"""导出几何文件，自动识别文件格式"""ifnot os.path.exists(os.path.dirname(filename)):        os.makedirs(os.path.dirname(filename), exist_ok=True)    file_ext = os.path.splitext(filename)[1].lower()if file_ext in ['.step', '.stp']:return export_step_file(shape, filename)elif file_ext in ['.iges', '.igs']:return export_iges_file(shape, filename)elif file_ext in ['.stl']:return export_stl_file(shape, filename)else:raise ValueError(f"不支持的文件格式: {file_ext}")

STEP导出示例：

defexport_step_file(shape: TopoDS_Shape, filename: str) -> bool:"""导出STEP文件"""try:from OCC.Core.STEPControl import STEPControl_Writerfrom OCC.Core.IFSelect import IFSelect_RetDoneexcept ImportError:raise ImportError("无法导入STEP模块，请确保已安装pythonocc-core")    step_writer = STEPControl_Writer()    step_writer.Transfer(shape, 1)    status = step_writer.Write(filename)return status == IFSelect_RetDone

6. GUI集成

在PyMeshGen的GUI中，几何导入导出功能通过异步线程实现，避免界面卡顿：

classGeometryImportThread(QThread):"""几何导入线程类"""    progress_updated = pyqtSignal(str, int)    import_finished = pyqtSignal(object)    import_failed = pyqtSignal(str)defrun(self):"""执行几何导入操作"""try:from fileIO.geometry_io import import_geometry_file, get_shape_statistics            self.progress_updated.emit(f"开始导入几何文件...", 5)            shape = import_geometry_file(self.file_path)            self.progress_updated.emit("提取几何统计信息...", 60)            stats = get_shape_statistics(shape)            self.progress_updated.emit("完成几何文件导入", 100)            result = {'shape': shape,'stats': stats,'file_path': self.file_path            }            self.import_finished.emit(result)except Exception as e:            self.import_failed.emit(f"几何导入失败: {str(e)}")

GUI主窗口中的导入导出函数：

defimport_geometry(self):"""导入几何文件"""    file_path, _ = QFileDialog.getOpenFileName(        self,"导入几何文件",        os.path.join(self.project_root, "geometries"),"几何文件 (*.step *.stp *.iges *.igs *.stl);;STEP文件 (*.step *.stp);;IGES文件 (*.iges *.igs);;STL文件 (*.stl);;所有文件 (*.*)"    )if file_path:        self.import_thread = GeometryImportThread(file_path)        self.import_thread.import_finished.connect(self._on_geometry_imported)        self.import_thread.import_failed.connect(self._on_geometry_import_failed)        self.import_thread.progress_updated.connect(self.update_progress)        self.import_thread.start()defexport_geometry(self):"""导出几何文件"""ifnot hasattr(self, 'current_geometry') or self.current_geometry isNone:        QMessageBox.warning(self, "警告", "请先导入几何文件")return    file_path, _ = QFileDialog.getSaveFileName(        self,"导出几何文件",        os.path.join(self.project_root, "geometries"),"几何文件 (*.step *.stp *.iges *.igs *.stl);;STEP文件 (*.step *.stp);;IGES文件 (*.iges *.igs);;STL文件 (*.stl);;所有文件 (*.*)"    )if file_path:from fileIO.geometry_io import export_geometry_file        success = export_geometry_file(self.current_geometry, file_path)if success:            self.log_info(f"已导出几何: {file_path}")

使用示例

命令行使用

from fileIO.geometry_io import import_geometry_file, export_geometry_filefrom fileIO.occ_to_vtk import shape_to_vtk_polydata# 导入几何文件shape = import_geometry_file("model.step")# 获取几何统计信息from fileIO.geometry_io import get_shape_statisticsstats = get_shape_statistics(shape)print(f"顶点数: {stats['vertex_count']}")print(f"边数: {stats['edge_count']}")print(f"面数: {stats['face_count']}")# 转换为VTK PolyData进行可视化polydata = shape_to_vtk_polydata(shape, mesh_quality=1.0)# 导出几何文件export_geometry_file(shape, "output.iges")

GUI使用

1. 启动PyMeshGen GUI
2. 点击"文件"菜单中的"导入几何"
3. 选择几何文件（STEP/IGES/STL格式）
4. 等待导入完成，查看几何统计信息
5. 如需导出，点击"导出几何"并选择目标格式

总结

通过引入PythonOCC，PyMeshGen实现了对主流CAD几何格式的支持，为网格生成流程提供了强大的几何处理能力。本文介绍了PyMeshGen中PythonOCC集成的技术细节，包括DLL预加载、几何导入导出、信息提取、可视化转换等功能。

PythonOCC的集成不仅扩展了PyMeshGen的功能，也为后续的几何处理（如几何修复、简化等）奠定了基础。

后续计划

实现从几何元素出发生成曲线网格、曲面网格。