Source code for pyobs.utils.fits
"""
TODO: write doc
"""
__title__ = "FITS utilities"
import logging
import re
from typing import Union, Any, cast, Optional, List, Callable, Dict
import numpy as np
from astropy.io import fits
from numpy.typing import NDArray
from pyobs.utils.time import Time
log = logging.getLogger(__name__)
def fitssec(hdu: Any, keyword: str = "TRIMSEC") -> NDArray[Any]:
"""Trim an image to TRIMSEC or BIASSEC.
Args:
hdu: HDU to take data from.
keyword: Header keyword for section.
Returns:
Numpy array with image data.
"""
# keyword not given?
if keyword not in hdu.header:
# return whole data
return cast(NDArray[Any], hdu.data)
# get value of section
sec = hdu.header[keyword]
# split values
s = sec[1:-1].split(",")
x = s[0].split(":")
y = s[1].split(":")
x0 = int(x[0]) - 1
x1 = int(x[1])
y0 = int(y[0]) - 1
y1 = int(y[1])
# return data
return cast(NDArray[Any], hdu.data[y0:y1, x0:x1])
[docs]class FilenameFormatter:
def __init__(self, fmt: Union[str, List[str]], keys: Optional[Dict[str, Any]] = None):
"""Initializes a new filename formatter.
Args:
fmt: Filename format or list of formats. If list is given, first valid one is used.
keys: Additional keys to pass to the formatter.
"""
self.format = fmt
self.keys: Dict[str, Any] = {} if keys is None else keys
# define functions
self.funcs: Dict[str, Callable[..., str]] = {
"lower": self._format_lower,
"time": self._format_time,
"date": self._format_date,
"filter": self._format_filter,
"string": self._format_string,
"type": self._format_type,
}
def _value(self, hdr: fits.Header, key: str) -> Any:
"""Returns value for given key.
Args:
hdr: fits.Header to take value from.
key: Key to return value for.
Returns:
Value for given key.
"""
# first check keys
if key in self.keys:
return self.keys[key]
# then check header
return hdr[key]
def __call__(self, hdr: fits.Header) -> Optional[str]:
"""Formats a filename given a format template and a FITS header.
Args:
hdr: FITS header to take values from.
Returns:
Formatted filename.
Raises:
KeyError: If either keyword could not be found in header or method could not be found.
"""
# make fmt a list
if self.format is None:
return None
if isinstance(self.format, str):
self.format = [self.format]
# loop formats
for f in self.format:
try:
# find all placeholders in format
placeholders = re.findall("\{[\w\d_-]+(?:\|[\w\d_-]+\:?(?:[\w\d_-]+)*)?\}", f)
if len(placeholders) == 0:
return f
# create output
output = f
# loop all placeholders
for ph in placeholders:
# call method and replace
output = output.replace(ph, str(self._format_placeholder(ph, hdr)))
# finished
return output
except KeyError:
# this format didn't work
pass
# still here?
raise KeyError("No valid format found.")
def _format_placeholder(self, placeholder: str, hdr: fits.Header) -> str:
"""Format a given placeholder.
Args:
placeholder: Placeholder to format.
hdr: FITS header to take values from.
Returns:
Formatted placeholder.
Raises:
KeyError: If method or FITS header keyword don't exist.
"""
# remove curly brackets
key = placeholder[1:-1]
method = None
params = []
# do we have a pipe in here?
if "|" in key:
key, method = key.split("|")
# parameters for method?
if method is not None and ":" in method:
method, *params = method.split(":")
# if no method is given, just replace
if method is None:
return str(self._value(hdr, key))
else:
# get function (may raise KeyError)
func = self.funcs[method]
# call method and replace
return func(hdr, key, *params)
def _format_lower(self, hdr: fits.Header, key: str) -> str:
"""Sets a given string to lowercase.
Args:
hdr: FITS header to take values from.
key: The name of the FITS header key to use.
Returns:
Formatted string.
"""
return str(self._value(hdr, key)).lower()
def _format_time(self, hdr: fits.Header, key: str, delimiter: str = "-") -> str:
"""Formats time using the given delimiter.
Args:
hdr: FITS header to take values from.
key: The name of the FITS header key to use.
delimiter: Delimiter for time formatting.
Returns:
Formatted string.
"""
fmt = "%H" + delimiter + "%M" + delimiter + "%S"
date_obs = Time(self._value(hdr, key))
return str(date_obs.datetime.strftime(fmt))
def _format_date(self, hdr: fits.Header, key: str, delimiter: str = "-") -> str:
"""Formats date using the given delimiter.
Args:
hdr: FITS header to take values from.
key: The name of the FITS header key to use.
delimiter: Delimiter for date formatting.
Returns:
Formatted string.
"""
fmt = "%Y" + delimiter + "%m" + delimiter + "%d"
date_obs = Time(self._value(hdr, key))
return str(date_obs.datetime.strftime(fmt))
def _format_filter(self, hdr: fits.Header, key: str, image_type: str = "IMAGETYP", prefix: str = "-") -> str:
"""Formats a filter, prefixed by a given separator, only if the image type requires it.
Args:
hdr: FITS header to take values from.
key: The name of the FITS header key to use.
image_type: FITS header key for IMAGETYP.
prefix: Prefix to add to filter.
Returns:
Formatted string.
"""
it = hdr[image_type].lower()
if it in ["light", "object", "flat", "skyflat"]:
return str(prefix + self._value(hdr, key))
else:
return ""
def _format_string(self, hdr: fits.Header, key: str, fmt: str) -> str:
"""Formats a string using Python string substitution.
Args:
hdr: FITS header to take values from.
key: The name of the FITS header key to use.
fmt: A Python string format like %d, %05d, or %4.1f.
Returns:
Formatted string.
"""
fmt = "%" + fmt
return str(fmt % self._value(hdr, key))
def _format_type(self, hdr: fits.Header, key: str) -> str:
"""Formats an image type to a one-letter code.
Args:
hdr: FITS header to take values from.
key: The name of the FITS header key to use.
Returns:
Formatted string.
"""
if self._value(hdr, key) == "bias":
return "b"
elif self._value(hdr, key) == "skyflat":
return "f"
elif self._value(hdr, key) == "dark":
return "d"
else:
return "e"
def format_filename(
hdr: fits.Header, fmt: Union[str, List[str]], keys: Optional[Dict[str, Any]] = None
) -> Optional[str]:
"""Formats a filename given a format template and a FITS header.
Args:
hdr: FITS header to take values from.
fmt: Filename format or list of formats. If multiple formats are given, the first valid one is used.
keys: Additional keys to pass to the format string.
Returns:
Filename
Raises:
KeyError: If either keyword could not be found in header or method could not be found.
"""
ff = FilenameFormatter(fmt, keys)
return ff(hdr)
__all__ = ["format_filename", "FilenameFormatter", "fitssec"]