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

.. only:: html

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

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

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

.. _sphx_glr_gallery_scene_isosurface.py:


Isosurface Visual
=================

This example demonstrates the use of the Isosurface visual.

.. GENERATED FROM PYTHON SOURCE LINES 13-60



.. image-sg:: /gallery/scene/images/sphx_glr_isosurface_001.png
   :alt: isosurface
   :srcset: /gallery/scene/images/sphx_glr_isosurface_001.png
   :class: sphx-glr-single-img


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

 .. code-block:: none

    Generating scalar field..






|

.. code-block:: Python


    import sys
    import numpy as np

    from vispy import app, scene

    # Create a canvas with a 3D viewport
    canvas = scene.SceneCanvas(keys='interactive')
    view = canvas.central_widget.add_view()


    # Define a scalar field from which we will generate an isosurface
    def psi(i, j, k, offset=(25, 25, 50)):
        x = i-offset[0]
        y = j-offset[1]
        z = k-offset[2]
        th = np.arctan2(z, (x**2+y**2)**0.5)
        r = (x**2 + y**2 + z**2)**0.5
        a0 = 1
        ps = ((1./81.) * 1./(6.*np.pi)**0.5 * (1./a0)**(3/2) * (r/a0)**2 *
              np.exp(-r/(3*a0)) * (3 * np.cos(th)**2 - 1))
        return ps

    print("Generating scalar field..")
    data = np.abs(np.fromfunction(psi, (50, 50, 100)))

    # Create isosurface visual
    surface = scene.visuals.Isosurface(data, level=data.max()/4.,
                                       color=(0.5, 0.6, 1, 1), shading='smooth',
                                       parent=view.scene)
    surface.transform = scene.transforms.STTransform(translate=(-25, -25, -50))

    # Add a 3D axis to keep us oriented
    axis = scene.visuals.XYZAxis(parent=view.scene)

    # Use a 3D camera
    # Manual bounds; Mesh visual does not provide bounds yet
    # Note how you can set bounds before assigning the camera to the viewbox
    cam = scene.TurntableCamera(elevation=30, azimuth=30)
    cam.set_range((-10, 10), (-10, 10), (-10, 10))
    view.camera = cam


    if __name__ == '__main__':
        canvas.show()
        if sys.flags.interactive == 0:
            app.run()


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

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


.. _sphx_glr_download_gallery_scene_isosurface.py:

.. only:: html

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

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

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

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

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


.. only:: html

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

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