Commit 9f589d78 authored by Leo Pound Singer's avatar Leo Pound Singer
Browse files

Use colormaps' full dynamic range in plot-airmass

parent d693bef4
...@@ -9,6 +9,9 @@ Changelog ...@@ -9,6 +9,9 @@ Changelog
for many sites at once. Conceptually, this tool is a variation of for many sites at once. Conceptually, this tool is a variation of
``ligo-skymap-plot-airmass`` in which the sky position is integrated out. ``ligo-skymap-plot-airmass`` in which the sky position is integrated out.
- The ``ligo-skymap-plot-airmass`` tool will now use the color map's full
dynamic range.
0.1.6 (2019-03-26) 0.1.6 (2019-03-26)
================== ==================
......
...@@ -82,6 +82,8 @@ def main(args=None): ...@@ -82,6 +82,8 @@ def main(args=None):
from astropy.time import Time from astropy.time import Time
from astropy import units as u from astropy import units as u
from matplotlib import dates from matplotlib import dates
from matplotlib.cm import ScalarMappable
from matplotlib.colors import Normalize
from matplotlib.patches import Patch from matplotlib.patches import Patch
from matplotlib import pyplot as plt from matplotlib import pyplot as plt
from tqdm import tqdm from tqdm import tqdm
...@@ -143,16 +145,16 @@ def main(args=None): ...@@ -143,16 +145,16 @@ def main(args=None):
weights=prob) weights=prob)
for t in tqdm(times)]) for t in tqdm(times)])
cmap = plt.get_cmap() cmap = ScalarMappable(Normalize(0, 100), plt.get_cmap())
for level, lo, hi in zip(levels, airmass[:nlevels], airmass[nlevels:]): for level, lo, hi in zip(levels, airmass[:nlevels], airmass[nlevels:]):
ax.fill_between( ax.fill_between(
times.plot_date, times.plot_date,
clip_verylarge(lo), # Clip infinities to large but finite values clip_verylarge(lo), # Clip infinities to large but finite values
clip_verylarge(hi), # because fill_between cannot handle inf clip_verylarge(hi), # because fill_between cannot handle inf
color=cmap(level), zorder=2) color=cmap.to_rgba(level), zorder=2)
ax.legend( ax.legend(
[Patch(facecolor=cmap(level)) for level in levels], [Patch(facecolor=cmap.to_rgba(level)) for level in levels],
['{}%'.format(level) for level in levels]) ['{}%'.format(level) for level in levels])
# ax.set_title('{} from {}'.format(m.meta['objid'], observer.name)) # ax.set_title('{} from {}'.format(m.meta['objid'], observer.name))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment