AnisoCADO 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

anisocado/tests/test_playing_around.py

@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*-
+"""Various tests to see if things run, no assertions."""
+
+import numpy as np
+
+from matplotlib import pyplot as plt
+
+from anisocado import AnalyticalScaoPsf
+from anisocado.misc import on_axis_strehl_for_kernel_size
+
+PLOTS = False
+
+
+class TestMakeStrehlMap:
+    def test_make_a_strehl_map_over_a_grid_on_the_focal_plane(self):
+        psf = AnalyticalScaoPsf(N=128, wavelength=2.15)
+        x, y = np.mgrid[-50:51:20, -50:51:20]
+
+        sr = np.zeros(x.shape)
+        for i in range(len(x)):
+            for j in range(len(y)):
+                psf.shift_off_axis(x[i, j], y[i, j])
+                sr[i, j] = psf.strehl_ratio
+        print(np.max(sr))
+
+        if PLOTS:
+            plt.imshow(sr)
+            plt.contourf(x, y, sr, np.arange(0, 1, 0.05))
+            plt.colorbar()
+            plt.show()
+
+
+class TestStrehlRatiosForDifferentSizedKernels:
+    def test_how_much_does_the_SR_change_for_kernel_size_increase(self):
+        side_length = [64, 128, 256, 512, 1024]
+        sr = on_axis_strehl_for_kernel_size(side_length, wavelength=2.15)
+        if PLOTS:
+            plt.plot(side_length, sr)
+            plt.show()
+
+
+class TestPsfsAlongXaxis:
+    def test_strehl_ratio_along_x_axis(self):
+        for wave in [0.8, 0.9, 1.1, 1.6, 2.15]:
+            psf = AnalyticalScaoPsf(N=128, wavelength=wave)
+            x, y = np.mgrid[0:41:1, 0:1:1]
+
+            sr = np.zeros(x.shape)
+            for i in range(x.shape[0]):
+                for j in range(x.shape[1]):
+                    psf.shift_off_axis(x[i, j], y[i, j])
+                    sr[i, j] = psf.strehl_ratio
+
+            if PLOTS:
+                print(wave, np.sum(psf._kernel_sum))
+                plt.plot(x.flatten(), sr, label=f"{wave} um")
+
+        if PLOTS:
+            plt.legend()
+            plt.xlabel("Distance from NGS [arcsec]")
+            plt.ylabel("Strehl Ratio")
+            plt.show()

anisocado/tests/test_psf_functions.py

