from safran_study.safran import Safran
from itertools import product


def load_all_safran(only_first_one=False):
    all_safran = []
    for safran_alti, nb_day in product([1800, 2400], [1, 3, 7]):
        print('alti: {}, nb_day: {}'.format(safran_alti, nb_day))
        all_safran.append(Safran(safran_alti, nb_day))
        if only_first_one:
            break
    return all_safran


def fit_mle_gev_independent(threshold=None):
    # Dump the result in a csv
    dfs = []
    for safran in load_all_safran(only_first_one=True):
        safran.visualize_margin_fits_with_cmap(threshold=threshold)
        # path = r'/home/erwan/Documents/projects/spatiotemporalextremes/local/spatio_temporal_datasets/results/fit_mle_massif/fit_mle_gev_{}.csv'
        # df.to_csv(path.format(VERSION))


def fit_max_stab():
    pass


if __name__ == '__main__':
    threshold = [None, 20, 40, 60][1]
    fit_mle_gev_independent(threshold=threshold)