30 lines
495 B
Python
30 lines
495 B
Python
import numpy as np
|
|
from dataclasses import dataclass
|
|
|
|
@dataclass
|
|
class JointState:
|
|
pos: np.ndarray
|
|
vel: np.ndarray
|
|
force: np.ndarray
|
|
|
|
@dataclass
|
|
class BaseState:
|
|
pos: np.ndarray
|
|
quat: np.ndarray
|
|
vel: np.ndarray
|
|
ang_vel: np.ndarray
|
|
|
|
@dataclass
|
|
class IMUState:
|
|
quat: np.ndarray
|
|
ang_vel: np.ndarray
|
|
acc: np.ndarray
|
|
pos: np.ndarray
|
|
lin_vel: np.ndarray
|
|
|
|
@dataclass
|
|
class RobotProprioception:
|
|
joint: JointState
|
|
base: BaseState
|
|
imu: IMUState
|