How to use WS2812B LED lights with ROS 2 node

edit

You need to Install the rpi_ws281x library on the Raspberry Pi.

pip install rpi-ws281x==1.1.1

And write a ROS 2 node in Python that imports the rpi_ws281x library and controls the WS2812B LED lights in response to ROS 2 messages.

import rclpy
from std_msgs.msg import ColorRGBA
from rpi_ws281x import Adafruit_NeoPixel, Color

# Define the number of WS2812B LED lights and the pin they are connected to
NUM_LEDS = 30  # Number of LED
LED_PIN = 18  # 0- number will form a color

# Initialize the WS2812B LED lights
strip = Adafruit_NeoPixel(NUM_LEDS, LED_PIN)
strip.begin()
strip.show()

# Callback function for the color message
def color_callback(msg):
    # Set the color of all the WS2812B LED lights
    for i in range(NUM_LEDS):
        strip.setPixelColor(i, Color(int(msg.r * 255), int(msg.g * 255), int(msg.b * 255)))
    strip.show()

def main(args=None):
    rclpy.init(args=args)
    node = rclpy.create_node('minimal_subscriber')
    subscription = node.create_subscription(ColorRGBA, 'color', color_callback, 10)
    subscription  # prevent unused variable warning
    rclpy.spin(node)
    node.destroy_node()
    rclpy.shutdown()

if __name__ == '__main__':
    main()

I hope it helps, fell free to drop a comment in case of questions.