|
|
@@ -1,4 +1,4 @@
|
|
|
-from charged_shells import patch_size, parameters, charge_distributions
|
|
|
+from charged_shells import patch_size, parameters, charge_distributions, expansion
|
|
|
from charged_shells import functions as fn
|
|
|
from scipy.special import eval_legendre
|
|
|
import numpy as np
|
|
|
@@ -19,7 +19,7 @@ def point_to_cap_magnitude(sigma_tilde: Array, a_bar: Array, theta0: Array, kapp
|
|
|
|
|
|
|
|
|
def ic_to_gauss(sigma_tilde, a_bar, params: ModelParams, l_max: int = 30,
|
|
|
- sigma0: float = 0) -> charge_distributions.create_gaussian_charge_expansion:
|
|
|
+ sigma0: float = 0) -> expansion.Expansion:
|
|
|
|
|
|
ex_mapped = charge_distributions.create_mapped_quad_expansion(a_bar=a_bar, kappaR=params.kappaR,
|
|
|
sigma_tilde=sigma_tilde, l_max=30, sigma0=sigma0)
|
|
|
@@ -36,7 +36,7 @@ def ic_to_gauss(sigma_tilde, a_bar, params: ModelParams, l_max: int = 30,
|
|
|
sigma1=gauss_sigma, l_max=l_max, sigma0=sigma0_mapped)
|
|
|
|
|
|
|
|
|
-def ic_to_cap(sigma_tilde, a_bar, params: ModelParams, l_max: int = 30, sigma0: float = 0) -> charge_distributions.create_spherical_cap_expansion:
|
|
|
+def ic_to_cap(sigma_tilde, a_bar, params: ModelParams, l_max: int = 30, sigma0: float = 0) -> expansion.Expansion:
|
|
|
|
|
|
ex_mapped = charge_distributions.create_mapped_quad_expansion(a_bar=a_bar, kappaR=params.kappaR,
|
|
|
sigma_tilde=sigma_tilde, l_max=30, sigma0=sigma0)
|
