libvisp-io-dev 3.0.0-4 / usr / include / x86_64-linux-gnu / visp3 / io / vpVideoReader.h

/****************************************************************************
 *
 * This file is part of the ViSP software.
 * Copyright (C) 2005 - 2015 by Inria. All rights reserved.
 *
 * This software is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * ("GPL") version 2 as published by the Free Software Foundation.
 * See the file LICENSE.txt at the root directory of this source
 * distribution for additional information about the GNU GPL.
 *
 * For using ViSP with software that can not be combined with the GNU
 * GPL, please contact Inria about acquiring a ViSP Professional
 * Edition License.
 *
 * See http://visp.inria.fr for more information.
 *
 * This software was developed at:
 * Inria Rennes - Bretagne Atlantique
 * Campus Universitaire de Beaulieu
 * 35042 Rennes Cedex
 * France
 *
 * If you have questions regarding the use of this file, please contact
 * Inria at visp@inria.fr
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 * Description:
 * Read videos and sequences of images .
 *
 * Authors:
 * Nicolas Melchior
 * Fabien Spindler
 *
 *****************************************************************************/

/*!
  \file vpVideoReader.h
  \brief Read videos and image sequences
*/

#ifndef vpVideoReader_H
#define vpVideoReader_H

#include <string>

#include <visp3/io/vpDiskGrabber.h>
#include <visp3/io/vpFFMPEG.h>

#if VISP_HAVE_OPENCV_VERSION >= 0x020200
#include "opencv2/highgui/highgui.hpp"
#elif VISP_HAVE_OPENCV_VERSION >= 0x020000
#include "opencv/highgui.h"
#endif

/*!
  \class vpVideoReader

  \ingroup group_io_video

  \brief Class that enables to manipulate easily a video file or a sequence of
  images. As it inherits from the vpFrameGrabber Class, it can be used like an
  other frame grabber class.
  
  The following example available in tutorial-video-reader.cpp shows how this
  class is really easy to use. It enables to read a video file named video.mpeg.
  \include tutorial-video-reader.cpp

  As shown in the next example, this class allows also to access to a specific
  frame. But be careful, for video files, the getFrame() method is not precise
  and returns the nearest intra key frame from the expected frame. You can use
  the getFrame() method to position the reader in the video and then use the
  acquire() method to get the following frames one by one.
  \code
#include <visp3/core/vpImage.h>
#include <visp3/core/vpRGBa.h>
#include <visp3/io/vpVideoReader.h>

int main()
{
#ifdef VISP_HAVE_FFMPEG
  vpImage<vpRGBa> I;

  vpVideoReader reader;

  // Initialize the reader.
  reader.setFileName("video.mpeg");
  reader.open(I);

  // Read the nearest key frame from the 3th frame
  reader.getFrame(I,2);

  // After positionning the video reader use acquire to read the video frame by frame
  reader.acquire(I);

  return 0;
#endif
}
  \endcode
  
  The other following example explains how to use the class to read a
  sequence of images. The images are stored in the folder "./image" and are
  named "image0000.jpeg", "image0001.jpeg", "image0002.jpeg", ... As explained
  in setFirstFrameIndex() and setLastFrameIndex() it is also possible to set the
  first and last image numbers to read a portion of the sequence. If these two
  functions are not used, first and last image numbers are set automatically to
  match the first and image images of the sequence.

  \code
#include <visp3/core/vpImage.h>
#include <visp3/core/vpRGBa.h>
#include <visp3/io/vpVideoReader.h>

int main()
{
  vpImage<vpRGBa> I;

  vpVideoReader reader;

  // Initialize the reader.
  reader.setFileName("./image/image%04d.jpeg");
  reader.setFirstFrameIndex(10);
  reader.setLastFrameIndex(20);
  reader.open(I);

  while (! reader.end() )
    reader.acquire(I);

  return 0;
}
  \endcode
  
  Note that it is also possible to access to a specific frame using getFrame().
  \code
#include <visp3/core/vpImage.h>
#include <visp3/core/vpRGBa.h>
#include <visp3/io/vpVideoReader.h>

int main()
{
  vpImage<vpRGBa> I;

  vpVideoReader reader;

  // Initialize the reader.
  reader.setFileName("./image/image%04d.jpeg");
  reader.open(I);

  // Read the 3th frame
  reader.getFrame(I,2);

  return 0;
}
  \endcode
*/

