Changeset 1745 in lmdz_wrf

Timestamp:

Dec 17, 2017, 2:14:25 PM (8 years ago)

Author:

lfita

Message:

Working on make it fine

File:

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

trunk/tools/ORforcing_reconstruct.py

@@ -74 +74 @@
 Ydimn = olon.dimensions[0]
 
+# Fortran indices, first 1
 indv = oind[:]
 lonv = olon[:]
 latv = olat[:]
+indv = indv - 1
 
 dimx = lonv.shape[1]
@@ -84 +86 @@
 veclat = np.zeros((len(indv)), dtype=np.float)
 
-# Constuct veclon, veclat
+# Construct veclon, veclat  (Fortran like, First 1)
 for iid in range(len(indv)):
-    iy = int(indv[iid]/dimx)
-    ix = indv[iid] - iy*dimx
+    iy = int((indv[iid]-1)/dimx)
+    ix = indv[iid] - iy*dimx - 1
     veclon[iid] = lonv[iy,ix]
     veclat[iid] = latv[iy,ix]
@@ -137 +139 @@
 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()
+
 # Creation of file
 onewnc = NetCDFFile(ofilen, 'w')
@@ -153 +164 @@
 ncvar.basicvardef(newvar, 'lat', 'Latitude', 'degrees_north')   
 onewnc.sync()
+
+# Variable indices
+newvar = onewnc.createVariable('lonlat_matind', 'i', ('y', 'x'), fill_value=-1)
+newvar[:] = matind
+ncvar.basicvardef(newvar, 'lonlat_matind', 'matrix with the equivalencies from ' +   \
+  '2D lon, lat matrices', '-')
+
+# Looking for equivalencies in the 1D vector
+lind = list(indv)
+vec1Dind = matind
+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()
 
 # Getting variables