STATE_VEC_IDX_MAPPING = { # [0, 10): right arm joint positions **{ "arm_joint_{}_pos".format(i): i for i in range(10) }, **{ "right_arm_joint_{}_pos".format(i): i for i in range(10) }, # [10, 15): right gripper joint positions **{ "gripper_joint_{}_pos".format(i): i + 10 for i in range(5) }, **{ "right_gripper_joint_{}_pos".format(i): i + 10 for i in range(5) }, "gripper_open": 10, # alias of right_gripper_joint_0_pos "right_gripper_open": 10, # [15, 25): right arm joint velocities **{ "arm_joint_{}_vel".format(i): i + 15 for i in range(10) }, **{ "right_arm_joint_{}_vel".format(i): i + 15 for i in range(10) }, # [25, 30): right gripper joint velocities **{ "gripper_joint_{}_vel".format(i): i + 25 for i in range(5) }, **{ "right_gripper_joint_{}_vel".format(i): i + 25 for i in range(5) }, "gripper_open_vel": 25, # alias of right_gripper_joint_0_vel "right_gripper_open_vel": 25, # [30, 33): right end effector positions "eef_pos_x": 30, "right_eef_pos_x": 30, "eef_pos_y": 31, "right_eef_pos_y": 31, "eef_pos_z": 32, "right_eef_pos_z": 32, # [33, 39): right end effector 6D pose "eef_angle_0": 33, "right_eef_angle_0": 33, "eef_angle_1": 34, "right_eef_angle_1": 34, "eef_angle_2": 35, "right_eef_angle_2": 35, "eef_angle_3": 36, "right_eef_angle_3": 36, "eef_angle_4": 37, "right_eef_angle_4": 37, "eef_angle_5": 38, "right_eef_angle_5": 38, # [39, 42): right end effector velocities "eef_vel_x": 39, "right_eef_vel_x": 39, "eef_vel_y": 40, "right_eef_vel_y": 40, "eef_vel_z": 41, "right_eef_vel_z": 41, # [42, 45): right end effector angular velocities "eef_angular_vel_roll": 42, "right_eef_angular_vel_roll": 42, "eef_angular_vel_pitch": 43, "right_eef_angular_vel_pitch": 43, "eef_angular_vel_yaw": 44, "right_eef_angular_vel_yaw": 44, # [45, 50): reserved # [50, 60): left arm joint positions **{ "left_arm_joint_{}_pos".format(i): i + 50 for i in range(10) }, # [60, 65): left gripper joint positions **{ "left_gripper_joint_{}_pos".format(i): i + 60 for i in range(5) }, "left_gripper_open": 60, # alias of left_gripper_joint_0_pos # [65, 75): left arm joint velocities **{ "left_arm_joint_{}_vel".format(i): i + 65 for i in range(10) }, # [75, 80): left gripper joint velocities **{ "left_gripper_joint_{}_vel".format(i): i + 75 for i in range(5) }, "left_gripper_open_vel": 75, # alias of left_gripper_joint_0_vel # [80, 83): left end effector positions "left_eef_pos_x": 80, "left_eef_pos_y": 81, "left_eef_pos_z": 82, # [83, 89): left end effector 6D pose "left_eef_angle_0": 83, "left_eef_angle_1": 84, "left_eef_angle_2": 85, "left_eef_angle_3": 86, "left_eef_angle_4": 87, "left_eef_angle_5": 88, # [89, 92): left end effector velocities "left_eef_vel_x": 89, "left_eef_vel_y": 90, "left_eef_vel_z": 91, # [92, 95): left end effector angular velocities "left_eef_angular_vel_roll": 92, "left_eef_angular_vel_pitch": 93, "left_eef_angular_vel_yaw": 94, # [95, 100): reserved # [100, 102): base linear velocities "base_vel_x": 100, "base_vel_y": 101, # [102, 103): base angular velocities "base_angular_vel": 102, # [103, 128): reserved } STATE_VEC_LEN = 128