Changeset 1817 in lmdz_wrf

Timestamp:

Mar 19, 2018, 7:04:49 PM (7 years ago)

Author:

lfita

Message:

Fixing `pinterp' on y-axis slices

File:

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

trunk/tools/nc_var_tools.py

@@ -18418 +18418 @@
     print '  ' + fname + ' interpolating at pressure levels _______' 
     print '     ', interplevs
+    tsaid = False
+    ysaid = False
     for vn in varns:
         print "      '" + vn + "' ..."
@@ -18520 +18522 @@
                 dimtfrac = 5
                 varinterp = np.zeros(tuple([dimx,dimy,len(interplevs),dimtfrac]), dtype=np.float)
-                print warnmsg
-                print '  ' + fname + ': variable to interpolate:', varin.shape,      \
-                  'too big!!'
-                print '    p-interpolation will be done by time-slices of 5 time-steps'
-                print range(0,dimtfrac*int(dimt/dimtfrac),dimtfrac)
+                if not tsaid:
+                    print warnmsg
+                    print '  ' + fname + ': variable to interpolate:', varin.shape,  \
+                      'too big!!'
+                    print '    p-interpolation will be done by time-slices of 5 time-steps'
+                    print range(0,dimtfrac*int(dimt/dimtfrac),dimtfrac)
             else:
                 dimtfrac = dimt
@@ -18533 +18536 @@
                 if np.prod([tend-tini+1, dimz, dimx, dimy]) > 150*150*39:
                     dimyfrac = 50
-                    print warnmsg
-                    print '  ' + fname + ': variable to interpolate:', varin.shape,      \
-                      'too big!!'
-                    print '    p-interpolation will be done by yaxis-slices of 10 grid-points'
-                    print range(0,dimyfrac*int(dimy/dimyfrac),dimyfrac)
+                    if not ysaid:
+                        print warnmsg
+                        print '  ' + fname + ': variable to interpolate:', varin.shape,\
+                          'too big!!'
+                        print '    p-interpolation will be done by yaxis-slices of', \
+                          dimyfrac,' 10 grid-points'
+                        print range(0,dimyfrac*int(dimy/dimyfrac),dimyfrac)
                     for iyy in range(0,dimyfrac*int(dimy/dimyfrac),dimyfrac):
                         yini = iyy
@@ -18557 +18562 @@
                         newvar[tini:tend,:,yini:yend,:] = varinterp.transpose()
                     if dimyfrac != dimy and yend != dimy-1:
-                        print '    finishing time-splitting:', yend, ', ', dimy
+                        if not ysaid:
+                            print '    finishing yaxis-splitting:', yend, ', ', dimy
+                            ysaid = True
                         varint = varin[tini:tend,:,yend:dimy,:].transpose()
                         prest = pres[tini:tend,:,yend:dimy,:].transpose()
@@ -18573 +18580 @@
                           ix=dimx, iy=Ndy, iz=dimz, it=Ndt, num_metgrid_levels=Nplevs)
         
-                        newvar[tend:dimt,:,yend:dimy,:] = varinterpyl.transpose()
+                        newvar[tini:tend,:,yend:dimy,:] = varinterpyl.transpose()
                 else:
                     varint = varin[tini:tend,:,:,:].transpose()
@@ -18591 +18598 @@
 
             if dimtfrac != dimt and tend != dimt-1:
-                print '    finishing time-splitting:', tend, ', ', dimt
+                if not tsaid:
+                    print '    finishing time-splitting:', tend, ', ', dimt
+                    tsaid = True
                 tini = tend
                 tend = dimt
                 if np.prod([dimt-tend+1, dimz, dimx, dimy]) > 150*150*39:
                     dimyfrac = 50
-                    print warnmsg
-                    print '  ' + fname + ': variable to interpolate:', varin.shape,      \
-                      'too big!!'
-                    print '    p-interpolation will be done by yaxis-slices of 10 grid-points'
-                    print range(0,dimyfrac*int(dimy/dimyfrac),dimyfrac)
                     for iyy in range(0,dimyfrac*int(dimy/dimyfrac),dimyfrac):
                         yini = iyy
@@ -18620 +18624 @@
                         newvar[tini:tend,:,yini:yend,:] = varinterp.transpose()
                     if dimyfrac != dimy and yend != dimy-1:
-                        print '    finishing time-splitting:', yend, ', ', dimy
                         varint = varin[tini:tend,:,yend:dimy,:].transpose()
                         prest = pres[tini:tend,:,yend:dimy,:].transpose()
@@ -18636 +18639 @@
                           ix=dimx, iy=Ndy, iz=dimz, it=Ndt, num_metgrid_levels=Nplevs)
         
-                        newvar[tend:dimt,:,yend:dimy,:] = varinterpyl.transpose()
+                        newvar[tini:tend,:,yend:dimy,:] = varinterpyl.transpose()
                 else:
                     varint = varin[tend:dimt,:,:,:].transpose()