Plotting update and new O scan

Author: elwaagaa

data/acc-models-sps

@@ -685,7 +685,7 @@ def fit_and_plot_transverse_profiles(self,
                             # Convert to m, normalize height
                             x_pos *= 1e-3
                             x_measured_bin_heights_sorted = np.array(sorted(x_prof_avg, reverse=True))
-                            x_measured_height_max_avg = np.mean(x_measured_bin_heights_sorted[:3]) # take average of top 3 values
+                            x_measured_height_max_avg = np.mean(x_measured_bin_heights_sorted[:2]) # take average of top 3 values
                             x_prof_avg_norm = x_prof_avg / x_measured_height_max_avg
 
                             # Fit Gaussian, center the profile and re-adjust heights
@@ -705,7 +705,7 @@ def fit_and_plot_transverse_profiles(self,
                             # Convert to m, normalize height
                             y_pos *= 1e-3
                             y_measured_bin_heights_sorted = np.array(sorted(y_prof_avg, reverse=True))
-                            y_measured_height_max_avg = np.mean(y_measured_bin_heights_sorted[:3]) # take average of top 3 values
+                            y_measured_height_max_avg = np.mean(y_measured_bin_heights_sorted[:2]) # take average of top 3 values
                             y_prof_avg_norm = y_prof_avg / y_measured_height_max_avg
                             # Fit Gaussian, center the profile and re-adjust heights
                             popt_Y_meas, _ = fits.fit_Gaussian(y_pos, y_prof_avg_norm, p0=(1.0, 0.0, 0.02))
@@ -740,7 +740,7 @@ def fit_and_plot_transverse_profiles(self,
                     for j, ind in enumerate(index_to_plot):
                         # Normalize bin heights
                         y_bin_heights_sorted = np.array(sorted(tbt_dict['monitorV_y_intensity'][ind], reverse=True))
-                        y_height_max_avg = np.mean(y_bin_heights_sorted[:3]) # take average of top 3 values
+                        y_height_max_avg = np.mean(y_bin_heights_sorted[:2]) # take average of top 3 values
                         Y_pos_data = tbt_dict['monitorV_y_grid']
                         if ind == 0:
                             Y0_profile_data = tbt_dict['monitorV_y_intensity'][ind] / y_height_max_avg ### if changes fast, take particle histogram instead

fma_ions/plotting/sps_plotting_classes.py

@@ -0,0 +1,90 @@
+"""
+Launcher script to HTCondor for GPU - generate python scripts for tune scan
+"""
+import fma_ions
+import os
+import pathlib
+import numpy as np
+import datetime
+
+# Find path of script being run
+dir_path = pathlib.Path(__file__).parent.absolute()
+
+# Define run files and which parameters to change
+master_name = 'Q26_O_ions_bunch_intensity_scan_adaptive_sc_ibs_with_compensated_50_150_300_600_Hz_ripple_130k_turns_TRANSFER_FUNCTION_high_Nb_same_emittance'
+num_turns = 130_000 # corresponds to about 3 s for SPS ions at flat bottom
+Qy = 26.19
+Qx = 26.31
+
+# Assume same emittances as Pb, but scaling from 28e8 as the SPS space charge limit
+# 28e8/3.94e8 = 7.1, so use this as scaling factor
+
+no_LEIR_inj = [2, 3, 4, 5, 6, 8]
+Nb_array = ['2*1.3e8', '2*1.75e8', '2*2.215e8', '2*2.93e8', '2*3.202e8', '2*3.94e8']
+exn_array = ['0.73e-6', '0.89e-6', '1.11e-6', '1.4e-6', '1.75e-6', '2.1e-6']
+eyn_array = ['0.48e-6', '0.63e-6', '0.74e-6', '0.85e-6', '1.05e-6', '1.175e-6']
+run_files = ['sps_run_leir_inj_{}.py'.format(i+1) for i in range(len(no_LEIR_inj))]
+
+# Define script and folder names
+script_names = run_files.copy()
+folder_names = ['sps_no_leir_inj_{}'.format(no_LEIR_inj[i]) for i in range(len(no_LEIR_inj))]
+string_array = ['No. LEIR inj. = {}'.format(no_LEIR_inj[i]) for i in range(len(no_LEIR_inj))]    
+
+# Generate the scripts to be submitted
+for i, run_file in enumerate(run_files):
+    
+    # Write run file for given tune
+    print('Generating launch script {}\n'.format(run_file))
+    run_file = open(run_file, 'w')
+    run_file.truncate(0)  # remove existing content, if any
+    run_file.write(
+    '''import fma_ions
+import numpy as np
+output_dir = './'
+
+n_turns = {}
+num_part = 20_000
+
+beamParams=fma_ions.BeamParameters_SPS()
+beamParams.Nb = {}*7.24399203
+beamParams.exn = {}
+beamParams.eyn = {}
+
+# Transfer function factors
+a_50 = 1.0 #1.7170
+a_150 = 0.5098
+a_300 = 0.2360
+a_600 = 0.1095
+
+# Desired ripple frequencies and amplitudes - from DCCT measurements 2024-10-30
+ripple_freqs = np.array([50.0, 150.0, 300.0, 600.0])
+kqf_amplitudes = np.array([1.79381965522221e-07*a_50, 1.7917856960711038e-07*a_150, 1.717715125357188e-07*a_300, 1.0613897587376263e-07*a_600])
+kqd_amplitudes = np.array([3.1433458408493135e-07*a_50, 4.125645646596158e-07*a_150, 2.6325770762187453e-07*a_300, 8.302889259074001e-08*a_600])
+kqf_phases = np.array([2.5699456856082965, -1.2707524434033985, 1.1509405507521766, -2.3897351868985552])
+kqd_phases = np.array([-1.6168418898711074, -1.5349070763197448, -2.145386063404577, 0.7431459693919794])
+
+# Tracking on GPU context
+sps = fma_ions.SPS_Flat_Bottom_Tracker(qx0={:.3f}, qy0={:.3f}, num_turns=n_turns, num_part=num_part)
+tbt = sps.track_SPS(ion_type='O', which_context='gpu', distribution_type='qgaussian', beamParams=beamParams, install_SC_on_line=True, add_beta_beat=True, 
+                    add_non_linear_magnet_errors=True, apply_kinetic_IBS_kicks=True, ibs_step = 2000, add_tune_ripple=True, ripple_freqs = ripple_freqs,
+                    kqf_amplitudes = kqf_amplitudes, kqd_amplitudes = kqd_amplitudes, kqf_phases=kqf_phases, kqd_phases=kqd_phases,
+                    SC_adaptive_interval_during_tracking=100)
+tbt.to_json(output_dir)
+    '''.format(num_turns, Nb_array[i], exn_array[i], eyn_array[i], Qx, Qy)
+    )
+    run_file.close()
+    
+    
+# Instantiate the submitter class and launch the jobs
+sub = fma_ions.Submitter() 
+master_job_name = '{:%Y_%m_%d__%H_%M_%S}_{}'.format(datetime.datetime.now(), master_name)
+
+# Launch the Python scripts in this folder
+for i, script in enumerate(script_names):
+    file_name = os.path.join(dir_path, script)
+    print(f"Submitting {file_name}")
+    sub.submit_GPU(file_name, master_job_name=master_job_name, job_name=folder_names[i])
+sub.copy_master_plot_script(folder_names, string_array)
+sub.copy_plot_script_emittances_for_scan(master_name, folder_names, scan_array_for_x_axis=Nb_array,
+                                             label_for_x_axis='Injected Pb ions per bunch', 
+                                             extra_text_string='$Q_{x, y}$ = 26.31, 26.19 - q-Gaussian beam\\nAdaptive SC, ~10% $\\beta$-beat + non-linear magnet errors')