<%inherit file="base.html"/> <% import matplotlib.pyplot as plt import os.path import numpy as np plt.figure() sourceid = context['sourceid'] spectrum = context['spectrum'] try: xaxis_data = spectrum.axis.data except: xaxis_data = np.arange(len(spectrum.data)) # sometimes xaxis is not monotonic xaxis_sort = np.argsort(xaxis_data) try: xaxis_title = spectrum.axis.title except: xaxis_title = None try: xaxis_units = spectrum.axis.units except: xaxis_units = None if spectrum.data.dtype==np.complex: plt.plot(xaxis_data[xaxis_sort], spectrum.data.real[xaxis_sort], color='blue') plt.plot(xaxis_data[xaxis_sort], spectrum.data.imag[xaxis_sort], color='red') plt.figtext(0.2, 0.25, 'real', color='blue') plt.figtext(0.2, 0.2, 'imag', color='red') else: plt.plot(xaxis_data[xaxis_sort], spectrum.data[xaxis_sort]) plt.title('Source %s' % sourceid) xlabel = ' '.join([v for v in [xaxis_title, xaxis_units] if v is not None]) plt.xlabel(xlabel) plt.savefig(os.path.join(context['dirname'], 'sourcespectrum%s.png' % sourceid)) plt.close() %>