PySDM-examples 2.107__tar.gz → 2.108__tar.gz

{pysdm_examples-2.107 → pysdm_examples-2.108}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pysdm-examples
-Version: 2.107
+Version: 2.108
 Summary: PySDM usage examples reproducing results from literature and depicting how to use PySDM from Python Jupyter notebooks
 Home-page: https://github.com/open-atmos/PySDM
 Author: https://github.com/open-atmos/PySDM/graphs/contributors

{pysdm_examples-2.107 → pysdm_examples-2.108}/PySDM_examples/Rozanski_and_Sonntag_1982/__init__.py

@@ -7,3 +7,4 @@ figs_4_5_6.ipynb:
 """
 
 # pylint: disable=invalid-name
+from PySDM_examples.Rozanski_and_Sonntag_1982.multibox import MultiBox

pysdm_examples-2.108/PySDM_examples/Rozanski_and_Sonntag_1982/multibox.py

@@ -0,0 +1,96 @@
+"""
+definition of a multi-box environment (parcel iteratively adjusting to a stationary state)
+plus helper equation for isotopic ratio evolution
+"""
+
+# pylint: disable=invalid-name
+
+import numpy as np
+
+from PySDM.environments import Parcel
+
+
+def Rv_prim(*, Rl, Nl, Rv, Nv, dNl, Rr, K, a):
+    """eq. (2) in [Rozanski and Sonntag 1982](https://doi.org/10.3402/tellusa.v34i2.10795)"""
+    return (Rl * Nl + Rv * Nv + dNl * Rr * K) / ((Nl + dNl * K) * a + Nv)
+
+
+class MultiBox(Parcel):
+    """iterative parcel model in which each new iteration operates with ambient isotopic profile
+    resultant from the previous iteration, leading to a stationary state
+    """
+
+    def __init__(
+        self,
+        *,
+        isotopes,
+        delta_nl,
+        rain_isotope_ratios,
+        nt,
+        autoconversion_mixrat_threshold,
+        isotope_exchange_factor,
+        **kwargs,
+    ):
+        super().__init__(
+            variables=[
+                f"{ratio}_{isotope}" for ratio in ("Rv", "Rr") for isotope in isotopes
+            ],
+            **kwargs,
+        )
+        self.isotopes = isotopes
+        self.delta_nl = delta_nl
+        self.rain_isotope_ratios = rain_isotope_ratios
+        self.nt = nt
+        self.autoconversion_mixrat_threshold = autoconversion_mixrat_threshold
+        self.isotope_exchange_factor = isotope_exchange_factor
+
+    def advance_parcel_vars(self):
+        """explicit Euler integration of isotope-ratio time derivative"""
+        assert self.delta_liquid_water_mixing_ratio >= 0
+        self._recalculate_temperature_pressure_relative_humidity(self._tmp)
+
+        alpha_old = {}
+        dRv__dt = {}
+        for isotope in self.isotopes:
+            alpha_fun = getattr(
+                self.particulator.formulae.isotope_equilibrium_fractionation_factors,
+                f"alpha_l_{isotope}",
+            )
+            alpha_old[isotope] = alpha_fun(self["T"][0])
+            alpha_new = alpha_fun(self._tmp["T"][0])
+
+            dRv__dt[isotope] = self[f"Rv_{isotope}"][
+                0
+            ] * self.particulator.formulae.isotope_ratio_evolution.d_Rv_over_Rv(
+                alpha=alpha_old[isotope],
+                d_alpha=(alpha_new - alpha_old[isotope]) / self.dt,
+                n_vapour=self["water_vapour_mixing_ratio"][0],
+                d_n_vapour=-self.delta_liquid_water_mixing_ratio / self.dt,
+                n_liquid=self.autoconversion_mixrat_threshold,  # TODO #1207
+            )
+
+        super().advance_parcel_vars()
+
+        for isotope in self.isotopes:
+            self.particulator.backend.explicit_euler(
+                self._tmp[f"Rv_{isotope}"], self.particulator.dt, dRv__dt[isotope]
+            )
+            level = self.particulator.n_steps
+            if self.delta_nl is not None:
+                self._tmp[f"Rv_{isotope}"][:] = Rv_prim(
+                    Rl=alpha_old[isotope] * self._tmp[f"Rv_{isotope}"][0],
+                    Nl=self.autoconversion_mixrat_threshold,
+                    Rv=self._tmp[f"Rv_{isotope}"][0],
+                    Nv=self["water_vapour_mixing_ratio"][0],
+                    dNl=np.sum(self.delta_nl[level:]),
+                    Rr=self.rain_isotope_ratios[isotope][
+                        min(
+                            level + 2, self.nt  # TODO #1207: this warrants a comment...
+                        )
+                    ],
+                    K=self.isotope_exchange_factor,
+                    a=alpha_old[isotope],
+                )
+            self._tmp[f"Rr_{isotope}"][:] = (
+                alpha_old[isotope] * self._tmp[f"Rv_{isotope}"][0]
+            )

{pysdm_examples-2.107 → pysdm_examples-2.108}/pyproject.toml

@@ -4,4 +4,4 @@ local_scheme = "no-local-version"
 version_scheme = "post-release"
 
 [build-system]
-requires = ['setuptools==75.8.2', 'setuptools-scm==8.2.0']
+requires = ['setuptools==76.0.0', 'setuptools-scm==8.2.0']

{pysdm_examples-2.107 → pysdm_examples-2.108}/pysdm_examples.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pysdm-examples
-Version: 2.107
+Version: 2.108
 Summary: PySDM usage examples reproducing results from literature and depicting how to use PySDM from Python Jupyter notebooks
 Home-page: https://github.com/open-atmos/PySDM
 Author: https://github.com/open-atmos/PySDM/graphs/contributors

{pysdm_examples-2.107 → pysdm_examples-2.108}/pysdm_examples.egg-info/SOURCES.txt

@@ -94,6 +94,7 @@ PySDM_examples/Pyrcel/simulation.py
 PySDM_examples/Pyrcel/tutorial_settings.py
 PySDM_examples/Pyrcel/tutorial_simulation.py
 PySDM_examples/Rozanski_and_Sonntag_1982/__init__.py
+PySDM_examples/Rozanski_and_Sonntag_1982/multibox.py
 PySDM_examples/Shima_et_al_2009/__init__.py
 PySDM_examples/Shima_et_al_2009/error_measure.py
 PySDM_examples/Shima_et_al_2009/example.py