tester_node.py

#!/usr/bin/env python3
 
# Copyright (c) 2018 Intel Corporation.
# Copyright (c) 2020 Samsung Research Russia
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
 
import argparse
from enum import Enum
import math
import sys
import time
from typing import Optional
 
from action_msgs.msg import GoalStatus
from geometry_msgs.msg import Pose, PoseStamped, PoseWithCovarianceStamped
from lifecycle_msgs.srv import GetState
from nav2_msgs.action import NavigateToPose
from nav2_msgs.msg import SpeedLimit
from nav2_msgs.srv import ManageLifecycleNodes
from nav_msgs.msg import OccupancyGrid, Path
import rclpy
from rclpy.action import ActionClient
from rclpy.client import Client
from rclpy.node import Node
from rclpy.qos import QoSDurabilityPolicy, QoSHistoryPolicy, QoSProfile, QoSReliabilityPolicy
from sensor_msgs.msg import PointCloud2
 
 
class TestType(Enum):
    KEEPOUT = 0
    SPEED = 1
 
 
class FilterMask:
 
    def __init__(self, filter_mask: OccupancyGrid):
        self.filter_mask: OccupancyGrid = filter_mask
 
    # Converts world coordinates into filter mask map coordinate.
    # Returns filter mask map coordinates or (-1, -1) in case
    # if world coordinates are out of mask bounds.
    def worldToMap(self, wx: float, wy: float) -> tuple[int, int]:
        origin_x = self.filter_mask.info.origin.position.x
        origin_y = self.filter_mask.info.origin.position.y
        size_x = self.filter_mask.info.width
        size_y = self.filter_mask.info.height
        resolution = self.filter_mask.info.resolution
 
        if wx < origin_x or wy < origin_y:
            return -1, -1
 
        mx = int((wx - origin_x) / resolution)
        my = int((wy - origin_y) / resolution)
 
        if mx < size_x and my < size_y:
            return mx, my
 
        return -1, -1
 
    # Gets filter_mask[mx, my] value
    def getValue(self, mx: int, my: int):  # type: ignore[no-untyped-def]
        size_x = self.filter_mask.info.width
        return self.filter_mask.data[mx + my * size_x]
 
 
