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

.. only:: html

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

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

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

.. _sphx_glr_examples_mdo_plot_sobieski_use_case.py:


Application: Sobieski's Super-Sonic Business Jet (MDO)
======================================================
.. _sobieski_mdo:

.. GENERATED FROM PYTHON SOURCE LINES 26-59

This section describes how to setup and solve the MDO problem relative to the
:ref:`Sobieski test case <sobieski_problem>` with |g|.

.. seealso::

   To begin with a more simple MDO problem, and have a detailed description
   of how to plug a test case to |g|, see :ref:`sellar_mdo`.


.. _sobieski_use_case:

Solving with an :ref:`MDF formulation <mdf_formulation>`
--------------------------------------------------------

In this example, we solve the range optimization using the following
:ref:`MDF formulation <mdf_formulation>`:

- The :ref:`MDF formulation <mdf_formulation>` couples all the disciplines
  during the :ref:`mda` at each optimization iteration.
- All the :term:`design variables` are equally treated, concatenated in a
  single vector and given to a single :term:`optimization algorithm` as the
  unknowns of the problem.
- There is no specific :term:`constraint` due to the :ref:`MDF formulation
  <mdf_formulation>`.
- Only the design :term:`constraints` :math:`g\_1`, :math:`g\_2` and
  :math:`g\_3` are added to the problem.
- The :term:`objective function` is the range (the :math:`y\_4` variable in
  the model), computed after the :ref:`mda`.

Imports
-------
All the imports needed for the tutorials are performed here.
Note that some of the imports are related to the Python 2/3 compatibility.

.. GENERATED FROM PYTHON SOURCE LINES 59-73

.. code-block:: default

    from __future__ import annotations

    from gemseo.api import configure_logger
    from gemseo.api import create_discipline
    from gemseo.api import create_scenario
    from gemseo.api import get_available_formulations
    from gemseo.core.jacobian_assembly import JacobianAssembly
    from gemseo.disciplines.utils import get_all_inputs
    from gemseo.disciplines.utils import get_all_outputs
    from gemseo.problems.sobieski.core.problem import SobieskiProblem
    from matplotlib import pyplot as plt

    configure_logger()





.. rst-class:: sphx-glr-script-out

 .. code-block:: none


    <RootLogger root (INFO)>



.. GENERATED FROM PYTHON SOURCE LINES 74-80

Step 1: Creation of :class:`.MDODiscipline`
-------------------------------------------

To build the scenario, we first instantiate the disciplines. Here, the
disciplines themselves have already been
developed and interfaced with |g| (see :ref:`benchmark_problems`).

.. GENERATED FROM PYTHON SOURCE LINES 80-90

.. code-block:: default


    disciplines = create_discipline(
        [
            "SobieskiPropulsion",
            "SobieskiAerodynamics",
            "SobieskiMission",
            "SobieskiStructure",
        ]
    )








.. GENERATED FROM PYTHON SOURCE LINES 91-98

.. tip::

   For the disciplines that are not interfaced with |g|, the |g|'s
   :mod:`~gemseo.api` eases the creation of disciplines without having
   to import them.

   See :ref:`api`.

.. GENERATED FROM PYTHON SOURCE LINES 100-116

Step 2: Creation of :class:`.Scenario`
--------------------------------------

The scenario delegates the creation of the optimization problem to the
:ref:`MDO formulation <mdo_formulations>`.

Therefore, it needs the list of :code:`disciplines`, the names of the formulation,
the name of the objective function and the design space.

- The :code:`design_space` (shown below for reference, as :code:`design_space.txt`)
  defines the unknowns of the optimization problem, and their bounds. It contains
  all the design variables needed by the :ref:`MDF formulation <mdf_formulation>`.
  It can be imported from a text file, or created from scratch with the methods
  :meth:`~gemseo.api.create_design_space` and
  :meth:`~gemseo.algos.design_space.DesignSpace.add_variable`. In this case,
  we will create it directly from the API.

.. GENERATED FROM PYTHON SOURCE LINES 116-117

.. code-block:: default

    design_space = SobieskiProblem().design_space







.. GENERATED FROM PYTHON SOURCE LINES 118-151

