Changeset 50

Timestamp:

Sep 19, 2014 4:53:31 PM (10 years ago)

Author:

JohnLightfoot

Message:

almost working

File:

• trunk/templates/sourcespectrum.html (modified) (2 diffs)

trunk/templates/sourcespectrum.html

@@ -11 +11 @@
 sourceid = context['sourceid']
 spectrum = context['spectrum']
-print 'spectrum', np.shape(spectrum.data)
 
 try:
     xaxis_data = spectrum.axis.data
-    print 'axis', len(xaxis_data)
 except:
     xaxis_data = np.arange(len(spectrum.data))
+
+# sometimes xaxis is not monotonic
+xaxis_sort = np.argsort(xaxis_data)
 
 try:
@@ -29 +30 @@
     xaxis_units = None
     
-plt.plot(xaxis_data, spectrum.data)
+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])