class NavTester(Node):
    def __init__(
        self,
        test_type: TestType,
        initial_pose: Pose,
        goal_pose: Pose,
        namespace: str = '',
    ):
        super().__init__(node_name='nav2_tester', namespace=namespace)
        self.initial_pose_pubinitial_pose_pub = self.create_publisher(
            PoseWithCovarianceStamped, 'initialpose', 10
        )
        self.goal_pubgoal_pub = self.create_publisher(PoseStamped, 'goal_pose', 10)
 
        transient_local_qos = QoSProfile(
            durability=QoSDurabilityPolicy.TRANSIENT_LOCAL,
            reliability=QoSReliabilityPolicy.RELIABLE,
            history=QoSHistoryPolicy.KEEP_LAST,
            depth=1,
        )
 
        volatile_qos = QoSProfile(
            durability=QoSDurabilityPolicy.VOLATILE,
            reliability=QoSReliabilityPolicy.RELIABLE,
            history=QoSHistoryPolicy.KEEP_LAST,
            depth=1,
        )
 
        self.model_pose_submodel_pose_sub = self.create_subscription(
            PoseWithCovarianceStamped,
            'amcl_pose',
            self.poseCallbackposeCallbackposeCallback,
            transient_local_qos,
        )
        self.clearing_ep_subclearing_ep_sub = self.create_subscription(
            PointCloud2,
            'local_costmap/clearing_endpoints',
            self.clearingEndpointsCallbackclearingEndpointsCallback,
            transient_local_qos,
        )
        self.test_typetest_type = test_type
        self.filter_test_resultfilter_test_result = True
        self.clearing_endpoints_receivedclearing_endpoints_received = False
        self.voxel_marked_receivedvoxel_marked_received = False
        self.voxel_unknown_receivedvoxel_unknown_received = False
        self.cost_cloud_receivedcost_cloud_received = False
 
        if self.test_typetest_type == TestType.KEEPOUT:
            self.plan_subplan_sub = self.create_subscription(
                Path, 'plan', self.planCallbackplanCallback, volatile_qos
            )
            self.voxel_marked_subvoxel_marked_sub = self.create_subscription(
                PointCloud2, 'voxel_marked_cloud', self.voxelMarkedCallbackvoxelMarkedCallback, 1
            )
            self.voxel_unknown_subvoxel_unknown_sub = self.create_subscription(
                PointCloud2, 'voxel_unknown_cloud', self.voxelUnknownCallbackvoxelUnknownCallback, 1
            )
            self.cost_cloud_subcost_cloud_sub = self.create_subscription(
                PointCloud2, 'cost_cloud', self.dwbCostCloudCallbackdwbCostCloudCallback, 1
            )
        elif self.test_typetest_type == TestType.SPEED:
            self.speed_itspeed_it = 0
            # Expected chain of speed limits
            self.limitslimits = [50.0, 0.0]
            # Permissive array: all received speed limits must match to 'limits' from above
            self.limit_passedlimit_passed = [False, False]
            self.plan_subplan_sub = self.create_subscription(
                SpeedLimit, 'speed_limit', self.speedLimitCallbackspeedLimitCallback, volatile_qos
            )
 
        self.mask_receivedmask_received = False
        self.mask_submask_sub = self.create_subscription(
            OccupancyGrid, 'filter_mask', self.maskCallbackmaskCallback, transient_local_qos
        )
 
        self.initial_pose_receivedinitial_pose_received = False
        self.initial_poseinitial_pose = initial_pose
        self.goal_posegoal_pose = goal_pose
        self.action_client: ActionClient[
            NavigateToPose.Goal, NavigateToPose.Result, NavigateToPose.Feedback] = \
            ActionClient(self, NavigateToPose, 'navigate_to_pose')
 
    def info_msg(self, msg: str) -> None:
        self.get_logger().info('\033[1;37;44m' + msg + '\033[0m')
 
    def warn_msg(self, msg: str) -> None:
        self.get_logger().warning('\033[1;37;43m' + msg + '\033[0m')
 
    def error_msg(self, msg: str) -> None:
        self.get_logger().error('\033[1;37;41m' + msg + '\033[0m')
 
    def setInitialPose(self) -> None:
        msg = PoseWithCovarianceStamped()
        msg.pose.pose = self.initial_poseinitial_pose
        msg.header.frame_id = 'map'
        self.info_msginfo_msginfo_msg('Publishing Initial Pose')
        self.initial_pose_pubinitial_pose_pub.publish(msg)
        self.currentPosecurrentPose = self.initial_poseinitial_pose
 
    def getStampedPoseMsg(self, pose: Pose) -> PoseStamped:
        msg = PoseStamped()
        msg.header.frame_id = 'map'
        msg.pose = pose
        return msg
 
    def publishGoalPose(self, goal_pose: Optional[Pose] = None) -> None:
        self.goal_posegoal_pose = goal_pose if goal_pose is not None else self.goal_posegoal_pose
        self.goal_pubgoal_pub.publish(self.getStampedPoseMsggetStampedPoseMsggetStampedPoseMsg(self.goal_posegoal_pose))
 
    def runNavigateAction(self, goal_pose: Optional[Pose] = None) -> bool:
        # Sends a `NavToPose` action request and waits for completion
        self.info_msginfo_msginfo_msg("Waiting for 'NavigateToPose' action server")
        while not self.action_client.wait_for_server(timeout_sec=1.0):
            self.info_msginfo_msginfo_msg("'NavigateToPose' action server not available, waiting...")
 
        self.goal_posegoal_pose = goal_pose if goal_pose is not None else self.goal_posegoal_pose
        goal_msg = NavigateToPose.Goal()
        goal_msg.pose = self.getStampedPoseMsggetStampedPoseMsggetStampedPoseMsg(self.goal_posegoal_pose)
 
        self.info_msginfo_msginfo_msg('Sending goal request...')
        send_goal_future = self.action_client.send_goal_async(goal_msg)
 
        rclpy.spin_until_future_complete(self, send_goal_future)
        goal_handle = send_goal_future.result()
 
        if not goal_handle or not goal_handle.accepted:
            self.error_msgerror_msgerror_msg('Goal rejected')
            return False
 
        self.info_msginfo_msginfo_msg('Goal accepted')
        get_result_future = goal_handle.get_result_async()
 
        self.info_msginfo_msginfo_msg("Waiting for 'NavigateToPose' action to complete")
        rclpy.spin_until_future_complete(self, get_result_future)
        status = get_result_future.result().status  # type: ignore[union-attr]
        if status != GoalStatus.STATUS_SUCCEEDED:
            self.info_msginfo_msginfo_msg(f'Goal failed with status code: {status}')
            return False
 
        self.info_msginfo_msginfo_msg('Goal succeeded!')
        return True
 
    def isInKeepout(self, x: float, y: float) -> bool:
        mx, my = self.filter_maskfilter_mask.worldToMap(x, y)
        if mx == -1 and my == -1:  # Out of mask's area
            return False
        if self.filter_maskfilter_mask.getValue(mx, my) == 100:  # Occupied
            return True
        return False
 
    # Checks that (x, y) position does not belong to a keepout zone.
    def checkKeepout(self, x: float, y: float) -> bool:
        if not self.mask_receivedmask_received:
            self.warn_msgwarn_msgwarn_msg('Filter mask was not received')
        elif self.isInKeepoutisInKeepout(x, y):
            self.filter_test_resultfilter_test_result = False
            self.error_msgerror_msgerror_msg(f'Pose ({x}, {y}) belongs to keepout zone')
            return False
        return True
 
    # Checks that currently received speed_limit is equal to the it-th item
    # of expected speed 'limits' array.
    # If so, sets it-th item of permissive array 'limit_passed' to be true.
    # Otherwise it will be remained to be false.
    # Also verifies that speed limit messages received no more than N-times
    # (where N - is the length of 'limits' array),
    # otherwise sets overall 'filter_test_result' to be false.
    def checkSpeed(self, it: int, speed_limit: float) -> None:
        if it >= len(self.limitslimits):
            self.error_msgerror_msgerror_msg('Got excess speed limit')
            self.filter_test_resultfilter_test_result = False
            return
        if speed_limit == self.limitslimits[it]:
            self.limit_passedlimit_passed[it] = True
        else:
            self.error_msgerror_msgerror_msg(
                'Incorrect speed limit received: '
                + str(speed_limit)
                + ', but should be: '
                + str(self.limitslimits[it])
            )
 
    def poseCallback(self, msg: PoseWithCovarianceStamped) -> None:
        self.info_msginfo_msginfo_msg('Received amcl_pose')
        self.current_posecurrent_pose = msg.pose.pose
        self.initial_pose_receivedinitial_pose_received = True
        if self.test_typetest_type == TestType.KEEPOUT:
            if not self.checkKeepoutcheckKeepout(
                msg.pose.pose.position.x, msg.pose.pose.position.y
            ):
                self.error_msgerror_msgerror_msg('Robot goes into keepout zone')
 
    def planCallback(self, msg: Path) -> None:
        self.info_msginfo_msginfo_msg('Received plan')
        for pose in msg.poses:
            if not self.checkKeepoutcheckKeepout(pose.pose.position.x, pose.pose.position.y):
                self.error_msgerror_msgerror_msg('Path plan intersects with keepout zone')
                return
 
    def clearingEndpointsCallback(self, msg: PointCloud2) -> None:
        if len(msg.data) > 0:
            self.clearing_endpoints_receivedclearing_endpoints_received = True
 
    def voxelMarkedCallback(self, msg: PointCloud2) -> None:
        if len(msg.data) > 0:
            self.voxel_marked_receivedvoxel_marked_received = True
 
    def voxelUnknownCallback(self, msg: PointCloud2) -> None:
        if len(msg.data) > 0:
            self.voxel_unknown_receivedvoxel_unknown_received = True
 
    def dwbCostCloudCallback(self, msg: PointCloud2) -> None:
        self.info_msginfo_msginfo_msg('Received cost_cloud points')
        if len(msg.data) > 0:
            self.cost_cloud_receivedcost_cloud_received = True
 
    def speedLimitCallback(self, msg: SpeedLimit) -> None:
        self.info_msginfo_msginfo_msg(f'Received speed limit: {msg.speed_limit}')
        self.checkSpeedcheckSpeed(self.speed_itspeed_it, msg.speed_limit)
        self.speed_itspeed_it += 1
 
    def maskCallback(self, msg: OccupancyGrid) -> None:
        self.info_msginfo_msginfo_msg('Received filter mask')
        self.filter_maskfilter_mask = FilterMask(msg)
        self.mask_receivedmask_received = True
 
    def wait_for_filter_mask(self, timeout: float) -> bool:
        start_time = time.time()
 
        while not self.mask_receivedmask_received:
            self.info_msginfo_msginfo_msg('Waiting for filter mask to be received ...')
            rclpy.spin_once(self, timeout_sec=1)
            if (time.time() - start_time) > timeout:
                self.error_msgerror_msgerror_msg('Time out to waiting filter mask')
                return False
        return True
 
    def wait_for_pointcloud_subscribers(self, timeout: float) -> bool:
        start_time = time.time()
        while (
            not self.voxel_unknown_receivedvoxel_unknown_received
            or not self.voxel_marked_receivedvoxel_marked_received
            or not self.clearing_endpoints_receivedclearing_endpoints_received
        ):
            self.info_msginfo_msginfo_msg(
                'Waiting for voxel_marked_cloud/voxel_unknown_cloud/\
                clearing_endpoints msg to be received ...'
            )
            rclpy.spin_once(self, timeout_sec=1)
            if (time.time() - start_time) > timeout:
                self.error_msgerror_msgerror_msg(
                    'Time out to waiting for voxel_marked_cloud/voxel_unknown_cloud/\
                    clearing_endpoints msgs'
                )
                return False
        return True
 
    def reachesGoal(self, timeout: float, distance: float) -> bool:
        goalReached = False
        start_time = time.time()
 
        while not goalReached:
            rclpy.spin_once(self, timeout_sec=1)
            if self.distanceFromGoaldistanceFromGoaldistanceFromGoal() < distance:
                goalReached = True
                self.info_msginfo_msginfo_msg('*** GOAL REACHED ***')
                return True
            elif timeout is not None:
                if (time.time() - start_time) > timeout:
                    self.error_msgerror_msgerror_msg('Robot timed out reaching its goal!')
                    return False
 
        self.info_msginfo_msginfo_msg('Robot reached its goal!')
        return True
 
    def distanceFromGoal(self) -> float:
        d_x = self.current_posecurrent_pose.position.x - self.goal_posegoal_pose.position.x
        d_y = self.current_posecurrent_pose.position.y - self.goal_posegoal_pose.position.y
        distance = math.sqrt(d_x * d_x + d_y * d_y)
        self.info_msginfo_msginfo_msg(f'Distance from goal is: {distance}')
        return distance
 
    def wait_for_node_active(self, node_name: str) -> None:
        # Waits for the node within the tester namespace to become active
        self.info_msginfo_msginfo_msg(f'Waiting for {node_name} to become active')
        node_service = f'{node_name}/get_state'
        state_client: Client[GetState.Request, GetState.Response] \
            = self.create_client(GetState, node_service)
        while not state_client.wait_for_service(timeout_sec=1.0):
            self.info_msginfo_msginfo_msg(f'{node_service} service not available, waiting...')
        req = GetState.Request()  # empty request
        state = 'UNKNOWN'
        while state != 'active':
            self.info_msginfo_msginfo_msg(f'Getting {node_name} state...')
            future = state_client.call_async(req)
            rclpy.spin_until_future_complete(self, future)
            if future.result() is not None:
                state = future.result().current_state.label  # type: ignore[union-attr]
                self.info_msginfo_msginfo_msg(f'Result of get_state: {state}')
            else:
                self.error_msgerror_msgerror_msg(
                    f'Exception while calling service: {future.exception()!r}'
                )
            time.sleep(5)
 
    def shutdown(self) -> None:
        self.info_msginfo_msginfo_msg('Shutting down')
        self.action_client.destroy()
 
        transition_service = 'lifecycle_manager_navigation/manage_nodes'
        mgr_client: Client[ManageLifecycleNodes.Request, ManageLifecycleNodes.Response] \
            = self.create_client(ManageLifecycleNodes, transition_service)
        while not mgr_client.wait_for_service(timeout_sec=1.0):
            self.info_msginfo_msginfo_msg(f'{transition_service} service not available, waiting...')
 
        req = ManageLifecycleNodes.Request()
        req.command = ManageLifecycleNodes.Request().SHUTDOWN
        future = mgr_client.call_async(req)
        try:
            self.info_msginfo_msginfo_msg('Shutting down navigation lifecycle manager...')
            rclpy.spin_until_future_complete(self, future)
            future.result()
            self.info_msginfo_msginfo_msg('Shutting down navigation lifecycle manager complete.')
        except Exception as e:  # noqa: B902
            self.error_msgerror_msgerror_msg(f'Service call failed {e!r}')
        transition_service = 'lifecycle_manager_localization/manage_nodes'
        mgr_client = self.create_client(ManageLifecycleNodes, transition_service)
        while not mgr_client.wait_for_service(timeout_sec=1.0):
            self.info_msginfo_msginfo_msg(f'{transition_service} service not available, waiting...')
 
        req = ManageLifecycleNodes.Request()
        req.command = ManageLifecycleNodes.Request().SHUTDOWN
        future = mgr_client.call_async(req)
        try:
            self.info_msginfo_msginfo_msg('Shutting down localization lifecycle manager...')
            rclpy.spin_until_future_complete(self, future)
            future.result()
            self.info_msginfo_msginfo_msg('Shutting down localization lifecycle manager complete')
        except Exception as e:  # noqa: B902
            self.error_msgerror_msgerror_msg(f'Service call failed {e!r}')
 
    def wait_for_initial_pose(self) -> None:
        self.initial_pose_receivedinitial_pose_received = False
        while not self.initial_pose_receivedinitial_pose_received:
            self.info_msginfo_msginfo_msg('Setting initial pose')
            self.setInitialPosesetInitialPosesetInitialPose()
            self.info_msginfo_msginfo_msg('Waiting for amcl_pose to be received')
            rclpy.spin_once(self, timeout_sec=1)
 
 
