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
53 class FFmpegFileEncoder
57 dcp::Size video_frame_size,
62 bool audio_stream_per_channel,
64 boost::filesystem::path output
67 ~FFmpegFileEncoder ();
69 void video (std::shared_ptr<PlayerVideo>, dcpomatic::DCPTime);
70 void audio (std::shared_ptr<AudioBuffers>);
71 void subtitle (PlayerText, dcpomatic::DCPTimePeriod);
75 static AVPixelFormat pixel_format (ExportFormat format);
81 void audio_frame (int size);
83 static void buffer_free(void* opaque, uint8_t* data);
84 void buffer_free2(uint8_t* data);
86 AVCodec const * _video_codec = nullptr;
87 AVCodecContext* _video_codec_context = nullptr;
88 std::vector<std::shared_ptr<ExportAudioStream>> _audio_streams;
89 bool _audio_stream_per_channel;
90 AVFormatContext* _format_context = nullptr;
91 AVStream* _video_stream = nullptr;
92 AVPixelFormat _pixel_format;
93 AVSampleFormat _sample_format;
94 AVDictionary* _video_options = nullptr;
95 std::string _video_codec_name;
96 std::string _audio_codec_name;
99 boost::filesystem::path _output;
100 dcp::Size _video_frame_size;
101 int _video_frame_rate;
102 int _audio_frame_rate;
104 int64_t _audio_frames = 0;
106 std::shared_ptr<AudioBuffers> _pending_audio;
108 /** Store of shared_ptr<Image> to keep them alive whilst raw pointers into
109 their data have been passed to FFmpeg.
111 std::map<uint8_t*, std::shared_ptr<const Image>> _pending_images;
112 boost::mutex _pending_images_mutex;
114 static int _video_stream_index;
115 static int _audio_stream_index_base;