WIP

Author: astrofrog

reproject/hips/dask_array.py

@@ -170,8 +170,11 @@ def __init__(self, directory_or_url, level=None):
 
         # Now determine what the indices would be for this at the given spectral order
 
-        self._index_min = freq2pix(self._level_freq, self._freq_min)
-        self._index_max = freq2pix(self._level_freq, self._freq_max) + 1
+
+        self._index_min = freq2pix(self._level_freq, self._freq_min) + 1
+        self._index_max = freq2pix(self._level_freq, self._freq_max) + 2
+
+        print(self._level_freq, self._index_min)
 
         image_depth = (self._index_max - self._index_min) * self._tile_depth
 
@@ -212,6 +215,8 @@ def __getitem__(self, item):
 
         spectral_index = ispec // self._tile_depth + self._index_min
 
+        print(spectral_index, self._index_min, self._index_max)
+
         return self._get_tile(
             level=self._level, spatial_index=spatial_index, spectral_index=spectral_index
         )

reproject/hips/utils.py

@@ -100,6 +100,19 @@ def make_tile_folders(*, level, indices, output_directory):
             os.makedirs(dirname)
 
 
+def make_tile_folders_3d(*, spatial_level, spectral_level, spatial_indices, spectral_indices, output_directory):
+
+    rounded_spatial_indices = np.unique(_rounded_spatial_index(spatial_indices))
+    for spatial_index in rounded_spatial_indices:
+        dirname = os.path.dirname(
+            tile_filename_3d(spatial_level=spatial_level, spectral_level=spectral_level,
+            spatial_index=spatial_index, spectral_index=spectral_indices,
+            output_directory=output_directory, extension="")
+        )
+        if not os.path.exists(dirname):
+            os.makedirs(dirname)
+
+
 def determine_healpix_level(wcs_in, tile_size):
     """
     Determine the appropriate HEALPix level by matching the HEALPix pixel size