.. code::

          vi design_space.txt

          name      lower_bound      value      upper_bound  type
          x_shared      0.01          0.05          0.09     float
          x_shared    30000.0       45000.0       60000.0    float
          x_shared      1.4           1.6           1.8      float
          x_shared      2.5           5.5           8.5      float
          x_shared      40.0          55.0          70.0     float
          x_shared     500.0         1000.0        1500.0    float
          x_1           0.1           0.25          0.4      float
          x_1           0.75          1.0           1.25     float
          x_2           0.75          1.0           1.25     float
          x_3           0.1           0.5           1.0      float
          y_14        24850.0    50606.9741711    77100.0    float
          y_14        -7700.0    7306.20262124    45000.0    float
          y_32         0.235       0.50279625      0.795     float
          y_31         2960.0    6354.32430691    10185.0    float
          y_24          0.44       4.15006276      11.13     float
          y_34          0.44       1.10754577       1.98     float
          y_23         3365.0    12194.2671934    26400.0    float
          y_21        24850.0    50606.9741711    77250.0    float
          y_12        24850.0      50606.9742     77250.0    float
          y_12          0.45          0.95          1.5      float

- The available :ref:`MDO formulations <mdo_formulations>` are located in the
  **gemseo.formulations** package, see :ref:`extending-gemseo` for extending
  GEMSEO with other formulations.
- The :code:`formulation` classname (here, :code:`"MDF"`) shall be passed to
  the scenario to select them.
- The list of available formulations can be obtained by using
  :meth:`~gemseo.api.get_available_formulations`.

.. GENERATED FROM PYTHON SOURCE LINES 151-152

.. code-block:: default

    get_available_formulations()




.. rst-class:: sphx-glr-script-out

 .. code-block:: none


    ['BiLevel', 'DisciplinaryOpt', 'IDF', 'MDF']



.. GENERATED FROM PYTHON SOURCE LINES 153-157

- :math:`y\_4` corresponds to the :code:`objective_name`. This name must be one
  of the disciplines outputs, here the "SobieskiMission" discipline. The list of
  all outputs of the disciplines can be obtained by using
  :meth:`~gemseo.disciplines.utils.get_all_outputs`:

.. GENERATED FROM PYTHON SOURCE LINES 157-159

.. code-block:: default

    get_all_outputs(disciplines)
    get_all_inputs(disciplines)




.. rst-class:: sphx-glr-script-out

 .. code-block:: none


    ['y_34', 'c_2', 'c_3', 'y_32', 'y_21', 'c_4', 'y_12', 'x_2', 'y_24', 'y_31', 'c_1', 'y_23', 'x_shared', 'x_3', 'c_0', 'x_1', 'y_14']



.. GENERATED FROM PYTHON SOURCE LINES 160-163

From these :class:`~gemseo.core.discipline.MDODiscipline`, design space filename,
:ref:`MDO formulation <mdo_formulations>` name and objective function name,
we build the scenario:

.. GENERATED FROM PYTHON SOURCE LINES 163-170

.. code-block:: default

    scenario = create_scenario(
        disciplines,
        formulation="MDF",
        maximize_objective=True,
        objective_name="y_4",
        design_space=design_space,
    )







.. GENERATED FROM PYTHON SOURCE LINES 171-185

The range function (:math:`y\_4`) should be maximized. However, optimizers
minimize functions by default. Which is why, when creating the scenario, the argument
:code:`maximize_objective` shall be set to :code:`True`.

Scenario options
~~~~~~~~~~~~~~~~

We may provide additional options to the scenario:


**Function derivatives.** As analytical disciplinary derivatives are
vailable for Sobieski test-case, they can be used instead of computing
the derivatives with finite-differences or with the complex-step method.
The easiest way to set a method is to let the optimizer determine it:

.. GENERATED FROM PYTHON SOURCE LINES 185-186

.. code-block:: default

    scenario.set_differentiation_method("user")







.. GENERATED FROM PYTHON SOURCE LINES 187-212

The default behavior of the optimizer triggers :term:`finite differences`.
It corresponds to:

.. code::

  scenario.set_differentiation_method("finite_differences",1e-7)

It it also possible to differentiate functions by means of the
:term:`complex step` method:

.. code::

  scenario.set_differentiation_method("complex_step",1e-30j)

Constraints
~~~~~~~~~~~

