def translate_from(self, src_expansion, src_coeff_exprs, src_rscale,
dvec, tgt_rscale, sac=None):
if not isinstance(src_expansion, type(self)):
raise RuntimeError("do not know how to translate %s to %s"
% (type(src_expansion).__name__,
type(self).__name__))
if not self.use_rscale:
src_rscale = 1
tgt_rscale = 1
from sumpy.symbolic import sym_real_norm_2
dvec_len = sym_real_norm_2(dvec)
bessel_j = sym.Function("bessel_j")
new_center_angle_rel_old_center = sym.atan2(dvec[1], dvec[0])
arg_scale = self.get_bessel_arg_scaling()
translated_coeffs = []
for j in self.get_coefficient_identifiers():
translated_coeffs.append(
sum(src_coeff_exprs[src_expansion.get_storage_index(m)]
* bessel_j(m - j, arg_scale * dvec_len)
* src_rscale ** abs(m)
/ tgt_rscale ** abs(j)
* sym.exp(sym.I * (m - j) * new_center_angle_rel_old_center)
for m in src_expansion.get_coefficient_identifiers()))
return translated_coeffs
def translate_from(self, src_expansion, src_coeff_exprs, src_rscale, dvec,
tgt_rscale):
from sumpy.symbolic import sym_real_norm_2
if not self.use_rscale:
src_rscale = 1
tgt_rscale = 1
arg_scale = self.get_bessel_arg_scaling()
if isinstance(src_expansion, type(self)):
dvec_len = sym_real_norm_2(dvec)
bessel_j = sym.Function("bessel_j")
new_center_angle_rel_old_center = sym.atan2(dvec[1], dvec[0])
translated_coeffs = []
for l in self.get_coefficient_identifiers():
translated_coeffs.append(
sum(src_coeff_exprs[src_expansion.get_storage_index(m)] *
bessel_j(m - l, arg_scale * dvec_len) /
src_rscale**abs(m) * tgt_rscale**abs(l) *
sym.exp(sym.I *
(m - l) * -new_center_angle_rel_old_center)
for m in src_expansion.get_coefficient_identifiers()))
return translated_coeffs
if isinstance(src_expansion, self.mpole_expn_class):
dvec_len = sym_real_norm_2(dvec)
hankel_1 = sym.Function("hankel_1")
new_center_angle_rel_old_center = sym.atan2(dvec[1], dvec[0])
translated_coeffs = []
for l in self.get_coefficient_identifiers():
translated_coeffs.append(
sum((-1)**l * hankel_1(m + l, arg_scale * dvec_len) *
src_rscale**abs(m) * tgt_rscale**abs(l) *
sym.exp(sym.I *
(m + l) * new_center_angle_rel_old_center) *
src_coeff_exprs[src_expansion.get_storage_index(m)]
for m in src_expansion.get_coefficient_identifiers()))
return translated_coeffs
raise RuntimeError("do not know how to translate %s to %s" %
(type(src_expansion).__name__, type(self).__name__))
def evaluate(self, coeffs, bvec, rscale):
if not self.use_rscale:
rscale = 1
from sumpy.symbolic import sym_real_norm_2
bessel_j = sym.Function("bessel_j")
bvec_len = sym_real_norm_2(bvec)
target_angle_rel_center = sym.atan2(bvec[1], bvec[0])
arg_scale = self.get_bessel_arg_scaling()
return sum(coeffs[self.get_storage_index(l)] *
self.kernel.postprocess_at_target(
bessel_j(l, arg_scale * bvec_len) / rscale**abs(l) *
sym.exp(sym.I * l * -target_angle_rel_center), bvec)
for l in self.get_coefficient_identifiers())
def evaluate(self, coeffs, bvec, rscale, sac=None):
if not self.use_rscale:
rscale = 1
from sumpy.symbolic import sym_real_norm_2
hankel_1 = sym.Function("hankel_1")
bvec_len = sym_real_norm_2(bvec)
target_angle_rel_center = sym.atan2(bvec[1], bvec[0])
arg_scale = self.get_bessel_arg_scaling()
return sum(coeffs[self.get_storage_index(c)] *
self.kernel.postprocess_at_target(
hankel_1(c, arg_scale * bvec_len) * rscale**abs(c) *
sym.exp(sym.I * c * target_angle_rel_center), bvec)
for c in self.get_coefficient_identifiers())
def coefficients_from_source(self, avec, bvec, rscale):
if not self.use_rscale:
rscale = 1
from sumpy.symbolic import sym_real_norm_2
hankel_1 = sym.Function("hankel_1")
arg_scale = self.get_bessel_arg_scaling()
# The coordinates are negated since avec points from source to center.
source_angle_rel_center = sym.atan2(-avec[1], -avec[0])
avec_len = sym_real_norm_2(avec)
return [
self.kernel.postprocess_at_source(
hankel_1(l, arg_scale * avec_len) * rscale**abs(l) *
sym.exp(sym.I * l * source_angle_rel_center), avec)
for l in self.get_coefficient_identifiers()
]
def coefficients_from_source(self, kernel, avec, bvec, rscale, sac=None):
if not self.use_rscale:
rscale = 1
if kernel is None:
kernel = self.kernel
from sumpy.symbolic import sym_real_norm_2
bessel_j = sym.Function("bessel_j")
avec_len = sym_real_norm_2(avec)
arg_scale = self.get_bessel_arg_scaling()
# The coordinates are negated since avec points from source to center.
source_angle_rel_center = sym.atan2(-avec[1], -avec[0])
return [
kernel.postprocess_at_source(
bessel_j(c, arg_scale * avec_len)
/ rscale ** abs(c)
* sym.exp(sym.I * c * -source_angle_rel_center),
avec)
for c in self.get_coefficient_identifiers()]
