ski roses with local difficulty colors

Author: dhimmel

openskistats/analyze.py

@@ -399,13 +399,19 @@ def aggregate_ski_areas_pl(
             vertical_drop=pl.max("max_elevation") - pl.min("min_elevation"),
             latitude=pl.mean("latitude"),
             longitude=pl.mean("longitude"),
+            osm_run_convention=pl.col("osm_run_convention").unique().drop_nulls(),
             _bearing_stats=pl.struct(
                 pl.col("bearing_mean").alias("bearing"),
                 "bearing_magnitude_cum",
                 "bearing_alignment",
                 "hemisphere",
             ).map_batches(_get_bearing_summary_stats_pl, returns_scalar=True),
         )
+        .with_columns(
+            osm_run_convention=pl.when(pl.col("osm_run_convention").list.len() == 1)
+            .then(pl.col("osm_run_convention").list.first())
+            .otherwise(None)
+        )
         .unnest("_bearing_stats")
         .join(bearings_pl, on=group_by)
         .sort(*group_by)
@@ -700,6 +706,7 @@ def _create_ski_area_rose(
     """Create a preview and a full rose for a ski area."""
     ski_area_id = info["ski_area_id"]
     ski_area_name = info["ski_area_name"]
+    color_convention = info["osm_run_convention"]
 
     # plot and save preview rose
     bearing_preview_pl = bearing_pl.filter(pl.col("num_bins") == 8)
@@ -736,7 +743,9 @@ def _create_ski_area_rose(
     fig, ax = plot_orientation(
         bin_counts=bearing_full_pl.get_column("bin_count").to_numpy(),
         bin_centers=bearing_full_pl.get_column("bin_center").to_numpy(),
-        color_to_bin_counts=get_difficulty_color_to_bearing_bin_counts(bearing_full_pl),
+        color_to_bin_counts=get_difficulty_color_to_bearing_bin_counts(
+            bearing_full_pl, convention=color_convention
+        ),
         title=ski_area_name,
         title_font_size=16,
         margin_text=_generate_margin_text(info),

openskistats/bearing.py

@@ -9,6 +9,7 @@
 from openskistats.models import (
     BearingStatsModel,
     SkiAreaBearingDistributionModel,
+    SkiRunConvention,
     SkiRunDifficulty,
 )
 
@@ -279,6 +280,7 @@ def get_bearing_histogram(
 
 def get_difficulty_color_to_bearing_bin_counts(
     bearing_pl: pl.DataFrame,
+    convention: SkiRunConvention = SkiRunConvention.north_america,
 ) -> dict[str, pl.Series]:
     return (  # type: ignore [no-any-return]
         bearing_pl.select(
@@ -290,7 +292,8 @@ def get_difficulty_color_to_bearing_bin_counts(
         .select(
             pl.all().name.map(
                 lambda x: SkiRunDifficulty(x.removeprefix("bin_count_")).color(
-                    subtle=True
+                    subtle=True,
+                    convention=convention,
                 )
             )
         )

openskistats/plot.py

@@ -15,6 +15,7 @@
 from osmnx.plot import _get_fig_ax
 
 from openskistats.bearing import get_difficulty_color_to_bearing_bin_counts
+from openskistats.models import SkiRunConvention
 from openskistats.sunlight import get_solar_location_band
 
 SUBPLOT_FIGSIZE = 4.0
@@ -385,6 +386,7 @@ def subplot_orientations(
     suptitle: str | None = None,
     sort_groups: bool = True,
     plot_solar_band: bool = False,
+    color_convention: SkiRunConvention | None = SkiRunConvention.north_america,
 ) -> plt.Figure:
     """
     Plot orientations from multiple graphs in a grid.
@@ -406,6 +408,9 @@ def subplot_orientations(
         This value must exist in the input groups_pl bearings.num_bins column.
     suptitle
         The figure's super title.
+    color_convention
+        The convention to use for color-coding the polar histogram by run difficulty.
+        If None, use the group specific conventions provided in groups_pl.
     """
     assert not groups_pl.select(grouping_col).is_duplicated().any()
     names = groups_pl.get_column(grouping_col).to_list()
@@ -443,7 +448,10 @@ def subplot_orientations(
             bin_counts=group_dist_pl.get_column("bin_count").to_numpy(),
             bin_centers=group_dist_pl.get_column("bin_center").to_numpy(),
             color_to_bin_counts=get_difficulty_color_to_bearing_bin_counts(
-                group_dist_pl
+                group_dist_pl,
+                convention=color_convention
+                or group_info["osm_run_convention"]
+                or SkiRunConvention.north_america,
             ),
             ax=ax,
             max_bin_count=max_bin_count,