Aris's preprocessing code

Author: Jammy2211

autoarray/dataset/interferometer/dataset.py

@@ -1,5 +1,7 @@
+from astropy.io import fits
 import logging
 import numpy as np
+from pathlib import Path
 from typing import Optional
 
 from autoconf import cached_property
@@ -27,6 +29,7 @@ def __init__(
         uv_wavelengths: np.ndarray,
         real_space_mask,
         transformer_class=TransformerNUFFT,
+        preprocessing_directory=None,
     ):
         """
         An interferometer dataset, containing the visibilities data, noise-map, real-space msk, Fourier transformer and
@@ -88,6 +91,8 @@ def __init__(
             uv_wavelengths=uv_wavelengths, real_space_mask=real_space_mask
         )
 
+        self.preprocessing_directory = Path(preprocessing_directory) if preprocessing_directory is not None else None
+
     @cached_property
     def grids(self):
         return GridsDataset(
@@ -134,6 +139,27 @@ def from_fits(
             transformer_class=transformer_class,
         )
 
+    def w_tilde_preprocessing(self):
+
+        if self.preprocessing_directory.is_dir():
+
+            filename = "{}/curvature_preload.fits".format(self.preprocessing_directory)
+
+            if not self.preprocessing_directory.isfile(filename):
+                print(
+                    "The file {} does not exist".format(filename)
+                )
+                logger.info("INTERFEROMETER - Computing W-Tilde... May take a moment.")
+
+                curvature_preload = inversion_interferometer_util.w_tilde_curvature_preload_interferometer_from(
+                    noise_map_real=self.noise_map.real,
+                    uv_wavelengths=self.uv_wavelengths,
+                    shape_masked_pixels_2d=self.transformer.grid.mask.shape_native_masked_pixels,
+                    grid_radians_2d=self.transformer.grid.mask.unmasked_grid_sub_1.in_radians.native,
+                )
+
+                fits.writeto(filename, data=curvature_preload)
+
     @cached_property
     def w_tilde(self):
         """

test_autoarray/inversion/inversion/interferometer/test_inversion_interferometer_util.py

@@ -144,7 +144,7 @@ def test__curvature_matrix_via_w_tilde_preload_from():
     )
 
     w_tilde_preload = (
-        aa.util.inversion.w_tilde_curvature_preload_interferometer_from(
+        aa.util.inversion_interferometer.w_tilde_curvature_preload_interferometer_from(
             noise_map_real=np.array(noise_map),
             uv_wavelengths=np.array(uv_wavelengths),
             shape_masked_pixels_2d=(3, 3),
@@ -163,7 +163,7 @@ def test__curvature_matrix_via_w_tilde_preload_from():
         [[0, 0], [0, 1], [0, 2], [1, 0], [1, 1], [1, 2], [2, 0], [2, 1], [2, 2]]
     )
 
-    curvature_matrix_via_preload = aa.util.inversion.curvature_matrix_via_w_tilde_curvature_preload_interferometer_from(
+    curvature_matrix_via_preload = aa.util.inversion_interferometer.curvature_matrix_via_w_tilde_curvature_preload_interferometer_from(
         curvature_preload=w_tilde_preload,
         pix_indexes_for_sub_slim_index=pix_indexes_for_sub_slim_index,
         pix_size_for_sub_slim_index=pix_size_for_sub_slim_index,
@@ -192,7 +192,7 @@ def test__curvature_matrix_via_w_tilde_two_methods_agree():
     )
 
     w_tilde_preload = (
-        aa.util.inversion.w_tilde_curvature_preload_interferometer_from(
+        aa.util.inversion_interferometer.w_tilde_curvature_preload_interferometer_from(
             noise_map_real=np.array(noise_map),
             uv_wavelengths=np.array(uv_wavelengths),
             shape_masked_pixels_2d=(3, 3),
@@ -204,7 +204,7 @@ def test__curvature_matrix_via_w_tilde_two_methods_agree():
         [[0, 0], [0, 1], [0, 2], [1, 0], [1, 1], [1, 2], [2, 0], [2, 1], [2, 2]]
     )
 
-    w_tilde_via_preload = aa.util.inversion.w_tilde_via_preload_from(
+    w_tilde_via_preload = aa.util.inversion_interferometer.w_tilde_via_preload_from(
         w_tilde_preload=w_tilde_preload,
         native_index_for_slim_index=native_index_for_slim_index,
     )