alac: output in planar sample format

Avoids unneeded interleaving and allows for reusing the AVFrame output buffer
as the internal buffer for 24-bit and 32-bit sample size.

1 files changed, 19 insertions, 27 deletions

diff --git a/libavcodec/alac.c b/libavcodec/alac.c
index 268d592325..5f2c8a0492 100644
--- a/libavcodec/alac.c
+++ b/libavcodec/alac.c
@@ -316,6 +316,10 @@ static int alac_decode_frame(AVCodecContext *avctx, void *data,
         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
         return ret;
     }
+    if (alac->sample_size > 16) {
+        for (ch = 0; ch < alac->channels; ch++)
+            alac->output_samples_buffer[ch] = (int32_t *)alac->frame.data[ch];
+    }
 
     readsamplesize = alac->sample_size - alac->extra_bits + channels - 1;
     if (readsamplesize > MIN_CACHE_BITS) {
@@ -409,33 +413,18 @@ static int alac_decode_frame(AVCodecContext *avctx, void *data,
 
     switch(alac->sample_size) {
     case 16: {
-        int16_t *outbuffer = (int16_t *)alac->frame.data[0];
-        for (i = 0; i < alac->nb_samples; i++) {
-            *outbuffer++ = alac->output_samples_buffer[0][i];
-            if (channels == 2)
-                *outbuffer++ = alac->output_samples_buffer[1][i];
+        for (ch = 0; ch < alac->channels; ch++) {
+            int16_t *outbuffer = (int16_t *)alac->frame.data[ch];
+            for (i = 0; i < alac->nb_samples; i++)
+                *outbuffer++ = alac->output_samples_buffer[ch][i];
         }}
         break;
     case 24: {
-        int32_t *outbuffer = (int32_t *)alac->frame.data[0];
-        for (i = 0; i < alac->nb_samples; i++) {
-            *outbuffer++ = alac->output_samples_buffer[0][i] << 8;
-            if (channels == 2)
-                *outbuffer++ = alac->output_samples_buffer[1][i] << 8;
+        for (ch = 0; ch < alac->channels; ch++) {
+            for (i = 0; i < alac->nb_samples; i++)
+                alac->output_samples_buffer[ch][i] <<= 8;
         }}
         break;
-    case 32:
-        if (channels == 2) {
-            int32_t *outbuffer = (int32_t *)alac->frame.data[0];
-            for (i = 0; i < alac->nb_samples; i++) {
-                *outbuffer++ = alac->output_samples_buffer[0][i];
-                *outbuffer++ = alac->output_samples_buffer[1][i];
-            }
-        } else {
-            memcpy(alac->frame.data[0], alac->output_samples_buffer[0],
-                   alac->nb_samples * sizeof(*alac->output_samples_buffer[0]));
-        }
-        break;
     }
 
     if (avpkt->size * 8 - get_bits_count(&alac->gb) > 8)
@@ -455,7 +444,8 @@ static av_cold int alac_decode_close(AVCodecContext *avctx)
     int ch;
     for (ch = 0; ch < alac->channels; ch++) {
         av_freep(&alac->predict_error_buffer[ch]);
-        av_freep(&alac->output_samples_buffer[ch]);
+        if (alac->sample_size == 16)
+            av_freep(&alac->output_samples_buffer[ch]);
         av_freep(&alac->extra_bits_buffer[ch]);
     }
 
@@ -471,8 +461,10 @@ static int allocate_buffers(ALACContext *alac)
         FF_ALLOC_OR_GOTO(alac->avctx, alac->predict_error_buffer[ch],
                          buf_size, buf_alloc_fail);
 
-        FF_ALLOC_OR_GOTO(alac->avctx, alac->output_samples_buffer[ch],
-                         buf_size, buf_alloc_fail);
+        if (alac->sample_size == 16) {
+            FF_ALLOC_OR_GOTO(alac->avctx, alac->output_samples_buffer[ch],
+                             buf_size, buf_alloc_fail);
+        }
 
         FF_ALLOC_OR_GOTO(alac->avctx, alac->extra_bits_buffer[ch],
                          buf_size, buf_alloc_fail);
@@ -530,10 +522,10 @@ static av_cold int alac_decode_init(AVCodecContext * avctx)
     }
 
     switch (alac->sample_size) {
-    case 16: avctx->sample_fmt    = AV_SAMPLE_FMT_S16;
+    case 16: avctx->sample_fmt = AV_SAMPLE_FMT_S16P;
              break;
     case 24:
-    case 32: avctx->sample_fmt    = AV_SAMPLE_FMT_S32;
+    case 32: avctx->sample_fmt = AV_SAMPLE_FMT_S32P;
              break;
     default: av_log_ask_for_sample(avctx, "Sample depth %d is not supported.\n",
                                    alac->sample_size);