########################################################################################
##
## BACKWARD DIFFERENTIATION FORMULAS
## (solvers/bdf.py)
##
########################################################################################
# IMPORTS ==============================================================================
from collections import deque
from ._solver import ImplicitSolver
from .dirk3 import DIRK3
# BASE BDF SOLVER ======================================================================
[docs]
class BDF(ImplicitSolver):
"""Base class for the backward differentiation formula (BDF) integrators.
Notes
-----
This solver class is not intended to be used directly
Attributes
----------
x : numeric, array[numeric]
internal 'working' state
n : int
order of integration scheme
s : int
number of internal intermediate stages
stage : int
counter for current intermediate stage
eval_stages : list[float]
rations for evaluation times of intermediate stages
opt : NewtonAnderson, Anderson, etc.
optimizer instance to solve the implicit update equation
K : dict[int: list[float]]
bdf coefficients for the state buffer for each order
F : dict[int: float]
bdf coefficients for the function 'func' for each order
history : deque[numeric]
internal history of past results
startup : Solver
internal solver instance for startup (building history)
of multistep methods (using 'DIRK3' for 'BDF' methods)
"""
def __init__(self, *solver_args, **solver_kwargs):
super().__init__(*solver_args, **solver_kwargs)
#integration order
self.n = None
#bdf coefficients for orders 1 to 6
self.K = {
1:[1.0],
2:[4/3, -1/3],
3:[18/11, -9/11, 2/11],
4:[48/25, -36/25, 16/25, -3/25],
5:[300/137, -300/137, 200/137, -75/137, 12/137],
6:[360/147, -450/147, 400/147, -225/147, 72/147, -10/147]
}
self.F = {1:1.0, 2:2/3, 3:6/11, 4:12/25, 5:60/137, 6:60/147}
#initialize startup solver from 'self' and flag
self._needs_startup = True
self.startup = DIRK3.cast(self, self.parent)
[docs]
@classmethod
def cast(cls, other, parent, **solver_kwargs):
"""cast to this solver needs special handling of startup method
Parameters
----------
other : Solver
solver instance to cast new instance of this class from
parent : None | Solver
solver instance to use as parent
solver_kwargs : dict
other args for the solver
Returns
-------
engine : BDF
instance of `BDF` solver with params and state from `other`
"""
engine = super().cast(other, parent, **solver_kwargs)
engine.startup = DIRK3.cast(engine, parent)
return engine
[docs]
def stages(self, t, dt):
"""Generator that yields the intermediate evaluation
time during the timestep 't + ratio * dt'.
Parameters
----------
t : float
evaluation time
dt : float
integration timestep
"""
#not enough history for full order -> stages of startup method
if self._needs_startup:
for self.stage, _t in enumerate(self.startup.stages(t, dt)):
yield _t
else:
for _t in super().stages(t, dt):
yield _t
[docs]
def reset(self):
""""Resets integration engine to initial state."""
#clear history (BDF solution buffer)
self.history.clear()
#overwrite state with initial value
self.x = self.initial_value
#reset startup solver
self.startup.reset()
[docs]
def buffer(self, dt):
"""buffer the state for the multistep method
Parameters
----------
dt : float
integration timestep
"""
#reset optimizer
self.opt.reset()
#add current solution to history
self.history.appendleft(self.x)
#flag for startup method, not enough history
self._needs_startup = len(self.history) < self.n
#buffer with startup method
if self._needs_startup:
self.startup.buffer(dt)
[docs]
def solve(self, f, J, dt):
"""Solves the implicit update equation using the optimizer of the engine.
Parameters
----------
f : array_like
evaluation of function
J : array_like
evaluation of jacobian of function
dt : float
integration timestep
Returns
-------
err : float
residual error of the fixed point update equation
"""
#not enough history for full order -> solve with startup method
if self._needs_startup:
err = self.startup.solve(f, J, dt)
self.x = self.startup.get()
return err
#fixed-point function update
g = self.F[self.n] * dt * f
for b, k in zip(self.history, self.K[self.n]):
g = g + b * k
#use the jacobian
if J is not None:
#optimizer step with block local jacobian
self.x, err = self.opt.step(self.x, g, self.F[self.n] * dt * J)
else:
#optimizer step (pure)
self.x, err = self.opt.step(self.x, g, None)
#return the fixed-point residual
return err
[docs]
def step(self, f, dt):
"""Performs the explicit timestep for (t+dt) based
on the state and input at (t).
Note
----
This is only required for the startup solver.
Parameters
----------
f : numeric, array[numeric]
evaluation of rhs function
dt : float
integration timestep
Returns
-------
success : bool
True if the timestep was successful
error : float
estimated error of the internal error controller
scale : float
estimated timestep rescale factor for error control
"""
#not enough histors -> step the startup solver
if self._needs_startup:
self.startup.step(f, dt)
self.x = self.startup.get()
return True, 0.0, 1.0
# SOLVERS ==============================================================================
[docs]
class BDF2(BDF):
"""Fixed-step 2nd order Backward Differentiation Formula (BDF).
Implicit linear multistep method. Uses the previous two solution points.
A-stable, suitable for stiff problems. Uses BDF1 for the first step.
Characteristics:
* Order: 2
* Implicit Multistep
* Fixed timestep only
* A-stable
"""
def __init__(self, *solver_args, **solver_kwargs):
super().__init__(*solver_args, **solver_kwargs)
#integration order (local)
self.n = 2
#longer history for BDF
self.history = deque([], maxlen=2)
[docs]
class BDF3(BDF):
"""Fixed-step 3rd order Backward Differentiation Formula (BDF).
Implicit linear multistep method. Uses the previous three solution points.
A(alpha)-stable, suitable for stiff problems. Uses lower orders for startup.
Characteristics:
* Order: 3
* Implicit Multistep
* Fixed timestep only
* A(alpha)-stable (:math:`\\alpha \\approx 86^\\circ`)
"""
def __init__(self, *solver_args, **solver_kwargs):
super().__init__(*solver_args, **solver_kwargs)
#integration order (local)
self.n = 3
#longer history for BDF
self.history = deque([], maxlen=3)
[docs]
class BDF4(BDF):
"""Fixed-step 4th order Backward Differentiation Formula (BDF).
Implicit linear multistep method. Uses the previous four solution points.
A(alpha)-stable, suitable for stiff problems. Uses lower orders for startup.
Characteristics:
* Order: 4
* Implicit Multistep
* Fixed timestep only
* A(alpha)-stable (:math:`\\alpha \\approx 73^\\circ`)
"""
def __init__(self, *solver_args, **solver_kwargs):
super().__init__(*solver_args, **solver_kwargs)
#integration order (local)
self.n = 4
#longer history for BDF
self.history = deque([], maxlen=4)
[docs]
class BDF5(BDF):
"""Fixed-step 5th order Backward Differentiation Formula (BDF).
Implicit linear multistep method. Uses the previous five solution points.
A(alpha)-stable, suitable for stiff problems. Uses lower orders for startup.
Characteristics:
* Order: 5
* Implicit Multistep
* Fixed timestep only
* A(alpha)-stable (:math:`\\alpha \\approx 51^\\circ`)
"""
def __init__(self, *solver_args, **solver_kwargs):
super().__init__(*solver_args, **solver_kwargs)
#integration order (local)
self.n = 5
#longer history for BDF
self.history = deque([], maxlen=5)
[docs]
class BDF6(BDF):
"""Fixed-step 6th order Backward Differentiation Formula (BDF).
Implicit linear multistep method. Uses the previous six solution points.
Not A-stable, stability region does not contain the entire left half-plane,
limiting its use for highly stiff problems compared to lower-order BDFs.
Uses lower orders for startup.
Characteristics:
* Order: 6
* Implicit Multistep
* Fixed timestep only
* Not A-stable (stability angle approx :math:`18^\\circ`)
"""
def __init__(self, *solver_args, **solver_kwargs):
super().__init__(*solver_args, **solver_kwargs)
#integration order (local)
self.n = 6
#longer history for BDF
self.history = deque([], maxlen=6)