fits2tmp 0.1.7__tar.gz → 0.1.8__tar.gz

{fits2tmp-0.1.7 → fits2tmp-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fits2tmp
-Version: 0.1.7
+Version: 0.1.8
 Summary: A way of loading FITS files into temporary filesystems to make them available in memory.
 Home-page: https://gitlab.developers.cam.ac.uk/phy/ra/vla22-phd/fits2tmp
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: astropy (>=6.1.4,<7.0.0)
-Requires-Dist: dask[complete] (>=2024.10.0,<2025.0.0)
+Requires-Dist: dask[distributed] (>=2025.1.0,<2026.0.0)
 Requires-Dist: h5py (>=3.12.1,<4.0.0)
 Requires-Dist: numpy (>=2.1.2,<3.0.0)
 Requires-Dist: pytest (>=8.3.3,<9.0.0)

{fits2tmp-0.1.7 → fits2tmp-0.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "fits2tmp"
-version = "0.1.7"
+version = "0.1.8"
 description = "A way of loading FITS files into temporary filesystems to make them available in memory."
 license = "BSD-3-Clause"
 authors = ["Verity Allan <vla22@cam.ac.uk>"]
@@ -9,7 +9,7 @@ repository = "https://gitlab.developers.cam.ac.uk/phy/ra/vla22-phd/fits2tmp"
 
 [tool.poetry.dependencies]
 python = "^3.10"
-dask = {extras = ["complete"], version = "^2024.10.0"}
+dask = {extras = ["distributed"], version = "^2025.1.0"}
 astropy = "^6.1.4"
 h5py = "^3.12.1"
 numpy = "^2.1.2"

{fits2tmp-0.1.7 → fits2tmp-0.1.8}/src/fits2tmp/fits2tmp.py

@@ -4,6 +4,7 @@ import json
 from astropy.io import fits
 from astropy import wcs
 import dask.array as da
+from dask import delayed
 from dask.distributed import Lock
 import numpy as np
 import os
@@ -78,11 +79,38 @@ def fits2tmpfiles(inputpath, destination, chunks):
 
     return filepath
 
-
-def disk2darray(filetype, filepath, arraytype, arraysize, chunks=None):
+@delayed
+def disk2tmpfiles(filetype, inputpath, destination, arraytype, arraysize, chunks=None):
+    fpath = []
     match filetype:
         case "fits":
-            foo
+            f = fits.open(inputpath)
+            d = da.from_array(f[0].data, chunks=chunks)
+            destination = tempfile.TemporaryDirectory(dir=destination, delete=False)
+            n_header = dict2header(header_dict(f[0].header))
+            wcs_obj = wcs.WCS(n_header)
+            print(d.shape)
+            for i in range(d.blocks.shape[0]):
+                x = i * chunks[0] 
+                x_delt = x + chunks[0] - 1
+                for j in range(d.blocks.shape[1]):
+                    data = d.blocks[i][j]
+                    y = j * chunks[1]
+                    y_delt = y + chunks[1] - 1
+                    if len(d.blocks.shape) == 2:
+                        dest = str(i) + str(j)
+                        wcs_coords = wcs_obj.slice((slice(x,x_delt), slice(y,y_delt)))
+                    elif len(d.blocks.shape) > 2:
+                        for k in range(d.blocks.shape[2]):
+                            data = d.blocks[i][j][k]
+                            z = k * chunks[2]
+                            z_delt = z + chunks[2] - 1
+                            dest = str(i) + str(j) + str(k)
+                            t =wcs_obj[x:x_delt, y:y_delt, z:z_delt]
+                            n_header.update(t.to_header())
+                            hdu = fits.PrimaryHDU(data=data, header=n_header)
+                            hdu.writeto(destination.name + dest + "fits")
+                            fpath.append(destination.name + dest + "fits")
         case "zarr":
             bar
         case "hdf5":
@@ -91,8 +119,7 @@ def disk2darray(filetype, filepath, arraytype, arraysize, chunks=None):
             weeble
         case _:
             raise ValueError("The specified file type is not supported by fits2tmp.")
-    return array
-
+    return fpath
 
 #def array2disk(array, metadata, outputtype, destination):