Changeset 2271 in lmdz_wrf for trunk

Timestamp:

Dec 27, 2018, 8:46:52 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `area_weighted': Function to provide an area weighted statistics within two different values of matrices assuming regular lon/lat projections without taking into account Earth's curvature

Location:

trunk/tools

Files:

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

trunk/tools/module_scientific.f90

@@ -5025 +5025 @@
     INTEGER                                              :: ix, iy, iu, minvalv
     LOGICAL                                              :: single
+    REAL(r_k), ALLOCATABLE, DIMENSION(:)                 :: uniques
+
 
 !!!!!!! Variables
@@ -5054 +5056 @@
       END DO
     END DO
-
-    CALL sortR_K(unique, Nunique)
+    IF (ALLOCATED(uniques)) DEALLOCATE(uniques)
+    ALLOCATE(uniques(Nunique)) 
+    uniques(1:Nunique) = unique(1:Nunique)
+    
+    CALL sortR_K(uniques(1:Nunique), Nunique)
+    unique(1:Nunique) = uniques(1:Nunique)
 
   END SUBROUTINE unique_matrixRK2D
@@ -5165 +5171 @@
         END DO
       CASE('count')
-        CALL unique_matrixRK2D(dxA, dyA, dxA*dyA, varin, Ncounts, icounts)
+        CALL unique_matrixRK2D(dxB, dyB, dxB*dyB, varin, Ncounts, icounts)
         IF (Lstats /= Ncounts) THEN
+          PRINT *,'  ' // TRIM(fname) // 'provided:', Lstats
+          PRINT *,'  ' // TRIM(fname) // 'found:', Ncounts, ' :', icounts
           WRITE(val1S,'(I3)')Lstats
           WRITE(val2S,'(I3)')Ncounts

trunk/tools/nc_var.py

@@ -67 +67 @@
 ## e.g. # nc_var.py -o compute_slices_stats -S 'XLONG,-74.,-36.,4.;XLAT,-62.,18.,2.;HGT,500.,7000.,500.@Time|Times:west_east|XLONG:south_north|XLAT@Time' -f wrfout_d01_1995-01-01_00:00:00 -v T2,Q2
 ## e.g. # nc_var.py -o dimrename -f /media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out_irene/simcdx_vars_cape_120lev_cdxwrf2.nc -S Time -v time0
+## e.g. # nc_var.py -o area_weighted -f 'reference_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1,get_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1' -S 'yes:min,max,mean,stddev,count' -v ct_values,xband_values,box_values,mosaic_values
 
 from optparse import OptionParser
@@ -80 +81 @@
 # addVar: Operation to add a new variable in a file using exisintg dimensions
 # addvals: Function to add values to a given variable at a given dimension
+# area_weighted: Function to provide an area weighted statistics within two different values 
+#   of matrices assuming regular lon/lat projections without taking into account Earth's curvature
 # CDO_toCF: Function to pass a CDO output file to CF-conventions
 # CFmorzization: Function to provide a CF-compilation version of a variable within a file
@@ -209 +212 @@
 # WRF_to_newCF: Function to pass a WRF original file to new CF-conventions file
 
-operations=['addDim', 'addVar', 'addvals', 'CDO_toCF', 'CFmorzization',              \
-  'chdimname', 'changevartype',                                                      \
+operations=['addDim', 'addVar', 'addvals', 'area_weighted', 'CDO_toCF',              \
+  'CFmorzization', 'chdimname', 'changevartype',                                     \
   'checkallvars', 'checkAllValues', 'checkNaNs',                                     \
   'chgtimestep', 'chvarname', 'cleaning_varsfile', 'compute_deaccum',                \
@@ -291 +294 @@
 
 # Operations which file name is not a real file
-NotCheckingFile = ['DatesFiles', 'compute_opersvarsfiles', 'file_creation',          \
+NotCheckingFile = ['area_weighted', 'DatesFiles', 'compute_opersvarsfiles',          \
+  'file_creation',                                                                   \
   'join_singlestation_obsfiles', 'join_sounding_obsfiles', 'list_operations',        \
   'merge_files',                                                                     \
@@ -327 +331 @@
 elif oper == 'addvals':
     ncvar.addvals(opts.values, opts.ncfile, opts.varname)
+elif oper == 'area_weighted':
+    ncvar.area_weighted(opts.values, opts.ncfile, opts.varname)
 elif oper == 'CDO_toCF':
     ncvar.CDO_toCF(opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -28 +28 @@
 # addVar: Operation to add a new variable in a file using exisintg dimensions
 # addvals: Function to add values to a given variable at a given dimension
+# area_weighted: Function to provide an area weighted statistics within two different values 
+#   of matrices assuming regular lon/lat projections without taking into account Earth's curvature
 # attrinvar: Check existence of an attribute in a netCDF variable object
 # basicvardef: Function to give the basic attributes to a variable (std_name, long_name, units)
@@ -27222 +27224 @@
         quit()
 
-    statsavail = ['mean', 'min', 'max', 'mean2', 'stddev']
+    statsavail = ['mean', 'min', 'max', 'mean2', 'stddev', 'count']
     # Characteristics of the statistics
     statnschars = {'mean': ['spcmean', 'spatial mean', 1],                           \
@@ -27238 +27240 @@
 
     for stn in statns:
-        if not gen.searchInlist(statsavail, stn):
+        if not gen.searchInlist(statnschars.keys(), stn):
             print errormsg
             print '  ' + fname + ": statistics '" + stn + "' not ready !!"
-            print '    available ones:', statnschar.keys()
+            print '    available ones:', statsavail
             quit(-1)
 
@@ -27574 +27576 @@
             varns.append(vn)
 
+    varrm = [refvarxn, refvaryn, refvarxbndsn, refvarybndsn, getvarxn, getvaryn,     \
+      getvarxbndsn, getvarybndsn]
+    for vn in varrm: 
+        if gen.searchInlist(varns, vn): varns.remove(vn)
+
     for vn in varns:
         print '  ' + fname + ": using '" + vn + "' ..."
@@ -27587 +27594 @@
 
             if stn == 'count': 
-                svals = np.unique(varin)
+                svals = np.unique(varvals)
                 statchar[2] = len(svals)
 
-            varout = fsci.module_scientific.spaceweightstats(varin=varvalst,         \
+            varoutt = fsci.module_scientific.spaceweightstats(varin=varvalst,        \
               ngridsin=Ngridsint, gridsin=gridsint, percentages=percenst,            \
               stats=stn, dxa=refdx, dya=refdy, dxb=getdx, dyb=getdy,                 \
               maxngridsin=Ngridsinmax, lstats=statchar[2])
 
-            varout = varout.transpose()
+            varout = varoutt.transpose()
             if statchar[2] == 1:
                 newvar = onewnc.createVariable(vn + statchar[0], 'f', ('lat','lon'))
@@ -27601 +27608 @@
             else:
                 if stn == 'count':
-                    if not gen.searchInlist(onewnc.dimensions, 'count'): 
-                        newdim = onewnc.createDimension('count', statchar[2])
-                        newvar = onewnc.createVariable(vn + 'count', 'f', ('count'))
+                    if not gen.searchInlist(onewnc.dimensions, vn+'count'): 
+                        newdim = onewnc.createDimension(vn+'count', statchar[2])
+                        newvar = onewnc.createVariable(vn+'count', 'f', (vn+'count'))
                         newvar[:] = svals
                         basicvardef(newvar, vn+'count', 'unique .get. values of ' +  \
                           vn, ovar.units)
 
-                    newvar = onewnc.createVariable(vn + statchar[0], 'f', ('count',  \
+                    newvar = onewnc.createVariable(vn+statchar[0], 'f', (vn+'count', \
                       'lat','lon'))
                     newvar[:] = varout[:]
@@ -27627 +27634 @@
     return
 
-fvals= 'reference_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1,get_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1'
-area_weighted('yes:min,max,mean,stddev',fvals,'ct_values')
+#fvals= 'reference_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1,get_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1'
+#area_weighted('yes:min,max,mean,stddev',fvals,'ct_values')