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 "util.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         bool flush ();
  61
  62         AVSampleFormat audio_sample_format (std::shared_ptr<FFmpegAudioStream> stream) const;
  63         int bytes_per_audio_sample (std::shared_ptr<FFmpegAudioStream> stream) const;
  64
  65         std::shared_ptr<FFmpegAudioStream> audio_stream_from_index (int index) const;
  66         void process_audio_frame (std::shared_ptr<FFmpegAudioStream> stream);
  67
  68         void process_video_frame ();
  69
  70         bool decode_and_process_video_packet (AVPacket* packet);
  71         void decode_and_process_audio_packet (AVPacket* packet);
  72         void decode_and_process_subtitle_packet (AVPacket* packet);
  73
  74         BitmapText process_bitmap_subtitle (AVSubtitleRect const * rect);
  75         void process_ass_subtitle (std::string ass, dcpomatic::ContentTime from);
  76
  77         void maybe_add_subtitle ();
  78
  79         std::list<std::shared_ptr<VideoFilterGraph>> _filter_graphs;
  80         boost::mutex _filter_graphs_mutex;
  81
  82         dcpomatic::ContentTime _pts_offset;
  83         boost::optional<dcpomatic::ContentTime> _current_subtitle_to;
  84         /** true if we have a subtitle which has not had emit_stop called for it yet */
  85         bool _have_current_subtitle = false;
  86
  87         std::shared_ptr<Image> _black_image;
  88
  89         std::map<std::shared_ptr<FFmpegAudioStream>, boost::optional<dcpomatic::ContentTime>> _next_time;
  90 };