funnel_output_engine.py
6c4eee04e0dfmaster

funnel_output_engine.py
View Options

			import numpy as np
			import fenics as fen
			import mshr
			from rrompy.hfengines.fenics_engines import ScatteringProblemEngine
			from rrompy.solver.fenics import fenics2Sparse

			class FunnelOutputEngine(ScatteringProblemEngine):
			def __init__(self, k0:float, n:int):
			super().__init__(mu0 = [k0])
			mesh = mshr.generate_mesh(mshr.Circle(fen.Point(0., 0.), 5.)
			- mshr.Rectangle(fen.Point(-5., -5.),
			fen.Point(-1., 5.))
			- mshr.Rectangle(fen.Point(-1., -5.),
			fen.Point(0., -1.))
			- mshr.Rectangle(fen.Point(-1., 1.),
			fen.Point(0., 5.)), n)
			self.V = fen.FunctionSpace(mesh, "P", 1)
			eps = 1e-4
			self.DirichletBoundary = (lambda x, on_boundary:
			on_boundary and x[0] < -1. + eps)
			self.NeumannBoundary = (lambda x, on_boundary:
			on_boundary and x[0] >= -1. + eps
			and x[0] < eps)
			self.RobinBoundary = "REST"
			y = fen.SpatialCoordinate(self.V.mesh())[1]
			self.DirichletDatum = 1. + .25 * fen.sin(.5 * np.pi * y)

			self.autoSetDS()
			l2R0 = fen.inner(self.u, self.v) * self.ds(1)
			L2R0 = fenics2Sparse(l2R0, {}, None, -1)
			bcR = np.where(np.abs(L2R0.dot(np.ones(L2R0.shape[1]))) > 1e-10)[0]
			bcR = bcR[np.argsort(self.V.tabulate_dof_coordinates()[bcR, 1])]
			self._C = np.zeros((len(bcR), L2R0.shape[1]))
			self._C[np.arange(len(bcR)), bcR] = 1.
			self.outputNormMatrix = L2R0[bcR][:, bcR]