from utils.data_processing.data_processing import get_discrete_data, get_raw_data
from utils.files import output
from utils.files.input import ScannedObject
from utils.math.position_manipulation import verticalise
import time

def main():
    # Create an object from the given file
    total_time = time.time()
    print("Loading file...")
    t = time.time()
    obj = ScannedObject.from_obj_file("datasets/Barette/1 - BARETTE.obj")
    print("File loaded in {} seconds".format(time.time()-t))
    t = time.time()
    print("Verticalising object...")
    verticalise(obj)
    print("Object verticalised in {} seconds".format(time.time()-t))
    t = time.time() 
    print("Normalising object...")
    obj.normalise()
    print("Object normalised in {} seconds".format(time.time()-t))

    # Calculate raw data and save it in a file
    t = time.time()
    data = get_raw_data(obj, 6,2)
    print("Raw data calculated in {} seconds".format(time.time()-t))
    t = time.time()
    print("Saving data...")
    output.save_output_file('analyse_brute.txt',
                             output.format_data(data,
                                                '\t',
                                                ["X (en mm)",
                                                "Y (en mm)",
                                                "Z (en mm)",
                                                "teta (en rad)",
                                                "rayon (en mm)",
                                                "Xi-Xmoy",
                                                "Yi-Ymoy"] ))
    print("Data saved in {} seconds".format(time.time()-t))
    
    # Calculate discrete data and save it in a file
    t = time.time()
    print("Calculating discrete data...")
    data = get_discrete_data(obj, 6,2)
    print("Discrete data calculated in {} seconds".format(time.time()-t))
    t = time.time()
    print("Saving data...")
    output.save_output_file('analyse_rayon.txt',
                             output.format_data(data,
                                                '\t',
                                                ["X moy (en mm)",
                                                "Y moy (en mm)",
                                                "Z moy (en mm)",
                                                "Delta z(en mm)",
                                                "Rayon moyen (en mm)",
                                                "Rayon ecart type (en mm)"] ))
    print("Data saved in {} seconds".format(time.time()-t))
    print("Total time : {} seconds".format(time.time()-total_time))

if __name__ == '__main__':
    main()