Changeset 466 for trunk/UTIL/PYTHON/gcm_transformations.py

Timestamp:

Dec 9, 2011, 5:10:15 PM (13 years ago)

Author:

acolaitis

Message:

Added localtime.e and streamfunction.e to the gcm wrapper.

File:

• trunk/UTIL/PYTHON/gcm_transformations.py (modified) (2 diffs)

trunk/UTIL/PYTHON/gcm_transformations.py

@@ -3 +3 @@
 ### A. Colaitis -- LMD -- 08/11/2011 <-- Added ZRECAST for altitude mode, and MCS and TES modes for Pressure
 ### A. Colaitis -- LMD -- 08/11/2011 <-- Added HRECAST for MCS and TES predefined grids
-###
-
-### TODO
-### add wrapper to streamfunction.e
-### add wrapper to localtime.e
+###                                      Added wrapper to streamfunction.e
+###                                      Added wrapper to localtime.e
 
 def call_zrecast (  interp_mode   = '4', \
@@ -119 +116 @@
     return output_name
 
+def call_localtime (  input_name = None, \
+                    fields  = 'all', \
+                    times = None):
+
+    import numpy as np
+    from myplot import separatenames
+    from os import system
+    system("rm -f localtime.auto.def")
+    system("touch localtime.auto.def")
+    indicefile=0
+    outputfilename=""
+    append="_LT"
+    f = open('localtime.auto.def', 'w')
+    for zfile in input_name:
+        f.write(zfile+"\n")
+        for zvar in fields:
+             f.write(zvar+"\n")
+        f.write("\n")
+    f.write(str(len(times))+"\n")
+    for t in times:
+        f.write(str(t)+"\n")
+    f.close()
+    system("localtime.e < localtime.auto.def")
+
+    if indicefile is 0:
+          outputfilename=input_name[indicefile][0:len(input_name[0])-3]+append+".nc"
+    else:
+          outputfilename=outputfilename+","+input_name[indicefile][0:len(input_name[0])-3]+append+".nc"
+    indicefile = indicefile +1
+
+    for i in range(len(outputfilename)):
+        output_name = separatenames(outputfilename)
+
+    return output_name
+
+def call_streamfunction (  input_name = None):
+    import numpy as np
+    from myplot import separatenames
+    from os import system
+    system("rm -f streamfunction.auto.def")
+    system("touch streamfunction.auto.def")
+    indicefile=0
+    outputfilename=""
+    append="_stream"
+    f = open('streamfunction.auto.def', 'w')
+    for zfile in input_name:
+        f.write(zfile+"\n")
+    f.close()
+    system("streamfunction.e < streamfunction.auto.def")
+
+    if indicefile is 0:
+          outputfilename=input_name[indicefile][0:len(input_name[0])-3]+append+".nc"
+    else:
+          outputfilename=outputfilename+","+input_name[indicefile][0:len(input_name[0])-3]+append+".nc"
+    indicefile = indicefile +1
+
+    for i in range(len(outputfilename)):
+        output_name = separatenames(outputfilename)
+
+    return output_name
+