scattering1.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	from scattering_1d 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))