refactor: add accumulated precip to dataset postprocessing

Author: yoctoyotta1024

scripts_for_plotting/src/load_ensemble_datasets.py

@@ -159,6 +159,17 @@ def postprocess_cleo_dataset(ds, config, gbxs, precip_rolling_window, is_ensembl
     )
     ds = ds.assign(**{arr.name: arr})
 
+    arr = xr.DataArray(
+        ds.surfprecip_rate.cumsum(dim="time") * config["OBSTSTEP"] / 3600,
+        name="surfprecip_cumulative",
+        dims=["ensemble", "time"] if is_ensemble else ["time"],
+        attrs={
+            "units": "mm",
+            "long_name": "accumulated surface precipitation",
+        },
+    )
+    ds = ds.assign(**{arr.name: arr})
+
     arr = xr.DataArray(
         calcs.mean_rolling_window(ds.surfprecip_rate, precip_rolling_window),
         name="surfprecip_rolling",
@@ -497,6 +508,18 @@ def postprocess_pysdm_dataset(ds, precip_rolling_window, is_ensemble=True):
             },
         )
         ds = ds.assign(**{arr.name: arr})
+
+        delta_time = np.mean(ds.time.values[1:] - ds.time.values[:-1])
+        arr = xr.DataArray(
+            ds.surfprecip_rate.cumsum(dim="time") * delta_time / 3600,
+            name="surfprecip_cumulative",
+            dims=["ensemble", "time"] if is_ensemble else ["time"],
+            attrs={
+                "units": "mm",
+                "long_name": "accumulated surface precipitation",
+            },
+        )
+        ds = ds.assign(**{arr.name: arr})
     except AttributeError:
         print("no precipitation in dataset")