forked from 3rdparty/wrf-python
Bill Ladwig
8 years ago
4 changed files with 0 additions and 466 deletions
@ -1,291 +0,0 @@ |
|||||||
#!/usr/bin/env python |
|
||||||
|
|
||||||
import traceback |
|
||||||
import sys |
|
||||||
import sqlite3 |
|
||||||
from datetime import datetime as dt |
|
||||||
|
|
||||||
import numpy as n |
|
||||||
import matplotlib |
|
||||||
matplotlib.use('agg') |
|
||||||
import matplotlib.pyplot as plt |
|
||||||
|
|
||||||
#from wrf.core import Constants |
|
||||||
#from wrf.var.temp import calc_temp |
|
||||||
#from wrf.plot.matplotlib.defaults import (get_basemap, get_default_map_opts, |
|
||||||
# get_null_opts) |
|
||||||
|
|
||||||
#from wrf.plot.matplotlib.helper import (add_plot_info_text, plot_map, |
|
||||||
# plot_contourf) |
|
||||||
|
|
||||||
#__all__ = ["plot_2d"] |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def get_basemap(wrfnc): |
|
||||||
#TODO: handle multiple projections |
|
||||||
lat2d = wrfnc.variables["XLAT"][0,:,:] |
|
||||||
lon2d = wrfnc.variables["XLONG"][0,:,:] |
|
||||||
|
|
||||||
ny = len(wrfnc.dimensions["south_north"]) |
|
||||||
nx = len(wrfnc.dimensions["west_east"]) |
|
||||||
nz = len(wrfnc.dimensions["bottom_top"]) |
|
||||||
|
|
||||||
dx = wrfnc.DX |
|
||||||
dy = wrfnc.DY |
|
||||||
center_lat = wrfnc.CEN_LAT |
|
||||||
center_lon = wrfnc.CEN_LON |
|
||||||
true_lat1 = wrfnc.TRUELAT1 |
|
||||||
true_lat2 = wrfnc.TRUELAT2 |
|
||||||
|
|
||||||
basemap = Basemap(projection="lcc", |
|
||||||
lat_0=center_lat, |
|
||||||
lon_0=center_lon, |
|
||||||
lat_1=true_lat1, |
|
||||||
lat_2=true_lat2, |
|
||||||
llcrnrlon=lon2d[0,0], |
|
||||||
llcrnrlat=lat2d[0,0], |
|
||||||
urcrnrlon=lon2d[ny-1, nx-1], |
|
||||||
urcrnrlat=lat2d[ny-1, nx-1], |
|
||||||
resolution='i') |
|
||||||
|
|
||||||
return basemap |
|
||||||
|
|
||||||
def get_default_map_opts(): |
|
||||||
landcolor = (204/255.0, 204/255.0, 153/255.0) |
|
||||||
oceancolor = (102/255.0, 204/255.0, 255/255.0) |
|
||||||
return MapOptions( |
|
||||||
coastargs = {"linewidth":1.0, "linestyle":'solid', "color":'k'}, |
|
||||||
countryargs = {"linewidth":0.5, "linestyle":'solid', "color":'k'}, |
|
||||||
stateargs = {"linewidth":0.5, "linestyle":'solid', "color":'k'}, |
|
||||||
mapboundaryargs = {"color":'k', "linewidth":1.0, |
|
||||||
"fill_color":oceancolor}, |
|
||||||
continentfillargs = {"color":landcolor, "lake_color":oceancolor, |
|
||||||
"zorder":0}) |
|
||||||
|
|
||||||
|
|
||||||
def get_null_opts(): |
|
||||||
return FilledContourOptions(fcontourargs={}, |
|
||||||
colorbarargs={"location" : "bottom", |
|
||||||
"size" : "5%", "pad" : "2%", |
|
||||||
"extend" : "both"}) |
|
||||||
|
|
||||||
def add_plot_info_text(ax, |
|
||||||
top_left_text="", top_right_text="", |
|
||||||
bot_left_text="", bot_right_text=""): |
|
||||||
|
|
||||||
plt.ioff() |
|
||||||
if top_left_text != "": |
|
||||||
plt.text(0.005,.995,top_left_text, |
|
||||||
bbox=dict(facecolor="white"), |
|
||||||
horizontalalignment="left", |
|
||||||
verticalalignment="top", |
|
||||||
transform = ax.transAxes, |
|
||||||
fontsize=10) |
|
||||||
|
|
||||||
if top_right_text != "": |
|
||||||
plt.text(.995,.995,top_right_text, |
|
||||||
bbox=dict(facecolor="white"), |
|
||||||
horizontalalignment="right", |
|
||||||
verticalalignment="top", |
|
||||||
transform = ax.transAxes, |
|
||||||
fontsize=10) |
|
||||||
|
|
||||||
if bot_left_text != "": |
|
||||||
plt.text(0.005,0.005,bot_left_text, |
|
||||||
bbox=dict(facecolor="white"), |
|
||||||
horizontalalignment="left", |
|
||||||
verticalalignment="bottom", |
|
||||||
transform = ax.transAxes, |
|
||||||
fontsize=10) |
|
||||||
|
|
||||||
if bot_right_text != "": |
|
||||||
plt.text(.995,0.005,bot_right_text, |
|
||||||
bbox=dict(facecolor="white"), |
|
||||||
horizontalalignment="right", |
|
||||||
verticalalignment="bottom", |
|
||||||
transform = ax.transAxes, |
|
||||||
fontsize=10) |
|
||||||
|
|
||||||
def plot_map(basemap, mapoptions): |
|
||||||
plt.ioff() |
|
||||||
if mapoptions.mapboundaryargs is not None: |
|
||||||
basemap.drawmapboundary(**mapoptions.mapboundaryargs) |
|
||||||
|
|
||||||
if mapoptions.continentfillargs is not None: |
|
||||||
basemap.fillcontinents(**mapoptions.continentfillargs) |
|
||||||
|
|
||||||
if mapoptions.coastargs is not None: |
|
||||||
basemap.drawcoastlines(**mapoptions.coastargs) |
|
||||||
|
|
||||||
if mapoptions.countryargs is not None: |
|
||||||
basemap.drawcountries(**mapoptions.countryargs) |
|
||||||
|
|
||||||
if mapoptions.stateargs is not None: |
|
||||||
basemap.drawstates(**mapoptions.stateargs) |
|
||||||
|
|
||||||
if mapoptions.countyargs is not None: |
|
||||||
basemap.drawcounties(**mapoptions.countyargs) |
|
||||||
|
|
||||||
if mapoptions.riverargs is not None: |
|
||||||
basemap.drawrivers(**mapoptions.riverargs) |
|
||||||
|
|
||||||
def plot_contourf(x,y,data,basemap, contourfoptions): |
|
||||||
plt.ioff() |
|
||||||
cs1 = None |
|
||||||
if contourfoptions.contourargs is not None: |
|
||||||
cs1 = basemap.contour(x,y,data,**contourfoptions.contourargs) |
|
||||||
|
|
||||||
cs2 = None |
|
||||||
if contourfoptions.fcontourargs is not None: |
|
||||||
cs2 = basemap.contourf(x,y,data,**contourfoptions.fcontourargs) |
|
||||||
|
|
||||||
cb = None |
|
||||||
if contourfoptions.colorbarargs is not None: |
|
||||||
cb = basemap.colorbar(cs2, **contourfoptions.colorbarargs) |
|
||||||
|
|
||||||
if contourfoptions.labelargs is not None: |
|
||||||
plt.clabel(cs1, **contourfoptions.labelargs) |
|
||||||
|
|
||||||
return cs1, cs2, cb |
|
||||||
|
|
||||||
def get_null_opts(): |
|
||||||
return FilledContourOptions(fcontourargs={}, |
|
||||||
colorbarargs={"location" : "bottom", |
|
||||||
"size" : "5%", "pad" : "2%", |
|
||||||
"extend" : "both"}) |
|
||||||
|
|
||||||
class MapOptions(object): |
|
||||||
def __init__(self, |
|
||||||
coastargs = None, |
|
||||||
countyargs = None, |
|
||||||
countryargs = None, |
|
||||||
riverargs = None, |
|
||||||
stateargs = None, |
|
||||||
mapboundaryargs = None, |
|
||||||
continentfillargs = None): |
|
||||||
self.coastargs = coastargs |
|
||||||
self.countyargs = countyargs |
|
||||||
self.countryargs = countryargs |
|
||||||
self.riverargs = riverargs |
|
||||||
self.stateargs = stateargs |
|
||||||
self.mapboundaryargs = mapboundaryargs |
|
||||||
self.continentfillargs = continentfillargs |
|
||||||
|
|
||||||
def plot_2d(wrfnc, varname=None, outfile=None, title=None, |
|
||||||
map_opts=None, plot_opts=None, |
|
||||||
top_left_info="", top_right_info="", |
|
||||||
bot_left_info="", bot_right_info="", |
|
||||||
wks_type="png", var=None, |
|
||||||
time_in=0): |
|
||||||
|
|
||||||
try: |
|
||||||
plt.ioff() |
|
||||||
print "generating %s.%s" % (outfile, wks_type) |
|
||||||
if var is not None: |
|
||||||
field = var |
|
||||||
elif varname is not None: |
|
||||||
field = wrfnc.variables[varname][time_in,:,:] |
|
||||||
|
|
||||||
lat2d = wrfnc.variables["XLAT"][time_in,:,:] |
|
||||||
lon2d = wrfnc.variables["XLONG"][time_in,:,:] |
|
||||||
times = wrfnc.variables["Times"][time_in,:] |
|
||||||
model_time = "".join(times) |
|
||||||
start_date = dt.strptime(model_time, "%Y-%m-%d_%H:%M:%S") |
|
||||||
|
|
||||||
ny = len(wrfnc.dimensions["south_north"]) |
|
||||||
nx = len(wrfnc.dimensions["west_east"]) |
|
||||||
nz = len(wrfnc.dimensions["bottom_top"]) |
|
||||||
|
|
||||||
fig = plt.figure(figsize=(8,8), dpi=200) |
|
||||||
ax = fig.add_axes([0.1,0.1,0.8,0.8]) |
|
||||||
|
|
||||||
bm = get_basemap(wrfnc) |
|
||||||
if map_opts is None: |
|
||||||
map_opts = get_default_map_opts() |
|
||||||
if plot_opts is None: |
|
||||||
plot_opts = get_null_opts() |
|
||||||
|
|
||||||
x,y = bm(lon2d, lat2d) |
|
||||||
|
|
||||||
plot_map(bm,map_opts) |
|
||||||
|
|
||||||
plt.xticks(rotation=70) |
|
||||||
tplot = plot_contourf(x,y,field,bm,plot_opts) |
|
||||||
|
|
||||||
|
|
||||||
add_plot_info_text(ax, |
|
||||||
top_left_info, top_right_info, |
|
||||||
bot_left_info, bot_right_info) |
|
||||||
|
|
||||||
ax.set_title(title,fontdict={"fontsize" : 20}) |
|
||||||
|
|
||||||
plt.savefig("%s.%s" % (outfile, wks_type)) |
|
||||||
|
|
||||||
plt.clf() |
|
||||||
plt.close(fig) |
|
||||||
except: |
|
||||||
# print the stack trace since it will be lost when used in a |
|
||||||
# multiprocessing worker. |
|
||||||
print traceback.format_exc() |
|
||||||
raise |
|
||||||
finally: |
|
||||||
sys.stdout.flush() |
|
||||||
|
|
||||||
def main(): |
|
||||||
parser = argparse.ArgumentParser(description="Generate meteorological " |
|
||||||
"plots for a specific data file") |
|
||||||
parser.add_argument("-v", "--var", required=True, |
|
||||||
help="variable name") |
|
||||||
parser.add_argument("-f", "--filename", required=True, |
|
||||||
help="WRF file to plot") |
|
||||||
parser.add_argument("-o", "--outdir", default=".", required=False, |
|
||||||
help="output directory for images") |
|
||||||
parser.add_argument("-l", "--levels", required=False, type=float, |
|
||||||
nargs="+", |
|
||||||
default=None, |
|
||||||
help=("the start, end, and increment for contour levels" |
|
||||||
" as a list of items with spaces between them" |
|
||||||
"example: 1 10 2 ")) |
|
||||||
parser.add_argument("-c", "--customlevels", required=False, type=float, |
|
||||||
nargs="+", |
|
||||||
default=None, |
|
||||||
help=("a list of space delimited contour levels" |
|
||||||
"example: 1 2 3 4 5 19 28 200 ")) |
|
||||||
|
|
||||||
args = parser.parse_args() |
|
||||||
|
|
||||||
if not os.path.exists(args.filename): |
|
||||||
raise RuntimeError ("%s not found" % args.filename) |
|
||||||
|
|
||||||
if not os.path.exists(args.outdir): |
|
||||||
os.makedirs(args.outdir) |
|
||||||
|
|
||||||
basename = os.path.basename(args.filename) |
|
||||||
wrfnc = NetCDF(args.filename, mode='r') |
|
||||||
|
|
||||||
outfile = os.path.join(args.outdir, domain, args.var, "%s.%s" % (basename,args.var)) |
|
||||||
if not os.path.exists(os.path.dirname(outfile)): |
|
||||||
os.makedirs(os.path.dirname(outfile)) |
|
||||||
|
|
||||||
if args.levels is not None or args.customlevels is not None: |
|
||||||
plot_opts = get_null_opts() |
|
||||||
if args.levels is not None: |
|
||||||
if len(args.levels) < 2 or len(args.levels) > 3: |
|
||||||
raise RuntimeError("levels argument is invalid") |
|
||||||
plot_opts.fcontourargs["levels"] = [x for x in n.arange(args.levels[0], |
|
||||||
args.levels[1], |
|
||||||
args.levels[2])] |
|
||||||
plot_opts.fcontourargs["extend"] = "both" |
|
||||||
elif args.customlevels is not None: |
|
||||||
plot_opts.fcontourargs["levels"] = args.customlevels |
|
||||||
plot_opts.fcontourargs["extend"] = "both" |
|
||||||
else: |
|
||||||
plot_opts = None |
|
||||||
|
|
||||||
plot_2d(wrfnc, args.var, outfile, "%s"%args.var, |
|
||||||
plot_opts = plot_opts) |
|
||||||
|
|
||||||
if __name__ == "__main__": |
|
||||||
main() |
|
||||||
@ -1,32 +0,0 @@ |
|||||||
#!/usr/bin/env python |
|
||||||
import os |
|
||||||
import argparse |
|
||||||
|
|
||||||
import Ngl |
|
||||||
|
|
||||||
from wrf.system import SOMMemberPlotSystem |
|
||||||
|
|
||||||
if __name__ == "__main__": |
|
||||||
|
|
||||||
parser = argparse.ArgumentParser(description="Generate meteorological " |
|
||||||
"plots for SOM members") |
|
||||||
parser.add_argument("-c", "--casename", required=True, |
|
||||||
help="the case name (e.g. 'site1-october')") |
|
||||||
parser.add_argument("-s", "--somid", required=True, |
|
||||||
help="the SOM ID to use (e.g. 'SOM001')") |
|
||||||
parser.add_argument("-p", "--caseparent", required=False, |
|
||||||
default="/projectw/reanalyses/1.2", |
|
||||||
help=("the case parent directory " |
|
||||||
"[default: /projectw/reanalyses/1.2]")) |
|
||||||
args = parser.parse_args() |
|
||||||
|
|
||||||
parentdir = os.path.expanduser(os.path.expandvars(args.caseparent)) |
|
||||||
casename = args.casename |
|
||||||
somid = args.somid |
|
||||||
|
|
||||||
sys = SOMMemberPlotSystem(parentdir, casename, somid) |
|
||||||
|
|
||||||
try: |
|
||||||
sys.run() |
|
||||||
finally: |
|
||||||
Ngl.end() |
|
||||||
@ -1,143 +0,0 @@ |
|||||||
|
|
||||||
import wrf.var as w |
|
||||||
import numpy as n |
|
||||||
|
|
||||||
from netCDF4 import Dataset as NetCDF |
|
||||||
|
|
||||||
def main(): |
|
||||||
wrfnc = NetCDF("/Users/bladwig/wrfout_d03_2003-05-07_09:00:00") |
|
||||||
|
|
||||||
# Cape NO RESULTS FOR LCL OR LFC |
|
||||||
cape, cin, lcl, lfc = w.getvar(wrfnc, "cape2d") |
|
||||||
#cape, cin = w.getvar(wrfnc, "cape3d") |
|
||||||
print n.amax(cape) |
|
||||||
print n.amax(cin) |
|
||||||
print n.amax(lcl) |
|
||||||
print n.amax(lfc) |
|
||||||
|
|
||||||
|
|
||||||
# DBZ |
|
||||||
dbz = w.getvar(wrfnc, "dbz") |
|
||||||
print n.amax(dbz) |
|
||||||
|
|
||||||
# DP |
|
||||||
dp = w.getvar(wrfnc, "dp", units="f") |
|
||||||
print n.amax(dp) |
|
||||||
|
|
||||||
dp2 = w.getvar(wrfnc, "dp2m", units="f") |
|
||||||
print n.amax(dp2) |
|
||||||
|
|
||||||
# Height |
|
||||||
ht = w.getvar(wrfnc, "height", msl=False, units="m") |
|
||||||
print n.amax(ht) |
|
||||||
|
|
||||||
geopt = w.getvar(wrfnc, "geopt") |
|
||||||
print n.amax(geopt) |
|
||||||
|
|
||||||
# Helicity |
|
||||||
srh = w.getvar(wrfnc, "srh") |
|
||||||
print n.amax(srh) |
|
||||||
|
|
||||||
uhel = w.getvar(wrfnc, "uhel") |
|
||||||
print n.amax(uhel) |
|
||||||
|
|
||||||
# Omega (Not sure if this is correct, and units aren't C) |
|
||||||
omega = w.getvar(wrfnc, "omega") |
|
||||||
print n.amax(omega) |
|
||||||
|
|
||||||
# Precip Water (NOT SURE) |
|
||||||
pw = w.getvar(wrfnc, "pw") |
|
||||||
print n.amax(pw) |
|
||||||
|
|
||||||
# RH |
|
||||||
rh = w.getvar(wrfnc, "rh") |
|
||||||
print n.amax(rh) |
|
||||||
|
|
||||||
rh2 = w.getvar(wrfnc, "rh2m") |
|
||||||
print n.amax(rh2) |
|
||||||
|
|
||||||
# SLP |
|
||||||
slp = w.getvar(wrfnc, "slp", units="hpa") |
|
||||||
print n.amax(slp) |
|
||||||
|
|
||||||
# TEMP |
|
||||||
t = w.getvar(wrfnc, "temp", units="f") |
|
||||||
print n.amax(t) |
|
||||||
|
|
||||||
# ETH VALUES SEEM HIGH.... |
|
||||||
eth = w.getvar(wrfnc, "theta_e", units="k") |
|
||||||
print n.amax(eth) |
|
||||||
|
|
||||||
tv = w.getvar(wrfnc, "tv", units="k") |
|
||||||
print n.amax(tv) |
|
||||||
|
|
||||||
# Note: NCL says this is in 'C', but appears to be 'K' |
|
||||||
tw = w.getvar(wrfnc, "tw", units="f") |
|
||||||
print n.amax(tw) |
|
||||||
|
|
||||||
# WIND |
|
||||||
umet,vmet = w.getvar(wrfnc, "uvmet", units="kts") |
|
||||||
print n.amax(umet) |
|
||||||
print n.amax(vmet) |
|
||||||
|
|
||||||
umet10,vmet10 = w.getvar(wrfnc, "uvmet10", units="kts") |
|
||||||
print n.amax(umet10) |
|
||||||
print n.amax(vmet10) |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# TERRAIN |
|
||||||
ter = w.getvar(wrfnc, "terrain", units="dm") |
|
||||||
print n.amax(ter) |
|
||||||
|
|
||||||
# VORTICITY |
|
||||||
avo = w.getvar(wrfnc, "avo") |
|
||||||
print n.amax(avo) |
|
||||||
|
|
||||||
pvo = w.getvar(wrfnc, "pvo") |
|
||||||
print n.amax(pvo) |
|
||||||
|
|
||||||
# LAT/LON |
|
||||||
lat = w.getvar(wrfnc, "lat") |
|
||||||
print n.amax(lat) |
|
||||||
print n.amin(lat) |
|
||||||
|
|
||||||
lon = w.getvar(wrfnc, "lon") |
|
||||||
print n.amax(lon) |
|
||||||
print n.amin(lon) |
|
||||||
|
|
||||||
i,j = w.get_ij(wrfnc, -97.516540, 35.467787) |
|
||||||
print i,j |
|
||||||
|
|
||||||
lon, lat = w.get_ll(wrfnc, 33.5, 33.5) |
|
||||||
print lon, lat |
|
||||||
|
|
||||||
#ETA -- Result somewhat different than geopt |
|
||||||
z = w.convert_eta(wrfnc, msl=False, units="m") |
|
||||||
print n.amax(z) |
|
||||||
|
|
||||||
diff = n.abs(z - ht)/ht * 100.0 |
|
||||||
print n.amin(diff), n.amax(diff) |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__": |
|
||||||
main() |
|
||||||
Loading…
Reference in new issue