Changeset 1816 in lmdz_wrf

Timestamp:

Mar 19, 2018, 6:45:03 PM (7 years ago)

Author:

lfita

Message:

Adding y-axis slice in the final proper way...

File:

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

trunk/tools/nc_var_tools.py

@@ -18550 +18550 @@
                         Ndt = len(range(tini,tend))
                         Ndy = len(range(yini,yend))
-                        print 'Lluis before!!!!'
                         varinterp = fin.module_forinterpolate.interp( data_in=varint,\
                           pres_field=prest, interp_levels=interplevs, psfc=psfct,    \
@@ -18556 +18555 @@
                           extrapolate=extrap, geopt=isgeop, missing=gen.fillValueF,  \
                           ix=dimx, iy=Ndy, iz=dimz, it=Ndt, num_metgrid_levels=Nplevs)
-                        print 'Lluis done!!!!'
                         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()
+                        psfct = psfc[tini:tend,yend:dimy,:].transpose()
+                        hgtt = hgt[yend:dimy].transpose()
+                        tempt = temp[tini:tend,:,yend:dimy,:].transpose()
+                        qvt = qv[tini:tend,:,yend:dimy,:].transpose()
+                        Nplevs = len(interplevs)
+                        Ndt = len(range(tini,tend))
+                        Ndy = len(range(yend,dimy))
+                        varinterpyl= fin.module_forinterpolate.interp(data_in=varint,\
+                          pres_field=prest, interp_levels=interplevs, psfc=psfct,    \
+                          ter=hgtt, tk=tempt, qv=qvt, linlog=linloginterp,           \
+                          extrapolate=extrap, geopt=isgeop, missing=gen.fillValueF,  \
+                          ix=dimx, iy=Ndy, iz=dimz, it=Ndt, num_metgrid_levels=Nplevs)
+        
+                        newvar[tend:dimt,:,yend:dimy,:] = varinterpyl.transpose()
                 else:
                     varint = varin[tini:tend,:,:,:].transpose()
@@ -18567 +18583 @@
                     Nplevs = len(interplevs)
                     Ndt = len(range(tini,tend))
-                    print 'Lluis before!!!!'
                     varinterp = fin.module_forinterpolate.interp( data_in=varint,    \
                       pres_field=prest, interp_levels=interplevs, psfc=psfct,        \
@@ -18573 +18588 @@
                       extrapolate=extrap, geopt=isgeop, missing=gen.fillValueF,      \
                       ix=dimx, iy=dimy, iz=dimz, it=Ndt, num_metgrid_levels=Nplevs)
-                    print 'Lluis done!!!!'
                     newvar[tini:tend,:,:,:] = varinterp.transpose()
 
             if dimtfrac != dimt and tend != dimt-1:
                 print '    finishing time-splitting:', tend, ', ', dimt
-                varint = varin[tend:dimt,:,:,:].transpose()
-                prest = pres[tend:dimt,:,:,:].transpose()
-                psfct = psfc[tend:dimt,:,:].transpose()
-                hgtt = hgt.transpose()
-                tempt = temp[tend:dimt,:,:,:].transpose()
-                qvt = qv[tend:dimt,:,:,:].transpose()
-
-                varinterplast = fin.module_forinterpolate.interp( data_in=varint,    \
-                  pres_field=prest, interp_levels=interplevs, psfc=psfct,            \
-                  ter=hgtt, tk=tempt, qv=qvt, linlog=linloginterp,                   \
-                  extrapolate=extrap, geopt=isgeop, missing=gen.fillValueF)
-
-                newvar[tend:dimt,:,:,:] = varinterplast.transpose()
+                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
+                        yend = iyy + dimyfrac
+                        varint = varin[tini:tend,:,yini:yend,:].transpose()
+                        prest = pres[tini:tend,:,yini:yend,:].transpose()
+                        psfct = psfc[tini:tend,yini:yend,:].transpose()
+                        hgtt = hgt[yini:yend,:].transpose()
+                        tempt = temp[tini:tend,:,yini:yend,:].transpose()
+                        qvt = qv[tini:tend,:,yini:yend,:].transpose()
+                        Nplevs = len(interplevs)
+                        Ndt = len(range(tini,tend))
+                        Ndy = len(range(yini,yend))
+                        varinterp = fin.module_forinterpolate.interp( data_in=varint,\
+                          pres_field=prest, interp_levels=interplevs, psfc=psfct,    \
+                          ter=hgtt, tk=tempt, qv=qvt, linlog=linloginterp,           \
+                          extrapolate=extrap, geopt=isgeop, missing=gen.fillValueF,  \
+                          ix=dimx, iy=Ndy, iz=dimz, it=Ndt, num_metgrid_levels=Nplevs)
+                        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()
+                        psfct = psfc[tini:tend,yend:dimy,:].transpose()
+                        hgtt = hgt[yend:dimy].transpose()
+                        tempt = temp[tini:tend,:,yend:dimy,:].transpose()
+                        qvt = qv[tini:tend,:,yend:dimy,:].transpose()
+                        Nplevs = len(interplevs)
+                        Ndt = len(range(tini,tend))
+                        Ndy = len(range(yend,dimy))
+                        varinterpyl= fin.module_forinterpolate.interp(data_in=varint,\
+                          pres_field=prest, interp_levels=interplevs, psfc=psfct,    \
+                          ter=hgtt, tk=tempt, qv=qvt, linlog=linloginterp,           \
+                          extrapolate=extrap, geopt=isgeop, missing=gen.fillValueF,  \
+                          ix=dimx, iy=Ndy, iz=dimz, it=Ndt, num_metgrid_levels=Nplevs)
+        
+                        newvar[tend:dimt,:,yend:dimy,:] = varinterpyl.transpose()
+                else:
+                    varint = varin[tend:dimt,:,:,:].transpose()
+                    prest = pres[tend:dimt,:,:,:].transpose()
+                    psfct = psfc[tend:dimt,:,:].transpose()
+                    hgtt = hgt.transpose()
+                    tempt = temp[tend:dimt,:,:,:].transpose()
+                    qvt = qv[tend:dimt,:,:,:].transpose()
+
+                    varinterplast = fin.module_forinterpolate.interp( data_in=varint,\
+                      pres_field=prest, interp_levels=interplevs, psfc=psfct,        \
+                      ter=hgtt, tk=tempt, qv=qvt, linlog=linloginterp,               \
+                      extrapolate=extrap, geopt=isgeop, missing=gen.fillValueF)
+    
+                    newvar[tend:dimt,:,:,:] = varinterplast.transpose()
 
             for attrn in newvarattr.keys():