# Copyright (c) 2023 Boston Dynamics, Inc. All rights reserved.
#
# Downloading, reproducing, distributing or otherwise using the SDK Software
# is subject to the terms and conditions of the Boston Dynamics Software
# Development Kit License (20191101-BDSDK-SL).
"""For clients to the power command service."""
import collections
import functools
import time
from concurrent.futures import TimeoutError
from deprecated.sphinx import deprecated
from google.protobuf.duration_pb2 import Duration
from bosdyn.api import (basic_command_pb2, full_body_command_pb2, license_pb2, power_pb2,
power_service_pb2_grpc, robot_command_pb2, robot_state_pb2)
from bosdyn.client.common import (BaseClient, common_license_errors, error_factory,
handle_common_header_errors, handle_lease_use_result_errors,
handle_unset_status_error)
from bosdyn.client.exceptions import (Error, InternalServerError, LicenseError, ResponseError,
TimedOutError)
from .lease import add_lease_wallet_processors
[docs]class PowerResponseError(ResponseError):
"""General class of errors for Power service."""
[docs]class ShorePowerConnectedError(PowerResponseError):
"""Robot cannot be powered on while on wall power."""
[docs]class BatteryMissingError(PowerResponseError):
"""Battery not inserted into robot."""
[docs]class CommandInProgressError(PowerResponseError):
"""Power command cannot be overwritten."""
[docs]class EstoppedError(PowerResponseError):
"""Cannot power on while estopped; inspect EStopState for more info."""
[docs]class OverriddenError(PowerResponseError):
"""The command was overridden and is no longer valid."""
[docs]class KeepaliveMotorsOffError(PowerResponseError):
"""Cannot power on while Keepalive requests motors off."""
[docs]class FaultedError(PowerResponseError):
"""Cannot power on due to a fault; inspect FaultState for more info."""
[docs]class PowerError(Error):
"""General class of errors to handle non-response non-grpc errors."""
[docs]class CommandTimedOutError(PowerError):
"""Timed out waiting for SUCCESS response from power command."""
[docs]class FanControlTemperatureError(PowerResponseError):
"""Current measured robot temperatures are too high to accept user fan command."""
[docs]class SafetyStopIncompatibleHardwareError(PowerResponseError):
"""SafetyStop command invalid because robot is not configured for SRSF."""
[docs]class SafetyStopFailedError(PowerResponseError):
"""SafetyStop command executed and failed."""
[docs]class SafetyStopUnknownStopTypeError(PowerResponseError):
"""SafetyStop command failed due to unknown stop type."""
[docs]class PowerClient(BaseClient):
"""A client for enabling / disabling robot motor power.
Commands are non-blocking. Clients are expected to issue a power command and then periodically
check the status of this command.
This service requires ownership over the robot, in the form of a lease.
"""
default_service_name = 'power'
service_type = 'bosdyn.api.PowerService'
def __init__(self):
super(PowerClient, self).__init__(power_service_pb2_grpc.PowerServiceStub)
[docs] def update_from(self, other):
super(PowerClient, self).update_from(other)
if self.lease_wallet:
add_lease_wallet_processors(self, self.lease_wallet)
[docs] def power_command(self, request, lease=None, **kwargs):
"""Issue a power request to the robot."""
req = self._power_command_request(lease, request)
return self.call(self._stub.PowerCommand, req, None, _power_command_error_from_response,
copy_request=False, **kwargs)
[docs] def power_command_async(self, request, lease=None, **kwargs):
"""Async version of power_command()."""
req = self._power_command_request(lease, request)
return self.call_async(self._stub.PowerCommand, req, None,
_power_command_error_from_response, copy_request=False, **kwargs)
[docs] def power_command_feedback(self, power_command_id, **kwargs):
"""Check the status of a previously issued power command."""
req = self._power_command_feedback_request(power_command_id)
return self.call(self._stub.PowerCommandFeedback, req, _power_status_from_response,
_power_feedback_error_from_response, copy_request=False, **kwargs)
[docs] def power_command_feedback_async(self, power_command_id, **kwargs):
"""Async version of power_command_feedback()"""
req = self._power_command_feedback_request(power_command_id)
return self.call_async(self._stub.PowerCommandFeedback, req, _power_status_from_response,
_power_feedback_error_from_response, copy_request=False, **kwargs)
[docs] def fan_power_command(self, percent_power, duration, lease=None, **kwargs):
"""Issue a fan power command request to the robot."""
req = self._fan_power_command_request(lease, percent_power, duration)
return self.call(self._stub.FanPowerCommand, req, None,
_fan_power_command_error_from_response, **kwargs)
[docs] def fan_power_command_async(self, percent_power, duration, lease=None, **kwargs):
"""Async version of fan_power_command()"""
req = self._fan_power_command_request(lease, percent_power, duration)
return self.call_async(self._stub.FanPowerCommand, req, None,
_fan_power_command_error_from_response, **kwargs)
[docs] def fan_power_command_feedback(self, command_id, **kwargs):
"""Check the status of a previously issued fan command"""
req = self._fan_power_command_feedback_request(command_id)
return self.call(self._stub.FanPowerCommandFeedback, req, None,
_fan_power_feedback_error_from_response, **kwargs)
[docs] def fan_power_command_feedback_async(self, command_id, **kwargs):
"""Async version of fan_power_command_feedback()"""
req = self._fan_power_command_feedback_request(command_id)
return self.call_async(self._stub.FanPowerCommandFeedback, req, None,
_fan_power_feedback_error_from_response, **kwargs)
[docs] def reset_safety_stop(self, safety_stop_type, lease=None, **kwargs):
"""Issue a reset safety stop request to the robot."""
req = self._reset_safety_stop_request(lease, safety_stop_type)
return self.call(self._stub.ResetSafetyStop, req, None,
_reset_safety_stop_error_from_response, **kwargs)
[docs] def reset_safety_stop_async(self, safety_stop_type, lease=None, **kwargs):
"""Async version of reset_safety_stop()"""
req = self._reset_safety_stop_request(lease, safety_stop_type)
return self.call_async(self._stub.ResetSafetyStop, req, None,
_reset_safety_stop_error_from_response, **kwargs)
@staticmethod
def _power_command_request(lease, request):
return power_pb2.PowerCommandRequest(lease=lease, request=request)
@staticmethod
def _power_command_feedback_request(power_command_id):
return power_pb2.PowerCommandFeedbackRequest(power_command_id=power_command_id)
@staticmethod
def _fan_power_command_request(lease, percent_power, duration):
return power_pb2.FanPowerCommandRequest(lease=lease, percent_power=percent_power,
duration=Duration(seconds=duration))
@staticmethod
def _fan_power_command_feedback_request(command_id):
return power_pb2.FanPowerCommandFeedbackRequest(command_id=command_id)
@staticmethod
def _reset_safety_stop_request(lease, safety_stop_type):
return power_pb2.ResetSafetyStopRequest(lease=lease, safety_stop_type=safety_stop_type)
def _handle_license_errors(func):
"""Decorate "error from response" functions to handle typical license errors."""
@functools.wraps(func)
def wrapper(*args, **kwargs):
return _common_license_errors(*args) or func(*args, **kwargs)
return wrapper
def _common_license_errors(response):
"""Return an exception based on license status. None if no error."""
if response.status != power_pb2.STATUS_LICENSE_ERROR:
return None
return common_license_errors(response)
@handle_common_header_errors
@handle_lease_use_result_errors
@_handle_license_errors
@handle_unset_status_error(unset='STATUS_UNKNOWN', statustype=power_pb2)
def _power_command_error_from_response(response):
"""Return a custom exception based on response, None if no error."""
return error_factory(response, response.status,
status_to_string=power_pb2.PowerCommandStatus.Name,
status_to_error=_STATUS_TO_ERROR)
@handle_common_header_errors
@handle_unset_status_error(unset='STATUS_UNKNOWN', statustype=power_pb2)
def _power_feedback_error_from_response(response):
return None
_STATUS_TO_ERROR = collections.defaultdict(lambda: (ResponseError, None))
_STATUS_TO_ERROR.update({
power_pb2.STATUS_SUCCESS: (None, None),
power_pb2.STATUS_IN_PROGRESS: (None, None),
power_pb2.STATUS_SHORE_POWER_CONNECTED:
(ShorePowerConnectedError, ShorePowerConnectedError.__doc__),
power_pb2.STATUS_BATTERY_MISSING: (BatteryMissingError, BatteryMissingError.__doc__),
power_pb2.STATUS_COMMAND_IN_PROGRESS: (CommandInProgressError, CommandInProgressError.__doc__),
power_pb2.STATUS_ESTOPPED: (EstoppedError, EstoppedError.__doc__),
power_pb2.STATUS_FAULTED: (FaultedError, FaultedError.__doc__),
power_pb2.STATUS_INTERNAL_ERROR: (InternalServerError, InternalServerError.__doc__),
power_pb2.STATUS_LICENSE_ERROR: (LicenseError, LicenseError.__doc__),
power_pb2.STATUS_OVERRIDDEN: (OverriddenError, OverriddenError.__doc__),
power_pb2.STATUS_KEEPALIVE_MOTORS_OFF:
(KeepaliveMotorsOffError, KeepaliveMotorsOffError.__doc__),
})
def _power_status_from_response(response):
return response.status
@handle_common_header_errors
@handle_lease_use_result_errors
@handle_unset_status_error(unset='STATUS_UNKNOWN', statustype=power_pb2.FanPowerCommandResponse)
def _fan_power_command_error_from_response(response):
return error_factory(response, response.status,
status_to_string=power_pb2.FanPowerCommandResponse.Status.Name,
status_to_error=_FAN_STATUS_TO_ERROR)
_FAN_STATUS_TO_ERROR = collections.defaultdict(lambda: (ResponseError, None))
_FAN_STATUS_TO_ERROR.update({
power_pb2.FanPowerCommandResponse.STATUS_OK: (None, None),
power_pb2.FanPowerCommandResponse.STATUS_TEMPERATURE_TOO_HIGH:
(FanControlTemperatureError, FanControlTemperatureError.__doc__),
})
@handle_common_header_errors
@handle_unset_status_error(unset='STATUS_UNKNOWN',
statustype=power_pb2.FanPowerCommandFeedbackResponse)
def _fan_power_feedback_error_from_response(response):
return None
@handle_common_header_errors
@handle_lease_use_result_errors
@handle_unset_status_error(unset='STATUS_UNKNOWN', statustype=power_pb2.ResetSafetyStopResponse)
def _reset_safety_stop_error_from_response(response):
return error_factory(response, response.status,
status_to_string=power_pb2.ResetSafetyStopResponse.Status.Name,
status_to_error=_RESET_SAFETY_STOP_STATUS_TO_ERROR)
_RESET_SAFETY_STOP_STATUS_TO_ERROR = collections.defaultdict(lambda: (ResponseError, None))
_RESET_SAFETY_STOP_STATUS_TO_ERROR.update({
power_pb2.ResetSafetyStopResponse.STATUS_OK: (None, None),
power_pb2.ResetSafetyStopResponse.STATUS_INCOMPATIBLE_HARDWARE_ERROR:
(SafetyStopIncompatibleHardwareError, SafetyStopIncompatibleHardwareError.__doc__),
power_pb2.ResetSafetyStopResponse.STATUS_FAILED:
(SafetyStopFailedError, SafetyStopFailedError.__doc__),
power_pb2.ResetSafetyStopResponse.STATUS_UNKNOWN_STOP_TYPE:
(SafetyStopUnknownStopTypeError, SafetyStopUnknownStopTypeError.__doc__)
})
[docs]@deprecated(reason='Replaced by the less ambiguous safe_power_off_motors function.',
version='3.0.0', action="ignore")
def safe_power_off(command_client, state_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Safely power off motors. See safe_power_off_motors()."""
safe_power_off_motors(command_client, state_client, timeout_sec, update_frequency, **kwargs)
[docs]def safe_power_off_motors(command_client, state_client, timeout_sec=30, update_frequency=1.0,
**kwargs):
"""Power off robot motors safely. This function blocks until robot safely powers off. This
means the robot will attempt to sit before powering motors off.
Args:
command_client (RobotCommandClient): client for calling RobotCommandService safe power off.
state_client (RobotStateClient): client for monitoring power state.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
RobotCommandResponseError: Something went wrong with the safe power off.
"""
start_time = time.time()
end_time = start_time + timeout_sec
update_time = 1.0 / update_frequency
full_body_command = full_body_command_pb2.FullBodyCommand.Request(
safe_power_off_request=basic_command_pb2.SafePowerOffCommand.Request())
command = robot_command_pb2.RobotCommand(full_body_command=full_body_command)
command_client.robot_command(command=command, **kwargs)
while time.time() < end_time:
time_until_timeout = end_time - time.time()
start_call_time = time.time()
future = state_client.get_robot_state_async(**kwargs)
try:
response = future.result(timeout=time_until_timeout)
if response.power_state.motor_power_state == robot_state_pb2.PowerState.STATE_OFF:
return
except TimeoutError:
raise CommandTimedOutError
call_time = time.time() - start_call_time
sleep_time = max(0.0, update_time - call_time)
time.sleep(sleep_time)
raise CommandTimedOutError
[docs]@deprecated(reason='Replaced by the less ambiguous power_on_motors function.', version='2.3.4',
action="ignore")
def power_on(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power on robot motors. See power_on_motors()."""
power_on_motors(power_client, timeout_sec, update_frequency, **kwargs)
[docs]@deprecated(reason='Replaced by the less ambiguous power_off_motors function.', version='2.3.4',
action="ignore")
def power_off(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power off the robot motors. See power_off_motors()."""
power_off_motors(power_client, timeout_sec, update_frequency, **kwargs)
[docs]def power_on_motors(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power on robot motors. This function blocks until the command returns success.
Args:
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
PowerResponseError: Something went wrong during the power off sequence.
"""
request = power_pb2.PowerCommandRequest.REQUEST_ON_MOTORS
_power_command(power_client, request, timeout_sec, update_frequency, **kwargs)
[docs]def power_off_motors(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power off the robot motors.
Args:
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
PowerResponseError: Something went wrong during the power off sequence.
"""
request = power_pb2.PowerCommandRequest.REQUEST_OFF_MOTORS
_power_command(power_client, request, timeout_sec, update_frequency, **kwargs)
[docs]def safe_power_off_robot(command_client, state_client, power_client, timeout_sec=30,
update_frequency=1.0, **kwargs):
"""Power off the robot motors and then the robot computers safely. This function blocks until
robot safely powers off. This means the robot will attempt to sit before powering motors off.
Args:
command_client (RobotCommandClient): client for calling RobotCommandService safe power off.
state_client (RobotStateClient): client for monitoring power state.
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
RobotCommandResponseError: Something went wrong with the safe power off.
"""
end_time = time.time() + timeout_sec
safe_power_off_motors(command_client, state_client, timeout_sec=end_time - time.time(),
update_frequency=update_frequency, **kwargs)
power_off_robot(power_client, timeout_sec=end_time - time.time(),
update_frequency=update_frequency, **kwargs)
[docs]def power_off_robot(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Fully power off the robot. Powering off the robot will stop API comms.
Args:
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
PowerResponseError: Something went wrong during the power off sequence.
"""
request = power_pb2.PowerCommandRequest.REQUEST_OFF_ROBOT
_power_command(power_client, request, timeout_sec, update_frequency, expect_grpc_timeout=True,
**kwargs)
[docs]def safe_power_cycle_robot(command_client, state_client, power_client, timeout_sec=30,
update_frequency=1.0, **kwargs):
"""Power cycle the robot safely. This function blocks until robot safely powers off. The robot
will attempt to sit before powering cycling.
Args:
command_client (RobotCommandClient): client for calling RobotCommandService safe power off.
state_client (RobotStateClient): client for monitoring power state.
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
RobotCommandResponseError: Something went wrong with the safe power off.
"""
end_time = time.time() + timeout_sec
safe_power_off_motors(command_client, state_client, timeout_sec=end_time - time.time(),
update_frequency=update_frequency, **kwargs)
power_cycle_robot(power_client, timeout_sec=end_time - time.time(),
update_frequency=update_frequency, **kwargs)
[docs]def power_cycle_robot(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power cycle the robot. Power cycling the robot will stop API comms.
Args:
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
PowerResponseError: Something went wrong during the power off sequence.
"""
request = power_pb2.PowerCommandRequest.REQUEST_CYCLE_ROBOT
_power_command(power_client, request, timeout_sec, update_frequency, expect_grpc_timeout=True,
**kwargs)
[docs]def power_off_payload_ports(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power off the robot payload ports.
Args:
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
PowerResponseError: Something went wrong during the power off sequence.
"""
request = power_pb2.PowerCommandRequest.REQUEST_OFF_PAYLOAD_PORTS
_power_command(power_client, request, timeout_sec, update_frequency, **kwargs)
[docs]def power_on_payload_ports(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power on the robot payload ports.
Args:
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
PowerResponseError: Something went wrong during the power off sequence.
"""
request = power_pb2.PowerCommandRequest.REQUEST_ON_PAYLOAD_PORTS
_power_command(power_client, request, timeout_sec, update_frequency, **kwargs)
[docs]def power_off_wifi_radio(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power off the robot Wi-Fi radio.
Args:
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
PowerResponseError: Something went wrong during the power off sequence.
"""
request = power_pb2.PowerCommandRequest.REQUEST_OFF_WIFI_RADIO
_power_command(power_client, request, timeout_sec, update_frequency, **kwargs)
[docs]def power_on_wifi_radio(power_client, timeout_sec=30, update_frequency=1.0, **kwargs):
"""Power off the robot Wi-Fi radio.
Args:
power_client (bosdyn.api.PowerClient): client for calling power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
Raises:
RpcError: Problem communicating with the robot.
power.CommandTimedOutError: Did not power off within timeout_sec
PowerResponseError: Something went wrong during the power off sequence.
"""
request = power_pb2.PowerCommandRequest.REQUEST_ON_WIFI_RADIO
_power_command(power_client, request, timeout_sec, update_frequency, **kwargs)
def _power_command(power_client, request, timeout_sec=30, update_frequency=1.0,
expect_grpc_timeout=False, **kwargs):
"""Helper function to issue command to power client.
Args:
power_client (bosdyn.api.PowerClient): Client for calling power service.
request (bosdyn.api.PowerCommandRequest): Request to make to power service.
timeout_sec (float): Max time this function will block for.
update_frequency (float): The frequency with which the robot should check if the command
has succeeded.
expect_timeout (bool): Expect API comms to drop on a success.
"""
start_time = time.time()
end_time = start_time + timeout_sec
update_time = 1.0 / update_frequency
try:
response = power_client.power_command(request, **kwargs)
except TimedOutError as e:
if expect_grpc_timeout:
return
else:
raise
if response.status == power_pb2.STATUS_SUCCESS:
return # Command succeeded immediately.
power_command_id = response.power_command_id
while time.time() < end_time:
time_until_timeout = end_time - time.time()
start_call_time = time.time()
future = power_client.power_command_feedback_async(power_command_id, **kwargs)
try:
response = future.result(timeout=time_until_timeout)
if response == power_pb2.STATUS_SUCCESS:
return
if response != power_pb2.STATUS_IN_PROGRESS:
error_type, message = _STATUS_TO_ERROR[response]
exc = error_type(response=None, error_message=message)
raise exc
except TimedOutError as e:
if expect_grpc_timeout:
return
else:
raise
except TimeoutError:
raise CommandTimedOutError
call_time = time.time() - start_call_time
sleep_time = max(0.0, update_time - call_time)
time.sleep(sleep_time)
raise CommandTimedOutError
[docs]def is_powered_on(state_client, **kwargs):
"""Returns true if robot is powered on, false otherwise.
Raises:
RpcError: Problem communicating with the robot
"""
response = state_client.get_robot_state(**kwargs)
return response.power_state.motor_power_state == robot_state_pb2.PowerState.STATE_ON