34-poisson…

static_solver.py
View Options

			import copy
			import scipy.sparse.linalg as spl
			from . import export
			from . import fe


			@export
			class StaticSolver(fe.Solver):
			def __init__(self, model):
			self._model = model

			def _assembleResidual(self):
			self._r = self._model.f_ext - self._model.f_int

			def _assembleJacobian(self):
			self._J = copy.copy(self._model.K)
			self._model.applyDirichletBC()

			self._zero_rows(self._J, self._model.blocked)

			def solveStep(self):
			self._assembleJacobian()
			self._assembleResidual()
			x = spl.spsolve(self._J, self._r)
			self._model.u += x