Merge pull request #17 from PyCOMPLETE/develop

v1.2.4

Author: giadarol

Simulation.py

@@ -130,6 +130,7 @@ def init_all(self):
                             enable_kick_y = pp.enable_kick_y)
 
 
+                        
         if self.ring_of_CPUs.I_am_the_master and pp.enable_arc_dip:
             with open('multigrid_config_dip.txt', 'w') as fid:
                 if hasattr(ecloud_dip.spacech_ele.PyPICobj, 'grids'):
@@ -155,6 +156,7 @@ def init_all(self):
                     pickle.dump(ecloud_quad.spacech_ele.PyPICobj.grids, fid)
                 else:
                     pickle.dump('Single grid.', fid)
+                
 
         # setup transverse losses (to "protect" the ecloud)
         import PyHEADTAIL.aperture.aperture as aperture
@@ -303,8 +305,12 @@ def init_master(self):
             if pp.N_turns!=pp.N_turns_target:
                 raise ValueError('In footprint mode you need to set N_turns_target=N_turns_per_run!')
 
+        check_for_resubmit = True
+        if hasattr(pp, 'check_for_resubmit'):
+            check_for_resubmit = pp.check_for_resubmit
         import PyPARIS_sim_class.Save_Load_Status as SLS
-        SimSt = SLS.SimulationStatus(N_turns_per_run=pp.N_turns, check_for_resubmit = True, N_turns_target=pp.N_turns_target)
+        SimSt = SLS.SimulationStatus(N_turns_per_run=pp.N_turns, check_for_resubmit=check_for_resubmit,
+                N_turns_target=pp.N_turns_target)
         SimSt.before_simulation()
         self.SimSt = SimSt
 
@@ -426,6 +432,7 @@ def execute_orders_from_master(self, orders_from_master):
             for ec in self.my_list_eclouds: ec.finalize_and_reinitialize()
 
 
+        
     def finalize_simulation(self):
         if pp.footprint_mode:
             # Tunes

examples/000_feedback_damping_time/clean_all

@@ -0,0 +1,5 @@
+rm simulation_status.sta
+rm bunch_evolution_*.h5
+rm slice_evolution_*.h5
+rm bunch_status_part*.h5
+rm *.pyc

examples/001_simulation_with_eclouds/000_run_serial.sh

@@ -0,0 +1,12 @@
+#!/usr/bin/bash
+
+export PYTHONPATH=$PYTHONPATH:../../../
+
+rm simulation_status.sta
+
+# Run Serial
+for i in 1 2 3
+do
+    echo "Run $i"
+    ../../../PyPARIS/serialexec.py sim_class=PyPARIS_sim_class.Simulation.Simulation
+done

examples/001_simulation_with_eclouds/001_run_multiproc.sh

@@ -0,0 +1,13 @@
+#!/usr/bin/bash
+
+export PYTHONPATH=$PYTHONPATH:../../../
+
+rm simulation_status.sta
+
+# Run Parallel without MPI
+for i in 1 2 3
+do
+    echo "Run $i"
+    ../../../PyPARIS/multiprocexec.py -n 3 sim_class=PyPARIS_sim_class.Simulation.Simulation
+done
+

examples/001_simulation_with_eclouds/002_run_MPI.sh

@@ -0,0 +1,12 @@
+#!/usr/bin/bash
+
+export PYTHONPATH=$PYTHONPATH:../../../
+
+rm simulation_status.sta
+
+# Run MPI
+for i in 1 2 3
+do
+    echo "Run $i"
+    mpiexec -n 3 ../../../PyPARIS/withmpi.py sim_class=PyPARIS_sim_class.Simulation.Simulation
+done

examples/001_simulation_with_eclouds/003_run_single_core.py

@@ -0,0 +1,7 @@
+import sys
+sys.path.append('../../../')
+
+import PyPARIS_sim_class.Simulation as sim_mod
+
+ring = sim_mod.get_serial_CPUring()
+ring.run()

examples/001_simulation_with_eclouds/004_some_checks.py

