From e4d09b4bfeacad1d5cd2f9542b5100174924f195 Mon Sep 17 00:00:00 2001
From: Le Roux Erwan <erwan.le-roux@irstea.fr>
Date: Tue, 10 Sep 2019 11:29:42 +0200
Subject: [PATCH] [POSTER EVAN] add poster part D for the EVAN poster

---
 .../poster_EVAN2019/main_poster_EVAN2019.py   | 19 ++++++++++++++++++-
 experiment/paper1_steps/utils.py              |  5 +++--
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/experiment/paper1_steps/poster_EVAN2019/main_poster_EVAN2019.py b/experiment/paper1_steps/poster_EVAN2019/main_poster_EVAN2019.py
index dbf691cd..e8d80b02 100644
--- a/experiment/paper1_steps/poster_EVAN2019/main_poster_EVAN2019.py
+++ b/experiment/paper1_steps/poster_EVAN2019/main_poster_EVAN2019.py
@@ -1,5 +1,9 @@
+from experiment.meteo_france_data.scm_models_data.crocus.crocus import CrocusDepth, CrocusSweTotal
+from experiment.meteo_france_data.scm_models_data.safran.safran import SafranSnowfall
 from experiment.meteo_france_data.scm_models_data.visualization.hypercube_visualization.altitude_year_hypercube_visualizer import \
     Altitude_Hypercube_Year_Visualizer, AltitudeHypercubeVisualizerWithoutTrendType
+from experiment.meteo_france_data.scm_models_data.visualization.study_visualization.main_study_visualizer import \
+    SCM_STUDIES
 from experiment.trend_analysis.univariate_test.gev_trend_test_one_parameter import GevScaleTrendTest, \
     GevLocationTrendTest
 from experiment.trend_analysis.univariate_test.gev_trend_test_two_parameters import GevLocationAndScaleTrendTest
@@ -53,6 +57,19 @@ def main_poster_B_starting_years_analysis():
             vizualiser.visualize_massif_trend_test_one_altitude(poster_plot=True, write_text_on_massif=True)
 
 
+def main_poster_D_other_quantities_analysis():
+    nb = 3
+    trend_test_class = GevLocationAndScaleTrendTest
+    for altitude in POSTER_ALTITUDES[:nb]:
+        for study_class in [SafranSnowfall, CrocusSweTotal, CrocusDepth][:nb]:
+            vizualiser = get_full_altitude_visualizer(Altitude_Hypercube_Year_Visualizer, altitude=altitude,
+                                                      exact_starting_year=1958, reduce_strength_array=False,
+                                                      trend_test_class=trend_test_class,
+                                                      study_class=study_class)
+            vizualiser.visualize_massif_trend_test_one_altitude(poster_plot=True, write_text_on_massif=False)
+
+
 if __name__ == '__main__':
     # main_poster_A_non_stationary_model_choice()
-    main_poster_B_starting_years_analysis()
+    # main_poster_B_starting_years_analysis()
+    main_poster_D_other_quantities_analysis()
diff --git a/experiment/paper1_steps/utils.py b/experiment/paper1_steps/utils.py
index 3cd69f13..24f7a9c6 100644
--- a/experiment/paper1_steps/utils.py
+++ b/experiment/paper1_steps/utils.py
@@ -11,12 +11,13 @@ FULL_ALTITUDES = [900, 1200, 1500, 1800, 2100, 2400, 2700, 3000]
 def get_full_altitude_visualizer(altitude_hypercube_class, exact_starting_year=None, altitude=900,
                                  reduce_strength_array=False,
                                  trend_test_class = GevLocationTrendTest,
-                                 offset_starting_year=10):
+                                 offset_starting_year=10,
+                                 study_class=CrocusSwe3Days):
     altitudes, first_starting_year, last_starting_year, nb_data_reduced_for_speed, only_first_one, save_to_file, _ = get_full_parameters(
         altitude=altitude, offset_starting_year=offset_starting_year)
     if exact_starting_year is not None:
         first_starting_year, last_starting_year = None, None
-    study_classes = [CrocusSwe3Days]
+    study_classes = [study_class]
     visualizer = load_altitude_visualizer(altitude_hypercube_class, altitudes, last_starting_year,
                                           nb_data_reduced_for_speed, only_first_one, save_to_file, study_classes,
                                           trend_test_class, first_starting_year=first_starting_year,
-- 
GitLab