import aidas
import cartopy.crs as ccrs
import matplotlib.pyplot as plt

fig, axes = plt.subplots(1, 3, figsize=(15, 5),
                         subplot_kw=dict(projection=ccrs.PlateCarree()),
                         gridspec_kw={'wspace': 0.25})

for ax, min_area in zip(axes, (0.25, 16., 400.)):
    rad_scan = aidas.model.detect_velocity_waves(
        'KOKX', rad_time='2010-12-26T23:45:15', max_range=137000.,
        min_area=min_area)
    aidas.vis.plot_velocity_wave_mask(
        rad_scan, ax=ax, title=f"min_area = {min_area:g} km$^2$")
plt.show()