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 extern "C" {
  29 #include <libavcodec/avcodec.h>
  30 #include <libpostproc/postprocess.h>
  31 }
  32 #include "util.h"
  33 #include "decoder.h"
  34
  35 struct AVFilterGraph;
  36 struct AVCodecContext;
  37 struct AVFilterContext;
  38 struct AVFormatContext;
  39 struct AVFrame;
  40 struct AVBufferContext;
  41 struct AVCodec;
  42 struct AVStream;
  43 class Job;
  44 class FilmState;
  45 class Options;
  46 class Image;
  47 class Log;
  48
  49 /** @class FFmpegDecoder
  50  *  @brief A decoder using FFmpeg to decode content.
  51  */
  52 class FFmpegDecoder : public Decoder
  53 {
  54 public:
  55         FFmpegDecoder (boost::shared_ptr<const FilmState>, boost::shared_ptr<const Options>, Job *, Log *, bool, bool);
  56         ~FFmpegDecoder ();
  57
  58         /* Methods to query our input video */
  59         int length_in_frames () const;
  60         int decoding_frames () const;
  61         float frames_per_second () const;
  62         Size native_size () const;
  63         int audio_channels () const;
  64         int audio_sample_rate () const;
  65         AVSampleFormat audio_sample_format () const;
  66         int64_t audio_channel_layout () const;
  67         bool has_subtitles () const;
  68         int bytes_per_audio_sample () const;
  69         int audio_to_discard () const;
  70
  71         std::vector<AudioStream> audio_streams () const;
  72         std::vector<SubtitleStream> subtitle_streams () const;
  73
  74 private:
  75
  76         bool do_pass ();
  77         PixelFormat pixel_format () const;
  78         int time_base_numerator () const;
  79         int time_base_denominator () const;
  80         int sample_aspect_ratio_numerator () const;
  81         int sample_aspect_ratio_denominator () const;
  82
  83         void setup_general ();
  84         void setup_video ();
  85         void setup_audio ();
  86         void setup_subtitle ();
  87
  88         void maybe_add_subtitle ();
  89
  90         std::string stream_name (AVStream* s) const;
  91
  92         AVFormatContext* _format_context;
  93         int _video_stream;
  94         int _audio_stream; ///< may be < 0 if there is no audio
  95         int _subtitle_stream; ///< may be < 0 if there is no subtitle
  96         AVFrame* _frame;
  97
  98         std::vector<AudioStream> _audio_streams;
  99         std::vector<SubtitleStream> _subtitle_streams;
 100
 101         AVCodecContext* _video_codec_context;
 102         AVCodec* _video_codec;
 103         AVCodecContext* _audio_codec_context;    ///< may be 0 if there is no audio
 104         AVCodec* _audio_codec;                   ///< may be 0 if there is no audio
 105         AVCodecContext* _subtitle_codec_context; ///< may be 0 if there is no subtitle
 106         AVCodec* _subtitle_codec;                ///< may be 0 if there is no subtitle
 107
 108         AVPacket _packet;
 109
 110         int64_t _first_video_pts;
 111         int64_t _first_audio_pts;
 112 };