]> git.sesse.net Git - ffmpeg/blobdiff - libavcodec/atrac1.c
Add function to export EDGE_WIDTH from libavcodec.
[ffmpeg] / libavcodec / atrac1.c
index 746deab2ee841e75193d33057eea1d5588757c09..202ed8f5ea994c37c423ac8813659fc730b4efb2 100644 (file)
@@ -23,7 +23,7 @@
 /**
  * @file libavcodec/atrac1.c
  * Atrac 1 compatible decoder.
- * This decoder handles raw ATRAC1 data.
+ * This decoder handles raw ATRAC1 data and probably SDDS data.
  */
 
 /* Many thanks to Tim Craig for all the help! */
@@ -35,6 +35,7 @@
 #include "avcodec.h"
 #include "get_bits.h"
 #include "dsputil.h"
+#include "fft.h"
 
 #include "atrac.h"
 #include "atrac1data.h"
@@ -58,11 +59,11 @@ typedef struct {
     int                 log2_block_count[AT1_QMF_BANDS];    ///< log2 number of blocks in a band
     int                 num_bfus;                           ///< number of Block Floating Units
     float*              spectrum[2];
-    DECLARE_ALIGNED_16(float, spec1[AT1_SU_SAMPLES]);       ///< mdct buffer
-    DECLARE_ALIGNED_16(float, spec2[AT1_SU_SAMPLES]);       ///< mdct buffer
-    DECLARE_ALIGNED_16(float, fst_qmf_delay[46]);           ///< delay line for the 1st stacked QMF filter
-    DECLARE_ALIGNED_16(float, snd_qmf_delay[46]);           ///< delay line for the 2nd stacked QMF filter
-    DECLARE_ALIGNED_16(float, last_qmf_delay[256+23]);      ///< delay line for the last stacked QMF filter
+    DECLARE_ALIGNED(16, float, spec1)[AT1_SU_SAMPLES];     ///< mdct buffer
+    DECLARE_ALIGNED(16, float, spec2)[AT1_SU_SAMPLES];     ///< mdct buffer
+    DECLARE_ALIGNED(16, float, fst_qmf_delay)[46];         ///< delay line for the 1st stacked QMF filter
+    DECLARE_ALIGNED(16, float, snd_qmf_delay)[46];         ///< delay line for the 2nd stacked QMF filter
+    DECLARE_ALIGNED(16, float, last_qmf_delay)[256+23];    ///< delay line for the last stacked QMF filter
 } AT1SUCtx;
 
 /**
@@ -70,13 +71,13 @@ typedef struct {
  */
 typedef struct {
     AT1SUCtx            SUs[AT1_MAX_CHANNELS];              ///< channel sound unit
-    DECLARE_ALIGNED_16(float, spec[AT1_SU_SAMPLES]);        ///< the mdct spectrum buffer
+    DECLARE_ALIGNED(16, float, spec)[AT1_SU_SAMPLES];      ///< the mdct spectrum buffer
 
-    DECLARE_ALIGNED_16(float,  low[256]);
-    DECLARE_ALIGNED_16(float,  mid[256]);
-    DECLARE_ALIGNED_16(float, high[512]);
+    DECLARE_ALIGNED(16, float,  low)[256];
+    DECLARE_ALIGNED(16, float,  mid)[256];
+    DECLARE_ALIGNED(16, float, high)[512];
     float*              bands[3];
-    DECLARE_ALIGNED_16(float, out_samples[AT1_MAX_CHANNELS][AT1_SU_SAMPLES]);
+    DECLARE_ALIGNED(16, float, out_samples)[AT1_MAX_CHANNELS][AT1_SU_SAMPLES];
     FFTContext          mdct_ctx[3];
     int                 channels;
     DSPContext          dsp;
@@ -108,6 +109,9 @@ static int at1_imdct_block(AT1SUCtx* su, AT1Ctx *q)
     unsigned int start_pos, ref_pos = 0, pos = 0;
 
     for (band_num = 0; band_num < AT1_QMF_BANDS; band_num++) {
+        float *prev_buf;
+        int j;
+
         band_samples = samples_per_band[band_num];
         log2_block_count = su->log2_block_count[band_num];
 
@@ -115,42 +119,38 @@ static int at1_imdct_block(AT1SUCtx* su, AT1Ctx *q)
         /* 4 for short mode(low/middle bands) and 8 for short mode(high band)*/
         num_blocks = 1 << log2_block_count;
 
-        /* mdct block size in samples: 128 (long mode, low & mid bands), */
-        /* 256 (long mode, high band) and 32 (short mode, all bands) */
-        block_size = band_samples >> log2_block_count;
+        if (num_blocks == 1) {
+            /* mdct block size in samples: 128 (long mode, low & mid bands), */
+            /* 256 (long mode, high band) and 32 (short mode, all bands) */
+            block_size = band_samples >> log2_block_count;
 
-        /* calc transform size in bits according to the block_size_mode */
-        nbits = mdct_long_nbits[band_num] - log2_block_count;
+            /* calc transform size in bits according to the block_size_mode */
+            nbits = mdct_long_nbits[band_num] - log2_block_count;
 
-        if (nbits != 5 && nbits != 7 && nbits != 8)
-            return -1;
+            if (nbits != 5 && nbits != 7 && nbits != 8)
+                return -1;
+        } else {
+            block_size = 32;
+            nbits = 5;
+        }
 
-        if (num_blocks == 1) {
-            /* long blocks */
-            at1_imdct(q, &q->spec[pos], &su->spectrum[0][ref_pos], nbits, band_num);
-            pos += block_size; // move to the next mdct block in the spectrum
+        start_pos = 0;
+        prev_buf = &su->spectrum[1][ref_pos + band_samples - 16];
+        for (j=0; j < num_blocks; j++) {
+            at1_imdct(q, &q->spec[pos], &su->spectrum[0][ref_pos + start_pos], nbits, band_num);
 
-            /* overlap and window long blocks */
-            q->dsp.vector_fmul_window(q->bands[band_num], &su->spectrum[1][ref_pos + band_samples - 16],
-                                      &su->spectrum[0][ref_pos], ff_sine_32, 0, 16);
-            memcpy(q->bands[band_num] + 32, &su->spectrum[0][ref_pos + 16], 240 * sizeof(float));
-        } else {
-            /* short blocks */
-            float *prev_buf;
-            start_pos = 0;
-            prev_buf = &su->spectrum[1][ref_pos + band_samples - 16];
-            for (; num_blocks != 0; num_blocks--) {
-                at1_imdct(q, &q->spec[pos], &su->spectrum[0][ref_pos + start_pos], 5, band_num);
-
-                /* overlap and window between short blocks */
-                q->dsp.vector_fmul_window(&q->bands[band_num][start_pos], prev_buf,
-                                          &su->spectrum[0][ref_pos + start_pos], ff_sine_32, 0, 16);
-
-                prev_buf = &su->spectrum[0][ref_pos+start_pos + 16];
-                start_pos += 32; // use hardcoded block_size
-                pos += 32;
-            }
+            /* overlap and window */
+            q->dsp.vector_fmul_window(&q->bands[band_num][start_pos], prev_buf,
+                                      &su->spectrum[0][ref_pos + start_pos], ff_sine_32, 0, 16);
+
+            prev_buf = &su->spectrum[0][ref_pos+start_pos + 16];
+            start_pos += block_size;
+            pos += block_size;
         }
+
+        if (num_blocks == 1)
+            memcpy(q->bands[band_num] + 32, &su->spectrum[0][ref_pos + 16], 240 * sizeof(float));
+
         ref_pos += band_samples;
     }
 
@@ -252,7 +252,7 @@ static int at1_unpack_dequant(GetBitContext* gb, AT1SUCtx* su,
 }
 
 
-void at1_subband_synthesis(AT1Ctx *q, AT1SUCtx* su, float *pOut)
+static void at1_subband_synthesis(AT1Ctx *q, AT1SUCtx* su, float *pOut)
 {
     float temp[256];
     float iqmf_temp[512 + 46];
@@ -340,7 +340,7 @@ static av_cold int atrac1_decode_init(AVCodecContext *avctx)
     ff_mdct_init(&q->mdct_ctx[1], 8, 1, -1.0/ (1 << 15));
     ff_mdct_init(&q->mdct_ctx[2], 9, 1, -1.0/ (1 << 15));
 
-    ff_sine_window_init(ff_sine_32, 32);
+    ff_init_ff_sine_windows(5);
 
     atrac_generate_tables();
 
@@ -372,7 +372,7 @@ static av_cold int atrac1_decode_end(AVCodecContext * avctx) {
 
 AVCodec atrac1_decoder = {
     .name = "atrac1",
-    .type = CODEC_TYPE_AUDIO,
+    .type = AVMEDIA_TYPE_AUDIO,
     .id = CODEC_ID_ATRAC1,
     .priv_data_size = sizeof(AT1Ctx),
     .init = atrac1_decode_init,