def __init__(self,Wctrl_fpath,Wmod_fpath,vrbl,utc,lv=False, accum_hr=False,radar_datadir=False,thresh=False, footprint=500): self.utc = utc self.C = {} self.M = {} self.thresh = thresh self.footprint = footprint self.M['WRFOut'] = WRFOut(Wmod_fpath) self.dx = self.M['WRFOut'].dx self.compute_d(self.M['WRFOut']) if Wctrl_fpath is False and (vrbl=='REFL_comp' or vrbl=='cref'): use_radar_obs = True self.C['data'] = self.get_radar_verif(utc,radar_datadir) else: use_radar_obs = False Wctrl = WRFOut(Wctrl_fpath) # Get 2D grids for ctrl and model if vrbl == 'accum_precip': if not accum_hr: raise Exception("Need to set accumulation hours.") self.C['data'] = Wctrl.compute_accum_rain(utc,accum_hr)[0,0,:,:] self.M['data'] = self.M['WRFOut'].compute_accum_rain(utc,accum_hr)[0,0,:,:] else: self.M['data'] = self.M['WRFOut'].get(vrbl,level=lv,utc=utc)[0,0,:,:] if not use_radar_obs: self.C['data'] = Wctrl.get(vrbl,level=lv,utc=utc)[0,0,:,:] # Set negative values to 0 # if vrbl == 'REFL_comp': self.C['data'][self.C['data']<0] = 0 self.M['data'][self.M['data']<0] = 0 self.vrbl = vrbl """ fig, ax = plt.subplots(1) ax.pcolor(self.C['data']) fig.savefig('/home/jrlawson/public_html/bowecho/SALtests/obs_pcolor.png') plt.close(fig) fig, ax = plt.subplots(1) ax.pcolor(self.M['data']) fig.savefig('/home/jrlawson/public_html/bowecho/SALtests/mod_pcolor.png') plt.close(fig) import pdb; pdb.set_trace() """ self.identify_objects() self.compute_amplitude() self.compute_location() self.compute_structure() print("S = {0} A = {1} L = {2}".format(self.S,self.A,self.L))