Addition of qvp functions from MeteoSwiss Py-ART + support of demixing in hydroclass_semisupervised

examples/correct/plot_attenuation.py

@@ -7,6 +7,7 @@
 for a polarimetric radar using a Z-PHI method implemented in Py-ART.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/correct/plot_dealias.py

@@ -6,6 +6,7 @@
 In this example doppler velocities are dealiased using the ial condition of the dealiasing,
 using the region-based dealiasing algorithm in Py-ART.
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/io/plot_nexrad_data_aws.py

@@ -7,6 +7,7 @@
 and plot quick looks of the datasets.
 
 """
+
 print(__doc__)
 
 # Author: Max Grover (mgrover@anl.gov)

examples/io/plot_older_nexrad_data_aws.py

@@ -7,6 +7,7 @@
 to 2008 that are missing some coordinate metadata.
 
 """
+
 print(__doc__)
 
 

examples/mapping/plot_compare_two_radars_gatemapper.py

@@ -7,6 +7,7 @@
 another radar in Antenna coordinates and compare the fields.
 
 """
+
 print(__doc__)
 
 # Author: Max Grover (mgrover@anl.gov) and Bobby Jackson (rjackson@anl.gov)

examples/mapping/plot_grid_single_sweep_ppi.py

@@ -13,6 +13,7 @@
 negligible at low elevation angles common to PPI sweeps.
 
 """
+
 print(__doc__)
 
 # =====================

examples/mapping/plot_map_one_radar_to_grid.py

@@ -7,6 +7,7 @@
 Cartesian grid.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/mapping/plot_map_two_radars_to_grid.py

@@ -7,6 +7,7 @@
 coordinates to a Cartesian grid.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/plotting/plot_choose_a_colormap.py

@@ -7,6 +7,7 @@
 and how to add them to your own plots.
 
 """
+
 print(__doc__)
 
 # Author: Max Grover (mgrover@anl.gov)

examples/plotting/plot_cross_section.py

@@ -7,6 +7,7 @@
 of your radar grid using the GridMapDisplay
 
 """
+
 print(__doc__)
 
 # Author: Max Grover (mgrover@anl.gov)

examples/plotting/plot_modify_colorbar.py

@@ -7,6 +7,7 @@
 within a Py-ART display object.
 
 """
+
 print(__doc__)
 
 # Author: Joe O'Brien (obrienj@anl.gov)

examples/plotting/plot_nexrad_multiple_moments.py

@@ -7,6 +7,7 @@
 NEXRAD Archive file.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/plotting/plot_nexrad_reflectivity.py

@@ -7,6 +7,7 @@
 NEXRAD file.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/plotting/plot_ppi_cfradial.py

@@ -6,6 +6,7 @@
 An example which creates a PPI plot of a Cfradial file.
 
 """
+
 print(__doc__)
 
 # Author: Max Grover (mgrover@anl.gov)

examples/plotting/plot_ppi_mdv.py

@@ -6,6 +6,7 @@
 An example which creates a PPI plot of a MDV file using a RadarDisplay object.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/plotting/plot_rhi_cfradial.py

@@ -7,6 +7,7 @@
 a RadarDisplay object.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/plotting/plot_rhi_cfradial_singlescan.py

@@ -7,6 +7,7 @@
 a RadarDisplay object.
 
 """
+
 print(__doc__)
 
 import matplotlib.pyplot as plt

examples/plotting/plot_rhi_contours_differential_reflectivity.py

@@ -7,6 +7,7 @@
 and adding differnential Reflectivity contours from the same MDV file.
 
 """
+
 print(__doc__)
 
 # Author: Cory Weber (cweber@anl.gov)

examples/plotting/plot_rhi_data_overlay.py

@@ -7,6 +7,7 @@
 and adding Reflectivity contours from the same MDV file.
 
 """
+
 print(__doc__)
 
 # Author: Cory Weber (cweber@anl.gov)

examples/plotting/plot_rhi_mdv.py

@@ -6,6 +6,7 @@
 An example which creates a RHI plot of a MDV file using a RadarDisplay object.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

examples/plotting/plot_rhi_two_panel.py

@@ -7,6 +7,7 @@
 included in the two panels are reflectivity and doppler velocity.
 
 """
+
 print(__doc__)
 
 # Author: Max Grover (mgrover@anl.gov)

examples/plotting/plot_xsect.py

@@ -7,6 +7,7 @@
 of PPI scans and plots both cross sections.
 
 """
+
 print(__doc__)
 
 # Author: Jonathan J. Helmus (jhelmus@anl.gov)

pyart/__check_build/__init__.py

@@ -1,6 +1,7 @@
 """ Module to give helpful messages to the user that did not
 compile Py-ART properly.
 """
+
 import os
 
 INPLACE_MSG = """

pyart/core/__init__.py

