| 321 | |
| 322 | |
| 323 | def _iter_indexers(model, data): |
| 324 | mujoco.mj_forward(model.ptr, data.ptr) |
| 325 | size_to_axis_indexer = index.make_axis_indexers(model) |
| 326 | all_fields = collections.OrderedDict() |
| 327 | for struct, struct_name in ((model, 'mjmodel'), (data, 'mjdata')): |
| 328 | indexer = index.struct_indexer(struct, struct_name, size_to_axis_indexer) |
| 329 | for field_name, field_indexer in indexer._asdict().items(): |
| 330 | if field_name not in all_fields: |
| 331 | all_fields[field_name] = field_indexer |
| 332 | for field_name, field_indexer in all_fields.items(): |
| 333 | yield field_name, field_indexer |
| 334 | |
| 335 | |
| 336 | class AllFieldsTest(parameterized.TestCase): |