.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "examples\04-advanced\01-solve_harmonic_problem.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

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

        :ref:`Go to the end <sphx_glr_download_examples_04-advanced_01-solve_harmonic_problem.py>`
        to download the full example code

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

.. _sphx_glr_examples_04-advanced_01-solve_harmonic_problem.py:


.. _ref_solve_modal_problem_advanced:

Solve harmonic problem (with damping) using matrix inverse
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

This example shows how to create a harmonic (over frequencies) fields
container for an analysis with damping. This fields container is then used to
solve the problem Ma+Dv+Ku=F by inverting the matrix

.. GENERATED FROM PYTHON SOURCE LINES 12-19

.. code-block:: Python


    import math

    from ansys.dpf import core as dpf
    from ansys.dpf.core import operators as ops









.. GENERATED FROM PYTHON SOURCE LINES 20-21

Create 2D (x,y) matrix fields for inertia, damping, and stiffness.

.. GENERATED FROM PYTHON SOURCE LINES 21-32

.. code-block:: Python


    freq = [25, 50, 100, 200, 400]
    dim = 2  # dimension of matrix

    fM0 = dpf.fields_factory.create_matrix_field(1, dim, dim)
    fM0.append([0.0, 1.0, 2.0, 3.0], 1)
    fK0 = dpf.fields_factory.create_matrix_field(1, dim, dim)
    fK0.append([4.0, 8.0, 0.0, 1.0], 1)
    fC0 = dpf.fields_factory.create_matrix_field(1, dim, dim)
    fC0.append([7.0, 5.0, 9.0, 1.0], 1)








.. GENERATED FROM PYTHON SOURCE LINES 33-35

Create a fields container for real and imaginary parts
for each frequency.

.. GENERATED FROM PYTHON SOURCE LINES 35-50

.. code-block:: Python


    reals = {}
    ims = {}
    for k, f in enumerate(freq):
        omega = 2.0 * math.pi * f
        omega2 = omega**2
        real = fK0 + fM0 * omega2
        imag = fC0 * omega
        reals[f] = real.outputs.field()
        ims[f] = imag.outputs.field()

    cplx_fc = dpf.fields_container_factory.over_time_freq_complex_fields_container(
        reals, ims, time_freq_unit="Hz"
    )








.. GENERATED FROM PYTHON SOURCE LINES 51-53

Use DPF operators to inverse the matrix and then compute the amplitude
and the phase.

.. GENERATED FROM PYTHON SOURCE LINES 53-59

.. code-block:: Python


    inverse = ops.math.matrix_inverse(cplx_fc)
    component = ops.logic.component_selector_fc(inverse, 0)
    amp = ops.math.amplitude_fc(component)
    phase = ops.math.phase_fc(component)








.. GENERATED FROM PYTHON SOURCE LINES 60-61

Get the phase and amplitude and then plot it over frequencies.

.. GENERATED FROM PYTHON SOURCE LINES 61-88

.. code-block:: Python

    amp_over_frequency = amp.outputs.fields_container()
    phase_over_frequency = phase.outputs.fields_container()
    time_freq_support = amp_over_frequency.time_freq_support

    amp_array = []
    phase_array = []
    for f in amp_over_frequency:
        amp_array.append(f.data)

    for f in phase_over_frequency:
        phase_array.append(f.data * 180.0 / math.pi)

    import matplotlib.pyplot as plt

    plt.figure()
    plt.plot(time_freq_support.time_frequencies.data, amp_array, "r", label="amplitude")
    plt.xlabel("Frequency (Hz)")
    plt.ylabel("Displacement ampliude (m)")
    plt.legend()
    plt.show()

    plt.figure()
    plt.plot(time_freq_support.time_frequencies.data, phase_array, "r", label="phase")
    plt.xlabel("Frequency (Hz)")
    plt.ylabel("Displacement phase (deg)")
    plt.legend()
    plt.show()



.. rst-class:: sphx-glr-horizontal


    *

      .. image-sg:: /examples/04-advanced/images/sphx_glr_01-solve_harmonic_problem_001.png
         :alt: 01 solve harmonic problem
         :srcset: /examples/04-advanced/images/sphx_glr_01-solve_harmonic_problem_001.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /examples/04-advanced/images/sphx_glr_01-solve_harmonic_problem_002.png
         :alt: 01 solve harmonic problem
         :srcset: /examples/04-advanced/images/sphx_glr_01-solve_harmonic_problem_002.png
         :class: sphx-glr-multi-img






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

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


.. _sphx_glr_download_examples_04-advanced_01-solve_harmonic_problem.py:

.. only:: html

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

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

      :download:`Download Jupyter notebook: 01-solve_harmonic_problem.ipynb <01-solve_harmonic_problem.ipynb>`

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

      :download:`Download Python source code: 01-solve_harmonic_problem.py <01-solve_harmonic_problem.py>`


.. only:: html

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

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