class TestOpticalElement(SyrisTest):
def setUp(self):
syris.init(device_index=0)
energies = range(10, 20) * q.keV
self.energy = energies[len(energies) / 2]
self.material = Material('foo', np.arange(len(energies), dtype=np.complex), energies)
def test_2d_conversion(self):
elem = DummyOpticalElement()
shape, ps = elem.transfer(1, 1 * q.m, 0 * q.keV)
self.assertEqual(len(shape), 2)
self.assertEqual(len(ps), 2)
@opencl
def test_transfer(self):
go = StaticBody(np.arange(4 ** 2).reshape(4, 4) * q.um, 1 * q.um, material=self.material)
transferred = go.transfer((4, 4), 1 * q.um, self.energy).get()
gt = transfer(go.thickness, self.material.get_refractive_index(self.energy),
energy_to_wavelength(self.energy)).get()
np.testing.assert_almost_equal(gt, transferred)
@opencl
def test_transfer_fourier(self):
elem = DummyOpticalElement()
print elem.__class__._transfer_fourier == OpticalElement._transfer_fourier
u = elem.transfer_fourier((4, 4), 1 * q.um, 10 * q.keV).get()
np.testing.assert_almost_equal(u, 0 + 0j)
class TestOpticalElement(SyrisTest):
def setUp(self):
default_syris_init()
energies = list(range(10, 20)) * q.keV
self.energy = energies[len(energies) // 2]
self.material = Material("foo", np.arange(len(energies), dtype=np.complex), energies)
def test_2d_conversion(self):
elem = DummyOpticalElement()
shape, ps = elem.transfer(1, 1 * q.m, 0 * q.keV)
self.assertEqual(len(shape), 2)
self.assertEqual(len(ps), 2)
def test_transfer(self):
go = StaticBody(np.arange(4 ** 2).reshape(4, 4) * q.um, 1 * q.um, material=self.material)
transferred = go.transfer((4, 4), 1 * q.um, self.energy).get()
gt = transfer(
go.thickness,
self.material.get_refractive_index(self.energy),
energy_to_wavelength(self.energy),
).get()
np.testing.assert_almost_equal(gt, transferred)
def test_transfer_fourier(self):
elem = DummyOpticalElement()
print(elem.__class__._transfer_fourier == OpticalElement._transfer_fourier)
u = elem.transfer_fourier((4, 4), 1 * q.um, 10 * q.keV).get()
np.testing.assert_almost_equal(u, 0 + 0j)
class TestOpticalElement(SyrisTest):
def setUp(self):
syris.init(device_index=0)
energies = range(10, 20) * q.keV
self.energy = energies[len(energies) / 2]
self.material = Material('foo',
np.arange(len(energies), dtype=np.complex),
energies)
def test_2d_conversion(self):
elem = DummyOpticalElement()
shape, ps = elem.transfer(1, 1 * q.m, 0 * q.keV)
self.assertEqual(len(shape), 2)
self.assertEqual(len(ps), 2)
@opencl
def test_transfer(self):
go = StaticBody(np.arange(4**2).reshape(4, 4) * q.um,
1 * q.um,
material=self.material)
transferred = go.transfer((4, 4), 1 * q.um, self.energy).get()
gt = transfer(go.thickness,
self.material.get_refractive_index(self.energy),
energy_to_wavelength(self.energy)).get()
np.testing.assert_almost_equal(gt, transferred)
class TestFilters(SyrisTest):
def setUp(self):
syris.init(device_index=0)
self.energies = np.arange(10, 20) * q.keV
self.energy = 15 * q.keV
delta = np.linspace(1e-5, 1e-6, len(self.energies))
beta = np.linspace(1e-8, 1e-9, len(self.energies))
self.material = Material('foo', delta + beta * 1j, self.energies)
self.thickness = 1 * q.mm
self.fltr = MaterialFilter(self.thickness, self.material)
def test_transfer(self):
thickness = self.thickness.simplified.magnitude
lam = energy_to_wavelength(self.energy).simplified.magnitude
coeff = -2 * np.pi * thickness / lam
ri = self.material.get_refractive_index(self.energy)
gt = np.exp(coeff * (ri.imag + ri.real * 1j))
fltr = self.fltr.transfer(None, None, self.energy)
self.assertAlmostEqual(gt, fltr)
def test_get_attenuation(self):
att = self.material.get_attenuation_coefficient(self.energy).simplified.magnitude
thickness = self.thickness.simplified.magnitude
gt = att * thickness
fltr = self.fltr.get_attenuation(self.energy)
self.assertAlmostEqual(gt, fltr)
def test_exponent(self):
transmitted = self.fltr.transfer(None, None, self.energy)
exponent = self.fltr.transfer(None, None, self.energy, exponent=True)
self.assertAlmostEqual(transmitted, np.exp(exponent), places=5)
def test_gaussian(self):
energies = np.arange(5, 30, 0.1) * q.keV
energy = 15 * q.keV
sigma = fwnm_to_sigma(1, n=2) * q.keV
energy_10 = energy - sigma_to_fwnm(sigma.magnitude, n=10) / 2 * q.keV
fltr = GaussianFilter(energies, energy, sigma, peak_transmission=0.5)
u_0 = 10 + 0j
u_f = fltr.transfer(None, None, energy)
u = u_0 * u_f
self.assertAlmostEqual(np.abs(u) ** 2, 50, places=4)
u_f = fltr.transfer(None, None, energy_10)
u = u_0 * u_f
self.assertAlmostEqual(np.abs(u) ** 2, 5, places=4)
# Test exponent
u_f = fltr.transfer(None, None, energy, exponent=True)
u = u_0 * np.exp(u_f)
self.assertAlmostEqual(np.abs(u) ** 2, 50, places=4)
u_f = fltr.transfer(None, None, energy_10, exponent=True)
u = u_0 * np.exp(u_f)
self.assertAlmostEqual(np.abs(u) ** 2, 5, places=4)
def test_scintillator_conservation(self):
"""Test if the integral of luminescence is really 1."""
wavelengths = np.linspace(100, 700, 512) * q.nm
wavelengths_dense = np.linspace(wavelengths[0].magnitude,
wavelengths[-1].magnitude, 4 * len(wavelengths)) * q.nm
d_wavelength_dense = wavelengths_dense[1] - wavelengths_dense[0]
sigma = 20. * q.nm
luminescence = np.exp(-(wavelengths - 400 * q.nm) ** 2 / (2 * sigma ** 2)) / \
(sigma * np.sqrt(2 * np.pi))
sc = Scintillator(1 * q.m, None, np.ones(30) / q.keV, np.arange(30) * q.keV, luminescence,
wavelengths, 1)
lum_orig = sc.get_luminescence(wavelengths)
self.assertAlmostEqual((np.sum(lum_orig) * sc.d_wavelength).simplified.magnitude, 1)
lum = sc.get_luminescence(wavelengths_dense)
self.assertAlmostEqual((np.sum(lum) * d_wavelength_dense).simplified.magnitude, 1)
def test_interpolation(self):
mat = Material('foo', self.refractive_indices, self.energies)
index = mat.get_refractive_index(2400 * q.eV)
self.assertAlmostEqual(2.4, index.real, places=5)
self.assertAlmostEqual(2.4, index.imag, places=5)
class TestFilters(SyrisTest):
def setUp(self):
syris.init(device_index=0)
self.energies = np.arange(10, 20) * q.keV
self.energy = 15 * q.keV
delta = np.linspace(1e-5, 1e-6, len(self.energies))
beta = np.linspace(1e-8, 1e-9, len(self.energies))
self.material = Material('foo', delta + beta * 1j, self.energies)
self.thickness = 1 * q.mm
self.fltr = MaterialFilter(self.thickness, self.material)
def test_transfer(self):
thickness = self.thickness.simplified.magnitude
lam = energy_to_wavelength(self.energy).simplified.magnitude
coeff = -2 * np.pi * thickness / lam
ri = self.material.get_refractive_index(self.energy)
gt = np.exp(coeff * (ri.imag + ri.real * 1j))
fltr = self.fltr.transfer(None, None, self.energy)
self.assertAlmostEqual(gt, fltr)
def test_get_attenuation(self):
att = self.material.get_attenuation_coefficient(
self.energy).simplified.magnitude
thickness = self.thickness.simplified.magnitude
gt = att * thickness
fltr = self.fltr.get_attenuation(self.energy)
self.assertAlmostEqual(gt, fltr)
def test_exponent(self):
transmitted = self.fltr.transfer(None, None, self.energy)
exponent = self.fltr.transfer(None, None, self.energy, exponent=True)
self.assertAlmostEqual(transmitted, np.exp(exponent), places=5)
def test_gaussian(self):
energies = np.arange(5, 30, 0.1) * q.keV
energy = 15 * q.keV
sigma = fwnm_to_sigma(1, n=2) * q.keV
energy_10 = energy - sigma_to_fwnm(sigma.magnitude, n=10) / 2 * q.keV
fltr = GaussianFilter(energies, energy, sigma, peak_transmission=0.5)
u_0 = 10 + 0j
u_f = fltr.transfer(None, None, energy)
u = u_0 * u_f
self.assertAlmostEqual(np.abs(u)**2, 50, places=4)
u_f = fltr.transfer(None, None, energy_10)
u = u_0 * u_f
self.assertAlmostEqual(np.abs(u)**2, 5, places=4)
# Test exponent
u_f = fltr.transfer(None, None, energy, exponent=True)
u = u_0 * np.exp(u_f)
self.assertAlmostEqual(np.abs(u)**2, 50, places=4)
u_f = fltr.transfer(None, None, energy_10, exponent=True)
u = u_0 * np.exp(u_f)
self.assertAlmostEqual(np.abs(u)**2, 5, places=4)
def test_scintillator_conservation(self):
"""Test if the integral of luminescence is really 1."""
wavelengths = np.linspace(100, 700, 512) * q.nm
wavelengths_dense = np.linspace(wavelengths[0].magnitude,
wavelengths[-1].magnitude,
4 * len(wavelengths)) * q.nm
d_wavelength_dense = wavelengths_dense[1] - wavelengths_dense[0]
sigma = 20. * q.nm
luminescence = np.exp(-(wavelengths - 400 * q.nm) ** 2 / (2 * sigma ** 2)) / \
(sigma * np.sqrt(2 * np.pi))
sc = Scintillator(1 * q.m, None,
np.ones(30) / q.keV,
np.arange(30) * q.keV, luminescence, wavelengths, 1)
lum_orig = sc.get_luminescence(wavelengths)
self.assertAlmostEqual(
(np.sum(lum_orig) * sc.d_wavelength).simplified.magnitude, 1)
lum = sc.get_luminescence(wavelengths_dense)
self.assertAlmostEqual(
(np.sum(lum) * d_wavelength_dense).simplified.magnitude, 1)
def test_interpolation(self):
mat = Material('foo', self.refractive_indices, self.energies)
index = mat.get_refractive_index(2400 * q.eV)
self.assertAlmostEqual(2.4, index.real, places=5)
self.assertAlmostEqual(2.4, index.imag, places=5)