mapmap/MediaImpl.h

/*
 * MediaImpl.h
 *
 * (c) 2013 Sofian Audry -- info(@)sofianaudry(.)com
 * (c) 2013 Alexandre Quessy -- alexandre(@)quessy(.)net
 * (c) 2012 Jean-Sebastien Senecal
 * (c) 2004 Mathieu Guindon, Julien Keable
 *           Based on code from Drone http://github.com/sofian/drone
 *           Based on code from the GStreamer Tutorials http://docs.gstreamer.com/display/GstSDK/Tutorials
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef VIDEO_IMPL_H_
#define VIDEO_IMPL_H_

#include <glib.h>
#include <gst/gst.h>
#include <gst/app/gstappsink.h>
#include <QtGlobal>
#include <QtOpenGL>
#if __APPLE__
#include <OpenGL/gl.h>
#else
#include <GL/gl.h>
#endif
#include <tr1/memory>
#include "AsyncQueue.h"

/**
 * Private declaration of the video player.
 * This is done this way so that GStreamer header don't need to be 
 * included in the whole project. (just in this file)
 */
class MediaImpl
{
public:
  /**
   * Constructor.
   * This media player works for both video files and shared memory sockets.
   * If live is true, it's a shared memory socket.
   */
  MediaImpl(const QString uri, bool live);
  ~MediaImpl();

//  void setUri(const QString uri);
  /**
   * Returns whether or not GStreamer video support is ok.
   */
  static bool hasVideoSupport();
  /**
   * Sets up the player.
   * Basically calls loadMovie().
   */
  void build();
  /**
   * Returns the width of the video image.
   */
  int getWidth() const;
  /**
   * Returns the height of the video image.
   */
  int getHeight() const;
  /**
   * Returns the path to the media file being played.
   */
  QString getUri() const;
  /**
   * When using the shared memory source, returns whether or not we
   * are attached to a shared memory socket.
   */
  bool getAttached();
  /**
   * Returns the raw image of the last video frame.
   * It is currently unused!
   */
  const uchar* getBits() const;

  /**
   * Checks if the pipeline is ready.
   *
   * Returns whether or not the elements in the pipeline are connected,
   * and if we are using shmsrc, if the shared memory socket is being read.
   */
  bool isReady() const { return _padHandlerData.videoIsConnected; }

  /**
   * Tries to pull a video frame from the queue. 
   *
   * Returns true if the image has changed.
   */
  bool runVideo();

  // void runAudio();

  /**
   * Loads a new movie file.
   * 
   * Creates a new GStreamer pipeline, opens a movie or a shmsrc socket.
   */
  bool loadMovie(QString filename);
  bool setPlayState(bool play);
  /**
   * Tells the MediaImpl that we are actually reading from a shmsrc.
   * Called from the GStreamer callback of the shmsrc.
   */
  void setAttached(bool attach);

  void resetMovie();

protected:
  void unloadMovie();
  void freeResources();

private:
  /**
   * Tries to pull a video frame from the asynchronous input buffer.
   *
   * Pushes the frame into the asynchronous output buffer.
   */
  bool _videoPull();
  /**
   * Checks if we reached the end of the video file.
   *
   * Returns false if the pipeline is not ready yet.
   */
  bool _eos() const;

  // void _finish();
  // void _init();

  bool _preRun();
  void _postRun();
  void _setReady(bool ready);
  void _setFinished(bool finished);

public:
  // GStreamer callbacks.

  struct GstPadHandlerData {
    GstElement* videoToConnect;
    GstElement* videoSink;
    bool videoIsConnected;
    int width;
    int height;

    GstPadHandlerData() :
      videoToConnect(NULL), videoSink(NULL),
      videoIsConnected(false),
      width(-1), height(-1)
    {}
  };

  // GStreamer callback that simply sets the #newSample# flag to point to TRUE.
  static GstFlowReturn gstNewSampleCallback(GstElement*, MediaImpl *p);
  static GstFlowReturn gstNewPreRollCallback (GstAppSink * appsink, gpointer user_data);

  // GStreamer callback that plugs the audio/video pads into the proper elements when they
  // are made available by the source.
  static void gstPadAddedCallback(GstElement *src, GstPad *newPad, MediaImpl::GstPadHandlerData* data);

  AsyncQueue<GstSample*> *getQueueInputBuffer() {
    return &this->_queueInputBuffer;
  }

  AsyncQueue<GstSample*> *getQueueOutputBuffer() {
    return &this->_queueOutputBuffer;
  }

private:
  //locals
  /**
   * Path of the movie being played.
   */
  QString _currentMovie;

  // gstreamer elements
  GstBus *_bus;
  GstElement *_pipeline;
  GstElement *_uridecodebin0;
  GstElement *_shmsrc0;
  GstElement *_gdpdepay0;
  //GstElement *_audioQueue;
  //GstElement *_audioConvert;
  //GstElement *_audioResample;
  GstElement *_queue0;
  GstElement *_videoconvert0;
  //GstElement *_audioSink;
  GstElement *_appsink0;
  /**
   * Temporary contains the image data of the last frame.
   */
  GstSample  *_frame;
  /**
   * shmsrc socket poller.
   */
  GSource *_pollSource;

  GstPadHandlerData _padHandlerData;

  // unused
  int _width;
  // unused
  int _height;

  /**
   * Raw image data of the last video frame.
   */
  uchar *_data;

  bool _seekEnabled;
  /**
   * Whether or not we are reading video from a shmsrc.
   */
  bool _isSharedMemorySource;
  /**
   * Whether or not we are attached to a shmsrc, if using a shmsrc.
   */
  bool _attached;

  int _audioNewBufferCounter;

  bool _terminate;
  bool _movieReady;
  /**
   * Input buffer.
   * Contains the raw data of last image.
   */
  AsyncQueue<GstSample*> _queueInputBuffer;
  /**
   * Output buffer.
   * Contains the raw data of last image.
   */
  AsyncQueue<GstSample*> _queueOutputBuffer;

private:
  /**
   * Path of the movie file being played.
   */
  QString _uri;
};

#endif /* ifndef */