X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fffmpeg_file_encoder.h;h=5bf501370ab6d9ad917b6808a0ffc1041b176378;hb=dba7e1137282b52a1bd6ad1d56fe6371a8c97e30;hp=5308947e6e2900b5f6d97a5f936518a67a5f3436;hpb=c4403784febdbdd42e9c32e67fadb147f11fe566;p=dcpomatic.git

diff --git a/src/lib/ffmpeg_file_encoder.h b/src/lib/ffmpeg_file_encoder.h
index 5308947e6..5bf501370 100644
--- a/src/lib/ffmpeg_file_encoder.h
+++ b/src/lib/ffmpeg_file_encoder.h
@@ -18,17 +18,37 @@
 
 */
 
+
 #ifndef DCPOMATIC_FFMPEG_FILE_ENCODER_H
 #define DCPOMATIC_FFMPEG_FILE_ENCODER_H
 
+
+#include "audio_mapping.h"
+#include "dcpomatic_time.h"
 #include "encoder.h"
 #include "event_history.h"
-#include "audio_mapping.h"
 #include "log.h"
+#include <dcp/key.h>
+#include <dcp/warnings.h>
+LIBDCP_DISABLE_WARNINGS
 extern "C" {
 #include <libavcodec/avcodec.h>
 #include <libavformat/avformat.h>
 }
+LIBDCP_ENABLE_WARNINGS
+
+
+class ExportAudioStream;
+
+
+enum class ExportFormat
+{
+	PRORES_4444,
+	PRORES_HQ,
+	H264_AAC,
+	SUBTITLES_DCP
+};
+
 
 class FFmpegFileEncoder
 {
@@ -39,12 +59,15 @@ public:
 		int audio_frame_rate,
 		int channels,
 		ExportFormat,
+		bool audio_stream_per_channel,
 		int x264_crf,
 		boost::filesystem::path output
 		);
 
-	void video (boost::shared_ptr<PlayerVideo>, dcpomatic::DCPTime);
-	void audio (boost::shared_ptr<AudioBuffers>);
+	~FFmpegFileEncoder ();
+
+	void video (std::shared_ptr<PlayerVideo>, dcpomatic::DCPTime);
+	void audio (std::shared_ptr<AudioBuffers>);
 	void subtitle (PlayerText, dcpomatic::DCPTimePeriod);
 
 	void flush ();
@@ -60,16 +83,15 @@ private:
 	static void buffer_free(void* opaque, uint8_t* data);
 	void buffer_free2(uint8_t* data);
 
-	AVCodec* _video_codec;
-	AVCodecContext* _video_codec_context;
-	AVCodec* _audio_codec;
-	AVCodecContext* _audio_codec_context;
-	AVFormatContext* _format_context;
-	AVStream* _video_stream;
-	AVStream* _audio_stream;
+	AVCodec const * _video_codec = nullptr;
+	AVCodecContext* _video_codec_context = nullptr;
+	std::vector<std::shared_ptr<ExportAudioStream>> _audio_streams;
+	bool _audio_stream_per_channel;
+	AVFormatContext* _format_context = nullptr;
+	AVStream* _video_stream = nullptr;
 	AVPixelFormat _pixel_format;
 	AVSampleFormat _sample_format;
-	AVDictionary* _video_options;
+	AVDictionary* _video_options = nullptr;
 	std::string _video_codec_name;
 	std::string _audio_codec_name;
 	int _audio_channels;
@@ -79,15 +101,18 @@ private:
 	int _video_frame_rate;
 	int _audio_frame_rate;
 
-	boost::shared_ptr<AudioBuffers> _pending_audio;
+	int64_t _audio_frames = 0;
+
+	std::shared_ptr<AudioBuffers> _pending_audio;
 
 	/** Store of shared_ptr<Image> to keep them alive whilst raw pointers into
 	    their data have been passed to FFmpeg.
 	*/
-	std::map<uint8_t*, boost::shared_ptr<const Image> > _pending_images;
+	std::map<uint8_t*, std::shared_ptr<const Image>> _pending_images;
+	boost::mutex _pending_images_mutex;
 
 	static int _video_stream_index;
-	static int _audio_stream_index;
+	static int _audio_stream_index_base;
 };
 
 #endif