rlightning.humanoid.utils.motion_viewer.robot_object_motion_viewer

class rlightning.humanoid.utils.motion_viewer.robot_object_motion_viewer.RobotObjectMotionViewer(robot_config, motion_path: str, rate_limit: bool = True, camera_follow: bool = True, transparent_robot: int = 0, record_video: bool = False, video_path: str = None, video_width: int = 640, video_height: int = 480, video_fps: float = 30.0, log_dir: str = './logs')

Bases: RobotMotionViewer

reset(fps=30)

setup()

step(human_motion_data=None, show_human_body_name=False, human_point_scale=0.1, human_pos_offset=MockModule('numpy.array'), obj_pos=None, obj_rot=None)

by default visualize robot motion. also support visualize human motion by providing human_motion_data, to compare with robot motion.

human_motion_data is a dict of {“human body name”: (3d global translation, 3d global rotation)}.

if rate_limit is True, the motion will be visualized at the same rate as the motion data. else, the motion will be visualized as fast as possible.

rlightning.humanoid.utils.motion_viewer.robot_object_motion_viewer.draw_frame(pos, mat, v, size, joint_name=None, orientation_correction=scipy.spatial.transform.Rotation.from_euler, pos_offset=MockModule('numpy.array'))