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

.. only:: html

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

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

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

.. _sphx_glr__auto_examples_containment_test_plot_capsule.py:


===================
Capsule Containment
===================

.. GENERATED FROM PYTHON SOURCE LINES 6-33



.. image-sg:: /_auto_examples/containment_test/images/sphx_glr_plot_capsule_001.png
   :alt: plot capsule
   :srcset: /_auto_examples/containment_test/images/sphx_glr_plot_capsule_001.png
   :class: sphx-glr-single-img


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

 .. code-block:: none


    0.006920337677001953 s






|

.. code-block:: default

    print(__doc__)
    import time
    import numpy as np
    import matplotlib.pyplot as plt
    import pytransform3d.plot_utils as ppu
    from distance3d import containment_test


    random_state = np.random.RandomState(0)
    capsule2origin = np.eye(4)
    radius = 0.5
    height = 1.0

    ax = ppu.make_3d_axis(ax_s=3)
    points = random_state.rand(100000, 3)
    points[:, 0] -= 0.5
    points[:, 0] *= 2.0
    points[:, 2] -= 0.5
    points[:, 2] *= 2.0
    start = time.time()
    contained = containment_test.points_in_capsule(points, capsule2origin, radius, height)
    stop = time.time()
    print(f"{stop - start} s")
    ax.scatter(points[::10, 0], points[::10, 1], points[::10, 2], c=contained[::10])
    ppu.plot_capsule(ax=ax, A2B=capsule2origin, radius=radius, height=height, wireframe=True, color="r")
    ppu.plot_capsule(ax=ax, A2B=capsule2origin, radius=radius, height=height, wireframe=False, alpha=0.5)
    plt.show()


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

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


.. _sphx_glr_download__auto_examples_containment_test_plot_capsule.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_capsule.py <plot_capsule.py>`

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

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


.. only:: html

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

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