X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=quicksync_encoder.cpp;h=b288fa6f7e7f035de8de9354cc1b1e34fbf33b10;hb=35efc9e3149028e6c65dceb46f60490448dcbd9f;hp=88d0fb401d514ff7422e1239a007e7d206eda319;hpb=f932301ce7f23ef4c14dd0d4f52dee662d0b4ef6;p=nageru

diff --git a/quicksync_encoder.cpp b/quicksync_encoder.cpp
index 88d0fb4..b288fa6 100644
--- a/quicksync_encoder.cpp
+++ b/quicksync_encoder.cpp
@@ -118,14 +118,6 @@ static constexpr unsigned int MaxFrameNum = (2<<16);
 static constexpr unsigned int MaxPicOrderCntLsb = (2<<8);
 static constexpr unsigned int Log2MaxFrameNum = 16;
 static constexpr unsigned int Log2MaxPicOrderCntLsb = 8;
-static constexpr int rc_default_modes[] = {  // Priority list of modes.
-    VA_RC_VBR,
-    VA_RC_CQP,
-    VA_RC_VBR_CONSTRAINED,
-    VA_RC_CBR,
-    VA_RC_VCM,
-    VA_RC_NONE,
-};
 
 using namespace std;
 
@@ -714,26 +706,6 @@ void encoding2display_order(
 }
 
 
-static const char *rc_to_string(int rc_mode)
-{
-    switch (rc_mode) {
-    case VA_RC_NONE:
-        return "NONE";
-    case VA_RC_CBR:
-        return "CBR";
-    case VA_RC_VBR:
-        return "VBR";
-    case VA_RC_VCM:
-        return "VCM";
-    case VA_RC_CQP:
-        return "CQP";
-    case VA_RC_VBR_CONSTRAINED:
-        return "VBR_CONSTRAINED";
-    default:
-        return "Unknown";
-    }
-}
-
 void QuickSyncEncoderImpl::enable_zerocopy_if_possible()
 {
 	if (global_flags.x264_video_to_disk) {
@@ -879,23 +851,13 @@ int QuickSyncEncoderImpl::init_va(const string &va_display)
     }
     
     if (attrib[VAConfigAttribRateControl].value != VA_ATTRIB_NOT_SUPPORTED) {
-        int tmp = attrib[VAConfigAttribRateControl].value;
-
-        if (rc_mode == -1 || !(rc_mode & tmp))  {
-            if (rc_mode != -1) {
-                printf("Warning: Don't support the specified RateControl mode: %s!!!, switch to ", rc_to_string(rc_mode));
-            }
-
-            for (i = 0; i < sizeof(rc_default_modes) / sizeof(rc_default_modes[0]); i++) {
-                if (rc_default_modes[i] & tmp) {
-                    rc_mode = rc_default_modes[i];
-                    break;
-                }
-            }
+        if (!(attrib[VAConfigAttribRateControl].value & VA_RC_CQP)) {
+            fprintf(stderr, "ERROR: VA-API encoder does not support CQP mode.\n");
+            exit(1);
         }
 
         config_attrib[config_attrib_num].type = VAConfigAttribRateControl;
-        config_attrib[config_attrib_num].value = rc_mode;
+        config_attrib[config_attrib_num].value = VA_RC_CQP;
         config_attrib_num++;
     }
     
@@ -1719,6 +1681,7 @@ bool QuickSyncEncoderImpl::begin_frame(int64_t pts, int64_t duration, YCbCrLumaC
 
 void QuickSyncEncoderImpl::add_audio(int64_t pts, vector<float> audio)
 {
+	lock_guard<mutex> lock(file_audio_encoder_mutex);
 	assert(!is_shutdown);
 	file_audio_encoder->encode_audio(audio, pts + global_delay());
 }
@@ -1800,7 +1763,10 @@ void QuickSyncEncoderImpl::shutdown()
 	storage_thread.join();
 
 	// Encode any leftover audio in the queues, and also any delayed frames.
-	file_audio_encoder->encode_last_audio();
+	{
+		lock_guard<mutex> lock(file_audio_encoder_mutex);
+		file_audio_encoder->encode_last_audio();
+	}
 
 	if (!global_flags.x264_video_to_disk) {
 		release_encode();
@@ -1837,11 +1803,14 @@ void QuickSyncEncoderImpl::open_output_file(const std::string &filename)
 
 	current_file_mux_metrics.reset();
 
-	AVCodecParametersWithDeleter audio_codecpar = file_audio_encoder->get_codec_parameters();
-	file_mux.reset(new Mux(avctx, frame_width, frame_height, Mux::CODEC_H264, video_extradata, audio_codecpar.get(), TIMEBASE,
-		std::bind(&DiskSpaceEstimator::report_write, disk_space_estimator, filename, _1),
-		Mux::WRITE_BACKGROUND,
-		{ &current_file_mux_metrics, &total_mux_metrics }));
+	{
+		lock_guard<mutex> lock(file_audio_encoder_mutex);
+		AVCodecParametersWithDeleter audio_codecpar = file_audio_encoder->get_codec_parameters();
+		file_mux.reset(new Mux(avctx, frame_width, frame_height, Mux::CODEC_H264, video_extradata, audio_codecpar.get(), TIMEBASE,
+			std::bind(&DiskSpaceEstimator::report_write, disk_space_estimator, filename, _1),
+			Mux::WRITE_BACKGROUND,
+			{ &current_file_mux_metrics, &total_mux_metrics }));
+	}
 	metric_current_file_start_time_seconds = get_timestamp_for_metrics();
 
 	if (global_flags.x264_video_to_disk) {