def test_monopole_M2P():
source_order = 0
order = source_order + TOTALORDER
cse = True
atomic = True
precision = 'double'
fmmgen.generate_code(order,
"MonopoleOrigin",
precision=precision,
CSE=cse,
cython=True,
potential=True,
field=True,
source_order=source_order,
atomic=atomic,
minpow=5,
harmonic_derivs=True,
language='c')
import pyximport
pyximport.install()
import MonopoleOrigin_wrap as fmm
# Locate charge +q at (0, 0, d/2)
# Locate charge -q at (0, 0, d/2)
# Compute multipole at origin
# We should get [0, 0, 0, q*d, ...] as the multipole moment
d = 2.0
q = 5.0
x1 = np.array([d, 0.0, 0.0])
x2 = np.array([20.0, 0.0, 0.0])
dx = x2 - x1
O = np.array([0.0, 0.0, 0.0])
Rinv = 1 / np.linalg.norm(dx)
F_python = np.array([
q * Rinv, q * dx[0] * (Rinv * Rinv * Rinv),
q * dx[1] * (Rinv * Rinv * Rinv), q * dx[2] * (Rinv * Rinv * Rinv)
])
print(f'F_python = {F_python}')
F0_last = 100
F1_last = 100
for Order in range(0, fmm.FMMGEN_MAXORDER):
Msize = Nterms(Order)
S = np.zeros(Msize)
S[0] = q
M = np.zeros(Msize)
F = np.zeros(4)
fmm.M2M(*(O - x1), S, M, Order)
print(M)
fmm.M2P(*(x2 - O), M, F, Order)
print(f'Order = {Order}, F = {F}')
assert np.abs((F[0] - F_python[0]) / F_python[0]) < F0_last
assert np.abs((F[1] - F_python[1]) / F_python[1]) < F1_last
assert F[2] == 0
assert F[3] == 0
def test_quadrupole_two_dipoles():
print('test_quadrupole_two_dipoles')
source_order = 0
order = source_order + TOTALORDER
cse = True
atomic = True
precision = 'double'
fmmgen.generate_code(order,
"QuadrupoleTwoDipoles",
precision=precision,
CSE=cse,
cython=True,
potential=True,
field=True,
source_order=source_order,
atomic=atomic,
minpow=5,
harmonic_derivs=True,
language='c')
import pyximport
pyximport.install()
import LinearQuadrupole_wrap as fmm
# Superimpose two dipoles, s.t. they have
# opposite orientations, and the +q charges
# sit on top of each other.
# Locate dipole (mux, 0, 0) at (-d/2, 0, 0)
# Locate charge (-mux, 0, 0) at (+d/2, 0, 0)
# Compute multipole at origin
# We should get [0, 0, 0, 0.0, ...] as the multipole moment
d = 2.0
mux = -5.0
xp = np.array([d, 0.0, 0.0])
xm = np.array([-d, 0.0, 0.0])
O = np.array([0.0, 0.0, 0.0])
for Order in range(2, fmm.FMMGEN_MAXORDER):
print(f'Quadrupole Test {Order}')
Msize = Nterms(Order)
S_px = np.zeros(Msize)
S_px[1] = -mux
S_mx = np.zeros(Msize)
S_mx[1] = mux
M = np.zeros(Msize)
fmm.M2M(*(O - xp), S_px, M, Order)
# print(f'M after dipole ({-mux}, 0, 0) at({}/2, 0, 0) = {M}')
fmm.M2M(*(O - xm), S_mx, M, Order)
print(M)
assert M[0] == 0.0
assert M[1] == 0.0
assert M[2] == 0.0
assert M[3] == 0.0
assert M[4] == 2 * mux * d
def test_linear_dipole():
source_order = 0
order = source_order + TOTALORDER
cse = True
atomic = True
precision = 'double'
fmmgen.generate_code(order,
"LinearDipole",
precision=precision,
CSE=cse,
cython=True,
potential=True,
field=True,
source_order=source_order,
atomic=atomic,
minpow=5,
harmonic_derivs=True,
language='c')
import pyximport
pyximport.install()
import LinearDipole_wrap as fmm
# Locate charge +q at (0, 0, d/2)
# Locate charge -q at (0, 0, d/2)
# Compute multipole at origin
# We should get [0, 0, 0, q*d, ...] as the multipole moment
d = 1.0
q = 5.0
x1 = np.array([d, 0.0, 0.0])
x2 = np.array([-d, 0.0, 0.0])
O = np.array([0.0, 0.0, 0.0])
for Order in range(1, fmm.FMMGEN_MAXORDER):
Msize = Nterms(Order)
S_px = np.zeros(Msize)
S_px[0] = q
S_mx = np.zeros(Msize)
S_mx[0] = -q
M = np.zeros(Msize)
fmm.M2M(*(O - x1), S_px, M, Order)
fmm.M2M(*(O - x2), S_mx, M, Order)
print(M)
assert M[0] == 0.0
assert M[1] == -2 * q * d
assert M[2] == 0.0
assert M[3] == 0.0
import fmmgen
source_order = 0
order = source_order + 12
cse = True
atomic = True
precision = 'double'
fmmgen.generate_code(order,
"operators",
precision=precision,
CSE=cse,
cython=False,
potential=True,
field=True,
source_order=source_order,
atomic=atomic,
minpow=11,
harmonic_derivs=True,
language='c++')
def test_linear_quadrupole():
print('test_linear_quadrupole')
source_order = 0
order = source_order + TOTALORDER
cse = True
atomic = True
precision = 'double'
fmmgen.generate_code(order,
"LinearQuadrupole",
precision=precision,
CSE=cse,
cython=True,
potential=True,
field=True,
source_order=source_order,
atomic=atomic,
minpow=5,
harmonic_derivs=True,
language='c')
import pyximport
pyximport.install()
import LinearQuadrupole_wrap as fmm
# Superimpose two dipoles, s.t. they have
# opposite orientations, and the +q charges
# sit on top of each other.
# Locate charge -q at (-d, 0, 0)
# Locate charge -q at (+d, 0, 0)
# Locate charge +2q at (0, 0, 0)
# Compute multipole at origin
# We should get [0, 0, 0, 0, ...] as the multipole moment
d = 2.0
q = 5.0
x1 = np.array([-d, 0.0, 0.0])
x2 = np.array([d, 0.0, 0.0])
O = np.array([0.0, 0.0, 0.0])
for Order in range(2, fmm.FMMGEN_MAXORDER):
Msize = Nterms(Order)
S_mx = np.zeros(Msize)
S_mx[0] = -q
S_px = np.zeros(Msize)
S_px[0] = -q
S_O = np.zeros(Msize)
S_O[0] = 2 * q
M = np.zeros(Msize)
fmm.M2M(*(O - x1), S_mx, M, Order)
print(M)
fmm.M2M(*(O - x2), S_px, M, Order)
fmm.M2M(*(O - O), S_O, M, Order)
print(M)
assert M[0] == 0.0
assert M[1] == 0.0
assert M[2] == 0.0
assert M[3] == 0.0
assert M[4] == -2 * d * q
import fmmgen
order = 8
source_order = 1
cse = True
atomic = True
precision='double'
fmmgen.generate_code(order, "operators",
precision=precision,
CSE=cse,
cython=False,
harmonic_derivs=True,
potential=False,
field=True,
source_order=source_order,
atomic=atomic)
print("Generating code for source order", k)
# fmmgen.generate_code(order, f"pot_m{k}",
# src_dir="../../cppe/core/fmm",
# include_dir="../../cppe/core/fmm",
# precision=precision,
# CSE=cse,
# cython=False,
# potential=True,
# field=False,
# source_order=k,
# atomic=atomic, minpow=11,
# harmonic_derivs=True,
# language='c++', name_prefix=f"pot_m{k}_",
# write_defines=False)
fmmgen.generate_code(order,
f"field_m{k}",
src_dir="../../cppe/core/fmm",
include_dir="../../cppe/core/fmm",
precision=precision,
CSE=cse,
cython=False,
potential=False,
field=True,
source_order=k,
atomic=atomic,
minpow=11,
harmonic_derivs=True,
language='c++',
name_prefix=f"field_m{k}_",
write_defines=False,
write_templates=True)