aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAaron LI <aly@aaronly.me>2017-08-13 11:44:54 +0800
committerAaron LI <aly@aaronly.me>2017-08-13 11:44:54 +0800
commit37094a6a90cc887e9a0ad94e59844fe43fe8507c (patch)
treea4a6dba1e372acda66595abb587c84ab6b2487b1
parentc36c990656133dc0b7c159aad4f7faaa6ecc7e18 (diff)
downloadfg21sim-37094a6a90cc887e9a0ad94e59844fe43fe8507c.tar.bz2
sky.py: Update to use "write_fits_{image,healpix}" functions
Signed-off-by: Aaron LI <aly@aaronly.me>
-rw-r--r--fg21sim/sky.py62
1 files changed, 19 insertions, 43 deletions
diff --git a/fg21sim/sky.py b/fg21sim/sky.py
index 99de963..d998bbd 100644
--- a/fg21sim/sky.py
+++ b/fg21sim/sky.py
@@ -6,7 +6,6 @@ Generic simulation sky supporting both sky patch and HEALPix all-sky
maps.
"""
-import os
import logging
import copy
from datetime import datetime, timezone
@@ -158,7 +157,7 @@ class SkyBase:
"""
raise NotImplementedError
- def write(self, outfile, clobber=False, checksum=False):
+ def write(self, outfile):
"""
Write the sky image (with current data) into a FITS file.
@@ -166,20 +165,6 @@ class SkyBase:
----------
outfile : str
The path/filename to the output FITS file.
- clobber : bool, optional
- Whether to overwrite the existing output file.
- Default: False
- checksum : bool, optional
- Whether to calculate the checksum data for the output
- FITS file, which may cost some time.
- Default: False
- """
- raise NotImplementedError
-
- @property
- def shape(self):
- """
- Numpy array shape of the (current/output) sky data.
"""
raise NotImplementedError
@@ -394,31 +379,27 @@ class SkyPatch(SkyBase):
return sky
def copy(self):
+ def write(self, outfile):
"""
Return a copy of this instance.
"""
return copy.deepcopy(self)
+ write_fits_image(outfile, image=self.data, header=self.header,
+ float32=self.float32_,
+ clobber=self.clobber_,
+ checksum=self.checksum_)
- def write(self, outfile, clobber=False, checksum=True):
+ @property
+ def header(self):
"""
Write current data to file.
+ FITS header of the sky for storing information in the output file.
"""
- outdir = os.path.dirname(outfile)
- if outdir and (not os.path.exists(outdir)):
- os.makedirs(outdir)
- logger.info("Created output directory: %s" % outdir)
- # NOTE: output image shape be (ysize, xsize)!
- image = self.data.reshape(self.ysize, self.xsize)
- if hasattr(self, "header"):
- header = self.header.copy(strip=True)
- wcs_header = self.wcs.to_header()
- header.extend(wcs_header, update=True)
- header["PIXSIZE"] = (self.pixelsize, "Pixel size [arcsec]")
- header["RA0"] = (self.center[0], "R.A. of patch center [deg]")
- header["DEC0"] = (self.center[1], "Dec. of patch center [deg]")
- hdu = fits.PrimaryHDU(data=image, header=header)
- hdu.writeto(outfile, clobber=clobber, checksum=checksum)
- logger.info("Write sky map to file: %s" % outfile)
+ hdr = super().header
+ hdr.extend(self.wcs.to_header(), update=True)
+ hdr["RA0"] = (self.center[0], "R.A. of patch center [deg]")
+ hdr["DEC0"] = (self.center[1], "Dec. of patch center [deg]")
+ return hdr
@property
def wcs(self):
@@ -634,19 +615,14 @@ class SkyHealpix(SkyBase):
"""
return copy.deepcopy(self)
- def write(self, outfile, clobber=False, checksum=True):
+ def write(self, outfile):
"""
Write current data to file.
"""
- outdir = os.path.dirname(outfile)
- if outdir and (not os.path.exists(outdir)):
- os.makedirs(outdir)
- logger.info("Created output directory: %s" % outdir)
- if hasattr(self, "header"):
- header = self.header
- write_fits_healpix(outfile, self.data, header=header,
- clobber=clobber, checksum=checksum)
- logger.info("Write sky map to file: %s" % outfile)
+ write_fits_healpix(outfile, hpmap=self.data, header=self.header,
+ float32=self.float32_,
+ clobber=self.clobber_,
+ checksum=self.checksum_)
def contains(self, skycoord):
"""