blob: 672da52a4d6879d871a8d562dc1d21ef8f7f17ee (
plain) (
blame)
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
|
#!/usr/bin/env python
import matplotlib.pyplot as plt
import warpx
warpx.initialize()
# run for ten time steps
warpx.evolve()
x = warpx.get_particle_x(0)
y = warpx.get_particle_y(0)
plt.plot(x[0], y[0], '.')
plt.savefig("particles.png")
# this returns a list of numpy arrays that hold the magnetic field
# data in the x-direction on each grid for level 0
grid_data = warpx.get_mesh_magnetic_field(0, 0, False)
# plot a slice through the second grid
plt.clf()
plt.pcolormesh(grid_data[1][9,:,:])
plt.savefig("field.png")
warpx.finalize()
|
