exp_key = 'Shape' diff_count += 1 if diff_count > 1: exp_key = 'Combined' return data.waveBand, exp_key def setupDefaults(grid): ax = grid[0] ax.xaxis.set_major_locator(plt.MultipleLocator(5)) ax.set_xlim(-15, 15) ax.set_ylim(0, 50) plt.rcParams['font.size'] = 12 plots.defaults.axisDefaults.setup = setupDefaults data_cache = DataCache('ref_runs', make_key, ('band', 'exp')) data_cache.key_sorter = sorter wavelengths,exps = data_cache.sub_keys() for lam_text, exp_text in [('C', 'Control'), ('X', 'Control')]: data = data_cache[lam_text, exp_text] with datatypes.PlotInfoContext(wavelength=data.wavelength): for desc,moms in [('Single', (datatypes.Reflectivity, datatypes.DopplerVelocity, datatypes.SpectrumWidth)), ('Dual', (datatypes.ZDR, datatypes.RhoHV, datatypes.PhiDP)), ('Attenuation', (datatypes.Attenuation, datatypes.DiffAtten))]: source = 'calc' if desc=='Attenuation' else 'ts' moments = [data.fields.grab(moment, pol='H', source=source) for moment in moms] fig = plt.figure(figsize=(11, 6), dpi=200) if len(moments) == 3: rect = [0.07, 0.05, 0.88, 0.95] else:
def load_spatial_experiments(data_dir, glob='*'): data_cache = DataCache(data_dir, make_spatial_key, ('band', 'exp'), pattern=glob) data_cache.key_sorter = spatial_sorter wavelengths,exps = data_cache.sub_keys() return data_cache