Add a MakiePlotsExt to be able to specify a plotting backend.

uses Preferences.jl for a persistent state.

Author: kellertuer

NEWS.md

@@ -5,7 +5,11 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [2.3.6] unreleased
+## [2.4.0] unreleased
+
+### Added
+
+* an extension to also perform the numerical checks plots with [`Makie.jl](https://https://makie.org/).
 
 ### Fixed
 

Project.toml

@@ -9,6 +9,7 @@ projects = ["test", "docs", "tutorials"]
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Markdown = "d6f4376e-aef5-505a-96c1-9c027394607a"
+Preferences = "21216c6a-2e73-6563-6e65-726566657250"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 
@@ -21,6 +22,7 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [extensions]
 ManifoldsBaseMakieExt = "Makie"
+ManifoldsBaseMakiePlotsExt = ["Makie", "Plots"]
 ManifoldsBasePlotsExt = "Plots"
 ManifoldsBaseQuaternionsExt = "Quaternions"
 ManifoldsBaseRecursiveArrayToolsExt = "RecursiveArrayTools"
@@ -31,9 +33,10 @@ LinearAlgebra = "1.6"
 Makie = "0.24"
 Markdown = "1.6"
 Plots = "1"
+Preferences = "1.4"
 Printf = "1.6"
+Quaternions = "0.7"
 Random = "1.6"
 RecursiveArrayTools = "2, 3, 4"
-Quaternions = "0.7"
 Statistics = "1.6"
 julia = "1.10"

ext/ManifoldsBaseMakiePlotsExt.jl

@@ -0,0 +1,24 @@
+module ManifoldsBaseMakiePlotsExt
+#
+#
+# For the case where one actually loads both plotting libraries,
+# this uses Preferences.jl to persistently specify which backend to use
+using ManifoldsBase
+
+function _preferred_plotting_ext()
+    if Symbol(ManifoldsBase.get_plotting_backend()) === :Makie
+        return Base.get_extension(ManifoldsBase, :ManifoldsBasePlotsExt)
+    else
+        return Base.get_extension(ManifoldsBase, :ManifoldsBaseMakieExt)
+    end
+end
+function ManifoldsBase.plot_slope(
+        x, y; slope = 2, line_base = 0, a = 0, b = 2.0, i = 1, j = length(x),
+    )
+    return _preferred_plotting_ext().plot_slope(x, y; slope, line_base, a, b, i, j)
+end
+function ManifoldsBase.plot_check_geodesic(T, N, e_norm, e_pt, e_alpha)
+    return _preferred_plotting_ext().plot_check_geodesic(T, N, e_norm, e_pt, e_alpha)
+end
+
+end # module ManifoldsBasePlotsMakieExt

src/ManifoldsBase.jl

@@ -24,6 +24,7 @@ import Random: rand, rand!
 
 using LinearAlgebra
 using Markdown: @doc_str
+using Preferences
 using Printf: @sprintf
 using Random
 
@@ -1219,6 +1220,18 @@ function zero_vector(M::AbstractManifold, p)
     zero_vector!(M, X, p)
     return X
 end
+
+#
+#
+# Internal function to set plotting backend
+function set_plotting_backend!(e::String)
+    (length(e) == 0) && return Preferences.@delete_preferences!("PlottingBackend")
+    return Preferences.@set_preferences!("PlottingBackend" => e)
+end
+# Internal function to set plotting backend
+function get_plotting_backend()
+    return Preferences.@load_preference("PlottingBackend", "Makie")
+end
 include("errors.jl")
 include("parallel_transport.jl")
 include("vector_transport.jl")