wehub-resource-sync
66 lines
2.1 KiB
Python
Executable File
66 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Shows how rerun can work with multiprocessing."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import multiprocessing
|
|
import os
|
|
import threading
|
|
|
|
import rerun as rr # pip install rerun-sdk
|
|
|
|
|
|
# Python does not guarantee that the normal atexit-handlers will be called at the
|
|
# termination of a multiprocessing.Process. Explicitly add the `shutdown_at_exit`
|
|
# decorator to ensure data is flushed when the task completes.
|
|
@rr.shutdown_at_exit # type: ignore[misc]
|
|
def task(child_index: int) -> None:
|
|
# In the new process, we always need to call init with the same `application_id`.
|
|
# By default, the `recording_id`` will match the `recording_id`` of the parent process,
|
|
# so all of these processes will have their log data merged in the viewer.
|
|
# Caution: if you manually specified `recording_id` in the parent, you also must
|
|
# pass the same `recording_id` here.
|
|
rr.init("rerun_example_multiprocessing")
|
|
|
|
# We then have to connect to the viewer instance.
|
|
rr.connect_grpc()
|
|
|
|
title = f"task_{child_index}"
|
|
rr.log(
|
|
"log",
|
|
rr.TextLog(
|
|
f"Logging from pid={os.getpid()}, thread={threading.get_ident()} using the rerun recording id {rr.get_recording_id()}",
|
|
),
|
|
)
|
|
if child_index == 0:
|
|
rr.log(title, rr.Boxes2D(array=[5, 5, 80, 80], array_format=rr.Box2DFormat.XYWH, labels=title))
|
|
else:
|
|
rr.log(
|
|
title,
|
|
rr.Boxes2D(
|
|
array=[10 + child_index * 10, 20 + child_index * 5, 30, 40],
|
|
array_format=rr.Box2DFormat.XYWH,
|
|
labels=title,
|
|
),
|
|
)
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Test multi-process logging to the same Rerun server")
|
|
parser.parse_args()
|
|
|
|
rr.init("rerun_example_multiprocessing")
|
|
rr.spawn(connect=False) # this is the viewer that each child process will connect to
|
|
|
|
task(0)
|
|
|
|
for i in [1, 2, 3]:
|
|
p = multiprocessing.Process(target=task, args=(i,))
|
|
p.start()
|
|
p.join()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|