# ENGG4900 Graph Generators
# Cal Wing 2023

import numpy as np
import pandas as pd
from makeGraph import makeGraph, UQ_COLOURS as UQC

DATA_FILE_PATH = ".\\data\\AERO4450-Group-10-Data-Analysis-v2.xls"
actualData = pd.read_excel(DATA_FILE_PATH, sheet_name="Actual Data")
calibrationData = pd.read_excel(DATA_FILE_PATH, sheet_name="Calibration Data Graphs")


def main():
    testRun = 0

    cols = [f"Unnamed: {3 + testRun*5}", f"Unnamed: {4+ testRun*5}"]
    data = actualData[cols].rename(columns = {cols[0]:'Time', cols[1]:'Strain'})
    
    data = data[5:505].astype({"Time":"int", "Strain":"float"})
    data1 = data[0::2]
    data2 = data[1::2]
    

    makeGraph({
        "title": f"Strain over Time\nRun: {testRun+1}",
        "xLabel": "Time (ms) - Δt = 10ms", 
        "yLabel": "Strain (ε)",
        "plots":[
            {"type":"scatter", "x": data1["Time"], "y": data1["Strain"], "args":{"s":10, "zorder":2}, "label":"Upper", "colour": UQC["purple"]},
            {"type":"scatter", "x": data2["Time"], "y": data2["Strain"], "args":{"s":10, "zorder":1}, "label":"Lower", "colour": UQC["dark_grey"]},
            {"type":"plot", "x": data1["Time"], "y": data1["Strain"], "args":{"zorder":0, "alpha":0.25}, "label":"Upper", "colour": UQC["purple"]},
            {"type":"plot", "x": data2["Time"], "y": data2["Strain"], "args":{"zorder":0, "alpha":0.25}, "label":"Lower", "colour": UQC["dark_grey"]},

        ]
    })
    

if __name__ == "__main__":
    main()