Reduce duplication in colorbar logic

Author: mariekeosw

CHANGELOG.md

@@ -1,6 +1,6 @@
 # Changelog
 
-## [1.18.1] - 2026-04-21
+## [1.18.2] - 2026-04-21
 - Add colorbar for cell regions in 2D `gridplot!` with `Py[thon]Plot`
 
 ## [1.18.0] - 2026-04-15

Project.toml

@@ -1,7 +1,7 @@
 name = "GridVisualize"
 uuid = "5eed8a63-0fb0-45eb-886d-8d5a387d12b8"
 authors = ["Juergen Fuhrmann <juergen.fuhrmann@wias-berlin.de>", "Patrick Jaap <patrick.jaap@wias-berlin.de>"]
-version = "1.18.1"
+version = "1.18.2"
 
 [deps]
 ColorSchemes = "35d6a980-a343-548e-a6ea-1d62b119f2f4"

src/pycommon.jl

@@ -274,41 +274,19 @@ function gridplot!(ctx, TP::Type{T}, ::Type{Val{2}}, grid) where {T <: AbstractP
     )
 
     if ctx[:show_colorbar]
-        if ctx[:colorbar] == :horizontal
-
+        if ctx[:colorbar] in (:horizontal, :vertical)
             fig.colorbar(
                 bcdata;
                 ax = ax,
-                ticks = collect(1:length(bcmap)),
-                orientation = "horizontal",
+                ticks = 1:length(bcmap),
+                orientation = "$(ctx[:colorbar])",
                 label = "boundary regions"
             )
-
-            fig.colorbar(
-                cdata;
-                ax = ax,
-                ticks = collect(1:length(cmap)),
-                orientation = "horizontal",
-                label = "cell regions"
-            )
-
-        end
-
-        if ctx[:colorbar] == :vertical
-
-            fig.colorbar(
-                bcdata;
-                ax = ax,
-                ticks = collect(1:length(bcmap)),
-                orientation = "vertical",
-                label = "boundary regions"
-            )
-
             fig.colorbar(
                 cdata;
                 ax = ax,
-                ticks = collect(1:length(cmap)),
-                orientation = "vertical",
+                ticks = 1:length(cmap),
+                orientation = "$(ctx[:colorbar])",
                 label = "cell regions"
             )
         end