.. DO NOT EDIT. .. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY. .. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE: .. "auto_examples/howto_plot/plot_animation.py" .. LINE NUMBERS ARE GIVEN BELOW. .. only:: html .. note:: :class: sphx-glr-download-link-note :ref:`Go to the end ` to download the full example code. .. rst-class:: sphx-glr-example-title .. _sphx_glr_auto_examples_howto_plot_plot_animation.py: How to create Animations ======================== .. sectionauthor:: Florian Zill (Helmholtz Centre for Environmental Research GmbH - UFZ) To demonstrate the creation of an animated plot we use a component transport example from the ogs benchmark gallery (https://www.opengeosys.org/docs/benchmarks/hydro-component/elder/). .. GENERATED FROM PYTHON SOURCE LINES 13-21 .. code-block:: Python import numpy as np import ogstools as ot from ogstools import examples ms = examples.load_meshseries_CT_2D_XDMF().scale(time=("s", "yrs")) saturation = ot.variables.saturation .. GENERATED FROM PYTHON SOURCE LINES 22-28 To read your own data as a mesh series you can do: .. code-block:: python from ogstools.meshlib import MeshSeries mesh_series = MeshSeries("filepath/filename_pvd_or_xdmf") .. GENERATED FROM PYTHON SOURCE LINES 30-33 You can choose which timesteps to render by either slicing the MeshSeries or by resampling it to new timesteps (this interpolates between existing timesteps). .. GENERATED FROM PYTHON SOURCE LINES 35-38 this would only use every second timesteps for animation ms = ms[::2] this uses equally spaced timesteps for a smooth animation .. GENERATED FROM PYTHON SOURCE LINES 38-41 .. code-block:: Python timevalues = np.linspace(ms.timevalues[0], ms.timevalues[-1], num=21) ms = ot.MeshSeries.resample(ms, timevalues) .. GENERATED FROM PYTHON SOURCE LINES 42-47 Now, let's animate the saturation solution. Note that rendering many frames in conjunction with large meshes might take a really long time. You need to setup a matplotlib figure first and a function, which is executed on each frame. This function has to take the individual values of the sequences passed as additional arguments, in this case the timevalues and the MeshSeries. .. GENERATED FROM PYTHON SOURCE LINES 50-51 clip to the right half .. GENERATED FROM PYTHON SOURCE LINES 51-66 .. code-block:: Python ms_r = ms.transform(lambda mesh: mesh.clip("-x", [-1, 0, 0])) # create initial figure with fixed colorbar fig = ot.plot.contourf(ms_r[0], saturation, vmin=0, vmax=100, dpi=50) fig.axes[0].set_title(f"{0} yrs", fontsize=32) def plot_contourf(timevalue: float, mesh: ot.Mesh) -> None: fig.axes[0].clear() ot.plot.contourf(mesh, saturation, ax=fig.axes[0], dpi=50) fig.axes[0].set_title(f"{timevalue:.1f} yrs", fontsize=32) anim = ot.plot.animate(fig, plot_contourf, ms_r.timevalues, ms_r) .. container:: sphx-glr-animation .. raw:: html

.. rst-class:: sphx-glr-script-out .. code-block:: none 0%| | 0/20 [00:00 None: fig.axes[0].clear() ot.plot.line(mesh, saturation, ax=fig.axes[0]) fig.axes[0].set_ylim([0, 100]) fig.tight_layout() anim = ot.plot.animate(fig, plot_line, ms_x) .. container:: sphx-glr-animation .. raw:: html

.. rst-class:: sphx-glr-script-out .. code-block:: none 0%| | 0/20 [00:00` .. container:: sphx-glr-download sphx-glr-download-python :download:`Download Python source code: plot_animation.py ` .. container:: sphx-glr-download sphx-glr-download-zip :download:`Download zipped: plot_animation.zip `