# Mech3410-P1-Graphs
#   Cal Wing - Aug 2024
#

import os, time
import numpy as np
import pandas as pd

from tqdm import tqdm
from numpy import sqrt

from makeGraph import makeGraph, pltKeyClose, UQ_COLOURS as UQC # Custom Graphing Lib

# Override Sin & Cos to use & return degrees
#def sin(angle): return np.sin(np.deg2rad(angle))
#def cos(angle): return np.cos(np.deg2rad(angle))

# Make sure the relevant folders folder exists
#folders = ["./images", "./tmp", "./data"]
folders = ["./images", './images/pressure', './images/cp']
for folder in folders:
    if not os.path.isdir(folder): os.mkdir(folder)


INCH_TO_M = 0.0254
GRAVITY = 9.81 #m/s^2

CHORD_LEN = 90 #mm
PITOT_PLACEMENT = np.array((0,4,8,16,25,34,43,53,61,70,5,9,17,25,34,43,53,61,70)) # mm from base of chord
PITOT_PLACEMENT_CHORD_NORM = PITOT_PLACEMENT / CHORD_LEN

print("="*15, "Loading Data", "="*15)
data_508rpm = {
    "rpm": 508,
    "airSpeed": 10,
    "data": pd.read_excel('.\\data\\508 RPM Results.xlsx', sheet_name=None, header=None),
    "AoA": (0,1,2,3,4,5,6,7,8,9)
}

data_1000rpm = {
    "rpm": 1000,
    "airSpeed": 20.40408122,
    "data": pd.read_excel('.\\data\\1000 RPM Results.xlsx', sheet_name=None, header=None),
    "AoA": (1,2,4,6,8,10,12,14,16)
}

print("="*15, "Loaded Data", "="*15)

def make_pressure_graph(sheet1, aoa, rpm, air_speed, doGraph=True):
    water_density = sheet1.iloc[0, 2] # kg/m^3
    air_density = sheet1.iloc[1, 2] # kg/m^3
    atm_presure_inch = sheet1.iloc[24, 12] # inch

    pitot_height_inch = sheet1.iloc[4:23, 11]
    pitot_height_m = (pitot_height_inch - atm_presure_inch)*INCH_TO_M

    pressure = water_density * GRAVITY * pitot_height_m

    #print(pressure)
    #print(pressure.min(), pressure.max())


    graph = {
                "title":  f"Pressure vs Pitot Placement along Chord\nfor a Clark Y 14% Aerofoil at:\nα = {int(aoa):d}° at {rpm:d} RPM ({air_speed:.1f}m/s)", 
                "windowTitle": f"Pressure along Clark Y 14% Airfoil - Alpha {int(aoa):d} Degrees - {rpm:d}rpm - {air_speed:.1f}m_s",
                "xLabel": "Pitot Placement [mm]", 
                "yLabel": "Pressure [Pa]",
                "grid": True,
                "yLim": (pressure.min()-10, pressure.max()+10),
                "plots": [
                    {"x": PITOT_PLACEMENT[:10], "y":pressure[:10], "colour": UQC["purple"]},
                    {"x": PITOT_PLACEMENT[10:], "y":pressure[10:], "colour": UQC["purple"]},
                    {"x": [PITOT_PLACEMENT[0], PITOT_PLACEMENT[10]],  "y": [pressure.iloc[0], pressure.iloc[10]], "colour": UQC["purple"]},
                    {"x": [PITOT_PLACEMENT[9], PITOT_PLACEMENT[18]], "y": [pressure.iloc[9], pressure.iloc[18]], "colour": UQC["purple"]},
                    
                    {"x": PITOT_PLACEMENT[:10], "y":pressure[:10], "colour": UQC["purple"], "alpha":0.2, "type":"fill"},
                    {"x": PITOT_PLACEMENT[10:], "y":pressure[10:], "colour": "w", "type":"fill"},
                    {"x":[PITOT_PLACEMENT[0], PITOT_PLACEMENT[10]], "y":[pressure.iloc[0], pressure.iloc[10]], "colour": "w", "type":"fill"},


                    {"x":PITOT_PLACEMENT, "y":pressure, "label":"Pressure Data", "type":"scatter"},
                    {"label":[str(i) for i in range(1, len(PITOT_PLACEMENT)+1)], "x":PITOT_PLACEMENT, "y":pressure, "type":"annotate"}
                ]
            }
    
    if doGraph:
        makeGraph(graph, False, figSavePath=f'./images/pressure/{{0}}.png')
    
    return pressure, graph, (water_density, air_density, atm_presure_inch, pitot_height_inch)


if __name__ == '__main__':
    print("Loading Data & Generating Pressure Graphs")
    data = {}
    for raw_data in tqdm((data_508rpm, data_1000rpm), position=0):
        aoa_data = {}
        for aoa in tqdm(raw_data["AoA"], position=1):
            sheet = raw_data["data"][f"{aoa} AoA"]

            aoa_data[aoa] = make_pressure_graph(sheet, aoa, raw_data["rpm"], raw_data["airSpeed"], False)
        
        data[raw_data["rpm"]] = aoa_data

    print("Data Loaded")