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

Timestamp:

May 25, 2012, 10:37:34 AM (13 years ago)

Author:

tnavarro

Message:

python wrapper for concat and lslin + improvements (verbose,bugs)

File:

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

trunk/UTIL/PYTHON/gcm_transformations.py

@@ -10 +10 @@
                     fields  = 'all', \
                     limites = None, \
-                    predefined = None):
+                    predefined = None, \
+                    verbose = True):
 
     import numpy as np
@@ -21 +22 @@
     indicefile=0
     outputfilename=""
-    f = open('zrecast.auto.def', 'w')
-
-    for zfile in input_name:
+
+    for zfile in input_name:
+        f = open('zrecast.auto.def', 'w')
         f.write(zfile+"\n")
         for zvar in fields:
@@ -62 +63 @@
              exit()
         f.close()
-        system("zrecast.e < zrecast.auto.def")
+        if verbose:
+           system("zrecast.e < zrecast.auto.def")
+        else:
+           system("zrecast.e < zrecast.auto.def > /dev/null")
 
         if indicefile is 0:
@@ -77 +81 @@
 def call_hrecast (  input_name = None, \
                     fields  = 'all', \
-                    predefined = None):
+                    predefined = None,\
+                    verbose = True):
 
     import numpy as np
@@ -107 +112 @@
             f.write(str(lat)+"\n")
         f.close()
-        system("hrecast.e < hrecast.auto.def")
-
+        if verbose:
+           system("hrecast.e < hrecast.auto.def")
+        else:
+           system("hrecast.e < hrecast.auto.def > /dev/null")
+           
         if indicefile is 0:
               outputfilename=input_name[indicefile][0:len(input_name[0])-3]+append+".nc"
@@ -120 +128 @@
     return output_name
 
-def call_localtime (  input_name = None, \
-                    fields  = 'all', \
-                    times = None):
+def call_localtime (input_name = None, \
+                    fields  = 'all', \
+                    times = None, \
+                    verbose = True):
 
     import numpy as np
@@ -129 +138 @@
     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:
+    output_name = []
+    #print input_name
+    for zfile in input_name:
+        print 'LOCALTIME FOR FILE: '+zfile
+        f = open('localtime.auto.def', 'w')
+        #print zfile
         f.write(zfile+"\n")
         for zvar in fields:
-             f.write(zvar+"\n")
+           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
+        f.write(str(len(times))+"\n")
+        for t in times:
+           f.write(str(t)+"\n")
+        f.close()
+        if verbose:
+          system("localtime.e < localtime.auto.def")
+        else:
+          system("localtime.e < localtime.auto.def >  /dev/null ")
+        #print input_name
+        zeoutput_name = zfile.replace('.nc','_LT'+str(times[0])+'.nc')
+    
+        system("mv -f "+zfile.replace('.nc','_LT.nc')+" "+zeoutput_name)
+        
+        output_name = np.append(output_name,zeoutput_name)
+
+    return output_name
+    
 
 def call_streamfunction (  input_name = None):
@@ -181 +194 @@
     return output_name
 
+
+def call_concat (  input_name = None, \
+                    fields  = 'all', \
+                    begin = 0, \
+                    verbose = True):
+
+    import numpy as np
+    from os import system
+    system("rm -f concatnc.auto.def")
+    system("touch concatnc.auto.def")
+        
+    f = open('concatnc.auto.def', 'w')
+    for zfile in input_name:
+        f.write(zfile+"\n")
+    
+    f.write("\n")
+    f.write(str(begin)+"\n")
+    f.write("sol"+"\n")
+    
+    for zvar in fields:
+        f.write(zvar+"\n")
+    f.write("\n")
+
+    f.close()
+    if verbose:
+       system("concatnc.e < concatnc.auto.def")
+    else:
+       system("concatnc.e < concatnc.auto.def > /dev/null")
+    
+    
+    #output_name = 'concat' + input_name[0].replace('diagfi','').replace('.nc','') \
+    #              + '_' + input_name[len(input_name)-1].replace('diagfi','').replace('.nc','') + '.nc'
+
+    #system('mv -f concat.nc '+output_name)
+
+    output_name = 'concat.nc'
+
+    return [output_name]
+
+
+
+def call_lslin ( input_name = None, \
+                 begin = 0,\
+                 timestep  = None,\
+                 isconcat = True, \
+                 verbose = True):
+
+    import numpy as np
+    from os import system
+    system("rm -f lslin.auto.def")
+    system("touch lslin.auto.def")
+        
+    f = open('lslin.auto.def', 'w')
+    
+    f.write(input_name[0]+"\n")
+    if isconcat:
+       f.write("y"+"\n")
+    else:
+       f.write("n"+"\n")
+
+    f.write(str(begin)+"\n")
+    if timestep is None:
+       f.write("y"+"\n")
+    else:
+       f.write("n"+"\n")
+       f.write(str(timestep)+"\n")
+       
+       
+    f.close()
+    if verbose:
+       system("lslin.e < lslin.auto.def")
+    else:
+       system("lslin.e < lslin.auto.def > /dev/null")
+    
+    output_name = input_name[0].replace('.nc','_Ls.nc')
+
+    return [output_name]
+
+
+
+