dora/examples/python-ros2-dataflow/control_node.py

#!/usr/bin/env python
"""TODO: Add docstring."""


import random

import pyarrow as pa
from dora import Node

node = Node()

for i in range(500):
    event = node.next()
    if event is None:
        break
    if event["type"] == "INPUT":
        event_id = event["id"]
        if event_id == "turtle_pose":
            print(
                f"""Pose: {event["value"].tolist()}""".replace("\r", "").replace(
                    "\n", " ",
                ),
            )
        elif event_id == "tick":
            direction = {
                "linear": {
                    "x": 1.0 + random.random(),
                },
                "angular": {"z": (random.random() - 0.5) * 5},
            }

            node.send_output(
                "direction",
                pa.array([direction]),
            )