@@ -0,0 +1,100 @@
+# -*- coding: utf-8 -*-
+"""Unit tests for misc functions."""
+
+import pytest
+
+import numpy as np
+import matplotlib.pyplot as plt
+from astropy.io import fits
+
+from anisocado import misc
+
+
+@pytest.fixture(scope="class")
+def basic_fv_psf():
+    n = 30
+    coords = [(-n, n),  (0, n),  (n, n),
+              (-n, 0),  (0, 0),  (n, 0),
+              (-n, -n), (0, -n), (n, -n)]
+    waves = [0.9, 1.1, 1.6, 2.15]
+    hdu = misc.make_simcado_psf_file(coords=coords, wavelengths=waves, N=128)
+    return hdu
+
+
+class TestSimcadoPsfFile:
+    def test_throws_error_with_no_input(self):
+        with pytest.raises(TypeError):
+            misc.make_simcado_psf_file()
+
+    def test_returns_hdulist_for_basic_input(self):
+        hdu = misc.make_simcado_psf_file(
+            coords=[(0, 0)],
+            wavelengths=[2.15],
+            N=128,
+        )
+        assert isinstance(hdu, fits.HDUList)
+        assert isinstance(hdu[1], fits.BinTableHDU)
+        assert len(hdu) == 3
+        assert hdu[2].data.shape == (1, 128, 128)
+
+    def test_returns_full_hdulist_for_full_input(self):
+        n = 30
+        coords = [(0, 0), (-n, -n), (n, -n), (n, n), (-n, n)]
+        waves = [0.9, 1.1, 1.6, 2.15]
+        hdu = misc.make_simcado_psf_file(coords=coords, wavelengths=waves,
+                                         N=128)
+        assert len(hdu) == len(waves) + 2
+        assert hdu[2].data.shape == (len(coords), 128, 128)
+        assert hdu[2].header["CDELT1"] == 0.004 / 3600.
+
+
+@pytest.mark.usefixtures("basic_fv_psf")
+class TestFvpsfFileConsistencyChecks:
+    def test_has_correct_keywords_in_ext0(self, basic_fv_psf):
+        for key in ["AUTHOR", "DATE_CRE", "DATE_MOD", "SOURCE", "STATUS",
+                    "ETYPE", "ECAT", "EDATA"]:
+            assert key in basic_fv_psf[0].header
+
+    def test_has_catalogue_psf(self, basic_fv_psf):
+        ecat = basic_fv_psf[0].header["ECAT"]
+        cat_hdu = basic_fv_psf[ecat]
+        for key in ["NUMPSFS", "CATTYPE", "CUNIT1"]:
+            assert key in cat_hdu.header
+
+        if cat_hdu.header["CATTYPE"] == "table":
+            assert isinstance(cat_hdu.data, (fits.BinTableHDU, fits.FITS_rec))
+
+        elif basic_fv_psf[ecat].header["CATTYPE"] == "image":
+            assert isinstance(cat_hdu.data, fits.ImageHDU)
+            for key in ["CRVAL1", "CRPIX1", "CDELT1"]:
+                assert key in cat_hdu.header
+
+    def test_has_the_right_number_of_layers_per_psf_hdu(self, basic_fv_psf):
+        ecat = basic_fv_psf[0].header["ECAT"]
+        cat_hdu = basic_fv_psf[ecat]
+        n_layers = len(set(cat_hdu.data["layer"]))
+
+        edata = basic_fv_psf[0].header["EDATA"]
+        data_indexes = range(edata, len(basic_fv_psf))
+        for ii in data_indexes:
+            assert basic_fv_psf[ii].data.shape[0] == n_layers
+            for key in ["CRVAL1", "CRPIX1", "CDELT1", "WAVE0"]:
+                assert key in basic_fv_psf[ii].header
+
+    def plot_psfs(self, basic_fv_psf):
+        for psf_hdu in basic_fv_psf[2:]:
+            for i in range(len(psf_hdu.data)):
+                plt.subplot(3, 3, i + 1)
+                plt.imshow(psf_hdu.data[i, :, :].T, origin="l", norm="log")
+
+            plt.show()
+
+    def print_coords(self):
+        # coords = psf.field_positions_for_simcado_psf()
+        # waves = [0.8, 0.9, 1.05, 1.2, 1.5, 1.7, 2.0, 2.2]
+        # hdu = psf.make_simcado_psf_file()
+
+        map = misc.make_strehl_map_from_table([(0, 0)])
+        plt.imshow(map)
+        plt.show()
+        print(np.unique(map))

anisocado/tests/test_scao_psf.py

