total rain between onset and cessation, multiplying size of problem by munber of years

Author: remicousin

enacts/config-defaults.yaml

@@ -66,19 +66,19 @@ onset:
             menu_label: How likely is the rainy season lasting less than...
             description: The map shows the probability of the rainy season
                 to last less than a certain number of days through all the years of available data.
-        #total_mean:
-        #    menu_label: How much rain falls in the rainy season?
-        #    description: The map shows the average precipitation (in mm)  during the rainy season,
-        #        defined as the period between onset and cessation dates,
-        #        over all years of available data.
-        #total_stddev:
-        #    menu_label: How uncertain is the amount of rain in the rainy season?
-        #    description: The map shows the standard deviation from the average precipitation in the season,
-        #        over all years of available data.
-        #total_pe:
-        #    menu_label: How likely is it raining less than...
-        #    description: The map shows the probability of precipitation in the rainy season
-        #        being less than a certain amount through all the years of available data.
+        total_mean:
+            menu_label: How much rain falls in the rainy season?
+            description: The map shows the average precipitation (in mm)  during the rainy season,
+                defined as the period between onset and cessation dates,
+                over all years of available data.
+        total_stddev:
+            menu_label: How uncertain is the amount of rain in the rainy season?
+            description: The map shows the standard deviation from the average precipitation in the season,
+                over all years of available data.
+        total_pe:
+            menu_label: How likely is it raining less than...
+            description: The map shows the probability of precipitation in the rainy season
+                being less than a certain amount through all the years of available data.
 
 
 # Water Balance Monitoring

enacts/config-zmd.yaml

@@ -104,4 +104,7 @@ onset:
             map_max: 180
         length_stddev:
             map_max: 40
-
+        total_mean:
+            map_max: 1100
+        total_stddev:
+            map_max: 300

enacts/onset/maproom.py

@@ -853,15 +853,19 @@ def onset_tile(tz, tx, ty):
                 cess_soil_moisture,
                 cess_dry_spell,
             )
+            if cess_dates["T"][0] < onset_dates["T"][0]:
+                cess_dates = cess_dates.isel({"T": slice(1, None)})
+                if cess_dates["T"].size != onset_dates["T"].size:
+                    onset_dates = onset_dates.isel({"T": slice(None, -2)})
+            cess_dates = (cess_dates["T"] + cess_dates["cess_delta"]).drop_indexes("T")
+            onset_dates = (onset_dates["T"] + onset_dates["onset_delta"]).drop_indexes("T")
             if "length" in map_choice:
-                if cess_dates["T"][0] < onset_dates["T"][0]:
-                    cess_dates = cess_dates.isel({"T": slice(1, None)})
-                    if cess_dates["T"].size != onset_dates["T"].size:
-                        onset_dates = onset_dates.isel({"T": slice(None, -2)})
-                seasonal_length = (
-                    (cess_dates["T"] + cess_dates["cess_delta"]).drop_indexes("T")
-                    - (onset_dates["T"] + onset_dates["onset_delta"]).drop_indexes("T")
-                ) #.astype("timedelta64[D]")
+                seasonal_length = cess_dates - onset_dates
+            else:
+                seasonal_total = precip_tile.where(
+                    (precip_tile["T"] >= onset_dates.rename({"T": "Ty"}))
+                        & (precip_tile["T"] <= cess_dates.rename({"T": "Ty"}))
+                ).sum("T")
         if map_choice == "mean":
             map_data = onset_dates.onset_delta.mean("T")
             map_max = np.timedelta64(search_days, 'D')
@@ -886,6 +890,17 @@ def onset_tile(tz, tx, ty):
             map_data = (seasonal_length < np.timedelta64(prob_exc_thresh_length, 'D')).mean("T") * 100
             map_max = 100
             colormap = pingrid.CORRELATION_COLORMAP
+        if map_choice == "total_mean":
+            map_data = seasonal_total.mean("Ty")
+            map_max = CONFIG["map_text"][map_choice]["map_max"]
+            colormap = pingrid.RAINFALL_COLORMAP
+        if map_choice == "total_stddev":
+            map_data = seasonal_total.std(dim="Ty", skipna=True)
+            map_max = CONFIG["map_text"][map_choice]["map_max"]
+        if map_choice == "total_pe":
+            map_data = (seasonal_total < prob_exc_thresh_tot).mean("Ty") * 100
+            map_max = 100
+            colormap = pingrid.CORRELATION_COLORMAP
     map_data.attrs["colormap"] = colormap
     map_data = map_data.rename(X="lon", Y="lat")
     map_data.attrs["scale_min"] = map_min
@@ -928,6 +943,16 @@ def set_colorbar(search_start_day, search_start_month, search_days, map_choice):
         tick_freq = 5
         map_max = CONFIG["map_text"][map_choice]["map_max"]
         unit = "days"
+    if map_choice == "total_mean":
+        colorbar = pingrid.to_dash_colorscale(pingrid.RAINFALL_COLORMAP)
+        tick_freq = 100
+        map_max = CONFIG["map_text"][map_choice]["map_max"]
+        unit = "mm"
+    if map_choice == "total_stddev":
+        tick_freq = 30
+                )
+        map_max = CONFIG["map_text"][map_choice]["map_max"]
+        unit = "mm"
     if map_choice == "monit":
         precip = rr_mrg.precip.isel({"T": slice(-366, None)})
         search_start_dm = calc.sel_day_and_month(