src/lib/ffmpeg_decoder.h

   1 /*
   2     Copyright (C) 2012-2016 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of DCP-o-matic.
   5
   6     DCP-o-matic 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     DCP-o-matic 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 DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21
  22 /** @file  src/ffmpeg_decoder.h
  23  *  @brief A decoder using FFmpeg to decode content.
  24  */
  25
  26
  27 #include "bitmap_text.h"
  28 #include "decoder.h"
  29 #include "ffmpeg.h"
  30 #include "video_filter_graph_set.h"
  31 extern "C" {
  32 #include <libavcodec/avcodec.h>
  33 }
  34 #include <boost/thread/mutex.hpp>
  35 #include <stdint.h>
  36
  37
  38 class AudioBuffers;
  39 class FFmpegAudioStream;
  40 class Image;
  41 class Log;
  42 class VideoFilterGraph;
  43 struct ffmpeg_pts_offset_test;
  44
  45
  46 /** @class FFmpegDecoder
  47  *  @brief A decoder using FFmpeg to decode content.
  48  */
  49 class FFmpegDecoder : public FFmpeg, public Decoder
  50 {
  51 public:
  52         FFmpegDecoder (std::shared_ptr<const Film> film, std::shared_ptr<const FFmpegContent>, bool fast);
  53
  54         bool pass () override;
  55         void seek (dcpomatic::ContentTime time, bool) override;
  56
  57 private:
  58         friend struct ::ffmpeg_pts_offset_test;
  59
  60         enum class FlushResult {
  61                 DONE,
  62                 AGAIN
  63         };
  64
  65         FlushResult flush();
  66
  67         AVSampleFormat audio_sample_format (std::shared_ptr<FFmpegAudioStream> stream) const;
  68         int bytes_per_audio_sample (std::shared_ptr<FFmpegAudioStream> stream) const;
  69
  70         std::shared_ptr<FFmpegAudioStream> audio_stream_from_index (int index) const;
  71         void process_audio_frame (std::shared_ptr<FFmpegAudioStream> stream);
  72
  73         void process_video_frame ();
  74
  75         bool decode_and_process_video_packet (AVPacket* packet);
  76         void decode_and_process_audio_packet (AVPacket* packet);
  77         void decode_and_process_subtitle_packet (AVPacket* packet);
  78
  79         BitmapText process_bitmap_subtitle (AVSubtitleRect const * rect);
  80         void process_ass_subtitle (std::string ass, dcpomatic::ContentTime from);
  81
  82         void maybe_add_subtitle ();
  83
  84         FlushResult flush_codecs();
  85         FlushResult flush_fill();
  86
  87         VideoFilterGraphSet _filter_graphs;
  88
  89         dcpomatic::ContentTime _pts_offset;
  90         boost::optional<dcpomatic::ContentTime> _current_subtitle_to;
  91         /** true if we have a subtitle which has not had emit_stop called for it yet */
  92         bool _have_current_subtitle = false;
  93
  94         std::shared_ptr<Image> _black_image;
  95
  96         std::map<std::shared_ptr<FFmpegAudioStream>, boost::optional<dcpomatic::ContentTime>> _next_time;
  97
  98         enum class FlushState {
  99                 CODECS,
 100                 AUDIO_DECODER,
 101                 FILL,
 102         };
 103
 104         FlushState _flush_state = FlushState::CODECS;
 105 };