imt-ring 1.6.27__tar.gz → 1.6.29__tar.gz

{imt_ring-1.6.27 → imt_ring-1.6.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.6.27
+Version: 1.6.29
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring
@@ -38,6 +38,10 @@ Requires-Dist: nbmake; extra == "dev"
 # Recurrent Inertial Graph-based Estimator (RING)
 <img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/coverage_badge.svg" height="20" />
 
+> **ℹ️ Tip:**
+> 
+> Check out my new plug-and-play interface for inertial motion tracking (RING included) [here](https://github.com/simon-bachhuber/imt.git).
+
 ## Installation
 
 Supports `Python=3.10/3.11/3.12` (tested).

{imt_ring-1.6.27 → imt_ring-1.6.29}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "imt-ring"
-version = "1.6.27"
+version = "1.6.29"
 authors = [
   { name="Simon Bachhuber", email="simon.bachhuber@fau.de" },
 ]

{imt_ring-1.6.27 → imt_ring-1.6.29}/readme.md

@@ -5,6 +5,10 @@
 # Recurrent Inertial Graph-based Estimator (RING)
 <img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/coverage_badge.svg" height="20" />
 
+> **ℹ️ Tip:**
+> 
+> Check out my new plug-and-play interface for inertial motion tracking (RING included) [here](https://github.com/simon-bachhuber/imt.git).
+
 ## Installation
 
 Supports `Python=3.10/3.11/3.12` (tested).

{imt_ring-1.6.27 → imt_ring-1.6.29}/src/imt_ring.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.6.27
+Version: 1.6.29
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring
@@ -38,6 +38,10 @@ Requires-Dist: nbmake; extra == "dev"
 # Recurrent Inertial Graph-based Estimator (RING)
 <img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/coverage_badge.svg" height="20" />
 
+> **ℹ️ Tip:**
+> 
+> Check out my new plug-and-play interface for inertial motion tracking (RING included) [here](https://github.com/simon-bachhuber/imt.git).
+
 ## Installation
 
 Supports `Python=3.10/3.11/3.12` (tested).

{imt_ring-1.6.27 → imt_ring-1.6.29}/src/ring/__init__.py

@@ -98,7 +98,10 @@ def RING(lam: list[int] | None, Ts: float | None, **kwargs) -> ml.AbstractFilter
         add_Ts = True
 
     ringnet = ml.RING(
-        params=params, lam=None if lam is None else tuple(lam), jit=False, name="RING"
+        params=params,
+        lam=None if lam is None else tuple(lam),
+        jit=config.pop("jit", False),
+        name="RING",
     )
     ringnet = ml.base.ScaleX_FilterWrapper(ringnet)
     if config["use_lpf"]:

{imt_ring-1.6.27 → imt_ring-1.6.29}/src/ring/algorithms/generator/base.py

@@ -55,6 +55,11 @@ class RCMG:
     ) -> None:
         "Random Chain Motion Generator"
 
+        # add some default values
+        randomize_hz_kwargs_defaults = dict(add_dt=True)
+        randomize_hz_kwargs_defaults.update(randomize_hz_kwargs)
+        randomize_hz_kwargs = randomize_hz_kwargs_defaults
+
         sys, config = utils.to_list(sys), utils.to_list(config)
         sys_ml = sys[0] if sys_ml is None else sys_ml
 
@@ -409,7 +414,11 @@ def _build_mconfig_batched_generator(
         @jax.vmap
         def _vmapped_context(key, q, sys):
             x, _ = jax.vmap(kinematics.forward_kinematics_transforms, (None, 0))(sys, q)
-            X = {"dt": jnp.array(sys.dt)} if randomize_hz else {}
+            X = (
+                {"dt": jnp.array(sys.dt)}
+                if (randomize_hz and randomize_hz_kwargs["add_dt"])
+                else {}
+            )
             Xy, extras = (X, {}), (key, q, x, sys)
             return _finalize_fn(Xy, extras)
 

{imt_ring-1.6.27 → imt_ring-1.6.29}/src/ring/base.py

@@ -558,6 +558,7 @@ class System(_Base):
         new_stif: Optional[jax.Array] = None,
         new_zero: Optional[jax.Array] = None,
         seed: int = 1,
+        warn: bool = True,
     ):
         "By default damping, stiffness are set to zero."
         from ring.algorithms import get_joint_model
@@ -587,7 +588,7 @@ class System(_Base):
 
         jm = get_joint_model(new_joint_type)
         if jm.init_joint_params is not None:
-            sys = sys.from_str(sys.to_str(), seed=seed)
+            sys = sys.from_str(sys.to_str(warn=warn), seed=seed)
 
         return sys
 
@@ -786,8 +787,8 @@ class System(_Base):
     def from_str(xml: str, seed: int = 1):
         return ring.io.load_sys_from_str(xml, seed)
 
-    def to_str(self) -> str:
-        return ring.io.save_sys_to_str(self)
+    def to_str(self, warn: bool = True) -> str:
+        return ring.io.save_sys_to_str(self, warn=warn)
 
     def to_xml(self, path: str) -> None:
         ring.io.save_sys_to_xml(self, path)

{imt_ring-1.6.27 → imt_ring-1.6.29}/src/ring/io/xml/to_xml.py

@@ -5,18 +5,19 @@ from xml.etree.ElementTree import SubElement
 from xml.etree.ElementTree import tostring
 
 import jax.numpy as jnp
-from ring import base
 from tree_utils import batch_concat
 
+from ring import base
+
 from . import abstract
 from .abstract import _to_str
 
 
-def save_sys_to_str(sys: base.System) -> str:
+def save_sys_to_str(sys: base.System, warn: bool = True) -> str:
     for joint_type in sys.links.joint_params:
         for i, link_name in enumerate(sys.link_names):
             joint_params_flat = batch_concat((sys.links[i]).joint_params[joint_type], 0)
-            if not jnp.all(joint_params_flat == 0.0):
+            if warn and (not jnp.all(joint_params_flat == 0.0)):
                 warnings.warn(
                     "The system has `sys.links.joint_params` unequal to the 'default'"
                     f" value (of zeros). In particular the link `{link_name}` has for"

{imt_ring-1.6.27 → imt_ring-1.6.29}/src/ring/maths.py

@@ -90,6 +90,19 @@ def angle_error(q, qhat):
     return jnp.abs(quat_angle(quat_mul(quat_inv(q), qhat)))
 
 
+def inclination_loss(q, qhat):
+    """Absolute inclination angle in radians. `q`'s are from body-to-eps.
+    This function fullfills
+        inclination_loss(q1, q2)
+        == inclination_loss(qmt.addHeading(q1, H), q2)
+        == inclination_loss(q1, qmt.addHeading(q2, H))`
+    for any q1, q2, H
+    """
+    q_rel = quat_mul(q, quat_inv(qhat))
+    q_rel_incl = quat_project(q_rel, [0, 0, 1.0])[1]
+    return jnp.abs(quat_angle(q_rel_incl))
+
+
 def unit_quats_like(array):
     "Array of *unit* quaternions of identical shape."
     if array.shape[-1] != 4:

{imt_ring-1.6.27 → imt_ring-1.6.29}/src/ring/rendering/mujoco_render.py

@@ -8,7 +8,10 @@ from ring import maths
 
 _skybox = """<texture name="skybox" type="skybox" builtin="gradient" rgb1=".4 .6 .8" rgb2="0 0 0" width="800" height="800" mark="random" markrgb="1 1 1"/>"""  # noqa: E501
 _skybox_white = """<texture name="skybox" type="skybox" builtin="gradient" rgb1="1 1 1" rgb2="1 1 1" width="800" height="800" mark="random" markrgb="1 1 1"/>"""  # noqa: E501
-_floor = """<geom name="floor" pos="0 0 -0.84" size="0 0 1" type="plane" material="matplane" mass="0"/>"""  # noqa: E501
+
+
+def _floor(floor_z: float) -> str:
+    return f"""<geom name="floor" pos="0 0 {floor_z}" size="0 0 1" type="plane" material="matplane" mass="0"/>"""  # noqa: E501
 
 
 def _build_model_of_geoms(
@@ -16,6 +19,7 @@ def _build_model_of_geoms(
     cameras: dict[int, Sequence[str]],
     lights: dict[int, Sequence[str]],
     floor: bool,
+    floor_z: float,
     stars: bool,
     debug: bool,
 ) -> mujoco.MjModel:
@@ -94,7 +98,7 @@ def _build_model_of_geoms(
 <camera pos="0 -1 1" name="target" mode="targetbodycom" target="{targetbody}"/>
 <camera pos="0 -3 3" name="targetfar" mode="targetbodycom" target="{targetbody}"/>
 <camera pos="0 -5 5" name="targetFar" mode="targetbodycom" target="{targetbody}"/>
-{_floor if floor else ''}
+{_floor(floor_z) if floor else ''}
 {inside_worldbody_cameras}
 {inside_worldbody_lights}
 {inside_worldbody}
@@ -171,6 +175,7 @@ class MujocoScene:
         add_lights: dict[int, str | Sequence[str]] = _default_lights,
         show_stars: bool = True,
         show_floor: bool = True,
+        floor_z: float = -0.84,
         debug: bool = False,
     ) -> None:
         self.debug = debug
@@ -185,6 +190,7 @@ class MujocoScene:
         self.add_cameras, self.add_lights = to_list(add_cameras), to_list(add_lights)
         self.show_stars = show_stars
         self.show_floor = show_floor
+        self.floor_z = floor_z
 
     def init(self, geoms: list[base.Geometry]):
         self._parent_ids = list(set([geom.link_idx for geom in geoms]))
@@ -193,6 +199,7 @@ class MujocoScene:
             self.add_cameras,
             self.add_lights,
             floor=self.show_floor,
+            floor_z=self.floor_z,
             stars=self.show_stars,
             debug=self.debug,
         )