From 45646740bc44c88bc20acaae00015318bd331211 Mon Sep 17 00:00:00 2001
From: lkugler <lukas.kugler@gmail.com>
Date: Mon, 13 Sep 2021 19:07:10 +0200
Subject: [PATCH] update

---
 config/clusters.py                 | 9 +++++----
 scripts/create_wbubble_wrfinput.py | 6 +++---
 scripts/obsseq_to_netcdf.py        | 8 +++++---
 scripts/prepare_wrfinput.py        | 2 +-
 scripts/run_obs_diag.py            | 8 ++------
 5 files changed, 16 insertions(+), 17 deletions(-)

diff --git a/config/clusters.py b/config/clusters.py
index 0a688b8..a51c84b 100755
--- a/config/clusters.py
+++ b/config/clusters.py
@@ -28,15 +28,16 @@ vsc.python = '/home/fs71386/lkugler/miniconda3/envs/DART/bin/python'
 vsc.ncks = '/home/fs71386/lkugler/miniconda3/envs/DART/bin/ncks'
 vsc.tmpfiledir = '/gpfs/data/fs71386/lkugler'
 vsc.userdir = '/home/fs71386/lkugler'
-vsc.srcdir = '/gpfs/data/fs71386/lkugler/compile/WRF/WRF-4.2.2/run'
+vsc.srcdir = '/gpfs/data/fs71386/lkugler/compile/WRF/WRF-4.3/run'
 vsc.archive_base = '/gpfs/data/fs71386/lkugler/sim_archive/'
-vsc.dart_srcdir = '/home/fs71386/lkugler/DART/DART-9.9.0/models/wrf/work'
+vsc.dart_srcdir = '/home/fs71386/lkugler/DART/DART-9.11.9/models/wrf/work'
 vsc.dartrundir = '/gpfs/data/fs71386/lkugler/run_DART'
 vsc.scriptsdir = '/home/fs71386/lkugler/DART-WRF/scripts/'
 
-vsc.nature_wrfout = '/home/fs71386/lkugler/data/sim_archive/exp_v1.16_P1_nature_tstep/2008-07-30_06:00/1/wrfout_d01_%Y-%m-%d_%H:%M:%S'
+vsc.nature_wrfout = '/home/fs71386/lkugler/data/sim_archive/exp_v1.16_Pwbub_nat/2008-07-30_09:00/1/wrfout_d01_%Y-%m-%d_%H:%M:%S'
+#vsc.nature_wrfout = '/home/fs71386/lkugler/data/sim_archive/wrfout_d01_%Y-%m-%d_%H:%M:%S'
 #vsc.input_profile = '/home/fs71386/lkugler/wrf_profiles/data/wrf/ens/from_LMU/raso.nat.<iens>.wrfprof'
-vsc.input_profile = '/home/fs71386/lkugler/wrf_profiles/data/wrf/ens/2021-05-04/raso.nat.1.wrfprof'
+vsc.input_profile = '/home/fs71386/lkugler/wrf_profiles/data/wrf/ens/2021-05-04/raso.nat.001.wrfprof'
 
 vsc.ideal = vsc.userdir+'/compile/bin/ideal-v4.2.2_v1.16.exe'
 vsc.wrfexe = vsc.userdir+'/compile/bin/wrf-v4.3_v1.16.exe'
diff --git a/scripts/create_wbubble_wrfinput.py b/scripts/create_wbubble_wrfinput.py
index 01732c8..984fcc4 100644
--- a/scripts/create_wbubble_wrfinput.py
+++ b/scripts/create_wbubble_wrfinput.py
@@ -24,8 +24,8 @@ for iens in range(1, exp.n_ens+1):
         z = (z[1:]+z[:-1])/2
         z = z[:, np.newaxis, np.newaxis]
 
-        cx = 100*dx_km #(80 + 40*np.random.rand())*dx_km
-        cy = 100*dx_km #(80 + 40*np.random.rand())*dx_km
+        cx = (80 + 40*np.random.rand())*dx_km
+        cy = (80 + 40*np.random.rand())*dx_km
         print(cx, cy)
         x = np.arange(0,nx)*dx_km
         y = np.arange(0,ny)*dx_km
