Merge branch 'develop'

001_reload_state_and_run.py

@@ -7,5 +7,10 @@
 
 
 sim = BuildupSimulation()
+
+# Optionally enable video saving
+sim.pyeclsaver.flag_video = False
+sim.pyeclsaver.flag_sc_video = False
+
 sim.load_state('simulation_state_0.pkl')
 sim.run()

other/make_distribution_movie/000_make_edistrib_movie.py

@@ -0,0 +1,183 @@
+import sys
+sys.path.append('../../../') 
+
+from mpl_toolkits.axes_grid1 import make_axes_locatable
+import pylab as pl
+import subprocess
+import numpy as np
+import PyECLOUD.mystyle as ms
+import os
+
+import scipy.io as sio
+import PyECLOUD.myloadmat_to_obj as mlo
+
+pl.close('all')
+
+fontsz=12
+
+bunch_spacing=25e-9
+
+first_passage=0
+num_passage=2
+
+# Please run the "LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns_stress_saver" befor using this script
+folder_sim = '../../testing/tests_buildup/LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns_stress_saver/'
+main_outp_filename = 'Pyecltest_angle3D.mat'
+
+passlist=range(first_passage,first_passage+num_passage)
+
+x_beam_pos = 0.
+y_beam_pos = 0.
+
+color_beam = 'r'
+tbeam_win_length = bunch_spacing*num_passage;
+
+########
+
+########
+
+xfield_cut_lim = 7 #5000.
+yfield_cut_lim = 7 #5000.
+
+denslim = 1.e11;
+
+N_dec=1
+lim_dens =  [10,14]
+flag_log=True
+outp_filname='pass%d.avi'%passlist[0]
+
+########
+
+########
+
+qe = 1.60217657e-19
+c = 299792458.
+
+i_photog=0
+
+ms.mystyle_arial(fontsz=fontsz, dist_tick_lab=10)
+
+chamber = mlo.myloadmat_to_obj(folder_sim+'/LHC_chm_ver.mat')
+Vx = chamber.Vx
+Vy = chamber.Vy
+
+##load output file
+obout = mlo.myloadmat_to_obj(folder_sim+'/'+main_outp_filename)
+t = obout.t
+lam_b1 = obout.lam_t_array
+cendens = np.squeeze(obout.cen_density)
+
+#every bunch passage
+for pass_ind in passlist:
+
+    filename_rho = folder_sim+'/rho_video/rho_pass%d.mat'%pass_ind
+    filename_efield = folder_sim+'/efield_video/efield_pass%d.mat'%pass_ind    
+    #new code
+
+    obout_ecl_video = mlo.myloadmat_to_obj(filename_rho)
+    obout_efield = mlo.myloadmat_to_obj(filename_efield)
+
+    xg_sc = obout_ecl_video.xg_sc
+    yg_sc = obout_ecl_video.yg_sc
+
+    xmin = np.min(xg_sc)
+    xmax = np.max(xg_sc)
+    ymin = np.min(yg_sc)
+    ymax = np.max(yg_sc)
+
+    ix_0 = np.argmin(np.abs(xg_sc-x_beam_pos))
+    iy_0 = np.argmin(np.abs(yg_sc-y_beam_pos))
+
+
+
+
+
+    rho_video = -obout_ecl_video.rho_video/qe
+    ex_video = obout_efield.efx_video
+    ey_video = obout_efield.efy_video
+    t_video = np.squeeze(obout_ecl_video.t_video.real)
+    b_spac = np.squeeze(obout.b_spac.real) 
+
+
+
+    (nphotog,_,_)=rho_video.shape
+
+
+
+
+    for ii in xrange(0, nphotog, N_dec):
+
+		fig = pl.figure(1,figsize=(4.5,6)) #,figsize=(4.5,6)
+		t_curr = t_video[ii]
+		cendens_curr = np.interp(t_curr, t, cendens)
+		lam_b1_curr = np.interp(t_curr, t, lam_b1)
+		cendens_curr = np.interp(t_curr, t, cendens)
+
+		print 'Pass %d %d/%d'%(pass_ind,ii,nphotog)
+
+		imm=np.squeeze(rho_video[ii,:,:])
+		imm_ex = np.squeeze(ex_video[ii,:,:])
+		imm_ey = np.squeeze(ey_video[ii,:,:])
+		immlin=imm.copy()
+
+		if flag_log:
+
+			imm = np.log10((imm))
+			imm[immlin<10**lim_dens[0]]=lim_dens[0]
+
+		pl.subplot(2,1,1)
+		ax = pl.gca()
+		im = ax.imshow(imm.T, cmap=None, norm=None, aspect='auto', interpolation=None,
+		   alpha=None, vmin=lim_dens[0], vmax=lim_dens[1], origin='lower', extent=[xmin*1e3, xmax*1e3, ymin*1e3, ymax*1e3])
+
+		pl.plot(Vx*1e3, Vy*1e3, 'y', linewidth=1.5)
+		pl.axis('equal')
+		pl.xlim(np.min(Vx*1e3)-np.max(Vx*1e3)/10, np.max(Vx*1e3)+np.max(Vx*1e3)/10)
+		pl.ylim(np.min(Vy*1e3)-np.max(Vy*1e3)/10, np.max(Vy*1e3)+np.max(Vy*1e3)/10)
+		pl.xlabel('x [mm]')
+		pl.ylabel('y [mm]')
+
+		divider = make_axes_locatable(ax)
+		cax = divider.append_axes("right", size="5%", pad=0.02)
+		clb = pl.colorbar(im, cax=cax)
+		clb.set_label('log10(e- dens.)')
+
+		pl.subplot(2,1,2)
+		mask_t_win = np.logical_and(t>t_curr-tbeam_win_length/2., t<t_curr+tbeam_win_length/2.)
+		pl.plot(c*(t[mask_t_win]-t_curr), lam_b1[mask_t_win])
+		#pl.plot(-c*(t[mask_t_win]-t_curr), lam_b2[mask_t_win], 'r')
+		pl.axis('tight')
+
+		pl.grid('on')
+
+		pl.axvline(0,linestyle='--',color='k')
+		pl.xlabel('(s-s0) [m]')
+		pl.ylabel('beam prof. [p/m]')
+
+		pl.subplots_adjust(top=0.85,right=0.8, left=0.15, hspace=0.3, wspace=0.5)	
+		filename = str('Pass%05d_%05d' % (pass_ind,ii)) + '.png'
+		pl.savefig(filename, dpi=150)
+		pl.clf()
+
+		i_photog+=1
+
+
+command = ('mencoder',
+           'mf://*.png',
+           '-mf',
+           'type=png:w=800:h=600:fps=5',
+           '-ovc',
+           'lavc',
+           '-lavcopts',
+           'vcodec=mpeg4',
+           '-oac',
+           'copy',
+           '-o',
+           outp_filname)
+
+
+subprocess.check_call(command)
+folderpngname = outp_filname.split('.avi')[0]+'_high_res_pngs'
+os.system('mkdir %s'%folderpngname)
+os.system('mv *.png '+folderpngname)
+

testing/tests_buildup/LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns_stress_saver/machine_parameters.input

@@ -11,9 +11,10 @@ filename_chm = 'LHC_chm_ver.mat'
 
 
 # Choose track_method= 'StrongBdip' / 'StrongBgen'/ 'Boris'
-track_method= 'BorisMultipole'
-N_sub_steps = 5
-B_multip = [0.53549999999999998]
+track_method= 'StrongBdip'
+B = 0.53549999999999998
+# N_sub_steps = 5
+# B_multip = [0.53549999999999998]