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

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

from tqdm import tqdm

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"]
for folder in folders:
    if not os.path.isdir(folder): os.mkdir(folder)


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

PITOT_PLACEMENT = ( # mm from base
    0, 
    4,
    8,
    16, 
    25, 
    34,
    43,
    53,
    61,
    70,
    5,
    9,
    17,
    25, 
    34,
    43,
    53,
    61,
    70,
)

print("Loading Data")
data = pd.read_excel('.\\data\\508 RPM Results.xlsx', sheet_name=None, header=None)
print("Loaded Data")

sheet1 = data['0 AoA']
#print(sheet1)

air_speed = 10 # m/s
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())


makeGraph({
            "title":  "Simple Plot", 
            "xLabel": "x label", 
            "yLabel": "y label",
            "plots": [
                {"x":PITOT_PLACEMENT, "y":pressure, "label":"Linear", "type":"scatter"},
            ]
        })

def main():
    pass

if __name__ == '__main__':
    main()