@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+"""Unit tests for AnalyticalScaoPsf."""
+
+import pytest
+
+import numpy as np
+from matplotlib import pyplot as plt
+from astropy.io import fits
+
+from anisocado.psf import AnalyticalScaoPsf
+
+
+PLOTS = False
+
+
+class TestInit:
+    def test_initialises_with_nothing(self):
+        assert isinstance(AnalyticalScaoPsf(), AnalyticalScaoPsf)
+
+    def test_initialises_with_bunch_of_keywords(self):
+        psf = AnalyticalScaoPsf(N=256, wavelengthIR=1.2e-6)
+        assert isinstance(psf, AnalyticalScaoPsf)
+
+    def test_throws_warning_if_keyword_not_spelt_correctly(self):
+        psf = AnalyticalScaoPsf(N=256, wavelengthIR=1.2e-6, hello="world")
+        assert isinstance(psf, AnalyticalScaoPsf)
+
+    def test_on_axis_psf_is_made(self):
+        psf = AnalyticalScaoPsf(N=512)
+        assert isinstance(psf.psf_on_axis, np.ndarray)
+
+        if PLOTS:
+            plt.imshow(psf.psf_on_axis.T,
+                       origin="lower", norm="log", vmin=1e-7)
+            plt.show()
+
+    def test_kernel_sums_to_one(self):
+        psf = AnalyticalScaoPsf(N=256)
+        assert psf.psf_latest.sum() == pytest.approx(1)
+
+
+class TestShiftPSF:
+    def test_psf_blurs_when_shifted(self):
+        psf = AnalyticalScaoPsf(N=1024)
+        sr_orig = psf.strehl_ratio
+
+        psf.shift_off_axis(0, 10)
+        sr_last = psf.strehl_ratio
+
+        assert sr_last < sr_orig
+
+        if PLOTS:
+            plt.subplot(121)
+            plt.imshow(psf.psf_on_axis.T,
+                       origin="lower", norm="log", vmin=3e-6)
+            plt.colorbar()
+
+            plt.subplot(122)
+            plt.imshow(psf.psf_latest.T,
+                       origin="lower", norm="log", vmin=3e-6)
+            plt.colorbar()
+            plt.show()
+
+    def test_kernel_sums_to_one(self):
+        psf = AnalyticalScaoPsf(N=512)
+        psf.shift_off_axis(0, 0)
+        assert psf.psf_latest.sum() == pytest.approx(1)
+
+
+class TestHDUProperty:
+    def test_returns_fits_imagehdu(self):
+        psf = AnalyticalScaoPsf(N=512)
+        psf.shift_off_axis(10, 10)
+        hdu = psf.hdu
+        assert isinstance(hdu, fits.ImageHDU)
+        assert hdu.header["CDELT1"] == 4 / (3600 * 1000.)
+        assert hdu.header["CRVAL1"] == 10 / 3600.
+
+        if PLOTS:
+            plt.imshow(hdu.data.T, origin="l", norm="log")
+            plt.show()
+
+
+class TestRNG:
+    def test_predictable(self):
+        psf_a = AnalyticalScaoPsf(N=512, seed=9999999)
+        psf_b = AnalyticalScaoPsf(N=512, seed=9999999)
+
+        np.testing.assert_array_equal(psf_a.psf_on_axis, psf_b.psf_on_axis)

anisocado-0.4.0.dist-info/METADATA

