wehub-resource-sync
45 lines
1.1 KiB
Python
Executable File
45 lines
1.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Demonstration of using rerun from multiple threads."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import random
|
|
import threading
|
|
|
|
import numpy as np
|
|
import numpy.typing as npt
|
|
|
|
import rerun as rr # pip install rerun-sdk
|
|
|
|
|
|
def rect_logger(path: str, color: npt.NDArray[np.float32]) -> None:
|
|
for _ in range(1000):
|
|
rects_xy = np.random.rand(5, 2) * 1024
|
|
rects_wh = np.random.rand(5, 2) * (1024 - rects_xy + 1)
|
|
rects = np.hstack((rects_xy, rects_wh))
|
|
rr.log(path, rr.Boxes2D(array=rects, array_format=rr.Box2DFormat.XYWH, colors=color))
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Demonstration of using Rerun from multiple threads.")
|
|
rr.script_add_args(parser)
|
|
args = parser.parse_args()
|
|
|
|
rr.script_setup(args, "rerun_example_multithreading")
|
|
|
|
threads = []
|
|
for i in range(10):
|
|
t = threading.Thread(target=rect_logger, args=(f"thread/{i}", [random.randrange(255) for _ in range(3)]))
|
|
t.start()
|
|
threads.append(t)
|
|
|
|
for t in threads:
|
|
t.join()
|
|
|
|
rr.script_teardown(args)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|