WP1.set_quadrature(None) params["basis_size"] = 6 WP2 = HagedornWavepacket(params) WP2.set_coefficient(0, 0, 1) WP2.set_quadrature(None) HQ1 = HomogeneousQuadrature() HQ1.build_qr(max(WP1.get_basis_size())) HQ2 = HomogeneousQuadrature() HQ2.build_qr(max(WP2.get_basis_size())) IHQ = InhomogeneousQuadrature() IHQ.build_qr(nmax) Pibra = WP1.get_parameters() x = linspace(-4, 4, 4000) positions = linspace(-0.5, 2.5, 61) quads1 = [] quads2 = [] quads12 = [] for index, pos in enumerate(positions): print(pos) # Moving Gaussian WP2.set_parameters((1.0j, 1.0, 0.0, 0.0, pos))
params["eps"] = 0.2 params["basis_size"] = 6 params["ncomponents"] = 3 WP2 = HagedornWavepacket(params) WP2.set_parameters((1.0j, 1.0, 0.0, 0.0, 0.125)) WP2.set_quadrature(None) HQ1 = HomogeneousQuadrature() HQ1.build_qr(max(WP1.get_basis_size())) HQ2 = HomogeneousQuadrature() HQ2.build_qr(max(WP2.get_basis_size())) IHQ = InhomogeneousQuadrature() IHQ.build_qr(10) M1 = HQ1.build_matrix(WP1) M2 = HQ2.build_matrix(WP2) figure() matshow(real(M1)) savefig("M1r.png") figure() matshow(imag(M1)) savefig("M1i.png") figure()