Changeset 792 for trunk/UTIL/PYTHON/mymath.py

Timestamp:

Sep 21, 2012, 3:03:27 PM (12 years ago)

Author:

acolaitis

Message:

PYTHON GRAPHICS

• Added option --finddevil which finds the steepest pressure drop in the PSFC field of the file, and add a cross in it's position.

This new function is intended to provide lon and lat of a dust devil to planetoplot. This could be used to drive a slice plot at the corresponding location.

• Removed dumpbdy for LES files (no relaxation zone)
• Added --mark mode to tiled plot
• Added routines in mymath to convert back and forth between greyscale Image objects and 2D arrays

File:

• trunk/UTIL/PYTHON/mymath.py (modified) (3 diffs)

trunk/UTIL/PYTHON/mymath.py

@@ -229 +229 @@
     return idx
 
+# Author: A.C.
 def fig2data ( fig ):
     import numpy
@@ -248 +249 @@
     return buf
 
+# Author: A.C.
 def fig2img ( fig ):
     import Image
@@ -260 +262 @@
     w, h, d = buf.shape
     return Image.fromstring( "RGBA", ( w ,h ), buf.tostring( ) )
+
+# Author: A.C.
+# Convert a single layer image object (greyscale) to an array
+def image2array(im):
+    import numpy as np
+    if im.mode not in ("L", "F"):
+        raise ValueError, ("can only convert single-layer images", im.mode)
+    if im.mode == "L":
+        a = np.fromstring(im.tostring(), np.uint8)
+    else:
+        a = np.fromstring(im.tostring(), np.float32)
+    a.shape = im.size[1], im.size[0]
+    return a
+
+# Author: A.C.
+# Convert a 2D array to a single layer image object (greyscale)
+def array2image(a):
+    import numpy as np
+    import Image
+    if a.dtype == np.uint8:
+        mode = "L"
+    elif a.dtype == np.float32:
+        mode = "F"
+    else:
+        raise ValueError, "unsupported image mode"
+    return Image.fromstring(mode, (a.shape[1], a.shape[0]), a.tostring())
+