Changeset 1750 in lmdz_wrf

Timestamp:

Dec 18, 2017, 3:02:22 AM (7 years ago)

Author:

lfita

Message:

Final working version of the script with multiple attributes

File:

• trunk/tools/ORforcing_reconstruct.py (modified) (7 diffs)

trunk/tools/ORforcing_reconstruct.py

@@ -15 +15 @@
 
 parser = OptionParser()
+parser.add_option("-f", "--fileHEader", dest="fh", help="header of files",           \
+  metavar="VALUE")
+parser.add_option("-L", "--latitude", dest="latn", help="name of the variable latitiude", \
+  metavar="VALUE")
+parser.add_option("-l", "--longitude", dest="lonn", help="name of the variable longitiude", \
+  metavar="VALUE")
+parser.add_option("-v", "--Variables", dest="varns", help="',' separated list of variables", \
+  metavar="VALUE")
 parser.add_option("-y", "--year", dest="year", help="year to process",               \
   metavar="VALUE")
@@ -22 +30 @@
 ####### ###### ##### #### ### ## #
 
-filen = 'cruncep_halfdeg_' + opts.year + '.nc'
+filen = opts.fh + opts.year + '.nc'
 
 # Variable whcih provides the indices of a 1D vector from the dimy, dimx space
@@ -28 +36 @@
 
 # 2D longitude, latitude matrices
-lonvar = 'nav_lon'
-latvar = 'nav_lat'
+lonvar = opts.lonn
+latvar = opts.latn
 
 # Range to retrieve
@@ -40 +48 @@
 # Variables to reconstruct
 #variable = 'all'
-variable = 'time,LWdown,Qair,PSurf,SWdown,Tair,Wind_E,Wind_N,Snowf,Rainf'
+variable = opts.varns
 
 # Resolution
@@ -79 +87 @@
 olat = onc.variables[latvar]
 
-vecdimn = oind.dimensions[0]
-Xdimn = olon.dimensions[1]
-Ydimn = olon.dimensions[0]
-
-# Fortran indices, first 1
 indv = oind[:]
 lonv = olon[:]
 latv = olat[:]
+
+vecdimn = oind.dimensions[0]
+if len(olon.dimensions) == 2:
+    Xdimn = olon.dimensions[1]
+    Ydimn = olon.dimensions[0]
+    dimx = lonv.shape[1]
+    dimy = lonv.shape[0]
+elif len(olon.dimensions) == 1:
+    Xdimn = olon.dimensions[0]
+    Ydimn = olat.dimensions[0]
+    dimx = lonv.shape[0]
+    dimy = latv.shape[0]
+    lonv, latv = np.meshgrid(lonv, latv)
+
+# Fortran indices, first 1
 indv = indv - 1
-
-dimx = lonv.shape[1]
-dimy = lonv.shape[0]
 
 veclon = np.zeros((len(indv)), dtype=np.float)
@@ -98 +113 @@
 for iid in range(len(indv)):
     iy = int((indv[iid]-1)/dimx)
-    ix = indv[iid] - iy*dimx - 1
+    ix = int(indv[iid] - iy*dimx - 1)
     veclon[iid] = lonv[iy,ix]
     veclat[iid] = latv[iy,ix]
@@ -117 +132 @@
         if not gen.searchInlist(ncvars, vn):
             print gen.errormsg
-            print '  ' + fname + ": file '" + ncfile + "' does not have " +          \
+            print '  ' + fname + ": file '" + filen + "' does not have " +           \
               " variable '" + vn + "' !!"
             print '  available ones:', ncvars