def create_from_code(self):
F0 = 933e6
K_RADAR = -2 * F0 * 2 * np.pi / const.c
self.legend_txt = [
"Maxwellian",
r"$\kappa = 20$",
r"$\kappa = 8$",
r"$\kappa = 3$",
]
kappa = [20, 8, 3]
sys_set = {
"K_RADAR": K_RADAR,
"B": 50000e-9,
"MI": 16,
"NE": 2e11,
"NU_E": 0,
"NU_I": 0,
"T_E": 5000,
"T_I": 2000,
"T_ES": 90000,
"THETA": 0 * np.pi / 180,
"Z": 599,
"mat_file": "fe_zmuE-07.mat",
}
params = {"kappa": 20, "vdf": "real_data", "area": False}
self.f, s, meta_data = isr.isr_spectrum("maxwell", sys_set, **params)
self.data.append(s)
for k in kappa:
params["kappa"] = k
self.f, s, meta_data = isr.isr_spectrum("kappa", sys_set, **params)
self.data.append(s)
meta_data["version"] = "both"
self.meta_data.append(meta_data)
def create_from_code(self):
F0 = 430e6
K_RADAR = -2 * F0 * 2 * np.pi / const.c # Radar wavenumber
self.legend_txt = [
r"$\lambda_{\mathrm{D}} = \lambda_{\mathrm{D},\kappa}$",
r"$\lambda_{\mathrm{D}} = \lambda_{\mathrm{D,M}}$",
]
sys_set = {
"K_RADAR": K_RADAR,
"B": 35000e-9,
"MI": 29,
"NE": 2e10,
"NU_E": 0,
"NU_I": 0,
"T_E": 200,
"T_I": 200,
"T_ES": 90000,
"THETA": 45 * np.pi / 180,
"Z": 599,
"mat_file": "fe_zmuE-07.mat",
}
params = {"kappa": 3, "vdf": "real_data", "area": False}
self.f, s, meta_data = isr.isr_spectrum("kappa", sys_set, **params)
self.data.append(s)
self.meta_data.append(meta_data)
params["debye"] = "maxwell"
self.f, s, meta_data = isr.isr_spectrum("kappa", sys_set, **params)
self.data.append(s)
self.meta_data.append(meta_data)
def create_from_code(self):
F0 = 430e6
K_RADAR = -2 * F0 * 2 * np.pi / const.c # Radar wavenumber
sys_set = {
"K_RADAR": K_RADAR,
"B": 35000e-9,
"MI": 16,
"NE": 1e11,
"NU_E": 100,
"NU_I": 100,
"T_E": 2000,
"T_I": 1500,
"T_ES": 90000,
"THETA": 30 * np.pi / 180,
"Z": 599,
"mat_file": "fe_zmuE-07.mat",
"pitch_angle": list(range(10)),
}
params = {"kappa": 8, "vdf": "real_data", "area": False}
# Ridge 1
ridge = []
# Line 1
self.f, s, meta_data = isr.isr_spectrum("a_vdf", sys_set, **params)
ridge.append(s)
self.meta_data.append(meta_data)
# Line 2
sys_set["NE"] = 1e12
self.f, s, meta_data = isr.isr_spectrum("a_vdf", sys_set, **params)
ridge.append(s)
self.data.append(ridge)
self.meta_data.append(meta_data)
# Ridge 2
ridge = []
# Line 1
sys_set["THETA"] = 60 * np.pi / 180
sys_set["NE"] = 1e11
self.f, s, meta_data = isr.isr_spectrum("a_vdf", sys_set, **params)
ridge.append(s)
self.meta_data.append(meta_data)
# Line 2
sys_set["NE"] = 1e12
self.f, s, meta_data = isr.isr_spectrum("a_vdf", sys_set, **params)
ridge.append(s)
self.data.append(ridge)
self.meta_data.append(meta_data)
self.legend_txt = ["1e11", "1e12"]
self.ridge_txt = ["30", "60"]
def test_isr_long_calc_gauss(self):
self.a, self.b, meta_data = isr.isr_spectrum("a_vdf", self.sys_set,
**self.params)
self.assertEqual(meta_data["kappa"], None)
self.assertEqual(meta_data["vdf"], "gauss_shell")
self.assertEqual(meta_data["T_ES"], 90000)
self.assertEqual(meta_data["Z"], None)
self.assertEqual(meta_data["mat_file"], None)
def test_isr_kappa(self):
self.a, self.b, meta_data = isr.isr_spectrum("kappa", self.sys_set,
**self.params)
self.assertEqual(meta_data["kappa"], 3)
self.assertEqual(meta_data["vdf"], None)
self.assertEqual(meta_data["T_ES"], None)
self.assertEqual(meta_data["Z"], None)
self.assertEqual(meta_data["mat_file"], None)
def test_isr_long_calc_real(self):
self.params["vdf"] = "real_data"
self.a, self.b, meta_data = isr.isr_spectrum("a_vdf", self.sys_set,
**self.params)
self.assertEqual(meta_data["kappa"], None)
self.assertEqual(meta_data["vdf"], "real_data")
self.assertEqual(meta_data["T_ES"], None)
self.assertEqual(meta_data["Z"], 599)
self.assertEqual(meta_data["mat_file"], "fe_zmuE-07.mat")
def create_from_code(self):
F0 = 933e6
K_RADAR = -2 * F0 * 2 * np.pi / const.c
T = [2000, 3000, 4000, 5000, 6000, 7000, 8000, 9000, 10000]
self.ridge_txt = [r"$T_{\mathrm{e}} = %d \mathrm{K}$" % j for j in T]
self.legend_txt = ["Maxwellian", r"$\kappa = 20$", r"$\kappa = 3$"]
sys_set = {
"K_RADAR": K_RADAR,
"B": 50000e-9,
"MI": 16,
"NE": 2e11,
"NU_E": 0,
"NU_I": 0,
"T_E": 2000,
"T_I": 2000,
"T_ES": 90000,
"THETA": 0 * np.pi / 180,
"Z": 599,
"mat_file": "fe_zmuE-07.mat",
}
params = {"kappa": 8, "vdf": "real_data", "area": False}
kappa = [20, 3]
for t in T:
ridge = []
sys_set["T_E"] = t
self.f, s, meta_data = isr.isr_spectrum("maxwell", sys_set, **params)
ridge.append(s)
for k in kappa:
params["kappa"] = k
self.f, s, meta_data = isr.isr_spectrum("kappa", sys_set, **params)
ridge.append(s)
self.data.append(ridge)
self.meta_data.append(meta_data)
for r in self.data:
peak = int(np.argwhere(r[0] == np.max(r[0])))
self.f_list[0].append(self.f[peak])
peak = int(np.argwhere(r[1] == np.max(r[1])))
self.f_list[1].append(self.f[peak])
peak = int(np.argwhere(r[2] == np.max(r[2])))
self.f_list[2].append(self.f[peak])
def create_from_code(self):
F0 = 430e6
K_RADAR = -2 * F0 * 2 * np.pi / const.c # Radar wavenumber
self.legend_txt = [
"Maxwellian",
r"$\kappa = 20$",
r"$\kappa = 8$",
r"$\kappa = 3$",
]
kappa = [20, 8, 3]
sys_set = {
"K_RADAR": K_RADAR,
"B": 35000e-9,
"MI": 29,
"NE": 2e10,
"NU_E": 0,
"NU_I": 0,
"T_E": 200,
"T_I": 200,
"T_ES": 90000,
"THETA": 45 * np.pi / 180,
"Z": 599,
"mat_file": "fe_zmuE-07.mat",
}
params = {"kappa": 20, "vdf": "real_data", "area": False}
t0 = time.perf_counter()
self.f, s, meta_data = isr.isr_spectrum("maxwell", sys_set, **params)
t1 = time.perf_counter()
print(f"Took {t1-t0:.2f} seconds.")
self.data.append(s)
for k in kappa:
params["kappa"] = k
t0 = time.perf_counter()
self.f, s, meta_data = isr.isr_spectrum("kappa", sys_set, **params)
t1 = time.perf_counter()
print(f"Took {t1-t0:.2f} seconds.")
self.data.append(s)
meta_data["version"] = "both"
self.meta_data.append(meta_data)
def create_from_code(self):
F0 = 430e6
K_RADAR = -2 * F0 * 2 * np.pi / const.c # Radar wavenumber
sys_set = {
"K_RADAR": K_RADAR,
"B": 35000e-9,
"MI": 16,
"NE": 1e12,
"NU_E": 100,
"NU_I": 100,
"T_E": 2000,
"T_I": 1500,
"T_ES": 90000,
"THETA": 30 * np.pi / 180,
"Z": 300,
"mat_file": "fe_zmuE-07.mat",
"pitch_angle": "all",
}
params = {"kappa": 3, "vdf": "maxwell", "area": False}
ridge = []
self.f, s1, meta_data = isr.isr_spectrum("maxwell", sys_set, **params)
ridge.append(s1)
self.meta_data.append(meta_data)
_, s2, _ = isr.isr_spectrum("a_vdf", sys_set, **params)
ridge.append(s2)
self.data.append(ridge)
ridge = []
params["vdf"] = "kappa"
self.f, s1, meta_data = isr.isr_spectrum("kappa", sys_set, **params)
ridge.append(s1)
self.meta_data.append(meta_data)
_, s2, _ = isr.isr_spectrum("a_vdf", sys_set, **params)
ridge.append(s2)
self.data.append(ridge)
def create_data(self):
if self.vol == 1:
sys_set = {
"K_RADAR": self.K_RADAR,
"B": 35000e-9,
"MI": 16,
"NE": 2e10,
"NU_E": 100,
"NU_I": 100,
"T_E": 2000,
"T_I": 1500,
"T_ES": 90000,
"THETA": 60 * np.pi / 180,
"Z": 599,
"mat_file": "fe_zmuE-07.mat",
"pitch_angle": list(range(10)),
}
else:
sys_set = {
"K_RADAR": self.K_RADAR,
"B": 35000e-9,
"MI": 16,
"NE": 2e10,
"NU_E": 100,
"NU_I": 100,
"T_E": 2000,
"T_I": 1500,
"T_ES": 90000,
"THETA": 60 * np.pi / 180,
"Z": 300,
"mat_file": "fe_zmuE-07.mat",
"pitch_angle": "all",
}
params = {"kappa": 8, "vdf": "real_data", "area": False}
with tqdm(total=len(self.Z) * len(self.A)) as pbar:
for i, z in enumerate(self.Z):
sys_set["NE"] = z
plasma_freq = (sys_set["NE"] * const.elementary_charge**2 /
(const.m_e * const.epsilon_0))**0.5 / (2 *
np.pi)
cf.I_P["F_MIN"] = plasma_freq
cf.I_P["F_MAX"] = plasma_freq + 4e5
cf.f = np.linspace(cf.I_P["F_MIN"], cf.I_P["F_MAX"],
int(cf.F_N_POINTS))
cf.w = 2 * np.pi * cf.f # Angular frequency
for j, a in enumerate(self.A):
sys_set["THETA"] = a * np.pi / 180
old_stdout = sys.stdout
f = open(os.devnull, "w")
sys.stdout = f
f, s, meta_data = isr.isr_spectrum("a_vdf", sys_set,
**params)
sys.stdout = old_stdout
plasma_power, energy_interval, fr = self.check_energy(
f, s, a)
if energy_interval != 0:
self.dots[0].append(energy_interval)
self.dots[1].append(j)
self.dots[2].append(z)
self.fr[i, j] = fr
self.g[i, j] = plasma_power
pbar.update(1)
self.meta.append(meta_data)