Changeset 27

Timestamp:

May 20, 2014 3:20:17 PM (10 years ago)

Author:

JohnLightfoot

Message:

various improvements

File:

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

trunk/templates/skymodelplane.html

@@ -4 +4 @@
 <%
 import matplotlib.pyplot as plt
+import numpy as np
 import os.path
+import string
+
+_valid_chars = "_.%s%s" % (string.ascii_letters, string.digits)
+
+def _char_replacer(s):
+    '''A small utility function that echoes the argument or returns '_' if the
+    argument is in a list of forbidden characters.
+    '''
+    if s not in _valid_chars:
+        return '_'
+    return s
+
+def sanitize(text):
+    filename = ''.join(_char_replacer(c) for c in text)
+    return filename
+
 
 plt.figure()
 
 iwn = context['iwn']
-sky_image = context['data']['skygenerator']['sky model'][:,:,iwn]
-spatial_axis = context['data']['skygenerator']['spatial axis']
+stagename = context['stagename']
+resultname = context['resultname']
+print 'resultname', resultname
+sky_image = context['data'][stagename][resultname][:,:,iwn]
+spatial_axis = context['data'][stagename]['spatial axis']
+wn = context['wn'] 
 
-plt.imshow(sky_image, interpolation='nearest', origin='lower',
+plt.subplot(211)
+plt.imshow(sky_image.real, interpolation='nearest', origin='lower',
   aspect='equal', extent=[spatial_axis[0], spatial_axis[-1],
-  spatial_axis[0], spatial_axis[-1]])
+  spatial_axis[0], spatial_axis[-1]],
+  vmax=np.max(sky_image.real)*1.1,
+  vmin=np.min(sky_image.real)*0.9)
+plt.colorbar(orientation='vertical')
 plt.axis('image')
+plt.title('Sky model real %6.4gcm-1' % wn)
 
-wn = context['wn'] 
-plt.title('Sky model plane %6.4gcm-1' % wn)
+plt.subplot(212)
+plt.imshow(sky_image.imag, interpolation='nearest', origin='lower',
+  aspect='equal', extent=[spatial_axis[0], spatial_axis[-1],
+  spatial_axis[0], spatial_axis[-1]],
+  vmax=np.max(sky_image.imag)*1.1,
+  vmin=np.min(sky_image.imag)*0.9)
+plt.colorbar(orientation='vertical')
+plt.axis('image')
+plt.title('Sky model imag %6.4gcm-1' % wn)
 
-plt.colorbar(orientation='vertical')
-plt.savefig(os.path.join(context['dirname'], 'skymodelplane%s.png' % wn))
+filename = '%splane%s.png' % (resultname, wn)
+filename = sanitize(filename)
+filename = os.path.join(context['dirname'], filename)
+print 'filename', filename
+plt.savefig(filename)
 plt.close()
 %>
@@ -27 +63 @@
 <!-- link to the plot from html -->
 
-<img src='skymodelplane${wn}.png' alt="The sky model should appear here<br>">
+<img src='${os.path.basename(filename)}' alt="The sky model should appear here<br>">