-
Le Roux Erwan authoreda3dd9ba5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
import matplotlib.pyplot as plt
from collections import OrderedDict
import numpy as np
from cached_property import cached_property
from extreme_data.meteo_france_data.adamont_data.abstract_adamont_study import AbstractAdamontStudy
from extreme_data.meteo_france_data.adamont_data.adamont_gcm_rcm_couples import get_gcm_rcm_couple_adamont_to_full_name, \
gcm_rcm_couple_to_color
from extreme_data.meteo_france_data.adamont_data.adamont_scenario import gcm_rcm_couple_to_str
from extreme_data.meteo_france_data.scm_models_data.abstract_study import AbstractStudy
from extreme_data.meteo_france_data.scm_models_data.visualization.main_study_visualizer import \
SCM_STUDY_CLASS_TO_ABBREVIATION, ADAMONT_STUDY_CLASS_TO_ABBREVIATION
from extreme_data.meteo_france_data.scm_models_data.visualization.study_visualizer import StudyVisualizer
class AdamontStudies(object):
def __init__(self, study_class, gcm_rcm_couples=None, adamont_version=2, **kwargs_study):
self.study_class = study_class
if gcm_rcm_couples is None:
gcm_rcm_couple_to_full_name = get_gcm_rcm_couple_adamont_to_full_name(adamont_version)
gcm_rcm_couples = list(gcm_rcm_couple_to_full_name.keys())
self.gcm_rcm_couples = gcm_rcm_couples
self.gcm_rcm_couple_to_study = OrderedDict() # type: OrderedDict[int, AbstractAdamontStudy]
for gcm_rcm_couple in self.gcm_rcm_couples:
study = study_class(gcm_rcm_couple=gcm_rcm_couple, adamont_version=adamont_version, **kwargs_study)
self.gcm_rcm_couple_to_study[gcm_rcm_couple] = study
@property
def study_list(self):
return list(self.gcm_rcm_couple_to_study.values())
@cached_property
def study(self) -> AbstractAdamontStudy:
return self.study_list[0]
@property
def nb_ensemble_members(self):
return len(self.gcm_rcm_couples)
# Some plots
def show_or_save_to_file(self, plot_name, show=False, no_title=False, tight_layout=None):
study_visualizer = StudyVisualizer(study=self.study, show=show, save_to_file=not show)
study_visualizer.plot_name = plot_name
study_visualizer.show_or_save_to_file(add_classic_title=False, dpi=500, no_title=no_title,
tight_layout=tight_layout)
def plot_maxima_time_series_adamont(self, massif_names=None, scm_study=None):
massif_names = massif_names if massif_names is not None else self.study.all_massif_names()
for massif_names in massif_names:
self._plot_maxima_time_series(massif_names, scm_study)
def _plot_maxima_time_series(self, massif_name, scm_study=None):
ax = plt.gca()
x = self.study.ordered_years
linewidth = 2
for gcm_rcm_couple, study in list(self.gcm_rcm_couple_to_study.items())[::-1]:
if massif_name in study.massif_name_to_annual_maxima:
y = study.massif_name_to_annual_maxima[massif_name]
label = gcm_rcm_couple_to_color[gcm_rcm_couple]
color = gcm_rcm_couple_to_color[gcm_rcm_couple]
ax.plot(x, y, linewidth=linewidth, label=label, color=color)
if scm_study is None:
y = np.array([study.massif_name_to_annual_maxima[massif_name] for study in self.study_list
if massif_name in study.massif_name_to_annual_maxima])
if len(y) > 0:
y = np.mean(y, axis=0)
label = 'Mean maxima'
color = 'black'
ax.plot(x, y, linewidth=linewidth * 2, label=label, color=color)
else:
# todo: otherwise display the mean in strong black
try:
y = scm_study.massif_name_to_annual_maxima[massif_name]
label = 'Reanalysis'
color = 'black'
ax.plot(x, y, linewidth=linewidth * 2, label=label, color=color)
except KeyError:
pass
ticks = [year for year in x if year % 10 == 0]
ax.xaxis.set_ticks(ticks)
ax.yaxis.grid()
ax.set_xlim((min(x), max(x)))
# Augment the ylim for the legend
ylim_min, ylim_max = ax.get_ylim()
ax.set_ylim((ylim_min, ylim_max * 1.5))
ax.tick_params(axis='both', which='major', labelsize=13)
handles, labels = ax.get_legend_handles_labels()
ncol = 2 if self.study.adamont_version == 1 else 3
ax.legend(handles[::-1], labels[::-1], ncol=ncol, prop={'size': 7})
plot_name = 'Annual maxima of {} in {} at {} m'.format(ADAMONT_STUDY_CLASS_TO_ABBREVIATION[self.study_class],
massif_name.replace('_', ' '),
self.study.altitude)
fontsize = 13
ax.set_ylabel('{} ({})'.format(plot_name, self.study.variable_unit), fontsize=fontsize)
ax.set_xlabel('years', fontsize=fontsize)
plot_name = 'time series/' + plot_name
self.show_or_save_to_file(plot_name=plot_name, show=False, no_title=True, tight_layout=True)
ax.clear()
plt.close()