# %%
import pvlib
import h5py
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

from pathlib import Path

keys = [
    "Canadian_270_poly",
    "Canadian_275_mono",
    "Itek_360_mono",
    "Jinko_260_poly",
    "LG_320_mono",
    "LG_400_mono",
    "Mission_300_mono",
    "Panasonic_325_hit",
    "Qcells_280_poly",
    "Qcells_300_mono",
    "Solaria_400_mono",
    "Trina_260_poly",
]

sr_dataset_path = Path(pvlib.__path__[0]).joinpath(
    "data", "Duramat_spectral_responses.nc"
)
datafile = h5py.File(sr_dataset_path, "r")

keys_grouped = {
    "polysi": [
        "Canadian_270_poly",
        "Jinko_260_poly",
        "Qcells_280_poly",
        "Trina_260_poly",
    ],
    "monosi": [
        "Canadian_275_mono",
        "Itek_360_mono",
        "LG_320_mono",
        "LG_400_mono",
        "Mission_300_mono",
        "Qcells_300_mono",
        "Solaria_400_mono",
    ],
    "hit": ["Panasonic_325_hit"],
}

grouped_srs = {}

# ncells = pd.Series(index=datafile.keys())

for cell_type, module_list in keys_grouped.items():
    wavelength = pd.DataFrame()
    sr = pd.DataFrame()
    print(cell_type)
    for module_name in module_list:
        mod = datafile[module_name]
        wavelength[module_name] = mod["wavelength"]
        sr[module_name] = mod["sr"]
        plt.plot(wavelength[module_name], sr[module_name], label=module_name)
        # ncells[module_list] = mod["qe"].shape[0]

    # print("Std Wv")
    # print(wavelength.std())
    # print("Mean SR")
    # print("Stdev SR")
    sr_std = sr.std(axis=1)
    grouped_srs[cell_type] = sr.mean(axis=1).to_numpy()
    plt.plot(
        wavelength[module_name],
        grouped_srs[cell_type],
        label="Mean",
        c="k",
        linestyle="--",
    )
    # print(sr_std)
    # print("Max std:", sr_std.max())
    # print("Max percent std:", (sr_std / grouped_srs[cell_type]).max())
    plt.grid()
    plt.legend()
    plt.show()
    # plt.savefig(f"duramat_sr_agg_{cell_type}.png")
    plt.cla()
    plt.clf()

# print("ncells")
# print(ncells.unique())

# %%
data_df = pd.DataFrame(index=wavelength.to_numpy().transpose()[0], data=grouped_srs)
data_df.index.name = "wavelength"
# %%
data_df.to_csv("duramat_spectral_responses.csv")