def test_RobotMovesToGoal(robot_tester: NavTester) -> bool:
    robot_tester.info_msg('Setting goal pose')
    robot_tester.publishGoalPose()
    robot_tester.info_msg('Waiting 60 seconds for robot to reach goal')
    return robot_tester.reachesGoal(timeout=60, distance=0.5)
 
 
# Tests that all received speed limits are correct:
# If overall 'filter_test_result' is true
# checks that all items in 'limit_passed' permissive array are also true.
# In other words, it verifies that all speed limits are received
# exactly (by count and values) as expected by 'limits' array.
def test_SpeedLimitsAllCorrect(robot_tester: NavTester) -> bool:
    if not robot_tester.filter_test_result:
        return False
    for passed in robot_tester.limit_passed:
        if not passed:
            robot_tester.error_msg('Did not meet one of the speed limit')
            return False
    return True
 
 
def run_all_tests(robot_tester: NavTester) -> bool:
    # set transforms to use_sim_time
    result = True
    if result:
        robot_tester.wait_for_node_active('amcl')
        robot_tester.wait_for_initial_pose()
        robot_tester.wait_for_node_active('bt_navigator')
        result = robot_tester.wait_for_filter_mask(10)
    if result:
        result = robot_tester.runNavigateAction()
 
    if robot_tester.test_type == TestType.KEEPOUT:
        result = result and robot_tester.wait_for_pointcloud_subscribers(10)
 
    if result:
        result = test_RobotMovesToGoal(robot_tester)
 
    if result:
        if robot_tester.test_type == TestType.KEEPOUT:
            result = robot_tester.filter_test_result
            result = result and robot_tester.cost_cloud_received
        elif robot_tester.test_type == TestType.SPEED:
            result = test_SpeedLimitsAllCorrect(robot_tester)
 
    # Add more tests here if desired
 
    if result:
        robot_tester.info_msg('Test PASSED')
    else:
        robot_tester.error_msg('Test FAILED')
 
    return result
 
 
