diff --git a/examples/asc_test/master.py b/examples/asc_test/master.py
index eed7854364da3bde46e000f63e1d84b60a41450d..9bfa5b249c079d93c7ccdc076d560e2399bc2102 100644
--- a/examples/asc_test/master.py
+++ b/examples/asc_test/master.py
@@ -6,7 +6,8 @@ from __future__ import unicode_literals
from pykat import finesse
from pykat.commands import *
import copy
-import shelve
+#import shelve
+import pickle
import sys
import scipy.optimize
@@ -89,10 +90,12 @@ def main():
print(" current results saved in: {0}".format(tmpresultfile))
# first the current kat file
kat.saveScript(tmpkatfile)
- # now the result variables:
- tmpfile = shelve.open(tmpresultfile, flag="c")
- tmpfile[str('result')]=result
- tmpfile.close()
+ with open(tmpresultfile, 'wb') as handle:
+ pickle.dump(result, handle)
+ # now the result variables (former version usuing shelve):
+ #tmpfile = shelve.open(tmpresultfile, flag="c")
+ #tmpfile[str('result')]=result
+ #tmpfile.close()
#---------------------------------------------------------------------------
diff --git a/examples/asc_test/master2.py b/examples/asc_test/master2.py
index 413d7e33440cf630a4dcfb1ada13f3c437f314fc..eb6330b52a4efad17b44c2736c1ae486cee73a07 100644
--- a/examples/asc_test/master2.py
+++ b/examples/asc_test/master2.py
@@ -9,7 +9,8 @@ import pylab as pl
import scipy
from scipy.optimize import fmin
import numpy as np
-import shelve
+import pickle
+#import shelve
import copy
import sys
@@ -51,9 +52,11 @@ def main():
# loading data saved by master.py
kat.loadKatFile('asc_base2.kat')
try:
- tmpfile = shelve.open(tmpresultfile, flag="c")
- result=tmpfile[str('result')]
- tmpfile.close()
+ #tmpfile = shelve.open(tmpresultfile, flag="c")
+ #result=tmpfile[str('result')]
+ #tmpfile.close()
+ with open(tmpresultfile, 'rb') as handle:
+ result = pickle.load(handle)
except: raise Exception("Could not open temprary results file {0}".format(tmpresultfile))
# overwriting some variables
@@ -125,9 +128,11 @@ def main():
# first the current kat file
kat.saveScript(tmpkatfile)
# now the result variables:
- tmpfile = shelve.open(tmpresultfile)
- tmpfile[str('result')]=result
- tmpfile.close()
+ #tmpfile = shelve.open(tmpresultfile)
+ #tmpfile[str('result')]=result
+ #tmpfile.close()
+ with open(tmpresultfile, 'wb') as handle:
+ pickle.dump(result, handle)
#-----------------------------------------------------------------------------------
@@ -186,7 +191,7 @@ def asc_phases(tmpkat):
kat.WFS1_I.phase1=x[0]
out = kat.run()
signal = out["WFS1_I"]
- print('\r minimising: function value %g ' % signal, end=' ')
+ print('\r minimising: function value {0:<16g}'.format(float(signal)), end='')
sys.stdout.flush()
return -1*abs(signal)
@@ -194,7 +199,7 @@ def asc_phases(tmpkat):
kat.WFS2_I.phase1=x[0]
out = kat.run()
signal = out["WFS2_I"]
- print('\r minimising: function value %g ' % signal, end=' ')
+ print('\r minimising: function value {0:<16g}'.format(float(signal)), end='')
sys.stdout.flush()
return -1*abs(signal)
diff --git a/examples/asc_test/master3.py b/examples/asc_test/master3.py
index b1c3f699a291b1d76621da032399aef4ae6b01b9..1fb2a7f09f88f8dd697a50f7800d9a1d2f380113 100644
--- a/examples/asc_test/master3.py
+++ b/examples/asc_test/master3.py
@@ -6,7 +6,7 @@ from __future__ import unicode_literals
from pykat import finesse
from pykat.commands import *
import numpy as np
-import shelve
+import pickle
import copy
import sys
import shutil
@@ -48,9 +48,11 @@ def main():
# loading data saved by master.py
kat.loadKatFile('asc_base3.kat')
try:
- tmpfile = shelve.open(tmpresultfile)
- result=tmpfile[str('result')]
- tmpfile.close()
+ with open(tmpresultfile, 'rb') as handle:
+ result = pickle.load(handle)
+ #tmpfile = shelve.open(tmpresultfile)
+ #result=tmpfile[str('result')]
+ #tmpfile.close()
except: raise Exception("Could not open temprary results file {0}".format(tmpresultfile))
kat.PDrefl_q.enabled = False
@@ -103,10 +105,13 @@ def asc_large(tmpkat, mir_name):
shutil.copyfile(tmpfilename, backupname)
print(" current results saved in: {0}".format(tmpfilename))
- tmpfile = shelve.open(tmpfilename)
- tmpfile[str('out')]=out
- tmpfile[str('maxtems')]=done_maxtems
- tmpfile.close()
+ with open(tmpfilename, 'wb') as handle:
+ pickle.dump({ "out": out, "maxtems": done_maxtems}, handle)
+
+ #tmpfile = shelve.open(tmpfilename)
+ #tmpfile[str('out')]=out
+ #tmpfile[str('maxtems')]=done_maxtems
+ #tmpfile.close()
if __name__ == '__main__':
diff --git a/examples/asc_test/master3_plot.py b/examples/asc_test/master3_plot.py
index b3f73a94d05f79f1c3425c276fe269f9d3823073..695c727e4815851fdaf172c72d6aa9feccb6965e 100644
--- a/examples/asc_test/master3_plot.py
+++ b/examples/asc_test/master3_plot.py
@@ -7,7 +7,7 @@ from pykat import finesse
from pykat.commands import *
import pylab as pl
import numpy as np
-import shelve
+import pickle
import sys
import matplotlib
@@ -16,94 +16,97 @@ formatter.ENG_PREFIXES[-6] = 'u'
import matplotlib.backends.backend_pdf
def printPDF(self, filename):
- pdfp = matplotlib.backends.backend_pdf.PdfPages(filename)
- pdfp.savefig(self,dpi=300,bbox_inches='tight')
- pdfp.close()
+ pdfp = matplotlib.backends.backend_pdf.PdfPages(filename)
+ pdfp.savefig(self,dpi=300,bbox_inches='tight')
+ pdfp.close()
def main():
- print("""
- --------------------------------------------------------------
- Example file for using PyKat to automate Finesse simulations
- Finesse: http://www.gwoptics.org/finesse
- PyKat: http://www.gwoptics.org/pykat
-
- Run this file to plot the data generated with master3.py.
-
- Andreas Freise 16.01.2014
- --------------------------------------------------------------
- """)
-
- # shall we clear the workspace?
- # %reset -f
- # maybe close all plot windows?
- # close('all')
-
- # making these global during testing and debugging
- #global kat
- #global out
- #global result
-
- print("--------------------------------------------------------")
- print(" Plotting ASC signals for large misalignments")
- asc_large('ITM')
- asc_large('ETM')
-
+ print("""
+ --------------------------------------------------------------
+ Example file for using PyKat to automate Finesse simulations
+ Finesse: http://www.gwoptics.org/finesse
+ PyKat: http://www.gwoptics.org/pykat
+
+ Run this file to plot the data generated with master3.py.
+
+ Andreas Freise 16.01.2014
+ --------------------------------------------------------------
+ """)
+
+ # shall we clear the workspace?
+ # %reset -f
+ # maybe close all plot windows?
+ # close('all')
+
+ # making these global during testing and debugging
+ #global kat
+ #global out
+ #global result
+
+ print("--------------------------------------------------------")
+ print(" Plotting ASC signals for large misalignments")
+ asc_large('ITM')
+ asc_large('ETM')
+
def asc_large(mir_name):
- xscale = 1e6
- yscale = 1e6
- if mir_name == 'ITM':
- ysign = 1.0
- else:
- ysign = -1.0
-
- tmpfilename = "datashelf_{0}.dat".format(mir_name)
- backupname = "datashelf_{0}.dat.bck".format(mir_name)
+ xscale = 1e6
+ yscale = 1e6
+ if mir_name == 'ITM':
+ ysign = 1.0
+ else:
+ ysign = -1.0
+
+ tmpfilename = "datashelf_{0}.dat".format(mir_name)
+ backupname = "datashelf_{0}.dat.bck".format(mir_name)
- try:
- tmpfile = shelve.open(tmpfilename)
- out=tmpfile[str('out')]
- maxtems=tmpfile[str('maxtems')]
- tmpfile.close()
- except: raise Exception("Could not open temprary results file {0}".format(tmpfilename))
+ try:
+ #tmpfile = shelve.open(tmpfilename)
+ #out=tmpfile[str('out')]
+ #maxtems=tmpfile[str('maxtems')]
+ #tmpfile.close()
+ readpickle = pickle.load(open(tmpfilename, "rb"))
+ out=readpickle['out']
+ maxtems=readpickle['maxtems']
+ except: raise Exception("Could not open temprary results file {0}".format(tmpfilename))
- fig=pl.figure()
- color_cycle=['b', 'c', 'r', 'k']
- N=len(maxtems)
- lw=np.ones(N)*3
- lw[-2]=2
- lw[-1]=1
- for i, tem in enumerate(maxtems):
- data=out[str(tem)]
- WFS1_idx=data.ylabels.index("WFS1_I")
- WFS2_idx=data.ylabels.index("WFS2_I")
- pl.plot(xscale*data.x,ysign*yscale*data.y[:,WFS1_idx],'-', color= color_cycle[i], linewidth=lw[i], label='maxtem {0}'.format(tem))
- line, = pl.plot(xscale*data.x,ysign*yscale*data.y[:,WFS2_idx],'-', color = color_cycle[i], linewidth=lw[i])
- #line.set_dashes([12, 4])
+ fig=pl.figure()
+ color_cycle=['b', 'c', 'r', 'k']
+ N=len(maxtems)
+ lw=np.ones(N)*3
+ lw[-2]=2
+ lw[-1]=1
+ for i, tem in enumerate(maxtems):
+ data=out[str(tem)]
+ WFS1_idx=data.ylabels.index("WFS1_I")
+ WFS2_idx=data.ylabels.index("WFS2_I")
+ pl.plot(xscale*data.x,ysign*yscale*data.y[:,WFS1_idx],'-', color= color_cycle[i], linewidth=lw[i], label='maxtem {0}'.format(tem))
+ line, = pl.plot(xscale*data.x,ysign*yscale*data.y[:,WFS2_idx],'-', color = color_cycle[i], linewidth=lw[i])
+ #line.set_dashes([12, 4])
- osc1=np.loadtxt("OSCAR_large_tilt_{0}.txt".format(mir_name),comments='%')
+ osc1=np.loadtxt("OSCAR_large_tilt_{0}.txt".format(mir_name),comments='%')
- x=xscale*osc1[:,0]
- y=yscale*osc1[:,1]
- pl.scatter(x,y,s=80,facecolors='none', edgecolors='k', label='OSCAR')
- y=yscale*osc1[:,2]
- pl.scatter(x,y,s=80,facecolors='none', edgecolors='k')
- pl.xlabel("{0} vertical misalignment [urad]".format(mir_name))
- pl.ylabel("Alignment signal [uW]")
- if mir_name == 'ITM':
- pl.annotate('WFS1',xy=[0.42,70])
- pl.annotate('WFS2',xy=[0.62,5])
- pl.legend(loc=2)
- else:
- pl.annotate('WFS1',xy=[0.42,10])
- pl.annotate('WFS2',xy=[0.62,-70])
- pl.legend(loc=3)
- pl.xlim([0,1])
- pl.grid()
- pl.draw()
- pl.show(block=0)
- printPDF(fig, "large_{0}.pdf".format(mir_name))
-
+ x=xscale*osc1[:,0]
+ y=yscale*osc1[:,1]
+ pl.scatter(x,y,s=80,facecolors='none', edgecolors='k', label='OSCAR')
+ y=yscale*osc1[:,2]
+ pl.scatter(x,y,s=80,facecolors='none', edgecolors='k')
+ pl.xlabel("{0} vertical misalignment [urad]".format(mir_name))
+ pl.ylabel("Alignment signal [uW]")
+ if mir_name == 'ITM':
+ pl.annotate('WFS1',xy=[0.42,70])
+ pl.annotate('WFS2',xy=[0.62,5])
+ pl.legend(loc=2)
+ else:
+ pl.annotate('WFS1',xy=[0.42,10])
+ pl.annotate('WFS2',xy=[0.62,-70])
+ pl.legend(loc=3)
+ pl.xlim([0,1])
+ pl.grid()
+ pl.draw()
+ pl.show(block=0)
+ printPDF(fig, "large_{0}.pdf".format(mir_name))
+
if __name__ == '__main__':
- main()
+ main()
diff --git a/examples/asc_test/master4.py b/examples/asc_test/master4.py
index 71f2eefc28578340605a847865f7dfbc2dcabf93..e4a2c0188e98fcdbc7b0a7aed1bb3100c24f2a4b 100644
--- a/examples/asc_test/master4.py
+++ b/examples/asc_test/master4.py
@@ -11,337 +11,339 @@ import pylab as pl
import scipy
from scipy.optimize import minimize_scalar
import numpy as np
-import shelve
+import pickle
import copy
import sys
def set_thermal_lens(kat, f):
- kat.ITM_TL.f=f
- # if a bs-based cavity is used, we need to set second lens
- if "ITM_TL_r" in kat._kat__components:
- kat.ITM_TL_r.f=f
- return (kat)
+ kat.ITM_TL.f=f
+ # if a bs-based cavity is used, we need to set second lens
+ if "ITM_TL_r" in kat._kat__components:
+ kat.ITM_TL_r.f=f
+ return (kat)
def main():
- print("""
- --------------------------------------------------------------
- Example file for using PyKat to automate Finesse simulations
- Finesse: http://www.gwoptics.org/finesse
- PyKat: https://pypi.python.org/pypi/PyKat/
-
- The file runs through the various Finesse simulations
- to generate the Finesse results reported in the document:
- `Comparing Finesse simulations, analytical solutions and OSCAR
- simulations of Fabry-Perot alignment signals', LIGO-T1300345,
- freely available online: http://arxiv.org/abs/1401.5727
-
- Run this file after master2.py to create data which can be
- plotted using master4_plot.py. Results are saved after
- each step and plots can be created at any time.
-
- Andreas Freise 16.01.2014
- --------------------------------------------------------------
- """)
-
- # shall we clear the workspace?
- # %reset -f
-
- # making these global during testing and debugging
- #global kat
- #global out
-
- kat = finesse.kat(tempdir=".",tempname="test")
- kat.verbose = False
-
- tmpresultfile = 'myshelf2.dat'
-
- # loading data saved by master.py
- kat.loadKatFile('asc_base3.kat')
- try:
- tmpfile = shelve.open(tmpresultfile)
- result=tmpfile[str('result')]
- tmpfile.close()
- except: raise Exception("Could not open temprary results file {0}".format(tmpresultfile))
-
- # this does not work yet due to the scale command
- kat.PDrefl_p.enabled = False
- kat.PDrefl_q.enabled = False
- kat.WFS1_I.enabled = False
- kat.WFS1_Q.enabled = False
- kat.WFS2_I.enabled = False
- kat.WFS2_Q.enabled = False
-
- kat.ETM.phi=result['phi_tuned']
-
- print("--------------------------------------------------------")
- print(" 11. Do beam tracing to measure beam parameters")
- # get beam parameters at nodes: "npsl", "nITM1", "nWFS1", "nWFS2", "npo2"
- global beam1, beam2, beam3
- beam1 = get_qs(kat,1e13)
- beam2 = get_qs(kat,50e3)
- beam3 = get_qs(kat,5e3)
-
- # starting with cold beam at npsl
- kat.psl.npsl.node.setGauss(kat.psl, beam1[0])
- kat.parseKatCode("startnode npsl")
-
-
- # if we use bs-based cavity we try to set good beam
- # parameter for reflected beam, first by
- # computing 'average' beam from cold and hot beams
- x1=0.70
- x2=0.30
- if "ITM_TL_r" in kat._kat__components:
- beam50 = beam_param(z=(x1*beam1[1].z+x2*beam2[1].z), w0=(x1*beam1[1].w0+x2*beam2[1].w0))
- beam5 = beam_param(z=(x1*beam1[1].z+x2*beam3[1].z), w0=(x1*beam1[1].w0+x2*beam3[1].w0))
- node_text = "at ITM->nITM1r"
- t_comp=kat.ITM
- t_node=kat.ITM.nITM1r
- else:
- beam50 = beam_param(z=(x1*beam1[4].z+x2*beam2[4].z), w0=(x1*beam1[4].w0+x2*beam2[4].w0))
- beam5 = beam_param(z=(x1*beam1[4].z+x2*beam3[4].z), w0=(x1*beam1[4].w0+x2*beam3[4].w0))
- node_text = "at s2->npo2"
- t_comp=kat.s2
- t_node=kat.s2.npo2
-
- kat = set_thermal_lens(kat,50e3)
- print("--------------------------------------------------------")
- print(" 12. computing beam tilt with thermal lens (f={0})".format(kat.ITM_TL.f))
-
- print(" Setting compromise beam parameter {0}: w0={1}, z={2}".format(node_text, beam50.w0, beam50.z))
- t_node.node.setGauss(t_comp, beam50)
- kat.maxtem=8
- print(" Calculating maxtem = %d " % kat.maxtem)
- tmp = gravity_tilt(kat)
-
- kat = set_thermal_lens(kat,5e3)
- print("--------------------------------------------------------")
- print(" 13. computing beam tilt with thermal lens (f={0})".format(kat.ITM_TL.f))
- print(" Setting compromise beam parameter {0}: w0={1}, z={2}".format(node_text, beam5.w0, beam5.z))
- t_node.node.setGauss(t_comp, beam5)
- #maxtems = [1, 3, 5, 9, 11, 13, 15, 19, 23, 25, 27, 29]
- maxtems = [1, 3, 5, 7]
- converge_tilt(kat, maxtems)
+ print("""
+ --------------------------------------------------------------
+ Example file for using PyKat to automate Finesse simulations
+ Finesse: http://www.gwoptics.org/finesse
+ PyKat: https://pypi.python.org/pypi/PyKat/
+
+ The file runs through the various Finesse simulations
+ to generate the Finesse results reported in the document:
+ `Comparing Finesse simulations, analytical solutions and OSCAR
+ simulations of Fabry-Perot alignment signals', LIGO-T1300345,
+ freely available online: http://arxiv.org/abs/1401.5727
+
+ Run this file after master2.py to create data which can be
+ plotted using master4_plot.py. Results are saved after
+ each step and plots can be created at any time.
+
+ Andreas Freise 16.01.2014
+ --------------------------------------------------------------
+ """)
+
+ # shall we clear the workspace?
+ # %reset -f
+
+ # making these global during testing and debugging
+ #global kat
+ #global out
+
+ kat = finesse.kat(tempdir=".",tempname="test")
+ kat.verbose = False
+
+ tmpresultfile = 'myshelf2.dat'
+
+ # loading data saved by master.py
+ kat.loadKatFile('asc_base3.kat')
+ try:
+ with open(tmpresultfile, 'rb') as handle:
+ result = pickle.load(handle)
+ #tmpfile = shelve.open(tmpresultfile)
+ #result=tmpfile[str('result')]
+ #tmpfile.close()
+ except: raise Exception("Could not open temprary results file {0}".format(tmpresultfile))
+
+ # this does not work yet due to the scale command
+ kat.PDrefl_p.enabled = False
+ kat.PDrefl_q.enabled = False
+ kat.WFS1_I.enabled = False
+ kat.WFS1_Q.enabled = False
+ kat.WFS2_I.enabled = False
+ kat.WFS2_Q.enabled = False
+
+ kat.ETM.phi=result['phi_tuned']
+
+ print("--------------------------------------------------------")
+ print(" 11. Do beam tracing to measure beam parameters")
+ # get beam parameters at nodes: "npsl", "nITM1", "nWFS1", "nWFS2", "npo2"
+ global beam1, beam2, beam3
+ beam1 = get_qs(kat,1e13)
+ beam2 = get_qs(kat,50e3)
+ beam3 = get_qs(kat,5e3)
+
+ # starting with cold beam at npsl
+ kat.psl.npsl.node.setGauss(kat.psl, beam1[0])
+ kat.parseKatCode("startnode npsl")
+
+
+ # if we use bs-based cavity we try to set good beam
+ # parameter for reflected beam, first by
+ # computing 'average' beam from cold and hot beams
+ x1=0.70
+ x2=0.30
+ if "ITM_TL_r" in kat._kat__components:
+ beam50 = beam_param(z=(x1*beam1[1].z+x2*beam2[1].z), w0=(x1*beam1[1].w0+x2*beam2[1].w0))
+ beam5 = beam_param(z=(x1*beam1[1].z+x2*beam3[1].z), w0=(x1*beam1[1].w0+x2*beam3[1].w0))
+ node_text = "at ITM->nITM1r"
+ t_comp=kat.ITM
+ t_node=kat.ITM.nITM1r
+ else:
+ beam50 = beam_param(z=(x1*beam1[4].z+x2*beam2[4].z), w0=(x1*beam1[4].w0+x2*beam2[4].w0))
+ beam5 = beam_param(z=(x1*beam1[4].z+x2*beam3[4].z), w0=(x1*beam1[4].w0+x2*beam3[4].w0))
+ node_text = "at s2->npo2"
+ t_comp=kat.s2
+ t_node=kat.s2.npo2
+
+ kat = set_thermal_lens(kat,50e3)
+ print("--------------------------------------------------------")
+ print(" 12. computing beam tilt with thermal lens (f={0})".format(kat.ITM_TL.f))
+
+ print(" Setting compromise beam parameter {0}: w0={1}, z={2}".format(node_text, beam50.w0, beam50.z))
+ t_node.node.setGauss(t_comp, beam50)
+ kat.maxtem=8
+ print(" Calculating maxtem = %d " % kat.maxtem)
+ tmp = gravity_tilt(kat)
+
+ kat = set_thermal_lens(kat,5e3)
+ print("--------------------------------------------------------")
+ print(" 13. computing beam tilt with thermal lens (f={0})".format(kat.ITM_TL.f))
+ print(" Setting compromise beam parameter {0}: w0={1}, z={2}".format(node_text, beam5.w0, beam5.z))
+ t_node.node.setGauss(t_comp, beam5)
+ #maxtems = [1, 3, 5, 9, 11, 13, 15, 19, 23, 25, 27, 29]
+ maxtems = [1, 3, 5, 7]
+ converge_tilt(kat, maxtems)
- kat.PDrefl_p.enabled = True
- kat.WFS1_I.enabled = True
- kat.WFS2_I.enabled = True
-
- kat = set_thermal_lens(kat,50e3)
- print("--------------------------------------------------------")
- print(" 12. computing alignment signal with thermal lens (f={0})".format(kat.ITM_TL.f))
- t_node.node.setGauss(t_comp, beam50)
- #maxtems = [1, 3, 5, 9, 11, 13, 15, 17, 19]
- maxtems = [1, 3, 5, 7]
- converge_asc(kat, maxtems, 'asc_signals_50.txt')
-
- kat = set_thermal_lens(kat,5e3)
- print("--------------------------------------------------------")
- print(" 13. computing alignment signal with thermal lens (f={0})".format(kat.ITM_TL.f))
- t_node.node.setGauss(t_comp, beam5)
- #maxtems = [1, 3, 5, 7, 9, 11, 13, 15, 17, 19, 21, 23, 25, 27, 29, 31]
- maxtems = [1, 3, 5, 7]
- converge_asc(kat, maxtems, 'asc_signals_5.txt')
+ kat.PDrefl_p.enabled = True
+ kat.WFS1_I.enabled = True
+ kat.WFS2_I.enabled = True
+
+ kat = set_thermal_lens(kat,50e3)
+ print("--------------------------------------------------------")
+ print(" 12. computing alignment signal with thermal lens (f={0})".format(kat.ITM_TL.f))
+ t_node.node.setGauss(t_comp, beam50)
+ #maxtems = [1, 3, 5, 9, 11, 13, 15, 17, 19]
+ maxtems = [1, 3, 5, 7]
+ converge_asc(kat, maxtems, 'asc_signals_50.txt')
+
+ kat = set_thermal_lens(kat,5e3)
+ print("--------------------------------------------------------")
+ print(" 13. computing alignment signal with thermal lens (f={0})".format(kat.ITM_TL.f))
+ t_node.node.setGauss(t_comp, beam5)
+ #maxtems = [1, 3, 5, 7, 9, 11, 13, 15, 17, 19, 21, 23, 25, 27, 29, 31]
+ maxtems = [1, 3, 5, 7]
+ converge_asc(kat, maxtems, 'asc_signals_5.txt')
#-----------------------------------------------------------------------------------
def converge_asc(tmpkat, maxtems, filename):
- kat = copy.deepcopy(tmpkat)
- savedata = np.zeros([len(maxtems),5])
- for idx, tem in enumerate(maxtems):
- savedata[idx,0]=tem
- print(" Calculating maxtem = %d " % tem)
- kat.maxtem = tem
- tmp = asc_signal(kat)
- savedata[idx,1:5]=tmp.reshape([1,4])
- print(" Saving results in file: {0}".format(filename))
- np.savetxt(filename, savedata[0:idx+1,:], fmt=b'%.18e', delimiter=' ')
+ kat = copy.deepcopy(tmpkat)
+ savedata = np.zeros([len(maxtems),5])
+ for idx, tem in enumerate(maxtems):
+ savedata[idx,0]=tem
+ print(" Calculating maxtem = %d " % tem)
+ kat.maxtem = tem
+ tmp = asc_signal(kat)
+ savedata[idx,1:5]=tmp.reshape([1,4])
+ print(" Saving results in file: {0}".format(filename))
+ np.savetxt(filename, savedata[0:idx+1,:], fmt=b'%.18e', delimiter=' ')
def converge_tilt(tmpkat, maxtems):
- kat = copy.deepcopy(tmpkat)
- savedata = np.zeros([len(maxtems),5])
- filename = "thermal_gravity.txt"
- for idx, tem in enumerate(maxtems):
- savedata[idx,0]=tem
- print(" Calculating maxtem = %d " % tem)
- kat.maxtem = tem
- tmp = gravity_tilt(kat)
- savedata[idx,1:5]=tmp
- print(" Saving results in file: {0}".format(filename))
- np.savetxt(filename, savedata[0:idx+1,:], fmt=b'%.18e', delimiter=' ')
+ kat = copy.deepcopy(tmpkat)
+ savedata = np.zeros([len(maxtems),5])
+ filename = "thermal_gravity.txt"
+ for idx, tem in enumerate(maxtems):
+ savedata[idx,0]=tem
+ print(" Calculating maxtem = %d " % tem)
+ kat.maxtem = tem
+ tmp = gravity_tilt(kat)
+ savedata[idx,1:5]=tmp
+ print(" Saving results in file: {0}".format(filename))
+ np.savetxt(filename, savedata[0:idx+1,:], fmt=b'%.18e', delimiter=' ')
def get_qs(tmpkat,f):
- kat = copy.deepcopy(tmpkat)
+ kat = copy.deepcopy(tmpkat)
- # measure beam parameter for the 'cold beam' i.e. the laser beam
- # matched to the cavity without any thermal lens
- nodenames=["npsl", "nITM1", "nWFS1", "nWFS2", "npo2"]
- for idx, nname in enumerate(nodenames):
- kat.parseKatCode('bp w{0} y q {1}'.format(idx, nname))
+ # measure beam parameter for the 'cold beam' i.e. the laser beam
+ # matched to the cavity without any thermal lens
+ nodenames=["npsl", "nITM1", "nWFS1", "nWFS2", "npo2"]
+ for idx, nname in enumerate(nodenames):
+ kat.parseKatCode('bp w{0} y q {1}'.format(idx, nname))
- kat.parseKatCode('yaxis re:im')
- kat.noxaxis = True
- kat.maxtem=0
+ kat.parseKatCode('yaxis re:im')
+ kat.noxaxis = True
+ kat.maxtem=0
- def beam_size(tmpkat, f, beam0):
- kat = copy.deepcopy(tmpkat)
- kat.psl.npsl.node.setGauss(kat.psl, beam0)
+ def beam_size(tmpkat, f, beam0):
+ kat = copy.deepcopy(tmpkat)
+ kat.psl.npsl.node.setGauss(kat.psl, beam0)
+
+ # add thermal lens and propagate input beam to ITM
+ kat = set_thermal_lens(kat, f)
+ out = kat.run()
+
+ # computing beam size at ITM
+ # and then we reflect of ITM, an set it as new startnode
+ q_in = out['w1']
+ from pykat.optics.ABCD import apply, mirror_refl
+ abcd = mirror_refl(1,-2500)
+ q_out = apply(abcd,q_in,1,1)
+ beam1 = beam_param(q=q_out)
+ kat.removeLine("startnode")
+ kat.psl.npsl.node.removeGauss()
+ if "ITM_TL_r" in kat._kat__components:
+ kat.ITM.nITM1r.node.setGauss(kat.ITM, beam1)
+ kat.parseKatCode("startnode nITM1r")
+ else:
+ kat.ITM.nITM1.node.setGauss(kat.ITM, beam1)
+ kat.parseKatCode("startnode nITM1")
+ out = kat.run()
+
+ # computing beam size at WFS1 and WFS2
+ q2 = out['w2']
+ beam2 = beam_param(q=q2)
+ q3 = out['w3']
+ beam3 = beam_param(q=q3)
+
+ # computing beam size at pick off
+ q4 = out['w4']
+ beam4 = beam_param(q=q4)
+ print(" Input mode beam size with thermal lens f={0}".format(f))
+ print(" - ITM w={0:.6}cm (w0={1}, z={2})".format(100.0*beam1.w,beam1.w0, beam1.z))
+ print(" - WFS1 w={0:.6}cm (w0={1}, z={2})".format(100.0*beam2.w,beam2.w0, beam2.z))
+ print(" - WFS2 w={0:.6}cm (w0={1}, z={2})".format(100.0*beam3.w,beam3.w0, beam3.z))
+ print(" - npo2 w={0:.6}cm (w0={1}, z={2})".format(100.0*beam4.w,beam4.w0, beam4.z))
+ #raw_input("Press enter to continue")
- # add thermal lens and propagate input beam to ITM
- kat = set_thermal_lens(kat, f)
- out = kat.run()
-
- # computing beam size at ITM
- # and then we reflect of ITM, an set it as new startnode
- q_in = out['w1']
- from pykat.optics.ABCD import apply, mirror_refl
- abcd = mirror_refl(1,-2500)
- q_out = apply(abcd,q_in,1,1)
- beam1 = beam_param(q=q_out)
- kat.removeLine("startnode")
- kat.psl.npsl.node.removeGauss()
- if "ITM_TL_r" in kat._kat__components:
- kat.ITM.nITM1r.node.setGauss(kat.ITM, beam1)
- kat.parseKatCode("startnode nITM1r")
- else:
- kat.ITM.nITM1.node.setGauss(kat.ITM, beam1)
- kat.parseKatCode("startnode nITM1")
- out = kat.run()
-
- # computing beam size at WFS1 and WFS2
- q2 = out['w2']
- beam2 = beam_param(q=q2)
- q3 = out['w3']
- beam3 = beam_param(q=q3)
-
- # computing beam size at pick off
- q4 = out['w4']
- beam4 = beam_param(q=q4)
- print(" Input mode beam size with thermal lens f={0}".format(f))
- print(" - ITM w={0:.6}cm (w0={1}, z={2})".format(100.0*beam1.w,beam1.w0, beam1.z))
- print(" - WFS1 w={0:.6}cm (w0={1}, z={2})".format(100.0*beam2.w,beam2.w0, beam2.z))
- print(" - WFS2 w={0:.6}cm (w0={1}, z={2})".format(100.0*beam3.w,beam3.w0, beam3.z))
- print(" - npo2 w={0:.6}cm (w0={1}, z={2})".format(100.0*beam4.w,beam4.w0, beam4.z))
- #raw_input("Press enter to continue")
-
- return [beam1, beam2, beam3, beam4]
- global out
- # run finesse with input laser mode matched to cavity (no thermal lens)
- out = kat.run()
-
- # beam at laser when matched to cold cavity
- # (note the sign flip of the real part to change direction of gauss param)
- q0 = -1.0*out['w0'].conjugate()
- beam0 = beam_param(q=q0)
- # compute beam sizes when tracing this beam back through the system
- (beam1,beam2,beam3, beam4)=beam_size(kat,f,beam0)
-
- return (beam0, beam1, beam2,beam3, beam4)
+ return [beam1, beam2, beam3, beam4]
+ global out
+ # run finesse with input laser mode matched to cavity (no thermal lens)
+ out = kat.run()
+
+ # beam at laser when matched to cold cavity
+ # (note the sign flip of the real part to change direction of gauss param)
+ q0 = -1.0*out['w0'].conjugate()
+ beam0 = beam_param(q=q0)
+ # compute beam sizes when tracing this beam back through the system
+ (beam1,beam2,beam3, beam4)=beam_size(kat,f,beam0)
+
+ return (beam0, beam1, beam2,beam3, beam4)
def asc_signal(tmpkat):
- kat = copy.deepcopy(tmpkat)
-
- code_lock = """
- set err PDrefl_p re
- lock z $err 900 1p
- put* ETM phi $z
- noplot z
- """
-
- kat.parseKatCode(code_lock)
- kat.parseKatCode('yaxis abs')
- kat.noxaxis = True
-
- signal=np.zeros((2, 2))
- kat.ITM.ybeta=1e-10
- kat.ETM.ybeta=0.0
- out = kat.run()
- signal[0,0] = out["WFS1_I"]
- signal[1,0] = out["WFS2_I"]
-
- kat.ITM.ybeta=0.0
- kat.ETM.ybeta=-1e-10
- out = kat.run()
- signal[0,1] = out["WFS1_I"]
- signal[1,1] = out["WFS2_I"]
- signal = signal *1e10
- sensors=('WFS1', 'WFS2')
- mirrors=('ITM', 'ETM')
- print(" ASC Matrix:")
- for i in range(2):
- print(" ", sensors[i], " ", end=' ')
- for j in range(2):
- print("%12.10g" % signal[i,j], end=' ')
- print(mirrors[i])
- return signal
-
+ kat = copy.deepcopy(tmpkat)
+
+ code_lock = """
+ set err PDrefl_p re
+ lock z $err 900 1p
+ put* ETM phi $z
+ noplot z
+ """
+
+ kat.parseKatCode(code_lock)
+ kat.parseKatCode('yaxis abs')
+ kat.noxaxis = True
+
+ signal=np.zeros((2, 2))
+ kat.ITM.ybeta=1e-10
+ kat.ETM.ybeta=0.0
+ out = kat.run()
+ signal[0,0] = out["WFS1_I"]
+ signal[1,0] = out["WFS2_I"]
+
+ kat.ITM.ybeta=0.0
+ kat.ETM.ybeta=-1e-10
+ out = kat.run()
+ signal[0,1] = out["WFS1_I"]
+ signal[1,1] = out["WFS2_I"]
+ signal = signal *1e10
+ sensors=('WFS1', 'WFS2')
+ mirrors=('ITM', 'ETM')
+ print(" ASC Matrix:")
+ for i in range(2):
+ print(" ", sensors[i], " ", end=' ')
+ for j in range(2):
+ print("%12.10g" % signal[i,j], end=' ')
+ print(mirrors[i])
+ return signal
+
def gravity_tilt(tmpkat):
- kat = copy.deepcopy(tmpkat)
-
- def compute_gravity_tilt(tmpkat):
- kat = copy.deepcopy(tmpkat)
- out = kat.run()
-
- y1 = out["b1"]
- y2 = out["b1_1k"]
- # shift of beam center on detector 1 (as m/w0y)
- x1 = np.sum(out.x*y1)/np.sum(y1)
- # shift of beam center on detector 2 (as m/w0y)
- x2 = np.sum(out.x*y2)/np.sum(y2)
- # calibrate this in meter by mutliplying with w0y
- # and compute the angle geometrically
- w0=out["w0y"][0]
- detector_distance = 1000.0
- tilt=w0*(x2-x1)/detector_distance
- #print " Wavefront tilt : %g nrad" % tilt
- return tilt
-
- code_WFS1 = """
- beam b1 nWFS1
- beam b1_1k nL1_in
- bp w0y y w0 nWFS1
- """
-
- code_WFS2 = """
- m md 0 1 0 nWFS2 nWFS2b
- s sd 1k nWFS2b nWFS2c
- beam b1 nWFS2*
- beam b1_1k nWFS2c
- bp w0y y w0 nWFS2
- """
-
- code_xaxis= """
- xaxis b1 y lin -40 40 800
- put b1_1k y $x1
- yaxis abs
- """
- kat.parseKatCode(code_WFS1)
- kat.parseKatCode(code_xaxis)
- kat.spo1.L=1000.0
- kat.ITM.ybeta=1e-10
- kat.ETM.ybeta=0.0
- t1=compute_gravity_tilt(kat)
- kat.ITM.ybeta=0.0
- kat.ETM.ybeta=-1e-10
- t2=compute_gravity_tilt(kat)
-
- kat = copy.deepcopy(tmpkat)
- kat.parseKatCode(code_WFS2)
- kat.parseKatCode(code_xaxis)
- kat.spo1.L=1.0e-9
- kat.ITM.ybeta=1e-10
- kat.ETM.ybeta=0.0
- t3=compute_gravity_tilt(kat)
- kat.ITM.ybeta=0.0
- kat.ETM.ybeta=-1e-10
- t4=compute_gravity_tilt(kat)
- print(" WFS1 ITM {0:.4} nrad".format(t1*1e9))
- print(" WFS2 ITM {0:.4} nrad".format(t3*1e9))
- print(" WFS1 ETM {0:.4} nrad".format(t2*1e9))
- print(" WFS2 ETM {0:.4} nrad".format(t4*1e9))
- return [t1,t2,t3,t4]
-
+ kat = copy.deepcopy(tmpkat)
+
+ def compute_gravity_tilt(tmpkat):
+ kat = copy.deepcopy(tmpkat)
+ out = kat.run()
+
+ y1 = out["b1"]
+ y2 = out["b1_1k"]
+ # shift of beam center on detector 1 (as m/w0y)
+ x1 = np.sum(out.x*y1)/np.sum(y1)
+ # shift of beam center on detector 2 (as m/w0y)
+ x2 = np.sum(out.x*y2)/np.sum(y2)
+ # calibrate this in meter by mutliplying with w0y
+ # and compute the angle geometrically
+ w0=out["w0y"][0]
+ detector_distance = 1000.0
+ tilt=w0*(x2-x1)/detector_distance
+ #print " Wavefront tilt : %g nrad" % tilt
+ return tilt
+
+ code_WFS1 = """
+ beam b1 nWFS1
+ beam b1_1k nL1_in
+ bp w0y y w0 nWFS1
+ """
+
+ code_WFS2 = """
+ m md 0 1 0 nWFS2 nWFS2b
+ s sd 1k nWFS2b nWFS2c
+ beam b1 nWFS2*
+ beam b1_1k nWFS2c
+ bp w0y y w0 nWFS2
+ """
+
+ code_xaxis= """
+ xaxis b1 y lin -40 40 800
+ put b1_1k y $x1
+ yaxis abs
+ """
+ kat.parseKatCode(code_WFS1)
+ kat.parseKatCode(code_xaxis)
+ kat.spo1.L=1000.0
+ kat.ITM.ybeta=1e-10
+ kat.ETM.ybeta=0.0
+ t1=compute_gravity_tilt(kat)
+ kat.ITM.ybeta=0.0
+ kat.ETM.ybeta=-1e-10
+ t2=compute_gravity_tilt(kat)
+
+ kat = copy.deepcopy(tmpkat)
+ kat.parseKatCode(code_WFS2)
+ kat.parseKatCode(code_xaxis)
+ kat.spo1.L=1.0e-9
+ kat.ITM.ybeta=1e-10
+ kat.ETM.ybeta=0.0
+ t3=compute_gravity_tilt(kat)
+ kat.ITM.ybeta=0.0
+ kat.ETM.ybeta=-1e-10
+ t4=compute_gravity_tilt(kat)
+ print(" WFS1 ITM {0:.4} nrad".format(t1*1e9))
+ print(" WFS2 ITM {0:.4} nrad".format(t3*1e9))
+ print(" WFS1 ETM {0:.4} nrad".format(t2*1e9))
+ print(" WFS2 ETM {0:.4} nrad".format(t4*1e9))
+ return [t1,t2,t3,t4]
+
if __name__ == '__main__':
- main()
+ main()