Similarly to the objective function, the constraints names are a subset
of the disciplines' outputs. They can be obtained by using
:meth:`~gemseo.disciplines.utils.get_all_outputs`.

The formulation has a powerful feature to automatically dispatch the constraints
(:math:`g\_1, g\_2, g\_3`) and plug them to the optimizers depending on
the formulation. To do that, we use the method
:meth:`~gemseo.core.scenario.Scenario.add_constraint`:

.. GENERATED FROM PYTHON SOURCE LINES 213-215

.. code-block:: default

    for constraint in ["g_1", "g_2", "g_3"]:
        scenario.add_constraint(constraint, "ineq")







.. GENERATED FROM PYTHON SOURCE LINES 216-221

Step 3: Execution and visualization of the results
--------------------------------------------------

The algorithm arguments are provided as a dictionary to the execution
method of the scenario:

.. GENERATED FROM PYTHON SOURCE LINES 221-222

.. code-block:: default

    algo_args = {"max_iter": 10, "algo": "SLSQP"}







.. GENERATED FROM PYTHON SOURCE LINES 223-233

.. warning::

   The mandatory arguments are the maximum number of iterations and the algorithm name.
   Any other options of the optimization algorithm can be prescribed through
   the argument :code:`algo_options` with a dictionary, e.g.
   :code:`algo_args =
   {"max_iter": 10, "algo": "SLSQP": "algo_options": {"ftol_rel": 1e-6}}`.
   This list of available algorithm options are detailed here: :ref:`gen_opt_algos`.

The scenario is executed by means of the line:

.. GENERATED FROM PYTHON SOURCE LINES 233-234

.. code-block:: default

    scenario.execute(algo_args)




