.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/oil_and_gas/plot_live_fluid_properties.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        Click :ref:`here <sphx_glr_download_auto_examples_oil_and_gas_plot_live_fluid_properties.py>`
        to download the full example code

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_oil_and_gas_plot_live_fluid_properties.py:


=========================================================
Calculate fluid properties given pressure and temperature
=========================================================

The calculation of fluid properties is a fundamental component of mass balance and other methods of conservation.
The goal of this feature is to calculate the fluid properties given the pressure and temperature conditions and
the composition of the fluid itself. Equation of state simulators output a fluid file that is a table of the fluid
properties for a range of pressure and temperature conditions. The input to these simulators is the composition of
the fluid obtained from lab tests. For this specific feature, the input fluid file is a .tab file used by
`OLGA <https://www.software.slb.com/products/olga>`_, a transient multiphase flow simulator.

.. GENERATED FROM PYTHON SOURCE LINES 16-71



.. image-sg:: /auto_examples/oil_and_gas/images/sphx_glr_plot_live_fluid_properties_001.png
   :alt: Pressure bottom hole [bara], Temperature bottom hole [C], Gas density [kg/m3], Oil density [kg/m3], Gas thermal capacity [J/kg·°C], Oil thermal capacity [J/kg·°C]
   :srcset: /auto_examples/oil_and_gas/images/sphx_glr_plot_live_fluid_properties_001.png
   :class: sphx-glr-single-img





.. code-block:: default

    import os
    import pickle as pkl

    import matplotlib.dates as mdates
    import matplotlib.pyplot as plt

    from indsl.oil_and_gas.live_fluid_properties import retrieve_fluid_properties


    base_path = "" if __name__ == "__main__" else os.path.dirname(__file__)
    df = pkl.load(open(os.path.join(base_path, "../../datasets/data/press_temp_bh&wh.pkl"), "rb"))
    pvt_data = pkl.load(open(os.path.join(base_path, "../../datasets/data/pvt_data.pkl"), "rb"))

    pbh = df.iloc[:, 0] * 100000
    tbh = df.iloc[:, 1]
    pwf = df.iloc[:, 2] * 100000
    twf = df.iloc[:, 3]

    pbh.name = "Pressure bottom hole [bara]"
    pwf.name = "Pressure wellhead [bara]"
    tbh.name = "Temperature bottom hole [C]"
    twf.name = "Temperature wellhead [C]"

    gas_den_bh = [retrieve_fluid_properties(pbh, tbh, pvt_data, param="ROG"), "red"]
    gas_den_wh = [retrieve_fluid_properties(pwf, twf, pvt_data, param="ROG"), "green"]
    liq_den_bh = [retrieve_fluid_properties(pbh, tbh, pvt_data, param="ROHL"), "red"]
    liq_den_wh = [retrieve_fluid_properties(pwf, twf, pvt_data, param="ROHL"), "green"]
    hl_bh = [retrieve_fluid_properties(pbh, tbh, pvt_data, param="CPHL"), "red"]
    hl_wh = [retrieve_fluid_properties(pwf, twf, pvt_data, param="CPHL"), "green"]
    hg_bh = [retrieve_fluid_properties(pbh, tbh, pvt_data, param="CPG"), "red"]
    hg_wh = [retrieve_fluid_properties(pwf, twf, pvt_data, param="CPG"), "green"]

    pbh = df.iloc[:, 0]
    pwf = df.iloc[:, 2]

    pbh.name = "Pressure bottom hole [bara]"
    pwf.name = "Pressure wellhead [bara]"
    pbh = [pbh, "red"]
    tbh = [tbh, "red"]
    pwf = [pwf, "green"]
    twf = [twf, "green"]

    data = [[pbh, pwf], [tbh, twf], [gas_den_bh, gas_den_wh], [liq_den_bh, liq_den_wh], [hg_bh, hg_wh], [hl_bh, hl_wh]]

    fig, ax = plt.subplots(figsize=(8, 2 * len(data)), nrows=len(data), ncols=1)
    ax = ax.ravel()
    for idx, i in enumerate(data):
        ax[idx].plot(i[0][0], color=i[0][1], label="bottom hole")
        ax[idx].plot(i[1][0], color=i[1][1], label="wellhead")
        ax[idx].set_title(i[0][0].name)
        ax[idx].legend()
        ax[idx].xaxis.set_major_locator(mdates.DayLocator(interval=3))

    plt.tight_layout()
    plt.show()


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** ( 0 minutes  0.809 seconds)


.. _sphx_glr_download_auto_examples_oil_and_gas_plot_live_fluid_properties.py:


.. only :: html

 .. container:: sphx-glr-footer
    :class: sphx-glr-footer-example



  .. container:: sphx-glr-download sphx-glr-download-python

     :download:`Download Python source code: plot_live_fluid_properties.py <plot_live_fluid_properties.py>`



  .. container:: sphx-glr-download sphx-glr-download-jupyter

     :download:`Download Jupyter notebook: plot_live_fluid_properties.ipynb <plot_live_fluid_properties.ipynb>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_