Source code for pgb_utils.pgb_utils.figures

#!/usr/bin/env python3
# -*- coding: UTF-8 -*-

"""``figures`` contains functions for plotting alert and history data.
"""

import aplpy
from astropy.io import fits
from astropy.time import Time
import gzip
import io
import matplotlib as mpl
from matplotlib import pyplot as plt
import numpy as np
import pandas as pd
from typing import Optional

from . import utils as pgbu


#--- Plot object history
[docs]def plot_lightcurve(dflc: pd.DataFrame, objectId: Optional[str] = None, ax: Optional[mpl.axes.Axes] = None, days_ago: bool = True, ): """Plot the lighcurve (per band) of a single ZTF object. Adapted from: https://github.com/ZwickyTransientFacility/ztf-avro-alert/blob/master/notebooks/Filtering_alerts.ipynb Args: dflc: Lightcurve history of a unique objectId. Must contain columns ['jd','fid','magpsf','sigmapsf','diffmaglim'] objectId: objectId identifying dflc. Used for plot title. ax: matplotlib axes for the figure. If None, a new figure will be created days_ago: If True, x-axis will be number of days in the past. Else x-axis will be Julian date. """ filter_code = pgbu.ztf_fid_names() # dict filter_color = {1:'green', 2:'red', 3:'pink'} # set the x-axis (time) details if days_ago: now = Time.now().jd t = dflc.jd - now xlabel = 'Days Ago' else: t = dflc.jd xlabel = 'Time (JD)' if ax is None: plt.figure() # plot lightcurves by band for fid, color in filter_color.items(): # plot detections in this filter: w = (dflc.fid == fid) & ~dflc.magpsf.isnull() if np.sum(w): label = f'{fid}: {filter_code[fid]}' kwargs = {'fmt':'.', 'color':color, 'label':label} plt.errorbar(t[w],dflc.loc[w,'magpsf'], dflc.loc[w,'sigmapsf'], **kwargs) # plot nondetections in this filter wnodet = (dflc.fid == fid) & dflc.magpsf.isnull() if np.sum(wnodet): plt.scatter(t[wnodet],dflc.loc[wnodet,'diffmaglim'], marker='v',color=color,alpha=0.25) plt.gca().invert_yaxis() plt.xlabel(xlabel) plt.ylabel('Magnitude') plt.legend() plt.title(f'objectId: {objectId}')
#--- Plot cutouts
[docs]def plot_stamp(stamp, fig=None, subplot=None, **kwargs): """Adapted from: https://github.com/ZwickyTransientFacility/ztf-avro-alert/blob/master/notebooks/Filtering_alerts.ipynb """ with gzip.open(io.BytesIO(stamp), 'rb') as f: with fits.open(io.BytesIO(f.read())) as hdul: if fig is None: fig = plt.figure(figsize=(4,4)) if subplot is None: subplot = (1,1,1) ffig = aplpy.FITSFigure(hdul[0],figure=fig, subplot=subplot, **kwargs) # the following has been throwing: `ValueError: a must be > 0 and <= 1` # this is fixed by requiring astropy==3.2.1 # Note: I see this related thing: https://github.com/aplpy/aplpy/issues/420 # but I am using the latest APLpy version (2.0.3). ffig.show_grayscale(stretch='arcsinh') return ffig
[docs]def plot_cutouts(alert_dict): """Adapted from: https://github.com/ZwickyTransientFacility/ztf-avro-alert/blob/master/notebooks/Filtering_alerts.ipynb """ #fig, axes = plt.subplots(1,3, figsize=(12,4)) fig = plt.figure(figsize=(12,4)) for i, cutout in enumerate(['Science','Template','Difference']): stamp = alert_dict['cutout{}'.format(cutout)]['stampData'] ffig = plot_stamp(stamp, fig=fig, subplot = (1,3,i+1)) ffig.set_title(cutout)
#--- Plot all
[docs]def plot_lightcurve_cutouts(alert_dict): """Adapted from: https://github.com/ZwickyTransientFacility/ztf-avro-alert/blob/master/notebooks/Filtering_alerts.ipynb """ fig = plt.figure(figsize=(16,4)) dflc = pgbu.alert_dict_to_dataframe(alert_dict) plot_lightcurve(dflc,ax = plt.subplot(1,4,1)) for i, cutout in enumerate(['Science','Template','Difference']): stamp = alert_dict['cutout{}'.format(cutout)]['stampData'] ffig = plot_stamp(stamp, fig=fig, subplot = (1,4,i+2)) ffig.set_title(cutout)