pythonOCC读取GLB文件自定义extras数据失败的问题咨询

1. RWGltf_CafReader是否支持导入extras？ #

OpenCASCADE（PythonOCC基于该库）的RWGltf_CafReader默认不支持将GLB节点的extras字段自动映射回TDataStd_NamedData属性。其核心导入逻辑仅聚焦于几何拓扑数据、材质、动画等标准GLTF/GLB元素，自定义的extras数据不会被自动解析并绑定到CAF的标签（Label）属性系统中。

2. 读取GLB的extras并关联对应形状的正确方式 #

需要通过「手动解析GLB文件提取extras + 建立GLB节点与OCCT形状的映射」的方式实现，具体步骤如下：

步骤1：导出时预留映射标识 #

导出GLB时，为每个面的OCCT标签（Label）设置唯一名称，确保与GLB节点的name字段对应，方便后续匹配：

# 导出时给每个面的label设置名称
face_label = shape_tool.NewShape()
shape_tool.SetShape(face_label, face_shape)
# 设置唯一名称，比如用面的索引或自定义ID
face_label.SetName(f"face_{face_index}")

# 添加自定义元数据
user_data = TDataStd_NamedData.Set(face_label)
user_data.SetString("some_variable", str(value))

# 执行GLB导出（省略常规导出代码）

步骤2：导入模型并收集形状与标签的映射 #

使用RWGltf_CafReader导入模型，遍历所有标签，记录「标签名称 - 标签对象」的映射：

from OCC.Core.RWGltf import RWGltf_CafReader
from OCC.Core.TDocStd import TDocStd_Document
from OCC.Core.TDataStd import TDataStd_Name
from OCC.Core.TopAbs import TopAbs_FACE

doc = TDocStd_Document()
reader = RWGltf_CafReader(doc)
reader.ReadFile("your_model.glb")

# 获取形状工具和标签遍历器
shape_tool = doc.Main().ShapeTool()
label_map = {}

# 遍历所有面的标签
def traverse_labels(label):
    if shape_tool.IsShape(label):
        shape = shape_tool.GetShape(label)
        if shape.ShapeType() == TopAbs_FACE:
            # 获取标签名称
            name_attr = TDataStd_Name()
            if label.FindAttribute(TDataStd_Name.GetID(), name_attr):
                label_name = name_attr.Get()
                label_map[label_name] = label
    # 递归遍历子标签
    child_label = label.FirstChild()
    while not child_label.IsNull():
        traverse_labels(child_label)
        child_label = child_label.NextSibling()

traverse_labels(doc.Main())

步骤3：手动解析GLB文件提取extras数据 #

用Python解析GLB的二进制结构，提取JSON块中的nodes数组，获取每个节点的name和extras：

import struct
import json

def parse_glb_extras(glb_path):
    with open(glb_path, "rb") as f:
        # 读取GLB文件头（12字节）
        magic, version, length = struct.unpack("<4sII", f.read(12))
        assert magic == b"glTF", "Not a valid GLB file"
        assert version == 2, "Only GLB version 2 is supported"

        # 读取第一个chunk（JSON块）
        chunk_length, chunk_type = struct.unpack("<II", f.read(8))
        assert chunk_type == 0x4E4F534A, "First chunk is not JSON"  # 对应小端编码的"JSON"

        # 读取JSON内容并解析
        json_data = json.loads(f.read(chunk_length))
        nodes = json_data.get("nodes", [])
        
        # 收集节点名称与extras的映射
        node_extras_map = {}
        for node in nodes:
            node_name = node.get("name")
            if node_name and "extras" in node:
                node_extras_map[node_name] = node["extras"]
        
        return node_extras_map

# 提取extras数据
extras_map = parse_glb_extras("your_model.glb")

步骤4：将extras数据绑定到对应标签的TDataStd_NamedData #

遍历映射关系，将解析到的extras数据添加到OCCT标签的属性中：

from OCC.Core.TDataStd import TDataStd_NamedData

for node_name, extras in extras_map.items():
    if node_name in label_map:
        label = label_map[node_name]
        # 创建或获取TDataStd_NamedData属性
        named_data = TDataStd_NamedData()
        if not label.FindAttribute(TDataStd_NamedData.GetID(), named_data):
            named_data = TDataStd_NamedData.Set(label)
        # 将extras中的键值对写入属性（这里以字符串为例，可根据实际类型调整）
        for key, value in extras.items():
            named_data.SetString(key, str(value))
        
        # 验证写入结果
        if named_data.HasStrings():
            print(f"Successfully added extras to label {node_name}")

注意事项 #

• 如果导出时未设置标签名称，也可通过形状的拓扑特征（如面的面积、顶点坐标哈希）建立映射，但稳定性不如名称匹配。
• GLB的extras支持任意JSON数据类型，写入TDataStd_NamedData时需根据实际类型选择SetString、SetInteger等对应方法。

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