Changeset 2269 in lmdz_wrf for trunk/tools

Timestamp:

Dec 27, 2018, 3:14:51 PM (6 years ago)

Author:

lfita

Message:

Getting more closer!

Location:

trunk/tools

Files:

• module_scientific.f90 (modified) (7 diffs)
• nc_var_tools.py (modified) (5 diffs)

trunk/tools/module_scientific.f90

@@ -57 +57 @@
 ! sort_polygon: Subroutine to sort a polygon using its center as average of the coordinates and remove duplicates
 ! SortR_K*: Subroutine receives an array x() r_K and sorts it into ascending order.
+! spacepercen: Subroutine to compute the space-percentages of a series of grid cells (B) into another series of grid-cells (A)
+! spaceweightstats: Subroutine to compute an spatial statistics value from a matrix B into a matrix A using weights
 ! StatsR_K: Subroutine to provide the minmum, maximum, mean, the quadratic mean, and the standard deviation of a 
 !   series of r_k numbers
@@ -4932 +4934 @@
   END SUBROUTINE spacepercen_within_reg
 
-  SUBROUTINE spacepercen(dxA, dyA, xCAvals, yCAvals, NAvertexmax, xBAvals, yBAvals, dxB, dyB, xCBvals,&
-    yCBvals, NBvertexmax, xBBvals, yBBvals, dxyB, strict, Ngridsin, gridsin, percentages)
+  SUBROUTINE spacepercen(xCAvals, yCAvals, xBAvals, yBAvals, xCBvals, yCBvals, xBBvals, yBBvals,      &
+    dxA, dyA, NAvertexmax, dxB, dyB, dxyB, NBvertexmax, strict, Ngridsin, gridsin, percentages)
   ! Subroutine to compute the space-percentages of a series of grid cells (B) into another series of 
   !   grid-cells (A)
@@ -4941 +4943 @@
     IMPLICIT NONE
 
-    INTEGER, INTENT(in)                                  :: dxA, dyA, NAvertexmax, dxB, dyB,          &
-      NBvertexmax, dxyB
+    INTEGER, INTENT(in)                                  :: dxA, dyA, NAvertexmax
+    INTEGER, INTENT(in)                                  :: dxB, dyB, NBvertexmax, dxyB
     REAL(r_k), DIMENSION(dxA,dyA), INTENT(in)            :: xCAvals, yCAvals
     REAL(r_k), DIMENSION(dxB,dyB), INTENT(in)            :: xCBvals, yCBvals
@@ -4954 +4956 @@
 ! Local
    INTEGER                                               :: iv, ix, iy
-   INTEGER                                               :: Nvertex, NvertexAgrid, Ncoin, Nsort
+   INTEGER                                               :: Nvertex
    INTEGER, ALLOCATABLE, DIMENSION(:,:)                  :: poinsin
    CHARACTER(len=20)                                     :: DS
-   REAL(r_k)                                             :: areapoly, areagpoly, totarea, totpercent
    REAL(r_k), ALLOCATABLE, DIMENSION(:,:)                :: vertexgrid
 
@@ -4982 +4983 @@
       Nvertex = 0
       DO iv=1, NAvertexmax
-        IF (xBAvals(ix,iy,iv) /= fillvalI) THEN
+        IF (xBAvals(ix,iy,iv) /= fillval64) THEN
           Nvertex = Nvertex + 1
         END IF
@@ -4993 +4994 @@
       CALL grid_within_polygon(Nvertex, vertexgrid, dxB, dyB, dxB*dyB, xCBvals, yCBvals, NBvertexmax, &
         xBBvals, yBBvals, Ngridsin(ix,iy), gridsin(ix,iy,:,:))
+   
       IF (ALLOCATED(poinsin)) DEALLOCATE(poinsin)
       ALLOCATE(poinsin(Ngridsin(ix,iy),2))
@@ -5004 +5006 @@
         Ngridsin(ix,iy), poinsin, strict, percentages(ix,iy,:))
 
-      END DO
     END DO
+    END DO
 
   END SUBROUTINE spacepercen
 
