"""
multi_xp.py
====================================
Module to plot multiple XP spectra.
"""
import matplotlib.pyplot as plt
from gaiaxpy.core.satellite import BANDS
from .plotter import Plotter
[docs]
class MultiXpPlotter(Plotter):
def _plot_multi_xp(self):
show_legend = self.legend
spectra_df = self.spectra
spectra_class = self.spectra_class
max_flux = 0
fig, ax = plt.subplots(ncols=2, figsize=(16, 8))
# Set titles
ax[0].set_title(BANDS.bp.upper())
ax[1].set_title(BANDS.rp.upper())
ax[1].yaxis.set_label_position('right')
ax[1].yaxis.tick_right()
for spectrum in spectra_df.to_dict('records'):
x, y, e = self._get_inputs(spectrum)
max_flux = max(y) if max(y) > max_flux else max_flux
axis = BANDS.index(spectrum['xp'].lower())
ax[axis].plot(x, y, lw=2, alpha=0.95, label=spectrum['source_id'])
for axis in ax:
axis.set_ylim(0, 1.05 * max_flux)
axis.set_xlabel(spectra_class.get_position_label())
axis.set_ylabel(spectra_class.get_flux_label())
if show_legend:
handles, labels = plt.gca().get_legend_handles_labels()
by_label = dict(zip(labels, handles))
fig.subplots_adjust(top=0.95, bottom=0.05, left=0.05, right=0.85, wspace=0.020)
fig.legend(by_label.values(), by_label.keys(), bbox_to_anchor=(1, 1), loc='upper right',
borderaxespad=0.1)
[docs]
def plot_fig(self):
n_spectra = len(self.spectra)
if self.show_plot and self.legend and n_spectra > self.max_spectra_on_multi:
raise ValueError(f'The legend can only be shown for a list of spectra no longer than '
f'{self.max_spectra_on_multi} elements. Try setting legend to False or retry with a '
f'shorter list.')
self._plot_multi_xp()
if self.output_path or self.output_file:
self._save_figure()
if self.show_plot:
plt.show()
plt.close()