lightstack 0.1.1__tar.gz → 0.1.3__tar.gz

{lightstack-0.1.1/src/lightstack.egg-info → lightstack-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightstack
-Version: 0.1.1
+Version: 0.1.3
 Summary: Tools for building and processing multi-filter astrophysical datacubes
 Author: Andressa Wille, Thallis Pessi
 License: MIT License
@@ -35,6 +35,8 @@ Requires-Dist: regions
 Requires-Dist: scipy
 Requires-Dist: photutils
 
+[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.19393439.svg)](https://doi.org/10.5281/zenodo.19393439)
+
 <img src="https://raw.githubusercontent.com/AndressaWille/lightstack/master/images/logo.png" width="120">
 
 # Lightstack

{lightstack-0.1.1 → lightstack-0.1.3}/README.md

@@ -1,3 +1,5 @@
+[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.19393439.svg)](https://doi.org/10.5281/zenodo.19393439)
+
 <img src="https://raw.githubusercontent.com/AndressaWille/lightstack/master/images/logo.png" width="120">
 
 # Lightstack

{lightstack-0.1.1 → lightstack-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lightstack"
-version = "0.1.1"
+version = "0.1.3"
 description = "Tools for building and processing multi-filter astrophysical datacubes"
 readme = "README.md"
 authors = [{name = "Andressa Wille"}, {name="Thallis Pessi"}]

{lightstack-0.1.1 → lightstack-0.1.3}/src/lightstack/datacube.py

@@ -6,7 +6,7 @@ from astropy.wcs import WCS
 
 from reproject import reproject_interp, reproject_exact
 
-from .utils import find_ext, infer_filter
+from .utils import find_ext, infer_filter, get_pixel_scale_from_wcs
 
 
 def align_reproject_fits(fits_list, ref_file, method="interp", crop=1):
@@ -24,7 +24,7 @@ def align_reproject_fits(fits_list, ref_file, method="interp", crop=1):
     method : str, optional
         Reprojection method:
         - "interp" (default): faster, interpolates values
-        - "exact": slower, conserves flux   --> but reproject_exact has precision issues with resolutions below ~0.05 arcsec, so the results may not be accurate.
+        - "exact": slower   --> but reproject_exact has precision issues with resolutions below ~0.05 arcsec, so the results may not be accurate.
 
     crop : int, optional
         Number of pixels to remove from each border after reprojection
@@ -36,6 +36,7 @@ def align_reproject_fits(fits_list, ref_file, method="interp", crop=1):
         List in the form [(aligned_fits_path, filter_name), ...].
     """
 
+
     # Choose reprojection method
     if method == "exact":
         reproj_func = reproject_exact
@@ -54,6 +55,9 @@ def align_reproject_fits(fits_list, ref_file, method="interp", crop=1):
         ref_wcs = WCS(ref_header)
         shape_out = hdul_ref[ext_ref].data.shape
 
+    # Pixel scale of reference
+    scale_out = get_pixel_scale_from_wcs(ref_wcs)
+
     aligned_list = []
 
     # Loop over all FITS
@@ -65,20 +69,30 @@ def align_reproject_fits(fits_list, ref_file, method="interp", crop=1):
                 print(f"No data extension found in {fpath}. Skipping.")
                 continue
 
-            data = hdul[ext].data
+            data = hdul[ext].data.astype(float)
             header = hdul[ext].header
             wcs_in = WCS(header)
 
             unit = header.get('BUNIT', 'unknown')
             print(f"Filter {filt}: unit = {unit}")
 
+            # Pixel scale of input
+            scale_in = get_pixel_scale_from_wcs(wcs_in)
+
             # Reproject
             data_aligned, footprint = reproj_func(
                 (data, wcs_in),
                 ref_wcs,
                 shape_out=shape_out)
 
-            # Crop border (interp method)
+            # Area correction
+            area_ratio = (scale_out / scale_in)**2
+            data_aligned *= area_ratio
+
+            print(f"Pixel scale in/out: {scale_in:.4f} -> {scale_out:.4f}")
+            print(f"Applied area correction: {area_ratio:.4f}")
+
+            # Crop border
             if crop > 0:
                 data_aligned = data_aligned[crop:-crop, crop:-crop]
 
@@ -90,10 +104,13 @@ def align_reproject_fits(fits_list, ref_file, method="interp", crop=1):
 
             # Header
             header_aligned = wcs_out.to_header()
+            header_aligned['BUNIT'] = unit
 
             ref_filter = infer_filter(ref_file)
             header_aligned.add_history(
                 f"Reprojected to {ref_filter} using {method} method (reproject package)")
+            header_aligned.add_history(
+                "Flux corrected by pixel area ratio")
 
             if crop > 0:
                 header_aligned.add_history(f"Cropped {crop} pixels from each border")
@@ -105,10 +122,16 @@ def align_reproject_fits(fits_list, ref_file, method="interp", crop=1):
             fits.PrimaryHDU(data_aligned, header=header_aligned).writeto(
                 out_name, overwrite=True)
 
+            # Flux sanity check
+            sum_in = np.nansum(data)
+            sum_out = np.nansum(data_aligned)
+            print(f"Flux ratio (out/in): {sum_out/sum_in:.4f}")
+
             aligned_list.append((out_name, filt))
             print(f"Saved: {out_name}")
 
     return aligned_list
+
     
     
 def build_datacube(aligned_fits_files, reference_file, output_path):

{lightstack-0.1.1 → lightstack-0.1.3}/src/lightstack/psf.py

@@ -125,8 +125,14 @@ def resample_psf(
             raise ValueError("Provide either zoom_factor OR both psf_pixel_scale and target_pixel_scale")
         zoom_factor = psf_pixel_scale / target_pixel_scale
 
+    if zoom_factor > 1:
+    	raise ValueError("Upsampling not recommended.")
+    
     # Resample
     psf_resampled = zoom(psf, zoom_factor, order=order)
+    
+    # Area correction
+    psf_resampled /= zoom_factor**2
 
     # Ensure odd shape
     if make_odd_shape:
@@ -143,6 +149,7 @@ def resample_psf(
         header["HISTORY"] = "PSF resampled using scipy.ndimage.zoom"
         if target_pixel_scale is not None:
             header["CDELT1"] = (target_pixel_scale / 3600, "deg/pix")
+            header["CDELT2"] = (target_pixel_scale / 3600, "deg/pix")
 
     # Save
     fits.PrimaryHDU(psf_resampled, header=header).writeto(

{lightstack-0.1.1 → lightstack-0.1.3}/src/lightstack/utils.py

@@ -210,7 +210,7 @@ def MJy_sr_to_jy(aligned_list):
 
 def get_pixel_scale(fits_path):
     """
-    Compute pixel scale in arcsec/pixel using WCS.
+    Compute pixel scale in arcsec/pixel using WCS. Assumes square pixels and no significant distortion.
     """
     with fits.open(fits_path) as hdul:
         ext = find_ext(hdul)
@@ -221,3 +221,11 @@ def get_pixel_scale(fits_path):
 
     pixscale = proj_plane_pixel_scales(wcs)[0] * 3600.0
     return pixscale
+    
+    
+def get_pixel_scale_from_wcs(wcs):
+    """
+    Compute pixel scale in arcsec/pixel from a WCS object.
+    Assumes square pixels and no significant distortion.
+    """
+    return proj_plane_pixel_scales(wcs)[0] * 3600.0

{lightstack-0.1.1 → lightstack-0.1.3/src/lightstack.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightstack
-Version: 0.1.1
+Version: 0.1.3
 Summary: Tools for building and processing multi-filter astrophysical datacubes
 Author: Andressa Wille, Thallis Pessi
 License: MIT License
@@ -35,6 +35,8 @@ Requires-Dist: regions
 Requires-Dist: scipy
 Requires-Dist: photutils
 
+[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.19393439.svg)](https://doi.org/10.5281/zenodo.19393439)
+
 <img src="https://raw.githubusercontent.com/AndressaWille/lightstack/master/images/logo.png" width="120">
 
 # Lightstack