OSDN Git Service

sbc: Rename sbc_analyze_4b_xx to sbc_analyze_xx
[android-x86/external-bluetooth-sbc.git] / sbc / sbc.c
index 86399dd..ffdf05d 100644 (file)
--- a/sbc/sbc.c
+++ b/sbc/sbc.c
@@ -47,6 +47,7 @@
 #include "sbc_tables.h"
 
 #include "sbc.h"
+#include "sbc_private.h"
 #include "sbc_primitives.h"
 
 #define SBC_SYNCWORD   0x9C
@@ -78,7 +79,7 @@ struct sbc_frame {
        uint8_t joint;
 
        /* only the lower 4 bits of every element are to be used */
-       uint32_t scale_factor[2][8];
+       uint32_t SBC_ALIGNED scale_factor[2][8];
 
        /* raw integer subband samples in the frame */
        int32_t SBC_ALIGNED sb_sample_f[16][2][8];
@@ -160,7 +161,8 @@ static uint8_t sbc_crc8(const uint8_t *data, size_t len)
  * Takes a pointer to the frame in question, a pointer to the bits array and
  * the sampling frequency (as 2 bit integer)
  */
-static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
+static SBC_ALWAYS_INLINE void sbc_calculate_bits_internal(
+               const struct sbc_frame *frame, int (*bits)[8], int subbands)
 {
        uint8_t sf = frame->frequency;
 
@@ -171,17 +173,17 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                for (ch = 0; ch < frame->channels; ch++) {
                        max_bitneed = 0;
                        if (frame->allocation == SNR) {
-                               for (sb = 0; sb < frame->subbands; sb++) {
+                               for (sb = 0; sb < subbands; sb++) {
                                        bitneed[ch][sb] = frame->scale_factor[ch][sb];
                                        if (bitneed[ch][sb] > max_bitneed)
                                                max_bitneed = bitneed[ch][sb];
                                }
                        } else {
-                               for (sb = 0; sb < frame->subbands; sb++) {
+                               for (sb = 0; sb < subbands; sb++) {
                                        if (frame->scale_factor[ch][sb] == 0)
                                                bitneed[ch][sb] = -5;
                                        else {
-                                               if (frame->subbands == 4)
+                                               if (subbands == 4)
                                                        loudness = frame->scale_factor[ch][sb] - sbc_offset4[sf][sb];
                                                else
                                                        loudness = frame->scale_factor[ch][sb] - sbc_offset8[sf][sb];
@@ -202,7 +204,7 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                                bitslice--;
                                bitcount += slicecount;
                                slicecount = 0;
-                               for (sb = 0; sb < frame->subbands; sb++) {
+                               for (sb = 0; sb < subbands; sb++) {
                                        if ((bitneed[ch][sb] > bitslice + 1) && (bitneed[ch][sb] < bitslice + 16))
                                                slicecount++;
                                        else if (bitneed[ch][sb] == bitslice + 1)
@@ -215,7 +217,7 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                                bitslice--;
                        }
 
-                       for (sb = 0; sb < frame->subbands; sb++) {
+                       for (sb = 0; sb < subbands; sb++) {
                                if (bitneed[ch][sb] < bitslice + 2)
                                        bits[ch][sb] = 0;
                                else {
@@ -225,7 +227,8 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                                }
                        }
 
-                       for (sb = 0; bitcount < frame->bitpool && sb < frame->subbands; sb++) {
+                       for (sb = 0; bitcount < frame->bitpool &&
+                                                       sb < subbands; sb++) {
                                if ((bits[ch][sb] >= 2) && (bits[ch][sb] < 16)) {
                                        bits[ch][sb]++;
                                        bitcount++;
@@ -235,7 +238,8 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                                }
                        }
 
-                       for (sb = 0; bitcount < frame->bitpool && sb < frame->subbands; sb++) {
+                       for (sb = 0; bitcount < frame->bitpool &&
+                                                       sb < subbands; sb++) {
                                if (bits[ch][sb] < 16) {
                                        bits[ch][sb]++;
                                        bitcount++;
@@ -251,7 +255,7 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                max_bitneed = 0;
                if (frame->allocation == SNR) {
                        for (ch = 0; ch < 2; ch++) {
-                               for (sb = 0; sb < frame->subbands; sb++) {
+                               for (sb = 0; sb < subbands; sb++) {
                                        bitneed[ch][sb] = frame->scale_factor[ch][sb];
                                        if (bitneed[ch][sb] > max_bitneed)
                                                max_bitneed = bitneed[ch][sb];
@@ -259,11 +263,11 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                        }
                } else {
                        for (ch = 0; ch < 2; ch++) {
-                               for (sb = 0; sb < frame->subbands; sb++) {
+                               for (sb = 0; sb < subbands; sb++) {
                                        if (frame->scale_factor[ch][sb] == 0)
                                                bitneed[ch][sb] = -5;
                                        else {
-                                               if (frame->subbands == 4)
+                                               if (subbands == 4)
                                                        loudness = frame->scale_factor[ch][sb] - sbc_offset4[sf][sb];
                                                else
                                                        loudness = frame->scale_factor[ch][sb] - sbc_offset8[sf][sb];
@@ -286,7 +290,7 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                        bitcount += slicecount;
                        slicecount = 0;
                        for (ch = 0; ch < 2; ch++) {
-                               for (sb = 0; sb < frame->subbands; sb++) {
+                               for (sb = 0; sb < subbands; sb++) {
                                        if ((bitneed[ch][sb] > bitslice + 1) && (bitneed[ch][sb] < bitslice + 16))
                                                slicecount++;
                                        else if (bitneed[ch][sb] == bitslice + 1)
@@ -301,7 +305,7 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                }
 
                for (ch = 0; ch < 2; ch++) {
-                       for (sb = 0; sb < frame->subbands; sb++) {
+                       for (sb = 0; sb < subbands; sb++) {
                                if (bitneed[ch][sb] < bitslice + 2) {
                                        bits[ch][sb] = 0;
                                } else {
@@ -325,7 +329,8 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                        if (ch == 1) {
                                ch = 0;
                                sb++;
-                               if (sb >= frame->subbands) break;
+                               if (sb >= subbands)
+                                       break;
                        } else
                                ch = 1;
                }
@@ -340,7 +345,8 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
                        if (ch == 1) {
                                ch = 0;
                                sb++;
-                               if (sb >= frame->subbands) break;
+                               if (sb >= subbands)
+                                       break;
                        } else
                                ch = 1;
                }
@@ -349,6 +355,14 @@ static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
 
 }
 
+static void sbc_calculate_bits(const struct sbc_frame *frame, int (*bits)[8])
+{
+       if (frame->subbands == 4)
+               sbc_calculate_bits_internal(frame, bits, 4);
+       else
+               sbc_calculate_bits_internal(frame, bits, 8);
+}
+
 /*
  * Unpacks a SBC frame at the beginning of the stream in data,
  * which has at most len bytes into frame.
@@ -370,7 +384,7 @@ static int sbc_unpack_frame(const uint8_t *data, struct sbc_frame *frame,
        int crc_pos = 0;
        int32_t temp;
 
-       int audio_sample;
+       uint32_t audio_sample;
        int ch, sb, blk, bit;   /* channel, subband, block and bit standard
                                   counters */
        int bits[2][8];         /* bits distribution */
@@ -480,23 +494,30 @@ static int sbc_unpack_frame(const uint8_t *data, struct sbc_frame *frame,
        for (blk = 0; blk < frame->blocks; blk++) {
                for (ch = 0; ch < frame->channels; ch++) {
                        for (sb = 0; sb < frame->subbands; sb++) {
-                               if (levels[ch][sb] > 0) {
-                                       audio_sample = 0;
-                                       for (bit = 0; bit < bits[ch][sb]; bit++) {
-                                               if (consumed > len * 8)
-                                                       return -1;
+                               uint32_t shift;
 
-                                               if ((data[consumed >> 3] >> (7 - (consumed & 0x7))) & 0x01)
-                                                       audio_sample |= 1 << (bits[ch][sb] - bit - 1);
+                               if (levels[ch][sb] == 0) {
+                                       frame->sb_sample[blk][ch][sb] = 0;
+                                       continue;
+                               }
 
-                                               consumed++;
-                                       }
+                               shift = frame->scale_factor[ch][sb] +
+                                               1 + SBCDEC_FIXED_EXTRA_BITS;
 
-                                       frame->sb_sample[blk][ch][sb] =
-                                               (((audio_sample << 1) | 1) << frame->scale_factor[ch][sb]) /
-                                               levels[ch][sb] - (1 << frame->scale_factor[ch][sb]);
-                               } else
-                                       frame->sb_sample[blk][ch][sb] = 0;
+                               audio_sample = 0;
+                               for (bit = 0; bit < bits[ch][sb]; bit++) {
+                                       if (consumed > len * 8)
+                                               return -1;
+
+                                       if ((data[consumed >> 3] >> (7 - (consumed & 0x7))) & 0x01)
+                                               audio_sample |= 1 << (bits[ch][sb] - bit - 1);
+
+                                       consumed++;
+                               }
+
+                               frame->sb_sample[blk][ch][sb] = (int32_t)
+                                       (((((uint64_t) audio_sample << 1) | 1) << shift) /
+                                       levels[ch][sb]) - (1 << shift);
                        }
                }
        }
@@ -667,30 +688,32 @@ static int sbc_analyze_audio(struct sbc_encoder_state *state,
        switch (frame->subbands) {
        case 4:
                for (ch = 0; ch < frame->channels; ch++) {
-                       x = &state->X[ch][state->position - 16 +
-                                                       frame->blocks * 4];
-                       for (blk = 0; blk < frame->blocks; blk += 4) {
-                               state->sbc_analyze_4b_4s(
-                                       x,
+                       x = &state->X[ch][state->position - 4 *
+                                       state->increment + frame->blocks * 4];
+                       for (blk = 0; blk < frame->blocks;
+                                               blk += state->increment) {
+                               state->sbc_analyze_4s(
+                                       state, x,
                                        frame->sb_sample_f[blk][ch],
                                        frame->sb_sample_f[blk + 1][ch] -
                                        frame->sb_sample_f[blk][ch]);
-                               x -= 16;
+                               x -= 4 * state->increment;
                        }
                }
                return frame->blocks * 4;
 
        case 8:
                for (ch = 0; ch < frame->channels; ch++) {
-                       x = &state->X[ch][state->position - 32 +
-                                                       frame->blocks * 8];
-                       for (blk = 0; blk < frame->blocks; blk += 4) {
-                               state->sbc_analyze_4b_8s(
-                                       x,
+                       x = &state->X[ch][state->position - 8 *
+                                       state->increment + frame->blocks * 8];
+                       for (blk = 0; blk < frame->blocks;
+                                               blk += state->increment) {
+                               state->sbc_analyze_8s(
+                                       state, x,
                                        frame->sb_sample_f[blk][ch],
                                        frame->sb_sample_f[blk + 1][ch] -
                                        frame->sb_sample_f[blk][ch]);
-                               x -= 32;
+                               x -= 8 * state->increment;
                        }
                }
                return frame->blocks * 8;
@@ -743,9 +766,10 @@ static int sbc_analyze_audio(struct sbc_encoder_state *state,
  * -99 not implemented
  */
 
-static SBC_ALWAYS_INLINE int sbc_pack_frame_internal(uint8_t *data,
+static SBC_ALWAYS_INLINE ssize_t sbc_pack_frame_internal(uint8_t *data,
                                        struct sbc_frame *frame, size_t len,
-                                       int frame_subbands, int frame_channels)
+                                       int frame_subbands, int frame_channels,
+                                       int joint)
 {
        /* Bitstream writer starts from the fourth byte */
        uint8_t *data_ptr = data + 4;
@@ -802,63 +826,6 @@ static SBC_ALWAYS_INLINE int sbc_pack_frame_internal(uint8_t *data,
        crc_pos = 16;
 
        if (frame->mode == JOINT_STEREO) {
-               /* like frame->sb_sample but joint stereo */
-               int32_t sb_sample_j[16][2];
-               /* scalefactor and scale_factor in joint case */
-               uint32_t scalefactor_j[2];
-               uint8_t scale_factor_j[2];
-
-               uint8_t joint = 0;
-               frame->joint = 0;
-
-               for (sb = 0; sb < frame_subbands - 1; sb++) {
-                       scale_factor_j[0] = 0;
-                       scalefactor_j[0] = 2 << SCALE_OUT_BITS;
-                       scale_factor_j[1] = 0;
-                       scalefactor_j[1] = 2 << SCALE_OUT_BITS;
-
-                       for (blk = 0; blk < frame->blocks; blk++) {
-                               uint32_t tmp;
-                               /* Calculate joint stereo signal */
-                               sb_sample_j[blk][0] =
-                                       ASR(frame->sb_sample_f[blk][0][sb], 1) +
-                                       ASR(frame->sb_sample_f[blk][1][sb], 1);
-                               sb_sample_j[blk][1] =
-                                       ASR(frame->sb_sample_f[blk][0][sb], 1) -
-                                       ASR(frame->sb_sample_f[blk][1][sb], 1);
-
-                               /* calculate scale_factor_j and scalefactor_j for joint case */
-                               tmp = fabs(sb_sample_j[blk][0]);
-                               while (scalefactor_j[0] < tmp) {
-                                       scale_factor_j[0]++;
-                                       scalefactor_j[0] *= 2;
-                               }
-                               tmp = fabs(sb_sample_j[blk][1]);
-                               while (scalefactor_j[1] < tmp) {
-                                       scale_factor_j[1]++;
-                                       scalefactor_j[1] *= 2;
-                               }
-                       }
-
-                       /* decide whether to join this subband */
-                       if ((frame->scale_factor[0][sb] +
-                                       frame->scale_factor[1][sb]) >
-                                       (scale_factor_j[0] +
-                                       scale_factor_j[1])) {
-                               /* use joint stereo for this subband */
-                               joint |= 1 << (frame_subbands - 1 - sb);
-                               frame->joint |= 1 << sb;
-                               frame->scale_factor[0][sb] = scale_factor_j[0];
-                               frame->scale_factor[1][sb] = scale_factor_j[1];
-                               for (blk = 0; blk < frame->blocks; blk++) {
-                                       frame->sb_sample_f[blk][0][sb] =
-                                                       sb_sample_j[blk][0];
-                                       frame->sb_sample_f[blk][1][sb] =
-                                                       sb_sample_j[blk][1];
-                               }
-                       }
-               }
-
                PUT_BITS(data_ptr, bits_cache, bits_count,
                        joint, frame_subbands);
                crc_header[crc_pos >> 3] = joint;
@@ -916,18 +883,23 @@ static SBC_ALWAYS_INLINE int sbc_pack_frame_internal(uint8_t *data,
        return data_ptr - data;
 }
 
-static int sbc_pack_frame(uint8_t *data, struct sbc_frame *frame, size_t len)
+static ssize_t sbc_pack_frame(uint8_t *data, struct sbc_frame *frame, size_t len,
+                                                               int joint)
 {
        if (frame->subbands == 4) {
                if (frame->channels == 1)
-                       return sbc_pack_frame_internal(data, frame, len, 4, 1);
+                       return sbc_pack_frame_internal(
+                               data, frame, len, 4, 1, joint);
                else
-                       return sbc_pack_frame_internal(data, frame, len, 4, 2);
+                       return sbc_pack_frame_internal(
+                               data, frame, len, 4, 2, joint);
        } else {
                if (frame->channels == 1)
-                       return sbc_pack_frame_internal(data, frame, len, 8, 1);
+                       return sbc_pack_frame_internal(
+                               data, frame, len, 8, 1, joint);
                else
-                       return sbc_pack_frame_internal(data, frame, len, 8, 2);
+                       return sbc_pack_frame_internal(
+                               data, frame, len, 8, 2, joint);
        }
 }
 
@@ -936,6 +908,7 @@ static void sbc_encoder_init(struct sbc_encoder_state *state,
 {
        memset(&state->X, 0, sizeof(state->X));
        state->position = (SBC_X_BUFFER_SIZE - frame->subbands * 9) & ~7;
+       state->increment = 4;
 
        sbc_init_primitives(state);
 }
@@ -963,7 +936,7 @@ static void sbc_set_defaults(sbc_t *sbc, unsigned long flags)
 #endif
 }
 
-int sbc_init(sbc_t *sbc, unsigned long flags)
+SBC_EXPORT int sbc_init(sbc_t *sbc, unsigned long flags)
 {
        if (!sbc)
                return -EIO;
@@ -984,12 +957,12 @@ int sbc_init(sbc_t *sbc, unsigned long flags)
        return 0;
 }
 
-ssize_t sbc_parse(sbc_t *sbc, const void *input, size_t input_len)
+SBC_EXPORT ssize_t sbc_parse(sbc_t *sbc, const void *input, size_t input_len)
 {
        return sbc_decode(sbc, input, input_len, NULL, 0, NULL);
 }
 
-ssize_t sbc_decode(sbc_t *sbc, const void *input, size_t input_len,
+SBC_EXPORT ssize_t sbc_decode(sbc_t *sbc, const void *input, size_t input_len,
                        void *output, size_t output_len, size_t *written)
 {
        struct sbc_priv *priv;
@@ -1016,6 +989,9 @@ ssize_t sbc_decode(sbc_t *sbc, const void *input, size_t input_len,
 
                priv->frame.codesize = sbc_get_codesize(sbc);
                priv->frame.length = framelen;
+       } else if (priv->frame.bitpool != sbc->bitpool) {
+               priv->frame.length = framelen;
+               sbc->bitpool = priv->frame.bitpool;
        }
 
        if (!output)
@@ -1055,11 +1031,12 @@ ssize_t sbc_decode(sbc_t *sbc, const void *input, size_t input_len,
        return framelen;
 }
 
-ssize_t sbc_encode(sbc_t *sbc, const void *input, size_t input_len,
-                       void *output, size_t output_len, size_t *written)
+SBC_EXPORT ssize_t sbc_encode(sbc_t *sbc, const void *input, size_t input_len,
+                       void *output, size_t output_len, ssize_t *written)
 {
        struct sbc_priv *priv;
-       int framelen, samples;
+       int samples;
+       ssize_t framelen;
        int (*sbc_enc_process_input)(int position,
                        const uint8_t *pcm, int16_t X[2][SBC_X_BUFFER_SIZE],
                        int nsamples, int nchannels);
@@ -1087,6 +1064,9 @@ ssize_t sbc_encode(sbc_t *sbc, const void *input, size_t input_len,
 
                sbc_encoder_init(&priv->enc_state, &priv->frame);
                priv->init = 1;
+       } else if (priv->frame.bitpool != sbc->bitpool) {
+               priv->frame.length = sbc_get_frame_length(sbc);
+               priv->frame.bitpool = sbc->bitpool;
        }
 
        /* input must be large enough to encode a complete frame */
@@ -1121,11 +1101,18 @@ ssize_t sbc_encode(sbc_t *sbc, const void *input, size_t input_len,
 
        samples = sbc_analyze_audio(&priv->enc_state, &priv->frame);
 
-       priv->enc_state.sbc_calc_scalefactors(
-               priv->frame.sb_sample_f, priv->frame.scale_factor,
-               priv->frame.blocks, priv->frame.channels, priv->frame.subbands);
-
-       framelen = sbc_pack_frame(output, &priv->frame, output_len);
+       if (priv->frame.mode == JOINT_STEREO) {
+               int j = priv->enc_state.sbc_calc_scalefactors_j(
+                       priv->frame.sb_sample_f, priv->frame.scale_factor,
+                       priv->frame.blocks, priv->frame.subbands);
+               framelen = sbc_pack_frame(output, &priv->frame, output_len, j);
+       } else {
+               priv->enc_state.sbc_calc_scalefactors(
+                       priv->frame.sb_sample_f, priv->frame.scale_factor,
+                       priv->frame.blocks, priv->frame.channels,
+                       priv->frame.subbands);
+               framelen = sbc_pack_frame(output, &priv->frame, output_len, 0);
+       }
 
        if (written)
                *written = framelen;
@@ -1133,7 +1120,7 @@ ssize_t sbc_encode(sbc_t *sbc, const void *input, size_t input_len,
        return samples * priv->frame.channels * 2;
 }
 
-void sbc_finish(sbc_t *sbc)
+SBC_EXPORT void sbc_finish(sbc_t *sbc)
 {
        if (!sbc)
                return;
@@ -1143,14 +1130,14 @@ void sbc_finish(sbc_t *sbc)
        memset(sbc, 0, sizeof(sbc_t));
 }
 
-size_t sbc_get_frame_length(sbc_t *sbc)
+SBC_EXPORT size_t sbc_get_frame_length(sbc_t *sbc)
 {
        int ret;
        uint8_t subbands, channels, blocks, joint, bitpool;
        struct sbc_priv *priv;
 
        priv = sbc->priv;
-       if (priv->init)
+       if (priv->init && priv->frame.bitpool == sbc->bitpool)
                return priv->frame.length;
 
        subbands = sbc->subbands ? 8 : 4;
@@ -1169,7 +1156,7 @@ size_t sbc_get_frame_length(sbc_t *sbc)
        return ret;
 }
 
-unsigned sbc_get_frame_duration(sbc_t *sbc)
+SBC_EXPORT unsigned sbc_get_frame_duration(sbc_t *sbc)
 {
        uint8_t subbands, blocks;
        uint16_t frequency;
@@ -1207,7 +1194,7 @@ unsigned sbc_get_frame_duration(sbc_t *sbc)
        return (1000000 * blocks * subbands) / frequency;
 }
 
-size_t sbc_get_codesize(sbc_t *sbc)
+SBC_EXPORT size_t sbc_get_codesize(sbc_t *sbc)
 {
        uint16_t subbands, channels, blocks;
        struct sbc_priv *priv;
@@ -1226,7 +1213,7 @@ size_t sbc_get_codesize(sbc_t *sbc)
        return subbands * blocks * channels * 2;
 }
 
-const char *sbc_get_implementation_info(sbc_t *sbc)
+SBC_EXPORT const char *sbc_get_implementation_info(sbc_t *sbc)
 {
        struct sbc_priv *priv;
 
@@ -1240,7 +1227,7 @@ const char *sbc_get_implementation_info(sbc_t *sbc)
        return priv->enc_state.implementation_info;
 }
 
-int sbc_reinit(sbc_t *sbc, unsigned long flags)
+SBC_EXPORT int sbc_reinit(sbc_t *sbc, unsigned long flags)
 {
        struct sbc_priv *priv;