Changeset 2023 in lmdz_wrf for trunk/tools

Timestamp:

Aug 3, 2018, 4:24:05 PM (7 years ago)

Author:

lfita

Message:

Adding retrieval of surface 2D maps

File:

• trunk/tools/get_WRFsounding-surface_data.bash (modified) (2 diffs)

trunk/tools/get_WRFsounding-surface_data.bash

@@ -294 +294 @@
   fi
 }
+
+function get_WRFsfc2D_data(){
+# Function to get the data to compute analysis in a surface map
+#   filen= name of the file
+#   sfcdiagnames= diagnostics to compute (as ':' separated list)
+#   nondiagvs= ':' list of variables which are not from a diagnostics ('None' for 
+#     without)
+
+  funcvals=$1
+
+  filen=`echo ${funcvals} | tr '#' ' ' | awk '{print $1}'`
+  sfcdiagnames=`echo ${funcvals} | tr '#' ' ' | awk '{print $2}'`
+  nondiagvs=`echo ${funcvals} | tr '#' ' ' | awk '{print $3}'`
+
+  # Computing sfc diagnostics
+  sfcmapdiagd='Time@WRFtime:south_north@XLAT:west_east@XLONG'
+  diagns=`echo ${sfcdiagnames} | tr ':' ' '`
+
+  # getting that variables only required for the diagnostics
+  vardiags=''
+  idiag=1
+  for diagn in ${diagns}; do
+    diagv=`cat $pyHOME/diagnostics.inf | grep WRF | grep ${diagn}',' | tr ',' ' ' |  \
+      awk '{print $2"|"$3}'`
+    if test ${idiag} -eq 1; then
+      sfcdiagvals=${diagv}
+      vardiags=`echo ${diagv} | tr '|' ' ' | awk '{print $2}' | tr '@' ':'`
+    else
+      sfcdiagvals=${sfcdiagvals}','${diagv}
+      vardiag=`echo ${diagv} | tr '|' ' ' | awk '{print $2}' | tr '@' ' '`
+      for vd in ${vardiag}; do
+        if ! $(isin_list ${vardiags} ${vd}); then vardiags=${vardiags}':'${vd}; fi
+      done
+    fi
+    idiag=`expr ${idiag} + 1`
+  done
+
+  # Inlcuding non-diagnostics
+  if test ! ${nondiagvs} = 'None'; then
+    nvns=`echo ${nondiagvs} | tr ':' ' '`
+    for vn in ${nvns}; do
+      if ! $(isin_list ${vardiags} ${vn}); then vardiags=${vardiags}':'${vn}; fi
+    done
+  fi
+  vardiag=`echo ${vardiags} | tr ':' ','`
+  
+  # Getting values
+  if ${fscratch}; then rm ${simsfcmapf} >& /dev/null; fi
+  if test ! -f ${simsfcmapf}; then
+    values='Time,0,-1,1@south_north,0,-1,1@west_east,0,-1,1@south_north_stag,0,-1,1@'
+    values=${values}'west_east_stag,0,-1,1:False'
+    python ${pyHOME}/nc_var.py -o DataSetSection_multidims -f ${filen}               \
+      -S ${values} -v ${vardiag}
+    if test $? -ne 0; then
+      echo ${errmsg}
+      echo "  python failed!!"
+      echo python ${pyHOME}/nc_var.py -o DataSetSection_multidims -f ${filen}        \
+        -S "'"${values}"'"  -v ${vardiag}
+      exit
+    fi
+    echo "* get values for 2D map diagnostics: " >> ${ofileins}
+    echo python ${pyHOME}/nc_var.py -o DataSetSection_multidims -f ${filen}          \
+      -S "'"${values}"'" -v ${vardiag} >> ${ofileins}
+    echo " " >> ${ofileins}
+
+    ofilen='newfile_multidims.nc'
+    mv ${ofilen} ${simsfcmapf}
+  fi
+
+  if ${fscratch}; then rm ${simsfcmapdiagsf}; fi
+  if test ! -f ${simsfcmapdiagsf}; then
+    python ${pyHOME}/diagnostics.py -f ${simsfcmapf} -d ${sfcmapdiagd}               \
+      -v ${sfcdiagvals}
+    if test $? -ne 0; then
+      echo ${errmsg}
+      echo "  python failed!!"
+      echo python ${pyHOME}/diagnostics.py -f ${simsfcmapf} -d "'"${sfcmapdiagd}"'"  \
+        -v "'"${sfcdiagvals}"'"
+      exit
+    fi
+    echo "* diagnostics for sfc 2D map: " >> ${ofileins}
+    echo python ${pyHOME}/diagnostics.py -f ${simsfcmapf} -d "'"${sfcmapdiagd}"'"    \
+      -v "'"${sfcdiagvals}"'" >> ${ofileins}
+    echo " " >> ${ofileins}
+    mv diagnostics.nc ${simsfcmapdiagsf}
+    if $(isin_list ${sfcdiagnames} wds); then
+      #To rad
+      python ${pyHOME}/nc_var.py -o valmod -S mulc,0.0174532925199                   \
+        -f ${simsfcmapdiagsf} -v wds
+      python ${pyHOME}/nc_var.py -o varaddattr -S 'units|rad' -f ${simsfcmapdiagsf}  \
+        -v wds
+    fi
+
+    #Adding non-diagnostic variables
+    if test ! ${nondiagvs} = 'None'; then
+      varsadd=`echo ${nondiagvs} | tr ':' ' '`
+      for vadd in ${varsadd}; do
+        python ${pyHOME}/nc_var.py -o fvaradd -S ${simsfcmapf},${vadd}               \
+          -f ${simsfcmapdiagsf}
+        if test $? -ne 0; then
+          echo ${errmsg}
+          echo "  python failed!!"
+          echo python ${pyHOME}/nc_var.py -o fvaradd -S "'"${simsfcmapf},${vadd}"'"  \
+            -f ${simsfcmapdiagsf}
+          exit
+          rm ${simsfcmapdiagsf}
+        fi
+        echo "* adding variable '"${vadd}"' at sfc map: " >> ${ofileins}
+        echo python ${pyHOME}/nc_var.py -o fvaradd -S "'"${simsfcmapf},${vadd}"'"    \
+          -f ${simsfcmapdiagsf} >> ${ofileins}
+        echo " " >> ${ofileins}
+        # CF varname
+        CFvarn=`python $pyHOME/generic.py -o variables_values -S ${vadd} |           \
+          tr ':' ' ' | awk '{print $1}'`
+        if test $? -ne 0; then
+          echo ${errmsg}
+          echo "  python failed!!"
+          echo python $pyHOME/generic.py -o variables_values -S ${vadd}
+          exit
+          rm ${simsfcmapdiagsf}
+        fi
+        python $pyHOME/nc_var.py -o chvarname -S ${CFvarn} -f ${simsfcmapdiagsf}     \
+          -v ${vadd}
+        if test $? -ne 0; then
+          echo ${errmsg}
+          echo "  python failed!!"
+          echo python ${pyHOME}/nc_var.py -o chvarname -S ${CFvarn}                  \
+            -f ${simsfcmapdiagsf} -v ${vadd}
+          exit
+          rm ${simsfcmapdiagsf}
+        fi
+      done
+    fi
+  fi
+}
+
 #######    #######
 ## MAIN
@@ -331 +467 @@
   fi
 
+  # Surface maps
+  if test ! ${sfcmapid} = 'None'; then
+    simsfcmapf=${odir}'/simout_vars_sfcmap_'${stid}'_'${expl}'_'${fdate}'.nc'
+    simsfcmapdiagsf=${odir}'/simout_sfcmapdiags_'${stid}'_'${expl}'_'${fdate}'.nc'
+
+    fvals=${infilen}'#'${wrfsfcdiags}'#'${wrfsfcnodiags}
+    get_WRFsfc2D_data ${fvals}
+
 fi