Skip to content

Commit

Permalink
[N/A] Expose height in Spot_Wrapper velocity command
Browse files Browse the repository at this point in the history
  • Loading branch information
fgillespie-bdai committed Oct 17, 2024
1 parent f410423 commit 401a09a
Showing 1 changed file with 19 additions and 6 deletions.
25 changes: 19 additions & 6 deletions spot_wrapper/wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -1299,7 +1299,7 @@ def get_mobility_params(self) -> spot_command_pb2.MobilityParams:
return self._mobility_params

def velocity_cmd(
self, v_x: float, v_y: float, v_rot: float, cmd_duration: float = 0.125
self, v_x: float, v_y: float, v_rot: float, cmd_duration: float = 0.125, body_height: float = 0.0
) -> typing.Tuple[bool, str]:
"""
Expand All @@ -1311,16 +1311,29 @@ def velocity_cmd(
v_rot: Angular velocity around the Z axis in radians
cmd_duration: (optional) Time-to-live for the command in seconds. Default is 125ms (assuming 10Hz command
rate).
body_height: Offset of the body relative to nominal stand height, in metres
Returns:
Tuple of bool success and a string message
"""
end_time = time.time() + cmd_duration
response = self._robot_command(
RobotCommandBuilder.synchro_velocity_command(v_x=v_x, v_y=v_y, v_rot=v_rot, params=self._mobility_params),
end_time_secs=end_time,
timesync_endpoint=self._robot.time_sync.endpoint,
)

if body_height:
# If there is an offset body_height, use it to pose the robot
response = self._robot_command(
RobotCommandBuilder.synchro_velocity_command(v_x=v_x, v_y=v_y, v_rot=v_rot, body_height=body_height),
end_time_secs=end_time,
timesync_endpoint=self._robot.time_sync.endpoint,
)
else:
# Otherwise just use the mobility params
response = self._robot_command(
RobotCommandBuilder.synchro_velocity_command(
v_x=v_x, v_y=v_y, v_rot=v_rot, params=self._mobility_params
),
end_time_secs=end_time,
timesync_endpoint=self._robot.time_sync.endpoint,
)
self.last_velocity_command_time = end_time
return response[0], response[1]

Expand Down

0 comments on commit 401a09a

Please sign in to comment.