X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Fquicksync_encoder.cpp;fp=nageru%2Fquicksync_encoder.cpp;h=eb144ceb2512f70044e5ae55d9b17b6e77ee4135;hb=8202dbe236c5e206989c383004f9dba116ea12bd;hp=992bf8f83e037e3bf8df40b02df23ede0fbcd2a3;hpb=70e2471c1297f5ee5592918a736224841d8e4d77;p=nageru

diff --git a/nageru/quicksync_encoder.cpp b/nageru/quicksync_encoder.cpp
index 992bf8f..eb144ce 100644
--- a/nageru/quicksync_encoder.cpp
+++ b/nageru/quicksync_encoder.cpp
@@ -1435,8 +1435,8 @@ void QuickSyncEncoderImpl::release_gl_resources()
 	has_released_gl_resources = true;
 }
 
-QuickSyncEncoderImpl::QuickSyncEncoderImpl(const std::string &filename, ResourcePool *resource_pool, QSurface *surface, const string &va_display, int width, int height, AVOutputFormat *oformat, X264Encoder *x264_encoder, DiskSpaceEstimator *disk_space_estimator)
-	: current_storage_frame(0), resource_pool(resource_pool), surface(surface), x264_encoder(x264_encoder), frame_width(width), frame_height(height), disk_space_estimator(disk_space_estimator)
+QuickSyncEncoderImpl::QuickSyncEncoderImpl(const std::string &filename, ResourcePool *resource_pool, QSurface *surface, const string &va_display, int width, int height, AVOutputFormat *oformat, X264Encoder *http_encoder, X264Encoder *disk_encoder, DiskSpaceEstimator *disk_space_estimator)
+	: current_storage_frame(0), resource_pool(resource_pool), surface(surface), x264_http_encoder(http_encoder), x264_disk_encoder(disk_encoder), frame_width(width), frame_height(height), disk_space_estimator(disk_space_estimator)
 {
 	file_audio_encoder.reset(new AudioEncoder(AUDIO_OUTPUT_CODEC_NAME, DEFAULT_AUDIO_OUTPUT_BIT_RATE, oformat));
 	open_output_file(filename);
@@ -1448,9 +1448,14 @@ QuickSyncEncoderImpl::QuickSyncEncoderImpl(const std::string &filename, Resource
 	//print_input();
 
 	if (global_flags.x264_video_to_http || global_flags.x264_video_to_disk) {
-		assert(x264_encoder != nullptr);
+		assert(x264_http_encoder != nullptr);
 	} else {
-		assert(x264_encoder == nullptr);
+		assert(x264_http_encoder == nullptr);
+	}
+	if (global_flags.x264_separate_disk_encode) {
+		assert(x264_disk_encoder != nullptr);
+	} else {
+		assert(x264_disk_encoder == nullptr);
 	}
 
 	enable_zerocopy_if_possible();
@@ -1731,7 +1736,7 @@ void QuickSyncEncoderImpl::open_output_file(const std::string &filename)
 
 	string video_extradata;  // FIXME: See other comment about global headers.
 	if (global_flags.x264_video_to_disk) {
-		video_extradata = x264_encoder->get_global_headers();
+		video_extradata = x264_disk_encoder->get_global_headers();
 	}
 
 	current_file_mux_metrics.reset();
@@ -1747,7 +1752,7 @@ void QuickSyncEncoderImpl::open_output_file(const std::string &filename)
 	metric_current_file_start_time_seconds = get_timestamp_for_metrics();
 
 	if (global_flags.x264_video_to_disk) {
-		x264_encoder->add_mux(file_mux.get());
+		x264_disk_encoder->add_mux(file_mux.get());
 	}
 }
 
@@ -1910,7 +1915,10 @@ void QuickSyncEncoderImpl::pass_frame(QuickSyncEncoderImpl::PendingFrame frame,
 	if (global_flags.uncompressed_video_to_http) {
 		add_packet_for_uncompressed_frame(pts, duration, data);
 	} else if (global_flags.x264_video_to_http || global_flags.x264_video_to_disk) {
-		x264_encoder->add_frame(pts, duration, frame.ycbcr_coefficients, data, received_ts);
+		x264_http_encoder->add_frame(pts, duration, frame.ycbcr_coefficients, data, received_ts);
+	}
+	if (global_flags.x264_separate_disk_encode) {
+		x264_disk_encoder->add_frame(pts, duration, frame.ycbcr_coefficients, data, received_ts);
 	}
 
 	if (v4l_output != nullptr) {
@@ -2012,8 +2020,8 @@ void QuickSyncEncoderImpl::encode_frame(QuickSyncEncoderImpl::PendingFrame frame
 }
 
 // Proxy object.
-QuickSyncEncoder::QuickSyncEncoder(const std::string &filename, ResourcePool *resource_pool, QSurface *surface, const string &va_display, int width, int height, AVOutputFormat *oformat, X264Encoder *x264_encoder, DiskSpaceEstimator *disk_space_estimator)
-	: impl(new QuickSyncEncoderImpl(filename, resource_pool, surface, va_display, width, height, oformat, x264_encoder, disk_space_estimator)) {}
+QuickSyncEncoder::QuickSyncEncoder(const std::string &filename, ResourcePool *resource_pool, QSurface *surface, const string &va_display, int width, int height, AVOutputFormat *oformat, X264Encoder *http_encoder, X264Encoder *disk_encoder, DiskSpaceEstimator *disk_space_estimator)
+	: impl(new QuickSyncEncoderImpl(filename, resource_pool, surface, va_display, width, height, oformat, http_encoder, disk_encoder, disk_space_estimator)) {}
 
 // Must be defined here because unique_ptr<> destructor needs to know the impl.
 QuickSyncEncoder::~QuickSyncEncoder() {}