class VISP_EXPORT vpVideoReader : public vpFrameGrabber
{    
  private:
    //!To read sequences of images
    vpDiskGrabber *imSequence;
#ifdef VISP_HAVE_FFMPEG
    //!To read video files
    vpFFMPEG *ffmpeg;
#elif VISP_HAVE_OPENCV_VERSION >= 0x020100
    //!To read video files with OpenCV
    cv::VideoCapture capture;
    cv::Mat frame;
#endif
    //!Types of available formats
    typedef enum
    {
      FORMAT_PGM,
      FORMAT_PPM,
      FORMAT_JPEG,
      FORMAT_PNG,
      // Formats supported by opencv
      FORMAT_TIFF,
      FORMAT_BMP,
      FORMAT_DIB,
      FORMAT_PBM,
      FORMAT_RASTER,
      FORMAT_JPEG2000,
      // Video format
      FORMAT_AVI,
      FORMAT_MPEG,
      FORMAT_MPEG4,
      FORMAT_MOV,
      FORMAT_OGV,
      FORMAT_WMV,
      FORMAT_FLV,
      FORMAT_MKV,
      FORMAT_UNKNOWN
    } vpVideoFormatType;
    
    //!Video's format which has to be read
    vpVideoFormatType formatType;
    
    //!Path to the video
    char fileName[FILENAME_MAX];
    //!Indicates if the path to the video is set.
    bool initFileName;
    //!Indicates if the video is "open".
    bool isOpen;
    //!Count the frame number when the class is used as a grabber.
    long frameCount; // Index of the next image
    //!The first frame index
    long firstFrame;
    //!The last frame index
    long lastFrame;
    bool firstFrameIndexIsSet;
    bool lastFrameIndexIsSet;

//private:
//#ifndef DOXYGEN_SHOULD_SKIP_THIS
//    vpVideoReader(const vpVideoReader &)
//      : vpFrameGrabber(), imSequence(NULL),
//    #ifdef VISP_HAVE_FFMPEG
//        ffmpeg(NULL),
//    #elif VISP_HAVE_OPENCV_VERSION >= 0x020100
//        capture(), frame(),
//    #endif
//        formatType(FORMAT_UNKNOWN), initFileName(false), isOpen(false), frameCount(0),
//        firstFrame(0), lastFrame(0), firstFrameIndexIsSet(false), lastFrameIndexIsSet(false)
//    {
//      throw vpException(vpException::functionNotImplementedError, "Not implemented!");
//    }
//    vpVideoReader &operator=(const vpVideoReader &){
//      throw vpException(vpException::functionNotImplementedError, "Not implemented!");
//      return *this;
//    }
//#endif

  public:
    vpVideoReader();
    ~vpVideoReader();
    
    void acquire(vpImage< vpRGBa > &I);
    void acquire(vpImage< unsigned char > &I);
    void close(){;}

    /*!
      \return true if the end of the sequence is reached.
    */
    inline bool end() {
      if (frameCount > lastFrame )
        return true;
      return false;
    }
    bool getFrame(vpImage<vpRGBa> &I, long frame);
    bool getFrame(vpImage<unsigned char> &I, long frame);
    double getFramerate();

    /*!
      Get the frame index of the next image. This index is updated at each call of the
      acquire method. It can be used to detect the end of a file (comparison
      with getLastFrameIndex()).

      \return Returns the current frame index.
    */
    inline long getFrameIndex() const { return frameCount;}

    /*!
      Gets the first frame index.

      \return Returns the first frame index.
    */
    inline long getFirstFrameIndex() const {return firstFrame;}
    /*!
      Gets the last frame index.

      \return Returns the last frame index.
    */
    inline long getLastFrameIndex() const {return lastFrame;}
    void open (vpImage< vpRGBa > &I);
    void open (vpImage< unsigned char > &I);

    /*!
      Reset the frame counter and sets it to the first image index.

      By default the first frame index is set to 0.

      This method is useful if you use the class like a frame grabber (ie with theacquire method).
    */
    inline void resetFrameCounter() {frameCount = firstFrame;}
    void setFileName(const char *filename);
    void setFileName(const std::string &filename);
    /*!
      Enables to set the first frame index if you want to use the class like a grabber (ie with the
      acquire method).

      \param first_frame : The first frame index.

      \sa setLastFrameIndex()
    */
    inline void setFirstFrameIndex(const long first_frame) {
      this->firstFrameIndexIsSet = true;
      this->firstFrame = first_frame;
    }
    /*!
      Enables to set the last frame index.

      \param last_frame : The last frame index.

      \sa setFirstFrameIndex()
    */
    inline void setLastFrameIndex(const long last_frame) {
      this->lastFrameIndexIsSet = true;
      this->lastFrame = last_frame;
    }

  private:
    vpVideoFormatType getFormat(const char *filename);
    static std::string getExtension(const std::string &filename);
    void findFirstFrameIndex();
    void findLastFrameIndex();
	bool isImageExtensionSupported();
	bool isVideoExtensionSupported();
};

#endif