MadisonBarks/weatherbot-radar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Dunno what I changed. I changed something.

Browse Source
This commit is contained in:
Madison
2025-06-21 11:34:14 -07:00
parent 7f382ba77f
commit e06dd6eb6d
1 changed files with 20 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
testfile cartopy.py
View File

@@ -39,7 +39,7 @@ def get_radar_frames(site, datetime_t, num_frames):
bucketListCurrentDate = list(bucket.objects.filter(Prefix = currentDateString)) bucketListCurrentDate = list(bucket.objects.filter(Prefix = currentDateString))
bucketListPreviousDate = list(bucket.objects.filter(Prefix = previousDateString)) bucketListPreviousDate = list(bucket.objects.filter(Prefix = previousDateString))
#we are going to create a list of keys and datetimes to allow easy searching # we are going to create a list of keys and datetimes to allow easy searching
keys = [] keys = []
datetimes = [] datetimes = []
@@ -102,7 +102,8 @@ def get_radar_from_aws(aws_object):
if not os.path.isfile(f'cache/rawradar/{filename}'): if not os.path.isfile(f'cache/rawradar/{filename}'):
printdebug('Cache Miss') printdebug('Cache Miss')
aws_object.download_file(f'cache/rawradar/{filename}') aws_object.download_file(f'cache/rawradar/{filename}')
radar = pyart.io.read(f'cache/rawradar/{filename}') #radar = pyart.io.read(f'cache/rawradar/{filename}')
radar = f'cache/rawradar/{filename}'
return radar return radar
def convert_fig_to_img(fig): def convert_fig_to_img(fig):
@@ -134,14 +135,14 @@ def generate_radar_image(radar_data, tiler, mercator):
radar_data.fields['reflectivity']['data'][:, -10:] = np.ma.masked radar_data.fields['reflectivity']['data'][:, -10:] = np.ma.masked
# gatefilter = pyart.filters.GateFilter(radar_data) gatefilter = pyart.filters.GateFilter(radar_data)
# gatefilter.exclude_transition() gatefilter.exclude_transition()
# gatefilter.exclude_masked("reflectivity") gatefilter.exclude_masked("reflectivity")
# gatefilter.exclude_below('reflectivity', 16) gatefilter.exclude_below('reflectivity', 5)
gatefilter = pyart.filters.moment_based_gate_filter(radar_data) #gatefilter = pyart.filters.moment_based_gate_filter(radar_data)
printdebug('Despeckling') printdebug('Despeckling')
despeckle_gatefilter = pyart.correct.despeckle_field(radar_data, 'reflectivity', gatefilter=gatefilter, size=20) despeckle_gatefilter = pyart.correct.despeckle_field(radar_data, 'reflectivity', gatefilter=gatefilter)
fig = plt.figure(figsize = [10,8]) fig = plt.figure(figsize = [10,8])
fig.patch.set_facecolor('none') fig.patch.set_facecolor('none')
@@ -152,13 +153,14 @@ def generate_radar_image(radar_data, tiler, mercator):
printdebug('Plotting!') printdebug('Plotting!')
display.plot_ppi_map('reflectivity', sweep = 0, resolution='10m', display.plot_ppi_map('reflectivity', sweep = 0, resolution='10m',
vmin = -8, vmax = 64, vmin = -8, vmax = 64,
cmap = "ChaseSpectral", #cmap = "ChaseSpectral",
cmap = "NWSRef",
min_lat = min_lat, min_lon = min_lon, min_lat = min_lat, min_lon = min_lon,
max_lat = max_lat, max_lon = max_lon, max_lat = max_lat, max_lon = max_lon,
lat_lines = lal, lon_lines = lol, lat_lines = lal, lon_lines = lol,
projection=projection, fig=fig, ax=ax, title_flag=False, projection=projection, fig=fig, ax=ax, title_flag=False,
lat_0=radar_data.latitude["data"][0], lon_0=radar_data.longitude["data"][0], lat_0=radar_data.latitude["data"][0], lon_0=radar_data.longitude["data"][0],
gatefilter=despeckle_gatefilter, add_grid_lines=False, alpha=0.8) gatefilter=gatefilter, add_grid_lines=False, alpha=0.8)
display.plot_range_ring(radar_data.range["data"][-1] / 1000.0) display.plot_range_ring(radar_data.range["data"][-1] / 1000.0)
@@ -166,11 +168,14 @@ def generate_radar_image(radar_data, tiler, mercator):
def _thread_generate_radar_image(radar_data, dataName, tiler, mercator, index, lock): def _thread_generate_radar_image(radar_data, dataName, tiler, mercator, index, lock):
if not os.path.isfile(f'cache/radarimages/{dataName}.png'): if not os.path.isfile(f'cache/radarimages/{dataName}.png'):
fig = generate_radar_image(radar_data, tiler, mercator) radarData = pyart.io.read(radar_data)
fig = generate_radar_image(radarData, tiler, mercator)
frame = convert_fig_to_img(fig) frame = convert_fig_to_img(fig)
frame.save(f'cache/radarimages/{dataName}.png') frame.save(f'cache/radarimages/{dataName}.png')
del radarData
lock.acquire() lock.acquire()
secondary_frames[index] = frame secondary_frames[index] = frame
#plt.clear()
lock.release() lock.release()
else: else:
frame = Image.open(f'cache/radarimages/{dataName}.png') frame = Image.open(f'cache/radarimages/{dataName}.png')
@@ -191,10 +196,12 @@ mercator = tiler.crs
radarKeys = list(radar.keys()) radarKeys = list(radar.keys())
if not os.path.isfile(f'cache/radarimages/{radarKeys[0]}.png'): if not os.path.isfile(f'cache/radarimages/{radarKeys[0]}.png'):
firstFrameFig = generate_radar_image(radar[radarKeys[0]], tiler, mercator) radarData = pyart.io.read(radar[radarKeys[0]])
firstFrameFig = generate_radar_image(radarData, tiler, mercator)
firstFrame = convert_fig_to_img(firstFrameFig) firstFrame = convert_fig_to_img(firstFrameFig)
firstFrame.save(f'cache/radarimages/{radarKeys[0]}.png') firstFrame.save(f'cache/radarimages/{radarKeys[0]}.png')
del radarData
else: else:
firstFrame = Image.open(f'cache/radarimages/{radarKeys[0]}.png') firstFrame = Image.open(f'cache/radarimages/{radarKeys[0]}.png')