Changeset 1746 in lmdz_wrf

Timestamp:

Dec 17, 2017, 4:07:39 PM (8 years ago)

Author:

lfita

Message:

Working version of the script!!

File:

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

trunk/tools/ORforcing_reconstruct.py

@@ -25 +25 @@
 Xmin=-90.25
 #Xmin='all'
-Xmax=-5.25
+Xmax=-33.25
 Ymin=-67.25
 Ymax=15.25
@@ -31 +31 @@
 # Variables to reconstruct
 #variable = 'all'
-variable = 'LWdown'
+variable = 'time,LWdown,Qair,PSurf,SWdown,Tair,Wind_E,Wind_N,Snowf,Rainf'
 
 # Resolution
@@ -86 +86 @@
 veclat = np.zeros((len(indv)), dtype=np.float)
 
-# Construct veclon, veclat  (Fortran like, First 1)
+# Construct veclon, veclat
 for iid in range(len(indv)):
     iy = int((indv[iid]-1)/dimx)
@@ -107 +107 @@
     for vn in varns:
         if not gen.searchInlist(ncvars, vn):
-            print errormsg
+            print gen.errormsg
             print '  ' + fname + ": file '" + ncfile + "' does not have " +          \
               " variable '" + vn + "' !!"
@@ -138 +138 @@
 
 matind = matindt.transpose()
-
-print 'Lluis veclon:', veclon[0:10]
-print 'Lluis matind: ___'
-for i in range(3):
-    print matind[50+i,50:55]
-
-print 'matind n:', np.min(matind), 'x:', np.max(matind), 'L1Dvec:', veclon.shape[0], 'vecind: n', np.min(indv), 'x:', np.max(indv)
-print 'vecind[matind]:', indv[matind[50:53,0]]
-quit()
+# Fortran like, First 1
+matind = np.where(matind != -1, matind - 1, matind)
 
 # Creation of file
@@ -166 +159 @@
 
 # Variable indices
-newvar = onewnc.createVariable('lonlat_matind', 'i', ('y', 'x'), fill_value=-1)
+newvar = onewnc.createVariable('vec1D_matind', 'i', ('y', 'x'), fill_value=-1)
 newvar[:] = matind
-ncvar.basicvardef(newvar, 'lonlat_matind', 'matrix with the equivalencies from ' +   \
-  '2D lon, lat matrices', '-')
+ncvar.basicvardef(newvar, 'vec1D_matind', 'matrix with the equivalencies from 1D ' + \
+  'vector indices', '-')
+ncvar.set_attribute(newvar, 'coordinates', 'lon lat')
 
 # Looking for equivalencies in the 1D vector
-lind = list(indv)
-vec1Dind = matind
+matlonlat = matind.copy()
 for j in range(dimy):
     for i in range(dimx):
         if matind[j,i] != -1:
-            vec1Dind[j,i] = lind.index(matind[j,i])
-
-newvar = onewnc.createVariable('vec1D_matind', 'i', ('y', 'x'), fill_value=-1)
-newvar[:] = vec1Dind
-ncvar.basicvardef(newvar, 'vec1D_matind', 'matrix with the equivalencies from 1D ' + \
-  'vector indices', '-')
-
-onewnc.sync()
-onc.clsoe()
-onewnc.close()
-quit()
+            matlonlat[j,i] = indv[matind[j,i]]
+
+newvar = onewnc.createVariable('lonlat_matind', 'i', ('y', 'x'), fill_value=-1)
+newvar[:] = matlonlat
+ncvar.basicvardef(newvar, 'lonlat_matind', 'matrix with the equivalencies from ' +   \
+  '2D lon, lat matrices', '-')
+ncvar.set_attribute(newvar, 'coordinates', 'lon lat')
+onewnc.sync()
 
 # Getting variables
@@ -245 +235 @@
             iix = vardims.index('x')
             iiy = vardims.index('y')
+            iislc = 0
+            Nslices = len(slices)
             for slc in slices:
+                gen.percendone(iislc,Nslices,0.25,vn + ' reconstructed:')
                 ival = slc[iix]
                 jval = slc[iiy]
@@ -257 +250 @@
                             oldvarslice.append(slc[iid])
 
-                    print vn, 'Slice new:', slc, 'old:', oldvarslice, 'shape:', newvar.shape, 'oldshape:', ovar.shape
                     newvar[tuple(slc)] = ovar[tuple(oldvarslice)]
                 else: