def show_table():
inch = 0.0254
lambda0 = 1.054e-6
spitfire_exit = Numeric.array((2 * inch, 5 * inch, -0.65))
compressor_entrance = Numeric.array((45.5 * inch, 2.5 * inch, 46.5 * inch))
optics = {}
BLUELINE = 'bl'
SPLIT = 'split'
START = 'start'
COMP = 'compressor'
QUAD = 'quadrupler'
IR = 'ir'
IR_COMP = 'ir compressor'
BE_TURN = 'be_turn'
IZ_SCREEN = 'iz'
optics[START] = null_optic("start", spitfire_exit)
optics[BLUELINE] = blueline = blue_through_ylf()
optics[SPLIT] = split1 = reflector("splitter1",
center=(2 * inch, 5 * inch, 3 * inch))
optics[IR] = irline = ir_line()
comp = optics[COMP] = blue_compressor(38.5,
1.28,
lambda0,
center=compressor_entrance,
angle=270)
mainbeam = beam(spitfire_exit, qtens(lambda0, spitfire.q0), lambda0)
# print mainbeam.q
split1.set_direction(spitfire_exit, blueline)
blueline.set_entrance_direction(split1)
blueline.set_exit_direction(comp)
comp.rotate_to_axis(blueline)
q = optics[QUAD] = quadrupler(center=(40 * inch, 2.5 * inch, 34.5 * inch),
from_optic=comp[blue_compressor.TM1])
comp.set_exit_direction(q)
q.set_entrance_direction(comp)
blue_sys = composite_optic("blue_sys", optics,
[START, SPLIT, BLUELINE, COMP, QUAD], (0, 0, 0),
(0, 0, 0), 0)
irline.set_entrance_direction(spitfire_exit)
# align compressor table to output of main IR line
ir_sys = composite_optic("ir_sys", optics, [START, IR], (0, 0, 0),
(0, 0, 0), 0).clone()
pointer = trace_path(ir_sys, mainbeam.clone())[-1]
pointer.free_drift(1.0)
optics[IR_COMP] = ircomp = ir_compressor(center=pointer.x0,
angle=math.atan(0.2) / deg)
ircomp.set_entrance_direction(irline)
# now, align IZ
ir_sys = composite_optic("ir_sys", optics, [START, IR, IR_COMP], (0, 0, 0),
(0, 0, 0), 0).clone()
pointer = trace_path(ir_sys, mainbeam.clone())[-1]
pointer.free_drift(1.0)
bemir = optics[BE_TURN] = reflector("be turn mirror",
center=pointer.x0,
angle=45.0)
bemir.transport_to_here(pointer).transform(pointer)
pointer.free_drift(0.1)
dz = pointer.q.next_waist()
pointer.free_drift(dz)
optics[IZ_SCREEN] = null_optic("IZ", pointer.x0)
ir_sys = composite_optic("ir_sys", optics,
[START, IR, IR_COMP, BE_TURN, IZ_SCREEN],
(0, 0, 0), (0, 0, 0), 0)
whole_table = composite_optic(
"whole table", optics,
[START, SPLIT, BLUELINE, COMP, QUAD, IR, IR_COMP, BE_TURN, IZ_SCREEN],
(0, 0, 0), (0, 0, 0), 0)
ir_trace = trace_path(ir_sys, mainbeam.clone())
ir_trace.color = graphite.red
blue_trace = trace_path(blue_sys, mainbeam.clone())
blue_trace.color = graphite.blue
if table_layout:
g = draw_everything({"sys": whole_table},
ir_trace, (-.25, 5), (0, 2.5),
three_d=0)
draw_trace(g, blue_trace)
#g=draw_everything({"sys":sys}, trace, (0.5,1.1), (0.8,1.1), three_d=0)
#g=draw_everything({"sys":sys}, trace, (0.7,1.3), (0.9,1.2), three_d=0)
if show_qd:
graphite.genOutput(g,
'QD',
canvasname="Compressor layout",
size=(900, 500))
if show_pdf:
graphite.genOutput(g,
'PDF',
canvasname="Tablelayout",
size=(900, 500))
# print trace[0]
plotq(blue_trace)
if 0:
glabel = ircomp.mark_label(ircomp.INPUT)
m = ir_trace[(glabel, 0)]
print("\n\n**start**\n\n")
print(m.incoming_direction, m.direction())
print(m.incoming_q)
print(m.incoming_q.qi_moments()[1:3])
print(m.incoming_q.q_moments()[1:3])
print(m.incoming_q.qit, "\n")
print(m.localize_transform_tensor)
print(m.footprint_q)
print(m.footprint_q.qi_moments()[1:3])
print(m.footprint_q.q_moments()[1:3])
print(m.footprint_q.qit, "\n")
print(m.globalize_transform_tensor)
print(m.q)
print(m.q.qi_moments()[1:3])
print(m.q.q_moments()[1:3])
print(m.q.qit, "\n")
glabel = ircomp.mark_label(ircomp.INPUT)
# endpoint=ir_trace[(glabel,0)] #get first hit on optic <glabel>
endpoint = ir_trace[-1]
q = endpoint.q
t, qx0, qy0 = q.qi_moments()
theta = math.atan2(t[0, 1].real, t[0, 0].real) / deg
qxx, qyy = endpoint.transform_q_to_table(q)
dzx, dzy = (1e6 / qxx).real, (1e6 / qyy).real
print((
"qxx = %.1f, dzx = %.0f, qyy=%.1f, dzy=%.0f, (in um), theta=%.1f deg" %
(q.rw(qxx)[1] * 1e6, dzx, q.rw(qyy)[1] * 1e6, dzy, theta)))
def show_table():
inch=0.0254
lambda0=1.054e-6
spitfire_exit=Numeric.array((2*inch, 5*inch, -0.65))
compressor_entrance=Numeric.array((45.5*inch, 2.5*inch, 46.5*inch))
optics={}
BLUELINE='bl'
SPLIT='split'
START='start'
COMP='compressor'
QUAD='quadrupler'
IR='ir'
IR_COMP='ir compressor'
BE_TURN='be_turn'
IZ_SCREEN='iz'
optics[START]=null_optic("start", spitfire_exit)
optics[BLUELINE]=blueline=blue_through_ylf()
optics[SPLIT]=split1=reflector("splitter1", center=(2*inch, 5*inch, 3*inch))
optics[IR]=irline=ir_line()
comp=optics[COMP]=blue_compressor(38.5, 1.28, lambda0, center=compressor_entrance, angle=270)
mainbeam=beam( spitfire_exit, qtens(lambda0, spitfire.q0), lambda0)
#print mainbeam.q
split1.set_direction(spitfire_exit, blueline)
blueline.set_entrance_direction(split1)
blueline.set_exit_direction(comp)
comp.rotate_to_axis(blueline)
q=optics[QUAD]=quadrupler(center=(40*inch, 2.5*inch, 34.5*inch), from_optic=comp[blue_compressor.TM1])
comp.set_exit_direction(q)
q.set_entrance_direction(comp)
blue_sys=composite_optic("blue_sys", optics, [START, SPLIT, BLUELINE, COMP, QUAD], (0,0,0), (0,0,0), 0)
irline.set_entrance_direction(spitfire_exit)
#align compressor table to output of main IR line
ir_sys=composite_optic("ir_sys", optics, [START, IR], (0,0,0), (0,0,0), 0).clone()
pointer=trace_path(ir_sys, mainbeam.clone())[-1]
pointer.free_drift(1.0)
optics[IR_COMP]=ircomp=ir_compressor(center=pointer.x0, angle=math.atan(0.2)/deg)
ircomp.set_entrance_direction(irline)
#now, align IZ
ir_sys=composite_optic("ir_sys", optics, [START, IR, IR_COMP], (0,0,0), (0,0,0), 0).clone()
pointer=trace_path(ir_sys, mainbeam.clone())[-1]
pointer.free_drift(1.0)
bemir=optics[BE_TURN]=reflector("be turn mirror", center=pointer.x0, angle=45.0)
bemir.transport_to_here(pointer).transform(pointer)
pointer.free_drift(0.1)
dz=pointer.q.next_waist()
pointer.free_drift(dz)
optics[IZ_SCREEN]=null_optic("IZ", pointer.x0)
ir_sys=composite_optic("ir_sys", optics, [START, IR, IR_COMP, BE_TURN, IZ_SCREEN], (0,0,0), (0,0,0), 0)
whole_table=composite_optic("whole table", optics, [START, SPLIT, BLUELINE, COMP, QUAD, IR, IR_COMP, BE_TURN, IZ_SCREEN], (0,0,0), (0,0,0), 0)
ir_trace=trace_path(ir_sys, mainbeam.clone())
ir_trace.color=graphite.red
blue_trace=trace_path(blue_sys, mainbeam.clone())
blue_trace.color=graphite.blue
if table_layout:
g=draw_everything({"sys":whole_table}, ir_trace, (-.25,5), (0,2.5), three_d=0)
draw_trace(g, blue_trace)
#g=draw_everything({"sys":sys}, trace, (0.5,1.1), (0.8,1.1), three_d=0)
#g=draw_everything({"sys":sys}, trace, (0.7,1.3), (0.9,1.2), three_d=0)
if show_qd:
graphite.genOutput(g,'QD',canvasname="Compressor layout", size=(900,500))
if show_pdf:
graphite.genOutput(g,'PDF',canvasname="Tablelayout", size=(900,500))
#print trace[0]
plotq(blue_trace)
if 0:
glabel=ircomp.mark_label(ircomp.INPUT)
m=ir_trace[(glabel,0)]
print "\n\n**start**\n\n"
print m.incoming_direction, m.direction()
print m.incoming_q
print m.incoming_q.qi_moments()[1:3]
print m.incoming_q.q_moments()[1:3]
print m.incoming_q.qit, "\n"
print m.localize_transform_tensor
print m.footprint_q
print m.footprint_q.qi_moments()[1:3]
print m.footprint_q.q_moments()[1:3]
print m.footprint_q.qit, "\n"
print m.globalize_transform_tensor
print m.q
print m.q.qi_moments()[1:3]
print m.q.q_moments()[1:3]
print m.q.qit, "\n"
glabel=ircomp.mark_label(ircomp.INPUT)
#endpoint=ir_trace[(glabel,0)] #get first hit on optic <glabel>
endpoint=ir_trace[-1]
q=endpoint.q
t, qx0, qy0=q.qi_moments()
theta=math.atan2(t[0,1].real, t[0,0].real)/deg
qxx, qyy = endpoint.transform_q_to_table(q)
dzx, dzy = (1e6/qxx).real, (1e6/qyy).real
print ("qxx = %.1f, dzx = %.0f, qyy=%.1f, dzy=%.0f, (in um), theta=%.1f deg" %
(q.rw(qxx)[1]*1e6, dzx, q.rw(qyy)[1]*1e6, dzy, theta ) )
def doit_bc(theta1, clen, lambda0, drawit=0):
print("\n\n***start blue line trace***\n")
grating_offset = -0.075
exit_height = 0
exit_z = -0.3
eta1 = 0.0
optics = {}
END = 'end'
optics[END] = null_optic("end", (0.15, exit_height, 0), 0)
START = 'start'
optics[START] = null_optic("start", (0, 0, -0.85))
tracebeam = basebeam.clone()
system_center = Numeric.array((-0.1, 0, -0.85))
basebeam = beam(system_center, qtens(lambda0, q=spitfire.q0), lambda0)
tracebeam.free_drift(-0.2)
COMP = 'comp'
comp = optics[COMP] = blue_compressor(theta1,
clen,
lambda0,
center=system_center,
angle=rotation)
TM2 = 'turn2'
optics[TM2] = reflector("output turn 2",
angle=0,
center=(0.15, exit_height, -0.6),
width=.025)
optics[COMP].set_exit_direction(optics[TM2])
optics[TM2].set_direction(optics[COMP], optics[END])
sys_order = (START, COMP, TM2, DEMAG1, DEMAG2, END)
GRATE = comp.mark_label(blue_compressor.GRATE)
IR1 = comp.mark_label(blue_compressor.IR1)
IR2 = comp.mark_label(blue_compressor.IR2)
optic_sys = composite_optic("system",
optics,
sys_order,
center=optics[START].center)
trace0 = trace_path(optic_sys, tracebeam.shift_lambda(0))
trace0.color = graphite.green
trace1 = trace_path(optic_sys, tracebeam.shift_lambda(-0.5e-9))
trace1.color = graphite.blue
trace2 = trace_path(optic_sys, tracebeam.shift_lambda(+0.5e-9))
trace2.color = graphite.red
try: # if one of the traces failed, this may not work
print("Theta1=%.3f eta=%.3f len=%.3f lambda=%.4f" %
(theta1, eta1, clen, lambda0 * 1e6))
d0 = trace0[-1]['total_drift']
d1 = trace1[-1]['total_drift']
d2 = trace2[-1]['total_drift']
print((
"d(lambda0-0.5nm)=%.3f m, d(lambda0+0.5nm)=%.3f m, dt/dl=%.0f ps/nm, "
% (d1, d2, (d2 - d1) * 1e12 / clight)),
end=' ')
print("d2t/dl2 = %.2f ps/nm^2" % ((d2 + d1 - 2.0 * d0) *
(1e12 / clight) * 4))
spot1info = trace0[(GRATE, 0)]
spot2info = trace0[(GRATE, 1)]
ir1info = trace0[(IR1, 0)]
ir2info = trace0[(IR2, 0)]
spot_offset = spot2info['position'] - spot1info['position']
retro_offset = ir2info['position'] - ir1info['position']
print("Grating offset = %.3f, retro_offset=%.3f" %
(vec_mag(spot_offset), vec_mag(retro_offset)))
print("measured vertex length = %.3f" %
vec_mag(comp[GRATE].center - comp[IR1].center))
print("dispersion offset on grating = %.3f m / nm" %
vec_mag(trace2[(GRATE, 1)]['position'] -
trace1[(GRATE, 1)]['position']))
print("final q = ", trace0[-1]['q'])
except:
traceback.print_exc()
pass
if drawit:
g = draw_everything({"sys": optic_sys},
trace0, (-.9, .1), (-.5, .5),
three_d=0)
#g=draw_everything(optics, trace0, (-2, 2), (-2, 2))
draw_trace(g, trace1)
draw_trace(g, trace2)
graphite.genOutput(g,
'QD',
canvasname="Compressor layout",
size=(600, 500))
def doit_bc(theta1, clen, lambda0, drawit=0):
print "\n\n***start blue line trace***\n"
grating_offset=-0.075
exit_height=0
exit_z=-0.3
eta1=0.0
optics={}
END='end'
optics[END]=null_optic("end", (0.15,exit_height,0), 0)
START='start'
optics[START]=null_optic("start", (0,0,-0.85))
tracebeam=basebeam.clone()
system_center=Numeric.array((-0.1, 0, -0.85))
basebeam=beam(system_center, qtens(lambda0, q=spitfire.q0), lambda0)
tracebeam.free_drift(-0.2)
COMP='comp'
comp=optics[COMP]=blue_compressor(theta1, clen, lambda0, center=system_center, angle=rotation)
TM2='turn2'
optics[TM2]=reflector("output turn 2", angle=0, center=(0.15,exit_height,-0.6), width=.025)
optics[COMP].set_exit_direction(optics[TM2])
optics[TM2].set_direction(optics[COMP], optics[END])
sys_order=(START, COMP, TM2, DEMAG1, DEMAG2, END)
GRATE=comp.mark_label(blue_compressor.GRATE)
IR1=comp.mark_label(blue_compressor.IR1)
IR2=comp.mark_label(blue_compressor.IR2)
optic_sys=composite_optic("system", optics, sys_order, center=optics[START].center)
trace0=trace_path(optic_sys, tracebeam.shift_lambda(0))
trace0.color=graphite.green
trace1=trace_path(optic_sys, tracebeam.shift_lambda(-0.5e-9))
trace1.color=graphite.blue
trace2=trace_path(optic_sys, tracebeam.shift_lambda(+0.5e-9))
trace2.color=graphite.red
try: #if one of the traces failed, this may not work
print "Theta1=%.3f eta=%.3f len=%.3f lambda=%.4f" % (theta1, eta1, clen, lambda0*1e6)
d0=trace0[-1]['total_drift']
d1=trace1[-1]['total_drift']
d2=trace2[-1]['total_drift']
print ("d(lambda0-0.5nm)=%.3f m, d(lambda0+0.5nm)=%.3f m, dt/dl=%.0f ps/nm, " % (d1, d2, (d2-d1)*1e12/clight)),
print "d2t/dl2 = %.2f ps/nm^2" % ((d2+d1-2.0*d0)*(1e12/clight)*4)
spot1info=trace0[(GRATE,0)]
spot2info=trace0[(GRATE,1)]
ir1info=trace0[(IR1,0)]
ir2info=trace0[(IR2,0)]
spot_offset=spot2info['position']-spot1info['position']
retro_offset=ir2info['position']-ir1info['position']
print "Grating offset = %.3f, retro_offset=%.3f" % (vec_mag(spot_offset), vec_mag(retro_offset))
print "measured vertex length = %.3f" % vec_mag(comp[GRATE].center-comp[IR1].center)
print "dispersion offset on grating = %.3f m / nm" % vec_mag(trace2[(GRATE,1)]['position']-trace1[(GRATE,1)]['position'])
print "final q = ", trace0[-1]['q']
except:
traceback.print_exc()
pass
if drawit:
g=draw_everything({"sys":optic_sys}, trace0, (-.9, .1), (-.5, .5), three_d=0)
#g=draw_everything(optics, trace0, (-2, 2), (-2, 2))
draw_trace(g, trace1)
draw_trace(g, trace2)
graphite.genOutput(g,'QD',canvasname="Compressor layout", size=(600,500))