else:
lossloc = "AP.UP.ZS21633"
if not os.path.exists(trackfolder+'/0'):
subprocess.check_call('./unpacker.sh', shell=True, cwd=trackfolder)
failed, messages = datproc.errorcheck(errfolder)
print (str(len(failed))+" processes failed")
#failed = [int(x) for x in failed]
#print failed
print messages
# print ""
# datproc.beamstats(lossfolder, lossloc=lossloc)
print ""
if not pycoll:
datproc.efficiency(lossfolder, aperturex=[0.06815,0.08815], aperturey=[-0.023,0.023], zs_len=18.77, zs_an=4.1635E-4, aperturex2=[0.04265,0.06265])
print ""
datproc.wireangle(lossfolder)
datproc.lossplot(lossfolder, ylim=[-0.0018, -0.0013], save=plotfolder+"/beam.png")
datproc.lossplot(lossfolder, ylim=[-0.0016, 0.0016], xax='TURN', yax='PT', cax='PX', save=plotfolder+"/sweep.png")
else:
datproc.efficiency(lossfolder, pycoll=True, aperturex=[0.04219,0.08219], aperturey=[-0.0100,0.0100])
print ""
datproc.lossstats(lossfolder)
#datproc.zsbacktrack(trackfolder, cax="TURN", save_d=plotfolder+"/zs_down_backtrack_"+name+".png", save_u=plotfolder+"/zs_up_backtrack_"+name+".png")
datproc.trackplot(trackfolder, obsloc="obs0002", cax="TURN", tpt=1, save=plotfolder+"/zs_up_"+name+".png")
datproc.trackplot(trackfolder, obsloc="obs0002", xlim=[0.065,0.090], ylim=[-0.0018, -0.0013], cax="TURN", tpt=1, save=plotfolder+"/zs_up_zoom_"+name+".png")
datproc.trackplot(trackfolder, obsloc="obs0002", xlim=[0.05,0.090], ylim=[-0.0018, -0.0009], cax="TURN", tpt=1, save=plotfolder+"/zs_up_zoom2_"+name+".png")
print ""
plotfolder = folder + "/../plots/" + name
if not os.path.exists(plotfolder):
os.makedirs(plotfolder)
if not os.path.exists(trackfolder + '/0'):
subprocess.check_call('./unpacker.sh', shell=True, cwd=trackfolder)
# Print efficiency and loss stats
stdout = sys.stdout
with open(plotfolder + "/stats.txt", 'w') as sys.stdout:
datproc.efficiency(lossfolder,
aperturex=[0.06815, 0.08815],
aperturey=[-0.023, 0.023],
zs_len=18.77,
zs_an=4.1635E-4,
aperturex2=[0.04140, 0.06140])
print ""
datproc.wireangle(lossfolder)
print ""
datproc.emittance(lossfolder,
lossloc=lossloc,
ap=[0.04219, 0.08219],
betagamma=426.3167)
sys.stdout = stdout
# Make plots
datproc.lossplot(lossfolder,
lossloc=lossloc,
xax='TURN',
mpl.use('Agg') #To use plotting routines on machines without display
import python.dataprocessing as datproc
if __name__ == "__main__":
folder = sys.argv[1]
lossfolder = folder+'/losses'
trackfolder = folder+'/tracks'
errfolder = folder+'/error'
plotfolder = folder+"/../plots"#"/../sweepspeed_plots"#
name = folder.split("/")[-1]
if name=="":
name = folder.split("/")[-2]
lossloc = "AP.UP.TPST21760"
if not os.path.exists(trackfolder+'/0'):
subprocess.check_call('./unpacker.sh', shell=True, cwd=trackfolder)
failed, messages = datproc.errorcheck(errfolder)
print (str(len(failed))+" processes failed")
#failed = [int(x) for x in failed]
#print failed
print messages
print ""
datproc.efficiency(lossfolder, pycoll=True, aperturex=[0.04219,0.08219], aperturey=[-0.0100,0.0100])
datproc.trackplot(trackfolder, obsloc="obs0002", cax="TURN", tpt=1, save=plotfolder+"/zs_up_"+name+".png")
datproc.trackplot(trackfolder, obsloc="obs0002", cax="TURN", tpt=9, save=plotfolder+"/zs_up_more_"+name+".png")
datproc.lossplot(lossfolder, lossloc=lossloc, cax="TURN", save=plotfolder+"/tpst_exbeam_"+name+".png")
plotfolder = folder + "/../plots" #"/../sweepspeed_plots"#
name = folder.split("/")[-1]
if name == "":
name = folder.split("/")[-2]
if not os.path.exists(trackfolder + '/0'):
subprocess.check_call('./unpacker.sh', shell=True, cwd=trackfolder)
failed, messages = datproc.errorcheck(errfolder)
print(str(len(failed)) + " processes failed")
#failed = [int(x) for x in failed]
#print failed
print messages
print ""
datproc.efficiency(lossfolder,
aperturex=[0.06815, 0.08815],
aperturey=[-0.023, 0.023])
datproc.trackplot(trackfolder,
obsloc="obs0002",
cax="TURN",
tpt=1,
save=plotfolder + "/qfa_" + name + ".png")
datproc.trackplot(trackfolder,
obsloc="obs0002",
cax="TURN",
tpt=9,
save=plotfolder + "/qfa_more_" + name + ".png")
datproc.trackplot(trackfolder,
obsloc="obs0003",
cax="TURN",
myap = {
'aperturex':
ap['zsupmid'] + ap['zsthick'] / 2 + np.array([0, ap['zsex']]),
'aperturex2':
ap['zsdomid'] + ap['zsthick'] / 2 + np.array([0, ap['zsex']]),
'aperturey': [-0.023, 0.023],
'zs_len':
18.77,
'zs_an':
4.1635E-4
}
stdout = sys.stdout
with open(plotfolder + "/stats.txt", 'w') as sys.stdout:
print 'Probabilities with 95% confidence intervals:'
alltags, _ = datproc.extr_tagger(data=losses, pycoll=pycoll, **myap)
datproc.efficiency(losses, alltags=alltags, silent=False)
print '\nLoss stats:'
datproc.lossstats(losses, merge=False, silent=False)
print 'Beam stats:'
emita = datproc.beamstats(losses[losses['tag'] == 'extracted'],
silent=False)
print '\nBeam fit (99.99%):'
emitb = datproc.get_ellipse(losses[losses['tag'] == 'extracted'],
x0=emita['X0'],
px0=emita['PX0'],
alpha0=emita['alpha'],
beta0=emita['beta'],
silent=False)
sys.stdout = stdout
if '1' in kind:
def makeplots(folder):
lossfolder = folder + '/losses'
trackfolder = folder + '/tracks'
errfolder = folder + '/error'
name = folder.split("/")[-1]
if name == "":
name = folder.split("/")[-2]
settings = datproc.getsettings(folder)
sliced = False if settings['slices'] == 'None' else True
pycoll = eval(settings['pycollimate'])
lossloc = "AP.UP.TPST21760" if pycoll else "AP.UP.ZS21633"
failed, messages = datproc.errorcheck(errfolder)
print(str(len(failed)) + " processes failed")
print "errors"
for message in messages:
print message
print ""
plotfolder = folder + "/../plots/" + name
if not os.path.exists(plotfolder):
os.makedirs(plotfolder)
if not os.path.exists(trackfolder + '/0'):
subprocess.check_call('./unpacker.sh', shell=True, cwd=trackfolder)
# Print efficiency and loss stats
stdout = sys.stdout
with open(plotfolder + "/stats.txt", 'w') as sys.stdout:
if pycoll:
datproc.efficiency(
lossfolder,
pycoll=True,
aperturex=[tpstcap + tpstblade, tpstcap + tpstblade + tpsteap],
aperturey=[-0.0100, 0.0100])
print ""
datproc.lossstats(lossfolder)
print ""
datproc.emittance(
lossfolder,
lossloc=lossloc,
ap=[tpstcap + tpstblade, tpstcap + tpstblade + tpsteap],
betagamma=426.3167)
else:
datproc.efficiency(lossfolder,
aperturex=[0.06815, 0.08815],
aperturey=[-0.023, 0.023],
zs_len=18.77,
zs_an=4.1635E-4,
aperturex2=[0.04160, 0.06160])
print ""
datproc.wireangle(lossfolder)
print ""
datproc.emittance(lossfolder,
lossloc=lossloc,
ap=[0.06815, 0.08815],
betagamma=426.3167)
sys.stdout = stdout
# Make plots
datproc.lossplot(lossfolder,
lossloc=lossloc,
xax='TURN',
yax='PT',
cax='PT',
ylim=[-0.0025, 0.0020],
clim=[-0.0025, 0.0020],
save=plotfolder + "/sweep_" + name + ".png")
datproc.trackplot(trackfolder,
obsloc="obs0002",
cax="PT",
clim=[-0.0025, 0.0020],
tpt=1,
save=plotfolder + "/zs_up_" + name + ".png")
datproc.trackplot(trackfolder,
obsloc="obs0003",
cax="PT",
clim=[-0.0025, 0.0020],
tpt=1,
save=plotfolder + "/zs_down_" + name + ".png")
datproc.trackplot(trackfolder,
obsloc="obs0004",
cax="PT",
clim=[-0.0025, 0.0020],
tpt=1,
save=plotfolder + "/tpst_circ_" + name + ".png")
if sliced:
datproc.trackplot(trackfolder,
obsloc="obs0002",
cax="PT",
clim=[-0.0025, 0.0020],
tpt=300,
save=plotfolder + "/zs_up_full_" + name + ".png")
datproc.trackplot(trackfolder,
obsloc="obs0003",
cax="PT",
clim=[-0.0025, 0.0020],
tpt=300,
save=plotfolder + "/zs_down_full_" + name + ".png")
datproc.trackplot(trackfolder,
obsloc="obs0004",
cax="PT",
clim=[-0.0025, 0.0020],
tpt=300,
save=plotfolder + "/tpst_circ_full_" + name + ".png")
if pycoll:
datproc.losshistscatter(lossfolder,
lossloc=lossloc,
xlim=[tpstcap, 0.095],
ylim=[0.0003, 0.0027],
cax="PT",
clim=[-0.0025, 0.0020],
xbin=0.001,
ybin=0.00005,
log=True,
save=plotfolder + "/tpst_losshist_h_" + name +
".png")
datproc.losshistscatter(lossfolder,
lossloc=lossloc,
xax='Y',
yax='PY',
xlim=[-0.02, 0.02],
ylim=[-0.0006, 0.0006],
cax="PT",
clim=[-0.0025, 0.0020],
xbin=0.0006,
ybin=0.00003,
log=True,
save=plotfolder + "/tpst_losshist_v_" + name +
".png")
datproc.losshistscatter(lossfolder,
lossloc=lossloc,
xax='X',
yax='Y',
xlim=[0.03759, 0.095],
ylim=[-0.02, 0.02],
cax="PT",
clim=[-0.0025, 0.0020],
xbin=0.001,
ybin=0.0006,
log=True,
save=plotfolder + "/tpst_losshist_s_" + name +
".png")
datproc.losshistcombo(lossfolder,
lossloc=lossloc,
xlim=[tpstcap, 0.095],
ylim=[0.0003, 0.0027],
xbin=0.0005,
ybin=0.000025,
cm='viridis',
log=True,
save=plotfolder + "/losshist2_" + name + ".png")
datproc.lossplot(lossfolder,
lossloc=None,
xlim=[1668, 1688],
xax='S',
yax='X',
cax="PT",
clim=[-0.0025, 0.0020],
save=plotfolder + "/zs_loss_" + name + ".png")
else:
datproc.lossplot(lossfolder,
xlim=[0.0675, 0.0885],
ylim=[-0.00185, -0.00135],
cax="PT",
clim=[-0.0025, 0.0020],
save=plotfolder + "/beam_" + name + ".png")
datproc.losshistscatter(lossfolder,
lossloc=lossloc,
xlim=[0.06815, 0.08815],
ylim=[-0.00185, -0.00135],
cax="PT",
clim=[-0.0025, 0.0020],
xbin=0.0002,
ybin=0.00001,
log=False,
save=plotfolder + "/losshist_" + name + ".png")
datproc.losshistcombo(lossfolder,
lossloc=lossloc,
xlim=[0.06815, 0.08815],
ylim=[-0.00185, -0.00135],
xbin=0.0002,
ybin=0.000005,
cm='viridis',
log=False,
save=plotfolder + "/losshist2_" + name + ".png")
datproc.losshistcombo(lossfolder,
lossloc=lossloc,
xlim=[0.06815, 0.07215],
ylim=[-0.00155, -0.0014],
xbin=0.00008,
ybin=0.000004,
cm='viridis',
log=False,
save=plotfolder + "/losshist2_zoom_" + name +
".png")
