fieldplot.py

"""Helper functions for creating K2 field plots."""
from matplotlib import pyplot as pl
import matplotlib.patheffects as path_effects
from matplotlib.patches import Ellipse


def annotate_target(ras, decs, texts, ha='left',
                    extended=False, padding=0.22, fontsize=16, colortext=False,
                    **kwargs):
    if not hasattr(ras, '__iter__'):
        ras = [ras]
        decs = [decs]
        texts = [texts]

    if extended:
        for ra, dec in zip(ras, decs):
            padding = 1.82 * padding
            if 'color' in kwargs:
                el = Ellipse((ra, dec), width=0.5, height=0.18, lw=2.,
                             color=kwargs['color'], zorder=kwargs['zorder'])
            else:
                el = Ellipse((ra, dec), width=0.5, height=0.18, lw=2., zorder=kwargs['zorder'])
            pl.axes().add_artist(el)
    else:
        pl.plot(ras, decs, **kwargs)
    if ha == 'left':
        padding = -padding
    for ra, dec, text in zip(ras, decs, texts):
        if 'zorder' in kwargs:
            if colortext:
                text = pl.text(ra + padding, dec, text, zorder=kwargs['zorder']+1,
                               fontsize=fontsize, va='center', ha=ha, color=kwargs['color'])
            else:
                text = pl.text(ra + padding, dec, text, zorder=kwargs['zorder']+1,
                               fontsize=fontsize, va='center', ha=ha)
        else:
            if colortext:
                text = pl.text(ra + padding, dec, text,
                               fontsize=fontsize, va='center', ha=ha, color=kwargs['color'])
            else:
                text = pl.text(ra + padding, dec, text,
                               fontsize=fontsize, va='center', ha=ha)

        text.set_path_effects([path_effects.Stroke(linewidth=4, foreground='white'),
                               path_effects.Normal()])