feat: improve AGV navigation performance

This commit is contained in:
clhchan
2026-05-26 11:15:41 +08:00
parent 39cb5cf26f
commit f966a123e4
24 changed files with 2208 additions and 62 deletions
@@ -0,0 +1,388 @@
#!/usr/bin/env python3
"""Yaw-only final pose refinement helper for AGV Pro."""
import math
import os
import shlex
import rclpy
from action_msgs.msg import GoalStatus, GoalStatusArray
from geometry_msgs.msg import PoseStamped, Twist
from rclpy.duration import Duration
from rclpy.node import Node
from rclpy.parameter import Parameter
from std_msgs.msg import String
from tf2_ros import Buffer, TransformException, TransformListener
class FinalPoseRefiner(Node):
"""Refine only the final map->base_footprint yaw with direct low-speed cmd_vel."""
def __init__(self):
super().__init__('final_pose_refiner')
self.param_prefix = 'final_pose_refiner_'
self.start_param = f'{self.param_prefix}start'
self.cancel_param = f'{self.param_prefix}cancel'
self.auto_start_param = f'{self.param_prefix}auto_start_on_nav_success'
self.log_separator = '------------------------------------------------------------'
self.cmd_vel_topic = '/cmd_vel'
self.goal_topic = '/goal_pose'
self.action_goal_topic = '/final_pose_refiner/goal_pose'
self.nav_status_topic = '/navigate_to_pose/_action/status'
self.nav2_status_topic = '/navigate_to_pose_nav2/_action/status'
self.global_frame = 'map'
self.base_frame = 'base_footprint'
self._declare_param('status_topic', '/final_pose_refiner/status')
self.declare_parameter(self.start_param, False)
self.declare_parameter(self.cancel_param, False)
self.declare_parameter(self.auto_start_param, False)
self._declare_param('handoff_distance', 0.20)
self._declare_param('yaw_tolerance', 0.04)
self._declare_param('settle_time', 0.5)
self._declare_param('timeout', 20.0)
self._declare_param('k_yaw', 0.5)
self._declare_param('max_wz', 0.35)
self._declare_param('min_cmd_w', 0.006)
self.cmd_vel_pub = self.create_publisher(Twist, self.cmd_vel_topic, 10)
self.status_pub = self.create_publisher(String, self._param('status_topic'), 10)
self.goal_sub = self.create_subscription(
PoseStamped,
self.goal_topic,
self._on_goal,
10,
)
self.action_goal_sub = self.create_subscription(
PoseStamped,
self.action_goal_topic,
self._on_goal,
10,
)
self.nav_status_sub = self.create_subscription(
GoalStatusArray,
self.nav_status_topic,
self._on_nav_status,
10,
)
self.nav2_status_sub = self.create_subscription(
GoalStatusArray,
self.nav2_status_topic,
self._on_nav_status,
10,
)
self.tf_buffer = Buffer()
self.tf_listener = TransformListener(self.tf_buffer, self)
self.state = 'idle'
self.goal = None
self.target = None
self.waiting_for_nav_success = False
self.active_nav_goal_ids = set()
self.refined_nav_goal_ids = set()
self.start_time = None
self.settle_start_time = None
self.last_log_time = self.get_clock().now()
self.timer = self.create_timer(1.0 / 20.0, self.on_timer)
self.get_logger().info(
'final_pose_refiner ready in yaw-only mode. A navigation proxy may submit the '
f'target and set {self.start_param}:=true, or these may be provided manually.'
)
def _declare_param(self, name, value):
self.declare_parameter(f'{self.param_prefix}{name}', value)
def _param(self, name):
return self.get_parameter(f'{self.param_prefix}{name}').value
def _on_goal(self, msg):
if msg.header.frame_id and msg.header.frame_id != self.global_frame:
self.get_logger().warn(
f'Ignoring goal in frame "{msg.header.frame_id}". Expected "{self.global_frame}".'
)
return
q = msg.pose.orientation
target_yaw = self._yaw_from_quaternion(q.x, q.y, q.z, q.w)
self.target = (msg.pose.position.x, msg.pose.position.y, target_yaw)
self.waiting_for_nav_success = True
self.active_nav_goal_ids.clear()
self.get_logger().info(
f'Updated refine target from topic: x={msg.pose.position.x:.4f}, '
f'y={msg.pose.position.y:.4f}, yaw={math.degrees(target_yaw):.2f} deg'
)
def _on_nav_status(self, msg):
if not self.get_parameter(self.auto_start_param).value:
return
if self.state != 'idle' or self.target is None or not self.waiting_for_nav_success:
return
for status in msg.status_list:
goal_id = tuple(status.goal_info.goal_id.uuid)
if status.status in (GoalStatus.STATUS_ACCEPTED, GoalStatus.STATUS_EXECUTING):
self.active_nav_goal_ids.add(goal_id)
elif status.status == GoalStatus.STATUS_SUCCEEDED:
if (
goal_id in self.active_nav_goal_ids and
goal_id not in self.refined_nav_goal_ids
):
self.refined_nav_goal_ids.add(goal_id)
self.get_logger().info(
'Detected Nav2 goal succeeded; starting final yaw refinement.'
)
self._start_refine()
return
elif status.status in (GoalStatus.STATUS_CANCELED, GoalStatus.STATUS_ABORTED):
if goal_id in self.active_nav_goal_ids:
self.waiting_for_nav_success = False
self.active_nav_goal_ids.discard(goal_id)
def on_timer(self):
if self.get_parameter(self.cancel_param).value:
if self.state == 'running':
self._finish_refine('canceled')
else:
self._reset_cancel_refine()
return
if self.state == 'running':
self._run_refine_step()
return
if self.get_parameter(self.start_param).value:
self._start_refine()
def _start_refine(self):
self._reset_cancel_refine()
if self.target is None:
self.get_logger().warn(
'Cannot start final refinement: no /goal_pose has been received yet.'
)
self._publish_status('no_goal')
self._reset_start_refine()
return
pose = self._lookup_pose()
if pose is None:
self.get_logger().warn('Cannot start final refinement: TF is not available.')
self._publish_status('failed_tf')
self._reset_start_refine()
return
target = self.target
distance, yaw_error = self._calculate_error(pose, target)
handoff_distance = max(self._param('handoff_distance'), 0.0)
if distance > handoff_distance:
self.get_logger().warn(
f'Cannot start final refinement: distance={distance:.3f} m exceeds '
f'handoff_distance={handoff_distance:.3f} m.'
)
self._publish_status('handoff_distance_exceeded')
self._reset_start_refine()
return
self.goal = target
self.waiting_for_nav_success = False
self.start_time = self.get_clock().now()
self.settle_start_time = None
self.last_log_time = self.get_clock().now()
self.state = 'running'
self._publish_status('running')
self.get_logger().info(
f'\n{self.log_separator}\n'
'FINAL YAW REFINE START\n'
f'target=({target[0]:.4f}, {target[1]:.4f}, {math.degrees(target[2]):.2f} deg)\n'
f'initial_distance={distance:.3f} m, '
f'initial_yaw_error={math.degrees(yaw_error):+.2f} deg\n'
f'{self.log_separator}'
)
def _run_refine_step(self):
pose = self._lookup_pose()
if pose is None:
self._finish_refine('failed_tf', warn=True)
return
distance, yaw_error = self._calculate_error(pose, self.goal)
elapsed = (self.get_clock().now() - self.start_time).nanoseconds / 1e9
yaw_tolerance = max(self._param('yaw_tolerance'), 0.0)
settle_time = max(self._param('settle_time'), 0.0)
timeout = self._param('timeout')
if timeout > 0.0 and elapsed > timeout:
self._finish_refine('timeout', pose, distance, yaw_error, warn=True)
return
if abs(yaw_error) <= yaw_tolerance:
now = self.get_clock().now()
if self.settle_start_time is None:
self.settle_start_time = now
self._publish_stop()
elif (now - self.settle_start_time).nanoseconds / 1e9 >= settle_time:
self._finish_refine('succeeded', pose, distance, yaw_error)
return
else:
self._publish_stop()
self._log_progress(pose, distance, yaw_error, elapsed, Twist())
return
self.settle_start_time = None
cmd = self._make_yaw_command(yaw_error)
self.cmd_vel_pub.publish(cmd)
self._log_progress(pose, distance, yaw_error, elapsed, cmd)
def _make_yaw_command(self, yaw_error):
cmd = Twist()
max_wz = max(abs(self._param('max_wz')), 0.0)
cmd.angular.z = self._clip(self._param('k_yaw') * yaw_error, -max_wz, max_wz)
cmd.angular.z = self._apply_min_abs(cmd.angular.z, self._param('min_cmd_w'))
return cmd
def _finish_refine(self, status, pose=None, distance=None, yaw_error=None, warn=False):
self._stop_robot()
self._reset_start_refine()
self._reset_cancel_refine()
self.state = 'idle'
self.settle_start_time = None
self._publish_status(status)
if pose is not None and distance is not None and yaw_error is not None:
msg = (
f'\n{self.log_separator}\n'
f'FINAL YAW REFINE END: {status}\n'
f'distance={distance:.4f} m, '
f'yaw_error={math.degrees(yaw_error):+.2f} deg, '
f'pose=({pose[0]:.4f}, {pose[1]:.4f}, {math.degrees(pose[2]):.2f} deg)\n'
f'{self.log_separator}'
)
else:
msg = (
f'\n{self.log_separator}\n'
f'FINAL YAW REFINE END: {status}\n'
f'{self.log_separator}'
)
if warn:
self.get_logger().warn(msg)
else:
self.get_logger().info(msg)
def _lookup_pose(self):
try:
trans = self.tf_buffer.lookup_transform(
self.global_frame,
self.base_frame,
rclpy.time.Time(),
timeout=Duration(seconds=0.3),
)
except TransformException as exc:
self.get_logger().warn(f'TF lookup failed: {exc}')
return None
translation = trans.transform.translation
rotation = trans.transform.rotation
return (
translation.x,
translation.y,
self._yaw_from_quaternion(rotation.x, rotation.y, rotation.z, rotation.w),
)
def _calculate_error(self, pose, target):
x, y, yaw = pose
target_x, target_y, target_yaw = target
distance = math.hypot(target_x - x, target_y - y)
yaw_error = self._normalize_angle(target_yaw - yaw)
return distance, yaw_error
def _log_progress(self, pose, distance, yaw_error, elapsed, cmd):
now = self.get_clock().now()
if (now - self.last_log_time).nanoseconds < 1e9:
return
self.get_logger().info(
f'[FINAL YAW REFINE RUNNING] '
f'distance={distance:.3f} m, yaw_error={math.degrees(yaw_error):+.2f} deg, '
f'elapsed={elapsed:.1f} s, cmd_wz={cmd.angular.z:+.3f}, '
f'pose=({pose[0]:.3f}, {pose[1]:.3f}, {math.degrees(pose[2]):.1f} deg)'
)
self.last_log_time = now
def _publish_status(self, status):
msg = String()
msg.data = status
self.status_pub.publish(msg)
def _publish_stop(self):
try:
self.cmd_vel_pub.publish(Twist())
except Exception:
pass
def _stop_robot(self):
for _ in range(5):
self._publish_stop()
def _stop_robot_with_ros_cli(self):
topic = shlex.quote(self.cmd_vel_topic)
zero_twist = (
'"{linear: {x: 0.0, y: 0.0, z: 0.0}, '
'angular: {x: 0.0, y: 0.0, z: 0.0}}"'
)
os.system(
f'timeout 2s ros2 topic pub --once {topic} '
f'geometry_msgs/msg/Twist {zero_twist} >/dev/null 2>&1'
)
def _reset_start_refine(self):
self.set_parameters([
Parameter(self.start_param, Parameter.Type.BOOL, False),
])
def _reset_cancel_refine(self):
self.set_parameters([
Parameter(self.cancel_param, Parameter.Type.BOOL, False),
])
@staticmethod
def _clip(value, low, high):
return max(low, min(high, value))
@staticmethod
def _apply_min_abs(value, min_abs):
min_abs = max(abs(min_abs), 0.0)
if value == 0.0 or abs(value) >= min_abs:
return value
return math.copysign(min_abs, value)
@staticmethod
def _normalize_angle(angle):
return math.atan2(math.sin(angle), math.cos(angle))
@staticmethod
def _yaw_from_quaternion(x, y, z, w):
siny_cosp = 2.0 * (w * z + x * y)
cosy_cosp = 1.0 - 2.0 * (y * y + z * z)
return math.atan2(siny_cosp, cosy_cosp)
def main(args=None):
rclpy.init(args=args)
node = FinalPoseRefiner()
try:
rclpy.spin(node)
except KeyboardInterrupt:
pass
finally:
node._stop_robot()
node._stop_robot_with_ros_cli()
node.destroy_node()
if rclpy.ok():
rclpy.shutdown()
if __name__ == '__main__':
main()
@@ -0,0 +1,397 @@
#!/usr/bin/env python3
"""Transparent final-refinement proxy for Nav2 pose navigation actions."""
import threading
import time
from copy import deepcopy
import rclpy
from action_msgs.msg import GoalStatus
from geometry_msgs.msg import PoseStamped
from nav2_msgs.action import NavigateThroughPoses, NavigateToPose
from rcl_interfaces.msg import Parameter as ParameterMsg
from rcl_interfaces.msg import ParameterType, ParameterValue
from rcl_interfaces.srv import SetParameters
from rclpy.action import ActionClient, ActionServer, CancelResponse, GoalResponse
from rclpy.callback_groups import ReentrantCallbackGroup
from rclpy.executors import MultiThreadedExecutor
from rclpy.node import Node
from std_msgs.msg import String
class NavigateToPoseRefinerProxy(Node):
"""Forward pose-navigation actions and complete them after final yaw refinement."""
TERMINAL_REFINER_STATUSES = {
'succeeded',
'timeout',
'failed_tf',
'no_goal',
'handoff_distance_exceeded',
'canceled',
}
STARTLESS_REFINER_FAILURES = {
'failed_tf',
'no_goal',
'handoff_distance_exceeded',
}
def __init__(self):
super().__init__('navigate_to_pose_refiner_proxy')
self.public_goal_topic = '/goal_pose'
self.refiner_goal_topic = '/final_pose_refiner/goal_pose'
self.refiner_status_topic = '/final_pose_refiner/status'
self.refiner_param_service = '/final_pose_refiner/set_parameters'
self.start_param = 'final_pose_refiner_start'
self.cancel_param = 'final_pose_refiner_cancel'
self.declare_parameter('nav2_server_timeout_sec', 5.0)
self.declare_parameter('refiner_service_timeout_sec', 2.0)
self.declare_parameter('refiner_wait_timeout_sec', 25.0)
self.declare_parameter('require_refinement', True)
self.declare_parameter('debug_print', False)
self.callback_group = ReentrantCallbackGroup()
self.goal_pub = self.create_publisher(PoseStamped, self.refiner_goal_topic, 10)
self.refiner_status_sub = self.create_subscription(
String,
self.refiner_status_topic,
self._on_refiner_status,
10,
callback_group=self.callback_group,
)
self.refiner_param_client = self.create_client(
SetParameters,
self.refiner_param_service,
callback_group=self.callback_group,
)
self._refinement_lock = threading.Lock()
self._status_condition = threading.Condition()
self._status_sequence = 0
self._status_history = []
self.routes = []
self._add_route(
'NavigateToPose',
NavigateToPose,
'/navigate_to_pose',
'/navigate_to_pose_nav2',
lambda request: request.pose,
)
self._add_route(
'NavigateThroughPoses',
NavigateThroughPoses,
'/navigate_through_poses',
'/navigate_through_poses_nav2',
lambda request: request.poses[-1] if request.poses else None,
)
self.topic_nav_client = ActionClient(
self,
NavigateToPose,
'/navigate_to_pose',
callback_group=self.callback_group,
)
self.goal_topic_sub = self.create_subscription(
PoseStamped,
self.public_goal_topic,
self._on_goal_pose,
10,
callback_group=self.callback_group,
)
self.get_logger().info(
'Navigation refinement proxy ready for NavigateToPose, NavigateThroughPoses, '
'and /goal_pose; public tasks complete after final yaw refinement.'
)
def _add_route(self, label, action_type, public_name, nav2_name, final_pose_getter):
route = {
'label': label,
'action_type': action_type,
'public_name': public_name,
'nav2_name': nav2_name,
'final_pose_getter': final_pose_getter,
}
route['client'] = ActionClient(
self,
action_type,
nav2_name,
callback_group=self.callback_group,
)
route['server'] = ActionServer(
self,
action_type,
public_name,
execute_callback=lambda handle, current=route: self._execute_callback(current, handle),
goal_callback=lambda request, current=route: self._goal_callback(current, request),
cancel_callback=self._cancel_callback,
callback_group=self.callback_group,
)
self.routes.append(route)
self._debug(f'{label} route: {public_name} -> {nav2_name}')
def _goal_callback(self, route, goal_request):
if not route['client'].server_is_ready():
self.get_logger().warn(
f"{route['label']} Nav2 action server {route['nav2_name']} is not ready; "
'rejecting goal.'
)
return GoalResponse.REJECT
final_pose = route['final_pose_getter'](goal_request)
if final_pose is not None:
self._publish_refiner_goal(final_pose)
return GoalResponse.ACCEPT
@staticmethod
def _cancel_callback(_goal_handle):
return CancelResponse.ACCEPT
def _on_goal_pose(self, pose):
goal = NavigateToPose.Goal()
goal.pose = deepcopy(pose)
if not self.topic_nav_client.server_is_ready():
self.get_logger().error(
'Cannot forward /goal_pose: public NavigateToPose is unavailable.'
)
return
send_future = self.topic_nav_client.send_goal_async(goal)
send_future.add_done_callback(self._on_topic_goal_response)
def _on_topic_goal_response(self, future):
try:
goal_handle = future.result()
except Exception as exc:
self.get_logger().error(f'Failed to forward /goal_pose to NavigateToPose: {exc}')
return
if goal_handle is None or not goal_handle.accepted:
self.get_logger().error('/goal_pose navigation goal was rejected.')
return
result_future = goal_handle.get_result_async()
result_future.add_done_callback(self._on_topic_goal_result)
def _on_topic_goal_result(self, future):
try:
action_result = future.result()
except Exception as exc:
self.get_logger().error(f'Failed to receive /goal_pose navigation result: {exc}')
return
if action_result.status != GoalStatus.STATUS_SUCCEEDED:
self.get_logger().warn(
f'/goal_pose navigation ended with action status {action_result.status}.'
)
def _execute_callback(self, route, goal_handle):
nav_result = self._forward_to_nav2(route, goal_handle)
if nav_result is None:
return route['action_type'].Result()
result, status = nav_result
if status == GoalStatus.STATUS_CANCELED:
goal_handle.canceled()
return result
if status != GoalStatus.STATUS_SUCCEEDED:
goal_handle.abort()
return result
final_pose = route['final_pose_getter'](goal_handle.request)
refine_status = 'succeeded'
if final_pose is not None:
refine_status = self._refine_final_pose(goal_handle, final_pose)
if refine_status == 'succeeded':
goal_handle.succeed()
elif refine_status == 'canceled':
goal_handle.canceled()
else:
self.get_logger().error(
f"{route['label']} completed in Nav2 but final refinement ended with "
f"status '{refine_status}'."
)
goal_handle.abort()
return result
def _forward_to_nav2(self, route, goal_handle):
timeout = float(self.get_parameter('nav2_server_timeout_sec').value)
if not route['client'].wait_for_server(timeout_sec=timeout):
self.get_logger().error(f"Nav2 action server {route['nav2_name']} is not available.")
goal_handle.abort()
return None
send_future = route['client'].send_goal_async(
deepcopy(goal_handle.request),
feedback_callback=lambda message: self._relay_feedback(goal_handle, message),
)
if not self._wait_for_future(send_future, timeout):
self.get_logger().error(f"Timed out forwarding {route['label']} goal to Nav2.")
goal_handle.abort()
return None
try:
nav_goal_handle = send_future.result()
except Exception as exc:
self.get_logger().error(f"Failed to forward {route['label']} goal to Nav2: {exc}")
goal_handle.abort()
return None
if nav_goal_handle is None or not nav_goal_handle.accepted:
self.get_logger().error(f"Forwarded {route['label']} goal was rejected by Nav2.")
goal_handle.abort()
return None
result_future = nav_goal_handle.get_result_async()
while rclpy.ok() and not result_future.done():
if goal_handle.is_cancel_requested:
self._cancel_nav_goal(nav_goal_handle)
goal_handle.canceled()
return None
time.sleep(0.05)
if not result_future.done():
goal_handle.abort()
return None
try:
nav_result = result_future.result()
except Exception as exc:
self.get_logger().error(f"Failed to get Nav2 {route['label']} result: {exc}")
goal_handle.abort()
return None
result = (
nav_result.result
if nav_result and nav_result.result
else route['action_type'].Result()
)
return result, nav_result.status
def _refine_final_pose(self, goal_handle, pose):
if not self.get_parameter('require_refinement').value:
return 'succeeded'
with self._refinement_lock:
if goal_handle.is_cancel_requested:
return 'canceled'
self._publish_refiner_goal(pose)
start_sequence = self._status_snapshot()
if not self._set_refiner_parameter(self.start_param, True):
return 'unavailable'
wait_timeout = float(self.get_parameter('refiner_wait_timeout_sec').value)
deadline = time.monotonic() + max(wait_timeout, 0.0)
saw_running = False
sequence = start_sequence
while rclpy.ok():
if goal_handle.is_cancel_requested:
self._set_refiner_parameter(self.cancel_param, True)
return 'canceled'
updates = self._wait_for_status_updates(sequence, deadline)
if updates is None:
self.get_logger().error('Timed out waiting for final pose refinement result.')
self._set_refiner_parameter(self.cancel_param, True)
return 'timeout'
for sequence, status in updates:
if status == 'running':
saw_running = True
elif status in self.TERMINAL_REFINER_STATUSES:
if saw_running or status in self.STARTLESS_REFINER_FAILURES:
return status
return 'canceled'
def _set_refiner_parameter(self, name, value):
timeout = float(self.get_parameter('refiner_service_timeout_sec').value)
if not self.refiner_param_client.wait_for_service(timeout_sec=timeout):
self.get_logger().error(
f'Final pose refiner parameter service {self.refiner_param_service} '
'is unavailable.'
)
return False
parameter = ParameterMsg()
parameter.name = name
parameter.value = ParameterValue(type=ParameterType.PARAMETER_BOOL, bool_value=value)
request = SetParameters.Request()
request.parameters = [parameter]
future = self.refiner_param_client.call_async(request)
if not self._wait_for_future(future, timeout):
self.get_logger().error(f'Timed out setting final pose refiner parameter {name}.')
return False
response = future.result()
if response is None or not response.results or not response.results[0].successful:
reason = response.results[0].reason if response and response.results else ''
self.get_logger().error(f'Failed to set final pose refiner parameter {name}: {reason}')
return False
return True
def _on_refiner_status(self, message):
with self._status_condition:
self._status_sequence += 1
self._status_history.append((self._status_sequence, message.data))
self._status_history = self._status_history[-32:]
self._status_condition.notify_all()
def _status_snapshot(self):
with self._status_condition:
return self._status_sequence
def _wait_for_status_updates(self, sequence, deadline):
with self._status_condition:
while rclpy.ok():
updates = [item for item in self._status_history if item[0] > sequence]
if updates:
return updates
remaining = deadline - time.monotonic()
if remaining <= 0.0:
return None
self._status_condition.wait(timeout=min(remaining, 0.1))
return None
def _publish_refiner_goal(self, pose):
refiner_goal = deepcopy(pose)
refiner_goal.header.stamp = self.get_clock().now().to_msg()
self.goal_pub.publish(refiner_goal)
@staticmethod
def _relay_feedback(goal_handle, feedback_message):
if goal_handle.is_active:
goal_handle.publish_feedback(feedback_message.feedback)
def _debug(self, message):
if self.get_parameter('debug_print').value:
self.get_logger().info(message)
@staticmethod
def _wait_for_future(future, timeout_sec):
done = threading.Event()
future.add_done_callback(lambda _: done.set())
return done.wait(timeout_sec)
def _cancel_nav_goal(self, nav_goal_handle):
cancel_future = nav_goal_handle.cancel_goal_async()
self._wait_for_future(cancel_future, 2.0)
def main(args=None):
rclpy.init(args=args)
node = NavigateToPoseRefinerProxy()
executor = MultiThreadedExecutor(num_threads=6)
try:
rclpy.spin(node, executor=executor)
except KeyboardInterrupt:
pass
finally:
node.destroy_node()
if rclpy.ok():
rclpy.shutdown()
if __name__ == '__main__':
main()
@@ -0,0 +1,469 @@
#!/usr/bin/env python3
"""X-axis odometry scale calibration helper for AGV Pro."""
import math
import os
import shlex
import statistics
import sys
import threading
import rclpy
from geometry_msgs.msg import Twist
from rclpy.duration import Duration
from rclpy.node import Node
from rclpy.parameter import Parameter
from tf2_ros import Buffer, TransformException, TransformListener
CONTROL_RATE_HZ = 20.0
MAX_SPEED_LIMIT = 0.30
TF_TIMEOUT_SEC = 0.5
STOP_REPEAT_COUNT = 5
class OdomLinearCalib(Node):
"""Run repeated X-axis odom tests and compute the final scale from cached samples."""
def __init__(self):
super().__init__('odom_linear_calib')
self.declare_parameter('cmd_vel_topic', '/cmd_vel')
self.declare_parameter('odom_frame', 'odom')
self.declare_parameter('base_frame', 'base_footprint')
self.declare_parameter('start_test', False)
self.declare_parameter('test_distance', 1.0)
self.declare_parameter('speed', 0.10)
self.declare_parameter('tolerance', 0.01)
self.declare_parameter('odom_linear_scale_correction', 1.0)
self.declare_parameter('timeout', 30.0)
self.cmd_vel_topic = self.get_parameter('cmd_vel_topic').value
self.cmd_vel_pub = self.create_publisher(Twist, self.cmd_vel_topic, 10)
self.tf_buffer = Buffer()
self.tf_listener = TransformListener(self.tf_buffer, self)
self.state = 'idle'
self.start_pose = None
self.direction_sign = 1.0
self.signed_target_distance = 1.0
self.target_distance = 1.0
self.command_speed = 0.10
self.tolerance = 0.01
self.odom_linear_scale_correction = 1.0
self.timeout = 30.0
self.start_time = None
self.last_odom_distance = 0.0
self.last_log_time = self.get_clock().now()
self.samples = []
self.samples_lock = threading.Lock()
self.pending_sample = None
self.timer = self.create_timer(1.0 / CONTROL_RATE_HZ, self.on_timer)
threading.Thread(target=self._stdin_loop, daemon=True).start()
self.get_logger().info(
'odom_linear_calib ready. Set params, set start_test:=true for each run, '
'use positive test_distance for forward and negative for backward, '
'then enter the measured ground error in cm after the robot stops. '
'Enter 0 to finish and print the cached scale summary; enter any text to skip a verification run.'
)
def _stdin_loop(self):
while True:
line = sys.stdin.readline()
if line == '':
return
self._handle_input_line(line.strip())
def _handle_input_line(self, text):
if not text:
self.get_logger().info(
'Input ignored. After a successful run enter ground error in cm '
'(+over target along motion direction, -short), or enter 0 to finish.'
)
return
try:
value = float(text)
except ValueError:
self._skip_pending_sample(text)
return
if value == 0.0 and not text.startswith(('+', '-')):
with self.samples_lock:
had_pending_sample = self.pending_sample is not None
self.pending_sample = None
if self.state == 'awaiting_input':
self.state = 'idle'
if had_pending_sample:
self.get_logger().warn('Pending run was not recorded because finish input 0 was entered.')
self._print_summary()
return
self._record_pending_sample(value)
def on_timer(self):
if self.state == 'running':
self._run_test_step()
return
if self.state == 'awaiting_input':
if self.get_parameter('start_test').value:
self.get_logger().warn(
'A finished run is waiting for ground-error input; record it before starting again.'
)
self._reset_start_test()
return
if self.get_parameter('start_test').value:
self._start_test()
def _start_test(self):
config = self._read_test_config()
if config is None:
self._reset_start_test()
return
pose = self._lookup_pose()
if pose is None:
self.get_logger().warn('Cannot start test: odom transform is not available.')
self._reset_start_test()
return
self.signed_target_distance = config['signed_test_distance']
self.target_distance = config['target_distance']
self.command_speed = config['speed']
self.tolerance = config['tolerance']
self.odom_linear_scale_correction = config['odom_linear_scale_correction']
self.timeout = config['timeout']
self.direction_sign = float(config['direction_sign'])
self.start_pose = pose
self.start_time = self.get_clock().now()
self.last_odom_distance = 0.0
self.state = 'running'
self.get_logger().info(
f'Start X odom calibration: direction={int(self.direction_sign)}, '
f'signed_target={self.signed_target_distance:.3f} m, '
f'target={self.target_distance:.3f} m, speed={self.command_speed:.3f} m/s, '
f'odom_linear_scale_correction={self.odom_linear_scale_correction:.6f}'
)
def _run_test_step(self):
pose = self._lookup_pose()
if pose is None:
self._finish_test('failed_tf', publish_warning=True)
return
raw_progress, lateral_drift = self._calculate_progress(pose)
corrected_progress = raw_progress * self.odom_linear_scale_correction
error = corrected_progress - self.target_distance
elapsed = (self.get_clock().now() - self.start_time).nanoseconds / 1e9
self.last_odom_distance = raw_progress
if corrected_progress >= self.target_distance - self.tolerance:
self._finish_test('succeeded', raw_progress, corrected_progress, lateral_drift, elapsed)
return
if elapsed > self.timeout:
self._finish_test('timeout', raw_progress, corrected_progress, lateral_drift, elapsed)
return
cmd = Twist()
cmd.linear.x = self.direction_sign * self.command_speed
self.cmd_vel_pub.publish(cmd)
self._log_progress(raw_progress, corrected_progress, error, lateral_drift, elapsed)
def _finish_test(
self,
status,
odom_distance=None,
corrected_distance=None,
lateral_drift=None,
elapsed=None,
publish_warning=False,
):
self._stop_robot()
self._reset_start_test()
if odom_distance is None:
odom_distance = self.last_odom_distance
if corrected_distance is None:
corrected_distance = odom_distance * self.odom_linear_scale_correction
if lateral_drift is None:
lateral_drift = 0.0
if elapsed is None and self.start_time is not None:
elapsed = (self.get_clock().now() - self.start_time).nanoseconds / 1e9
if elapsed is None:
elapsed = 0.0
if status == 'succeeded' and odom_distance > 0.0:
pending_sample = {
'direction': int(self.direction_sign),
'signed_target_distance': self.signed_target_distance,
'target_distance': self.target_distance,
'odom_distance': odom_distance,
'corrected_distance': corrected_distance,
'lateral_drift': lateral_drift,
'elapsed': elapsed,
'used_correction': self.odom_linear_scale_correction,
}
with self.samples_lock:
self.pending_sample = pending_sample
self.state = 'awaiting_input'
self.get_logger().info(
'Run is waiting for measured ground error. '
'Enter cm error now: +over target along motion direction, -short of target, '
'+0/-0 for exact target, 0 to finish, or any text to skip this run.'
)
else:
self.state = 'idle'
msg = (
f'Calibration {status}: odom_distance={odom_distance:.4f} m, '
f'corrected_distance={corrected_distance:.4f} m, '
f'lateral_drift={lateral_drift:.4f} m, elapsed={elapsed:.2f} s, '
f'target={self.target_distance:.4f} m, '
f'used_correction={self.odom_linear_scale_correction:.6f}.'
)
if publish_warning:
self.get_logger().warn(msg)
else:
self.get_logger().info(msg)
def _read_test_config(self):
test_distance = self.get_parameter('test_distance').value
speed = abs(self.get_parameter('speed').value)
tolerance = max(self.get_parameter('tolerance').value, 0.0)
correction = self.get_parameter('odom_linear_scale_correction').value
timeout = self.get_parameter('timeout').value
if test_distance == 0.0:
self.get_logger().error(
'test_distance must not be 0.0 m. Use a positive value for forward, negative for backward.'
)
return None
if speed <= 0.0:
self.get_logger().error('speed must be greater than 0.0 m/s.')
return None
if timeout <= 0.0:
self.get_logger().error('timeout must be greater than 0.0 s.')
return None
if correction <= 0.0:
self.get_logger().error('odom_linear_scale_correction must be greater than 0.0.')
return None
if speed > MAX_SPEED_LIMIT:
self.get_logger().warn(
f'speed {speed:.3f} m/s exceeds internal safety limit '
f'{MAX_SPEED_LIMIT:.3f} m/s; clipping command speed.'
)
speed = MAX_SPEED_LIMIT
direction_sign = 1 if test_distance > 0.0 else -1
return {
'direction_sign': direction_sign,
'signed_test_distance': test_distance,
'target_distance': abs(test_distance),
'speed': speed,
'tolerance': tolerance,
'odom_linear_scale_correction': correction,
'timeout': timeout,
}
def _lookup_pose(self):
odom_frame = self.get_parameter('odom_frame').value
base_frame = self.get_parameter('base_frame').value
try:
trans = self.tf_buffer.lookup_transform(
odom_frame,
base_frame,
rclpy.time.Time(),
timeout=Duration(seconds=TF_TIMEOUT_SEC),
)
except TransformException as exc:
self.get_logger().warn(f'TF lookup failed: {exc}')
return None
translation = trans.transform.translation
rotation = trans.transform.rotation
return (
translation.x,
translation.y,
self._yaw_from_quaternion(rotation.x, rotation.y, rotation.z, rotation.w),
)
def _calculate_progress(self, pose):
x, y, _ = pose
start_x, start_y, start_yaw = self.start_pose
dx = x - start_x
dy = y - start_y
cos_yaw = math.cos(start_yaw)
sin_yaw = math.sin(start_yaw)
forward_delta = dx * cos_yaw + dy * sin_yaw
lateral_drift = -dx * sin_yaw + dy * cos_yaw
progress = self.direction_sign * forward_delta
return progress, lateral_drift
def _log_progress(self, raw_progress, corrected_progress, error, lateral_drift, elapsed):
now = self.get_clock().now()
if (now - self.last_log_time).nanoseconds < 1e9:
return
self.get_logger().info(
f'odom_distance={raw_progress:.3f} m, '
f'corrected_distance={corrected_progress:.3f} m, '
f'error={error:+.3f} m, lateral_drift={lateral_drift:.3f} m, '
f'elapsed={elapsed:.1f} s'
)
self.last_log_time = now
def _skip_pending_sample(self, reason):
with self.samples_lock:
if self.pending_sample is None:
self.get_logger().warn(
f'Input "{reason}" ignored. No pending successful run is waiting for input.'
)
return
skipped_sample = self.pending_sample
self.pending_sample = None
self.state = 'idle'
self.get_logger().info(
f'Skipped pending run by input "{reason}": '
f'direction={skipped_sample["direction"]:+d}, '
f'signed_target={skipped_sample["signed_target_distance"]:.4f} m, '
f'odom={skipped_sample["odom_distance"]:.4f} m, '
f'corrected={skipped_sample["corrected_distance"]:.4f} m, '
f'used_correction={skipped_sample["used_correction"]:.6f}. '
'This run will not be used in the final scale summary.'
)
def _record_pending_sample(self, ground_error_cm):
with self.samples_lock:
if self.pending_sample is None:
self.get_logger().warn(
'No pending successful run. Set start_test:=true first, wait for the robot to stop, '
'then enter the measured cm error.'
)
return
actual_distance = self.pending_sample['target_distance'] + ground_error_cm / 100.0
if actual_distance <= 0.0:
self.get_logger().error(
f'Invalid measured result: target + error = {actual_distance:.4f} m. '
'Re-enter the cm error for this pending run.'
)
return
sample = dict(self.pending_sample)
sample['ground_error_cm'] = ground_error_cm
sample['actual_distance'] = actual_distance
sample['scale'] = actual_distance / sample['odom_distance']
self.samples.append(sample)
sample_index = len(self.samples)
direction_index = sum(
1 for recorded_sample in self.samples
if recorded_sample['direction'] == sample['direction']
)
self.pending_sample = None
self.state = 'idle'
self.get_logger().info(
f'Recorded sample #{sample_index} overall, direction {sample["direction"]:+d} #{direction_index}: '
f'actual={actual_distance:.4f} m, '
f'ground_error={ground_error_cm:+.2f} cm, odom={sample["odom_distance"]:.4f} m, '
f'scale={sample["scale"]:.6f}. Set start_test:=true for the next run, or enter 0 to finish.'
)
def _print_summary(self):
with self.samples_lock:
samples = list(self.samples)
if not samples:
self.get_logger().warn('No successful calibration samples have been recorded yet.')
return
self.get_logger().info('========== X ODOM SCALE SUMMARY ==========')
for index, sample in enumerate(samples, start=1):
self.get_logger().info(
f'#{index:02d} direction={sample["direction"]:+d}, '
f'signed_target={sample["signed_target_distance"]:.4f} m, '
f'target={sample["target_distance"]:.4f} m, '
f'actual={sample["actual_distance"]:.4f} m, '
f'ground_error={sample["ground_error_cm"]:+.2f} cm, '
f'odom={sample["odom_distance"]:.4f} m, '
f'corrected={sample["corrected_distance"]:.4f} m, '
f'lateral_drift={sample["lateral_drift"]:.4f} m, '
f'used_correction={sample["used_correction"]:.6f}, '
f'scale={sample["scale"]:.6f}'
)
self._print_scale_stats('all', samples)
for direction in (1, -1):
direction_samples = [sample for sample in samples if sample['direction'] == direction]
if direction_samples:
self._print_scale_stats(f'direction={direction:+d}', direction_samples)
self.get_logger().info('Restart this node to clear cached samples.')
def _print_scale_stats(self, label, samples):
scales = [sample['scale'] for sample in samples]
mean_scale = statistics.fmean(scales)
std_scale = statistics.pstdev(scales) if len(scales) > 1 else 0.0
self.get_logger().info(
f'{label}: samples={len(scales)}, recommended_odometry.scale_x={mean_scale:.6f}, '
f'std={std_scale:.6f}, min={min(scales):.6f}, max={max(scales):.6f}'
)
def _publish_stop(self):
try:
self.cmd_vel_pub.publish(Twist())
except Exception:
pass
def _stop_robot(self):
for _ in range(STOP_REPEAT_COUNT):
self._publish_stop()
def _stop_robot_with_ros_cli(self):
topic = shlex.quote(self.cmd_vel_topic)
zero_twist = (
'"{linear: {x: 0.0, y: 0.0, z: 0.0}, '
'angular: {x: 0.0, y: 0.0, z: 0.0}}"'
)
os.system(
f'timeout 2s ros2 topic pub --once {topic} '
f'geometry_msgs/msg/Twist {zero_twist} >/dev/null 2>&1'
)
def _reset_start_test(self):
self.set_parameters([
Parameter('start_test', Parameter.Type.BOOL, False),
])
@staticmethod
def _yaw_from_quaternion(x, y, z, w):
siny_cosp = 2.0 * (w * z + x * y)
cosy_cosp = 1.0 - 2.0 * (y * y + z * z)
return math.atan2(siny_cosp, cosy_cosp)
def main(args=None):
rclpy.init(args=args)
node = OdomLinearCalib()
try:
rclpy.spin(node)
except KeyboardInterrupt:
pass
finally:
node._stop_robot()
node._stop_robot_with_ros_cli()
node.destroy_node()
if rclpy.ok():
rclpy.shutdown()
if __name__ == '__main__':
main()
@@ -0,0 +1,466 @@
#!/usr/bin/env python3
"""Yaw odometry scale calibration helper for AGV Pro."""
import math
import os
import shlex
import statistics
import sys
import threading
import rclpy
from geometry_msgs.msg import Twist
from rclpy.duration import Duration
from rclpy.node import Node
from rclpy.parameter import Parameter
from tf2_ros import Buffer, TransformException, TransformListener
CONTROL_RATE_HZ = 20.0
MAX_ANGULAR_SPEED_LIMIT = 0.50
TF_TIMEOUT_SEC = 0.5
STOP_REPEAT_COUNT = 5
class OdomYawCalib(Node):
"""Run repeated yaw odom tests and compute the final scale from cached samples."""
def __init__(self):
super().__init__('odom_yaw_calib')
self.declare_parameter('cmd_vel_topic', '/cmd_vel')
self.declare_parameter('odom_frame', 'odom')
self.declare_parameter('base_frame', 'base_footprint')
self.declare_parameter('start_test', False)
self.declare_parameter('test_angle', 360.0)
self.declare_parameter('speed', 0.20)
self.declare_parameter('tolerance', 2.0)
self.declare_parameter('odom_yaw_scale_correction', 1.0)
self.declare_parameter('timeout', 60.0)
self.cmd_vel_topic = self.get_parameter('cmd_vel_topic').value
self.cmd_vel_pub = self.create_publisher(Twist, self.cmd_vel_topic, 10)
self.tf_buffer = Buffer()
self.tf_listener = TransformListener(self.tf_buffer, self)
self.state = 'idle'
self.direction_sign = 1.0
self.signed_target_angle_deg = 360.0
self.target_angle_deg = 360.0
self.target_angle = math.radians(360.0)
self.command_speed = 0.20
self.tolerance_deg = 2.0
self.tolerance = math.radians(2.0)
self.odom_yaw_scale_correction = 1.0
self.timeout = 60.0
self.start_time = None
self.prev_yaw = None
self.accumulated_yaw = 0.0
self.last_odom_angle = 0.0
self.last_log_time = self.get_clock().now()
self.samples = []
self.samples_lock = threading.Lock()
self.pending_sample = None
self.timer = self.create_timer(1.0 / CONTROL_RATE_HZ, self.on_timer)
threading.Thread(target=self._stdin_loop, daemon=True).start()
self.get_logger().info(
'odom_yaw_calib ready. Set params, set start_test:=true for each run, '
'use positive test_angle for positive angular.z and negative for negative angular.z, '
'then enter the measured ground yaw error in deg after the robot stops. '
'Enter 0 to finish and print the cached scale summary; enter any text to skip a verification run.'
)
def _stdin_loop(self):
while True:
line = sys.stdin.readline()
if line == '':
return
self._handle_input_line(line.strip())
def _handle_input_line(self, text):
if not text:
self.get_logger().info(
'Input ignored. After a successful run enter ground yaw error in deg '
'(+over target along rotation direction, -short), or enter 0 to finish.'
)
return
try:
value = float(text)
except ValueError:
self._skip_pending_sample(text)
return
if value == 0.0 and not text.startswith(('+', '-')):
with self.samples_lock:
had_pending_sample = self.pending_sample is not None
self.pending_sample = None
if self.state == 'awaiting_input':
self.state = 'idle'
if had_pending_sample:
self.get_logger().warn('Pending run was not recorded because finish input 0 was entered.')
self._print_summary()
return
self._record_pending_sample(value)
def on_timer(self):
if self.state == 'running':
self._run_test_step()
return
if self.state == 'awaiting_input':
if self.get_parameter('start_test').value:
self.get_logger().warn(
'A finished run is waiting for ground-yaw-error input; record it before starting again.'
)
self._reset_start_test()
return
if self.get_parameter('start_test').value:
self._start_test()
def _start_test(self):
config = self._read_test_config()
if config is None:
self._reset_start_test()
return
pose = self._lookup_pose()
if pose is None:
self.get_logger().warn('Cannot start test: odom transform is not available.')
self._reset_start_test()
return
self.signed_target_angle_deg = config['signed_test_angle']
self.target_angle_deg = config['target_angle']
self.target_angle = math.radians(config['target_angle'])
self.command_speed = config['speed']
self.tolerance_deg = config['tolerance']
self.tolerance = math.radians(config['tolerance'])
self.odom_yaw_scale_correction = config['odom_yaw_scale_correction']
self.timeout = config['timeout']
self.direction_sign = float(config['direction_sign'])
self.prev_yaw = pose[2]
self.accumulated_yaw = 0.0
self.start_time = self.get_clock().now()
self.last_odom_angle = 0.0
self.state = 'running'
self.get_logger().info(
f'Start yaw odom calibration: direction={int(self.direction_sign)}, '
f'signed_target={self.signed_target_angle_deg:.1f} deg, '
f'target={self.target_angle_deg:.1f} deg, speed={self.command_speed:.3f} rad/s, '
f'odom_yaw_scale_correction={self.odom_yaw_scale_correction:.6f}'
)
def _run_test_step(self):
pose = self._lookup_pose()
if pose is None:
self._finish_test('failed_tf', publish_warning=True)
return
raw_progress = self._calculate_yaw_progress(pose)
raw_angle = max(raw_progress, 0.0)
corrected_angle = raw_angle * self.odom_yaw_scale_correction
error = corrected_angle - self.target_angle
elapsed = (self.get_clock().now() - self.start_time).nanoseconds / 1e9
self.last_odom_angle = raw_angle
if corrected_angle >= self.target_angle - self.tolerance:
self._finish_test('succeeded', raw_angle, corrected_angle, elapsed)
return
if elapsed > self.timeout:
self._finish_test('timeout', raw_angle, corrected_angle, elapsed)
return
cmd = Twist()
cmd.angular.z = self.direction_sign * self.command_speed
self.cmd_vel_pub.publish(cmd)
self._log_progress(raw_angle, corrected_angle, error, elapsed)
def _finish_test(
self,
status,
odom_angle=None,
corrected_angle=None,
elapsed=None,
publish_warning=False,
):
self._stop_robot()
self._reset_start_test()
if odom_angle is None:
odom_angle = self.last_odom_angle
if corrected_angle is None:
corrected_angle = odom_angle * self.odom_yaw_scale_correction
if elapsed is None and self.start_time is not None:
elapsed = (self.get_clock().now() - self.start_time).nanoseconds / 1e9
if elapsed is None:
elapsed = 0.0
if status == 'succeeded' and odom_angle > 0.0:
pending_sample = {
'direction': int(self.direction_sign),
'signed_target_angle_deg': self.signed_target_angle_deg,
'target_angle_deg': self.target_angle_deg,
'odom_angle_deg': math.degrees(odom_angle),
'corrected_angle_deg': math.degrees(corrected_angle),
'elapsed': elapsed,
'used_correction': self.odom_yaw_scale_correction,
}
with self.samples_lock:
self.pending_sample = pending_sample
self.state = 'awaiting_input'
self.get_logger().info(
'Run is waiting for measured ground yaw error. '
'Enter deg error now: +over target along rotation direction, -short of target, '
'+0/-0 for exact target, 0 to finish, or any text to skip this run.'
)
else:
self.state = 'idle'
msg = (
f'Calibration {status}: odom_angle={math.degrees(odom_angle):.2f} deg, '
f'corrected_angle={math.degrees(corrected_angle):.2f} deg, '
f'elapsed={elapsed:.2f} s, target={self.target_angle_deg:.2f} deg, '
f'used_correction={self.odom_yaw_scale_correction:.6f}.'
)
if publish_warning:
self.get_logger().warn(msg)
else:
self.get_logger().info(msg)
def _read_test_config(self):
test_angle = self.get_parameter('test_angle').value
speed = abs(self.get_parameter('speed').value)
tolerance = max(self.get_parameter('tolerance').value, 0.0)
correction = self.get_parameter('odom_yaw_scale_correction').value
timeout = self.get_parameter('timeout').value
if test_angle == 0.0:
self.get_logger().error(
'test_angle must not be 0.0 deg. Use a positive value for one yaw direction, '
'negative for the opposite direction.'
)
return None
if speed <= 0.0:
self.get_logger().error('speed must be greater than 0.0 rad/s.')
return None
if timeout <= 0.0:
self.get_logger().error('timeout must be greater than 0.0 s.')
return None
if correction <= 0.0:
self.get_logger().error('odom_yaw_scale_correction must be greater than 0.0.')
return None
if speed > MAX_ANGULAR_SPEED_LIMIT:
self.get_logger().warn(
f'speed {speed:.3f} rad/s exceeds internal safety limit '
f'{MAX_ANGULAR_SPEED_LIMIT:.3f} rad/s; clipping command speed.'
)
speed = MAX_ANGULAR_SPEED_LIMIT
direction_sign = 1 if test_angle > 0.0 else -1
return {
'direction_sign': direction_sign,
'signed_test_angle': test_angle,
'target_angle': abs(test_angle),
'speed': speed,
'tolerance': tolerance,
'odom_yaw_scale_correction': correction,
'timeout': timeout,
}
def _lookup_pose(self):
odom_frame = self.get_parameter('odom_frame').value
base_frame = self.get_parameter('base_frame').value
try:
trans = self.tf_buffer.lookup_transform(
odom_frame,
base_frame,
rclpy.time.Time(),
timeout=Duration(seconds=TF_TIMEOUT_SEC),
)
except TransformException as exc:
self.get_logger().warn(f'TF lookup failed: {exc}')
return None
rotation = trans.transform.rotation
return (
trans.transform.translation.x,
trans.transform.translation.y,
self._yaw_from_quaternion(rotation.x, rotation.y, rotation.z, rotation.w),
)
def _calculate_yaw_progress(self, pose):
current_yaw = pose[2]
delta = math.atan2(
math.sin(current_yaw - self.prev_yaw),
math.cos(current_yaw - self.prev_yaw),
)
self.accumulated_yaw += delta
self.prev_yaw = current_yaw
return self.direction_sign * self.accumulated_yaw
def _log_progress(self, raw_angle, corrected_angle, error, elapsed):
now = self.get_clock().now()
if (now - self.last_log_time).nanoseconds < 1e9:
return
self.get_logger().info(
f'odom_angle={math.degrees(raw_angle):.1f} deg, '
f'corrected_angle={math.degrees(corrected_angle):.1f} deg, '
f'error={math.degrees(error):+.1f} deg, elapsed={elapsed:.1f} s'
)
self.last_log_time = now
def _skip_pending_sample(self, reason):
with self.samples_lock:
if self.pending_sample is None:
self.get_logger().warn(
f'Input "{reason}" ignored. No pending successful run is waiting for input.'
)
return
skipped_sample = self.pending_sample
self.pending_sample = None
self.state = 'idle'
self.get_logger().info(
f'Skipped pending run by input "{reason}": '
f'direction={skipped_sample["direction"]:+d}, '
f'signed_target={skipped_sample["signed_target_angle_deg"]:.2f} deg, '
f'odom={skipped_sample["odom_angle_deg"]:.2f} deg, '
f'corrected={skipped_sample["corrected_angle_deg"]:.2f} deg, '
f'used_correction={skipped_sample["used_correction"]:.6f}. '
'This run will not be used in the final scale summary.'
)
def _record_pending_sample(self, ground_error_deg):
with self.samples_lock:
if self.pending_sample is None:
self.get_logger().warn(
'No pending successful run. Set start_test:=true first, wait for the robot to stop, '
'then enter the measured deg error.'
)
return
actual_angle_deg = self.pending_sample['target_angle_deg'] + ground_error_deg
if actual_angle_deg <= 0.0:
self.get_logger().error(
f'Invalid measured result: target + error = {actual_angle_deg:.2f} deg. '
'Re-enter the deg error for this pending run.'
)
return
sample = dict(self.pending_sample)
sample['ground_error_deg'] = ground_error_deg
sample['actual_angle_deg'] = actual_angle_deg
sample['scale'] = math.radians(actual_angle_deg) / math.radians(sample['odom_angle_deg'])
self.samples.append(sample)
sample_index = len(self.samples)
direction_index = sum(
1 for recorded_sample in self.samples
if recorded_sample['direction'] == sample['direction']
)
self.pending_sample = None
self.state = 'idle'
self.get_logger().info(
f'Recorded sample #{sample_index} overall, direction {sample["direction"]:+d} #{direction_index}: '
f'actual={actual_angle_deg:.2f} deg, '
f'ground_error={ground_error_deg:+.2f} deg, odom={sample["odom_angle_deg"]:.2f} deg, '
f'scale={sample["scale"]:.6f}. Set start_test:=true for the next run, or enter 0 to finish.'
)
def _print_summary(self):
with self.samples_lock:
samples = list(self.samples)
if not samples:
self.get_logger().warn('No successful calibration samples have been recorded yet.')
return
self.get_logger().info('========== YAW ODOM SCALE SUMMARY ==========')
for index, sample in enumerate(samples, start=1):
self.get_logger().info(
f'#{index:02d} direction={sample["direction"]:+d}, '
f'signed_target={sample["signed_target_angle_deg"]:.2f} deg, '
f'target={sample["target_angle_deg"]:.2f} deg, '
f'actual={sample["actual_angle_deg"]:.2f} deg, '
f'ground_error={sample["ground_error_deg"]:+.2f} deg, '
f'odom={sample["odom_angle_deg"]:.2f} deg, '
f'corrected={sample["corrected_angle_deg"]:.2f} deg, '
f'used_correction={sample["used_correction"]:.6f}, '
f'scale={sample["scale"]:.6f}'
)
self._print_scale_stats('all', samples)
for direction in (1, -1):
direction_samples = [sample for sample in samples if sample['direction'] == direction]
if direction_samples:
self._print_scale_stats(f'direction={direction:+d}', direction_samples)
self.get_logger().info('Restart this node to clear cached samples.')
def _print_scale_stats(self, label, samples):
scales = [sample['scale'] for sample in samples]
mean_scale = statistics.fmean(scales)
std_scale = statistics.pstdev(scales) if len(scales) > 1 else 0.0
self.get_logger().info(
f'{label}: samples={len(scales)}, recommended_odometry.scale_theta={mean_scale:.6f}, '
f'std={std_scale:.6f}, min={min(scales):.6f}, max={max(scales):.6f}'
)
def _publish_stop(self):
try:
self.cmd_vel_pub.publish(Twist())
except Exception:
pass
def _stop_robot(self):
for _ in range(STOP_REPEAT_COUNT):
self._publish_stop()
def _stop_robot_with_ros_cli(self):
topic = shlex.quote(self.cmd_vel_topic)
zero_twist = (
'"{linear: {x: 0.0, y: 0.0, z: 0.0}, '
'angular: {x: 0.0, y: 0.0, z: 0.0}}"'
)
os.system(
f'timeout 2s ros2 topic pub --once {topic} '
f'geometry_msgs/msg/Twist {zero_twist} >/dev/null 2>&1'
)
def _reset_start_test(self):
self.set_parameters([
Parameter('start_test', Parameter.Type.BOOL, False),
])
@staticmethod
def _yaw_from_quaternion(x, y, z, w):
siny_cosp = 2.0 * (w * z + x * y)
cosy_cosp = 1.0 - 2.0 * (y * y + z * z)
return math.atan2(siny_cosp, cosy_cosp)
def main(args=None):
rclpy.init(args=args)
node = OdomYawCalib()
try:
rclpy.spin(node)
except KeyboardInterrupt:
pass
finally:
node._stop_robot()
node._stop_robot_with_ros_cli()
node.destroy_node()
if rclpy.ok():
rclpy.shutdown()
if __name__ == '__main__':
main()
+28
View File
@@ -0,0 +1,28 @@
<?xml version="1.0"?>
<?xml-model href="http://download.ros.org/schema/package_format3.xsd" schematypens="http://www.w3.org/2001/XMLSchema"?>
<package format="3">
<name>agv_pro_calibration</name>
<version>0.0.0</version>
<description>AGV Pro calibration tools for odom, IMU, and TF health check</description>
<maintainer email="elephant@todo.todo">elephant</maintainer>
<license>TODO: License declaration</license>
<depend>rclpy</depend>
<depend>geometry_msgs</depend>
<depend>nav_msgs</depend>
<depend>sensor_msgs</depend>
<depend>tf2_ros</depend>
<depend>std_msgs</depend>
<depend>action_msgs</depend>
<depend>nav2_msgs</depend>
<depend>rcl_interfaces</depend>
<test_depend>ament_copyright</test_depend>
<test_depend>ament_flake8</test_depend>
<test_depend>ament_pep257</test_depend>
<test_depend>python3-pytest</test_depend>
<export>
<build_type>ament_python</build_type>
</export>
</package>
+4
View File
@@ -0,0 +1,4 @@
[develop]
script_dir=$base/lib/agv_pro_calibration
[install]
install_scripts=$base/lib/agv_pro_calibration
+29
View File
@@ -0,0 +1,29 @@
from setuptools import find_packages, setup
package_name = 'agv_pro_calibration'
setup(
name=package_name,
version='0.0.0',
packages=find_packages(exclude=['test']),
data_files=[
('share/ament_index/resource_index/packages',
['resource/' + package_name]),
('share/' + package_name, ['package.xml']),
],
install_requires=['setuptools'],
zip_safe=True,
maintainer='elephant',
maintainer_email='elephant@todo.todo',
description='AGV Pro calibration tools for odom, IMU, and TF health check',
license='TODO: License declaration',
tests_require=['pytest'],
entry_points={
'console_scripts': [
'odom_linear_calib = agv_pro_calibration.odom_linear_calib:main',
'odom_yaw_calib = agv_pro_calibration.odom_yaw_calib:main',
'final_pose_refiner = agv_pro_calibration.final_pose_refiner:main',
'navigate_to_pose_refiner_proxy = agv_pro_calibration.navigate_to_pose_refiner_proxy:main',
],
},
)
@@ -0,0 +1,25 @@
# Copyright 2015 Open Source Robotics Foundation, Inc.
#
# 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.
from ament_copyright.main import main
import pytest
# Remove the `skip` decorator once the source file(s) have a copyright header
@pytest.mark.skip(reason='No copyright header has been placed in the generated source file.')
@pytest.mark.copyright
@pytest.mark.linter
def test_copyright():
rc = main(argv=['.', 'test'])
assert rc == 0, 'Found errors'
+25
View File
@@ -0,0 +1,25 @@
# Copyright 2017 Open Source Robotics Foundation, Inc.
#
# 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.
from ament_flake8.main import main_with_errors
import pytest
@pytest.mark.flake8
@pytest.mark.linter
def test_flake8():
rc, errors = main_with_errors(argv=[])
assert rc == 0, \
'Found %d code style errors / warnings:\n' % len(errors) + \
'\n'.join(errors)
+23
View File
@@ -0,0 +1,23 @@
# Copyright 2015 Open Source Robotics Foundation, Inc.
#
# 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.
from ament_pep257.main import main
import pytest
@pytest.mark.linter
@pytest.mark.pep257
def test_pep257():
rc = main(argv=['.', 'test'])
assert rc == 0, 'Found code style errors / warnings'
@@ -3,11 +3,16 @@ import os
from ament_index_python.packages import get_package_share_directory
from launch import LaunchDescription
from launch.substitutions import LaunchConfiguration
from launch.actions import DeclareLaunchArgument,IncludeLaunchDescription
from launch.actions import (
DeclareLaunchArgument,
GroupAction,
IncludeLaunchDescription,
)
from launch.conditions import IfCondition
from launch.launch_description_sources import PythonLaunchDescriptionSource
from launch_ros.actions import Node
from launch.substitutions import LaunchConfiguration
from launch_ros.actions import Node, SetRemap
def generate_launch_description():
use_sim_time = LaunchConfiguration('use_sim_time', default='false')
@@ -27,7 +32,8 @@ def generate_launch_description():
'param',
param_file_name))
nav2_launch_file_dir = os.path.join(get_package_share_directory('nav2_bringup'), 'launch')
nav2_launch_file_dir = os.path.join(
get_package_share_directory('nav2_bringup'), 'launch')
rviz_config_dir = os.path.join(
get_package_share_directory('agv_pro_navigation2'),
@@ -45,12 +51,41 @@ def generate_launch_description():
default_value=param_dir,
description='Full path to param file to load'),
GroupAction(actions=[SetRemap(
src='/goal_pose',
dst='/goal_pose_nav2',
)] + [
SetRemap(
src=f'/{action}/_action/{suffix}',
dst=f'/{action}_nav2/_action/{suffix}')
for action in ('navigate_to_pose', 'navigate_through_poses')
for suffix in ('send_goal', 'get_result', 'cancel_goal', 'feedback', 'status')
] + [
IncludeLaunchDescription(
PythonLaunchDescriptionSource([nav2_launch_file_dir, '/bringup_launch.py']),
PythonLaunchDescriptionSource(
[nav2_launch_file_dir, '/bringup_launch.py']),
launch_arguments={
'map': map_dir,
'params_file': param_dir}.items(),
),
'params_file': param_dir,
}.items()),
], scoped=True),
Node(
package='agv_pro_calibration',
executable='navigate_to_pose_refiner_proxy',
name='navigate_to_pose_refiner_proxy',
output='screen',
parameters=[{'use_sim_time': use_sim_time}]),
Node(
package='agv_pro_calibration',
executable='final_pose_refiner',
name='final_pose_refiner',
output='screen',
parameters=[{
'use_sim_time': use_sim_time,
'final_pose_refiner_auto_start_on_nav_success': False,
}]),
Node(
package='rviz2',
File diff suppressed because one or more lines are too long
+1 -1
View File
@@ -1,7 +1,7 @@
image: map.pgm
mode: trinary
resolution: 0.05
origin: [-10, -24.4, 0]
origin: [-22.8, -10, 0]
negate: 0
occupied_thresh: 0.65
free_thresh: 0.25
File diff suppressed because one or more lines are too long
+7
View File
@@ -0,0 +1,7 @@
image: map1.pgm
mode: trinary
resolution: 0.05
origin: [-10, -10, 0]
negate: 0
occupied_thresh: 0.65
free_thresh: 0.25
File diff suppressed because one or more lines are too long
+7
View File
@@ -0,0 +1,7 @@
image: map.pgm
mode: trinary
resolution: 0.05
origin: [-21.2, -22.8, 0]
negate: 0
occupied_thresh: 0.65
free_thresh: 0.25
+30 -16
View File
@@ -14,10 +14,13 @@ amcl:
global_frame_id: "map"
lambda_short: 0.1
laser_likelihood_max_dist: 2.0
laser_max_range: 100.0
laser_min_range: -1.0
# /* 阶段4修改:AMCL 激光匹配最大有效距离,限制远距离无效数据影响粒子权重,初始值为 100.0m */
laser_max_range: 10.0
# /* 阶段4修改:AMCL 激光匹配最小有效距离,过滤雷达近距盲区数据,初始值为 -1.0 */
laser_min_range: 0.2
laser_model_type: "likelihood_field"
max_beams: 60
# /* 阶段4修改:AMCL 每次更新采样的激光束数量,提高定位匹配信息量,初始值为 60 */
max_beams: 90
max_particles: 2000
min_particles: 500
odom_frame_id: "odom"
@@ -25,14 +28,19 @@ amcl:
pf_z: 0.99
recovery_alpha_fast: 0.0
recovery_alpha_slow: 0.0
resample_interval: 2
robot_model_type: "nav2_amcl::OmniMotionModel"
# /* 阶段4修改:AMCL 粒子重采样间隔,提高低速接近目标时的定位收敛及时性,初始值为 2 */
resample_interval: 1
# /* 阶段4修改:AMCL 里程计运动模型,减少未使用 y 方向速度时的横向运动假设,初始值为 nav2_amcl::OmniMotionModel */
robot_model_type: "nav2_amcl::DifferentialMotionModel"
save_pose_rate: 0.5
sigma_hit: 0.02
# /* 阶段4修改:AMCL 激光命中似然标准差,降低真实车地图匹配误差导致的权重过敏感,初始值为 0.02 */
sigma_hit: 0.04
tf_broadcast: true
transform_tolerance: 0.3
update_min_a: 0.06
update_min_d: 0.025
# /* 方案2修改:降低 AMCL 角度更新触发阈值,让目标附近低速小角度调整时定位更新更及时,初始值为 0.06rad */
update_min_a: 0.04
# /* 方案2修改:降低 AMCL 位移更新触发阈值,让目标附近低速接近时更频繁使用激光修正定位,初始值为 0.025m */
update_min_d: 0.015
z_hit: 0.7
z_max: 0.001
z_rand: 0.059
@@ -132,13 +140,15 @@ controller_server:
general_goal_checker:
stateful: True
plugin: "nav2_controller::SimpleGoalChecker"
xy_goal_tolerance: 0.25
yaw_goal_tolerance: 0.25
# /* 阶段2修改:最终位置成功阈值,避免 25cm 内提前判定到点,初始值为 0.25m */
xy_goal_tolerance: 0.05
# /* 阶段2临时验证修改:最终姿态成功阈值,放宽到约 5.7 度以验证大角度掉头时末端旋转振荡是否由 0.05rad 过严导致,初始值为 0.25rad*/
yaw_goal_tolerance: 0.8
# DWB parameters
FollowPath:
plugin: "dwb_core::DWBLocalPlanner"
debug_trajectory_details: True
min_vel_x: 0.0
min_vel_x: -0.03
min_vel_y: 0.0
max_vel_x: 0.26
max_vel_y: 0.0
@@ -150,10 +160,10 @@ controller_server:
# https://github.com/ROBOTIS-GIT/turtlebot3_simulations/issues/75
acc_lim_x: 2.5
acc_lim_y: 0.0
acc_lim_theta: 3.2
acc_lim_theta: 2.5
decel_lim_x: -2.5
decel_lim_y: 0.0
decel_lim_theta: -3.2
decel_lim_theta: -2.5
vx_samples: 20
vy_samples: 5
vtheta_samples: 40
@@ -161,8 +171,10 @@ controller_server:
linear_granularity: 0.05
angular_granularity: 0.025
transform_tolerance: 0.1
xy_goal_tolerance: 0.25
trans_stopped_velocity: 0.1
# /* 阶段2修改:DWB RotateToGoal 进入末端减速/旋转模式的位置窗口,避免 25cm 内过早进入末端旋转逻辑,初始值为 0.25m */
xy_goal_tolerance: 0.03 #0.03
# /* 阶段2修改:进入只旋转阶段前允许的最大平移速度,避免平移速度小于 0.1m/s 时过早停止平移修正,初始值为 0.1m/s */
trans_stopped_velocity: 0.01 #0.01
short_circuit_trajectory_evaluation: True
stateful: True
critics: ["RotateToGoal", "Oscillation", "BaseObstacle", "GoalAlign", "PathAlign", "PathDist", "GoalDist"]
@@ -292,8 +304,10 @@ planner_server:
planner_plugins: ["GridBased"]
GridBased:
plugin: "nav2_navfn_planner/NavfnPlanner"
tolerance: 2.0
# /* 阶段2修改:全局规划目标替代容差,避免 2.0m 范围内替代终点影响精度测试,初始值为 2.0m */
tolerance: 0.05 #0.05
use_astar: false
# /* 阶段2修改:是否允许规划到未知区域,避免精度测试阶段出现不可控路径终点,初始值为 true */
allow_unknown: true
planner_server_rclcpp_node:
@@ -1,17 +1,36 @@
#! /usr/bin/env python3
import argparse
import sys
import yaml
import rclpy
from geometry_msgs.msg import PoseStamped
from nav2_simple_commander.robot_navigator import BasicNavigator, TaskResult
import rclpy
from rclpy.duration import Duration
"""
Basic navigation demo to go to pose.
"""
def parse_arguments():
parser = argparse.ArgumentParser(description='Send navigation test goals.')
parser.add_argument(
'--localization',
choices=('amcl', 'slam'),
default='amcl',
help='Localization backend started by navigation2_active.launch.py.')
parser.add_argument(
'targets',
nargs='*',
help='Waypoint letters to execute once, for example AB. Omit to loop ABCDE.')
return parser.parse_args()
def set_initial_pose(navigator: BasicNavigator, x: float, y: float, oz: float, ow: float):
"""
Set the initial pose of the robot for AMCL localization.
Set the initial pose of the robot for the active localization backend.
Args:
navigator (BasicNavigator): The navigator instance controlling the robot.
@@ -30,21 +49,7 @@ def set_initial_pose(navigator: BasicNavigator, x: float, y: float, oz: float, o
navigator.setInitialPose(initial_pose)
def navigate_to_goal(navigator: BasicNavigator, x: float, y: float, oz: float, ow: float, verbose: bool = False) -> bool:
"""
Navigate the robot to a target goal pose.
Args:
navigator (BasicNavigator): The navigator instance controlling the robot.
x (float): Goal X position in the map frame.
y (float): Goal Y position in the map frame.
oz (float): Orientation Z component (quaternion).
ow (float): Orientation W component (quaternion).
verbose (bool, optional): If True, prints navigation feedback such as estimated arrival time. Default is False.
Returns:
bool: True if navigation succeeded, False otherwise.
"""
def make_goal_pose(navigator: BasicNavigator, x: float, y: float, oz: float, ow: float) -> PoseStamped:
goal_pose = PoseStamped()
goal_pose.header.frame_id = 'map'
goal_pose.header.stamp = navigator.get_clock().now().to_msg()
@@ -52,6 +57,22 @@ def navigate_to_goal(navigator: BasicNavigator, x: float, y: float, oz: float, o
goal_pose.pose.position.y = y
goal_pose.pose.orientation.z = oz
goal_pose.pose.orientation.w = ow
return goal_pose
def navigate_to_goal(navigator: BasicNavigator, goal_pose: PoseStamped, verbose: bool = False) -> bool:
"""
Navigate the robot to a target goal pose.
Args:
navigator (BasicNavigator): The navigator instance controlling the robot.
goal_pose (PoseStamped): Goal pose in the map frame.
verbose (bool, optional): If True, prints navigation feedback such as estimated arrival time. Default is False.
Returns:
bool: True if navigation succeeded, False otherwise.
"""
goal_pose.header.stamp = navigator.get_clock().now().to_msg()
navigator.goToPose(goal_pose)
@@ -74,24 +95,76 @@ def navigate_to_goal(navigator: BasicNavigator, x: float, y: float, oz: float, o
return False
if __name__ == '__main__':
cli_args = parse_arguments()
rclpy.init()
navigator = BasicNavigator()
# Set robot initial pose
# set_initial_pose(navigator, x=-1.9248794317245483, y=-0.5366987586021423, oz=-1.8463129131030735e-06, ow=0.9999999999982956)
if cli_args.localization == 'slam':
# slam_toolbox on Humble is not a Nav2 lifecycle-managed localizer.
navigator._waitForNodeToActivate('bt_navigator')
navigator.info('Nav2 is ready for use!')
else:
# AMCL obtains its initial origin pose from agvpro.yaml.
navigator.initial_pose_received = True
navigator.waitUntilNav2Active()
# Wait for navigation to fully activate, since autostarting nav2
# navigator.waitUntilNav2Active()
# Try to load waypoints from YAML, fallback to hardcoded defaults
waypoints = {}
try:
with open('waypoints.yaml', 'r') as f:
waypoints = (yaml.safe_load(f) or {}).get('waypoints', {})
except FileNotFoundError:
with open('waypoints.yaml', 'w') as f:
yaml.dump({'waypoints': {}}, f, default_flow_style=False)
goal_A = [1.6766083240509033,0.37930558800697327,-0.03491306994337919, 0.9993903529387947]
goal_B = [-0.5062443017959595,1.559376835823059,0.6869307039904945,0.7267229237578264]
goals = {
'A': waypoints.get('A', [4.89649,-0.617371,0.706899,0.707315]),
'B': waypoints.get('B', [0.90387,-0.446105,0.273676,0.961822]),
'C': waypoints.get('C', [4.46734,-0.532388,0.969886,-0.243558]),
'D': waypoints.get('D', [-0.0233348,0.00798563,0.999322,0.0368173]),
'E': waypoints.get('E', [2.33651,-0.440663,0.937234,-0.3487]),
}
x_goal, y_goal, orientation_z, orientation_w = goal_A
success = navigate_to_goal(navigator, x_goal, y_goal, orientation_z, orientation_w)
print("Navigation result:", success)
x_goal, y_goal, orientation_z, orientation_w = goal_B
success = navigate_to_goal(navigator, x_goal, y_goal, orientation_z, orientation_w)
print("Navigation result:", success)
args = cli_args.targets
loop_targets = False
if args:
targets = []
for arg in args:
for c in arg.upper():
if c in goals:
targets.append(c)
else:
targets = ['A', 'B', 'C', 'D', 'E']
loop_targets = True
print('No target arguments provided; running A-B-C-D-E repeatedly. Press Ctrl+C to stop.')
if not targets:
print('No valid target names provided. Use names such as A, B, C, D, E or AB.')
rclpy.shutdown()
sys.exit(1)
try:
cycle_index = 1
while rclpy.ok():
if loop_targets:
print(f'============= cycle {cycle_index}: ABCDE =============')
for name in targets:
if not rclpy.ok():
break
x_goal, y_goal, orientation_z, orientation_w = goals[name]
input(f'============={name}==================\n')
goal_pose = make_goal_pose(navigator, x_goal, y_goal, orientation_z, orientation_w)
success = navigate_to_goal(navigator, goal_pose)
print("Navigation result:", goals[name], success)
if not loop_targets:
break
cycle_index += 1
except KeyboardInterrupt:
print('Navigation loop interrupted by user.')
navigator.cancelTask()
finally:
if rclpy.ok():
rclpy.shutdown()
+3
View File
@@ -0,0 +1,3 @@
from pymycobot import MyAGVPro
m = MyAGVPro('/dev/agvpro_controller')
m.power_on()
@@ -0,0 +1,119 @@
#!/usr/bin/env python3
"""Record waypoints by sampling stable map->base_footprint pose."""
import sys
import threading
import time
import rclpy
from rclpy.node import Node
from tf2_ros import Buffer, TransformListener
import yaml
class WaypointRecorder(Node):
def __init__(self):
super().__init__('waypoint_recorder')
self.tf_buffer = Buffer()
self.tf_listener = TransformListener(self.tf_buffer, self)
self.yaml_path = 'waypoints.yaml'
self.waypoints = {}
self._load_existing()
def _load_existing(self):
try:
with open(self.yaml_path, 'r') as f:
data = yaml.safe_load(f) or {}
self.waypoints = data.get('waypoints', {})
n = len(self.waypoints)
if n > 0:
self.get_logger().info(f'Loaded {n} existing waypoints from {self.yaml_path}')
except FileNotFoundError:
self.waypoints = {}
def _save(self):
data = {'waypoints': self.waypoints}
with open(self.yaml_path, 'w') as f:
yaml.dump(data, f, default_flow_style=False, sort_keys=False)
self.get_logger().info(f'Saved waypoints to {self.yaml_path}')
def _sample_pose(self, duration_sec=3.0, rate_hz=20):
xs, ys, zs, ws = [], [], [], []
dt = 1.0 / rate_hz
start = time.time()
while time.time() - start < duration_sec:
try:
trans = self.tf_buffer.lookup_transform(
'map', 'base_footprint', rclpy.time.Time()
)
t = trans.transform.translation
r = trans.transform.rotation
xs.append(t.x)
ys.append(t.y)
zs.append(r.z)
ws.append(r.w)
except Exception:
pass
time.sleep(dt)
if not xs:
return None
xs.sort()
ys.sort()
zs.sort()
ws.sort()
n = len(xs)
mid = n // 2
if n % 2 == 1:
return [xs[mid], ys[mid], zs[mid], ws[mid]]
return [
(xs[mid - 1] + xs[mid]) / 2,
(ys[mid - 1] + ys[mid]) / 2,
(zs[mid - 1] + zs[mid]) / 2,
(ws[mid - 1] + ws[mid]) / 2,
]
def run(self):
self.get_logger().info('Waypoint recorder ready.')
self.get_logger().info('Enter A/B/C/D/E to record, q to quit.')
while rclpy.ok():
try:
cmd = input('> ').strip().upper()
except EOFError:
break
if cmd == 'Q':
break
if cmd in 'ABCDE':
self.get_logger().info(
f'Sampling pose for {cmd} ({3}s, keep still)...'
)
pose = self._sample_pose()
if pose is None:
self.get_logger().error(
'Failed to sample pose. Is AMCL running?'
)
continue
self.waypoints[cmd] = [float(f'{v:.6f}') for v in pose]
self.get_logger().info(f'{cmd}: {self.waypoints[cmd]}')
self._save()
else:
self.get_logger().warn('Use A/B/C/D/E or q.')
def main(args=None):
rclpy.init(args=args)
node = WaypointRecorder()
spin_thread = threading.Thread(target=rclpy.spin, args=(node,), daemon=True)
spin_thread.start()
try:
node.run()
except KeyboardInterrupt:
pass
finally:
node.destroy_node()
rclpy.shutdown()
if __name__ == '__main__':
main()
@@ -0,0 +1,26 @@
waypoints:
A:
- 4.24531
- 1.866955
- 0.584964
- 0.811059
B:
- 3.624886
- -2.221018
- -0.569358
- 0.82209
C:
- 1.900274
- 1.940145
- -0.737521
- 0.675324
D:
- 0.882323
- -0.456797
- 0.745742
- 0.666235
E:
- -1.63798
- -0.847909
- 0.978385
- 0.206792