Week 1 Publisher Node
The Code¶
Copy all the code below into your publisher.py
file and review the annotations to understand how it all works.
Tip
Don't forget the Shebang! See below for further details...
#!/usr/bin/env python3
# A simple ROS publisher node in Python
import rospy # (1)!
from std_msgs.msg import String # (2)!
class Publisher(): # (3)!
def __init__(self): # (4)!
self.node_name = "simple_publisher" # (5)!
topic_name = "chatter" # (6)!
self.pub = rospy.Publisher(topic_name, String, queue_size=10) # (7)!
rospy.init_node(self.node_name, anonymous=True) # (8)!
self.rate = rospy.Rate(10) # (9)!
self.ctrl_c = False # (10)!
rospy.on_shutdown(self.shutdownhook)
rospy.loginfo(f"The '{self.node_name}' node is active...") # (11)!
def shutdownhook(self): # (12)!
print(f"Stopping the '{self.node_name}' node at: {rospy.get_time()}")
self.ctrl_c = True
def main_loop(self):
while not self.ctrl_c: # (13)!
publisher_message = f"rospy time is: {rospy.get_time()}"
self.pub.publish(publisher_message)
self.rate.sleep()
if __name__ == '__main__': # (14)!
publisher_instance = Publisher() # (15)!
try:
publisher_instance.main_loop() # (16)!
except rospy.ROSInterruptException:
pass
-
rospy
is the Python client library for ROS, and we need to import this in order to create ROS Nodes in Python. -
We also need to import the
String
message type from thestd_msgs.msg
library for publishing our messages. -
We create a Python class called
Publisher()
to encapsulate all the functionality of our node. -
The
__init__()
method is called as soon as an instance of thePublisher()
class is created. -
We define a name for this node and assign it to
self.node_name
. We can call the node anything that we want, but it's good to give it a meaningful name as this is the name that will be used to register the node on the ROS Network. -
We also define the name of a topic that we want to publish messages to (
"chatter"
in this case). If this is the name of a topic that already exists on the ROS Network, then we need to ensure that we use the correct message type, in order to be able to publish to it. In this case however, a topic called "chatter" shouldn't currently exist, so it will be created for us, and we can choose whatever type of message we want to use (aString
message from thestd_msgs
library in our case). -
We then create an instance of a
rospy.Publisher()
object within our class: this creates the topic on the ROS Network (if it doesn't already exist). We therefore need to tell it the name of the topic that we want to create, and we also need to specify that we will be publishingString
type messages. -
Then, we initialise our Python node, using the name that we defined earlier (
"simple_publisher"
), setting theanonymous
flag toTrue
to ensure that the node name is unique, by appending random numbers to it (we'll observe this when we run the node shortly). -
Then, we instantiate a
rospy.Rate()
object and set the frequency to 10 Hz, so that our publisher will publish messages at this frequency. -
This is used to shut down a ROS node effectively:
- First, we create a
ctrl_c
variable within the parent class and initialise it toFalse
. - Then, we use the
rospy.on_shutdown()
method to register a shutdown hook (in this case a function calledshutdownhook
). This will be called when rospy detects that the node has been asked to stop (i.e. by a user enteringCtrl+C
in the terminal, for example). The shutdown hook function must take no arguments.
- First, we create a
-
Finally, we issue a message to indicate that our node is active (this will appear in the terminal that we run the node in):
-
This method is called by the
rospy.on_shutdown()
method when the node is stopped.Here, we can include any important shutdown processes (making a robot stop moving, for instance). In this case, we just print a message to the terminal and then set the
ctrl_c
variable toTrue
, which will stop themain_loop()
method... -
The
while
loop here makes sure that themain_loop()
runs continuously, until the node is shut down (via theself.ctrl_c
flag):Inside the
while
loop we create a publisher message (a simple string in this case), publish it using thepub
object we created in the initialisation stage, and then use therate
object (also created earlier) to then make the node "sleep" for as long as required to satisfy the frequency that we defined earlier. -
This
__name__
check, ensures that our node is the main executable (i.e. it has been executed directly (viarosrun
), and hasn't been called by another script): -
We create an instance of the
Publisher
class that we created above (which executes the__init__
method automatically). -
We call the
main_loop()
of ourPublisher()
class to execute the core functionality of the node. We wrap this inside atry-except-pass
statement to look for arospy.ROSInterruptException
error, which can be output by rospy when the user pressesCtrl+C
or the node is shutdown in some other way.
The Shebang¶
The very first line of code looks like a comment, but it is actually a very crucial part of the script:
This is called the Shebang, and it tells the operating system which interpreter to use to execute the code. In our case here, it tells the operating system where to find the right Python interpreter that should be used to actually run the code.