Enhance type flexibility for PoincareMap & Add a Bigfloat test (#239)

* Enhance type flexibility for PoincareMap and add a Bigfloat test for Poincaré mapping.

* Remove `setprecision` and designate precision locally.

* Increment the patch version from 3.15.1 to 3.15.2.

Author: hibiki-kato

Project.toml

@@ -1,7 +1,7 @@
 name = "DynamicalSystemsBase"
 uuid = "6e36e845-645a-534a-86f2-f5d4aa5a06b4"
 repo = "https://github.com/JuliaDynamics/DynamicalSystemsBase.jl.git"
-version = "3.15.1"
+version = "3.15.2"
 
 [deps]
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"

src/derived_systems/poincare/poincaremap.jl

@@ -95,18 +95,18 @@ next_state_on_psos = current_state(pmap)
     Datseris & Parlitz 2022, _Nonlinear Dynamics: A Concise Introduction Interlaced with Code_,
     [Springer Nature, Undergrad. Lect. Notes In Physics](https://doi.org/10.1007/978-3-030-91032-7)
 """
-mutable struct PoincareMap{I<:ContinuousTimeDynamicalSystem, F, P, R, V} <: DiscreteTimeDynamicalSystem
+mutable struct PoincareMap{I<:ContinuousTimeDynamicalSystem, F, P, R, U<:Real, V} <: DiscreteTimeDynamicalSystem
 	ds::I
 	plane_distance::F
  	planecrossing::P
-	Tmax::Float64
+	Tmax::U
 	rootkw::R
 	state_on_plane::V
-    tcross::Float64
+    tcross::U
 	t::Int
     # These two fields are for setting the state of the pmap from the plane
     # (i.e., given a D-1 dimensional state, create the full D-dimensional state)
-    dummy::Vector{Float64}
+    dummy::Vector{U}
     diffidxs::Vector{Int}
     state_initial::V
 end
@@ -125,11 +125,13 @@ function PoincareMap(
 	planecrossing = PlaneCrossing(plane, direction > 0)
 	plane_distance = (t) -> planecrossing(ds(t))
     v = recursivecopy(current_state(ds))
-    dummy = zeros(D)
+    tcross = current_time(ds)
+    Tmax = convert(typeof(tcross), Tmax)
+    dummy = zeros(eltype(v), D)
     diffidxs = _indices_on_poincare_plane(plane, D)
 	pmap = PoincareMap(
         ds, plane_distance, planecrossing, Tmax, rootkw,
-        v, current_time(ds), 0, dummy, diffidxs, recursivecopy(v),
+        v, tcross, 0, dummy, diffidxs, recursivecopy(v),
     )
     step!(pmap) # this ensures that the state is on the hyperplane
     pmap.state_initial = recursivecopy(current_state(pmap))

test/poincare.jl

@@ -99,3 +99,23 @@ end
     @test all(x -> abs(x) < 1e-12, B[:, 1])
     @test vec(A) == vec(B)
 end
+
+@testset "PoincareMap BigFloat" begin
+    u0Big = BigFloat.(u0, 113)
+    pBig = BigFloat.(p, 113)
+    ds = CoupledODEs(gissinger_rule, u0Big, pBig)
+    rootkw = (xrtol = BigFloat(1e-25, 113), atol = BigFloat(1e-25, 113))
+    pmap = PoincareMap(ds, plane1;
+        rootkw = rootkw,
+    )
+
+    # check if everything is BigFloat
+    @test typeof(current_crossing_time(pmap)) == BigFloat
+    @test typeof(pmap.Tmax) == BigFloat
+    @test eltype(current_state(pmap)) == BigFloat
+
+    # check if BigFloat works
+    pmap = poincaresos(ds, plane1, 100; rootkw = rootkw)
+    @test eltype(pmap[1]) == BigFloat
+    @test all(x -> abs(x) < 1e-20, pmap[:, 1])
+end