@@ -0,0 +1,105 @@
+import sys, os
+sys.path.append('../../../')
+
+
+import numpy as np
+import matplotlib.pyplot as plt
+from matplotlib.ticker import MaxNLocator
+
+import PyPARIS.myfilemanager as mfm
+
+import mystyle as ms
+
+tag = 'noecloud'
+
+flag_check_damp_time = False 
+tau_damp_x = 50.
+tau_damp_y = 100.
+
+flag_check_Qs = True
+Q_s = np.nan
+
+#ob = mfm.monitorh5_to_obj('bunch_evolution_00.h5')
+ob = mfm.monitorh5list_to_obj(
+        ['bunch_evolution_%02d.h5'%ii for ii in range(3)])
+
+
+plt.close('all')
+
+# Plot transverse positions
+fig1 = plt.figure(1, figsize=(8,6*1.2))
+fig1.set_facecolor('w')
+ms.mystyle_arial(fontsz=16)
+
+axx = fig1.add_subplot(2,1,1)
+axy = fig1.add_subplot(2,1,2, sharex=axx)
+
+axx.plot(ob.mean_x)
+axy.plot(ob.mean_y)
+
+if flag_check_damp_time:
+    turn_num = np.arange(0, len(ob.mean_x), dtype=np.float)
+    ix_max = np.argmax(ob.mean_x)
+    iy_max = np.argmax(ob.mean_y)
+
+    axx.plot(ob.mean_x[ix_max]*np.exp(-(turn_num-ix_max)/tau_damp_x), 
+            linewidth=2, color='red', linestyle='--',
+            label=r'Damping time = %.0f turns'%tau_damp_x)
+    axy.plot(ob.mean_y[iy_max]*np.exp(-(turn_num-iy_max)/tau_damp_y), 
+            linewidth=2, color='red', linestyle='--',
+            label=r'Damping time = %.0f turns'%tau_damp_y)
+
+axx.legend(prop={'size':14}).draggable()
+axy.legend(prop={'size':14}).draggable()
+
+axx.set_ylabel('x [m]')
+axy.set_ylabel('y [m]')
+axy.set_xlabel('Turn')
+
+# Plot transverse spectra
+fig2 = plt.figure(2, figsize=(8,6*1.2))
+fig2.set_facecolor('w')
+
+axfx = fig2.add_subplot(2,1,1)
+axfy = fig2.add_subplot(2,1,2, sharex=axfx)
+
+spectx = np.abs(np.fft.rfft(ob.mean_x))
+specty = np.abs(np.fft.rfft(ob.mean_y))
+freq = np.fft.rfftfreq(len(ob.mean_x))
+
+axfx.plot(freq, spectx)
+axfy.plot(freq, specty)
+
+# Check longitudinal plane
+fig3 = plt.figure(3, figsize=(8,6*1.2))
+fig3.set_facecolor('w')
+axz = fig3.add_subplot(2,1,1, sharex=axx)
+axfz = fig3.add_subplot(2,1,2)
+
+axz.plot(ob.mean_z[:-10])
+spectz = np.abs(np.fft.rfft(ob.mean_z[:-10]))
+spectz[0] = 0. # I am non interested in the average
+freqz = np.fft.rfftfreq(len(ob.mean_x[:-10]))
+axfz.plot(freqz, spectz)
+axfz.axvline(x=Q_s)
+
+for ax in [axx, axy, axfx, axfy, axz, axfz]:
+    ax.ticklabel_format(style='sci', scilimits=(0,0),axis='y')
+
+for ax in [axx, axy, axfx, axfy, axz, axfz]:
+    ax.yaxis.set_major_locator(MaxNLocator(5))
+
+for ax in [axx, axy, axfx, axfy, axz, axfz]:
+    ax.grid(True)
+
+for fig in [fig1, fig2, fig3]:
+    fig.suptitle('Bunch')
+    fig.subplots_adjust(
+            top=0.885,
+            bottom=0.1,
+            left=0.125,
+            right=0.9,
+            hspace=0.345,
+            wspace=0.2)
+
+plt.show()

examples/001_simulation_with_eclouds/Simulation_parameters.py

