chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Split MjSpec out into header to allow use by other pybind modules.

Browse Source 
This should enable other pybind modules to accept arguments of type MjSpec.

PiperOrigin-RevId: 745580987
Change-Id: I435a8d264149369e3fbee14c5d2c628ef74b2541
This commit is contained in:
Sam Haves
2025-04-09 07:29:44 -07:00
committed by Copybara-Service
parent 85b323c7fb
commit 5c12ea871f
4 changed files with 172 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/mujoco/specs.cc
+1 -94
View File
@@ -30,6 +30,7 @@
#include <mujoco/mujoco.h>
#include "errors.h"
#include "indexers.h" // IWYU pragma: keep
#include "specs_wrapper.h" // IWYU pragma: keep
#include "raw.h"
#include "structs.h" // IWYU pragma: keep
#include <pybind11/cast.h>
@@ -70,100 +71,6 @@ using MjDoubleRef10 = Eigen::Ref<const Eigen::Matrix<double, 10, 1>>;
using MjDoubleRef11 = Eigen::Ref<const Eigen::Matrix<double, 11, 1>>;
using MjDoubleRefVec = Eigen::Ref<const Eigen::VectorXd>;
struct MjSpec {
MjSpec() : ptr(mj_makeSpec()) {}
MjSpec(raw::MjSpec* ptr, const py::dict& assets_ = {}) : ptr(ptr) {
for (const auto [key, value] : assets_) {
assets[key] = value;
}
}
// copy constructor and assignment
MjSpec(const MjSpec& other) : ptr(mj_copySpec(other.ptr)) {
override_assets = other.override_assets;
for (const auto [key, value] : other.assets) {
assets[key] = value;
}
parent = other.parent;
}
MjSpec& operator=(const MjSpec& other) {
override_assets = other.override_assets;
ptr = mj_copySpec(other.ptr);
for (const auto [key, value] : other.assets) {
assets[key] = value;
}
parent = other.parent;
return *this;
}
// move constructor and move assignment
MjSpec(MjSpec&& other) : ptr(other.ptr) {
override_assets = other.override_assets;
other.ptr = nullptr;
for (const auto [key, value] : other.assets) {
assets[key] = value;
}
other.assets.clear();
parent = other.parent;
other.parent = nullptr;
}
MjSpec& operator=(MjSpec&& other) {
override_assets = other.override_assets;
ptr = other.ptr;
other.ptr = nullptr;
for (const auto [key, value] : other.assets) {
assets[key] = value;
}
other.assets.clear();
parent = other.parent;
other.parent = nullptr;
return *this;
}
~MjSpec() {
mj_deleteSpec(ptr);
}
raw::MjModel* Compile() {
if (assets.empty()) {
auto m = mj_compile(ptr, 0);
if (!m || mjs_isWarning(ptr)) {
throw py::value_error(mjs_getError(ptr));
}
return m;
}
mjVFS vfs;
mj_defaultVFS(&vfs);
for (const auto& asset : assets) {
std::string buffer_name =
_impl::StripPath(py::cast<std::string>(asset.first).c_str());
std::string buffer = py::cast<std::string>(asset.second);
const int vfs_error = InterceptMjErrors(mj_addBufferVFS)(
&vfs, buffer_name.c_str(), buffer.c_str(), buffer.size());
if (vfs_error) {
mj_deleteVFS(&vfs);
if (vfs_error == 2) {
throw py::value_error("Repeated file name in assets dict: " +
buffer_name);
} else {
throw py::value_error("Asset failed to load: " + buffer_name);
}
}
}
auto m = mj_compile(ptr, &vfs);
if (!m || mjs_isWarning(ptr)) {
throw py::value_error(mjs_getError(ptr));
}
mj_deleteVFS(&vfs);
return m;
}
raw::MjSpec* ptr;
py::dict assets;
bool override_assets = true;
MjSpec* parent = nullptr;
};
template <typename LoadFunc>
static raw::MjSpec* LoadSpecFileImpl(
const std::string& filename,