"""This example illustrates the usage of creator, evaluator and generator.

All the steps of airfoil creation & evaluation are detailed here;
furthermore, the generator.py module contains certain presets
(default airfoils).

Create an airfoil;
Evaluate an airfoil;
Generate a population of airfoils & optimize.
"""

import matplotlib.pyplot as plt

import creator
import evaluator
import generator
import resources.materials as mt

import time
start_time = time.time()

# Thicknesses
SPAR_THICKNESS = 0.4
SKIN_THICKNESS = 0.1

# Component masses (lbs)
AIRFOIL_MASS = 10
SPAR_MASS = 10
STRINGER_MASS = 5

# Area (sqin)
SPAR_CAP_AREA = 0.3
STRINGER_AREA = 0.1

# Amount of stringers
TOP_STRINGERS = 6
BOTTOM_STRINGERS = 4
NOSE_TOP_STRINGERS = 3
NOSE_BOTTOM_STRINGERS = 5

SAVE_PATH = '/home/blendux/Projects/Aircraft_Studio/save'

eval = evaluator.Evaluator("eval")

ac = creator.base.Aircraft(eval, "ac")
af = creator.wing.Airfoil(ac, 'af')
af.add_naca(2412)
spar1 = creator.wing.Spar(af, 'spar1')
spar2 = creator.wing.Spar(af, 'spar2', 0.57)
# spar2 = creator.wing.Spar(af, 'spar2', 0.7)
stringer = creator.wing.Stringer(af, 'stringer', 5, 6, 5, 4)
stringer.info_save(SAVE_PATH)

ac2 = creator.base.Aircraft(eval, "ac2")
af2 = creator.wing.Airfoil(ac2, 'af2')
af2.add_naca(3412)
af2.info_save()
spar3 = creator.wing.Spar(af2, 'spar3', 0.23)
spar4 = creator.wing.Spar(af2, 'spar4', 0.67)
stringer2 = creator.wing.Stringer(af2, 'stringer2')
stringer2.info_save(SAVE_PATH)


for _ in range(5):
    aircraft = generator.default_aircraft(eval)
    aircraft2 = generator.default_aircraft(eval)
    eval.tree_print(aircraft, aircraft2)
    eval.tree_save(aircraft)

print(eval.analysis())

# creator.wing.plot_geom(af)
# eval.tree_print()

# Final execution time
final_time = time.time() - start_time
print(f"--- {round(final_time, 4)}s seconds ---")