+  SUBROUTINE spaceweightstats(varin, Ngridsin, gridsin, percentages, stats, varout, dxA, dyA, dxB,    &
+    dyB, maxNgridsin, Lstats)
+  ! Subroutine to compute an spatial statistics value from a matrix B into a matrix A using weights
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: dxA, dyA, dxB, dyB, maxNgridsin, Lstats
+    CHARACTER(len=60), INTENT(in)                        :: stats
+    INTEGER, DIMENSION(dxA,dyA), INTENT(in)              :: Ngridsin
+    INTEGER, DIMENSION(dxA,dyA,maxNgridsin,2), INTENT(in):: gridsin
+    REAL(r_k), DIMENSION(dxB,dyB), INTENT(out)           :: varin
+    REAL(r_k), DIMENSION(dxA,dyA,maxNgridsin), INTENT(in):: percentages
+    REAL(r_k), DIMENSION(dxA,dyA,Lstats), INTENT(out)    :: varout
+
+! Local
+    INTEGER                                              :: ix, iy, iv, ii, jj
+
+!!!!!!! Variables
+! dxA, dyA: length of dimensions of matrix A
+! dxB, dyB: length of dimensions of matrix B
+! maxNgridsin: maximum number of grid points from B to be used to compute into a grid of matrix A
+! Lstats: length of the dimension of the statistics
+! varin: variable from matrix B to be used
+! Ngridsin: number of grids from matrix B for each grid of matrix A
+! gridsin: coordinates of grids of matrix B for each grid of matrix A
+! percentages: weights as percentages of space of grid in matrix A covered by grid of matrix B
+! stats: name of the spatial statistics to compute inside each grid of matrix A using values from 
+!     matrix B. Avaialbe ones:
+!   'min': minimum value
+!   'max': maximum value
+!   'mean': mean value
+!   'mean2': quadratic mean value
+!   'stddev': standard deviation value
+!   'count': percentage of the space of matrix A covered by each different value of matrix B
+! varout: output statistical variable
+
+    fname = 'spaceweightstats'
+
+    ! Let's be efficvient?
+    statn: SELECT CASE(TRIM(stats))
+      CASE('min')
+        varout = fillVal64
+        DO ix=1, dxA
+          DO iy=1, dyA
+            DO iv=1, Ngridsin(ix,iy)
+              ii = gridsin(ix,iy,iv,1)
+              jj = gridsin(ix,iy,iv,2)
+              IF (varin(ii,jj) < varout(ix,iy,Lstats)) varout(ix,iy,1) = varin(ii,jj)
+            END DO
+          END DO
+        END DO
+      CASE DEFAULT
+        msg = "statisitcs '" // TRIM(stats) // "' not ready !!" // CHAR(44) // " available ones: " // &
+          "'min', 'max', 'mean', 'mean2', 'stddev', 'count'" 
+        CALL ErrMsg(msg, fname, -1)
+    END SELECT statn
+
+  END SUBROUTINE spaceweightstats
+
 END MODULE module_scientific

trunk/tools/nc_var_tools.py

@@ -27351 +27351 @@
             refdx = reflon.shape[1]
             refdy = reflon.shape[0]
-            refmaxvert = refvarbndsx.shape[2]
+            refmaxvert = refvarbndsx.shape[0]
 
         else:
@@ -27441 +27441 @@
             ovarxbnds = oncget.variables[getvarxbndsn]
             ovary = oncget.variables[getvaryn]
+            ovarybnds = oncget.variables[getvarybndsn]
 
             sgetvarx, dims = SliceVarDict(ovarx,slicedict)
@@ -27452 +27453 @@
             getdx = getlon.shape[1]
             getdy = getlon.shape[0]
-            getmaxvert = getvarbndsx.shape[2]
+            getmaxvert = getvarbndsx.shape[0]
 
         iif = iif + 1
@@ -27465 +27466 @@
     getvarybndst = getvarbndsy.transpose()
 
-    Ngridsint, gridsint, percenst = fsci.module_scientific.spacepercen(dxa=refdx,    \
-      dya=refdy, xcavals=reflont, ycavals=reflatt, navertexmax=refmaxvert,           \
-      xbavals=refvarxbndst, ybavals=refvarybndst,                                    \
-      dxb=getdx, dyb=getdy, dxyb=getdx*getdy, xcbvals=getlont, ycbvals=getlatt,      \
-      nbvertexmax=getmaxvert, xbbvals=getvarxbndst, ybbvals=getvarybndst,            \
-      strict=strict)
-
-    Ngridsin = Ngridint.transpose()
-    gridsin = gridint.transpose()
+    Ngridsint, gridsint, percenst = fsci.module_scientific.spacepercen(              \
+      xcavals=reflont, ycavals=reflatt, xbavals=refvarxbndst, ybavals=refvarybndst,  \
+      xcbvals=getlont, ycbvals=getlatt, xbbvals=getvarxbndst, ybbvals=getvarybndst,  \
+      strict=strict, dxa=refdx, dya=refdy, navertexmax=refmaxvert,                   \
+      dxb=getdx, dyb=getdy, dxyb=getdx*getdy, nbvertexmax=getmaxvert)
+
+    Ngridsin = Ngridsint.transpose()
+    gridsin = gridsint.transpose()
     percens = percenst.transpose()
 
@@ -27484 +27484 @@
     newdim = onewnc.createDimension('lat',refdy)
     newdim = onewnc.createDimension('bnds',refmaxvert)
