generic_problem_engine.py
c6af74908c2cmaster

generic_problem_engine.py
View Options

			#!/usr/bin/python
			# Copyright (C) 2018 by the RROMPy authors
			#
			# This file is part of RROMPy.
			#
			# RROMPy is free software: you can redistribute it and/or modify
			# it under the terms of the GNU Lesser General Public License as published by
			# the Free Software Foundation, either version 3 of the License, or
			# (at your option) any later version.
			#
			# RROMPy is distributed in the hope that it will be useful,
			# but WITHOUT ANY WARRANTY; without even the implied warranty of
			# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
			# GNU Lesser General Public License for more details.
			#
			# You should have received a copy of the GNU Lesser General Public License
			# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
			#

			from abc import abstractmethod
			import numpy as np
			import scipy.sparse as scsp
			from rrompy.utilities.types import Np1D, Np2D, Tuple, List

			__all__ = ['GenericProblemEngine']

			class GenericProblemEngine:
			"""
			Generic solver for parametric problems.

			Attributes:
			As: Scipy sparse array representation (in CSC format) of As.
			bs: Numpy array representation of bs.
			"""
			Aterms, bterms = 1, 1

			functional = lambda self, u: 0.

			@abstractmethod
			def energyNormMatrix(self) -> Np2D:
			"""Sparse matrix (in CSR format) representative of scalar product."""
			pass

			@abstractmethod
			def assembleA(self):
			"""Assemble matrix blocks of linear system."""
			self.As = [None] * 1
			pass

			@abstractmethod
			def assembleb(self):
			"""Assemble matrix blocks of linear system."""
			self.bs = [None] * 1
			pass

			def A(self, mu:complex) -> Np2D:
			"""Assemble matrix of linear system."""
			self.assembleA()
			A = self.As[0]
			for j in range(1, len(self.As)):
			A = A + np.power(mu, j) * self.As[j]
			return A

			def b(self, mu:complex) -> Np1D:
			"""Assemble RHS of linear system."""
			self.assembleb()
			b = self.bs[0]
			for j in range(1, len(self.bs)):
			b = b + np.power(mu, j) * self.bs[j]
			return b

			def solve(self, mu:complex) -> Np1D:
			"""Find solution of linear system."""
			return scsp.linalg.spsolve(self.A(mu), self.b(mu))

			def getLSBlocks(self) -> Tuple[List[Np2D], List[Np1D]]:
			"""Get blocks of linear system."""
			self.assembleA()
			self.assembleb()
			return self.As, self.bs

RROMPy/rrompy/hfengines/fenics/generic_problem_engine.pyc6af74908c2cmaster

generic_problem_engine.pyView Options

RROMPy/rrompy/hfengines/fenics/generic_problem_engine.py
c6af74908c2cmaster

generic_problem_engine.py
View Options