def fwd_pose(x: float = 0.0, y: float = 0.0, z: float = 0.01) -> Pose:
    initial_pose = Pose()
    initial_pose.position.x = x
    initial_pose.position.y = y
    initial_pose.position.z = z
    initial_pose.orientation.x = 0.0
    initial_pose.orientation.y = 0.0
    initial_pose.orientation.z = 0.0
    initial_pose.orientation.w = 1.0
    return initial_pose
 
 
def get_tester(args: argparse.Namespace) -> NavTester:
 
    # Requested tester for one robot
    type_str = args.type
    init_x, init_y, final_x, final_y = args.robot[0]
    test_type = TestType.KEEPOUT  # Default value
    if type_str == 'speed':
        test_type = TestType.SPEED
 
    tester = NavTester(
        test_type,
        initial_pose=fwd_pose(float(init_x), float(init_y)),
        goal_pose=fwd_pose(float(final_x), float(final_y)),
    )
    tester.info_msg(
        'Starting tester, robot going from '
        + init_x
        + ', '
        + init_y
        + ' to '
        + final_x
        + ', '
        + final_y
        + '.'
    )
    return tester
 
 
def main(argv: list[str] = sys.argv[1:]):  # type: ignore[no-untyped-def]
    # The robot(s) positions from the input arguments
    parser = argparse.ArgumentParser(
        description='System-level costmap filters tester node'
    )
    parser.add_argument(
        '-t',
        '--type',
        type=str,
        action='store',
        dest='type',
        help='Type of costmap filter being tested.',
    )
    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument(
        '-r',
        '--robot',
        action='append',
        nargs=4,
        metavar=('init_x', 'init_y', 'final_x', 'final_y'),
        help='The robot starting and final positions.',
    )
 
    args, unknown = parser.parse_known_args()
 
    rclpy.init()
 
    # Create tester for the robot
    tester = get_tester(args)
 
    # wait a few seconds to make sure entire stacks are up
    time.sleep(10)
 
    passed = run_all_tests(tester)
 
    # stop and shutdown the nav stack to exit cleanly
    tester.shutdown()
    tester.info_msg('Done Shutting Down.')
 
    if not passed:
        tester.info_msg('Exiting failed')
        exit(1)
    else:
        tester.info_msg('Exiting passed')
        exit(0)
 
 
if __name__ == '__main__':
    main()