.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/04_demos/viz_play_video.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_04_demos_viz_play_video.py>`
        to download the full example code

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

.. _sphx_glr_auto_examples_04_demos_viz_play_video.py:


=======================================================
Play a video in the 3D world
=======================================================

The goal of this demo is to show how to visualize a video
on a rectangle by updating a texture.

.. GENERATED FROM PYTHON SOURCE LINES 9-94



.. image-sg:: /auto_examples/04_demos/images/sphx_glr_viz_play_video_001.png
   :alt: viz play video
   :srcset: /auto_examples/04_demos/images/sphx_glr_viz_play_video_001.png
   :class: sphx-glr-single-img





.. code-block:: Python


    import time

    import cv2
    import numpy as np

    from fury import actor, window


    # The VideoCapturer Class
    # This Class wraps OpenCV Videocapture
    class VideoCapturer:
        def __init__(self, video, time):
            self.path = video
            self.video = cv2.VideoCapture(self.path)
            self.fps = int(self.video.get(cv2.CAP_PROP_FPS))
            self.frames = int(self.video.get(cv2.CAP_PROP_FRAME_COUNT))
            self.time = time

        # A generator to yield video frames on every call
        def get_frame(self):
            start = time.time()
            for _ in range(self.frames):
                isframe, frame = self.video.read()
                dur = time.time() - start
                if dur > self.time:
                    break
                if isframe:
                    yield cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            self.video.release()
            yield None


    class VideoPlayer:
        def __init__(self, video, time=10):
            # Initializes the Video object with the given Video
            self.video = VideoCapturer(video, time)
            self.video_generator = self.video.get_frame()
            self.current_video_frame = next(self.video_generator)
            # Initialize Scene
            self.initialize_scene()
            # Create a Show Manager and Initialize it
            self.show_manager = window.ShowManager(
                self.scene, size=(900, 768), reset_camera=False, order_transparent=True
            )

        # Initialize the Scene with actors
        def initialize_scene(self):
            self.scene = window.Scene()
            # Initialize a Plane actor with the 1st video frame along with
            # the actor grid which is to be updated in each iteration
            self.plane_actor = actor.texture(self.current_video_frame)
            self.scene.add(self.plane_actor)

        # The timer_callback function getting called by the show manager
        def timer_callback(self, _obj, _event):
            self.current_video_frame = next(self.video_generator)
            if isinstance(self.current_video_frame, np.ndarray):
                # update texture of the actor with the current frame image
                # by updating the actor grid
                actor.texture_update(self.plane_actor, self.current_video_frame)
                self.show_manager.scene.azimuth(1.5)  # to rotate the camera
            else:
                self.show_manager.exit()

            self.show_manager.render()

        def run(self):
            # Add a timer callback to show manager after with
            # video frame duration as the interval
            self.frame_duration = int(1000 / self.video.fps)
            self.show_manager.add_timer_callback(
                True, self.frame_duration, self.timer_callback
            )
            self.show_manager.start()


    # Create VideoPlayer Object and run it
    video_url = (
        'http://commondatastorage.googleapis.com/'
        + 'gtv-videos-bucket/sample/BigBuckBunny.mp4'
    )
    vp = VideoPlayer(video_url)
    vp.run()
    window.record(vp.show_manager.scene, out_path='viz_play_video.png', size=(600, 600))


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

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


.. _sphx_glr_download_auto_examples_04_demos_viz_play_video.py:

.. only:: html

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

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

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

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

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


.. only:: html

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

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