From a46c8d5a1a3ea71f21bc7a3d0af9db5b29604884 Mon Sep 17 00:00:00 2001 From: Meher Malladi Date: Fri, 21 Feb 2025 10:29:48 +0100 Subject: [PATCH] fix conversion to ros2, because rosbags api changed --- python/vbr_devkit/tools/run.py | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/python/vbr_devkit/tools/run.py b/python/vbr_devkit/tools/run.py index de7a7ee..cc20bcb 100644 --- a/python/vbr_devkit/tools/run.py +++ b/python/vbr_devkit/tools/run.py @@ -75,13 +75,20 @@ def convert(to: Annotated[OutputDataInterface, typer.Argument(help="Desired data show_default=True)] = True) -> None: console.print(f"Converting {input_dir} to {to} format at {output_dir}") if to == OutputDataInterface.ros2: - if not input_dir.is_dir(): - print("Processing...") - rosconvert.convert(input_dir, output_dir / input_dir.stem) - else: - for item in track(list(input_dir.iterdir()), description="Processing..."): - if item.suffix == '.bag': - rosconvert.convert(item, output_dir / item.stem) + print("Processing...") + rosconvert.convert( + sorted([f for f in input_dir.glob("*.bag")]) if input_dir.is_dir() else [input_dir], + output_dir / input_dir.stem, + dst_version=None, + compress=None, + compress_mode="file", + default_typestore=None, + typestore=None, + exclude_topics=[], + include_topics=[], + exclude_msgtypes=[], + include_msgtypes=[] + ) else: with RosReader(input_dir) as reader: with OutputDataInterface_lut[to](output_dir, rgb_convert=rgb_conversion,