From 2de146e93dd4796ddcc9a155bf519b985c726745 Mon Sep 17 00:00:00 2001
From: Le Roux Erwan <erwan.le-roux@irstea.fr>
Date: Mon, 17 Jun 2019 15:06:48 +0200
Subject: [PATCH] [HYPERCUBE] add maxima plot for poster. rename some unit for
the variable
---
.../scm_models_data/abstract_study.py | 1 +
.../crocus/crocus_variables.py | 2 +-
.../scm_models_data/safran/safran_variable.py | 2 +-
.../main_study_visualizer.py | 56 +++++++++++++++----
.../study_visualization/study_visualizer.py | 27 +++++++--
5 files changed, 69 insertions(+), 19 deletions(-)
diff --git a/experiment/meteo_france_data/scm_models_data/abstract_study.py b/experiment/meteo_france_data/scm_models_data/abstract_study.py
index 1b179543..ebe76ecf 100644
--- a/experiment/meteo_france_data/scm_models_data/abstract_study.py
+++ b/experiment/meteo_france_data/scm_models_data/abstract_study.py
@@ -46,6 +46,7 @@ class AbstractStudy(object):
def __init__(self, variable_class: type, altitude: int = 1800, year_min=1000, year_max=3000,
multiprocessing=True):
+ assert isinstance(altitude, int), type(altitude)
assert altitude in ALTITUDES, altitude
self.altitude = altitude
self.model_name = None
diff --git a/experiment/meteo_france_data/scm_models_data/crocus/crocus_variables.py b/experiment/meteo_france_data/scm_models_data/crocus/crocus_variables.py
index 065d641e..7b3296f1 100644
--- a/experiment/meteo_france_data/scm_models_data/crocus/crocus_variables.py
+++ b/experiment/meteo_france_data/scm_models_data/crocus/crocus_variables.py
@@ -12,7 +12,7 @@ class CrocusVariable(AbstractVariable):
class CrocusSweVariable(CrocusVariable):
NAME = 'Snow Water Equivalent'
- UNIT = 'kg per m2 or mm'
+ UNIT = 'kg per square meter'
@classmethod
def keyword(cls):
diff --git a/experiment/meteo_france_data/scm_models_data/safran/safran_variable.py b/experiment/meteo_france_data/scm_models_data/safran/safran_variable.py
index 9484b1de..a0c29dc6 100644
--- a/experiment/meteo_france_data/scm_models_data/safran/safran_variable.py
+++ b/experiment/meteo_france_data/scm_models_data/safran/safran_variable.py
@@ -22,7 +22,7 @@ class SafranSnowfallVariable(AbstractVariable):
"""
NAME = 'Snowfall'
- UNIT = 'kg per m2 or mm'
+ UNIT = 'mm'
@classmethod
def keyword(cls):
diff --git a/experiment/meteo_france_data/scm_models_data/visualization/study_visualization/main_study_visualizer.py b/experiment/meteo_france_data/scm_models_data/visualization/study_visualization/main_study_visualizer.py
index 7aa2bc04..199a5626 100644
--- a/experiment/meteo_france_data/scm_models_data/visualization/study_visualization/main_study_visualizer.py
+++ b/experiment/meteo_france_data/scm_models_data/visualization/study_visualization/main_study_visualizer.py
@@ -7,7 +7,8 @@ from experiment.trend_analysis.abstract_score import MannKendall
from experiment.meteo_france_data.scm_models_data.abstract_study import AbstractStudy
from experiment.meteo_france_data.scm_models_data.crocus.crocus import CrocusDepth, CrocusSwe, ExtendedCrocusDepth, \
ExtendedCrocusSwe, CrocusDaysWithSnowOnGround
-from experiment.meteo_france_data.scm_models_data.safran.safran import SafranSnowfall, ExtendedSafranSnowfall, SafranRainfall, \
+from experiment.meteo_france_data.scm_models_data.safran.safran import SafranSnowfall, ExtendedSafranSnowfall, \
+ SafranRainfall, \
SafranTemperature, SafranTotalPrecip
from collections import OrderedDict
@@ -23,16 +24,27 @@ SCM_STUDY_CLASS_TO_ABBREVIATION = {
CrocusSwe: 'SWE',
CrocusDepth: 'SD',
}
-SCM_STUDY_NAME_TO_ABBREVIATION = {get_display_name_from_object_type(k): v for k, v in SCM_STUDY_CLASS_TO_ABBREVIATION.items()}
+
+altitude_massif_name_and_study_class_for_poster = [
+ (900, 'Chartreuse', CrocusSwe),
+ (1800, 'Vercors', CrocusDepth),
+ (2700, 'Parpaillon', SafranSnowfall),
+]
+
+SCM_STUDY_NAME_TO_ABBREVIATION = {get_display_name_from_object_type(k): v for k, v in
+ SCM_STUDY_CLASS_TO_ABBREVIATION.items()}
+SCM_COLORS = ['tab:orange', 'y', 'tab:purple']
+SCM_STUDY_CLASS_TO_COLOR = dict(zip(SCM_STUDIES, SCM_COLORS))
SCM_STUDY_NAME_TO_COLOR = {get_display_name_from_object_type(s): color
- for s, color in zip(SCM_STUDIES, ['tab:orange', 'y', 'tab:purple'])}
+ for s, color in zip(SCM_STUDIES, SCM_COLORS)}
SCM_EXTENDED_STUDIES = [ExtendedSafranSnowfall, ExtendedCrocusSwe, ExtendedCrocusDepth]
SCM_STUDY_TO_EXTENDED_STUDY = OrderedDict(zip(SCM_STUDIES, SCM_EXTENDED_STUDIES))
ALL_ALTITUDES = [0, 300, 600, 900, 1200, 1500, 1800, 2100, 2400, 2700, 3000, 3300, 3600, 3900, 4200, 4500, 4800]
ALTITUDES_LOW_MIDDLE_HIGH = [900, 1800, 2700]
-ALL_ALTITUDES_WITHOUT_NAN = [300, 600, 900, 1200, 1500, 1800, 2100, 2400, 2700, 3000, 3300, 3600, 3900, 4200, 4500, 4800]
+ALL_ALTITUDES_WITHOUT_NAN = [300, 600, 900, 1200, 1500, 1800, 2100, 2400, 2700, 3000, 3300, 3600, 3900, 4200, 4500,
+ 4800]
full_altitude_with_at_least_2_stations = [0, 300, 600, 900, 1200, 1500, 1800, 2100, 2400, 2700, 3000, 3300, 3600, 3900,
4200]
ALL_ALTITUDES_WITH_20_STATIONS_AT_LEAST = ALL_ALTITUDES[3:-6][:]
@@ -41,7 +53,7 @@ ALL_STUDIES = SCM_STUDIES + [SafranTemperature, SafranRainfall]
def study_iterator_global(study_classes, only_first_one=False, verbose=True, altitudes=None, nb_days=None) -> \
-List[AbstractStudy]:
+ List[AbstractStudy]:
for study_class in study_classes:
for study in study_iterator(study_class, only_first_one, verbose, altitudes, nb_days):
yield study
@@ -49,7 +61,8 @@ List[AbstractStudy]:
break
-def study_iterator(study_class, only_first_one=False, verbose=True, altitudes=None, nb_consecutive_days=3) -> List[AbstractStudy]:
+def study_iterator(study_class, only_first_one=False, verbose=True, altitudes=None, nb_consecutive_days=3) -> List[
+ AbstractStudy]:
# Default argument
altis = [1800] if altitudes is None else altitudes
@@ -126,8 +139,10 @@ def all_normal_vizu():
study_visualizer = StudyVisualizer(study, save_to_file=True, temporal_non_stationarity=True)
study_visualizer.visualize_all_mean_and_max_graphs()
+
def case_study():
- for study in study_iterator(study_class=SafranSnowfall, only_first_one=False, altitudes=[2100], nb_consecutive_days=3):
+ for study in study_iterator(study_class=SafranSnowfall, only_first_one=False, altitudes=[2100],
+ nb_consecutive_days=3):
study_visualizer = StudyVisualizer(study, save_to_file=False, temporal_non_stationarity=False)
study_visualizer.visualize_all_mean_and_max_graphs()
massif_id = study.study_massif_names.index('Chablais')
@@ -149,11 +164,14 @@ def scores_vizu():
def all_scores_vizu():
save_to_file = True
only_first_one = False
- for study in study_iterator_global(study_classes=[SafranSnowfall], only_first_one=only_first_one, altitudes=ALL_ALTITUDES):
- study_visualizer = StudyVisualizer(study, save_to_file=save_to_file, temporal_non_stationarity=True, verbose=True)
+ for study in study_iterator_global(study_classes=[SafranSnowfall], only_first_one=only_first_one,
+ altitudes=ALL_ALTITUDES):
+ study_visualizer = StudyVisualizer(study, save_to_file=save_to_file, temporal_non_stationarity=True,
+ verbose=True)
# study_visualizer.visualize_all_mean_and_max_graphs()
study_visualizer.visualize_all_score_wrt_starting_year()
+
def complete_analysis(only_first_one=False):
"""An overview of everything that is possible with study OR extended study"""
for study_class, extended_study_class in list(SCM_STUDY_TO_EXTENDED_STUDY.items())[:]:
@@ -191,13 +209,14 @@ def trend_analysis():
study_visualizer.df_trend_spatio_temporal(GevLocationChangePointTest,
starting_years=[1958, 1980], nb_massif_for_fast_mode=2)
+
def maxima_analysis():
save_to_file = False
only_first_one = True
durand_altitude = [1800]
altitudes = durand_altitude
normalization_class = BetweenZeroAndOneNormalization
- study_classes = [ SafranSnowfall][:]
+ study_classes = [SafranSnowfall][:]
for study in study_iterator_global(study_classes, only_first_one=only_first_one, altitudes=altitudes):
study_visualizer = StudyVisualizer(study, save_to_file=save_to_file,
transformation_class=normalization_class,
@@ -208,7 +227,19 @@ def maxima_analysis():
score_class=MannKendall)
study_visualizer.visualize_all_score_wrt_starting_year()
# study_visualizer.visualize_all_independent_temporal_trend()
- # study_visualizer.visualize_all_mean_and_max_graphs()
+ study_visualizer.visualize_all_mean_and_max_graphs()
+
+
+def max_graph_annual_maxima_poster():
+ save_to_file = True
+ for altitude, massif_name, study_class in altitude_massif_name_and_study_class_for_poster:
+ for study in study_iterator_global([study_class], altitudes=[altitude]):
+ study_visualizer = StudyVisualizer(study, save_to_file=save_to_file,
+ verbose=True,
+ multiprocessing=True)
+ snow_abbreviation = SCM_STUDY_CLASS_TO_ABBREVIATION[study_class]
+ color = SCM_STUDY_CLASS_TO_COLOR[study_class]
+ study_visualizer.visualize_max_graphs_poster(massif_name, altitude, snow_abbreviation, color)
def altitude_analysis():
@@ -222,7 +253,8 @@ def main_run():
# normal_visualization(temporal_non_stationarity=True)
# trend_analysis()
- altitude_analysis()
+ # altitude_analysis()
+ max_graph_annual_maxima_poster()
# case_study()
# all_scores_vizu()
# maxima_analysis()
diff --git a/experiment/meteo_france_data/scm_models_data/visualization/study_visualization/study_visualizer.py b/experiment/meteo_france_data/scm_models_data/visualization/study_visualization/study_visualizer.py
index 17e181c2..fdf51de3 100644
--- a/experiment/meteo_france_data/scm_models_data/visualization/study_visualization/study_visualizer.py
+++ b/experiment/meteo_france_data/scm_models_data/visualization/study_visualization/study_visualizer.py
@@ -432,7 +432,6 @@ class StudyVisualizer(VisualizationParameters):
else:
massif_names = sample(self.study.study_massif_names, k=self.nb_massif_for_change_point_test)
-
for massif_id, massif_name in enumerate(massif_names):
years, smooth_maxima = self.smooth_maxima_x_y(massif_id)
gev_change_point_test_results = compute_gev_change_point_test_results(self.multiprocessing, smooth_maxima,
@@ -459,7 +458,7 @@ class StudyVisualizer(VisualizationParameters):
trend_test_class_for_change_point_test,
starting_years_for_change_point_test,
nb_massif_for_change_point_test,
- sample_one_massif_from_each_region)
+ sample_one_massif_from_each_region)
for massif_name, gev_change_point_test_results in massif_name_to_gev_change_point_test_results.items():
trend_test_res, best_idxs = gev_change_point_test_results
trend_test_res = [(a, b, c) if i in best_idxs else (np.nan, np.nan, c)
@@ -586,6 +585,21 @@ class StudyVisualizer(VisualizationParameters):
self.plot_name = plot_name
self.show_or_save_to_file()
+ def visualize_max_graphs_poster(self, massif_name, altitude, snow_abbreviation, color):
+ massif_names = self.study.study_massif_names
+ # Display the graph of the max on top
+ ax = plt.gca()
+ x, y = self.smooth_maxima_x_y(massif_names.index(massif_name))
+ ax.plot(x, y, color=color)
+ ax.set_ylabel('{} (in {})'.format(snow_abbreviation, self.study.variable_unit), color=color)
+ ax.xaxis.set_ticks(x[2::10])
+ # self.visualize_massif_graphs(self.visualize_mean_and_max_graph,
+ # specified_massif_ids=specified_massif_ids)
+ plot_name = 'Annual maxima of {} in {} at {}'.format(snow_abbreviation, massif_name, altitude)
+ self.plot_name = plot_name
+ self.show_or_save_to_file(add_classic_title=False)
+ ax.clear()
+
def visualize_mean_and_max_graph(self, ax, massif_id):
# Display the graph of the max on top
color_maxima = 'r'
@@ -719,10 +733,13 @@ class StudyVisualizer(VisualizationParameters):
label_function(full_title)
ax0.tick_params(axis=u'both', which=u'both', length=0)
- def show_or_save_to_file(self):
+ def show_or_save_to_file(self, add_classic_title=True):
assert self.plot_name is not None
- title = self.study.title
- title += '\n' + self.plot_name
+ if add_classic_title:
+ title = self.study.title
+ title += '\n' + self.plot_name
+ else:
+ title = self.plot_name
if self.only_one_graph:
plt.suptitle(self.plot_name)
else:
--
GitLab