rlevo-environments 0.2.0

RL benchmark environments and landscapes for rlevo (internal crate — use `rlevo` for the full API)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

//! Physics state for [`super::Swimmer`].

use rapier3d::prelude::{MultibodyJointHandle, RigidBodyHandle};
use rlevo_core::base::State;

use super::observation::SwimmerObservation;

/// Physics state — body + joint handles plus the last observation. The
/// non-`Clone` world lives on the env struct directly.
///
/// The two revolute-z joints live in Rapier's `MultibodyJointSet` (not the
/// impulse set): a free-floating serial chain with no grounded reference is
/// a stiff problem for the PGS impulse solver, which injects kinetic energy
/// when constraint drift is corrected aggressively. The Featherstone-style
/// multibody solver parameterises the chain in reduced coordinates
/// (analogous to MuJoCo's generalised coordinates) and stays conservative.
#[derive(Debug, Clone)]
pub struct SwimmerState {
    pub segment0: RigidBodyHandle,
    pub segment1: RigidBodyHandle,
    pub segment2: RigidBodyHandle,
    pub joint1: MultibodyJointHandle,
    pub joint2: MultibodyJointHandle,
    pub last_obs: SwimmerObservation,
}

impl State<1> for SwimmerState {
    type Observation = SwimmerObservation;

    /// Returns `[8]`, matching the 8-dimensional observation vector.
    fn shape() -> [usize; 1] {
        [8]
    }

    /// Returns `true` when the last extracted observation is fully finite.
    /// A `false` result indicates a physics divergence (NaN or `±∞` in one
    /// or more of the observation elements).
    fn is_valid(&self) -> bool {
        self.last_obs.is_finite()
    }

    /// Returns the last observation cached by the environment after the most
    /// recent `reset` or `step` call.
    fn observe(&self) -> SwimmerObservation {
        self.last_obs
    }
}