Cleanup: move some stuff into process_video_frame().
[dcpomatic.git] / src / lib / ffmpeg_decoder.h
index 2d295db7b68b7bf3a1e6dd18fd604b77090f0e9f..29e0d22e6339d6bc957c8f9a9552805c468bdd16 100644 (file)
@@ -1,21 +1,20 @@
-/* -*- c-basic-offset: 8; default-tab-width: 8; -*- */
-
 /*
-    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2016 Carl Hetherington <cth@carlh.net>
+
+    This file is part of DCP-o-matic.
 
-    This program is free software; you can redistribute it and/or modify
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
  *  @brief A decoder using FFmpeg to decode content.
  */
 
-#include <vector>
-#include <string>
-#include <stdint.h>
-#include <boost/shared_ptr.hpp>
-#include <boost/optional.hpp>
-#include <boost/thread/mutex.hpp>
+#include "util.h"
+#include "decoder.h"
+#include "ffmpeg.h"
 extern "C" {
 #include <libavcodec/avcodec.h>
-#include <libpostproc/postprocess.h>
 }
-#include "util.h"
-#include "decoder.h"
-#include "video_decoder.h"
-#include "audio_decoder.h"
-
-struct AVFilterGraph;
-struct AVCodecContext;
-struct AVFilterContext;
-struct AVFormatContext;
-struct AVFrame;
-struct AVBufferContext;
-struct AVCodec;
-struct AVStream;
-class Job;
-class Options;
-class Image;
+#include <boost/thread/mutex.hpp>
+#include <stdint.h>
+
 class Log;
-class FFmpegContent;
-class Film;
+class VideoFilterGraph;
+class FFmpegAudioStream;
+class AudioBuffers;
+class Image;
+struct ffmpeg_pts_offset_test;
 
 /** @class FFmpegDecoder
  *  @brief A decoder using FFmpeg to decode content.
  */
-class FFmpegDecoder : public VideoDecoder, public AudioDecoder
+class FFmpegDecoder : public FFmpeg, public Decoder
 {
 public:
-       FFmpegDecoder (boost::shared_ptr<const Film>, boost::shared_ptr<const FFmpegContent>, bool video, bool audio, bool subtitles);
-       ~FFmpegDecoder ();
-
-       /* Decoder */
-
-       void pass ();
-       void seek (Time);
-       void seek_back ();
-       void seek_forward ();
-       Time next () const;
-       bool done () const;
-
-       /* VideoDecoder */
-
-       float video_frame_rate () const;
-       libdcp::Size video_size () const;
-       ContentVideoFrame video_length () const;
-
-       /* FFmpegDecoder */
+       FFmpegDecoder (std::shared_ptr<const Film> film, std::shared_ptr<const FFmpegContent>, bool fast);
 
-       std::vector<boost::shared_ptr<FFmpegSubtitleStream> > subtitle_streams () const {
-               return _subtitle_streams;
-       }
-       
-       std::vector<boost::shared_ptr<FFmpegAudioStream> > audio_streams () const {
-               return _audio_streams;
-       }
-
-       boost::shared_ptr<const FFmpegContent> ffmpeg_content () const {
-               return _ffmpeg_content;
-       }
+       bool pass () override;
+       void seek (dcpomatic::ContentTime time, bool) override;
 
 private:
+       friend struct ::ffmpeg_pts_offset_test;
 
-       /* No copy construction */
-       FFmpegDecoder (FFmpegDecoder const &);
-       FFmpegDecoder& operator= (FFmpegDecoder const &);
-
-       PixelFormat pixel_format () const;
-       AVSampleFormat audio_sample_format () const;
-       int bytes_per_audio_sample () const;
-       void do_seek (Time, bool, bool);
+       bool flush ();
 
-       void setup_general ();
-       void setup_video ();
-       void setup_audio ();
-       void setup_subtitle ();
+       static std::shared_ptr<AudioBuffers> deinterleave_audio (AVFrame* frame);
 
-       bool decode_video_packet ();
-       void decode_audio_packet ();
+       AVSampleFormat audio_sample_format (std::shared_ptr<FFmpegAudioStream> stream) const;
+       int bytes_per_audio_sample (std::shared_ptr<FFmpegAudioStream> stream) const;
 
-       void maybe_add_subtitle ();
-       boost::shared_ptr<AudioBuffers> deinterleave_audio (uint8_t** data, int size);
+       std::shared_ptr<FFmpegAudioStream> audio_stream_from_index (int index) const;
+       void process_audio_frame (std::shared_ptr<FFmpegAudioStream> stream);
 
-       std::string stream_name (AVStream* s) const;
+       void process_video_frame ();
 
-       boost::shared_ptr<const FFmpegContent> _ffmpeg_content;
+       bool decode_and_process_video_packet (AVPacket* packet);
+       void decode_and_process_audio_packet (AVPacket* packet);
+       void decode_and_process_subtitle_packet (AVPacket* packet);
 
-       AVFormatContext* _format_context;
-       int _video_stream;
-       
-       AVFrame* _frame;
+       void process_bitmap_subtitle (AVSubtitleRect const * rect, dcpomatic::ContentTime from);
+       void process_ass_subtitle (std::string ass, dcpomatic::ContentTime from);
 
-       AVCodecContext* _video_codec_context;
-       AVCodec* _video_codec;
-       AVCodecContext* _audio_codec_context;    ///< may be 0 if there is no audio
-       AVCodec* _audio_codec;                   ///< may be 0 if there is no audio
-       AVCodecContext* _subtitle_codec_context; ///< may be 0 if there is no subtitle
-       AVCodec* _subtitle_codec;                ///< may be 0 if there is no subtitle
-
-       AVPacket _packet;
+       void maybe_add_subtitle ();
 
-       std::list<boost::shared_ptr<FilterGraph> > _filter_graphs;
+       std::list<std::shared_ptr<VideoFilterGraph> > _filter_graphs;
        boost::mutex _filter_graphs_mutex;
 
-        std::vector<boost::shared_ptr<FFmpegSubtitleStream> > _subtitle_streams;
-        std::vector<boost::shared_ptr<FFmpegAudioStream> > _audio_streams;
+       dcpomatic::ContentTime _pts_offset;
+       boost::optional<dcpomatic::ContentTime> _current_subtitle_to;
+       /** true if we have a subtitle which has not had emit_stop called for it yet */
+       bool _have_current_subtitle = false;
 
-       bool _decode_video;
-       bool _decode_audio;
-       bool _decode_subtitles;
+       std::shared_ptr<Image> _black_image;
 
-       /* It would appear (though not completely verified) that one must have
-          a mutex around calls to avcodec_open* and avcodec_close... and here
-          it is.
-       */
-       static boost::mutex _mutex;
+       std::map<std::shared_ptr<FFmpegAudioStream>, boost::optional<dcpomatic::ContentTime>> _next_time;
 };