@@ -0,0 +1,149 @@
+from scipy.constants import c
+
+check_for_resubmit = False
+
+####################
+# Machine Settings #
+####################
+
+machine_configuration = 'LHC-collision'
+
+# # Use this part for optics from file
+# # n_segments needs to be None if optics_pickle_file is specified
+# optics_pickle_file = 'lhc2018_25cm_only_triplets_IR15_b1_optics.pkl'
+# n_segments = None
+# beta_x =  None
+# beta_y =  None
+# Q_x = None
+# Q_y = None
+
+# # Use this part for smooth machine
+optics_pickle_file = None
+n_segments = 3 
+beta_x =  400.0
+beta_y =  400.0
+Q_x = 62.27
+Q_y = 60.295
+
+Qp_x = 0.
+Qp_y = 0.
+
+octupole_knob = 0.
+
+V_RF = 12e6
+
+n_non_parallelizable = 2 #rf and aperture
+
+# Transverse Damper Settings
+enable_transverse_damper = False 
+dampingrate_x = 50. 
+dampingrate_y = 100.
+if enable_transverse_damper: n_non_parallelizable += 1
+
+
+###################
+# Beam Parameters #
+###################
+
+bunch_from_file = None
+
+intensity = 1.2e+11
+
+epsn_x = 2.5e-6
+epsn_y = 2.5e-6
+
+sigma_z = 1.2e-9/4*c
+
+x_kick_in_sigmas = 0.1
+y_kick_in_sigmas = 0.1
+
+# Numerical Parameters
+n_slices = 100
+z_cut = 2.5e-9/2*c # For slicing
+macroparticles_per_slice = 1000
+n_macroparticles = macroparticles_per_slice*n_slices
+
+
+#################
+# Stop Criteria #  
+#################
+
+# 1. Turns
+N_turns = 10 # Per job
+N_turns_target = 150 
+# 2. Losses
+sim_stop_frac = 0.9
+# 3. Emittance Growth
+flag_check_emittance_growth = True
+epsn_x_max_growth_fraction = 0.5
+epsn_y_max_growth_fraction = epsn_x_max_growth_fraction
+
+
+######################
+# Footprint Settings #
+######################
+
+footprint_mode = False
+n_macroparticles_for_footprint_map = 500000
+n_macroparticles_for_footprint_track = 5000
+
+
+####################
+# E-Cloud Settings #
+####################
+
+# General E-Cloud Settings
+chamb_type = 'polyg'
+x_aper = 2.300000e-02
+y_aper = 1.800000e-02
+filename_chm = 'LHC_chm_ver.mat'
+Dt_ref = 25e-12
+pyecl_input_folder = './pyecloud_config'
+sey = 1.30
+
+# Transverse Multigrid Parameters
+PyPICmode = 'ShortleyWeller_WithTelescopicGrids'
+N_min_Dh_main = 10.
+Dh_sc_ext = .8e-3
+f_telescope = 0.3
+N_nodes_discard = 5.
+target_size_internal_grid_sigma = 10.
+target_Dh_internal_grid_sigma = 0.5
+custom_target_grid_arcs = None
+
+# # Uncomment for custom grid
+# custom_target_grid_arcs = {
+#     'x_min_target': -3e-3,
+#     'x_max_target': 3e-3,
+#     'y_min_target': -3.1e-3,
+#     'y_max_target': 3.1e-3,
+#     'Dh_target': 7e-5}
+
+# Enable Kicks Different Planes
+enable_kick_x = True
+enable_kick_y = True
+
+# Dedicated Dipole E-Cloud Settings
+enable_arc_dip = True
+fraction_device_dip = 0.65
+init_unif_edens_flag_dip = 1
+init_unif_edens_dip = 1.000000e+12
+N_MP_ele_init_dip = 50000
+N_mp_max_dip = N_MP_ele_init_dip*4
+B_multip_dip = [8.33] #T
+
+# Dedicated Quadrupole E-Cloud Settings
+enable_arc_quad = False 
+fraction_device_quad = 26.000000e-02 #7.000000e-02
+N_mp_max_quad = 2000000 
+B_multip_quad = [0., 188.2] #T
+folder_path = '../../LHC_ecloud_distrib_quads/'
+filename_state = 'combined_distribution_sey%.2f_%.1fe11ppb_7tev.mat'%(sey,intensity/1e11)
+filename_init_MP_state_quad = folder_path + filename_state
+
+# Dedicated Kick Element Settings
+enable_eclouds_at_kick_elements = False 
+path_buildup_simulations_kick_elements = '/home/kparasch/workspace/Triplets/ec_headtail_triplets/simulations_PyECLOUD/!!!NAME!!!_sey1.35'
+name_MP_state_file_kick_elements = 'MP_state_9.mat'
+orbit_factor = 6.250000e-01
+

examples/001_simulation_with_eclouds/clean_all

@@ -0,0 +1,6 @@
+rm multigrid_config*
+rm simulation_status.sta
+rm bunch_evolution_*.h5
+rm slice_evolution_*.h5
+rm bunch_status_part*.h5
+rm *.pyc

examples/001_simulation_with_eclouds/mystyle.py

@@ -0,0 +1,31 @@
+from matplotlib import rc, rcdefaults
+import pylab as pl
+from colorsys import hsv_to_rgb
+
+
+def mystyle(fontsz=16):
+	
+	font = {#'family' : 'normal',
+			#'weight' : 'bold',
+			'size'   : fontsz}
+	print fontsz
+	rcdefaults()
+	rc('font', **font)
+	
+def mystyle_arial(fontsz=16, dist_tick_lab=10):
+	
+	rcdefaults()
+	rc('font',**{'family':'sans-serif','sans-serif':['arial'], 'size':fontsz})
+	rc(('xtick.major','xtick.minor','ytick.major','ytick.minor'), pad=dist_tick_lab)
+	
+
+
+def sciy():
+	pl.gca().ticklabel_format(style='sci', scilimits=(0,0),axis='y') 
+	
+def scix():
+	pl.gca().ticklabel_format(style='sci', scilimits=(0,0),axis='x') 
+
+
+def colorprog(i_prog, Nplots, v1 = .9, v2 = 1.):
+	return hsv_to_rgb(float(i_prog)/float(Nplots), v1, v2)