Merge

Author: PatrickSampaioUSP

data/weather/EuroC_pressure_levels_reanalysis_2001-2021.nc

@@ -27,7 +27,7 @@ The following packages are needed in order to run RocketPy:
 - timezonefinder
 - simplekml
 - ipywidgets >= 7.6.3
-- json
+- jsonpickle
 
 
 All of these packages, with the exception of netCDF4, should be automatically installed when RocketPy is installed using either ``pip`` or ``conda``.
@@ -50,6 +50,7 @@ The packages needed can be installed via ``pip`` by running the following lines
     pip install pytz
     pip install timezonefinder
     pip install simplekml
+    pip install jsonpickle
 
 Installing Required Packages Using ``conda``
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

data/weather/EuroC_single_level_reanalysis_2002_2021.nc

@@ -8,4 +8,4 @@ requests
 pytz
 timezonefinder
 simplekml
-json
+jsonpickle

docs/notebooks/environment_analysis_EuroC_example.ipynb

@@ -3,10 +3,12 @@
 __copyright__ = "Copyright 20XX, RocketPy Team"
 __license__ = "MIT"
 
+import matplotlib.pyplot as plt
 import numpy as np
 from scipy.integrate import solve_ivp
 
 from .Environment import Environment
+from .EnvironmentAnalysis import EnvironmentAnalysis
 from .Function import Function
 
 
@@ -197,3 +199,135 @@ def du(z, u):
         velocityFunction()
 
     return altitudeFunction, velocityFunction, final_sol
+
+
+def compareEnvironments(env1, env2, names=["env1", "env2"]):
+    """Compares two environments and plots everything.
+    Useful when comparing Environment Analysis results against forecasts.
+
+    Parameters
+    ----------
+    env1 : Environment or EnvironmentAnalysis
+        Environment to compare with.
+    env2: Environment or EnvironmentAnalysis
+        Environment to compare with.
+
+    Returns
+    -------
+    diff: Dict
+        Dictionary with the differences.
+    """
+
+    # Raise TypeError if env1 or env2 are not Environment nor EnvironmentAnalysis
+    if not isinstance(env1, Environment) and not isinstance(env1, EnvironmentAnalysis):
+        raise TypeError("env1 must be an Environment or EnvironmentAnalysis object")
+
+    if not isinstance(env2, Environment) and not isinstance(env2, EnvironmentAnalysis):
+        raise TypeError("env2 must be an Environment or EnvironmentAnalysis object")
+
+    # If instances are Environment Analysis, get the equivalent environment
+    if isinstance(env1, EnvironmentAnalysis):
+        env1.process_temperature_profile_over_average_day()
+        env1.process_pressure_profile_over_average_day()
+        env1.process_wind_velocity_x_profile_over_average_day()
+        env1.process_wind_velocity_y_profile_over_average_day()
+        print()
+    if isinstance(env2, EnvironmentAnalysis):
+        env2.process_temperature_profile_over_average_day()
+        env2.process_pressure_profile_over_average_day()
+        env2.process_wind_velocity_x_profile_over_average_day()
+        env2.process_wind_velocity_y_profile_over_average_day()
+        print()
+
+    # Plot graphs
+    print("\n\nAtmospheric Model Plots")
+    # Create height grid
+    grid = np.linspace(env1.elevation, env1.maxExpectedHeight)
+
+    # Create figure
+    plt.figure(figsize=(9, 9))
+
+    # Create wind speed and wind direction subplot
+    ax1 = plt.subplot(221)
+    ax1.plot(
+        [env1.windSpeed(i) for i in grid],
+        grid,
+        "#ff7f0e",
+        label="Speed of Sound " + names[0],
+    )
+    ax1.plot(
+        [env2.windSpeed(i) for i in grid],
+        grid,
+        "#ff7f0e",
+        label="Speed of Sound " + names[1],
+    )
+    ax1.set_xlabel("Wind Speed (m/s)", color="#ff7f0e")
+    ax1.tick_params("x", colors="#ff7f0e")
+    # ax1up = ax1.twiny()
+    # ax1up.plot(
+    #     [env1.windDirection(i) for i in grid],
+    #     grid,
+    #     color="#1f77b4",
+    #     label="Density "+names[0],
+    # )
+    # ax1up.plot(
+    #     [env2.windDirection(i) for i in grid],
+    #     grid,
+    #     color="#1f77b4",
+    #     label="Density "+names[1],
+    # )
+    # ax1up.set_xlabel("Wind Direction (°)", color="#1f77b4")
+    # ax1up.tick_params("x", colors="#1f77b4")
+    # ax1up.set_xlim(0, 360)
+    ax1.set_ylabel("Height Above Sea Level (m)")
+    ax1.grid(True)
+
+    # # Create density and speed of sound subplot
+    # ax2 = plt.subplot(222)
+    # ax2.plot(
+    #     [env1.speedOfSound(i) for i in grid],
+    #     grid,
+    #     "#ff7f0e",
+    #     label="Speed of Sound",
+    # )
+    # ax2.set_xlabel("Speed of Sound (m/s)", color="#ff7f0e")
+    # ax2.tick_params("x", colors="#ff7f0e")
+    # ax2up = ax2.twiny()
+    # ax2up.plot([env1.density(i) for i in grid], grid, color="#1f77b4", label="Density")
+    # ax2up.set_xlabel("Density (kg/m³)", color="#1f77b4")
+    # ax2up.tick_params("x", colors="#1f77b4")
+    # ax2.set_ylabel("Height Above Sea Level (m)")
+    # ax2.grid(True)
+
+    # Create wind u and wind v subplot
+    ax3 = plt.subplot(223)
+    ax3.plot([env1.windVelocityX(i) for i in grid], grid, label="Wind U " + names[0])
+    ax3.plot([env1.windVelocityY(i) for i in grid], grid, label="Wind V " + names[0])
+    ax3.plot([env2.windVelocityX(i) for i in grid], grid, label="Wind U " + names[1])
+    ax3.plot([env2.windVelocityY(i) for i in grid], grid, label="Wind V " + names[1])
+    ax3.legend(loc="best").set_draggable(True)
+    ax3.set_ylabel("Height Above Sea Level (m)")
+    ax3.set_xlabel("Wind Speed (m/s)")
+    ax3.grid(True)
+
+    # Create pressure and temperature subplot
+    ax4 = plt.subplot(224)
+    ax4.plot([env1.pressure(i) / 100 for i in grid], grid, "#ff7f0e", label="Pressure")
+    ax4.set_xlabel("Pressure (hPa)", color="#ff7f0e")
+    ax4.tick_params("x", colors="#ff7f0e")
+    ax4up = ax4.twiny()
+    ax4up.plot(
+        [env1.temperature(i) for i in grid],
+        grid,
+        color="#1f77b4",
+        label="Temperature",
+    )
+    ax4up.set_xlabel("Temperature (K)", color="#1f77b4")
+    ax4up.tick_params("x", colors="#1f77b4")
+    ax4.set_ylabel("Height Above Sea Level (m)")
+    ax4.grid(True)
+
+    plt.subplots_adjust(wspace=0.5, hspace=0.3)
+    plt.show()
+
+    return None  # diff

docs/user/requirements.rst

@@ -16,7 +16,7 @@
         "pytz",
         "timezonefinder",
         "simplekml",
-        "json",
+        "jsonpickle",
     ],
     maintainer="RocketPy Developers",
     author="Giovani Hidalgo Ceotto",