2 Copyright (C) 2017-2018 Carl Hetherington <cth@carlh.net>
4 This file is part of DCP-o-matic.
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.
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.
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/>.
21 #ifndef DCPOMATIC_FFMPEG_FILE_ENCODER_H
22 #define DCPOMATIC_FFMPEG_FILE_ENCODER_H
25 #include "event_history.h"
26 #include "audio_mapping.h"
30 #include <libavcodec/avcodec.h>
31 #include <libavformat/avformat.h>
35 class ExportAudioStream;
38 class FFmpegFileEncoder
42 dcp::Size video_frame_size,
47 bool audio_stream_per_channel,
49 boost::filesystem::path output
50 #ifdef DCPOMATIC_VARIANT_SWAROOP
51 , boost::optional<dcp::Key> key
52 , boost::optional<std::string> id
56 ~FFmpegFileEncoder ();
58 void video (boost::shared_ptr<PlayerVideo>, dcpomatic::DCPTime);
59 void audio (boost::shared_ptr<AudioBuffers>);
60 void subtitle (PlayerText, dcpomatic::DCPTimePeriod);
64 static AVPixelFormat pixel_format (ExportFormat format);
70 void audio_frame (int size);
72 static void buffer_free(void* opaque, uint8_t* data);
73 void buffer_free2(uint8_t* data);
75 AVCodec* _video_codec;
76 AVCodecContext* _video_codec_context;
77 std::vector<boost::shared_ptr<ExportAudioStream> > _audio_streams;
78 bool _audio_stream_per_channel;
79 AVFormatContext* _format_context;
80 AVStream* _video_stream;
81 AVPixelFormat _pixel_format;
82 AVSampleFormat _sample_format;
83 AVDictionary* _video_options;
84 std::string _video_codec_name;
85 std::string _audio_codec_name;
88 boost::filesystem::path _output;
89 dcp::Size _video_frame_size;
90 int _video_frame_rate;
91 int _audio_frame_rate;
93 int64_t _audio_frames;
95 boost::shared_ptr<AudioBuffers> _pending_audio;
97 /** Store of shared_ptr<Image> to keep them alive whilst raw pointers into
98 their data have been passed to FFmpeg.
100 std::map<uint8_t*, boost::shared_ptr<const Image> > _pending_images;
101 boost::mutex _pending_images_mutex;
103 static int _video_stream_index;
104 static int _audio_stream_index_base;