source: lmdz_wrf/trunk/tools/getting_fields-trajectories.bash @ 647

Last change on this file since 647 was 622, checked in by lfita, 10 years ago

Script to obtain trajectory fields and values
  • Property svn:executable set to *
File size: 13.9 KB
Line 
1#!/bin/bash -x
2# Script to analyse results from model outputs following trajectories
3# L. Fita, LMD May 2015
4
5# isim: folder with the model outputs
6# osim: folder for the outputs of the analysis
7
8# A file called 'trajectory.dat' is mandatory to proceed
9# Concatenated file `netcdf_fold_concatenated.nc' is used to retrieve the trajectory
10# Output
11##
12# netcdf_fold_concatenated.nc: File with only `P' to retrieve the trajectory
13# full_concatenated.nc: File with a given set of variables concatenated
14# diagnostics.nc: file with the computed diagnostics
15# fullout.nc: merge of `full_concatenated.nc' and `diagnostics.nc'
16# tevol_boxtraj.nc: file with the values following the trajectory
17
18models='wrf'
19
20#WRFexps='control:mp1:pbl1:pbl2:rad1'
21WRFexps='control:pbl2:rad1'
22LMDZexps='lmdza:lmdzb:lmdzora:lmdzorb'
23LWRFexps='lmdza:lmdzb'
24
25
26# Domain to analyse
27dom='2'
28
29ofold=/homedata/lfita/etudes/WRFdynamicoSENS
30HOMEpy=/home/lfita/etudes/WRF_LMDZ/svn/LMDZ_WRF/tools
31
32use_concatenated=false
33
34####### ###### ##### #### ### ## #
35errormsg='ERROR -- error -- ERROR -- error'
36main='getting_fields-trajectories.bash'
37
38#WRFvars='all'
39WRFvars='Times:XLONG:XLAT:T2,U10,V10,P,PB,PSFC,HGT,T,QVAPOR,RAINC,RAINNC'
40
41#LMDZvars='all'
42LMDZars='Time_counter:longitude:latitude:u10m:v10m:t2m,wind10m,slp,precip'
43
44LWRFvars='Times:XLONG:XLAT:T2,U10,V10,P,PB,PSFC,HGT,T,QVAPOR,RAINC,RAINNC'
45#LWRFvars='LU_INDEX,ZNU,ZNW,ZS,DZS,U,RU_TEND,V,RV_TEND,W,PH,PHB,T,HFX_FORCE,LH_FORCE,'
46#LWRFvars=${LWRFvars}'TSK_FORCE,HFX_FORCE_TEND,LH_FORCE_TEND,TSK_FORCE_TEND,MU,MUB,'
47#LWRFvars=${LWRFvars}'NEST_POS,P,PB,FNM,FNP,RDNW,RDN,DNW,DN,CFN,CFN1,P_HYD,Q2,T2,TH2,'
48#LWRFvars=${LWRFvars}'PSFC,U10,V10,RDX,RDY,RESM,ZETATOP,CF1,CF2,CF3,XTIME,QVAPOR,'
49#LWRFvars=${LWRFvars}'LANDMASK,TOPOSTDV,TSLB,SMOIS,SH2O,SMCREL,SEAICE,XICEM,SFROFF,'
50#LWRFvars=${LWRFvars}'UDROFF,VEGFRA,GRDFLX,ACGRDFLX,SNOW,SNOWH,CANWAT,SST,SSTSK,LAI,'
51#LWRFvars=${LWRFvars}'H_DIABATIC,MAPFAC_M,MAPFAC_U,MAPFAC_V,MAPFAC_MX,MAPFAC_MY,'
52#LWRFvars=${LWRFvars}'MAPFAC_UX,MAPFAC_UY,MAPFAC_VX,MF_VX_INV,MAPFAC_VY,F,E,SINALPHA,'
53#LWRFvars=${LWRFvars}'COSALPHA,HGT,TSK,P_TOP,T00,P00,TLP,TISO,MAX_MSTFX,MAX_MSTFY,'
54#LWRFvars=${LWRFvars}'RAINC,RAINSH,RAINNC,SNOWNC,GRAUPELNC,HAILNC,CLDFRA,SWDOWN,GLW,'
55#LWRFvars=${LWRFvars}'SWNORM,OLR,XLAT,XLONG,XLAT_U,XLONG_U,XLAT_V,XLONG_V,ALBEDO,'
56#LWRFvars=${LWRFvars}'ALBBCK,EMISS,NOAHRES,TMN,XLAND,ZNT,UST,PBLH,HFX,QFX,LH,ACHFX,'
57#LWRFvars=${LWRFvars}'ACLHF,SNOWC,SR,POTEVP,SNOPCX,SOILTB,LAGES_LIC,LAGES_OCE,'
58#LWRFvars=${LWRFvars}'LAGES_SIC,LAGES_TER,LAIRE,LAIRETER,LALB1,LALB2,LALBE_LIC,'
59#LWRFvars=${LWRFvars}'LALBE_OCE,LALBE_SIC,LALBE_TER,LALE,LALE_BL,LALE_WK,LALP,LALP_BL,'
60#LWRFvars=${LWRFvars}'LALP_WK,LA_TH,LBETA_PREC,LBILS,LBILS_DISS,LBILS_EC,'
61#LWRFvars=${LWRFvars}'LBILS_ENTHALP,LBILS_KINETIC,LBILS_LATENT,LBILS_TKE,LCAPE,'
62#LWRFvars=${LWRFvars}'LCAPE_MAX,LCDRH,LCDRM,LCLDEMI,LCLDH,LCLDL,LCLDM,LCLDQ,LCLDT,'
63#LWRFvars=${LWRFvars}'LCLDTAU,LCLWCON,LCONTFRACATM,LCONTFRACOR,LCUMPB,LCUMRN,LDMC,'
64#LWRFvars=${LWRFvars}'LDNWD,LDNWD0,LDQAJS,LDQCON,LDQDYN,LDQEVA,LDQLSC,LDQLSCST,'
65#LWRFvars=${LWRFvars}'LDQLSCTH,LDQPHY,LDQTHE,LDQVDF,LDQWAK,LDTAJS,LDTCON,LDTDIS,'
66#LWRFvars=${LWRFvars}'LDTDYN,LDTEC,LDTEVA,LD_TH,LDTHMIN,LDTLIF,LDTLSC,LDTLSCHR,'
67#LWRFvars=${LWRFvars}'LDTLSCST,LDTLSCTH,LDTLW0,LDTLWR,LDTORO,LDTPHY,LDTSVDFG,LDTSVDFI,'
68#LWRFvars=${LWRFvars}'LDTSVDFO,LDTSVDFT,LDTSW0,LDTSWR,LDTTHE,LDTVDF,LDTWAK,LDUCON,'
69#LWRFvars=${LWRFvars}'LDUDYN,LDULIF,LDUORO,LDUVDF,LDVCON,LDVDYN,LDVLIF,LDVORO,LDVVDF,'
70#LWRFvars=${LWRFvars}'LEC550AER,LE_TH,LEVAP,LEVAP_LIC,LEVAP_OCE,LEVAPPOT_LIC,'
71#LWRFvars=${LWRFvars}'LEVAPPOT_OCE,LEVAPPOT_SIC,LEVAPPOT_TER,LEVAP_SIC,LEVAP_TER,LEVU,'
72#LWRFvars=${LWRFvars}'LFBASE,LFDER,LFFONTE,LFL,LFLAT,LFLW_LIC,LFLW_OCE,LFLW_SIC,'
73#LWRFvars=${LWRFvars}'LFLW_TER,LFQCALVING,LFQFONTE,LFRACT_LIC,LFRACT_OCE,LFRACT_SIC,'
74#LWRFvars=${LWRFvars}'LFRACT_TER,LFSNOW,LFSW_LIC,LFSW_OCE,LFSW_SIC,LFSW_TER,'
75#LWRFvars=${LWRFvars}'LFTIME_CON,LFTIME_TH,LGEOP,LH2O,LIWCON,LIWP,LKZ,LKZ_MAX,'
76#LWRFvars=${LWRFvars}'LLAMBDA_TH,LLAT_LIC,LLAT_OCE,LLAT_SIC,LLAT_TER,LLMAXTH,LLWCON,'
77#LWRFvars=${LWRFvars}'LLWDN200,LLWDN200CLR,LLWDNSFC,LLWDNSFCCLR,LLWDOWNOR,LLWP,'
78#LWRFvars=${LWRFvars}'LLWUP200,LLWUP200CLR,LLWUPSFC,LLWUPSFCCLR,LMA,LMASS,LMC,LMCD,'
79#LWRFvars=${LWRFvars}'LMSNOW,LNDAYRAIN,LNETTOP,LOLIQ,LOVAP,LOVAPINIT,LOZONE,LPAPRS,'
80#LWRFvars=${LWRFvars}'LPB,LPBASE,LPHIS,LPLCL,LPLFC,LPLUC,LPLUL,LPLULST,LPLULTH,'
81#LWRFvars=${LWRFvars}'LPOURC_LIC,LPOURC_OCE,LPOURC_SIC,LPOURC_TER,LPR_CON_I,LPR_CON_L,'
82#LWRFvars=${LWRFvars}'LPRECIP,LPRES,LPRESNIVS,LPR_LSC_I,LPR_LSC_L,LPRW,LPSOL,LPTCONV,'
83#LWRFvars=${LWRFvars}'LPTOP,LQ10,LQ100,LQ200,LQ2M,LQ50,LQ500,LQ700,LQ850,LQSAT2M,'
84#LWRFvars=${LWRFvars}'LQSOL,LQSURF,LQ_TH,LRADSOL,LRATQS,LRE,LREF_ICE,LREF_LIQ,LRH2M,'
85#LWRFvars=${LWRFvars}'LRH2M_MAX,LRH2M_MIN,LRHUM,LRLD,LRLDCS,LRLU,LRLUCS,LRN,LRNEB,'
86#LWRFvars=${LWRFvars}'LRNEBCON,LRNEBLS,LRSD,LRSDCS,LRSU,LRSUCS,LRUGS,LRUGS_LIC,'
87#LWRFvars=${LWRFvars}'LRUGS_OCE,LRUGS_SIC,LRUGS_TER,LSENS,LSENS_LIC,LSENS_OCE,'
88#LWRFvars=${LWRFvars}'LSENS_SIC,LSENS_TER,LSICF,LS_LCL,LSLP,LSNOW,LSNOWL,LSOLL,LSOLL0,'
89#LWRFvars=${LWRFvars}'LSOLLWDOWN,LSOLS,LSOLS0,LS_PBLH,LS_PBLT,LS_THERM,LSWDN200,'
90#LWRFvars=${LWRFvars}'LSWDN200CLR,LSWDNSFC,LSWDNSFCCLR,LSWDNTOA,LSWDNTOACLR,LSWDOWNOR,'
91#LWRFvars=${LWRFvars}'LSWNETOR,LSWUP200,LSWUP200CLR,LSWUPSFC,LSWUPSFCCLR,LSWUPTOA,'
92#LWRFvars=${LWRFvars}'LSWUPTOACLR,LT10,LT100,LT200,LT2M,LT2M_LIC,LT2M_MAX,LT2M_MIN,'
93#LWRFvars=${LWRFvars}'LT2M_OCE,LT2M_SIC,LT2M_TER,LT50,LT500,LT700,LT850,LTAUX,'
94#LWRFvars=${LWRFvars}'LTAUX_LIC,LTAUX_OCE,LTAUX_SIC,LTAUX_TER,LTAUY,LTAUY_LIC,'
95#LWRFvars=${LWRFvars}'LTAUY_OCE,LTAUY_SIC,LTAUY_TER,LTEMP,LTHETA,LTKE,LTKE_LIC,'
96#LWRFvars=${LWRFvars}'LTKE_MAX,LTKE_MAX_LIC,LTKE_MAX_OCE,LTKE_MAX_SIC,LTKE_MAX_TER,'
97#LWRFvars=${LWRFvars}'LTKE_OCE,LTKE_SIC,LTKE_TER,LTNHUS,LTNHUSC,LTNHUSSCPBL,LTNT,'
98#LWRFvars=${LWRFvars}'LTNTC,LTNTR,LTNTSCPBL,LT_OCE_SIC,LTOPL,LTOPL0,LTOPS,LTOPS0,'
99#LWRFvars=${LWRFvars}'LTPOT,LTPOTE,LTSOL,LTSOL_LIC,LTSOL_OCE,LTSOL_SIC,LTSOL_TER,LU10,'
100#LWRFvars=${LWRFvars}'LU100,LU10M,LU10M_LIC,LU10M_OCE,LU10M_SIC,LU10M_TER,LU200,LU50,'
101#LWRFvars=${LWRFvars}'LU500,LU700,LU850,LUE,LUPWD,LUQ,LUSTAR,LUSTAR_LIC,LUSTAR_OCE,'
102#LWRFvars=${LWRFvars}'LUSTAR_SIC,LUSTAR_TER,LV10,LV100,LV10M,LV10M_LIC,LV10M_OCE,'
103#LWRFvars=${LWRFvars}'LV10M_SIC,LV10M_TER,LV200,LV50,LV500,LV700,LV850,LVE,LVITU,'
104#LWRFvars=${LWRFvars}'LVITV,LVITW,LVPRECIP,LVQ,LW10,LW100,LW200,LW50,LW500,LW700,'
105#LWRFvars=${LWRFvars}'LW850,LWAKE_DELTAQ,LWAKE_DELTAT,LWAKE_H,LWAKE_OMG,LWAKE_S,LWAPE,'
106#LWRFvars=${LWRFvars}'LWBEFF,LWBILO_LIC,LWBILO_OCE,LWBILO_SIC,LWBILO_TER,LWBILS_LIC,'
107#LWRFvars=${LWRFvars}'LWBILS_OCE,LWBILS_SIC,LWBILS_TER,LWDTRAINA,LWDTRAINM,LWEAKINV,'
108#LWRFvars=${LWRFvars}'LWIND10M,LWIND10MAX,LZ10,LZ100,LZ200,LZ50,LZ500,LZ700,LZ850,'
109#LWRFvars=${LWRFvars}'LZFULL,LZHALF,LZMAX_TH,LCONTROL,LMSK,LTER,LLIC,LOCE,LSIC,'
110#LWRFvars=${LWRFvars}'LTKSOIL,LOTTER,LOTLIC,LOTOCE,LOTSIC,LQKSOIL,LWSOL,LALBKSOIL,'
111#LWRFvars=${LWRFvars}'LLWALBKSOIL,LEVAPKSOIL,LSNOWKSOIL,LRADS,LSOLSW,LSOLLW,LRAIN,'
112#LWRFvars=${LWRFvars}'LRUGKSOIL,LAGESNOKSOIL,LZMEA,LZSTD,LZSIG,LZGAM,LZTHE,LZPIC,'
113#LWRFvars=${LWRFvars}'LZVAL,LZRUGSREL,LRUGSEA,LRUNOFFLIC,LZMAX0,LF0,LEMA_WORK1,'
114#LWRFvars=${LWRFvars}'LEMA_WORK2,LWAKE_CSTAR,LWAKE_PE,LWAKE_FIP,Times,'
115#LWRFvars=${LWRFvars}'ITIMESTEP,IVGTYP,ISLTYP,STEPAVE_COUNT,SAVE_TOPO_FROM_REAL'
116
117LMDZtrjvars='time_counter,u10m,v10m,precip,slp,psol'
118#LMDZ lon/lat file
119if test ! -f lmdz_lonlat.nc; then
120  python ${HOMEpy}/nc_var.py -o selvar -v t2m -f \
121    lmdz/run/lmdza/lmdzout/histins.nc -S 'lon@lon,lat@lat'
122  mv selvar_new.nc lmdz_lonlat.nc
123fi
124
125WRFvdims='XLONG,XLAT,ZNU,time'
126LMDZvdims='lon,lat,presnivs,time_counter'
127LWRFvdims='XLONG,XLAT,ZNU,time'
128
129WRFtrajvars='T2,wss,psl,pr'
130LMDZtrajvars='t2m,wind10m,slp,precip'
131LWRFtrajvars='T2,wss,psl,pr'
132
133#######    #######
134## MAIN
135    #######
136
137mods=`expr ${models} | tr ':' ' '`
138
139
140for mod in ${mods}; do
141  echo ${mod}"..."
142  if test ${mod} = 'wrf'; then 
143    experiments=${WRFexps}
144    allvars=${WRFvars}
145    dimvars=${WRFvdims}
146    trjvars=${WRFtrajvars}
147  fi
148  if test ${mod} = 'lmdz'; then 
149    experiments=${LMDZexps}
150    allvars=${LMDZvars}
151    dimvars=${LMDZvdims}
152    trjvars=${LMDZtrajvars}
153  fi
154  if test ${mod} = 'lwrf'; then 
155    experiments=${LWRFexps}
156    allvars=${LWRFvars}
157    dimvars=${LWRFvdims}
158    trjvars=${LWRFtrajvars}
159  fi
160
161  exps=`expr ${experiments} | tr ':' ' '`
162
163  for exp in ${exps}; do
164    echo "  "${exp}"..."
165# Destination folder with the outputs of the analysis
166#    osim=${ofold}/${mod}/run/${exp}
167    osim=${ofold}/${exp}
168    if test ${mod} = 'wrf' || test ${mod} = 'lwrf'; then
169      isim=${osim}/wrfout
170# Trajectory file
171##
172      if test ! -f ${osim}/P_lev1_values.nc; then
173        python ${HOMEpy}/nc_var.py -o netcdf_fold_concatenation -f wrfout_d0${dom}   \
174          -S ${isim},Time -v P
175        if test $? -ne 0; then
176          echo ${errormsg}
177          echo "  "${main}": python does not work !!"
178          echo "  "${HOMEpy}"/nc_var.py -o netcdf_fold_concatenation -f wrfout_d0"${dom}\
179            " -S "${isim}",Time -v P"
180          exit
181        else
182          python ${HOMEpy}/nc_var.py -o DataSetSection_multidims                     \
183            -f netcdf_fold_concatenated.nc -S bottom_top,0,0,1
184          if test $? -ne 0; then
185            echo ${errormsg}
186            echo "  "${main}": python does not work !!"
187            echo "  "${HOMEpy}"/nc_var.py -o DataSetSection_multidims -f "\
188              "netcdf_fold_concatenated.nc -S bottom_top,0,0,1"
189            exit
190          fi
191          mv netcdf_fold_concatenated_bottom_top_B0-E0-I1.nc ${osim}/P_lev1_values.nc
192          rm netcdf_fold_concatenated.nc
193        fi
194      fi
195
196# concatenating all file
197##
198      if test ! -f ${osim}/full_concatenated.nc; then
199        python ${HOMEpy}/nc_var.py -o netcdf_fold_concatenation -f wrfout_d0${dom}   \
200          -S ${isim},Time -v ${allvars}
201        if test $? -ne 0; then
202          echo ${errormsg}
203          echo "  "${main}": python does not work !!"
204          echo "  "${HOMEpy}"/nc_var.py -o netcdf_fold_concatenation -f wrfout_d0"${dom}\
205            " -S "${isim}",Time -v "${allvars}
206          exit
207        else
208          mv netcdf_fold_concatenated.nc ${osim}/full_concatenated.nc
209        fi
210        python ${HOMEpy}/nc_var.py -o WRF_CFtime_creation -S 19491201000000,hours    \
211          -f ${osim}/full_concatenated.nc -v time
212      fi
213
214# diagnostics
215##
216      if test ! -f ${osim}/diagnostics.nc; then
217        if ${use_concatenated}; then
218          diags='RAINTOT|RAINC@RAINNC@time,WRFmslp|WRFp@PSFC@HGT@WRFt@QVAPOR,wss|U10@V10'
219          python ${HOMEpy}/diagnostics.py -f ${osim}/full_concatenated.nc              \
220            -d 'Time@time,bottom_top@ZNU,south_north@XLAT,west_east@XLONG' -v ${diags} \
221            -f ${osim}/full_concatenated.nc
222          if test $? -ne 0; then
223            echo ${errormsg}
224            echo "  "${main}": python does not work !!"
225            echo "  python "${HOMEpy}"/diagnostics.py -f "${osim}"/full_concatenated.nc \
226              -d 'Time@time,bottom_top@ZNU,south_north@XLAT,west_east@XLONG' -v "${diags} \
227              "-f ${osim}/full_concatenated.nc"
228            exit
229          else
230            mv diagnostics.nc ${osim}/diagnostics.nc
231          fi
232        else
233          echo "Computing diagnostics file by file..."
234          diags='RAINTOT|RAINC@RAINNC@WRFtime,WRFmslp|WRFp@PSFC@HGT@WRFt@QVAPOR,wss|U10@V10'
235          for file in ${isim}/wrfout_d0${dom}*; do
236            fdate=`echo ${file} | tr '_' ' ' | awk '{print $3"_"$4}'`
237            echo "  "${file}
238            python ${HOMEpy}/diagnostics.py -f ${osim}/full_concatenated.nc              \
239              -d 'Time@Times,bottom_top@ZNU,south_north@XLAT,west_east@XLONG' -v ${diags}\
240              -f ${file}
241            #exit
242            if test $? -ne 0; then
243              echo ${errormsg}
244              echo "  "${main}": python does not work !!"
245              echo "  python "${HOMEpy}"/diagnostics.py -f "${osim}"/full_concatenated.nc \
246                -d 'Time@Times,bottom_top@ZNU,south_north@XLAT,west_east@XLONG' -v "${diags} \
247                "-f "${file}
248              exit
249            else
250              mv diagnostics.nc ${osim}/diagnostics_${fdate}.nc
251            fi
252          done
253          python ${HOMEpy}/nc_var.py -o netcdf_fold_concatenation -f diagnostics_    \
254            -S ${isim},Time -v ${allvars}
255          if test $? -ne 0; then
256            echo ${errormsg}
257            echo "  "${main}": python does not work !!"
258            echo "  "${HOMEpy}"/nc_var.py -o netcdf_fold_concatenation -f diagnostics_"\
259              " -S "${isim}",Time -v all"
260            exit
261          else
262            mv netcdf_fold_concatenated.nc ${osim}/full_concatenated.nc
263          fi
264          python ${HOMEpy}/nc_var.py -o WRF_CFtime_creation -S 19491201000000,hours    \
265            -f ${osim}/full_concatenated.nc -v time
266        fi
267      fi
268# full file
269##
270      if test ! -f ${osim}/fullout.nc; then
271        cdo merge ${osim}/full_concatenated.nc ${osim}/diagnostics.nc                \
272          ${osim}/fullout.nc
273      fi
274
275# End of if 'wrf', 'lwrf'
276    else
277      osim=${ofold}/${mod}/run/erai/${exp}
278      isim=${osim}/lmdzout
279# Trajectory file
280##
281      if test ! -f ${osim}/trajectory.nc; then
282        cdo selvar,${LMDZtrjvars} ${isim}/histins.nc ${osim}/trajectory.nc
283      fi
284
285    fi
286    if test ${mod} = 'lmdz'; then
287      trjfile=${osim}/lmdzout/histins.nc
288    else
289      trjfile=${osim}/fullout.nc
290    fi
291
292# Following trajectory values
293##
294    if test ! -f ${osim}/tevol_boxtraj.nc && -f ${osim}/trajectory.dat; then
295      tfirst=`cat ${osim}/trajectory.dat | grep -v '#' | head -n 1 | awk '{print $1-1}'`
296      echo "tfirst: "${tfirst}
297      trajvalues=${osim}'/trajectory.dat@'${tfirst}','${dimvars}',cf,5,5'
298      python ${HOMEpy}/nc_var.py -f ${trjfile} -o compute_tevolboxtraj               \
299        -S ${trajvalues} -v ${trjvars}
300      if test $? -ne 0; then
301        echo ${errormsg}
302        echo "  "${main}": python does not work !!"
303        echo "  python "${HOMEpy}"nc_var.py -f ${trjfile} -o compute_tevolboxtraj -S \
304          ${trajvalues}  -v ${trjvars}"
305        exit
306      else
307        mv tevolboxtraj_multi-var.nc ${osim}/tevol_boxtraj.nc
308      fi
309    fi
310#    exit
311# end experiments
312  done
313#  exit
314# end models
315done
Note: See TracBrowser for help on using the repository browser.