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
27 #include "event_history.h"
28 #include "audio_mapping.h"
32 #include <libavcodec/avcodec.h>
33 #include <libavformat/avformat.h>
37 class ExportAudioStream;
40 class FFmpegFileEncoder
44 dcp::Size video_frame_size,
49 bool audio_stream_per_channel,
51 boost::filesystem::path output
54 ~FFmpegFileEncoder ();
56 void video (std::shared_ptr<PlayerVideo>, dcpomatic::DCPTime);
57 void audio (std::shared_ptr<AudioBuffers>);
58 void subtitle (PlayerText, dcpomatic::DCPTimePeriod);
62 static AVPixelFormat pixel_format (ExportFormat format);
68 void audio_frame (int size);
70 static void buffer_free(void* opaque, uint8_t* data);
71 void buffer_free2(uint8_t* data);
73 AVCodec* _video_codec = nullptr;
74 AVCodecContext* _video_codec_context = nullptr;
75 std::vector<std::shared_ptr<ExportAudioStream>> _audio_streams;
76 bool _audio_stream_per_channel;
77 AVFormatContext* _format_context = nullptr;
78 AVStream* _video_stream = nullptr;
79 AVPixelFormat _pixel_format;
80 AVSampleFormat _sample_format;
81 AVDictionary* _video_options = nullptr;
82 std::string _video_codec_name;
83 std::string _audio_codec_name;
86 boost::filesystem::path _output;
87 dcp::Size _video_frame_size;
88 int _video_frame_rate;
89 int _audio_frame_rate;
91 int64_t _audio_frames = 0;
93 std::shared_ptr<AudioBuffers> _pending_audio;
95 /** Store of shared_ptr<Image> to keep them alive whilst raw pointers into
96 their data have been passed to FFmpeg.
98 std::map<uint8_t*, std::shared_ptr<const Image>> _pending_images;
99 boost::mutex _pending_images_mutex;
101 static int _video_stream_index;
102 static int _audio_stream_index_base;