.. rst-class:: sphx-glr-script-out

 .. code-block:: none

        INFO - 14:47:17:  
        INFO - 14:47:17: *** Start MDOScenario execution ***
        INFO - 14:47:17: MDOScenario
        INFO - 14:47:17:    Disciplines: SobieskiAerodynamics SobieskiMission SobieskiPropulsion SobieskiStructure
        INFO - 14:47:17:    MDO formulation: MDF
        INFO - 14:47:17: Optimization problem:
        INFO - 14:47:17:    minimize -y_4(x_shared, x_1, x_2, x_3)
        INFO - 14:47:17:    with respect to x_1, x_2, x_3, x_shared
        INFO - 14:47:17:    subject to constraints:
        INFO - 14:47:17:       g_1(x_shared, x_1, x_2, x_3) <= 0.0
        INFO - 14:47:17:       g_2(x_shared, x_1, x_2, x_3) <= 0.0
        INFO - 14:47:17:       g_3(x_shared, x_1, x_2, x_3) <= 0.0
        INFO - 14:47:17:    over the design space:
        INFO - 14:47:17:    +-------------+-------------+-------+-------------+-------+
        INFO - 14:47:17:    | name        | lower_bound | value | upper_bound | type  |
        INFO - 14:47:17:    +-------------+-------------+-------+-------------+-------+
        INFO - 14:47:17:    | x_shared[0] |     0.01    |  0.05 |     0.09    | float |
        INFO - 14:47:17:    | x_shared[1] |    30000    | 45000 |    60000    | float |
        INFO - 14:47:17:    | x_shared[2] |     1.4     |  1.6  |     1.8     | float |
        INFO - 14:47:17:    | x_shared[3] |     2.5     |  5.5  |     8.5     | float |
        INFO - 14:47:17:    | x_shared[4] |      40     |   55  |      70     | float |
        INFO - 14:47:17:    | x_shared[5] |     500     |  1000 |     1500    | float |
        INFO - 14:47:17:    | x_1[0]      |     0.1     |  0.25 |     0.4     | float |
        INFO - 14:47:17:    | x_1[1]      |     0.75    |   1   |     1.25    | float |
        INFO - 14:47:17:    | x_2         |     0.75    |   1   |     1.25    | float |
        INFO - 14:47:17:    | x_3         |     0.1     |  0.5  |      1      | float |
        INFO - 14:47:17:    +-------------+-------------+-------+-------------+-------+
        INFO - 14:47:17: Solving optimization problem with algorithm SLSQP:
        INFO - 14:47:17: ...   0%|          | 0/10 [00:00<?, ?it]
        INFO - 14:47:18: ...  20%|██        | 2/10 [00:00<00:00, 40.75 it/sec, obj=-2.12e+3]
     WARNING - 14:47:18: MDAJacobi has reached its maximum number of iterations but the normed residual 1.4486313079508508e-06 is still above the tolerance 1e-06.
        INFO - 14:47:18: ...  30%|███       | 3/10 [00:00<00:00, 23.47 it/sec, obj=-3.75e+3]
        INFO - 14:47:18: ...  40%|████      | 4/10 [00:00<00:00, 17.13 it/sec, obj=-4.01e+3]
     WARNING - 14:47:18: MDAJacobi has reached its maximum number of iterations but the normed residual 2.928004141058104e-06 is still above the tolerance 1e-06.
        INFO - 14:47:18: ...  50%|█████     | 5/10 [00:00<00:00, 13.02 it/sec, obj=-4.49e+3]
        INFO - 14:47:18: ...  60%|██████    | 6/10 [00:00<00:00, 11.02 it/sec, obj=-3.4e+3]
        INFO - 14:47:18: ...  80%|████████  | 8/10 [00:01<00:00,  9.37 it/sec, obj=-4.76e+3]
        INFO - 14:47:19: ... 100%|██████████| 10/10 [00:01<00:00,  8.17 it/sec, obj=-4.56e+3]
        INFO - 14:47:19: ... 100%|██████████| 10/10 [00:01<00:00,  8.15 it/sec, obj=-4.56e+3]
        INFO - 14:47:19: Optimization result:
        INFO - 14:47:19:    Optimizer info:
        INFO - 14:47:19:       Status: None
        INFO - 14:47:19:       Message: Maximum number of iterations reached. GEMSEO Stopped the driver
        INFO - 14:47:19:       Number of calls to the objective function by the optimizer: 12
        INFO - 14:47:19:    Solution:
        INFO - 14:47:19:       The solution is feasible.
        INFO - 14:47:19:       Objective: -3749.8868975554387
        INFO - 14:47:19:       Standardized constraints:
        INFO - 14:47:19:          g_1 = [-0.01671296 -0.03238836 -0.04350867 -0.05123129 -0.05681738 -0.13780658
        INFO - 14:47:19:  -0.10219342]
        INFO - 14:47:19:          g_2 = -0.0004062839430756249
        INFO - 14:47:19:          g_3 = [-0.66482546 -0.33517454 -0.11023156 -0.183255  ]
        INFO - 14:47:19:       Design space: 
        INFO - 14:47:19:       +-------------+-------------+---------------------+-------------+-------+
        INFO - 14:47:19:       | name        | lower_bound |        value        | upper_bound | type  |
        INFO - 14:47:19:       +-------------+-------------+---------------------+-------------+-------+
        INFO - 14:47:19:       | x_shared[0] |     0.01    | 0.05989842901423112 |     0.09    | float |
        INFO - 14:47:19:       | x_shared[1] |    30000    |  59853.73840058666  |    60000    | float |
        INFO - 14:47:19:       | x_shared[2] |     1.4     |         1.4         |     1.8     | float |
        INFO - 14:47:19:       | x_shared[3] |     2.5     |  2.527371250092273  |     8.5     | float |
        INFO - 14:47:19:       | x_shared[4] |      40     |  69.86825198198687  |      70     | float |
        INFO - 14:47:19:       | x_shared[5] |     500     |  1495.734648986894  |     1500    | float |
        INFO - 14:47:19:       | x_1[0]      |     0.1     |         0.4         |     0.4     | float |
        INFO - 14:47:19:       | x_1[1]      |     0.75    |  0.7521124139939552 |     1.25    | float |
        INFO - 14:47:19:       | x_2         |     0.75    |  0.7520888531444992 |     1.25    | float |
        INFO - 14:47:19:       | x_3         |     0.1     |  0.1398000762238233 |      1      | float |
        INFO - 14:47:19:       +-------------+-------------+---------------------+-------------+-------+
        INFO - 14:47:19: *** End MDOScenario execution (time: 0:00:01.243061) ***

    {'max_iter': 10, 'algo': 'SLSQP'}



.. GENERATED FROM PYTHON SOURCE LINES 235-242

