OCCT 的OCAF之遍历TDF

OCCT 的OCAF之遍历TDF_Label树

0.概述

如果想从文档 (TDocStd_Document) 中获取单独的TopoDS_Shape和对应的颜色信息等，那就需要遍历TDF_Label 树中储存的信息，如果不想麻烦的去遍历，可以直接使用XCAFPrs_AISObject来直接渲染TDF_Label ，XCAFPrs_AISObject内部已经处理好了。

1.遍历TDF_Label树

1.1 TDF_Label简介

标签是 OCAF 的核心数据结构，用于存储和组织数据。
每个标签都有一个唯一的路径（Entry()），例如 0:1:2。
标签支持层次结构，可以嵌套子标签，形成树形结构。

1.2 示例代码:

#include <TDocStd_Application.hxx>
#include <TDocStd_Document.hxx>
#include <BinLDrivers.hxx>
#include <XCAFDoc_ShapeTool.hxx>
#include <XCAFDoc_ColorTool.hxx>
#include <TDF_Label.hxx>
#include <Quantity_Color.hxx>
#include <iostream>// 提取颜色
void ExtractColor(const TDF_Label& label, const Handle(XCAFDoc_ColorTool)& colorTool) {Quantity_Color color;if (colorTool->GetColor(label, XCAFDoc_ColorSurf, color) ||colorTool->GetColor(label, XCAFDoc_ColorGen, color) ||colorTool->GetColor(label, XCAFDoc_ColorCurv, color)) {std::cout << "  Color found on label: ";TCollection_AsciiString entry;label.Entry(entry);std::cout << entry.ToCString()<< " - R: " << color.Red()<< ", G: " << color.Green()<< ", B: " << color.Blue()<< std::endl;}
}// 遍历并提取形状和颜色
void TraverseAndExtract(const TDF_Label& label, const Handle(XCAFDoc_ShapeTool)& shapeTool, const Handle(XCAFDoc_ColorTool)& colorTool, int depth = 0) {// 打印标签路径TCollection_AsciiString entry;label.Entry(entry);std::cout << std::string(depth * 2, ' ') << "Label: " << entry.ToCString() << std::endl;// 提取形状if (shapeTool->IsShape(label)) {TopoDS_Shape shape = shapeTool->GetShape(label);if (!shape.IsNull()) {std::cout << std::string(depth * 2, ' ') << "  Shape found!" << std::endl;}}// 提取颜色ExtractColor(label, colorTool);// 遍历子标签for (TDF_ChildIterator it(label); it.More(); it.Next()) {TraverseAndExtract(it.Value(), shapeTool, colorTool, depth + 1); // 递归子标签}
}int main() {// 初始化 OCAF 文档Handle(TDocStd_Application) app = new TDocStd_Application();BinLDrivers::DefineFormat(app);Handle(TDocStd_Document) doc;app->Open("assembly.caf", doc); // 加载 OCAF 文件if (doc.IsNull()) {std::cerr << "Failed to load document!" << std::endl;return 1;}// 获取 ShapeTool 和 ColorToolHandle(XCAFDoc_ShapeTool) shapeTool = XCAFDoc_DocumentTool::ShapeTool(doc->Main());Handle(XCAFDoc_ColorTool) colorTool = XCAFDoc_DocumentTool::ColorTool(doc->Main());// 从根标签开始遍历TDF_Label rootLabel = doc->Main();TraverseAndExtract(rootLabel, shapeTool, colorTool);return 0;
}