import os,sys
import numpy as np
from slam.slam import compose_ply
from utils import read_ply,write_ply,file_fliter

def run(folder,alpha,beta,gamma,savename):
    
    # literate the folder
    dir_List_ = os.listdir(folder)
    sort_lambda = lambda x:int(x.split("°")[0])
    dir_List_.sort(key=sort_lambda)
    
    dir_List=[]
    for k in dir_List_:
        dir_List.append(k[:-1])


    angle = []
    index = 0 
    for i,degree in enumerate([alpha,beta,gamma]):
        if degree == "auto":
            degree = dir_List
            index = i
        angle.append(degree)

    data = []
    
    # iterate the auto angle degree
    for an in angle[index]:
        ply_with_mutiangle = []
        for z in angle:
            if isinstance(z,list):
                z = an
            ply_with_mutiangle.append(z)
        ply_name_list = file_fliter(os.path.join(folder,an+"°"), "ply",lambda x:int(x.split(".")[0]))
        # select the first ply file
        ply = read_ply(os.path.join(folder,an+"°",ply_name_list[0]))
        ply_with_mutiangle.append(ply)
        data.append(ply_with_mutiangle)

    # compose muti ply files to one ply file
    cloudpoint = compose_ply(data)

    # write the file
    write_ply(cloudpoint,".", savename)






if __name__ == "__main__":
    def help():
        print("Usage: mannul_slam.py [folder] [alpha] [beta] [gamma] [savename]")
        print("EXAMPLE: mannul_slam.py T:\\Data\\chuizhi60°_shuiping120° 0 60 auto test60120")
        exit(1)

    if len(sys.argv) != 6:
        help()
    try:
        folder = str(sys.argv[1])
        alpha = str(sys.argv[2])
        beta = str(sys.argv[3])
        gamma = str(sys.argv[4])
        savename = str(sys.argv[5])

    except Exception as e:
        print(e)
        help()

    run(folder,alpha,beta,gamma,savename)
    print("Done")