blob: 8918c79486be7f7027edd753a4322236bf7bf67d (
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
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
|
#! /usr/bin/env python
import sys
import yt
import numpy as np
import scipy.constants as scc
yt.funcs.mylog.setLevel(0)
filename = sys.argv[1]
# WarpX headers include more data when rigid injection is used,
# which gives an error with the last yt release.
# To avoid this issue, the three last lines of WarpXHeader are removed if
# needed.
def remove_rigid_lines(plotfile, nlines_if_rigid):
header_name = plotfile + '/WarpXHeader'
f = open(header_name, 'r')
file_lines = f.readlines()
nlines = len(file_lines)
f.close()
if nlines == nlines_if_rigid:
f = open(header_name, 'w')
f.writelines(file_lines[:-3])
f.close()
# Remove rigid injection header lines
remove_rigid_lines(filename, 18)
# Read beam parameters
ds = yt.load( filename )
ad = ds.all_data()
# Beam longitudinal position
z = np.mean(ad['beam', 'particle_position_y'].v)
# Beam width
w = np.std(ad['beam', 'particle_position_x'].v)
# initial parameters
z0 = 20.e-6
z0_no_rigid = -5.e-6
w0 = 1.e-6
theta0 = np.arcsin(0.1)
# Theoretical beam width after propagation if rigid OFF
# Inform the user if rigid injection simply off (just to be kind)
wth_no_rigid = np.sqrt( w0**2 + (z-z0_no_rigid)**2*theta0**2 )
error_no_rigid = np.abs((w-wth_no_rigid)/wth_no_rigid)
if ( error_no_rigid < 0.05):
print("error no rigid: " + str(error_no_rigid))
print("Looks like the beam defocuses as if rigid injection were OFF")
# Theoretical beam width after propagation if rigid ON
wth = np.sqrt( w0**2 + (z-z0)**2*theta0**2 )
error = np.abs((w-wth)/wth)
# Print error and assert small error
print("error: " + str(error))
assert( error < 0.05 )
|