@@ -36,6 +36,6 @@ for iens in range(1, exp.n_ens+1):
 
         pert = 3*np.exp(-(dr/cr)**2)*np.exp(-(z/cz)**2)
 
-        ds.variables['T'][0,...] = ds.variables['T'][0,...] + pert
+        ds.variables['THM'][0,...] = ds.variables['THM'][0,...] + pert
 
     print(wrfin, 'wbubble inserted.')
diff --git a/scripts/obsseq_to_netcdf.py b/scripts/obsseq_to_netcdf.py
index f401e1a..290d8d7 100644
--- a/scripts/obsseq_to_netcdf.py
+++ b/scripts/obsseq_to_netcdf.py
@@ -8,14 +8,16 @@ def listdir_dirs(path):
 if __name__ == '__main__':
 
     datadir = cluster.archive_base
-    ddir = datadir+exp.expname+'/obs_seq_final/'
+    expname = 'exp_v1.16_Pwbub-1_Radar_soe2' 
+    expname = exp.expname
+    ddir = datadir+expname+'/obs_seq_final/'
 
     for dir_name in listdir_dirs(ddir):
         files = sorted(glob.glob(ddir+'/'+dir_name+'/*.final'))  
-        # rod.run_obsdiag(files, f_out=ddir+'/obsdiag_'+dir_name+'.nc')
+        rod.run_obsdiag(files, f_out=ddir+'/obsdiag_'+dir_name+'.nc')
         rod.run_obs_seq_to_netcdf(files, f_out=ddir+'/obs_epoch-'+dir_name+'.nc') 
 
-    ddir = datadir+exp.expname+'/obs_seq_final_1min/'
+    ddir = datadir+expname+'/obs_seq_final_1min/'
 
     for dir_name in listdir_dirs(ddir):
         files = sorted(glob.glob(ddir+'/'+dir_name+'/*.final'))
diff --git a/scripts/prepare_wrfinput.py b/scripts/prepare_wrfinput.py
index 41f8217..1a8aca5 100755
--- a/scripts/prepare_wrfinput.py
+++ b/scripts/prepare_wrfinput.py
@@ -19,7 +19,7 @@ for iens in range(1, exp.n_ens+1):
     symlink(cluster.wrfexe, rundir+'/wrf.exe')
 
     # time not important, but general settings
-    prepare_namelist.run(iens, begin=init_time, end=dt.datetime(2008, 7, 30, 12),
+    prepare_namelist.run(iens, begin=init_time, end=dt.datetime(2008, 7, 30, 23),
                          archive=False)
 
     symlink(input_prof, rundir+'/input_sounding')
diff --git a/scripts/run_obs_diag.py b/scripts/run_obs_diag.py
index 6aadeb9..d3d3feb 100644
--- a/scripts/run_obs_diag.py
+++ b/scripts/run_obs_diag.py
@@ -2,7 +2,7 @@ import os, sys, shutil, glob
 from config.cfg import exp, cluster
 from utils import symlink, copy, sed_inplace, append_file
 
-rundir_program = '/home/fs71386/lkugler/data/DART-WRF/rundir/'
+rundir_program = '/home/fs71386/lkugler/data/run_DART/'
 
 
 def prepare(obserr_iszero='.true.'):
@@ -38,10 +38,6 @@ def run_obsdiag(filepaths, f_out='./obsdiag.nc'):
         print('------ running obs_diag program')
         os.chdir(rundir_program)
         symlink(cluster.dart_srcdir+'/obs_diag', rundir_program+'/obs_diag')
-        try:
-            os.remove(rundir_program+'/obs_seq_to_netcdf')
-        except:
-            pass
         os.system('./obs_diag >& obs_diag.log')  # caution, this overwrites obs_seq_to_netcdf
 
         # move output to archive
@@ -71,5 +67,5 @@ if __name__ == '__main__':
     folder_obs_seq_final = str(sys.argv[1])
     files = sorted(glob.glob(folder_obs_seq_final+'/*.final'))  # input for obs_diag program
     
-    run_obsdiag(files, outdir=folder_obs_seq_final)  # input must be files with posterior data!!
+    run_obsdiag(files, f_out='./test.nc')  # input must be files with posterior data!!
     run_obs_seq_to_netcdf(files, outdir=folder_obs_seq_final)  # input can be files without posterior data
-- 
GitLab