MuJoCo C++ 示例:列出所有刚体

这个简单示例展示了如何从 XML 文件加载 MuJoCo 模型,并列出模型中的所有刚体及其 ID。

list_bodies.cpp
#include <mujoco/mujoco.h>
#include <cstdio>

int main(int argc, char** argv) {
    // 检查参数
    if (argc < 2) {
        printf("Usage: %s model.xml\n", argv[0]);
        return 1;
    }

    // 加载模型
    char error[1000] = "Could not load binary model";
    mjModel* m = mj_loadXML(argv[1], nullptr, error, 1000);
    if (!m) {
        printf("%s\n", error);
        return 1;
    }

    // 创建数据
    mjData* d = mj_makeData(m);

    // 遍历所有刚体
    printf("Bodies in the model:\n");
    for (int i = 0; i < m->nbody; i++) {
        const char* name = mj_id2name(m, mjOBJ_BODY, i);
        if (name)
            printf("  id %d: %s\n", i, name);
        else
            printf("  id %d: (unnamed)\n", i);
    }

    // 释放 MuJoCo 结构
    mj_deleteData(d);
    mj_deleteModel(m);

    return 0;
}

如何编译

build.sh
g++ -o list_bodies list_bodies.cpp -lmujoco -I/opt/mujoco/include -L/opt/mujoco/lib -Wl,-rpath,/opt/mujoco/lib

用法示例

run_example.sh
./list_bodies ~/mujoco_menagerie/franka_fr3/fr3.xml
Bodies in the model:
  id 0: world
  id 1: base
  id 2: fr3_link0
  id 3: fr3_link1
  id 4: fr3_link2
  id 5: fr3_link3
  id 6: fr3_link4
  id 7: fr3_link5
  id 8: fr3_link6
  id 9: fr3_link7

Check out similar posts by category: Mujoco, Robotics, C++