Function get_clean_mesh_filenames

pointersect/data/dataset_helper.py:106–147

Get the clean the meshes' filenames in mesh_filenames. Args: mesh_filenames: a list containing the meshes to be cleaned. The mesh_filenames are relative filepath within `dataset_root_dir`. cleaned_root_dir: only for cleaning obj Retu

(
        mesh_filenames: T.List[str],
        dataset_root_dir: str,
        cleaned_root_dir: str,
)

def get_clean_mesh_filenames(
        mesh_filenames: T.List[str],
        dataset_root_dir: str,
        cleaned_root_dir: str,
) -> T.List[str]:
    """
    Get the clean the meshes' filenames in mesh_filenames.

    Args:
        mesh_filenames:
            a list containing the meshes to be cleaned.
            The mesh_filenames are relative filepath within `dataset_root_dir`.
        cleaned_root_dir:
            only for cleaning obj

    Returns:
        cleaned_mesh_filenames:
            a list containing the cleaned meshes
    """

    if isinstance(mesh_filenames, str):  # str, list, or tuple
        mesh_filenames = [mesh_filenames]

    cleaned_mesh_filenames = []
    for mesh_filename in mesh_filenames:
        name, ext = os.path.splitext(mesh_filename)
        if ext.lower() == '.obj':
            rel_mesh_filename = os.path.relpath(mesh_filename, dataset_root_dir)
            cleaned_filename = os.path.join(
                cleaned_root_dir,
                rel_mesh_filename,
            )
        elif ext.lower() in {'.glb', '.gltf'}:
            name, ext = os.path.splitext(mesh_filename)
            cleaned_filename = f'{name}.gltf'
        else:
            cleaned_filename = mesh_filename

        if os.path.exists(cleaned_filename):
            cleaned_mesh_filenames.append(cleaned_filename)

    return cleaned_mesh_filenames


def gather_and_clean_dataset(