如何从USD文件生成MuJoCo XML样板代码？链轮链条模拟求助

1. 直接导入USD/URDF模型（推荐） #

MuJoCo支持直接加载USD（包括.usdz）或URDF格式的模型，无需手动编写XML：

• 先确认Fusion360导出的.usdz文件保留了链节的关节约束，若识别异常，可改用Fusion360的URDF导出插件导出模型，该插件能自动映射连杆、关节结构。
• 使用MuJoCo Python绑定直接加载模型，还可批量调整物理参数后导出为XML：

import mujoco

# 加载.usdz模型
model = mujoco.MjModel.from_path("chain_system.usdz")

# 批量修改链节关节参数（示例：设置铰链关节阻尼）
for i in range(model.njnt):
    if model.jnt_type[i] == mujoco.mjtJoint.mjJNT_HINGE and "chain_link" in model.jnt_name[i]:
        model.dof_damping[i] = 0.1
        model.jnt_range[i] = [-3.14, 3.14]

# 保存修改后的模型为XML，方便后续使用
mujoco.mj_saveXML("chain_system_modified.xml", model, None, 0)

• 也可直接将.usdz拖入MuJoCo的Simulate应用，快速验证模型加载效果。

2. 脚本化批量生成XML #

若需高度自定义XML结构，可编写Python脚本自动生成链节代码：

• 从Fusion360导出单个链节的几何文件（如.stl），并导出所有链节的坐标数据为CSV。
• 用循环生成链节的<body>和<joint>标签，示例脚本：

# 基础XML框架
base_xml = """
<mujoco model="chain_system">
    <option timestep="0.001"/>
    <default>
        <joint type="hinge" damping="0.1"/>
        <geom conaffinity="0" condim="3" friction="1 0.1 0.1"/>
    </default>
    <worldbody>
        <!-- 固定链轮部件 -->
        <body name="sprocket1" pos="0 0 0">
            <geom type="mesh" file="sprocket1.stl"/>
        </body>
"""

# 链节模板
chain_link_template = """
        <body name="chain_link_{idx}" pos="{x} {y} {z}">
            <joint name="joint_{idx}" axis="0 0 1" parent="chain_link_{prev_idx}"/>
            <geom type="mesh" file="chain_link.stl"/>
        </body>
"""

# 从CSV读取链节坐标（示例用模拟数据）
chain_positions = [(i*0.05, 0, 0) for i in range(130)]

# 生成链节XML片段
chain_xml = ""
for idx, pos in enumerate(chain_positions):
    if idx == 0:
        # 第一个链节连接到链轮
        chain_xml += f"""
        <body name="chain_link_0" pos="{pos[0]} {pos[1]} {pos[2]}">
            <joint name="joint_0" axis="0 0 1" parent="sprocket1"/>
            <geom type="mesh" file="chain_link.stl"/>
        </body>
        """
    else:
        chain_xml += chain_link_template.format(idx=idx, x=pos[0], y=pos[1], z=pos[2], prev_idx=idx-1)

# 拼接并保存完整XML
full_xml = base_xml + chain_xml + """
    </worldbody>
</mujoco>
"""
with open("chain_system.xml", "w") as f:
    f.write(full_xml)

3. 利用<default>标签减少重复代码 #

即使手动编写XML，也可通过<default>标签统一定义链节的公共属性，仅需在每个链节中指定差异化内容：

<default>
    <joint type="hinge" damping="0.1" armature="0.01"/>
    <geom type="mesh" file="chain_link.stl" friction="0.8 0.1 0.1" mass="0.05"/>
</default>

后续每个链节仅需编写name、pos、parent等个性化参数即可。

4. 修正Fusion360导出的关节结构 #

若加载.usdz后关节未被正确识别，需在Fusion360中为链节添加铰链关节约束，确保每个链节与相邻部件的关节关系正确，再重新导出USD/URDF格式。

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