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/>.
22 #ifndef DCPOMATIC_FFMPEG_FILE_ENCODER_H
23 #define DCPOMATIC_FFMPEG_FILE_ENCODER_H
26 #include "audio_mapping.h"
27 #include "dcpomatic_time.h"
29 #include "event_history.h"
32 #include <dcp/warnings.h>
33 LIBDCP_DISABLE_WARNINGS
35 #include <libavcodec/avcodec.h>
36 #include <libavformat/avformat.h>
38 LIBDCP_ENABLE_WARNINGS
41 class ExportAudioStream;
44 enum class ExportFormat
52 class FFmpegFileEncoder
56 dcp::Size video_frame_size,
61 bool audio_stream_per_channel,
63 boost::filesystem::path output
66 ~FFmpegFileEncoder ();
68 void video (std::shared_ptr<PlayerVideo>, dcpomatic::DCPTime);
69 void audio (std::shared_ptr<AudioBuffers>);
70 void subtitle (PlayerText, dcpomatic::DCPTimePeriod);
74 static AVPixelFormat pixel_format (ExportFormat format);
80 void audio_frame (int size);
82 static void buffer_free(void* opaque, uint8_t* data);
83 void buffer_free2(uint8_t* data);
85 AVCodec const * _video_codec = nullptr;
86 AVCodecContext* _video_codec_context = nullptr;
87 std::vector<std::shared_ptr<ExportAudioStream>> _audio_streams;
88 bool _audio_stream_per_channel;
89 AVFormatContext* _format_context = nullptr;
90 AVStream* _video_stream = nullptr;
91 AVPixelFormat _pixel_format;
92 AVSampleFormat _sample_format;
93 AVDictionary* _video_options = nullptr;
94 std::string _video_codec_name;
95 std::string _audio_codec_name;
98 boost::filesystem::path _output;
99 dcp::Size _video_frame_size;
100 int _video_frame_rate;
101 int _audio_frame_rate;
103 int64_t _audio_frames = 0;
105 std::shared_ptr<AudioBuffers> _pending_audio;
107 /** Store of shared_ptr<Image> to keep them alive whilst raw pointers into
108 their data have been passed to FFmpeg.
110 std::map<uint8_t*, std::shared_ptr<const Image>> _pending_images;
111 boost::mutex _pending_images_mutex;
113 static int _video_stream_index;
114 static int _audio_stream_index_base;