Add upstream pressure to the control variables (#257)

Author: RemiLehe

dashboard/config/variables.yml

@@ -2,6 +2,12 @@ staging_injector:
 
   input_variables:
 
+    input0:
+      name: "Cap Up (torr)"
+      type: scalar
+      default: 16.5
+      value_range: [10, 25]
+
     input1:
       name: "Cap downstream (torr)"
       type: scalar
@@ -50,20 +56,27 @@ staging_injector:
       beta: 0.0
 
     input4:
-      name: "Background density [1e18/cm^3]"
+      name: "Upstream density [1e18/cm^3]"
       unit: "cm^-3"
-      depends_on: "Cap downstream (torr)"
+      depends_on: "Cap Up (torr)"
       alpha: 0.0644
       beta: 0.0
 
     input5:
+      name: "Downstream density [1e18/cm^3]"
+      unit: "cm^-3"
+      depends_on: "Cap downstream (torr)"
+      alpha: 0.0644
+      beta: 0.0
+
+    input6:
       name: "Trapped charge [pC]"
       unit: "pC"
       depends_on: "EBeamPrf charge [pC]"
       alpha: 1.0
       beta: 0.0
 
-    input6:
+    input7:
       name: "Beam RMS div x [mrad]"
       unit: "mrad"
       depends_on: "EBeamPrf fwhm div x [mrad]"

simulation_data/staging_injector/submission_script_multi

@@ -1,6 +1,6 @@
 #!/bin/bash -l
 
-#SBATCH -t 02:15:00
+#SBATCH -t 04:00:00
 #SBATCH -N 60
 #SBATCH -A m558_g
 #SBATCH -q regular

simulation_data/staging_injector/templates/analyze_simulation.py

@@ -44,7 +44,8 @@ def analyze_simulation():
         ramp_length = float( re.findall(r'my_constants\.ramp_length = (.+)', text)[0] )
         dopant_length = float( re.findall(r'my_constants\.dopant_length = (.+)', text)[0] )
         dopant_fraction = float( re.findall(r'my_constants\.dopant_fraction = (.+)', text)[0] )
-        n_atom = float( re.findall(r'my_constants\.n_atom = (.+)', text)[0] )
+        n_upstream_atom = float( re.findall(r'my_constants\.n_upstream_atom = (.+)', text)[0] )
+        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]
 

simulation_data/staging_injector/templates/inputs

@@ -63,7 +63,8 @@ my_constants.ppc_z1  = 3
 ##### plasma parameters #########
 #################################
 
-my_constants.n_atom = {{background_density}}e24 # number of electrons per m^3
+my_constants.n_upstream_atom = {{upstream_density}}e24 # number of electrons per m^3
+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.dopant_length = 0.4e-2
@@ -83,7 +84,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))*n_atom*( (z<ramp_length)*sqrt(z/ramp_length) + (z>=ramp_length)*(z<stage_length-ramp_length)*1 + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_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.profile = parse_density_function
 electrons1.momentum_distribution_type = constant
 electrons1.do_continuous_injection=1
@@ -96,7 +97,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))*n_atom*( (z<ramp_length)*sqrt(z/ramp_length) + (z>=ramp_length)*(z<stage_length-ramp_length)*1 + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_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.profile = parse_density_function
 hydrogen1.momentum_distribution_type = constant
 hydrogen1.do_continuous_injection=1
@@ -109,7 +110,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*n_atom*( (z<ramp_length)*sqrt(z/ramp_length) + (z>=ramp_length)*(z<stage_length-ramp_length)*1 + (z>=stage_length-ramp_length)*(z<stage_length)*sqrt((stage_length-z)/ramp_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.profile = parse_density_function
 nitrogen1.momentum_distribution_type = constant
 nitrogen1.do_field_ionization = 1

simulation_data/staging_injector/templates/prepare_simulation.py

@@ -18,7 +18,8 @@ def record_simulation_parameters():
         "Laser energy [J]": {{laser_energy}},
         "Target-to-focus distance [cm]": {{target_to_focus_distance}},
         "Dopant concentration [%]": {{dopant_concentration}},
-        "Background density [1e18/cm^3]": {{background_density}},
+        "Upstream density [1e18/cm^3]": {{upstream_density}},
+        "Downstream density [1e18/cm^3]": {{downstream_density}},
     }
     # Dump input parameters, to be read in analysis file
     with open('input_params.json', 'w') as file:

simulation_data/staging_injector/templates/run_grid_scan.py

@@ -13,10 +13,11 @@ def analysis_func_main(work_dir, output_params):
 var_1 = VaryingParameter("laser_energy", 12.5, 13.5)
 var_2 = VaryingParameter("target_to_focus_distance", 0, 1)
 var_3 = VaryingParameter("dopant_concentration", 0, 3)
-var_4 = VaryingParameter("background_density", 0.33, 1.1)
+var_4 = VaryingParameter("upstream_density", 0.9, 1.54)
+var_5 = VaryingParameter("downstream_density", 0.33, 1.1)
 obj = Objective("f", minimize=False)
 
-n_steps = [3, 5, 4, 6]
+n_steps = [3, 5, 4, 6, 6]
 sim_workers = 240
 
 # Compute total number of steps
@@ -26,7 +27,7 @@ def analysis_func_main(work_dir, output_params):
 
 # Create generator
 gen = GridSamplingGenerator(
-    varying_parameters=[var_1,var_2,var_3,var_4],
+    varying_parameters=[var_1,var_2,var_3,var_4,var_5],
     objectives=[obj],
     n_steps=n_steps,
 )