Plot meteograms

config/showcase.yaml

@@ -26,10 +26,10 @@ runs:
 
 baselines:
   - baseline:
-      baseline_id: COSMO-E
+      baseline_id: COSMO-E-1h
       label: COSMO-E
-      root: /store_new/mch/msopr/ml/COSMO-E
-      steps: 0/120/6
+      root: /store_new/mch/msopr/ml/COSMO-E_hourly
+      steps: 0/120/1
 
 stratification:
   regions:
@@ -43,7 +43,7 @@ stratification:
 
 analysis:
   label: COSMO KENDA
-  analysis_zarr: /scratch/mch/fzanetta/data/anemoi/datasets/mch-co2-an-archive-0p02-2015-2020-6h-v3-pl13.zarr
+  analysis_zarr: /scratch/mch/fzanetta/data/anemoi/datasets/mch-co2-an-archive-0p02-2015-2020-1h-v3-pl13.zarr
 
 locations:
   output_root: output/

pyproject.toml

@@ -24,6 +24,7 @@ dependencies = [
     "earthkit-plots",
     "marimo>=0.16.5",
     "geopandas>=0.14.0",
+    "peakweather",
 ]
 
 [project.optional-dependencies]
@@ -58,3 +59,6 @@ packages = [
   "src/verification",
   "src/data_input"
 ]
+
+[tool.uv.sources]
+peakweather = { git = "https://github.com/MeteoSwiss/PeakWeather.git" }

workflow/Snakefile

@@ -57,9 +57,16 @@ rule showcase_all:
             / "showcases/{run_id}/{init_time}/{init_time}_{param}_{region}.gif",
             init_time=[t.strftime("%Y%m%d%H%M") for t in REFTIMES],
             run_id=collect_all_candidates(),
-            param=["TOT_PREC", "T_2M", "SP_10M"],
+            param=["T_2M", "SP_10M"],
             region=["globe", "europe", "switzerland"],
         ),
+        expand(
+            OUT_ROOT / "showcases/{run_id}/{init_time}/{init_time}_{param}_{sta}.png",
+            init_time=[t.strftime("%Y%m%d%H%M") for t in REFTIMES],
+            run_id=collect_all_candidates(),
+            param=["T_2M", "SP_10M"],
+            sta=["GVE", "KLO", "LUG"],
+        ),
         expand(
             OUT_ROOT / "data/runs/{run_id}/summary.md",
             run_id=collect_all_candidates(),

workflow/rules/data.smk

@@ -56,3 +56,14 @@ if "extract_cosmo1e" in config.get("include-optional-rules", []):
                 --steps {params.steps} \
                     > {log} 2>&1
             """
+
+
+rule download_obs_from_peakweather:
+    localrule: True
+    output:
+        peakweather=directory(OUT_ROOT / "data/observations/peakweather"),
+    run:
+        from peakweather.dataset import PeakWeatherDataset
+
+        # Download the data from Huggingface
+        ds = PeakWeatherDataset(root=output.peakweather)

workflow/rules/plot.smk

@@ -9,6 +9,52 @@ include: "common.smk"
 import pandas as pd
 
 
+def _use_first_baseline_zarr(wc):
+    """Get the first available baseline zarr for the given init time."""
+    for baseline_id in BASELINE_CONFIGS:
+        root = BASELINE_CONFIGS[baseline_id].get("root")
+        year = wc.init_time[2:4]
+        baseline_zarr = f"{root}/FCST{year}.zarr"
+        if Path(baseline_zarr).exists():
+            return baseline_zarr
+    raise ValueError(f"No baseline zarr found for init time {wc.init_time}")
+
+
+rule plot_meteogram:
+    input:
+        script="workflow/scripts/plot_meteogram.mo.py",
+        inference_okfile=rules.execute_inference.output.okfile,
+        analysis_zarr=config["analysis"].get("analysis_zarr"),
+        baseline_zarr=lambda wc: _use_first_baseline_zarr(wc),
+        peakweather_dir=rules.download_obs_from_peakweather.output.peakweather,
+    output:
+        OUT_ROOT / "showcases/{run_id}/{init_time}/{init_time}_{param}_{sta}.png",
+    # localrule: True
+    resources:
+        slurm_partition="postproc",
+        cpus_per_task=1,
+        runtime="5m",
+    params:
+        grib_out_dir=lambda wc: (
+            Path(OUT_ROOT) / f"data/runs/{wc.run_id}/{wc.init_time}/grib"
+        ).resolve(),
+    shell:
+        """
+        export ECCODES_DEFINITION_PATH=$(realpath .venv/share/eccodes-cosmo-resources/definitions)
+        python {input.script} \
+            --forecast {params.grib_out_dir}  --analysis {input.analysis_zarr} \
+            --baseline {input.baseline_zarr} --peakweather {input.peakweather_dir} \
+            --date {wildcards.init_time} --outfn {output[0]} \
+            --param {wildcards.param}  --station {wildcards.sta}
+        # interactive editing (needs to set localrule: True and use only one core)
+        # marimo edit {input.script} -- \
+        #     --forecast {params.grib_out_dir}  --analysis {input.analysis_zarr} \
+        #     --baseline {input.baseline_zarr} --peakweather {input.peakweather_dir} \
+        #     --date {wildcards.init_time} --outfn {output[0]} \
+        #     --param {wildcards.param}  --station {wildcards.sta}
+        """
+
+
 rule plot_forecast_frame:
     input:
         script="workflow/scripts/plot_forecast_frame.mo.py",