#export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:../lib
import pyctmm
import sys
stack = pyctmm.create_stack(3, 633e-9, 0)
pyctmm.set_ind(stack, 0, 1, 0)
pyctmm.set_ind(stack, 1, 3, -2.999)
pyctmm.set_ind(stack, 2, 1, 0)
pyctmm.set_d(stack, 0, 0)
pyctmm.set_d(stack, 1, 50e-9)
pyctmm.set_d(stack, 2, 0)
pyctmm.evaluate(stack)
power_coefs = pyctmm.get_power(stack)
print('\n', power_coefs)
model.components['qwpMes'].update() model.components['qwpCos'].retardance = 2 * np.pi / 4 model.components['qwpCos'].rotation = 0 #20*np.pi/180 model.components['qwpCos'].update() model.components['polCos'].rotation = np.pi / 4 model.components['polCos'].update() model.components['polSin'].rotation = np.pi / 4 model.components['polSin'].update() stack = pyctmm.create_stack(2, model.wavelength, 0) pyctmm.set_ind(stack, 0, 3, 0) pyctmm.set_ind(stack, 1, 1., 0) pyctmm.set_d(stack, 0, 0) pyctmm.set_d(stack, 1, 0) model.components['sMesA'].set_pyctmm(stack) nPoints = 100 xs = np.linspace(0, 1, nPoints) ints1 = np.empty(xs.shape, dtype=float) ints2 = np.empty(xs.shape, dtype=float) ints1analytic = np.empty(xs.shape, dtype=float) ints2analytic = np.empty(xs.shape, dtype=float) iAir = 1 iGlass = 3 rag = (iAir - iGlass) / (iGlass + iAir)
def test_set_pyctmm_free_space(self):
"""Test that the set_ctmm method produces the correct results.
This does not aim to fully test the pyctmm library, but tests a few
simple cases:
phase change through free space
reflection from air-glass interface
destructive interference at air-glass interface (half-wave layer)
"""
model = ts.Model()
model.wavelength = 633e-9
model.add_component(ts.components.Source, 'laser', 'n0')
model.add_component(ts.components.Dump, 'dump', 'n1')
model.add_component(ts.components.Stack, 'stack', ('n0', 'n1'))
model.add_detector('out', 'n1', ('amplitude', 'intensity'))
model.components['laser'].amplitude[0] = 0
model.components['laser'].amplitude[1] = 1
cstack = pyctmm.create_stack(1, model.wavelength, 0)
pyctmm.set_ind(cstack, 0, 1, 0)
pyctmm.set_d(cstack, 0, model.wavelength / 4)
model.components['stack'].set_pyctmm(cstack)
model.build()
model.evaluate()
self.assertAlmostEqual(model.detectors['out'].amplitudes[0], 0)
self.assertAlmostEqual(model.detectors['out'].amplitudes[1],
np.exp(-1j * np.pi / 2))
self.assertAlmostEqual(model.detectors['out'].amplitudes[2], 0)
self.assertAlmostEqual(model.detectors['out'].amplitudes[3], 0)
del cstack
del model
model = ts.Model()
model.wavelength = 633e-9
model.add_component(ts.components.Source, 'laser', 'n0')
model.add_component(ts.components.Dump, 'dump', 'n1')
model.add_component(ts.components.Stack, 'stack', ('n0', 'n1'))
model.add_detector('in', 'n0', ('amplitude', 'intensity'))
model.add_detector('out', 'n1', ('amplitude', 'intensity'))
model.components['laser'].amplitude[0] = 0
model.components['laser'].amplitude[1] = 1
cstack = pyctmm.create_stack(2, model.wavelength, 0)
pyctmm.set_ind(cstack, 0, 1, 0)
pyctmm.set_ind(cstack, 1, 1.5, 0)
pyctmm.set_d(cstack, 0, 0)
pyctmm.set_d(cstack, 1, 0)
model.components['stack'].set_pyctmm(cstack)
model.build()
model.evaluate()
self.assertAlmostEqual(model.detectors['in'].intensity - 1,
((1 - 1.5) / (1 + 1.5))**2)
del cstack
del model
model = ts.Model()
model.wavelength = 633e-9
model.add_component(ts.components.Source, 'laser', 'n0')
model.add_component(ts.components.Dump, 'dump', 'n1')
model.add_component(ts.components.Stack, 'stack', ('n0', 'n1'))
model.add_detector('in', 'n0', ('amplitude', 'intensity'))
model.add_detector('out', 'n1', ('amplitude', 'intensity'))
model.components['laser'].amplitude[0] = 0
model.components['laser'].amplitude[1] = 1
cstack = pyctmm.create_stack(3, model.wavelength, 0)
pyctmm.set_ind(cstack, 0, 1, 0)
pyctmm.set_ind(cstack, 1, 1.515, 0)
pyctmm.set_ind(cstack, 2, 1, 0)
pyctmm.set_d(cstack, 0, 0)
pyctmm.set_d(cstack, 1, (model.wavelength / 1.515) / 2)
pyctmm.set_d(cstack, 2, 0)
model.components['stack'].set_pyctmm(cstack)
model.build()
model.evaluate()
self.assertAlmostEqual(model.detectors['in'].intensity - 1, 0)
self.assertAlmostEqual(model.detectors['out'].intensity, 1)
