fa156ed22f
Add accelerometer to MJX.
...
PiperOrigin-RevId: 671355424
Change-Id: I8593bcad6ba311f3c555b118889e2a5c6b367ec7
2024-09-05 06:57:09 -07:00
20296466b1
Add subtreelinvel and subtreeangmom sensors to MJX.
...
PiperOrigin-RevId: 671347528
Change-Id: I8a32590c650ca6d0bb6df43d0347f707ffc05f12
2024-09-05 06:21:50 -07:00
7b073b60c4
Add rne_postconstraint function to MJX. This function matches mj_rnePostConstraint and computes cacc, cfrc_ext, and cfrc_int.
...
PiperOrigin-RevId: 671336319
Change-Id: I263b46c56a7609b9a8a30895c012798579d3dffa
2024-09-05 05:35:26 -07:00
cd8ff4401d
Add eq_objtype to mjx.Model and support for connect constraint using sites in MJX.
...
PiperOrigin-RevId: 671225109
Change-Id: I8e2e6eb4c539d318658d3d7dfe0a7f03b411aabc
2024-09-04 22:00:36 -07:00
8954a088ce
Add automatic free-joint alignment.
...
PiperOrigin-RevId: 671010083
Change-Id: Ib7d60f75c7f545b1886703715663db8de77feeef
2024-09-04 10:07:33 -07:00
303e5e7718
Fix test threshold for external tests.
...
PiperOrigin-RevId: 670720396
Change-Id: I28bff46535ffdd99724c75f8137834d84c3ec534
2024-09-03 15:06:42 -07:00
49711fa14f
Add support for joint and tendon frictionloss to MJX.
...
PiperOrigin-RevId: 670679777
Change-Id: I2eb56d013c11d4e1851f1c2edd6d98faaf854db9
2024-09-03 13:17:41 -07:00
9805df616b
Add framelinvel and frameangvel sensors to MJX.
...
PiperOrigin-RevId: 669647542
Change-Id: I612230e81b456c26f29173d1cb5a60975791875a
2024-08-31 06:12:37 -07:00
5391e6011a
Add sensor cutoff to MJX.
...
PiperOrigin-RevId: 669363779
Change-Id: Ia5c3f913c1b84e1893aa6072268a334123f00aeb
2024-08-30 10:09:17 -07:00
cf02171bf8
Add velocimeter and gyro sensors to MJX.
...
PiperOrigin-RevId: 669337697
Change-Id: I5e7f245b96eb334ffddd01757a9f631d9ad862cd
2024-08-30 08:39:40 -07:00
494e166f28
Fix a bug where mjx.make_data qLDiagInv has wrong size for sparse mass matrices.
...
PiperOrigin-RevId: 669087189
Change-Id: I39e97264a192a621e67ae5e079334fc8be38a25d
2024-08-29 15:40:08 -07:00
088079eff0
Rename mpr_tolerance and mpr_iterations to ccd_tolerance and ccd_iterations.
...
PiperOrigin-RevId: 666438658
Change-Id: I23dbdf939b9ab581f0fca4445f95c5bb36f6b463
2024-08-22 12:01:54 -07:00
a2649d6cec
Add efc_margin to MJX.
...
PiperOrigin-RevId: 666273240
Change-Id: I12d6cc0d3edd3c3faff5a876335910b42e9cc9bf
2024-08-22 03:27:02 -07:00
b2174a7ec3
Add framequat sensor to MJX.
...
PiperOrigin-RevId: 666254481
Change-Id: I50be8574e1e4677ef5dc23a2bdcbc296f301d7fc
2024-08-22 02:21:20 -07:00
b42780a5f4
Add subtree_vel function to MJX. This function matches mj_subtreeVel.
...
PiperOrigin-RevId: 666078305
Change-Id: Ia0ecbf8279e914f49ee60b862541d697e8d4abb1
2024-08-21 16:21:57 -07:00
806b8c8e2e
Add support for actuatorfrc and jointactfrc sensors in MJX.
...
PiperOrigin-RevId: 665804850
Change-Id: I99371fc4da055382f1564b2c67d25f7b30a0a0cf
2024-08-21 03:46:26 -07:00
a68141eeff
Implicitfast integration for MJX.
...
PiperOrigin-RevId: 664889120
Change-Id: Id3bd3916fe821ab2af79c52e53e14837ae829b2a
2024-08-19 11:12:36 -07:00
6241814ad7
Add velocity-dependent sensors to MJX. Includes: joint velocity, actuator velocity, and ball joint angular velocity.
...
PiperOrigin-RevId: 664461284
Change-Id: I5063f662d707fbe342493f9cd7993f009fac9c76
2024-08-18 08:47:35 -07:00
6a12787a32
Add camera projection sensor to MJX.
...
PiperOrigin-RevId: 664426007
Change-Id: I0f2681c156b12ab62503d45beb2cd8752e7202d6
2024-08-18 05:57:33 -07:00
51733c2a8a
Add rangefinder sensor to MJX.
...
PiperOrigin-RevId: 664417458
Change-Id: I1ad7720e48541349f155984584314b2b49ccbf36
2024-08-18 05:15:58 -07:00
6acf40613d
Add support for disabling MJX sensors.
...
PiperOrigin-RevId: 664410169
Change-Id: Icb278e9d7a9a493d5a61ab04c1300d8e412d0e83
2024-08-18 04:41:24 -07:00
390bce2352
Avoid placing arrays on device for unused MJX fields. Add `device` parameter to mjx.make_data.
...
PiperOrigin-RevId: 663808411
Change-Id: Ic55875f87f5c36b9dfe21e5ed5891d64c7003cab
2024-08-16 12:02:21 -07:00
ce9c0ea351
Modify MJX sensor_pos to handle unsupported sensors.
...
PiperOrigin-RevId: 663768186
Change-Id: I2e7600e77855a96a47f012fb64b6cc43abbf6f64
2024-08-16 10:10:37 -07:00
6db96e07e4
Add position-dependent sensors to MJX: magnetometer, ballquat, subtreecom, framepos, framexaxis, frameyaxis, framezaxis, and clock.
...
PiperOrigin-RevId: 663668212
Change-Id: Ifc9adc6bff4544172ab22572da04a093aebfdbd4
2024-08-16 03:10:24 -07:00
a74c184f9d
Add efc_pos to MJX.
...
fixes #1388
PiperOrigin-RevId: 663358494
Change-Id: I63885f9208d12edbaa4b7d123ce4c6d16f40dcbe
2024-08-15 10:33:44 -07:00
1141b4abbc
Force dtype of mjx.Contact.{geom1,geom2,geom} to int32.
...
This improves compatibility with MuJoCo.
PiperOrigin-RevId: 662655282
Change-Id: I2a336d9453f40cf0bfa85a7203b55709e23a07fe
2024-08-13 14:34:43 -07:00
ebf887e14d
Add sensors to MJX. Implements jointpos and actuatorpos as examples.
...
PiperOrigin-RevId: 660891329
Change-Id: Ie6c282a397de2f52843775ec747495e9be46d307
2024-08-08 10:33:19 -07:00
2d24c58819
Adds support for fixed tendons to MJX.
...
PiperOrigin-RevId: 654816559
Change-Id: I16ab379b0dc96d62a530e1c3ccd98a76b78c7f5a
2024-07-22 10:49:13 -07:00
d49d4bf64c
Make unsupported feature checking optional in put_model.
...
PiperOrigin-RevId: 654141976
Change-Id: I35b5f1e4c5b07e32075e847d4a757698a0b603c1
2024-07-19 15:23:32 -07:00
db1d51a1f5
Add more fields to mjx.Option and mjx.Statistic for MuJoCo compatibility.
...
Also add metadata fields to the Python bindings of mjModel and mjOption,
and fix dtype of a number of fields.
PiperOrigin-RevId: 654138876
Change-Id: If2bbe41bf3dfdb350d60b5289afc65fffb397ce9
2024-07-19 15:13:01 -07:00
85532b74e1
Speed up hfield JIT time by reducing XLA code generation overhead.
...
PiperOrigin-RevId: 653766462
Change-Id: I9d7e7fa3a4d33ae0fa9d00babd5c5478eb4eff88
2024-07-18 15:01:43 -07:00
dbe18f5740
Add additional MuJoCo fields to MJX structs.
...
PiperOrigin-RevId: 653270676
Change-Id: I5f7a62d3bd542abc0d618e63d2426663e7d6406f
2024-07-17 10:04:52 -07:00
c145b79f6e
Using typing instead of collections.abc. Fixes #1763
...
PiperOrigin-RevId: 651186508
Change-Id: I844ec884b802837554b0b60f2414d6bc37f2da14
2024-07-10 16:21:00 -07:00
28e5ba2d29
Copybara import of the project:
...
--
8857485483508e2c82c6a827daee8b457f88d1c8 by Taylor <taylor.athaniel.howell@gmail.com >:
velcoity -> velocity
COPYBARA_INTEGRATE_REVIEW=https://github.com/google-deepmind/mujoco/pull/1733 from thowell:docs 8857485483508e2c82c6a827daee8b457f88d1c8
PiperOrigin-RevId: 642779993
Change-Id: I84d15237314af8b823fe616f72bd79a088a21d61
2024-06-12 16:42:30 -07:00
238736393b
#mjx Make Data.time strongly typed.
...
PiperOrigin-RevId: 641774798
Change-Id: I5e13e3cefeb7942083f623f29757a459656062dd
2024-06-09 20:56:33 -07:00
4c3d9461ae
Elliptic friction in MJX.
...
PiperOrigin-RevId: 641384367
Change-Id: I510c565940324fbbf392ce537ce27e0cb9af3eb1
2024-06-07 16:05:58 -07:00
c511d02265
Add hfield. Fixes #1655 #1491 #1695
...
PiperOrigin-RevId: 638447127
Change-Id: Ib1e5020a8407bc100145a6b382e985c03dd4a848
2024-05-29 16:21:34 -07:00
7ca8a629a2
Speed up primitive sdf slightly.
...
PiperOrigin-RevId: 637105250
Change-Id: Iec73b0b30d5222617d01622771afbc95a646da45
2024-05-24 20:01:54 -07:00
86e6fc6de7
Fixes #1668
...
PiperOrigin-RevId: 634483090
Change-Id: I30b55bbf419e57a7b2966dfca4da6bb57d5879ad
2024-05-16 11:58:09 -07:00
c6b1293e58
Add contact force support function. #1555
...
PiperOrigin-RevId: 632591177
Change-Id: I15aa00686d9f885330506d2d63443c700cc029f6
2024-05-10 13:41:45 -07:00
8d1f65c01a
Add act to mjx viewer.
...
PiperOrigin-RevId: 629451047
Change-Id: I0baf546d3d91632291e93dec39f4dba7c329d370
2024-04-30 10:13:20 -07:00
719476c274
Fix bug in `mjx.ray` and speed up ray-mesh by ~30% for large meshes.
...
PiperOrigin-RevId: 629209303
Change-Id: I39466e9a2dc55ec48dcca1780e8abef903d1c754
2024-04-29 15:48:00 -07:00
e9709900b4
Add gravcomp.
...
PiperOrigin-RevId: 627791303
Change-Id: I87c9254967eb492d0c38153ff41d7458ffb69136
2024-04-24 11:19:34 -07:00
0a60d555c6
Fix small bug in testspeed.
...
PiperOrigin-RevId: 627773755
Change-Id: Iebbcaa41cfdde126312e388d368843a4754abfb8
2024-04-24 10:30:45 -07:00
d15db54537
Add support functions for id2name and name2id.
...
PiperOrigin-RevId: 627525330
Change-Id: Ic8069f64c94ef4b8f96f62b88ece62a8b296d870
2024-04-23 15:35:59 -07:00
c6ab354bd7
Remove type annotations unavailable in python < 3.10. Fixes #1590
...
PiperOrigin-RevId: 627188913
Change-Id: I9adc627349a4001291ac6d05080b761fee155ab0
2024-04-22 16:22:36 -07:00
71333938ec
Add support for condim to MJX.
...
PiperOrigin-RevId: 627136131
Change-Id: Ia62eb20cb80395b569c7fc7f4d4ef87e12eaaf40
2024-04-22 13:20:57 -07:00
c2d0c5dd1b
Add cylinder collisions using SDFs.
...
PiperOrigin-RevId: 626998709
Change-Id: I669a639b1cabf5ed522c73a07e54684d675fa396
2024-04-22 05:07:43 -07:00
3b64217b96
Remove deprecated mjx.device_* functions. They are not suited to upcoming condim support.
...
PiperOrigin-RevId: 626486447
Change-Id: Ibd41874bb3903123567168ebbac563399788d2f2
2024-04-19 15:34:28 -07:00
a4df912018
Prepare MJX for condim.
...
This is a refactor of collision_driver and some of its surrounding code in order to prepare for condim in MJX. In this change we reify types that are needed for condim: dim, efc_address, efc_type. We make explicit the way contacts are organized and grouped to guarantee that dim and efc_type are statically defined.
This change simplifies the way meshes are organized on device and slightly speeds up mesh collisions for cases where a single mesh is instanced across many geoms.
PiperOrigin-RevId: 626119500
Change-Id: Ic0c8599bcda2326f2e19cd3246a673e56097886b
2024-04-18 12:43:36 -07:00
Taylor Howell