@@ -0,0 +1,53 @@
+Metadata-Version: 2.3
+Name: AnisoCADO
+Version: 0.4.0
+Summary: Generate off-axis SCAO PSFs for the ELT
+License: GPL-3.0-or-later
+Keywords: astronomy
+Author: Kieran Leschinski
+Author-email: kieran.leschinski@unive.ac.at
+Maintainer: Kieran Leschinski
+Maintainer-email: kieran.leschinski@unive.ac.at
+Requires-Python: >=3.10,<3.14
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Astronomy
+Requires-Dist: astropy (>=6.1.7,<8.0.0)
+Requires-Dist: matplotlib (>=3.10.1,<4.0.0)
+Requires-Dist: numpy (>=1.26.4,<2.3.0) ; python_version >= "3.10" and python_version < "3.13"
+Requires-Dist: numpy (>=2.2.6,<3.0.0) ; python_version >= "3.13"
+Project-URL: Bug Tracker, https://github.com/AstarVienna/AnisoCADO/issues
+Project-URL: Changelog, https://github.com/AstarVienna/AnisoCADO/releases
+Project-URL: Documentation, https://anisocado.readthedocs.io/en/latest/
+Project-URL: Repository, https://github.com/AstarVienna/AnisoCADO/
+Description-Content-Type: text/markdown
+
+# AnisoCADO
+
+[![Tests](https://github.com/AstarVienna/AnisoCADO/actions/workflows/tests.yml/badge.svg)](https://github.com/AstarVienna/AnisoCADO/actions/workflows/tests.yml)
+[![Documentation Status](https://readthedocs.org/projects/anisocado/badge/?version=latest)](https://anisocado.readthedocs.io/en/latest/?badge=latest)
+[![Poetry](https://img.shields.io/endpoint?url=https://python-poetry.org/badge/v0.json)](https://python-poetry.org/)
+![dev version](https://img.shields.io/badge/dynamic/toml?url=https%3A%2F%2Fraw.githubusercontent.com%2FAstarVienna%2FAnisoCADO%2Fmain%2Fpyproject.toml&query=%24.project.version&label=dev%20version&color=teal)
+
+[![License: GPL v3](https://img.shields.io/badge/License-GPLv3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0)
+![GitHub Release Date](https://img.shields.io/github/release-date/AstarVienna/AnisoCADO)
+[![PyPI - Version](https://img.shields.io/pypi/v/AnisoCADO)](https://pypi.org/project/AnisoCADO/)
+![PyPI - Python Version](https://img.shields.io/pypi/pyversions/AnisoCADO)
+
+
+A python package to generate off-axis PSFs for the SCAO mode for the ELT
+
+Please note: this package is not yet finished yet! The code is fine, but the
+documentation is lacking.
+
+## Documentation
+
+Apropos documentation. It can be found here:
+[https://anisocado.readthedocs.io/en/latest/index.html](https://anisocado.readthedocs.io/en/latest/index.html)
+

anisocado-0.4.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+anisocado/__init__.py,sha256=J-25dIVp8MNV4Wvzg_RX9373kccqDVhnuAW6FeYSZGs,309
+anisocado/_anisocado.py,sha256=-SbAZ9jATckuKtDC8qko0gJl9YM0PXmf7e6xuXdcDRM,9668
+anisocado/misc.py,sha256=4HRkwr1JavEcsvMB3Q7zJmwTwZs324uikdu3bAgfags,6243
+anisocado/psf.py,sha256=flXK2EvTYGhwQsd3nUZX2JvhghBK9FDjBIxAyeeOsu4,17410
+anisocado/psf_utils.py,sha256=D4fbm8w60LtJiDT5-OZzpwBnTSGoiRngimYMg-kKl6M,23178
+anisocado/pupil_utils.py,sha256=c-uRxucVtJMbKSgHcnev52aumpmprkOLko224bqKPjc,37172
+anisocado/tests/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+anisocado/tests/test_playing_around.py,sha256=cppxriCdCnQUjtkJ8DZmdyaZ-0csw1UFpofK3ldDhpI,1899
+anisocado/tests/test_psf_functions.py,sha256=i_Ke0NjapEgiRIAVVKPRpQrUrJaSDjJj1ebnf0FZAhY,3480
+anisocado/tests/test_scao_psf.py,sha256=6BvAjqAkh1hRg9AtW8Mf001zmET-6uDUUjcOf0jeIls,2556
+anisocado-0.4.0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+anisocado-0.4.0.dist-info/METADATA,sha256=SMee_1eUTNhfKPCv73dEcpnxqudVc-PpmWtOjw8Qtms,2770
+anisocado-0.4.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+anisocado-0.4.0.dist-info/RECORD,,

{AnisoCADO-0.3.0.dist-info → anisocado-0.4.0.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.40.0)
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any
-

AnisoCADO-0.3.0.dist-info/METADATA

@@ -1,52 +0,0 @@
-Metadata-Version: 2.1
-Name: AnisoCADO
-Version: 0.3.0
-Summary: Generate off-axis SCAO PSFs for the ELT
-Author: Eric Gendron
-Author-email: Kieran Leschinski <kieran.leschinski@unive.ac.at>
-Maintainer-email: Kieran Leschinski <kieran.leschinski@unive.ac.at>, Hugo Buddelmeijer <hugo@buddelmeijer.nl>
-License: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
-Project-URL: Homepage, https://anisocado.readthedocs.io/en/latest/
-Project-URL: Source, https://github.com/AstarVienna/AnisoCADO/
-Project-URL: Bug Reports, https://github.com/AstarVienna/AnisoCADO/issues
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
-Classifier: Operating System :: OS Independent
-Classifier: Intended Audience :: Science/Research
-Classifier: Topic :: Scientific/Engineering :: Astronomy
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: numpy (>=1.18.0)
-Requires-Dist: astropy (>=4.0)
-Requires-Dist: matplotlib (>=3.2.0)
-Provides-Extra: dev
-Requires-Dist: scipy ; extra == 'dev'
-Provides-Extra: docs
-Requires-Dist: sphinx (>=4.3.0) ; extra == 'docs'
-Requires-Dist: sphinx-rtd-theme (>=0.5.1) ; extra == 'docs'
-Requires-Dist: sphinxcontrib-apidoc ; extra == 'docs'
-Requires-Dist: numpydoc ; extra == 'docs'
-Provides-Extra: test
-Requires-Dist: pytest (>=5.0.0) ; extra == 'test'
-Requires-Dist: pytest-cov ; extra == 'test'
-
-# AnisoCADO
-
-[![Tests](https://github.com/AstarVienna/AnisoCADO/actions/workflows/tests.yml/badge.svg)](https://github.com/AstarVienna/AnisoCADO/actions/workflows/tests.yml)
-[![Documentation Status](https://readthedocs.org/projects/anisocado/badge/?version=latest)](https://anisocado.readthedocs.io/en/latest/?badge=latest)
-
-[![Build Status](http://github-actions.40ants.com/AstarVienna/AnisoCADO/matrix.svg)](https://github.com/AstarVienna/AnisoCADO)
-
-[![License: GPL v3](https://img.shields.io/badge/License-GPLv3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0)
-
-
-A python package to generate off-axis PSFs for the SCAO mode for the ELT
-
-Please note: this package is not yet finished yet! The code is fine, but the 
-documentation is lacking.
-
-## Documentation
-
-Apropos documentation. It can be found here:
-[https://anisocado.readthedocs.io/en/latest/index.html](https://anisocado.readthedocs.io/en/latest/index.html)

AnisoCADO-0.3.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-anisocado/__init__.py,sha256=k9knIPJ4-n-Ec52GePTgttwEmpQPSxaA_AI77Ky17aA,107
-anisocado/_anisocado.py,sha256=bc5pffn7-yHVy3iZUrHQugb40kcqw0i8dMLNR60Q--U,10116
-anisocado/misc.py,sha256=2CH1dOxUmdOZDUMFDbjRVSMLfQkMbmWphcdugYy5Q5o,6101
-anisocado/psf.py,sha256=F_4prXDKP_eI8C9B-vDnIcHkQ1srXt9j5UJahs9YBz0,16782
-anisocado/psf_utils.py,sha256=kX-RGhGLDBRc31CyOUMdZoLJXI6lcRbwsWHMuQ-P36Q,23379
-anisocado/pupil_utils.py,sha256=Qao9jv_c2smOrcdqmyxBme5V00NYkIVX1VGLK-Is8oc,36858
-anisocado/version.py,sha256=ta0Fge42PF4lOHwxmBa6ij23TVXjpCQBB46tB5SE020,104
-AnisoCADO-0.3.0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-AnisoCADO-0.3.0.dist-info/METADATA,sha256=jO9kpegkZDIWi1n6g4qUsJgv5AlPBgCNRBuw5hhAvg0,2359
-AnisoCADO-0.3.0.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-AnisoCADO-0.3.0.dist-info/top_level.txt,sha256=PEq0LVCCihooKMxK5ZCHLIl_7gTMnUgPXZsxNzW417Q,10
-AnisoCADO-0.3.0.dist-info/RECORD,,

AnisoCADO-0.3.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-anisocado

anisocado/version.py

@@ -1,3 +0,0 @@
-from importlib import metadata
-version = metadata.version(__package__)
-date = '2023-07-10 10:00:00 GMT'