"""
These solvers solve the (pure) bidomain equations on the form: find
the transmembrane potential :math:`v = v(x, t)` and the extracellular
potential :math:`u = u(x, t)` such that
.. math::
v_t - \mathrm{div} ( G_i v + G_i u) = I_s
\mathrm{div} (G_i v + (G_i + G_e) u) = I_a
where the subscript :math:`t` denotes the time derivative; :math:`G_x`
denotes a weighted gradient: :math:`G_x = M_x \mathrm{grad}(v)` for
:math:`x \in \{i, e\}`, where :math:`M_i` and :math:`M_e` are the
intracellular and extracellular cardiac conductivity tensors,
respectively; :math:`I_s` and :math:`I_a` are prescribed input. In
addition, initial conditions are given for :math:`v`:
.. math::
v(x, 0) = v_0
Finally, boundary conditions must be prescribed. For now, this solver
assumes pure homogeneous Neumann boundary conditions for :math:`v` and
:math:`u` and enforces the additional average value zero constraint
for u.
"""
# Copyright (C) 2013 Marie E. Rognes (meg@simula.no)
# Use and modify at will
# Last changed: 2013-04-18
__all__ = ["BasicBidomainSolver", "BidomainSolver"]
from cbcbeat.dolfinimport import *
from cbcbeat.markerwisefield import *
from cbcbeat.utils import end_of_time, annotate_kwargs
[docs]class BasicBidomainSolver(object):
"""This solver is based on a theta-scheme discretization in time
and CG_1 x CG_1 (x R) elements in space.
.. note::
For the sake of simplicity and consistency with other solver
objects, this solver operates on its solution fields (as state
variables) directly internally. More precisely, solve (and
step) calls will act by updating the internal solution
fields. It implies that initial conditions can be set (and are
intended to be set) by modifying the solution fields prior to
simulation.
*Arguments*
mesh (:py:class:`dolfin.Mesh`)
The spatial domain (mesh)
time (:py:class:`dolfin.Constant` or None)
A constant holding the current time. If None is given, time is
created for you, initialized to zero.
M_i (:py:class:`ufl.Expr`)
The intracellular conductivity tensor (as an UFL expression)
M_e (:py:class:`ufl.Expr`)
The extracellular conductivity tensor (as an UFL expression)
I_s (:py:class:`dict`, optional)
A typically time-dependent external stimulus given as a dict,
with domain markers as the key and a
:py:class:`dolfin.Expression` as values. NB: it is assumed
that the time dependence of I_s is encoded via the 'time'
Constant.
I_a (:py:class:`dolfin.Expression`, optional)
A (typically time-dependent) external applied current
v\_ (:py:class:`ufl.Expr`, optional)
Initial condition for v. A new :py:class:`dolfin.Function`
will be created if none is given.
params (:py:class:`dolfin.Parameters`, optional)
Solver parameters
"""
def __init__(self, mesh, time, M_i, M_e, I_s=None, I_a=None, v_=None,
params=None):
# Check some input
assert isinstance(mesh, Mesh), \
"Expecting mesh to be a Mesh instance, not %r" % mesh
assert isinstance(time, Constant) or time is None, \
"Expecting time to be a Constant instance (or None)."
assert isinstance(params, Parameters) or params is None, \
"Expecting params to be a Parameters instance (or None)"
self._nullspace_basis = None
# Store input
self._mesh = mesh
self._time = time
self._M_i = M_i
self._M_e = M_e
self._I_s = I_s
self._I_a = I_a
# Initialize and update parameters if given
self.parameters = self.default_parameters()
if params is not None:
self.parameters.update(params)
# Set-up function spaces
k = self.parameters["polynomial_degree"]
Ve = FiniteElement("CG", self._mesh.ufl_cell(), k)
V = FunctionSpace(self._mesh, "CG", k)
Ue = FiniteElement("CG", self._mesh.ufl_cell(), k)
U = FunctionSpace(self._mesh, "CG", k)
use_R = self.parameters["use_avg_u_constraint"]
if use_R:
Re = FiniteElement("R", self._mesh.ufl_cell(), 0)
R = FunctionSpace(self._mesh, "R", 0)
self.VUR = FunctionSpace(mesh, MixedElement((Ve, Ue, Re)))
else:
self.VUR = FunctionSpace(mesh, MixedElement((Ve, Ue)))
self.V = V
# Set-up solution fields:
if v_ is None:
self.merger = FunctionAssigner(V, self.VUR.sub(0))
self.v_ = Function(V, name="v_")
else:
debug("Experimental: v_ shipped from elsewhere.")
self.merger = None
self.v_ = v_
self.vur = Function(self.VUR, name="vur")
# Figure out whether we should annotate or not
self._annotate_kwargs = annotate_kwargs(self.parameters)
@property
def time(self):
"The internal time of the solver."
return self._time
[docs] def solution_fields(self):
"""
Return tuple of previous and current solution objects.
Modifying these will modify the solution objects of the solver
and thus provides a way for setting initial conditions for
instance.
*Returns*
(previous v, current vur) (:py:class:`tuple` of :py:class:`dolfin.Function`)
"""
return (self.v_, self.vur)
[docs] def solve(self, interval, dt=None):
"""
Solve the discretization on a given time interval (t0, t1)
with a given timestep dt and return generator for a tuple of
the interval and the current solution.
*Arguments*
interval (:py:class:`tuple`)
The time interval for the solve given by (t0, t1)
dt (int, optional)
The timestep for the solve. Defaults to length of interval
*Returns*
(timestep, solution_fields) via (:py:class:`genexpr`)
*Example of usage*::
# Create generator
solutions = solver.solve((0.0, 1.0), 0.1)
# Iterate over generator (computes solutions as you go)
for (interval, solution_fields) in solutions:
(t0, t1) = interval
v_, vur = solution_fields
# do something with the solutions
"""
timer = Timer("PDE step")
# Initial set-up
# Solve on entire interval if no interval is given.
(T0, T) = interval
if dt is None:
dt = (T - T0)
t0 = T0
t1 = T0 + dt
# Step through time steps until at end time
while (True) :
info("Solving on t = (%g, %g)" % (t0, t1))
self.step((t0, t1))
# Yield solutions
yield (t0, t1), self.solution_fields()
# Break if this is the last step
if end_of_time(T, t0, t1, dt):
break
# If not: update members and move to next time
# Subfunction assignment would be good here.
if isinstance(self.v_, Function):
self.merger.assign(self.v_, self.vur.sub(0))
else:
debug("Assuming that v_ is updated elsewhere. Experimental.")
t0 = t1
t1 = t0 + dt
[docs] def step(self, interval):
"""
Solve on the given time interval (t0, t1).
*Arguments*
interval (:py:class:`tuple`)
The time interval (t0, t1) for the step
*Invariants*
Assuming that v\_ is in the correct state for t0, gives
self.vur in correct state at t1.
"""
timer = Timer("PDE step")
# Extract interval and thus time-step
(t0, t1) = interval
k_n = Constant(t1 - t0)
theta = self.parameters["theta"]
# Extract conductivities
M_i, M_e = self._M_i, self._M_e
# Define variational formulation
use_R = self.parameters["use_avg_u_constraint"]
if use_R:
(v, u, l) = TrialFunctions(self.VUR)
(w, q, lamda) = TestFunctions(self.VUR)
else:
(v, u) = TrialFunctions(self.VUR)
(w, q) = TestFunctions(self.VUR)
Dt_v = (v - self.v_)/k_n
v_mid = theta*v + (1.0 - theta)*self.v_
# Set time
t = t0 + theta*(t1 - t0)
self.time.assign(t)
# Define spatial integration domains:
(dz, rhs) = rhs_with_markerwise_field(self._I_s, self._mesh, w)
theta_parabolic = (inner(M_i*grad(v_mid), grad(w))*dz()
+ inner(M_i*grad(u), grad(w))*dz())
theta_elliptic = (inner(M_i*grad(v_mid), grad(q))*dz()
+ inner((M_i + M_e)*grad(u), grad(q))*dz())
G = Dt_v*w*dz() + theta_parabolic + theta_elliptic
if use_R:
G += (lamda*u + l*q)*dz()
# Add applied current as source in elliptic equation if
# applicable
if self._I_a:
G -= self._I_a*q*dz()
# Add applied stimulus as source in parabolic equation if
# applicable
G -= rhs
# Define variational problem
a, L = system(G)
pde = LinearVariationalProblem(a, L, self.vur)
# Set-up solver
solver = LinearVariationalSolver(pde)
solver.parameters.update(self.parameters["linear_variational_solver"])
solver.solve()
[docs] @staticmethod
def default_parameters():
"""Initialize and return a set of default parameters
*Returns*
A set of parameters (:py:class:`dolfin.Parameters`)
To inspect all the default parameters, do::
info(BasicBidomainSolver.default_parameters(), True)
"""
params = Parameters("BasicBidomainSolver")
params.add("enable_adjoint", True)
params.add("theta", 0.5)
params.add("polynomial_degree", 1)
params.add("use_avg_u_constraint", True)
params.add(LinearVariationalSolver.default_parameters())
return params
[docs]class BidomainSolver(BasicBidomainSolver):
__doc__ = BasicBidomainSolver.__doc__
def __init__(self, mesh, time, M_i, M_e, I_s=None, I_a=None, v_=None,
params=None):
# Call super-class
BasicBidomainSolver.__init__(self, mesh, time, M_i, M_e,
I_s=I_s, I_a=I_a, v_=v_,
params=params)
# Check consistency of parameters first
if self.parameters["enable_adjoint"] and not dolfin_adjoint:
warning("'enable_adjoint' is set to True, but no "\
"dolfin_adjoint installed.")
# Mark the timestep as unset
self._timestep = None
@property
def linear_solver(self):
"""The linear solver (:py:class:`dolfin.LUSolver` or
:py:class:`dolfin.PETScKrylovSolver`)."""
return self._linear_solver
def _create_linear_solver(self):
"Helper function for creating linear solver based on parameters."
solver_type = self.parameters["linear_solver_type"]
if solver_type == "direct":
solver = LUSolver(self._lhs_matrix)
solver.parameters.update(self.parameters["lu_solver"])
solver.parameters["reuse_factorization"] = True
update_routine = self._update_lu_solver
elif solver_type == "iterative":
# Initialize KrylovSolver with matrix
alg = self.parameters["algorithm"]
prec = self.parameters["preconditioner"]
debug("Creating PETSCKrylovSolver with %s and %s" % (alg, prec))
if prec == "fieldsplit":
# Argh. DOLFIN won't let you construct a PETScKrylovSolver with fieldsplit. Sigh ..
solver = PETScKrylovSolver()
# FIXME: work around DOLFIN bug #583. Just deleted this when fixed.
solver.parameters.convergence_norm_type = "preconditioned"
#solver.parameters["preconditioner"]["structure"] = "same" # MER this should be set by user, and is below
solver.parameters.update(self.parameters["petsc_krylov_solver"])
solver.set_operator(self._lhs_matrix)
# Initialize the KSP directly:
ksp = solver.ksp()
ksp.setType(alg)
ksp.pc.setType(prec)
ksp.setOptionsPrefix("bidomain_") # it's really stupid, solver.set_options_prefix() doesn't work
# Set various options (by default) for the fieldsplit
# approach to solving the bidomain equations.
# FIXME: This needs a try
from petsc4py import PETSc
# Patrick believes that the fieldsplit index sets
# should already be set from the assembled matrix.
# Now let's set some default options for the solver.
opts = PETSc.Options("bidomain_")
if "pc_fieldsplit_type" not in opts: opts["pc_fieldsplit_type"] = "symmetric_multiplicative"
if "fieldsplit_0_ksp_type" not in opts: opts["fieldsplit_0_ksp_type"] = "preonly"
if "fieldsplit_1_ksp_type" not in opts: opts["fieldsplit_1_ksp_type"] = "preonly"
if "fieldsplit_0_pc_type" not in opts: opts["fieldsplit_0_pc_type"] = "hypre"
if "fieldsplit_1_pc_type" not in opts: opts["fieldsplit_1_pc_type"] = "hypre"
ksp.setFromOptions()
ksp.setUp()
else:
solver = PETScKrylovSolver(alg, prec)
solver.set_operator(self._lhs_matrix)
# Still waiting for that bug fix:
solver.parameters.convergence_norm_type = "preconditioned"
solver.parameters.update(self.parameters["petsc_krylov_solver"])
# Set nullspace if present. We happen to know that the
# transpose nullspace is the same as the nullspace (easy
# to prove from matrix structure).
if self.parameters["use_avg_u_constraint"]:
# Nothing to do, no null space
pass
else:
# If dolfin-adjoint is enabled and installled: set the solver nullspace
if dolfin_adjoint:
solver.set_nullspace(self.nullspace)
solver.set_transpose_nullspace(self.nullspace)
# Otherwise, set the nullspace in the operator
# directly.
else:
A = as_backend_type(self._lhs_matrix)
A.set_nullspace(self.nullspace)
update_routine = self._update_krylov_solver
else:
error("Unknown linear_solver_type given: %s" % solver_type)
return (solver, update_routine)
@property
def nullspace(self):
if self._nullspace_basis is None:
null_vector = Vector(self.vur.vector())
self.VUR.sub(1).dofmap().set(null_vector, 1.0)
null_vector *= 1.0/null_vector.norm("l2")
self._nullspace_basis = VectorSpaceBasis([null_vector])
return self._nullspace_basis
[docs] @staticmethod
def default_parameters():
"""Initialize and return a set of default parameters
*Returns*
A set of parameters (:py:class:`dolfin.Parameters`)
To inspect all the default parameters, do::
info(BidomainSolver.default_parameters(), True)
"""
params = Parameters("BidomainSolver")
params.add("enable_adjoint", True)
params.add("theta", 0.5)
params.add("polynomial_degree", 1)
# Set default solver type to be iterative
params.add("linear_solver_type", "iterative")
params.add("use_avg_u_constraint", False)
# Set default iterative solver choices (used if iterative
# solver is invoked)
params.add("algorithm", "cg")
params.add("preconditioner", "petsc_amg")
#params.add("preconditioner", "fieldsplit") # This seg faults
# Add default parameters from both LU and Krylov solvers
params.add(LUSolver.default_parameters())
petsc_params = PETScKrylovSolver.default_parameters()
# FIXME: work around DOLFIN bug #583. Just deleted this when fixed.
petsc_params.convergence_norm_type = "preconditioned"
params.add(petsc_params)
# Customize default parameters for LUSolver
params["lu_solver"]["same_nonzero_pattern"] = True
# Customize default parameters for PETScKrylovSolver
#params["petsc_krylov_solver"]["preconditioner"]["structure"] = "same"
return params
[docs] def variational_forms(self, k_n):
"""Create the variational forms corresponding to the given
discretization of the given system of equations.
*Arguments*
k_n (:py:class:`ufl.Expr` or float)
The time step
*Returns*
(lhs, rhs) (:py:class:`tuple` of :py:class:`ufl.Form`)
"""
# Extract theta parameter and conductivities
theta = self.parameters["theta"]
M_i = self._M_i
M_e = self._M_e
# Define variational formulation
use_R = self.parameters["use_avg_u_constraint"]
if use_R:
(v, u, l) = TrialFunctions(self.VUR)
(w, q, lamda) = TestFunctions(self.VUR)
else:
(v, u) = TrialFunctions(self.VUR)
(w, q) = TestFunctions(self.VUR)
# Set-up measure and rhs from stimulus
(dz, rhs) = rhs_with_markerwise_field(self._I_s, self._mesh, w)
# Set-up variational problem
Dt_v_k_n = (v - self.v_)
v_mid = theta*v + (1.0 - theta)*self.v_
theta_parabolic = (inner(M_i*grad(v_mid), grad(w))*dz()
+ inner(M_i*grad(u), grad(w))*dz())
theta_elliptic = (inner(M_i*grad(v_mid), grad(q))*dz()
+ inner((M_i + M_e)*grad(u), grad(q))*dz())
G = (Dt_v_k_n*w*dz() + k_n*theta_parabolic + k_n*theta_elliptic
- k_n*rhs)
if use_R:
G += k_n*(lamda*u + l*q)*dz()
# Add applied current as source in elliptic equation if
# applicable
if self._I_a:
G -= k_n*self._I_a*q*dz()
(a, L) = system(G)
return (a, L)
[docs] def step(self, interval):
"""
Solve on the given time step (t0, t1).
*Arguments*
interval (:py:class:`tuple`)
The time interval (t0, t1) for the step
*Invariants*
Assuming that v\_ is in the correct state for t0, gives
self.vur in correct state at t1.
"""
timer = Timer("PDE step")
solver_type = self.parameters["linear_solver_type"]
# Extract interval and thus time-step
(t0, t1) = interval
dt = t1 - t0
theta = self.parameters["theta"]
t = t0 + theta*dt
self.time.assign(t)
# Update matrix and linear solvers etc as needed
if self._timestep is None:
self._timestep = Constant(dt)
(self._lhs, self._rhs) = self.variational_forms(self._timestep)
# Preassemble left-hand side and initialize right-hand side vector
debug("Preassembling bidomain matrix (and initializing vector)")
self._lhs_matrix = assemble(self._lhs, **self._annotate_kwargs)
self._rhs_vector = Vector(self._mesh.mpi_comm(), self._lhs_matrix.size(0))
self._lhs_matrix.init_vector(self._rhs_vector, 0)
# Create linear solver (based on parameter choices)
self._linear_solver, self._update_solver = self._create_linear_solver()
else:
timestep_unchanged = (abs(dt - float(self._timestep)) < 1.e-12)
self._update_solver(timestep_unchanged, dt)
# Assemble right-hand-side
assemble(self._rhs, tensor=self._rhs_vector, **self._annotate_kwargs)
# Solve problem
self.linear_solver.solve(self.vur.vector(), self._rhs_vector,
**self._annotate_kwargs)
def _update_lu_solver(self, timestep_unchanged, dt):
"""Helper function for updating an LUSolver depending on
whether timestep has changed."""
# Update reuse of factorization parameter in accordance with
# changes in timestep
if timestep_unchanged:
debug("Timestep is unchanged, reusing LU factorization")
else:
debug("Timestep has changed, updating LU factorization")
if dolfin_adjoint and self.parameters["enable_adjoint"]:
raise ValueError("dolfin-adjoint doesn't support changing timestep (yet)")
# Update stored timestep
# FIXME: dolfin_adjoint still can't annotate constant assignment.
self._timestep.assign(Constant(dt))#, annotate=annotate)
# Reassemble matrix
assemble(self._lhs, tensor=self._lhs_matrix,
**self._annotate_kwargs)
(self._linear_solver, dummy) = self._create_linear_solver()
def _update_krylov_solver(self, timestep_unchanged, dt):
"""Helper function for updating a KrylovSolver depending on
whether timestep has changed."""
# Update reuse of preconditioner parameter in accordance with
# changes in timestep
if timestep_unchanged:
debug("Timestep is unchanged, reusing preconditioner")
else:
debug("Timestep has changed, updating preconditioner")
if dolfin_adjoint and self.parameters["enable_adjoint"]:
raise ValueError("dolfin-adjoint doesn't support changing timestep (yet)")
# Update stored timestep
self._timestep.assign(Constant(dt))#, annotate=annotate)
# Reassemble matrix
assemble(self._lhs, tensor=self._lhs_matrix, **self._annotate_kwargs)
# Make new Krylov solver
(self._linear_solver, dummy) = self._create_linear_solver()
# Set nonzero initial guess if it indeed is nonzero
if (self.vur.vector().norm("l2") > 1.e-12):
debug("Initial guess is non-zero.")
self.linear_solver.parameters["nonzero_initial_guess"] = True