Require prefix for the attach tag.

For `mjs_attachFrame` and `mjs_attachBody`, require either a prefix or suffix.

It is tempting for users to not use a prefix (see #2162), but this causes issues. For example, it creates a duplicate "main" default class in the parent as a child of the already existing "main" default.

PiperOrigin-RevId: 688535755
Change-Id: I653181ca35943f1e4eb89c70fca8b5289818324e
This commit is contained in:
Alessio Quaglino
2024-10-22 07:07:33 -07:00
committed by Copybara-Service
parent 94e9912d28
commit bfa78614c8
5 changed files with 23 additions and 15 deletions
+5 -5
View File
@@ -3861,19 +3861,19 @@ all attachments will appear in the saved XML file.
.. _body-attach-model: .. _body-attach-model:
:at:`model`: :at-val:`string, optional` :at:`model`: :at-val:`string, required`
The sub-model from which to attach a subtree. The sub-model from which to attach a subtree.
.. _body-attach-body: .. _body-attach-body:
:at:`body`: :at-val:`string, optional` :at:`body`: :at-val:`string, required`
Name of the body in the sub-model to attach here. The body and its subtree will be attached. Name of the body in the sub-model to attach here. The body and its subtree will be attached.
.. _body-attach-prefix: .. _body-attach-prefix:
:at:`prefix`: :at-val:`string, optional` :at:`prefix`: :at-val:`string, required`
Prefix to prepend to names of elements in the sub-model. If empty, the names are unchanged. This attribute is Prefix to prepend to names of elements in the sub-model. This attribute is required to prevent name collisions with
required to prevent name collisions with the parent or when attaching the same sub-tree multiple times. the parent or when attaching the same sub-tree multiple times.
.. _body-frame: .. _body-frame:
+2 -2
View File
@@ -850,7 +850,7 @@ class SpecsTest(absltest.TestCase):
body = parent.worldbody.add_body() body = parent.worldbody.add_body()
frame = child.worldbody.add_frame() frame = child.worldbody.add_frame()
with self.assertRaises(ValueError) as cm: with self.assertRaises(ValueError) as cm:
body.attach_frame(frame, '', '') body.attach_frame(frame, '_', '')
self.assertEqual( self.assertEqual(
str(cm.exception), str(cm.exception),
'Error: cannot attach mjSpecs with incompatible compiler/angle' 'Error: cannot attach mjSpecs with incompatible compiler/angle'
@@ -862,7 +862,7 @@ class SpecsTest(absltest.TestCase):
parent = mujoco.MjSpec() parent = mujoco.MjSpec()
site = parent.worldbody.add_site(pos=[1, 2, 3]) site = parent.worldbody.add_site(pos=[1, 2, 3])
body = child.worldbody.add_body() body = child.worldbody.add_body()
self.assertIsNotNone(site.attach(body, '', '')) self.assertIsNotNone(site.attach(body, '_', ''))
model = parent.compile() model = parent.compile()
np.testing.assert_array_equal(model.body_pos[1], [1, 2, 3]) np.testing.assert_array_equal(model.body_pos[1], [1, 2, 3])
+8
View File
@@ -850,6 +850,10 @@ mjCBody& mjCBody::operator+=(const mjCFrame& other) {
other.model->suffix = other.suffix; other.model->suffix = other.suffix;
other.model->StoreKeyframes(model); other.model->StoreKeyframes(model);
if (other.prefix.empty() && other.suffix.empty()) {
throw mjCError(this, "either prefix or suffix must be non-empty");
}
// attach defaults // attach defaults
if (other.model != model) { if (other.model != model) {
mjCDef* subdef = new mjCDef(*other.model->Default()); mjCDef* subdef = new mjCDef(*other.model->Default());
@@ -1812,6 +1816,10 @@ mjCFrame& mjCFrame::operator+=(const mjCBody& other) {
other.model->prefix = ""; other.model->prefix = "";
other.model->suffix = ""; other.model->suffix = "";
if (other.prefix.empty() && other.suffix.empty()) {
throw mjCError(this, "either prefix or suffix must be non-empty");
}
mjCBody* subtree = new mjCBody(other, model); mjCBody* subtree = new mjCBody(other, model);
other.ForgetKeyframes(); other.ForgetKeyframes();
other.model->prefix = subtree->prefix; other.model->prefix = subtree->prefix;
+5 -5
View File
@@ -3674,9 +3674,9 @@ void mjXReader::Body(XMLElement* section, mjsBody* body, mjsFrame* frame,
// attachment // attachment
else if (name=="attach") { else if (name=="attach") {
string model_name, body_name, prefix; string model_name, body_name, prefix;
ReadAttrTxt(elem, "model", model_name); ReadAttrTxt(elem, "model", model_name, /*required=*/true);
ReadAttrTxt(elem, "body", body_name); ReadAttrTxt(elem, "body", body_name, /*required=*/true);
ReadAttrTxt(elem, "prefix", prefix); ReadAttrTxt(elem, "prefix", prefix, /*required=*/true);
mjsBody* child = mjs_findBody(spec, (prefix+body_name).c_str()); mjsBody* child = mjs_findBody(spec, (prefix+body_name).c_str());
mjsFrame* pframe = frame ? frame : mjs_addFrame(body, nullptr); mjsFrame* pframe = frame ? frame : mjs_addFrame(body, nullptr);
@@ -3684,11 +3684,11 @@ void mjXReader::Body(XMLElement* section, mjsBody* body, mjsFrame* frame,
if (!child) { if (!child) {
mjSpec* asset = mjs_findSpec(spec, model_name.c_str()); mjSpec* asset = mjs_findSpec(spec, model_name.c_str());
if (!asset) { if (!asset) {
throw mjXError(0, "could not find model '%s'", model_name.c_str()); throw mjXError(elem, "could not find model '%s'", model_name.c_str());
} }
child = mjs_findBody(asset, body_name.c_str()); child = mjs_findBody(asset, body_name.c_str());
if (!child) { if (!child) {
throw mjXError(0, "could not find body '%s''%s'", body_name.c_str()); throw mjXError(elem, "could not find body '%s''%s'", body_name.c_str());
} }
if (!mjs_attachBody(pframe, child, prefix.c_str(), "")) { if (!mjs_attachBody(pframe, child, prefix.c_str(), "")) {
throw mjXError(elem, mjs_getError(spec)); throw mjXError(elem, mjs_getError(spec));
+3 -3
View File
@@ -1544,12 +1544,12 @@ TEST_F(XMLReaderTest, InvalidAttach) {
<body name="parent"> <body name="parent">
<joint name="joint1"/> <joint name="joint1"/>
<geom size="2"/> <geom size="2"/>
<attach model="other" body="body"/> <attach model="other" body="body" prefix="_"/>
</body> </body>
</worldbody> </worldbody>
<actuator> <actuator>
<motor name="actuator" joint="joint1"/> <motor name="_actuator" joint="joint1"/>
</actuator> </actuator>
</mujoco> </mujoco>
)"; )";
@@ -1578,7 +1578,7 @@ TEST_F(XMLReaderTest, InvalidAttach) {
LoadModelFromString(xml_parent, er.data(), er.size(), vfs.get()); LoadModelFromString(xml_parent, er.data(), er.size(), vfs.get());
EXPECT_THAT(model, IsNull()) << er.data(); EXPECT_THAT(model, IsNull()) << er.data();
EXPECT_THAT(er.data(), HasSubstr("repeated name 'actuator' in actuator")); EXPECT_THAT(er.data(), HasSubstr("repeated name '_actuator' in actuator"));
EXPECT_THAT(er.data(), HasSubstr("Element 'attach'")); EXPECT_THAT(er.data(), HasSubstr("Element 'attach'"));
mj_deleteVFS(vfs.get()); mj_deleteVFS(vfs.get());
} }