Changeset 1688 in lmdz_wrf

Timestamp:

Dec 5, 2017, 1:12:34 PM (7 years ago)

Author:

lfita

Message:

Final working version of the scripts and its configuration for WRF

Location:

trunk/tools

Files:

• config_get_stations.WRF.py (modified) (1 diff)
• get_stations.py (modified) (6 diffs)

trunk/tools/config_get_stations.WRF.py

@@ -30 +30 @@
 # simulation file
 simfilen = '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00'
+
+# dimension CFequivalences
+CFdims = {'lon': 'west_east', 'lat': 'south_north', 'plev': 'bottom_top',            \
+  'time': 'Time'}
 
 # axes dimensions

trunk/tools/get_stations.py

@@ -20 +20 @@
 ####### ###### ##### #### ### ## #
 
+# Gneric variables prepared to be computed
+Cvars = ['C_td']
+
+# WRF specific diagnostics
+WRFvars = ['WRFp', 'WRFta', 'WRFtd', 'WRFtime', 'WRFua', 'WRFuas', 'WRFva', 'WRFvas',\
+  'WRFzg']
+
 # Variables not to check their existence inside file
-NONcheckingvars = ['WRFp', 'WRFta', 'WRFtd', 'WRFtime', 'WRFua', 'WRFuas', 'WRFva',  \
-  'WRFvas', 'WRFzg']
+NONcheckingvars = Cvars + WRFvars
 
 def creation_sfcstation_file(filen, lv, Lv, hv, lab, tunits, sfcvars, dimt, ifilen):
@@ -254 +260 @@
 ## MAIN
     #######
+
+Wcomputed = {}
+for vn in diag.Wavailablediags:
+    Wcomputed[vn] = False
+
+Ccomputed = {}
+for vn in diag.Cavailablediags:
+    Ccomputed[vn] = False
 
 yrref = ReferenceDate[0:4]
@@ -366 +380 @@
 # Retrieving surface data
 ##
-
-# Had already uas & vas been computed?
-computeduasvas = False
 
 # Diagnosted variables
@@ -465 +476 @@
         if gen.searchInlist(NONcheckingvars,fvn):
             if ist == 0:
-                if fvn == 'WRFuas' or fvn == 'WRFvas':
-                    if not computeduasvas:
-                        computeduasvas = True
-                        ua, va, dmn, dmvn = diag.compute_WRFuasvas(                  \
-                          onc.variables['U10'][:], onc.variables['V10'][:],          \
-                          onc.variables['SINALPHA'][:], onc.variables['COSALPHA'][:],\
-                          getdims, getvdims)
-                        diagvars['WRFuas'] = ua
-                        diagvars['WRFvas'] = va
-                    ogetvar = diagvars[fvn]
+                if fvn[0:3] == 'WRF':
+                    fvn_diag = fvn[3:len(fvn)]
+                    if not Wcomputed[fvn_diag]:
+                        # Using a given series of prepared diagnostics
+                        Wcomputed[fvn_diag] = True
+                        vardiag = diag.W_diagnostic(fvn_diag, onc,                  \
+                          sfcvariables, sndvariables, getdims, getvdims, CFdims)
+                        diagvars[fvn] = vardiag.values
+                elif fvn[0:2] == 'C_':
+                    fvn_diag = fvn.split('_')[1]
+                    if not Ccomputed[fvn_diag]:
+                        # Using a given series of prepared diagnostics
+                        Ccomputed[fvn_diag] = True
+                        vardiag = C_diagnostic(fvn_diag, onc, sfcvariables,          \
+                          sndvariables, CFdims)
+                        diagvars[fvn] = vardiag.values
+                ogetvar = diagvars[fvn]
             else:
                 ogetvar = diagvars[fvn]
@@ -499 +517 @@
 
 # Retrieving sounding data
-
-# Had already diagnostics been computed?
-computedta = False
-computedtd = False
-computeduava = False
-computedp = False
-computedzg = False
 
 # Diagnosted variables
@@ -602 +613 @@
         if gen.searchInlist(NONcheckingvars,fvn):
             if ist == 0:
-                if fvn == 'WRFp':
-                    if not computedp:
-                        computedp = True
-                        diagvars['WRFp'] = onc.variables['P'][:] +                   \
-                          onc.variables['PB'][:]
-                elif fvn == 'WRFta':
-                    if not computedta:
-                        computedta = True
-                        ta, dmn, dmvn = diag.compute_WRFta(onc.variables['T'][:],    \
-                          onc.variables['P'][:]+onc.variables['PB'][:],              \
-                          getdims, getvdims)
-                        diagvars['WRFta'] = ta
-                elif fvn == 'WRFtd':
-                    if not computedtd:
-                        computedtd = True
-                        td, dmn, dmvn = diag.compute_WRFtd(onc.variables['T'][:],    \
-                          onc.variables['P'][:]+onc.variables['PB'][:],              \
-                          onc.variables['QVAPOR'][:], getdims, getvdims)
-                        diagvars['WRFtd'] = td
-                elif fvn == 'WRFua' or fvn == 'WRFva':
-                    if not computeduava:
-                        computeduasvas = True
-                        ua, va, dmn, dmvn = diag.compute_WRFuava(                    \
-                          onc.variables['U'][:], onc.variables['V'][:],              \
-                          onc.variables['SINALPHA'][:], onc.variables['COSALPHA'][:],\
-                          getdims, getvdims)
-                        diagvars['WRFua'] = ua
-                        diagvars['WRFva'] = va
-                elif fvn == 'WRFzg':
-                    if not computedzg:
-                        computedzg = True
-                        diagvars['WRFzg'] = onc.variables['PH'][:] +                 \
-                          onc.variables['PHB'][:]
+                if fvn[0:3] == 'WRF':
+                    fvn_diag = fvn[3:len(fvn)]
+                    if not Wcomputed[fvn_diag]:
+                        # Using a given series of prepared diagnostics
+                        Wcomputed[fvn_diag] = True
+                        vardiag = diag.W_diagnostic(fvn_diag, onc,                   \
+                          sfcvariables, sndvariables, getdims, getvdims, CFdims)
+                        diagvars[fvn] = vardiag.values
+                elif fvn[0:2] == 'C_':
+                    fvn_diag = fvn.split('_')[1]
+                    if not Ccomputed[fvn_diag]:
+                        # Using a given series of prepared diagnostics
+                        Ccomputed[fvn_diag] = True
+                        vardiag = diag.C_diagnostic(fvn_diag, onc, sfcvariables,     \
+                          sndvariables, CFdims)
+                        diagvars[fvn] = vardiag.values
+
                 ogetvar = diagvars[fvn]
             else: