tf2.md

rclpy: TF2

The TF2 library provides easy access to transformations. All of the examples below require a dependency on the tf2_ros package.

Listening for Transforms

import rclpy
from rclpy.node import Node
from tf2_ros.buffer import Buffer
from tf2_ros.transform_listener import TransformListener

class MyNode(Node):
    def __init__(self):
        super().__init__("my_node")

        self.buffer = Buffer()
        self.listener = TransformListener(self.buffer, self)

Applying Transforms

TF2 can be extended by packages that provide implementations of transform. The _tf2_geometry_msgs package provides these for geometry_msgs. The example below uses geometry_msgs.msg.PointStamped - but this should work for any data type in geometry_msgs:

from geometry_msgs.msg import PointStamped
from tf2_ros.buffer import Buffer
from tf2_ros.transform_listener import TransformListener
import tf2_geometry_msgs

# Setup buffer/listener as above

p1 = PointStamped()
p1.header.frame_id = "source_frame"
# fill in p1

p2 = buffer.transform(p1, "target_frame")

Transformations

In ROS1, tf included the transformations module. tf2 has no similar module. It is recommended to use transforms3d Python package, which is available through apt on Ubuntu 22.04:

sudo apt-get install python3-transforms3d

Or via pip on other operating systems:

sudo pip3 install transforms3d

For instance, to rotate a point:

import numpy as np
from transforms3d.quaternion import quat2mat

# Create rotation matrix from quaternion
R = quat2mat([w, x, y, z])
# Create a vector to rotate
V = np.array([x, y, z]).reshape((3, 1))
# Rotate the vector
M = np.dot(R, V)

p = PointStamped()
p.point.x = M[0, 0]
p.point.y = M[1, 0]
p.point.z = M[2, 0]

Additionally, in ROS2 Humble there is the tf_transformations package which should make it as easy as changing your imports from tf.transformations to tf_transformations.