src/lib/ffmpeg_decoder.h

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 /** @file  src/ffmpeg_decoder.h
  21  *  @brief A decoder using FFmpeg to decode content.
  22  */
  23
  24 #include <vector>
  25 #include <string>
  26 #include <stdint.h>
  27 #include <boost/shared_ptr.hpp>
  28 #include <boost/optional.hpp>
  29 extern "C" {
  30 #include <libavcodec/avcodec.h>
  31 #include <libpostproc/postprocess.h>
  32 }
  33 #include "util.h"
  34 #include "decoder.h"
  35 #include "video_decoder.h"
  36 #include "audio_decoder.h"
  37
  38 struct AVFilterGraph;
  39 struct AVCodecContext;
  40 struct AVFilterContext;
  41 struct AVFormatContext;
  42 struct AVFrame;
  43 struct AVBufferContext;
  44 struct AVCodec;
  45 struct AVStream;
  46 class Job;
  47 class Options;
  48 class Image;
  49 class Log;
  50
  51 /** @class FFmpegDecoder
  52  *  @brief A decoder using FFmpeg to decode content.
  53  */
  54 class FFmpegDecoder : public VideoDecoder, public AudioDecoder
  55 {
  56 public:
  57         FFmpegDecoder (boost::shared_ptr<Film>, boost::shared_ptr<const Options>, Job *);
  58         ~FFmpegDecoder ();
  59
  60         float frames_per_second () const;
  61         Size native_size () const;
  62         int time_base_numerator () const;
  63         int time_base_denominator () const;
  64         int sample_aspect_ratio_numerator () const;
  65         int sample_aspect_ratio_denominator () const;
  66
  67         void set_audio_stream (boost::optional<AudioStream>);
  68         void set_subtitle_stream (boost::optional<SubtitleStream>);
  69
  70 private:
  71
  72         bool pass ();
  73         PixelFormat pixel_format () const;
  74         AVSampleFormat audio_sample_format () const;
  75         int bytes_per_audio_sample () const;
  76
  77         void filter_and_emit_video (AVFrame *);
  78
  79         void setup_general ();
  80         void setup_video ();
  81         void setup_audio ();
  82         void setup_subtitle ();
  83
  84         void maybe_add_subtitle ();
  85         boost::shared_ptr<AudioBuffers> deinterleave_audio (uint8_t* data, int size);
  86
  87         std::string stream_name (AVStream* s) const;
  88
  89         AVFormatContext* _format_context;
  90         int _video_stream;
  91
  92         AVFrame* _frame;
  93
  94         AVCodecContext* _video_codec_context;
  95         AVCodec* _video_codec;
  96         AVCodecContext* _audio_codec_context;    ///< may be 0 if there is no audio
  97         AVCodec* _audio_codec;                   ///< may be 0 if there is no audio
  98         AVCodecContext* _subtitle_codec_context; ///< may be 0 if there is no subtitle
  99         AVCodec* _subtitle_codec;                ///< may be 0 if there is no subtitle
 100
 101         AVPacket _packet;
 102
 103         boost::optional<double> _first_video;
 104         boost::optional<double> _first_audio;
 105
 106         std::list<boost::shared_ptr<FilterGraph> > _filter_graphs;
 107 };