Changeset 2293 in lmdz_wrf for trunk/tools

Timestamp:

Jan 28, 2019, 3:40:08 PM (6 years ago)

Author:

lfita

Message:

First version with slicing two variables !!

File:

• trunk/tools/nc_var_tools.py (modified) (2 diffs)

trunk/tools/nc_var_tools.py

@@ -28414 +28414 @@
     inpointsB = inpB.transpose()
 
+    # Remembering that it is python (C-like...)
+    inpointsA = inpointsA-1
+    inpointsB = inpointsB-1
+
     maxNpointsAB = np.max(NpointsAB)
     print "  Joined slice '" + newslcvarns[0] + "' & '" + newslcvarns[1] + "':",     \
@@ -28452 +28456 @@
                     Nin = NpointsAB[jB,iB,jA,iA]
                     innewvar[:,0:Nin,jB,iB,jA,iA] = pointsAB[:,0:Nin,jB,iB,jA,iA]
-                    print Nin, 'Lluis shapes sliceaA:', osliceaA.shape, 'sliceaB:', osliceaB.shape
+                    print Nin, 'Lluis shapes sliceaA:', osliceaA.shape, 'sliceaB:', osliceaB.shape, \
+                    'pnewvar:', pnewvar.shape, 'anewvar:', anewvar.shape
                     for iv in range(Nin):
-                        pA = oslicepA[inpA[jA,iA,iv]-1,jA,iA]
-                        pB = oslicepB[inpB[jB,iB,iv]-1,jB,iB]
+                        print 'Lluis jB, iB, jA, iA:', jB,iB,jA,iA
+                        print 'Lluis shapes inpointsA:', inpointsA.shape, 'inpointsB:', inpointsB.shape
+                        print 'Lluis inpointsA:', inpointsA[iv,jA,iA], 'inpointsB:', inpointsB[iv,jB,iB]
+                        pA = oslicepA[inpointsA[iv,jA,iA],jA,iA]
+                        pB = oslicepB[inpointsB[iv,jB,iB],jB,iB]
+                        print 'Lluis shapes pnewvar:', pnewvar.shape, 'pA:', pA, 'pB:', pB
                         pnewvar[iv,jB,iB,jA,iA]= pA*pB
-                        ixA = sliceinA[1,inpointsA[jA,iA,iv]-1,jA,iA]
-                        iyA = sliceinA[0,inpointsA[jA,iA,iv]-1,jA,iA]
+                        ixA = sliceinA[1,inpointsA[iv,jA,iA],jA,iA]
+                        iyA = sliceinA[0,inpointsA[iv,jA,iA],jA,iA]
                         print '  A:', ixA, iyA
                         aA = osliceaA[iyA,ixA]
-                        ixB = sliceinB[1,inpointsB[jB,iB,iv]-1,jB,iB]
-                        iyB = sliceinB[0,inpointsB[jB,iB,iv]-1,jB,iB]
+                        ixB = sliceinB[1,inpointsB[iv,jB,iB],jB,iB]
+                        iyB = sliceinB[0,inpointsB[iv,jB,iB],jB,iB]
                         print '  B:', ixB, iyB
                         aB = osliceaB[iyB,ixB]
-                        anewvar[jB,iB,jA,iA]= aA*aB
+                        anewvar[jB,iB,jA,iA]= (aA*pA)*(aB*pB)
 
     onewnc.sync()