@@ -13,6 +13,7 @@
 from .transforms import cartesian_vectors_to_geographic  # noqa
 from .transforms import geographic_to_cartesian  # noqa
 from .transforms import geographic_to_cartesian_aeqd  # noqa
+from .transforms import cartesian_to_antenna  # noqa
 from .wind_profile import HorizontalWindProfile  # noqa
 
 __all__ = [s for s in dir() if not s.startswith("_")]

pyart/core/transforms.py

@@ -78,6 +78,35 @@ def antenna_to_cartesian(ranges, azimuths, elevations):
     return x, y, z
 
 
+def cartesian_to_antenna(x, y, z):
+    """
+    Returns antenna coordinates from Cartesian coordinates.
+
+    Parameters
+    ----------
+    x, y, z : array
+        Cartesian coordinates in meters from the radar.
+
+    Returns
+    -------
+    ranges : array
+        Distances to the center of the radar gates (bins) in m.
+    azimuths : array
+        Azimuth angle of the radar in degrees. [-180., 180]
+    elevations : array
+        Elevation angle of the radar in degrees.
+    ke : float, optional
+        Effective radius scale factor
+
+    """
+    ranges = np.sqrt(x**2.0 + y**2.0 + z**2.0)
+    elevations = np.rad2deg(np.arctan(z / np.sqrt(x**2.0 + y**2.0)))
+    azimuths = np.rad2deg(np.arctan2(x, y))  # [-180, 180]
+    azimuths[azimuths < 0.0] += 360.0  # [0, 360]
+
+    return ranges, azimuths, elevations
+
+
 def antenna_vectors_to_cartesian(ranges, azimuths, elevations, edges=False):
     """
     Calculate Cartesian coordinate for gates from antenna coordinate vectors.

pyart/core/wind_profile.py

@@ -3,7 +3,6 @@
 
 """
 
-
 import numpy as np
 
 

pyart/correct/phase_proc.py

@@ -108,11 +108,7 @@ def fzl_index(fzl, ranges, elevation, radar_height):
     p_r = 4.0 * Re / 3.0
     z = (
         radar_height
-        + (
-            ranges**2
-            + p_r**2
-            + 2.0 * ranges * p_r * np.sin(elevation * np.pi / 180.0)
-        )
+        + (ranges**2 + p_r**2 + 2.0 * ranges * p_r * np.sin(elevation * np.pi / 180.0))
         ** 0.5
         - p_r
     )

pyart/graph/_cm_colorblind.py

@@ -2,6 +2,7 @@
 Data for colorblind friendly radar colormaps
 
 """
+
 import os
 
 import numpy as np

pyart/graph/gridmapdisplay_basemap.py

@@ -3,7 +3,6 @@
 
 """
 
-
 import matplotlib.pyplot as plt
 import numpy as np
 

pyart/io/mdv_grid.py

@@ -3,7 +3,6 @@
 
 """
 
-
 import datetime
 import warnings
 

pyart/io/nexrad_common.py

@@ -2,6 +2,7 @@
 Data and functions common to all types of NEXRAD files.
 
 """
+
 # The functions in this module are intended to be used in other
 # nexrad related modules. The functions are not and should not be
 # exported into the pyart.io namespace.

pyart/io/nexrad_level2.py

@@ -2,7 +2,6 @@
 Functions for reading NEXRAD level 2 files.
 """
 
-
 # This file is part of the Py-ART, the Python ARM Radar Toolkit
 # https://github.com/ARM-DOE/pyart
 

pyart/io/nexrad_level3.py

@@ -761,7 +761,7 @@ def _int16_to_float16(val):
     ("block_length", INT4),  # Length of block in bytes
     ("layers", INT2),  # Number of data layers
     ("layer_divider", INT2),  # Delineate data layers, -1
-    ("layer_length", INT4)  # Length of data layer in bytes
+    ("layer_length", INT4),  # Length of data layer in bytes
     # Display data packets
 )
 

pyart/map/gates_to_grid.py

@@ -2,6 +2,7 @@
 Generate a Cartesian grid by mapping from radar gates onto the grid.
 
 """
+
 import gc
 import warnings
 

pyart/retrieve/__init__.py

@@ -21,7 +21,8 @@
 from .qpe import est_rain_rate_zkdp  # noqa
 from .qpe import est_rain_rate_zpoly  # noqa
 from .qpe import ZtoR  # noqa
-from .qvp import quasi_vertical_profile  # noqa
+from .qvp import quasi_vertical_profile, compute_qvp, compute_rqvp  # noqa
+from .qvp import compute_evp, compute_svp, compute_vp, compute_ts_along_coord  # noqa
 from .simple_moment_calculations import calculate_snr_from_reflectivity  # noqa
 from .simple_moment_calculations import calculate_velocity_texture  # noqa
 from .simple_moment_calculations import compute_cdr  # noqa

pyart/retrieve/_echo_class_wt.py

@@ -13,7 +13,6 @@
     atwt2d
 """
 
-
 import numpy as np