fits2tmp 0.1.0__py3-none-any.whl

fits2tmp-0.1.0.dist-info/LICENSE

@@ -0,0 +1,28 @@
+BSD 3-Clause License
+
+Copyright (c) 2024, Verity Allan
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

fits2tmp-0.1.0.dist-info/METADATA

@@ -0,0 +1,31 @@
+Metadata-Version: 2.1
+Name: fits2tmp
+Version: 0.1.0
+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
+Author: Verity Allan
+Author-email: vla22@cam.ac.uk
+Requires-Python: >=3.10,<4.0
+Classifier: License :: OSI Approved :: BSD License
+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: 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)
+Requires-Dist: zarr (>=2.18.3,<3.0.0)
+Project-URL: Repository, https://gitlab.developers.cam.ac.uk/phy/ra/vla22-phd/fits2tmp
+Description-Content-Type: text/markdown
+
+# Sample GitLab Project
+
+This sample project shows how a project in GitLab looks for demonstration purposes. It contains issues, merge requests and Markdown files in many branches,
+named and filled with lorem ipsum.
+
+You can look around to get an idea how to structure your project and, when done, you can safely delete this project.
+
+[Learn more about creating GitLab projects.](https://docs.gitlab.com/ee/gitlab-basics/create-project.html)
+

fits2tmp-0.1.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+src/fits2tmp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/fits2tmp/fits2tmp.py,sha256=v3vg7XXHLFFCSyHMu6dDa72lZqPn74tkk_nYXhYM6no,3363
+src/fits2tmp/sdc2_100.fits,sha256=-1154P1-muCt9x1iycnbZd8yLqtlFrOSbJ1HXeatbHk,855360
+src/fits2tmp/sofiatestrewrite1002_cat.txt,sha256=ksqBuW2MIUmMRGPMabHxoiA24vuknfHyfSccZF6nhjA,8624
+src/fits2tmp/sofiatestrewrite1002_cat.xml,sha256=k2nJbCAzJ8UttI9iufQnSN8leEzsM7D85ZfwOcMFpjs,27739
+src/fits2tmp/sofiatestrewrite100_cat.txt,sha256=k5bjtLfU_8VaHg5My1-VzS4QLpIHz4VAX-CQk0In8B4,8624
+src/fits2tmp/sofiatestrewrite100_cat.xml,sha256=vlEDIiNls4xSPWbd93trR7grAWUWkjTg5OPqEfJ05Bw,27738
+fits2tmp-0.1.0.dist-info/LICENSE,sha256=Ok-auf5W6wYUmfqD3NURQIvpq09FNV_fU39rYwCq8_4,1499
+fits2tmp-0.1.0.dist-info/METADATA,sha256=cIsOcE5SutbiXZjIOIiXLzOXt5qS3R2GqAZHz8iVtpo,1344
+fits2tmp-0.1.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+fits2tmp-0.1.0.dist-info/RECORD,,

fits2tmp-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.6.1
+Root-Is-Purelib: true
+Tag: py3-none-any

src/fits2tmp/fits2tmp.py

@@ -0,0 +1,105 @@
+import h5py
+import zarr
+import json
+from astropy.io import fits
+from astropy import wcs
+import dask.array as da
+import numpy as np
+import os
+import dask 
+import tempfile
+
+def arr_size_check(source_arr, desired_arr: tuple) -> bool:
+    if len(source_arr.shape) < len(desired_arr):
+        raise ValueError("You've asked for an array with more dimensions than the array stored")
+    else:
+        for i in range(min(len(source_arr.shape), len(desired_arr))):
+            # using min here allows us to avoid an IndexError when we have more dimensions in the source array
+            j = source_arr.shape[i] - desired_arr[i]
+            if j < 0:
+                raise IndexError("Desired array size is larger than the stored array in at least position %s: source array has dimensions %s, and you have requested $s", i, source_arr.shape, desired_arr)
+        return True
+
+#def fits_add_header(orig_header, sliced_data) -> fits.Header:
+#    coords = wcs.WCS(orig_header)
+#    new_header = orig_header
+#    new_header.update(coords[data_sli].to_header())
+#    return new_header
+
+def header_dict(header) -> dict:
+    d = {}
+    for i in header.cards:
+        d[i.keyword] = i.value
+    return d
+
+def dict2header(input_dict: dict) -> fits.Header:
+    nh = fits.Header(cards=input_dict)
+    return nh
+
+def dict2file(input_dict: dict, filepath):
+    with open(filepath, 'w') as f:
+        f.write(json.dumps(input_dict))
+
+def file2dict(input_file, filepath) -> dict:
+    with open(filepath, 'r') as j:
+        contents = json.loads(j.read())
+    return contents
+filepath = []
+
+def fun(darr, destination, header, block_info=None):
+    arr_loc = block_info[None]['array-location']
+    print(arr_loc)
+    ch_size = block_info[None]['chunk-shape']
+    chunk_loc = block_info[None]['chunk-location']
+    coords = []
+    for i in range(len(arr_loc)):
+        coords.append(slice(arr_loc[i][0],arr_loc[i][1]))
+    block_id = str(chunk_loc).replace(',', '-').replace('(', '-').replace(')', '').replace(' ', '')
+    n_header = dict2header(header_dict(header))
+    wcs_obj = wcs.WCS(n_header)
+    n_header.update(wcs_obj[coords].to_header())
+    out = fits.PrimaryHDU(data=darr, header=n_header)
+    outpath = destination.name  + '/' + block_id + ".fits"
+    out.writeto(outpath)
+    filepath.append(outpath)
+    return darr 
+
+def fits2tmpfiles(inputpath, destination, chunks):
+    f = fits.open(inputpath)
+    destination = tempfile.TemporaryDirectory(dir=destination, delete=False)
+    d = da.from_array(f[0].data, chunks=chunks)
+    # TODO overlap
+    de = da.overlap.overlap(d, depth=None, boundary=None)
+    da.map_blocks(fun, d, destination, f[0].header, dtype=float).compute()
+
+    return filepath
+
+
+def disk2darray(filetype, filepath, arraytype, arraysize, chunks=None):
+    match filetype:
+        case "fits":
+            foo
+        case "zarr":
+            bar
+        case "hdf5":
+            baz
+        case "xradio":
+            weeble
+        case _:
+            raise ValueError("The specified file type is not supported by fits2tmp.")
+    return array
+
+
+#def array2disk(array, metadata, outputtype, destination):
+
+#def tmpfitsfile2disk(fitsfile, outputtype, destination):
+
+#def tmpfitsfile2array(fitsfile, arraytype, arraysize, chunks=None):
+
+#def makezarrmetadata(fitsheader):
+
+#def makefitsmetadata(fitsheader, location):
+
+#makehdf5metadata
+
+#makexradiometadata