improve handling of mesh adaptation of collocation

bifurcationkit · Apr 15, 2024 · 6d00223 · 6d00223
1 parent eafc49b
commit 6d00223
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/periodicorbit/NormalForms.jl b/src/periodicorbit/NormalForms.jl
@@ -425,7 +425,7 @@ function period_doubling_normal_form(pbwrap::WrapPOColl,
     # computation of c
     # we need B(t, v₁(t), h₂(t))
     for i=1:size(Bₛ, 2)
-        Bₛ[:,i]  .= B(u₀ₛ[:,i], par, v₁ₛ[:,i], h₂ₛ[:,i])
+        Bₛ[:,i] .= B(u₀ₛ[:,i], par, v₁ₛ[:,i], h₂ₛ[:,i])
     end
                 # _plot(vcat(vec( Bₛ ),1), label = "Bₛ for h2")
                 # _plot(vcat(vec(@. Bₛ * v₁★ₛ ),1), label = "Bₛ*v1star")

diff --git a/src/periodicorbit/PeriodicOrbitUtils.jl b/src/periodicorbit/PeriodicOrbitUtils.jl
@@ -82,9 +82,16 @@ function (finalizer::Finaliser{ <: Union{ <: PeriodicOrbitOCollProblem,
     state = get(kF, :state, nothing)
     success = converged(state)
     bisection = in_bisection(state)
+    is_mesh_updated = false
+
     # mesh adaptation
-    if success && coll.meshadapt && bisection == false
+    if success &&
+            coll.meshadapt && 
+            bisection == false && 
+            mod_counter(step, updateSectionEveryStep) == 1 &&
+            step > 2
         @debug "[Collocation] update mesh"
+        is_mesh_updated = true
         oldsol = _copy(z) # avoid possible overwrite in compute_error!
         oldmesh = get_times(coll) .* getperiod(coll, oldsol.u, nothing)
         adapt = compute_error!(coll, oldsol.u;
@@ -95,6 +102,7 @@ function (finalizer::Finaliser{ <: Union{ <: PeriodicOrbitOCollProblem,
             return false
         end
     end
+
     if success && mod_counter(step, updateSectionEveryStep) == 1 && bisection == false
         @debug "[collocation] update section"
         updatesection!(coll, z.u, setparam(contResult, z.p))