evalio_2.0_file_renaming.py

import os
from pathlib import Path

EVALIO_DATA = os.environ.get("EVALIO_DATA")

if EVALIO_DATA is None:
    raise ValueError(
        "EVALIO_DATA environment variable is not set. Please set it to the path where evalio data is stored."
    )

# Fix newer_college_2020 dataset
newer_college_path = Path(EVALIO_DATA) / "newer_college_2020"
if newer_college_path.exists() and newer_college_path.is_dir():
    print("Fixing newer_college_2020 dataset...")
    for subdirectory in newer_college_path.iterdir():
        if subdirectory.is_dir():
            for item in subdirectory.iterdir():
                if item.name == "ground_truth.csv":
                    print(
                        f"---Moving {item} to {subdirectory / 'registered_poses.csv'}"
                    )
                    item.rename(subdirectory / "registered_poses.csv")

# Fix newer_college_2021 dataset
newer_college_2021_path = Path(EVALIO_DATA) / "newer_college_2021"
if newer_college_2021_path.exists() and newer_college_2021_path.is_dir():
    print("Fixing newer_college_2021 dataset...")
    for subdirectory in newer_college_2021_path.iterdir():
        if subdirectory.is_dir():
            for item in subdirectory.iterdir():
                if item.name == "ground_truth.csv":
                    if "maths" in subdirectory.name:
                        difficulty = subdirectory.name.split("_")[1]
                        gt_file = f"gt_state_{difficulty}.csv"
                    else:
                        name = subdirectory.name.replace("_", "-")
                        gt_file = f"gt-nc-{name}.csv"

                    print(f"---Moving {item} to {subdirectory / gt_file}")
                    item.rename(subdirectory / gt_file)


# Fix multi_campus dataset
multi_campus_path = Path(EVALIO_DATA) / "multi_campus"
if multi_campus_path.exists() and multi_campus_path.is_dir():
    print("Fixing multi_campus dataset...")
    for subdirectory in multi_campus_path.iterdir():
        if subdirectory.is_dir():
            if "ntu" in subdirectory.name:
                beams = 128
                imu = "vn100"
            else:
                beams = 64
                imu = "vn200"

            for item in subdirectory.iterdir():
                if item.name == "ouster.bag":
                    ouster_name = f"{subdirectory.name}_os1_{beams}.bag"
                    print(f"---Moving {item} to {subdirectory / ouster_name}")
                    item.rename(subdirectory / ouster_name)

                elif item.name == "vectornav.bag":
                    vectornav_name = f"{subdirectory.name}_{imu}.bag"
                    print(f"---Moving {item} to {subdirectory / vectornav_name}")
                    item.rename(subdirectory / vectornav_name)

# fix oxford spires
oxford_spires_path = Path(EVALIO_DATA) / "oxford_spires"
if oxford_spires_path.exists() and oxford_spires_path.is_dir():
    print("Fixing oxford spires dataset...")
    for subdirectory in oxford_spires_path.iterdir():
        if subdirectory.is_dir():
            for item in subdirectory.iterdir():
                if item.name == "gt-tum.csv":
                    print(f"---Moving {item} to {subdirectory / 'gt-tum.txt'}")
                    item.rename(subdirectory / "gt-tum.txt")