Post-processing options
~~~~~~~~~~~~~~~~~~~~~~~
A whole variety of visualizations may be displayed for both MDO and DOE
scenarios. These features are illustrated on the SSBJ use case in
:ref:`post_processing`.

To visualize the optimization history:

.. GENERATED FROM PYTHON SOURCE LINES 242-245

.. code-block:: default

    scenario.post_process("OptHistoryView", save=False, show=False)
    # Workaround for HTML rendering, instead of ``show=True``
    plt.show()



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


    *

      .. image-sg:: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_001.png
         :alt: Evolution of the optimization variables
         :srcset: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_001.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_002.png
         :alt: Evolution of the objective value
         :srcset: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_002.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_003.png
         :alt: Distance to the optimum
         :srcset: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_003.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_004.png
         :alt: Hessian diagonal approximation
         :srcset: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_004.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_005.png
         :alt: Evolution of the inequality constraints
         :srcset: /examples/mdo/images/sphx_glr_plot_sobieski_use_case_005.png
         :class: sphx-glr-multi-img





.. GENERATED FROM PYTHON SOURCE LINES 246-254

Influence of gradient computation method on performance
-------------------------------------------------------

As mentioned in :ref:`jacobian_assembly`, several methods
are available in order to perform  the gradient computations: classical finite
differences, complex step and :ref:`mda` linearization in direct or adjoint mode.
These modes are automatically selected by |g| to minimize the CPU time. Yet, they
can be forced on demand in each :ref:`mda`:

.. GENERATED FROM PYTHON SOURCE LINES 254-256

.. code-block:: default

    scenario.formulation.mda.linearization_mode = JacobianAssembly.DIRECT_MODE
    scenario.formulation.mda.matrix_type = JacobianAssembly.LINEAR_OPERATOR







.. GENERATED FROM PYTHON SOURCE LINES 257-262

The method used to solve the adjoint or direct linear problem may also be selected.
|g| can either assemble a sparse residual Jacobian matrix of the :ref:`mda` from the
disciplines matrices. This has the advantage that LU factorizations may be stored to
solve multiple right hand sides problems in a cheap way. But this requires
extra memory.

.. GENERATED FROM PYTHON SOURCE LINES 262-264

.. code-block:: default

    scenario.formulation.mda.matrix_type = JacobianAssembly.SPARSE
    scenario.formulation.mda.use_lu_fact = True







.. GENERATED FROM PYTHON SOURCE LINES 265-269

Altenatively, |g| can implicitly create a matrix-vector product operator,
which is sufficient for GMRES-like solvers. It avoids to create an additional
data structure. This can also be mandatory if the disciplines do not provide
full Jacobian matrices but only matrix-vector product operators.

.. GENERATED FROM PYTHON SOURCE LINES 269-270

.. code-block:: default

    scenario.formulation.mda.matrix_type = JacobianAssembly.LINEAR_OPERATOR







.. GENERATED FROM PYTHON SOURCE LINES 271-292

The next table shows the performance of each method for solving the Sobieski use case
with :ref:`MDF <mdf_formulation>` and :ref:`IDF <idf_formulation>` formulations.
Efficiency of linearization is clearly visible has it takes from 10 to 20 times
less CPU time to compute analytic derivatives of an :ref:`mda` compared
to finite difference and complex step.
For :ref:`IDF <idf_formulation>`, improvements are less consequent,
but direct linearization is more than 2.5 times faster than other methods.

.. tabularcolumns:: |l|c|c|

+-----------------------+------------------------------+------------------------------+
|                       |  Execution time (s)                                         |
+  Derivation Method    +------------------------------+------------------------------+
|                       | :ref:`MDF <mdf_formulation>` | :ref:`IDF <idf_formulation>` |
+=======================+==============================+==============================+
| Finite differences    | 8.22                         | 1.93                         |
+-----------------------+------------------------------+------------------------------+
| Complex step          | 18.11                        | 2.07                         |
+-----------------------+------------------------------+------------------------------+
| Linearized (direct)   | 0.90                         | 0.68                         |
+-----------------------+------------------------------+------------------------------+


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

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


.. _sphx_glr_download_examples_mdo_plot_sobieski_use_case.py:

.. only:: html

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


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

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

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

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


.. only:: html

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

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