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")
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")
datproc.trackplot(trackfolder, obsloc="obs0003", cax="TURN", tpt=1, save=plotfolder+"/zs_down_"+name+".png")
trackfolder = folder + '/tracks'
errfolder = folder + '/error'
name = folder.split("/")[-1]
if name == "":
name = folder.split("/")[-2]
plotfolder = folder + "/../plots/" + name
if not os.path.exists(plotfolder):
os.makedirs(plotfolder)
lossloc = "AP.UP.ZS21633"
for width in [200, 500, 1000, 2000]:
print width
datproc.wireangle(lossfolder, wiremax=(0.06795 + width * 1E-6))
datproc.losshistscatter(lossfolder,
lossloc=lossloc,
cax="PT",
xlim=[0.06795, 0.06995],
ylim=[-0.0016, -0.00135],
xbin=0.00005,
ybin=0.000005,
datalim=[[None, 0.06795 + width * 1E-6],
[None, None], [None, None]],
log=False,
save=plotfolder + "/losshist_" + str(width) +
".png")
datproc.efficiency(lossfolder,
aperturex=[0.06815, 0.08815],