src/lib/ffmpeg_decoder.h

   1 /* -*- c-basic-offset: 8; default-tab-width: 8; -*- */
   2
   3 /*
   4     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with this program; if not, write to the Free Software
  18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19
  20 */
  21
  22 /** @file  src/ffmpeg_decoder.h
  23  *  @brief A decoder using FFmpeg to decode content.
  24  */
  25
  26 #include <vector>
  27 #include <string>
  28 #include <stdint.h>
  29 #include <boost/shared_ptr.hpp>
  30 #include <boost/optional.hpp>
  31 #include <boost/thread/mutex.hpp>
  32 extern "C" {
  33 #include <libavcodec/avcodec.h>
  34 #include <libpostproc/postprocess.h>
  35 }
  36 #include "util.h"
  37 #include "decoder.h"
  38 #include "video_decoder.h"
  39 #include "audio_decoder.h"
  40 #include "film.h"
  41 #include "ffmpeg_content.h"
  42
  43 struct AVFilterGraph;
  44 struct AVCodecContext;
  45 struct AVFilterContext;
  46 struct AVFormatContext;
  47 struct AVFrame;
  48 struct AVBufferContext;
  49 struct AVCodec;
  50 struct AVStream;
  51 class Job;
  52 class Options;
  53 class Image;
  54 class Log;
  55
  56 /** @class FFmpegDecoder
  57  *  @brief A decoder using FFmpeg to decode content.
  58  */
  59 class FFmpegDecoder : public VideoDecoder, public AudioDecoder
  60 {
  61 public:
  62         FFmpegDecoder (boost::shared_ptr<const Film>, boost::shared_ptr<const FFmpegContent>, bool video, bool audio, bool subtitles);
  63         ~FFmpegDecoder ();
  64
  65         /* Decoder */
  66
  67         void pass ();
  68         void seek (Time);
  69         void seek_back ();
  70         void seek_forward ();
  71         Time next () const;
  72
  73         /* VideoDecoder */
  74
  75         float video_frame_rate () const;
  76         libdcp::Size video_size () const;
  77         ContentVideoFrame video_length () const;
  78
  79         /* FFmpegDecoder */
  80
  81         std::vector<boost::shared_ptr<FFmpegSubtitleStream> > subtitle_streams () const {
  82                 return _subtitle_streams;
  83         }
  84
  85         std::vector<boost::shared_ptr<FFmpegAudioStream> > audio_streams () const {
  86                 return _audio_streams;
  87         }
  88
  89         boost::shared_ptr<const FFmpegContent> ffmpeg_content () const {
  90                 return _ffmpeg_content;
  91         }
  92
  93 private:
  94
  95         /* No copy construction */
  96         FFmpegDecoder (FFmpegDecoder const &);
  97         FFmpegDecoder& operator= (FFmpegDecoder const &);
  98
  99         PixelFormat pixel_format () const;
 100         AVSampleFormat audio_sample_format () const;
 101         int bytes_per_audio_sample () const;
 102         void do_seek (Time, bool, bool);
 103
 104         void setup_general ();
 105         void setup_video ();
 106         void setup_audio ();
 107         void setup_subtitle ();
 108
 109         bool decode_video_packet ();
 110         void decode_audio_packet ();
 111
 112         void maybe_add_subtitle ();
 113         boost::shared_ptr<AudioBuffers> deinterleave_audio (uint8_t** data, int size);
 114
 115         std::string stream_name (AVStream* s) const;
 116
 117         boost::shared_ptr<const FFmpegContent> _ffmpeg_content;
 118
 119         AVFormatContext* _format_context;
 120         int _video_stream;
 121
 122         AVFrame* _frame;
 123
 124         AVCodecContext* _video_codec_context;
 125         AVCodec* _video_codec;
 126         AVCodecContext* _audio_codec_context;    ///< may be 0 if there is no audio
 127         AVCodec* _audio_codec;                   ///< may be 0 if there is no audio
 128         AVCodecContext* _subtitle_codec_context; ///< may be 0 if there is no subtitle
 129         AVCodec* _subtitle_codec;                ///< may be 0 if there is no subtitle
 130
 131         AVPacket _packet;
 132
 133         std::list<boost::shared_ptr<FilterGraph> > _filter_graphs;
 134         boost::mutex _filter_graphs_mutex;
 135
 136         std::vector<boost::shared_ptr<FFmpegSubtitleStream> > _subtitle_streams;
 137         std::vector<boost::shared_ptr<FFmpegAudioStream> > _audio_streams;
 138
 139         bool _decode_video;
 140         bool _decode_audio;
 141         bool _decode_subtitles;
 142
 143         /* It would appear (though not completely verified) that one must have
 144            a mutex around calls to avcodec_open* and avcodec_close... and here
 145            it is.
 146         */
 147         static boost::mutex _mutex;
 148 };