import glob
import os
import time

import pytest

from pymskt.statistics import ProcrustesRegistration

file_path = os.path.abspath(__file__)
# Construct the path to the package directory
package_dir = os.path.dirname(os.path.dirname(os.path.dirname(file_path)))

LIST_MESH_PATHS = [
    os.path.join(
        package_dir, "data", "femur_meshes_registration", f"{idx}_RIGHT_femur_Nov_02_2021.vtk"
    )
    for idx in range(1, 6)
]
REF_MESH_PATH = LIST_MESH_PATHS.pop(0)

for path in LIST_MESH_PATHS:
    print(os.path.exists(path))


@pytest.mark.skip(reason="Takes too long - and probably still fails?")
def test_ProcrustesRegistration_runs():
    folder_save = os.path.join(
        package_dir, "data", "femur_meshes_registration", "registered_meshes"
    )

    procrustes_reg = ProcrustesRegistration(
        path_ref_mesh=REF_MESH_PATH,  # using the idx of the best mesh from the previous step
        list_mesh_paths=LIST_MESH_PATHS,  # This will automatically remove the ref_mesh path if it is in the list.
        max_n_registration_steps=1,
        n_coords_spectral_ordering=10000,
        n_coords_spectral_registration=1000,
        n_extra_spectral=6,
        include_points_as_features=True,
        vertex_features=["thickness (mm)"],
        multiprocessing=True,
        verbose=False,
        save_meshes_during_registration=True,
        folder_save=folder_save,
    )

    tic = time.time()
    procrustes_reg.execute()
    toc = time.time()

    print(f"Time taken for ProcrustesRegistration: {toc - tic} seconds")

    # procrustes_reg.save_meshes(folder=folder_save)

    list_meshes = glob.glob(os.path.join(folder_save, "*.vtk"))
    assert len(list_meshes) == len(LIST_MESH_PATHS)

    # delete the folder of registered meshes
    os.system(f"rm -r {folder_save}")

    print(procrustes_reg.registered_pt_coords.shape)
    print(procrustes_reg.registered_vertex_features.shape)


if __name__ == "__main__":
    # pytest.main([__file__])
    test_ProcrustesRegistration_runs()