Update density function (#263)

Author: RemiLehe

simulation_data/staging_injector/templates/analyze_simulation.py

@@ -48,7 +48,9 @@ def analyze_simulation():
         n_downstream_atom = float( re.findall(r'my_constants\.n_downstream_atom = (.+)', text)[0] )
         hydrogen_density_function = re.findall(r'hydrogen1\.density_function\(x,y,z\) = (.+)', text)[0]
         nitrogen_density_function = re.findall(r'nitrogen1\.density_function\(x,y,z\) = (.+)', text)[0]
+        plateau_length = eval( re.findall(r'my_constants\.plateau_length = (.+)', text)[0] )
 
+    
     # Compute red/blue shift: wavelength such that 13.5%/86.5% of the spectrum energy is below
     S, info = ts.get_laser_spectral_intensity(
         iteration=ts.iterations[-1], pol=pol)

simulation_data/staging_injector/templates/inputs

@@ -67,6 +67,7 @@ my_constants.n_upstream_atom = {{upstream_density}}e24 # number of electrons per
 my_constants.n_downstream_atom = {{downstream_density}}e24 # number of electrons per m^3
 my_constants.ramp_length = 0.4e-2
 my_constants.stage_length = 3.3e-2
+my_constants.plateau_length = stage_length - 2*ramp_length
 my_constants.dopant_length = 0.4e-2
 my_constants.dopant_fraction = {{dopant_concentration}}e-2
 
@@ -84,7 +85,7 @@ electrons1.xmin = 0
 electrons1.xmax = 0.9*x_max
 electrons1.zmin = 0
 electrons1.zmax = stage_length
-electrons1.density_function(x,y,z) = "(1 + 4*dopant_fraction*(z<dopant_length))*( (z<ramp_length)*sqrt(z/ramp_length)*n_upstream_atom + (z>=ramp_length)*(z<stage_length-ramp_length)*(n_upstream_atom + n_downstream_atom*(z-ramp_length)/stage_length) + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_length)*n_downstream_atom)"
+electrons1.density_function(x,y,z) = "(1 + 4*dopant_fraction*(z<dopant_length))*( (z<ramp_length)*sqrt(z/ramp_length)*n_upstream_atom + (z>=ramp_length)*(z<stage_length-ramp_length)*(n_upstream_atom + (n_downstream_atom-n_upstream_atom)*(z-ramp_length)/plateau_length) + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_length)*n_downstream_atom)"
 electrons1.profile = parse_density_function
 electrons1.momentum_distribution_type = constant
 electrons1.do_continuous_injection=1
@@ -97,7 +98,7 @@ hydrogen1.xmin = 0
 hydrogen1.xmax = 0.9*x_max
 hydrogen1.zmin = 0
 hydrogen1.zmax = stage_length
-hydrogen1.density_function(x,y,z) = "(1 - dopant_fraction*(z<dopant_length))*( (z<ramp_length)*sqrt(z/ramp_length)*n_upstream_atom + (z>=ramp_length)*(z<stage_length-ramp_length)*(n_upstream_atom + n_downstream_atom*(z-ramp_length)/stage_length) + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_length)*n_downstream_atom)"
+hydrogen1.density_function(x,y,z) = "(1 - dopant_fraction*(z<dopant_length))*( (z<ramp_length)*sqrt(z/ramp_length)*n_upstream_atom + (z>=ramp_length)*(z<stage_length-ramp_length)*(n_upstream_atom + (n_downstream_atom-n_upstream_atom)*(z-ramp_length)/plateau_length) + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_length)*n_downstream_atom)"
 hydrogen1.profile = parse_density_function
 hydrogen1.momentum_distribution_type = constant
 hydrogen1.do_continuous_injection=1
@@ -110,7 +111,7 @@ nitrogen1.xmin = 0
 nitrogen1.xmax = 0.9*x_max
 nitrogen1.zmin = 0
 nitrogen1.zmax = dopant_length
-nitrogen1.density_function(x,y,z) = "(z<dopant_length)*dopant_fraction*( (z<ramp_length)*sqrt(z/ramp_length)*n_upstream_atom + (z>=ramp_length)*(z<stage_length-ramp_length)*(n_upstream_atom + n_downstream_atom*(z-ramp_length)/stage_length) + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_length)*n_downstream_atom)"
+nitrogen1.density_function(x,y,z) = "(z<dopant_length)*dopant_fraction*( (z<ramp_length)*sqrt(z/ramp_length)*n_upstream_atom + (z>=ramp_length)*(z<stage_length-ramp_length)*(n_upstream_atom + (n_downstream_atom-n_upstream_atom)*(z-ramp_length)/plateau_length) + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_length)*n_downstream_atom)"
 nitrogen1.profile = parse_density_function
 nitrogen1.momentum_distribution_type = constant
 nitrogen1.do_field_ionization = 1