Source code for tests.test_case_0dparcel.run_0dparcel_model

"""
Copyright (c) 2024 MPI-M, Clara Bayley

----- Microphysics Test Cases -----
File: run_0dparcel_model.py
Project: test_case_0dparcel
Created Date: Thursday 29th February 2024
Author: Clara Bayley (CB)
Additional Contributors:
-----
Last Modified: Sunday 1st September 2024
Modified By: CB
-----
License: BSD 3-Clause "New" or "Revised" License
https://opensource.org/licenses/BSD-3-Clause
-----
File Description:
"""

from .adiabatic_motion import AdiabaticMotion
from libs.src_mock_py.output_thermodynamics import OutputThermodynamics




[docs]
def run_0dparcel_model(time, time_end, timestep, thermo, microphys_scheme):
    """Run a 0-D parcel model with a specified microphysics scheme and parcel dynamics.

    This function runs a 0-D parcel model with the given initial thermodynamic conditions, and
    microphysics scheme from time to time_end with a constant timestep using an instance of
    AdiabaticMotion for the parcel dynamics.

    Parameters:
        time (float):
          Initial time for the simulation (s).
        time_end (float):
          End time for the simulation (s).
        timestep (float):
          Timestep for the simulation (s).
        thermo (Thermodynamics):
          Initial thermodynamic conditions.
        microphys_scheme:
          Microphysics scheme to use.

    Returns:
          OutputThermodynamics: Output containing thermodynamic data from the model run.
    """

    ### data to output during model run
    out = OutputThermodynamics()

    ### type of dynamics parcel will undergo
    amp = 11325  # amplitude of pressure sinusoid [Pa]
    tau = 120  # time period of pressure sinusiod [s]
    parcel_dynamics = AdiabaticMotion(amp, tau)

    ### run dynamics + microphysics from time to time_end
    microphys_scheme.initialize()

    out.output_thermodynamics(time, thermo)
    while time <= time_end:
        thermo = parcel_dynamics.run(time, timestep, thermo)
        thermo = microphys_scheme.run(timestep, thermo)

        out.output_thermodynamics(time, thermo)

        time += timestep

    microphys_scheme.finalize()

    out.finalize()
    return out