Changeset 1371 in lmdz_wrf

Timestamp:

Dec 6, 2016, 11:28:51 AM (8 years ago)

Author:

lfita

Message:

Adding:

`smean': spatial-weighted mean values
making use of `gen.Str_Bool' on scratch, add definitions

File:

• trunk/tools/model_graphics.py (modified) (12 diffs)

trunk/tools/model_graphics.py

@@ -119 +119 @@
     fname = 'scratches'
 
-    if config['scratch'] == 'true':
+    if gen.Str_Bool(config['scratch']):
         scr = True
         print warnmsg
@@ -134 +134 @@
     else:
         scr = False
-        if config['filescratch'] == 'true':
+        if gen.Str_Bool(config['filescratch']):
             filescr = True
             print warnmsg
@@ -143 +143 @@
             filescr = False
 
-        if config['figscratch'] == 'true':
+        if gen.Str_Bool(config['figscratch']):
             figscr = True
             print warnmsg
@@ -152 +152 @@
             figscr = False
 
-        if config['diffscratch'] == 'true':
+        if gen.Str_Bool(config['diffscratch']):
             difscr = True
             print warnmsg
@@ -161 +161 @@
             difscr = False
 
-        if config['figdiffscratch'] == 'true':
+        if gen.Str_Bool(config['figdiffscratch']):
             figdifscr = True
             print warnmsg
@@ -170 +170 @@
             figdifscr = False
 
-        if config['moddiffscratch'] == 'true':
+        if gen.Str_Bool(config['moddiffscratch']):
             moddifscr = True
             print warnmsg
@@ -179 +179 @@
             moddifscr = False
 
-        if config['figmoddiffscratch'] == 'true':
+        if gen.Str_Bool(config['figmoddiffscratch']):
             figmoddifscr = True
             print warnmsg
@@ -188 +188 @@
             figmoddifscr = False
 
-        if config['figallmodexpscratch'] == 'true':
+        if gen.Str_Bool(config['figallmodexpscratch']):
             figallmodexpscr = True
             print warnmsg
@@ -197 +197 @@
             figallmodexpscr = False
 
-    if config['addall'] == 'true':
+    if gen.Str_Bool(config['addall']):
         print warnmsg
         print "  " + fname + ": adding new values everywhere !!"
@@ -208 +208 @@
         addallmodexpfigs = True
     else:
-        if config['addfiles'] == 'true':
-            addfils = True
-        else:
-            addfils = False
-
-        if config['addfigures'] == 'true':
-            addfigs = True
-        else:
-            addfigs = False
-
-        if config['adddiffs'] == 'true':
-            adddiffs = True
-        else:
-            adddiffs = False
-
-        if config['adddifffigures'] == 'true':
-            adddifffigs = True
-        else:
-            adddifffigs = False
-
-        if config['addmoddiffs'] == 'true':
-            addmoddiffs = True
-        else:
-            addmoddiffs = False
-
-        if config['addmoddifffigures'] == 'true':
-            addmoddifffigs = True
-        else:
-            addmoddifffigs = False
-
-        if config['addallmodexpfigures'] == 'true':
-            addallmodexpfigs = True
-        else:
-            addallmodexpfigs = False
-
-    if config['debug'] == 'true':
-        debug = True
-    else:
-        debug = False
+        addfils = gen.Str_Bool(config['addfiles'])
+        addfigs = gen.Str_Bool(config['addfigures'])
+        adddiffs = gen.Str_Bool(config['adddiffs'])
+        adddifffigs = gen.Str_Bool(config['adddifffigures'])
+        addmoddiffs = gen.Str_Bool(config['addmoddiffs'])
+        addmoddifffigs =  gen.Str_Bool(config['addmoddifffigures'])
+        addallmodexpfigs = gen.Str_Bool(config['addallmodexpfigures'])
+
+    debug = gen.Str_Bool(config['debug'])
 
     return scr, filescr, figscr, difscr, figdifscr, moddifscr, figmoddifscr,         \
@@ -1013 +983 @@
         lsec: longitudinal section (longitudinal value must be given, [var]@[lat])
         pinterp: pressure interpolation (to the given $plevels)
+        smean: spatial-weighted mean values
         tmean: temporal mean values
         tstd: temporal standard deviation values
@@ -1270 +1241 @@
             CFvarnp = vninF
             gP = 'none'
+
+        elif op == 'smean':
+            # spatial-weighted mean values
+            noLlvdims = list(varnCFs)
+            noLlvdims.remove('lon')
+            noLlvdims.remove('lat')
+            Saddvars = ':'.join(noLlvdims)
+            if len(Saddvars) > 1:
+                vals = 'reglonlat,lon,lat,lon,lat,' + Saddvars
+            else:
+                vals = 'reglonlat,lon,lat,lon,lat,None'
+
+            if dofile:
+                try:
+                    with gen.Capturing() as output:
+                        ncvar.SpatialWeightedMean(vals,prevfile,vninF)
+                except:
+                    print errmsg
+                    print 'SpatialWeightedMean('+vals+',', prevfile, ','+vninF+')'
+                    for s1out in output: print s1out
+                    quit(-1)
+
+                if db:
+                    for s1out in output: print s1out
+
+                # renaming variable
+                ncvar.chvarname(vninF+'mean', 'SpatialWeightedMean_reglonlat.nc',    \
+                  vninF+'spaceweightmean')
+                sout = sub.call('mv SpatialWeightedMean_reglonlat.nc '+fileon,       \
+                  shell=True)
+
+                # Keeping the operations
+                pyins=pyH + "/nc_var.py -o SpatialWeightedMean -S '" + vals +        \
+                  "' -f " + prevfile + " -v " + vninF
+                otrackf.write("\n")
+                otrackf.write("# " + vninF + " " + Fopers + "\n")
+                otrackf.write(pyins + "\n")
+
+            # removing dimension variable-dimension 'lon', 'lat'
+            varnCFs.remove('lon')
+            varnCFs.remove('lat')
+
+            varkeep.append(vninF + 'mean')
 
         elif op == 'tmean':