Function clean_meshes

pointersect/data/dataset_helper.py:39–103

Clean the meshes in mesh_filenames. Note that the actual cleaning will only be executed by rank=0. The rest of the ranks should wait for rank=0 to finish. Args: mesh_filenames: a list containing the meshes to be cleaned. The mesh_filenames are relat

(
        mesh_filenames: T.List[str],
        dataset_root_dir: str,
        cleaned_root_dir: str,
        skip_exist: bool,
        printout: bool = True,
)

def clean_meshes(
        mesh_filenames: T.List[str],
        dataset_root_dir: str,
        cleaned_root_dir: str,
        skip_exist: bool,
        printout: bool = True,
) -> T.List[str]:
    """
    Clean the meshes in mesh_filenames.

    Note that the actual cleaning will only be executed by rank=0.
    The rest of the ranks should wait for rank=0 to finish.

    Args:
        mesh_filenames:
            a list containing the meshes to be cleaned.
            The mesh_filenames are relative filepath within `dataset_root_dir` or full path.
        dataset_root_dir:
            the root dir where all meshes locate
        cleaned_root_dir:
            only for cleaning obj

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

    # print(f'clean_meshes: dataset_root_dir = {dataset_root_dir}', flush=True)
    # print(f'clean_meshes: cleaned_root_dir = {cleaned_root_dir}', flush=True)

    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 = preprocess_obj.generated_cleaned_mesh(
                input_dir=dataset_root_dir,
                output_dir=cleaned_root_dir,
                relative_filepath=rel_mesh_filename,
                skip_exist=skip_exist,
                printout=printout,
            )
        elif ext.lower() in {'.glb', '.gltf'}:
            cleaned_filename = utils.clean_up_glb_write_gltf(
                filename_glb=mesh_filename,
                overwrite_gltf=not skip_exist,
                exists_ok=True,
                image_ext='.png',  # assume the texture maps are of png format
            )
        elif ext.lower() in {'off'}:
            # off meshes do not need cleaning
            cleaned_filename = mesh_filename
        else:
            warnings.warn(f'{mesh_filename} not cleaned')
            cleaned_filename = mesh_filename