Commit Graph

169 Commits

Author SHA1 Message Date
Baruch Tabanpour 47ebb0614e Set some visual params as jax.Arrays for visual domain rando.
PiperOrigin-RevId: 692273879
Change-Id: I80475aa65744de06a6bfebbfb95a44faa751595f
2024-11-01 13:24:29 -07:00
Taylor Howell a51f346059 Use sparse (uncompressed) actuator_moment in mj_transmission.
PiperOrigin-RevId: 692179704
Change-Id: Ic30ac5a98dc13de2028e378df65dc88ba3912bf5
2024-11-01 08:06:28 -07:00
Yuval Tassa c9f78ad8a8 Rename nnzJ to nJ in mjData.
This naming is more consistent.

PiperOrigin-RevId: 691858223
Change-Id: Ife3db822afa0b51b30d0792548b35b3742da388a
2024-10-31 10:43:10 -07:00
Taylor Howell 554641060b Add support for computing elliptic cone contact forces to MJX. Fixes #2153.
PiperOrigin-RevId: 691829630
Change-Id: I862ac9c7db07d0a74b718d2a41e19d5ae636a260
2024-10-31 09:22:59 -07:00
Baruch Tabanpour 26ccaeb78e Don't populate _qM_sparse, _qLD_sparse, _qLDiagInv_sparse fields if they were never set via _full_compat. Fixes #2188
PiperOrigin-RevId: 691538016
Change-Id: Ib0385166af2dfbbb46cf1986cb692e1fda467a0b
2024-10-30 13:43:02 -07:00
Baruch Tabanpour 03d04f6096 Add texture fields to mjx.Model for madrona collaborators.
PiperOrigin-RevId: 691170517
Change-Id: Icb8e2d6ed65a13d17dc2565d3d84de629a10c37a
2024-10-29 14:49:12 -07:00
Baruch Tabanpour d01fb81b50 Use eq_active0 in MJX.
PiperOrigin-RevId: 690727109
Change-Id: If60a99d7644e0376a27039478319ff5300b15cc1
2024-10-28 13:21:05 -07:00
Taylor Howell 3c21abc0e5 Add ray intersection with ellipsoid to MJX.
PiperOrigin-RevId: 690557087
Change-Id: Icd2f0be1243574720baeac82b8bdd92e57c6a61c
2024-10-28 04:25:11 -07:00
Taylor Howell 1c424644dd Fix MJX touch sensor.
PiperOrigin-RevId: 690544118
Change-Id: Ic909c6f6ce0e31237cbe317fbbfbb1fd267bd462
2024-10-28 03:39:53 -07:00
Baruch Tabanpour 078c7bb42e internal change
PiperOrigin-RevId: 689931872
Change-Id: Icc7b261b6d98e181a6bf0845f44a050af2833db2
2024-10-25 15:29:39 -07:00
Baruch Tabanpour f24de91cc9 Use eq_active in MJX. Fixes #2173.
PiperOrigin-RevId: 689549368
Change-Id: I14d9817c5ca5dfc2a735ab8bacfb11b75c9feadc
2024-10-24 16:08:14 -07:00
Baruch Tabanpour 2de430a61f Use newton by default in benchmark.
PiperOrigin-RevId: 689538944
Change-Id: I99b5588ae5980eee43fb33ac390122ce38000dd3
2024-10-24 15:35:01 -07:00
Kyle Bayes ca348632a9 Change tolerances for ellipsoid-capsule and ellipsoid-sphere collision tests in MJX.
PiperOrigin-RevId: 689301610
Change-Id: I86be65524d13a19abd0b378b0933a5fbbe0ecab8
2024-10-24 02:42:16 -07:00
Taylor Howell b00a7c6705 Add touch sensor to MJX.
PiperOrigin-RevId: 688056696
Change-Id: I50b0ebf3141aab0a5b07e5ae770101ff8b7737d4
2024-10-21 02:48:36 -07:00
Baruch Tabanpour 680fb3e5dd Add more public support functions for MJX.
PiperOrigin-RevId: 687469475
Change-Id: I4218cc2f6d51d3f2f63b1a9a96953d550a5503a5
2024-10-18 17:15:18 -07:00
Taylor Howell 096853e192 Add JOINTINPARENT actuator transmission type to MJX.
PiperOrigin-RevId: 686083761
Change-Id: I5159d050812d32e15ced06df132d3e434fced814
2024-10-15 06:38:02 -07:00
Taylor Howell ac91a7639d Add tendon velocity sensor to MJX.
PiperOrigin-RevId: 685689974
Change-Id: Ib5e65b19c20b74e20cffa66c6baca0d8faaaf3ae
2024-10-14 06:47:59 -07:00
Taylor Howell 1945926351 Fix computation of tangential contact forces in MJX _decode_pyramid.
PiperOrigin-RevId: 685665382
Change-Id: Icdbba4a3ee5fec804340d73d525a36ee570c2aa8
2024-10-14 05:01:00 -07:00
Taylor Howell 223892c051 Add tendon position sensor to MJX.
PiperOrigin-RevId: 684851437
Change-Id: Ica0a30762a9cb3d612357b40b6f87cd2fc544fdb
2024-10-11 08:47:52 -07:00
Baruch Tabanpour 22e4f7fc31 Fix bug in MJX frictionloss.
PiperOrigin-RevId: 684198925
Change-Id: I5f6575759729712550bcb8d7659aaa7a657c66fc
2024-10-09 15:50:58 -07:00
Taylor Howell 356d172f56 Update MJX SolverType docstring with Newton.
PiperOrigin-RevId: 684040839
Change-Id: I27bc87b387b6b405f5c76350438d1fdd6867450a
2024-10-09 08:08:36 -07:00
Baruch Tabanpour fa22e6d001 Add sphere-ellipsoid, sphere-cylinder. #2126
PiperOrigin-RevId: 683747371
Change-Id: Ic11803bb395944380d25ce848d9fbcf19afaad77
2024-10-08 13:37:00 -07:00
Taylor Howell 160ed9bfe2 Add pulley support to spatial tendons in MJX.
PiperOrigin-RevId: 683103659
Change-Id: I22544071a3fc6fbd09f9a59690106701b101ee6a
2024-10-07 03:00:53 -07:00
Copybara-Service 3ea8be2dcb Merge pull request #2078 from hartikainen:enable-mjb-for-mjx-testspeed
PiperOrigin-RevId: 682328610
Change-Id: I1329d8f7a0a47234e3eda61c457567825c89bd88
2024-10-04 08:22:53 -07:00
Taylor Howell c7c2edf758 Add external sphere and cylinder wrapping for spatial tendons to MJX.
PiperOrigin-RevId: 681808698
Change-Id: I37612e2c6b781b759f9009e1bc6ae10b4e861c8d
2024-10-03 03:56:24 -07:00
Baruch Tabanpour 1a9d307042 Use mocap pos/quat in smooth.kinematics.
PiperOrigin-RevId: 681136974
Change-Id: I0ece009ed1e68b61b0b96da2e685e86000f7f6b8
2024-10-01 13:12:33 -07:00
Baruch Tabanpour cb745db412 Ensure ten_wrapnum/wrapadr are int type.
PiperOrigin-RevId: 680709773
Change-Id: Iebc0c796e66e657b8bebcc5fe4b2d347d9bdf42c
2024-09-30 14:02:48 -07:00
Kristian Hartikainen 6f031f2293 Enable binary .mjb in mjx/viewer.py 2024-09-26 12:40:20 +03:00
Kristian Hartikainen 827401554e Enable binary .mjb in mjx/testspeed.py 2024-09-26 12:40:13 +03:00
Erik Frey 2785dbc0da Add tendon types to MJX name2id and id2name.
PiperOrigin-RevId: 678424789
Change-Id: I2634be0810f8f7dafec597919c512752d2efcfa6
2024-09-24 15:46:33 -07:00
Baruch Tabanpour 306e52c30f internal change.
PiperOrigin-RevId: 678355240
Change-Id: I271bece139de5d6e2e629d7f2fe84a4e2dd71138
2024-09-24 12:37:58 -07:00
Baruch Tabanpour b09c1a977e Add jit flag to viewer (easier debugging). Make convex.xml stable.
PiperOrigin-RevId: 678312872
Change-Id: I0d65d7c2d05a7b5103fa0be2233cbc5ba232817c
2024-09-24 10:49:34 -07:00
Baruch Tabanpour 3a17be2d24 Add pause key to MJX viewer.
PiperOrigin-RevId: 677959701
Change-Id: I7f816eb6bf702461f0d749b2aa6cd1664c258f3d
2024-09-23 15:23:13 -07:00
Erik Frey 5838f84777 Fix bug where MJX data tendon fields are incorrect after calling put_data.
PiperOrigin-RevId: 677911639
Change-Id: I75f536fa348e2941d9ec2d454f475cd067e6d3af
2024-09-23 13:12:49 -07:00
Google DeepMind bbe8a0c516 Strip weak_types in mjx.put_model.
PiperOrigin-RevId: 677859138
Change-Id: I9ef010c0c37d04f9931adf884ab1dce0521b8567
2024-09-23 10:54:53 -07:00
Taylor Howell 76b48b8e61 Add site semantics to weld constraint in MJX.
PiperOrigin-RevId: 677823461
Change-Id: Iab20e042695dfad416d61fa7cc938faba8c7a6b6
2024-09-23 09:25:08 -07:00
Taylor Howell ab3954d892 Set actuator_force in MJX. Fixes #2068.
PiperOrigin-RevId: 676867005
Change-Id: I80514a050ff40cb6d43a95b74c4a36c4991db588
2024-09-20 09:16:55 -07:00
Taylor Howell f9401776d4 Add framelinacc and frameangacc sensors to MJX.
PiperOrigin-RevId: 675141513
Change-Id: Ie38dba3e8be82e0554c7c9e47c6d07f3cdad29b8
2024-09-16 07:34:16 -07:00
Taylor Howell 9d73211757 Add force and torque sensors to MJX.
PiperOrigin-RevId: 675130978
Change-Id: I8a3b925c57f317aebc0c345578cebeb4210dc11d
2024-09-16 06:57:00 -07:00
Taylor Howell e3d3a24b36 Add spatial tendons with site wrapping to MJX.
PiperOrigin-RevId: 673783473
Change-Id: I09b13787e3e0914c8de21d81db0347c4638b2f1d
2024-09-12 04:17:16 -07:00
Taylor Howell 8710126d08 Raise NotImplementedError for unsupported sensors in MJX. Update MJX feature parity docs to include fully supported sensors.
PiperOrigin-RevId: 673325233
Change-Id: Ib262bdd37937c5230765c11d1f61581e55e8c397
2024-09-11 03:50:46 -07:00
Taylor Howell fa156ed22f Add accelerometer to MJX.
PiperOrigin-RevId: 671355424
Change-Id: I8593bcad6ba311f3c555b118889e2a5c6b367ec7
2024-09-05 06:57:09 -07:00
Taylor Howell 20296466b1 Add subtreelinvel and subtreeangmom sensors to MJX.
PiperOrigin-RevId: 671347528
Change-Id: I8a32590c650ca6d0bb6df43d0347f707ffc05f12
2024-09-05 06:21:50 -07:00
Taylor Howell 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
Taylor Howell 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
Yuval Tassa 8954a088ce Add automatic free-joint alignment.
PiperOrigin-RevId: 671010083
Change-Id: Ib7d60f75c7f545b1886703715663db8de77feeef
2024-09-04 10:07:33 -07:00
Baruch Tabanpour 303e5e7718 Fix test threshold for external tests.
PiperOrigin-RevId: 670720396
Change-Id: I28bff46535ffdd99724c75f8137834d84c3ec534
2024-09-03 15:06:42 -07:00
Baruch Tabanpour 49711fa14f Add support for joint and tendon frictionloss to MJX.
PiperOrigin-RevId: 670679777
Change-Id: I2eb56d013c11d4e1851f1c2edd6d98faaf854db9
2024-09-03 13:17:41 -07:00
Taylor Howell 9805df616b Add framelinvel and frameangvel sensors to MJX.
PiperOrigin-RevId: 669647542
Change-Id: I612230e81b456c26f29173d1cb5a60975791875a
2024-08-31 06:12:37 -07:00
Taylor Howell 5391e6011a Add sensor cutoff to MJX.
PiperOrigin-RevId: 669363779
Change-Id: Ia5c3f913c1b84e1893aa6072268a334123f00aeb
2024-08-30 10:09:17 -07:00