如何使用OpenCascade获取STEP模型所有子元素并展示为树形视图

你已经完成了STEP模型的加载与显示，要实现递归遍历所有子元素并生成树形视图，可以借助OpenCascade的TopExp_Explorer工具来遍历拓扑形状的层级结构，以下是具体实现方案：

1. 核心递归遍历函数 #

编写递归函数，通过TopExp_Explorer遍历每个形状的子元素，区分不同拓扑类型（如复合形状、实体、面、边等），同时记录层级用于树形结构的缩进展示：

#include <TopExp_Explorer.hxx>
#include <TopoDS_Shape.hxx>
#include <TopAbs_ShapeEnum.hxx>
#include <iostream>

// 递归遍历形状并输出树形结构
void TraverseShape(const TopoDS_Shape& aShape, int level = 0)
{
    // 输出当前形状的类型和层级缩进
    std::string indent(level * 2, ' ');
    const char* shapeTypeStr = "";
    switch(aShape.ShapeType())
    {
        case TopAbs_COMPOUND: shapeTypeStr = "Compound"; break;
        case TopAbs_COMPSOLID: shapeTypeStr = "CompSolid"; break;
        case TopAbs_SOLID: shapeTypeStr = "Solid"; break;
        case TopAbs_SHELL: shapeTypeStr = "Shell"; break;
        case TopAbs_FACE: shapeTypeStr = "Face"; break;
        case TopAbs_WIRE: shapeTypeStr = "Wire"; break;
        case TopAbs_EDGE: shapeTypeStr = "Edge"; break;
        case TopAbs_VERTEX: shapeTypeStr = "Vertex"; break;
        default: shapeTypeStr = "Unknown"; break;
    }
    std::cout << indent << "- " << shapeTypeStr << std::endl;

    // 遍历子形状并递归处理
    TopExp_Explorer explorer(aShape, TopAbs_SHAPE);
    for (; explorer.More(); explorer.Next())
    {
        const TopoDS_Shape& subShape = explorer.Current();
        TraverseShape(subShape, level + 1);
    }
}

2. 整合到现有加载流程 #

在你的onSTEPLoad函数加载完成STEPShape后，直接调用上述遍历函数即可输出树形结构：

void OcctGtkViewer::onSTEPLoad(std::string filename)
{
    // ... 原有加载代码保持不变 ...

    shape = reader.OneShape();
    STEPShape = reader.OneShape();

    cout << "STEP: файл загружен " << endl;

    // 调用遍历函数输出树形结构
    TraverseShape(STEPShape);

    return;
}

3. 扩展为GTK树形视图 #

如果要在GTK界面中展示树形视图，需要先将遍历到的形状信息存储为树形节点数据，再绑定到GtkTreeView：

步骤1：定义树形节点数据结构 #

#include <gtk/gtk.h>

// 存储形状信息的结构体
struct ShapeNode {
    gchar* typeName;
    TopoDS_Shape shape;
};

步骤2：修改遍历函数填充树形存储模型 #

void PopulateTreeStore(GtkTreeStore* store, GtkTreeIter* parentIter, const TopoDS_Shape& aShape)
{
    // 获取形状类型名称
    const char* shapeTypeStr = "";
    switch(aShape.ShapeType())
    {
        case TopAbs_COMPOUND: shapeTypeStr = "Compound"; break;
        case TopAbs_COMPSOLID: shapeTypeStr = "CompSolid"; break;
        case TopAbs_SOLID: shapeTypeStr = "Solid"; break;
        case TopAbs_SHELL: shapeTypeStr = "Shell"; break;
        case TopAbs_FACE: shapeTypeStr = "Face"; break;
        case TopAbs_WIRE: shapeTypeStr = "Wire"; break;
        case TopAbs_EDGE: shapeTypeStr = "Edge"; break;
        case TopAbs_VERTEX: shapeTypeStr = "Vertex"; break;
        default: shapeTypeStr = "Unknown"; break;
    }

    // 添加当前节点到树形存储
    GtkTreeIter currentIter;
    gtk_tree_store_append(store, &currentIter, parentIter);
    gtk_tree_store_set(store, &currentIter, 0, shapeTypeStr, -1);

    // 递归添加子节点
    TopExp_Explorer explorer(aShape, TopAbs_SHAPE);
    for (; explorer.More(); explorer.Next())
    {
        const TopoDS_Shape& subShape = explorer.Current();
        PopulateTreeStore(store, &currentIter, subShape);
    }
}

步骤3：在UI初始化时创建树形视图并绑定数据 #

// 假设你已经有GTK窗口和容器
void InitTreeView(GtkWidget* container)
{
    // 创建树形存储模型（仅一列用于显示类型名称）
    GtkTreeStore* store = gtk_tree_store_new(1, G_TYPE_STRING);
    // 加载完STEP模型后调用PopulateTreeStore填充数据
    // PopulateTreeStore(store, nullptr, STEPShape);

    // 创建树形视图并设置模型
    GtkWidget* treeView = gtk_tree_view_new_with_model(GTK_TREE_MODEL(store));
    g_object_unref(store); // 模型由视图引用，此处释放

    // 添加列
    GtkTreeViewColumn* column = gtk_tree_view_column_new_with_attributes("形状类型",
        gtk_cell_renderer_text_new(), "text", 0, nullptr);
    gtk_tree_view_append_column(GTK_TREE_VIEW(treeView), column);

    // 添加到容器
    gtk_container_add(GTK_CONTAINER(container), treeView);
}

注意事项 #

• TopExp_Explorer可以通过第二个参数指定要遍历的拓扑类型，比如只遍历实体可以传入TopAbs_SOLID，按需调整即可。
• 如果需要保留STEP文件中的原始名称（如部件名），可以通过STEPControl_Reader的GiveListOfRoots获取STEPConstruct_Styles，进而提取形状的名称信息。

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