def test_many_particles_periodic_deposition(N, _velocity):
NG = 10
L = NG
g = Grid(1, L=L, NG=NG, periodic=True)
s = Species(1.0 / N, 1, N, g)
s.distribute_uniformly(L)
s.v[:, 0] = _velocity
s.v[:, 1] = 1
s.v[:, 2] = -1
dt = g.dx / s.c
g.gather_current([s])
longitudinal_collected_weights = g.current_density_x[1:-2].sum(
axis=0) / s.v[0, 0]
transversal_collected_weights = g.current_density_yz[2:-2].sum(
axis=0) / s.v[0, 1:]
label = {0: 'x', 1: 'y', 2: 'z'}
def plot():
fig, ax = plt.subplots()
i = 0
fig.suptitle(
f"Longitudinal weights: {longitudinal_collected_weights}, transversal: {transversal_collected_weights}"
)
ax.plot(g.x,
g.current_density_x[1:-2],
alpha=(3 - i) / 3,
lw=1 + i,
label=f"{label[i]}")
ax.scatter(s.x, np.zeros_like(s.x))
ax.legend(loc='lower right')
ax.set_xticks(g.x)
ax.grid()
for i in range(1, 3):
ax.plot(g.x,
g.current_density_yz[2:-2, i - 1],
alpha=(3 - i) / 3,
lw=1 + i,
label=f"{label[i]}")
ax.scatter(s.x, np.zeros_like(s.x))
ax.legend(loc='lower right')
ax.set_xticks(g.x)
ax.grid()
filename = f"data_analysis/test/periodic_multiple_{N}_{_velocity:.2f}.png"
make_sure_path_exists(filename)
fig.savefig(filename)
fig.clf()
plt.close(fig)
assert np.allclose(longitudinal_collected_weights,
1), ("Longitudinal weights don't match!", plot())
assert np.allclose(transversal_collected_weights,
1), ("Transversal weights don't match!", plot())
def test_single_particle_deposition_simulation(periodic):
g = Grid(T=100, L=1, NG=100, c=1, periodic=periodic)
s = Particle(g, g.L / 2, g.dx, g.dx, -g.dx)
sim = Simulation(
g, [s],
category_type="test",
filename=
f"test_single_particle_depo_sim_{'' if periodic else 'a'}periodic")
try:
sim.run().postprocess()
except Exception as E:
assert False, (E, plots(sim, show_animation=True, show_static=True))
def test_single_particle_above_lightspeed():
g = Grid(1, L=7, NG=7)
s = Particle(g, 1 * g.dx, g.c * 4)
dt = g.dx / s.c
g.current_density_x[...] = 0
g.current_density_yz[...] = 0
with pytest.raises(Exception):
longitudinal_current_deposition(g.current_density_x, s.v[:, 0], s.x,
g.dx, dt, s.q)
with pytest.raises(Exception):
transversal_current_deposition(g.current_density_yz, s.v, s.x, g.dx,
dt, s.q)
def test_single_particle_transversal_deposition(_position, _velocity):
g = Grid(1, L=7, NG=7)
s = Particle(g, _position * g.dx, _velocity, 1, -1)
dt = g.dx / s.c
new_positions = s.x + s.v[:, 0] * dt
g.current_density_x[...] = 0
g.current_density_yz[...] = 0
print("\n\n=====Start run===")
# current_deposition(g, s, dt)
transversal_current_deposition(g.current_density_yz, s.v, s.x, g.dx, dt,
s.q)
total_currents = g.current_density_yz.sum(axis=0) / s.v[0, 1:]
total_sum_currents = g.current_density_yz.sum()
x_velocity = s.v[0, 0]
collected_transversal_weights = total_currents
print("total", total_currents)
print("x velocity", x_velocity)
print("weights", collected_transversal_weights)
def plot_transversal():
fig, ax = plt.subplots()
ax.scatter(s.x, 0)
title = f"x0: {s.x[0]} v: {s.v[0,0]} x1: {new_positions[0]:.3f} w: {collected_transversal_weights}"
ax.annotate(title,
xy=(new_positions[0], 0),
xytext=(s.x[0], 0.3),
arrowprops=dict(facecolor='black',
shrink=0.05,
linewidth=0.5),
horizontalalignment='right')
ax.set_xticks(g.x)
ax.set_xticks(np.arange(0, g.L, g.dx / 2), minor=True)
ax.grid(which='minor', alpha=0.3)
ax.grid(which='major', alpha=0.7)
ax.set_title(title)
ax.scatter(new_positions, 0)
for i, label in {1: 'y', 2: 'z'}.items():
ax.plot(g.x + 0.5,
g.current_density_yz[1:-1, i - 1],
"o-",
alpha=0.7,
linewidth=i + 3,
label=f"j{label}")
ax.legend()
plt.show()
return title + " instead of 1"
assert np.allclose(collected_transversal_weights, 1), plot_transversal()
assert np.isclose(total_sum_currents,
0), (plot_transversal(),
f"Currents do not zero out at {total_sum_currents}")
def test_single_particle_longitudinal_deposition(_position, _velocity):
g = Grid(T=1, L=7, NG=7)
s = Particle(g, _position * g.dx, _velocity)
dt = g.dt
g.current_density_x[...] = 0
g.current_density_yz[...] = 0
# current_deposition(g, s, dt)
longitudinal_current_deposition(g.current_density_x, s.v[:, 0], s.x, g.dx,
dt, s.q)
collected_longitudinal_weights = g.current_density_x.sum() / s.v[0, 0]
def plot_longitudinal():
fig, ax = plt.subplots()
ax.scatter(s.x, 0)
new_positions = s.x + s.v[:, 0] * dt
title = f"x0: {s.x[0]} v: {s.v[0,0]} x1: {new_positions[0]} w: {collected_longitudinal_weights}"
ax.annotate(title,
xy=(new_positions[0], 0),
xytext=(s.x[0], 0.3),
arrowprops=dict(facecolor='black',
shrink=0.05,
linewidth=0.5),
horizontalalignment='right')
ax.set_xticks(g.x)
ax.set_xticks(np.arange(0, g.L, g.dx / 2), minor=True)
ax.grid(which='minor', alpha=0.3)
ax.grid(which='major', alpha=0.7)
ax.set_title(title)
ax.scatter(new_positions, 0)
ax.plot(g.x,
g.current_density_x[1:-2],
"go-",
alpha=0.7,
linewidth=3,
label=f"jx")
ax.legend()
plt.show()
return title + " instead of 1"
assert np.isclose(collected_longitudinal_weights, 1), plot_longitudinal()
def test_two_particles_deposition(_position, _velocity, _truefalse,
_truefalse2):
NG = 7
L = NG
g = Grid(1, L=L, NG=NG)
c = 1
dt = g.dx / c
positions = [_position * g.dx, (L - _position * g.dx) % L]
# print(positions)
for position in positions:
s = Particle(g, position, _velocity, 1, -1)
# print(f"\n======PARTICLE AT {position}=======")
# print(s)
# print(s.x)
# print(s.v)
if _truefalse:
longitudinal_current_deposition(g.current_density_x, s.v[:, 0],
s.x, g.dx, dt, s.q)
if _truefalse2:
transversal_current_deposition(g.current_density_yz, s.v, s.x,
g.dx, dt, s.q)
# print(g.current_density)
collected_weights_x = g.current_density_x.sum(axis=0) / _velocity
collected_weights_yz = g.current_density_yz.sum(axis=0) / np.array(
[1, -1], dtype=float)
g2 = Grid(1, L=L, NG=NG)
s = Species(1, 1, 2, g2)
s.x[:] = positions
s.v[:, 0] = _velocity
s.v[:, 1] = 1
s.v[:, 2] = -1
# print("\n\n======TWO PARTICLES=======")
# print(s)
# print(s.x)
# print(s.v)
if _truefalse:
longitudinal_current_deposition(g2.current_density_x, s.v[:, 0], s.x,
g2.dx, dt, s.q)
if _truefalse2:
transversal_current_deposition(g2.current_density_yz, s.v, s.x, g2.dx,
dt, s.q)
# print(g2.current_density)
collected_weights_x2 = g2.current_density_x.sum(axis=0) / s.v[0, 0]
collected_weights_yz2 = g2.current_density_yz.sum(axis=0) / np.array(
[1, -1], dtype=float)
label = {0: 'x', 1: 'y', 2: 'z'}
def plot():
fig, (ax1, ax2) = plt.subplots(2)
plt.suptitle(f"x: {positions}, vx: {_velocity}")
i = 0
ax1.plot(g.x,
g.current_density_x[1:-2],
alpha=(3 - i) / 3,
lw=1 + i,
label=f"1 {label[i]}")
ax1.plot(g.x,
g2.current_density_x[1:-2],
alpha=(3 - i) / 3,
lw=1 + i,
label=f"2 {label[i]}")
ax2.plot(g.x, (g.current_density_x - g2.current_density_x)[1:-2],
alpha=(3 - i) / 3,
lw=1 + i,
label=f"1 - 2 {label[i]}")
for i in range(1, 3):
ax1.plot(g.x,
g.current_density_yz[2:-2, i - 1],
alpha=(3 - i) / 3,
lw=1 + i,
label=f"1 {label[i]}")
ax1.plot(g.x,
g2.current_density_yz[2:-2, i - 1],
alpha=(3 - i) / 3,
lw=1 + i,
label=f"2 {label[i]}")
ax2.plot(g.x,
(g.current_density_yz - g2.current_density_yz)[2:-2,
i - 1],
alpha=(3 - i) / 3,
lw=1 + i,
label=f"1 - 2 {label[i]}")
for ax in [ax1, ax2]:
ax.scatter(s.x, np.zeros_like(s.x))
ax.legend(loc='lower right')
ax.set_xticks(g.x)
ax.grid()
fig.savefig(
f"data_analysis/deposition/{_position:.2f}_{_velocity:.2f}.png")
assert np.allclose(
g.current_density_x,
g2.current_density_x), ("Longitudinal currents don't match!", plot())
assert np.allclose(
g.current_density_yz,
g2.current_density_yz), ("Transversal currents don't match!", plot())
assert np.allclose(
collected_weights_x,
collected_weights_x2), ("Longitudinal weights don't match!", plot())
assert np.allclose(
collected_weights_yz,
collected_weights_yz2), ("Transversal weights don't match!", plot())
def __init__(self,
filename,
n_macroparticles,
impulse_duration,
laser_intensity,
perturbation_amplitude,
additional_scaling=1):
"""
A simulation of laser-hydrogen shield interaction.
Parameters
----------
filename : str
Filename for the simulation.
n_macroparticles : int
Number of macroparticles for each species. The simulation is
normalized to 75000 macroparticles by default,
impulse_duration : float
Duration of the laser impulse.
laser_intensity : float
Laser impulse intensity, in W/m^2. A good default is 1e21.
perturbation_amplitude : float
Amplitude of the initial position perturbation.
"""
if laser_intensity:
bc_laser = BoundaryCondition.Laser(
laser_intensity=laser_intensity,
laser_wavelength=laser_wavelength,
envelope_center_t=total_time / 2,
envelope_width=impulse_duration,
envelope_power=6,
c=lightspeed,
epsilon_0=epsilon_zero,
)
print(f"Laser amplitude: {bc_laser.laser_amplitude:e}")
bc = bc_laser.laser_pulse
else:
bc = lambda x: None
grid = Grid(T=total_time,
L=length,
NG=number_cells,
c=lightspeed,
epsilon_0=epsilon_zero,
bc=bc,
periodic=False)
if n_macroparticles:
scaling = default_scaling * N_MACROPARTICLES / n_macroparticles * additional_scaling
electrons = Species(-electric_charge, electron_rest_mass,
n_macroparticles, grid, "electrons", scaling)
protons = Species(electric_charge, proton_mass, n_macroparticles,
grid, "protons", scaling)
list_species = [electrons, protons]
else:
list_species = []
self.perturbation_amplitude = perturbation_amplitude # in units of dx
description = "Hydrogen shield-laser interaction"
super().__init__(grid,
list_species,
filename=filename,
category_type="laser-shield",
config_version=VERSION,
title=description,
considered_large=True)
print("Simulation prepared.")