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

.. only:: html

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

        :ref:`Go to the end <sphx_glr_download_auto_examples_ellipsoid.py>`
        to download the full example code

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

.. _sphx_glr_auto_examples_ellipsoid.py:


Ellipsoid example
-----------------

Updating the size and shape of an object interactively

.. GENERATED FROM PYTHON SOURCE LINES 41-150



.. image-sg:: /auto_examples/images/sphx_glr_ellipsoid_001.png
   :alt: ellipsoid
   :srcset: /auto_examples/images/sphx_glr_ellipsoid_001.png
   :class: sphx-glr-single-img





.. code-block:: default


    from __future__ import absolute_import
    from soma import aims
    import anatomist.direct.api as anatomist
    import time
    import numpy
    import sys
    from six.moves import range
    from soma.qt_gui import qt_backend
    from soma.qt_gui.qt_backend import QtGui, Qt

    if 'IPython' in sys.modules and Qt.QApplication.instance() is None:
        run_loop = False
    else:
        run_loop = True

    a = anatomist.Anatomist()


    def rotation_3d(a1, a2, a3):
        '''
        Return matrix 4x4 of rotation along 3 canonic axis based on 3 angles.

        - a1, a2, a3 : rotation angles.
        '''
        c1, s1 = numpy.cos(a1), numpy.sin(a1)
        c2, s2 = numpy.cos(a2), numpy.sin(a2)
        c3, s3 = numpy.cos(a3), numpy.sin(a3)
        m1 = numpy.matrix([
            [c1, -s1, 0, 0],
            [s1, c1,  0, 0],
            [0,   0,  1, 0],
            [0,   0,  0, 1]])
        m2 = numpy.matrix([
            [1,  0,   0, 0],
            [0, c2, -s2, 0],
            [0, s2,  c2, 0],
            [0,  0,   0, 1]])
        m3 = numpy.matrix([
            [c3, 0, -s3, 0],
            [0,  1,   0, 0],
            [s3, 0,  c3, 0],
            [0,  0,   0, 1]])
        return m1 * m2 * m3


    def apply_transform(mesh, scaling, rot, translate):
        '''
        Transform mesh with scaling, translation and rotation matrix.
        '''
        mesh2 = aims.AimsTimeSurface_3(mesh)
        # create transformation
        scale = numpy.asmatrix(numpy.diag(scaling))
        rot[:, 3] = numpy.asmatrix(translate).T
        cov = rot * scale * rot.I
        # apply transformation
        motion = aims.Motion(numpy.asarray(cov).flatten())
        aims.SurfaceManip.meshTransform(mesh2, motion)
        # update mesh in anatomist object
        asphere.setSurface(mesh2)


    def transform(mesh, angles, scaling, translate):
        '''
        Create and apply transformation to mesh.
        '''
        angles[0] += 0.02
        angles[1] += 0.03
        angles[2] += 0.05
        a1, a2, a3 = angles
        scaling = numpy.array([1.1 + numpy.cos(a1 * 3), 1.1 + numpy.cos(a2 * 2),
                               1.1 + numpy.cos(a3), 0]) * 0.5
        scaling[3] = 1.
        rot = rotation_3d(a1, a2, a3)
        apply_transform(mesh, scaling, rot, translate)


    mesh = aims.SurfaceGenerator.sphere(aims.Point3df(0, 0, 0), 1, 500)
    asphere = a.toAObject(mesh)
    asphere.releaseAppRef()
    aw = a.createWindow('3D')
    aw.setHasCursor(0)
    a.addObjects([asphere], [aw])


    if __name__ == '__main__':
        angles = numpy.array([0., 0., 0.])
        scaling = numpy.array([1., 1., 1., 0.])
        translate = numpy.array([0, 0, 0, 1])
        if 'sphinx_gallery' in sys.modules:
            for i in range(90):
                transform(mesh, angles, scaling, translate)
            asphere.setChanged()
            asphere.notifyObservers()
            QtGui.qApp.processEvents()
            # display in matplotlib for sphinx_gallery
            import matplotlib
            matplotlib.use('agg', force=True)  # force agg
            aw.imshow(show=True)

        elif run_loop:
            while a.getControlWindow().isVisible():
                transform(mesh, angles, scaling, translate)
                asphere.setChanged()
                asphere.notifyObservers()
                QtGui.qApp.processEvents()
                time.sleep(0.01)

        del aw, asphere, mesh


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

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


.. _sphx_glr_download_auto_examples_ellipsoid.py:

.. only:: html

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




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

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

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

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


.. only:: html

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

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