scattering1.py
3ed8cef19cfcmaster

scattering1.py
View Options

			from rrompy.hfengines.linear_problem.tridimensional import Scattering1d as S1D
			import fenics as fen
			import numpy as np

			verb = 100

			mu0 = [4., np.pi, 0.]
			mu = [4.5, np.pi, 1.]

			n = 100

			solver = S1D(mu0 = mu0, n = n, verbosity = verb)
			u0 = solver.liftDirichletData()
			uh = solver.solve(mu)[0]
			#solver.plotmesh(figsize = (12, 2))
			#solver.plot(u0, what = ['REAL', 'IMAG'], figsize = (12, 4))
			print(solver.norm(uh))
			#solver.plot(uh, what = ['REAL', 'IMAG'], figsize = (12, 4))
			#solver.plot(solver.residual(mu0, uh)[0], name = 'res',
			# what = ['REAL', 'IMAG'], figsize = (12, 4))

			x = fen.SpatialCoordinate(solver.V.mesh())
			warps = [x * mu[1] / solver._L - x, x * solver._L / mu[1] - x]

			#solver.plotmesh(warps, figsize = (12, 2))
			#solver.plot(u0, warps, what = ['REAL', 'IMAG'], figsize = (12, 4))
			solver.plot(uh, warps, what = ['REAL', 'IMAG'], figsize = (12, 4))
			#solver.plot(solver.residual(mu0, uh)[0], warps, name = 'res',
			# what = ['REAL', 'IMAG'], figsize = (12, 4))