Skip to content

Commit

Permalink
plot soca difference directly
Browse files Browse the repository at this point in the history
  • Loading branch information
weihuang-jedi committed Jul 25, 2022
1 parent fdbc65d commit b6aef1c
Show file tree
Hide file tree
Showing 5 changed files with 17 additions and 14 deletions.
4 changes: 2 additions & 2 deletions genplot.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,12 +164,12 @@ def plot(self, pvar, addmark=0, marker='x', size=3, color='green'):
self.fig = self.plt.figure()
self.ax = self.plt.subplot()

msg = ('plot variable min: %s, max: %s' % (pvar.min(), pvar.max()))
msg = ('plot variable min: %s, max: %s' % (np.min(pvar), np.max(pvar)))
print(msg)

(self.x, self.y) = self.basemap(self.lon1d, self.lat1d)
#(self.x, self.y) = np.meshgrid(lon1d, lat1d)
v1d = np.reshape(pvar, (pvar.size, ))
v1d = np.reshape(pvar, (len(pvar), ))

contfill = self.basemap.contourf(self.x, self.y, v1d, tri=True,
levels=self.clevs, extend=self.extend,
Expand Down
2 changes: 1 addition & 1 deletion plot-utils/plottools.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ def plot(self, pvar, addmark=0, marker='x', size=3, color='green'):
self.fig = self.plt.figure()
self.ax = self.plt.subplot()

msg = ('plot variable min: %s, max: %s' % (pvar.min(), pvar.max()))
msg = ('plot variable min: %s, max: %s' % (np.min(pvar), np.max(pvar)))
print(msg)

(self.x, self.y) = self.basemap(self.lon1d, self.lat1d)
Expand Down
11 changes: 6 additions & 5 deletions soca-diff.py
Original file line number Diff line number Diff line change
Expand Up @@ -202,16 +202,16 @@ def plot_coast_lat_lon_line(self):
#------------------------------------------------------------------
if __name__== '__main__':
debug = 1
output = 0
output = 1
gridf = 'regrid/grids/ocn_2014_01.nc'
#file1 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t4n_80p/ocn.LETKF.an.2015-12-01T12:00:00Z.nc'
file1 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t4n_80p/ocn.LETKF.an.2015-12-01T12:00:00Z.nc'
#file1 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.24t2n_48p/ocn.LETKF.an.2015-12-01T12:00:00Z.nc'
#file1 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t2n_40p/new-run-output/ocn.LETKF.an.2015-12-01T12:00:00Z.nc'
#file2 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t2n_40p/ocn.LETKF.an.2015-12-01T12:00:00Z.nc'
file2 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t2n_40p/ocn.LETKF.an.2015-12-01T12:00:00Z.nc'
#file2 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.40t6n_240p/ocn.LETKF.an.2015-12-01T12:00:00Z.nc'

file1 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t2n_40p/new-run-output/ocn.LETKF.fc.2015-12-01T12:00:00Z.PT0S.nc'
file2 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t2n_40p/ocn.LETKF.fc.2015-12-01T12:00:00Z.PT0S.nc'
#file1 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t2n_40p/new-run-output/ocn.LETKF.fc.2015-12-01T12:00:00Z.PT0S.nc'
#file2 = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t2n_40p/ocn.LETKF.fc.2015-12-01T12:00:00Z.PT0S.nc'

ncg = netCDF4.Dataset(gridf, 'r')
lon = ncg.variables['geolon'][:,:]
Expand All @@ -232,6 +232,7 @@ def plot_coast_lat_lon_line(self):

lat1d = lat.flatten()
lon1d = lon.flatten()
lon1d = np.where(lon1d > 0, lon1d, lon1d+360.0)
gp.set_grid(lat1d, lon1d)

nlay, nlat, nlon = temp1.shape
Expand Down
9 changes: 5 additions & 4 deletions soca-tile-timing.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,9 +139,6 @@ def plot(self, pvar):
print('self.y.shape = ', self.y.shape)
print('v1d.shape = ', v1d.shape)

#contfill = self.basemap.contourf(self.x, self.y, v1d, tri=True,
# levels=self.clevs, extend=self.extend,
# alpha=self.alpha, cmap=self.cmapname)
contfill = self.plt.tricontourf(self.x, self.y, v1d,
alpha=self.alpha, cmap=self.cmapname)

Expand Down Expand Up @@ -290,12 +287,16 @@ def get_stats(self, flnm):

nl = 0
while(nl < num_lines):
if(lines[nl].find('Longitude:') > 0):
if(lines[nl].find('Longitude:') >= 0):
#print('line[%d]: %s' %(nl, lines[nl]))
item = lines[nl].split(':')
lonitem = item[1].split(',')
lonval = float(lonitem[0].strip())
latval = float(item[2].strip())
#print('item: ', item)
#print('lon: %f, lat: %f' %(lonval, latval))
if(lonval < 0.0):
lonval += 360.0
lon.append(lonval)
lat.append(latval)
#if(lines[nl].find('Longitude:') > 0):
Expand Down
5 changes: 3 additions & 2 deletions tile-timing.py
Original file line number Diff line number Diff line change
Expand Up @@ -296,6 +296,8 @@ def get_stats(self, flnm):
lonval = float(lonitem[0].strip())
latval = float(item[2].strip())
#print('lon: %f, lat: %f' %(lonval, latval))
if(lonval < 0.0):
lonval += 360.0
lon.append(lonval)
lat.append(latval)
#if(lines[nl].find('Longitude:') > 0):
Expand Down Expand Up @@ -383,8 +385,7 @@ def get_idx(self, name):
output = 0
#workdir = '/work2/noaa/gsienkf/weihuang/jedi/case_study/sondes/run_80.40t1n_36p/stdoutNerr'
#workdir = '/work2/noaa/gsienkf/weihuang/jedi/case_study/sondes/run_80.40t8n_312p/stdoutNerr'
#workdir = '/work2/noaa/gsienkf/weihuang/jedi/case_study/sondes/run_80.40t2n_78p/stdoutNerr'
workdir = '/work2/noaa/gsienkf/weihuang/ufs/soca/new-soca-solver/soca_solver.20t2n_40p/stdoutNerr'
workdir = '/work2/noaa/gsienkf/weihuang/jedi/case_study/sondes/run_80.40t2n_78p/stdoutNerr'

opts, args = getopt.getopt(sys.argv[1:], '', ['debug=', 'output=', 'workdir='])

Expand Down

0 comments on commit b6aef1c

Please sign in to comment.