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 /** @file  src/ffmpeg_decoder.h
  22  *  @brief A decoder using FFmpeg to decode content.
  23  */
  24
  25 #include "util.h"
  26 #include "decoder.h"
  27 #include "ffmpeg.h"
  28 extern "C" {
  29 #include <libavcodec/avcodec.h>
  30 }
  31 #include <boost/thread/mutex.hpp>
  32 #include <stdint.h>
  33
  34 class Log;
  35 class VideoFilterGraph;
  36 class FFmpegAudioStream;
  37 class AudioBuffers;
  38 class Image;
  39 struct ffmpeg_pts_offset_test;
  40
  41 /** @class FFmpegDecoder
  42  *  @brief A decoder using FFmpeg to decode content.
  43  */
  44 class FFmpegDecoder : public FFmpeg, public Decoder
  45 {
  46 public:
  47         FFmpegDecoder (std::shared_ptr<const Film> film, std::shared_ptr<const FFmpegContent>, bool fast);
  48
  49         bool pass () override;
  50         void seek (dcpomatic::ContentTime time, bool) override;
  51
  52 private:
  53         friend struct ::ffmpeg_pts_offset_test;
  54
  55         bool flush ();
  56
  57         static std::shared_ptr<AudioBuffers> deinterleave_audio (AVFrame* frame);
  58
  59         AVSampleFormat audio_sample_format (std::shared_ptr<FFmpegAudioStream> stream) const;
  60         int bytes_per_audio_sample (std::shared_ptr<FFmpegAudioStream> stream) const;
  61
  62         std::shared_ptr<FFmpegAudioStream> audio_stream_from_index (int index) const;
  63         void process_audio_frame (std::shared_ptr<FFmpegAudioStream> stream);
  64
  65         bool decode_and_process_video_packet (AVPacket* packet);
  66         void decode_and_process_audio_packet (AVPacket* packet);
  67         void decode_and_process_subtitle_packet (AVPacket* packet);
  68
  69         void process_bitmap_subtitle (AVSubtitleRect const * rect, dcpomatic::ContentTime from);
  70         void process_ass_subtitle (std::string ass, dcpomatic::ContentTime from);
  71
  72         void maybe_add_subtitle ();
  73
  74         std::list<std::shared_ptr<VideoFilterGraph> > _filter_graphs;
  75         boost::mutex _filter_graphs_mutex;
  76
  77         dcpomatic::ContentTime _pts_offset;
  78         boost::optional<dcpomatic::ContentTime> _current_subtitle_to;
  79         /** true if we have a subtitle which has not had emit_stop called for it yet */
  80         bool _have_current_subtitle = false;
  81
  82         /** number of errors from avcodec_send_packet to ignore */
  83         int _errors_to_ignore = 0;
  84
  85         std::shared_ptr<Image> _black_image;
  86
  87         std::map<std::shared_ptr<FFmpegAudioStream>, boost::optional<dcpomatic::ContentTime>> _next_time;
  88 };