1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
#!/usr/bin/env python3
from pywarpx import picmi
# Physical constants
c = picmi.constants.c
q_e = picmi.constants.q_e
# Number of time steps
max_steps = 84
# Number of cells
nx = 16
ny = 16
nz = 16
# Physical domain
xmin = -1.
xmax = 1.
ymin = -1.
ymax = 1.
zmin = 0.
zmax = 2.
# Create grid
grid = picmi.Cartesian3DGrid(number_of_cells = [nx, ny, nz],
lower_bound = [xmin, ymin, zmin],
upper_bound = [xmax, ymax, zmax],
lower_boundary_conditions = ['dirichlet', 'dirichlet', 'dirichlet'],
upper_boundary_conditions = ['dirichlet', 'dirichlet', 'dirichlet'],
lower_boundary_conditions_particles = ['absorbing', 'absorbing', 'absorbing'],
upper_boundary_conditions_particles = ['absorbing', 'absorbing', 'absorbing'])
# Particles
vel_z = 0.5*c
multiparticles_distribution = picmi.ParticleListDistribution(x = [0.05, 0.],
y = [0., 0.04],
z = [0.05, 0.05],
ux = [0., 0.],
uy = [0., 0.],
uz = [vel_z, vel_z],
weight = [1., 1.])
electrons = picmi.Species(particle_type = 'electron',
name = 'electrons',
initial_distribution = multiparticles_distribution)
# Plasma lenses
plasma_lenses = picmi.PlasmaLens(period = 0.5,
starts = [0.1, 0.11, 0.12, 0.13],
lengths = [0.1, 0.11, 0.12, 0.13],
strengths_E = [600000., 800000., 600000., 200000.],
strengths_B = [0.0, 0.0, 0.0, 0.0])
# Electromagnetic solver
solver = picmi.ElectromagneticSolver(grid = grid,
method = 'Yee',
cfl = 0.7)
# Diagnostics
part_diag1 = picmi.ParticleDiagnostic(name = 'diag1',
period = max_steps,
species = [electrons],
data_list = ['ux', 'uy', 'uz'],
write_dir = '.',
warpx_file_prefix = 'Python_plasma_lens_plt')
# Set up simulation
sim = picmi.Simulation(solver = solver,
max_steps = max_steps,
verbose = 1,
particle_shape = 'linear',
warpx_serialize_initial_conditions = 1,
warpx_do_dynamic_scheduling = 0)
# Add plasma electrons
sim.add_species(electrons, layout = None)
# Add the plasma lenses
sim.add_applied_field(plasma_lenses)
# Add diagnostics
sim.add_diagnostic(part_diag1)
# Write input file that can be used to run with the compiled version
#sim.write_input_file(file_name = 'inputs_3d_picmi')
# Advance simulation until last time step
sim.step(max_steps)
|