-    newdim = onewnc.createDimension('gridin',np.max(gridsin))
-    newdim = onewnc.createDimension('coords',2)
+    newdim = onewnc.createDimension('gridin',np.max(Ngridsin))
+    newdim = onewnc.createDimension('coord',2)
 
     # variable-dimensions
     newvar = onewnc.createVariable('lon','f8',('lat','lon'))
-    newvar[:] = reflon
+    newvar[:] = reflon[:]
     basicvardef(newvar,'longitude','Longitude','degrees_east')
-    newavr.setncattr('bounds','lon_bnds')
+    newvar.setncattr('bounds','lon_bnds')
 
     newvar = onewnc.createVariable('lon_bnds','f8',('bnds', 'lat','lon'))
-    newvar[:] = refvarxbnds
+    newvar[:] = refvarbndsx[:]
     basicvardef(newvar,'longitude_bnds','Bounds of Longitude','degrees_east')
 
     newvar = onewnc.createVariable('lat','f8',('lat','lon'))
-    newvar[:] = reflat
+    newvar[:] = reflat[:]
     basicvardef(newvar,'latitude','Latitude','degrees_north')
-    newavr.setncattr('bounds','lat_bnds')
+    newvar.setncattr('bounds','lat_bnds')
 
     newvar = onewnc.createVariable('lat_bnds','f8',('bnds', 'lat','lon'))
-    newvar[:] = refvarybnds
+    newvar[:] = refvarbndsy[:]
     basicvardef(newvar,'latitude_bnds','Bounds of Latitude','degrees_north')
 
     # variables space-weight
-    newvar = onewnc.createVariable('Ngrid','i',('gridin', 'lat','lon'))
-    newvar[:] = Ngridsin
+    newvar = onewnc.createVariable('Ngrid','i',('lat','lon'))
+    newvar[:] = Ngridsin[:]
     basicvardef(newvar, 'Ngrid', "number of grids cells grom 'get' laying within" +  \
       " 'ref'",'-')
-    newavr.setncattr('coordinates','lon lat')
-
-    newvar = onewnc.createVariable('gridin','i',('coord', 'gridin', 'lat', 'lon'))
-    newvar[:] = Ngridsin
-    basicvardef(newvar,'gridin',"coordinates of the grids cells grom 'get' laying "+ \
+    newvar.setncattr('coordinates','lon lat')
+
+    innewvar = onewnc.createVariable('gridin','i',('coord', 'gridin', 'lat', 'lon'))
+    basicvardef(innewvar,'gridin',"coordinates of the grids cells grom 'get' laying "+ \
       "within 'ref'",'-')
-    newavr.setncattr('coordinates','lon lat')
-
-    newvar = onewnc.createVariable('gridpercen','i',('coord', 'gridin', 'lat', 'lon'))
-    newvar[:] = percens
-    basicvardef(newvar,'gridpercen',"percentages of the grids cells grom 'get' " +   \
+    innewvar.setncattr('coordinates','lon lat')
+
+    pnewvar = onewnc.createVariable('gridpercen','f',('gridin', 'lat', 'lon'))
+    basicvardef(pnewvar,'gridpercen',"percentages of the grids cells grom 'get' " +   \
       "laying within 'ref'", '1')
-    newavr.setncattr('coordinates','lon lat')
+    pnewvar.setncattr('coordinates','lon lat')
+    for j in range(refdy):
+        for i in range(refdx):
+            innewvar[:,0:Ngridsin[j,i],j,i] = gridsin[:,0:Ngridsin[j,i],j,i]
+            pnewvar[0:Ngridsin[j,i],j,i] = percens[0:Ngridsin[j,i],j,i]
+    onewnc.sync()
 
     # Getting values
+    if variable == 'all':
+        variables = oncget.variables.keys()
+    else:
+        if variable.count(',') == 0: variables = [variable]
+        else: variables = variable.split(',')
+
+    # Removing variables
+    varns = ''
+    for vn in variables:
+        if not oncget.variables.has_key(vn):
+            print errormsg
+            print '  ' + fname + ": get file '" + netcdfgetn + "' does not have " +  \
+              "variable '" + vn + "' !!"
+            varns0 = list(oncget.variables.keys())
+            varns0.sort()
+            print '    available ones:', varns0
+            quit(-1)
+
+        ovar = oncget.variables[vn]
+        vardims = ovar.dimensions
+        if not searchInlist(vardims, getdimxn) or searchInlist(vardims, getdimyn):
+            print '  ' + fname + ": get variable '" + vn + "' does not have space "+ \
+              " dimensions '" + getdimxn + "', '" +  getdimxn + "' not using it !!"
+            print "    variables' dimensions:", vardims
+        else:
+            varns.append(vn)
+
+    for vn in varns:
+        ovar = onget.variables[vn]
+        
 
     add_global_PyNCplot(onewnc, 'nc_var_tools', fname, '1.0')
+
+    print fname + ": Successfull witting of file '" + ofilen + "' !!"
 
     return