X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Faudio_encoder.cpp;h=b236c002e7df44288651c7e6557afbb35844dcc7;hb=d0c6ebbc951adbfb04d5ee21c564c9e5e861a269;hp=7b72932f0b9ea1b4a6a4248001500f204326cf1d;hpb=1501c53153cb0daa846e4de7a73cfbfc797fd543;p=nageru

diff --git a/nageru/audio_encoder.cpp b/nageru/audio_encoder.cpp
index 7b72932..b236c00 100644
--- a/nageru/audio_encoder.cpp
+++ b/nageru/audio_encoder.cpp
@@ -32,7 +32,7 @@ AudioEncoder::AudioEncoder(const string &codec_name, int bit_rate, const AVOutpu
 	AVCodec *codec = avcodec_find_encoder_by_name(codec_name.c_str());
 	if (codec == nullptr) {
 		fprintf(stderr, "ERROR: Could not find codec '%s'\n", codec_name.c_str());
-		exit(1);
+		abort();
 	}
 
 	ctx = avcodec_alloc_context3(codec);
@@ -47,7 +47,7 @@ AudioEncoder::AudioEncoder(const string &codec_name, int bit_rate, const AVOutpu
 	}
 	if (avcodec_open2(ctx, codec, NULL) < 0) {
 		fprintf(stderr, "Could not open codec '%s'\n", codec_name.c_str());
-		exit(1);
+		abort();
 	}
 
 	resampler = swr_alloc_set_opts(nullptr,
@@ -61,12 +61,12 @@ AudioEncoder::AudioEncoder(const string &codec_name, int bit_rate, const AVOutpu
 	                               /*log_ctx=*/nullptr);
 	if (resampler == nullptr) {
 		fprintf(stderr, "Allocating resampler failed.\n");
-		exit(1);
+		abort();
 	}
 
 	if (swr_init(resampler) < 0) {
 		fprintf(stderr, "Could not open resample context.\n");
-		exit(1);
+		abort();
 	}
 
 	audio_frame = av_frame_alloc();
@@ -110,24 +110,25 @@ void AudioEncoder::encode_audio_one_frame(const float *audio, size_t num_samples
 {
 	audio_frame->pts = audio_pts;
 	audio_frame->nb_samples = num_samples;
+	audio_frame->channels = 2;
 	audio_frame->channel_layout = AV_CH_LAYOUT_STEREO;
 	audio_frame->format = ctx->sample_fmt;
 	audio_frame->sample_rate = OUTPUT_FREQUENCY;
 
 	if (av_samples_alloc(audio_frame->data, nullptr, 2, num_samples, ctx->sample_fmt, 0) < 0) {
-		fprintf(stderr, "Could not allocate %ld samples.\n", num_samples);
-		exit(1);
+		fprintf(stderr, "Could not allocate %zu samples.\n", num_samples);
+		abort();
 	}
 
 	if (swr_convert(resampler, audio_frame->data, num_samples, reinterpret_cast<const uint8_t **>(&audio), num_samples) < 0) {
 		fprintf(stderr, "Audio conversion failed.\n");
-		exit(1);
+		abort();
 	}
 
 	int err = avcodec_send_frame(ctx, audio_frame);
 	if (err < 0) {
 		fprintf(stderr, "avcodec_send_frame() failed with error %d\n", err);
-		exit(1);
+		abort();
 	}
 
 	for ( ;; ) {  // Termination condition within loop.
@@ -147,7 +148,7 @@ void AudioEncoder::encode_audio_one_frame(const float *audio, size_t num_samples
 			break;
 		} else {
 			fprintf(stderr, "avcodec_receive_frame() failed with error %d\n", err);
-			exit(1);
+			abort();
 		}
 	}
 
@@ -183,7 +184,7 @@ void AudioEncoder::encode_last_audio()
 				break;
 			} else {
 				fprintf(stderr, "avcodec_receive_frame() failed with error %d\n", err);
-				exit(1);
+				abort();
 			}
 		}
 	}