Homec4science
Diffusion RationalROMPy (master) Browse

RROMPy/examples/3d/base/scattering_1d.py
5b3b6ab16a8emaster

Last Change
Raw File

scattering_1d.py
View Options

# 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/>.
#
import numpy as np
import fenics as fen
from rrompy.utilities.base.types import paramVal
from rrompy.solver.fenics import fenZERO, fenONE
from rrompy.hfengines.linear_problem.scattering_problem_engine import (
ScatteringProblemEngine)
from rrompy.utilities.base import verbosityManager as vbMng
from rrompy.solver.fenics import fenics2Sparse
class Scattering1d(ScatteringProblemEngine):
def __init__(self, mu0 : paramVal = [5.5, np.pi, 0.], n : int = 50,
degree_threshold : int = np.inf, verbosity : int = 10,
timestamp : bool = True):
super().__init__(mu0 = mu0, degree_threshold = degree_threshold,
verbosity = verbosity, timestamp = timestamp)
self._affinePoly = False
self.nAs = 3
self.npar = 3
self.rescalingExp = [1., 1., 1.]
self._L = np.real(mu0[1])
self.V = fen.FunctionSpace(fen.IntervalMesh(n, 0., self._L), "P", 1)
self.RobinBoundary = lambda x, on_b: (on_b and x[0] >= .5 * np.pi)
self.DirichletBoundary = "REST"
self.DirichletDatum = fenONE
def buildA(self):
"""Build terms of operator of linear system."""
if self.As[0] is None:
vbMng(self, "INIT", "Assembling operator term A0.", 20)
DirichletBC0 = fen.DirichletBC(self.V, fenZERO,
self.DirichletBoundary)
a0 = fen.dot(fen.grad(self.u), fen.grad(self.v)) * fen.dx
self.As[0] = fenics2Sparse(a0, {}, DirichletBC0, 1)
self.thAs[0] = self.getMonomialSingleWeight([0, 0, 0])
vbMng(self, "DEL", "Done assembling operator term.", 20)
if self.As[1] is None:
self.autoSetDS()
vbMng(self, "INIT", "Assembling operator term A1.", 20)
DirichletBC0 = fen.DirichletBC(self.V, fenZERO,
self.DirichletBoundary)
a1 = - self._L ** -1. * fen.dot(self.u, self.v) * self.ds(1)
self.As[1] = 1.j * fenics2Sparse(a1, {}, DirichletBC0, 0)
self.thAs[1] = self.getMonomialSingleWeight([1, 1, 1])
vbMng(self, "DEL", "Done assembling operator term.", 20)
if self.As[2] is None:
vbMng(self, "INIT", "Assembling operator term A2.", 20)
DirichletBC0 = fen.DirichletBC(self.V, fenZERO,
self.DirichletBoundary)
a2 = - self._L ** -2. * fen.dot(self.u, self.v) * fen.dx
self.As[2] = fenics2Sparse(a2, {}, DirichletBC0, 0)
self.thAs[2] = self.getMonomialSingleWeight([2, 2, 0])
vbMng(self, "DEL", "Done assembling operator term.", 20)
c4science · Help