Merge remote-tracking branch 'shariman/wmall'
[ffmpeg.git] / libavcodec / wmalosslessdec.c
index b108807929eacfdd753436e19cf68ffa38bfc793..2b32bccba10a4503b26be305796d645a35d1913a 100644 (file)
@@ -222,7 +222,7 @@ typedef struct WmallDecodeCtx {
     WmallChannelCtx channel[WMALL_MAX_CHANNELS];    ///< per channel data
 
     // WMA lossless
-    
+
     uint8_t do_arith_coding;
     uint8_t do_ac_filter;
     uint8_t do_inter_ch_decorr;
@@ -245,11 +245,11 @@ typedef struct WmallDecodeCtx {
     int quant_stepsize;
 
     struct {
-       int order;
-       int scaling;
-       int coefsend;
-       int bitsend;
-       int16_t coefs[256];
+        int order;
+        int scaling;
+        int coefsend;
+        int bitsend;
+        int16_t coefs[256];
     int16_t lms_prevvalues[512];    // FIXME: see above
     int16_t lms_updates[512];   // and here too
     int recent;
@@ -514,12 +514,12 @@ static int decode_tilehdr(WmallDecodeCtx *s)
         for (c = 0; c < s->num_channels; c++) {
             if (num_samples[c] == min_channel_len) {
                 if (fixed_channel_layout || channels_for_cur_subframe == 1 ||
-                   (min_channel_len == s->samples_per_frame - s->min_samples_per_subframe)) {
+                    (min_channel_len == s->samples_per_frame - s->min_samples_per_subframe)) {
                     contains_subframe[c] = 1;
-               }                   
+                }
                 else {
                     contains_subframe[c] = get_bits1(&s->gb);
-               }
+                }
             } else
                 contains_subframe[c] = 0;
         }
@@ -544,7 +544,7 @@ static int decode_tilehdr(WmallDecodeCtx *s)
                 if (num_samples[c] > s->samples_per_frame) {
                     av_log(s->avctx, AV_LOG_ERROR, "broken frame: "
                            "channel len(%d) > samples_per_frame(%d)\n",
-                          num_samples[c], s->samples_per_frame);
+                           num_samples[c], s->samples_per_frame);
                     return AVERROR_INVALIDDATA;
                 }
             } else if (num_samples[c] <= min_channel_len) {
@@ -574,7 +574,7 @@ static int my_log2(unsigned int i)
 {
     unsigned int iLog2 = 0;
     while ((i >> iLog2) > 1)
-       iLog2++;
+        iLog2++;
     return iLog2;
 }
 
@@ -589,7 +589,7 @@ static void decode_ac_filter(WmallDecodeCtx *s)
     s->acfilter_scaling = get_bits(&s->gb, 4);
 
     for(i = 0; i < s->acfilter_order; i++) {
-       s->acfilter_coeffs[i] = get_bits(&s->gb, s->acfilter_scaling) + 1;
+        s->acfilter_coeffs[i] = get_bits(&s->gb, s->acfilter_scaling) + 1;
     }
 }
 
@@ -602,26 +602,26 @@ static void decode_mclms(WmallDecodeCtx *s)
     s->mclms_order = (get_bits(&s->gb, 4) + 1) * 2;
     s->mclms_scaling = get_bits(&s->gb, 4);
     if(get_bits1(&s->gb)) {
-       // mclms_send_coef
-       int i;
-       int send_coef_bits;
-       int cbits = av_log2(s->mclms_scaling + 1);
-       assert(cbits == my_log2(s->mclms_scaling + 1));
-       if(1 << cbits < s->mclms_scaling + 1)
-           cbits++;
-
-       send_coef_bits = (cbits ? get_bits(&s->gb, cbits) : 0) + 2;
-
-       for(i = 0; i < s->mclms_order * s->num_channels * s->num_channels; i++) {
-           s->mclms_coeffs[i] = get_bits(&s->gb, send_coef_bits);
-       }           
-
-       for(i = 0; i < s->num_channels; i++) {
-           int c;
-           for(c = 0; c < i; c++) {
-               s->mclms_coeffs_cur[i * s->num_channels + c] = get_bits(&s->gb, send_coef_bits);
-           }
-       }
+        // mclms_send_coef
+        int i;
+        int send_coef_bits;
+        int cbits = av_log2(s->mclms_scaling + 1);
+        assert(cbits == my_log2(s->mclms_scaling + 1));
+        if(1 << cbits < s->mclms_scaling + 1)
+            cbits++;
+
+        send_coef_bits = (cbits ? get_bits(&s->gb, cbits) : 0) + 2;
+
+        for(i = 0; i < s->mclms_order * s->num_channels * s->num_channels; i++) {
+            s->mclms_coeffs[i] = get_bits(&s->gb, send_coef_bits);
+        }
+
+        for(i = 0; i < s->num_channels; i++) {
+            int c;
+            for(c = 0; c < i; c++) {
+                s->mclms_coeffs_cur[i * s->num_channels + c] = get_bits(&s->gb, send_coef_bits);
+            }
+        }
     }
 }
 
@@ -635,36 +635,36 @@ static void decode_cdlms(WmallDecodeCtx *s)
     int cdlms_send_coef = get_bits1(&s->gb);
 
     for(c = 0; c < s->num_channels; c++) {
-       s->cdlms_ttl[c] = get_bits(&s->gb, 3) + 1;
-       for(i = 0; i < s->cdlms_ttl[c]; i++) {
-           s->cdlms[c][i].order = (get_bits(&s->gb, 7) + 1) * 8;
-       }
-
-       for(i = 0; i < s->cdlms_ttl[c]; i++) {
-           s->cdlms[c][i].scaling = get_bits(&s->gb, 4);
-       }
-
-       if(cdlms_send_coef) {
-           for(i = 0; i < s->cdlms_ttl[c]; i++) {
-               int cbits, shift_l, shift_r, j;
-               cbits = av_log2(s->cdlms[c][i].order);
-               if(1 << cbits < s->cdlms[c][i].order)
-                   cbits++;
-               s->cdlms[c][i].coefsend = get_bits(&s->gb, cbits) + 1;
-
-               cbits = av_log2(s->cdlms[c][i].scaling + 1);
-               if(1 << cbits < s->cdlms[c][i].scaling + 1)
-                   cbits++;
-               
-               s->cdlms[c][i].bitsend = get_bits(&s->gb, cbits) + 2;
-               shift_l = 32 - s->cdlms[c][i].bitsend;
-               shift_r = 32 - 2 - s->cdlms[c][i].scaling;
-               for(j = 0; j < s->cdlms[c][i].coefsend; j++) {
-                   s->cdlms[c][i].coefs[j] = 
-                       (get_bits(&s->gb, s->cdlms[c][i].bitsend) << shift_l) >> shift_r;
-               }
-           }
-       }
+        s->cdlms_ttl[c] = get_bits(&s->gb, 3) + 1;
+        for(i = 0; i < s->cdlms_ttl[c]; i++) {
+            s->cdlms[c][i].order = (get_bits(&s->gb, 7) + 1) * 8;
+        }
+
+        for(i = 0; i < s->cdlms_ttl[c]; i++) {
+            s->cdlms[c][i].scaling = get_bits(&s->gb, 4);
+        }
+
+        if(cdlms_send_coef) {
+            for(i = 0; i < s->cdlms_ttl[c]; i++) {
+                int cbits, shift_l, shift_r, j;
+                cbits = av_log2(s->cdlms[c][i].order);
+                if(1 << cbits < s->cdlms[c][i].order)
+                    cbits++;
+                s->cdlms[c][i].coefsend = get_bits(&s->gb, cbits) + 1;
+
+                cbits = av_log2(s->cdlms[c][i].scaling + 1);
+                if(1 << cbits < s->cdlms[c][i].scaling + 1)
+                    cbits++;
+
+                s->cdlms[c][i].bitsend = get_bits(&s->gb, cbits) + 2;
+                shift_l = 32 - s->cdlms[c][i].bitsend;
+                shift_r = 32 - 2 - s->cdlms[c][i].scaling;
+                for(j = 0; j < s->cdlms[c][i].coefsend; j++) {
+                    s->cdlms[c][i].coefs[j] =
+                        (get_bits(&s->gb, s->cdlms[c][i].bitsend) << shift_l) >> shift_r;
+                }
+            }
+        }
     }
 }
 
@@ -677,47 +677,47 @@ static int decode_channel_residues(WmallDecodeCtx *s, int ch, int tile_size)
     unsigned int ave_mean;
     s->transient[ch] = get_bits1(&s->gb);
     if(s->transient[ch]) {
-           s->transient_pos[ch] = get_bits(&s->gb, av_log2(tile_size));
+            s->transient_pos[ch] = get_bits(&s->gb, av_log2(tile_size));
         if (s->transient_pos[ch])
-               s->transient[ch] = 0;
-           s->channel[ch].transient_counter =
-               FFMAX(s->channel[ch].transient_counter, s->samples_per_frame / 2);
-       } else if (s->channel[ch].transient_counter)
-           s->transient[ch] = 1;
+                s->transient[ch] = 0;
+            s->channel[ch].transient_counter =
+                FFMAX(s->channel[ch].transient_counter, s->samples_per_frame / 2);
+        } else if (s->channel[ch].transient_counter)
+            s->transient[ch] = 1;
 
     if(s->seekable_tile) {
-       ave_mean = get_bits(&s->gb, s->bits_per_sample);
-       s->ave_sum[ch] = ave_mean << (s->movave_scaling + 1);
-//     s->ave_sum[ch] *= 2;
+        ave_mean = get_bits(&s->gb, s->bits_per_sample);
+        s->ave_sum[ch] = ave_mean << (s->movave_scaling + 1);
+//        s->ave_sum[ch] *= 2;
     }
 
     if(s->seekable_tile) {
-       if(s->do_inter_ch_decorr)
-           s->channel_residues[ch][0] = get_sbits(&s->gb, s->bits_per_sample + 1);
-    else
-           s->channel_residues[ch][0] = get_sbits(&s->gb, s->bits_per_sample);
-       i++;
+        if(s->do_inter_ch_decorr)
+            s->channel_residues[ch][0] = get_sbits(&s->gb, s->bits_per_sample + 1);
+        else
+            s->channel_residues[ch][0] = get_sbits(&s->gb, s->bits_per_sample);
+        i++;
     }
     //av_log(0, 0, "%8d: ", num_logged_tiles++);
     for(; i < tile_size; i++) {
-       int quo = 0, rem, rem_bits, residue;
-       while(get_bits1(&s->gb))
-           quo++;
-       if(quo >= 32)
-           quo += get_bits_long(&s->gb, get_bits(&s->gb, 5) + 1);
-
-               ave_mean = (s->ave_sum[ch] + (1 << s->movave_scaling)) >> (s->movave_scaling + 1);
-       rem_bits = av_ceil_log2(ave_mean);
-       rem = rem_bits ? get_bits(&s->gb, rem_bits) : 0;
-       residue = (quo << rem_bits) + rem;
-
-       s->ave_sum[ch] = residue + s->ave_sum[ch] - (s->ave_sum[ch] >> s->movave_scaling);
-
-       if(residue & 1)
-           residue = -(residue >> 1) - 1;
-       else
-           residue = residue >> 1;
-       s->channel_residues[ch][i] = residue;
+        int quo = 0, rem, rem_bits, residue;
+        while(get_bits1(&s->gb))
+            quo++;
+        if(quo >= 32)
+            quo += get_bits_long(&s->gb, get_bits(&s->gb, 5) + 1);
+
+               ave_mean = (s->ave_sum[ch] + (1 << s->movave_scaling)) >> (s->movave_scaling + 1);
+        rem_bits = av_ceil_log2(ave_mean);
+        rem = rem_bits ? get_bits(&s->gb, rem_bits) : 0;
+        residue = (quo << rem_bits) + rem;
+
+        s->ave_sum[ch] = residue + s->ave_sum[ch] - (s->ave_sum[ch] >> s->movave_scaling);
+
+        if(residue & 1)
+            residue = -(residue >> 1) - 1;
+        else
+            residue = residue >> 1;
+        s->channel_residues[ch][i] = residue;
     }
     dump_int_buffer(s->channel_residues[ch], 4, tile_size, 16);
 
@@ -738,9 +738,9 @@ decode_lpc(WmallDecodeCtx *s)
     s->lpc_intbits = get_bits(&s->gb, 3) + 1;
     cbits = s->lpc_scaling + s->lpc_intbits;
     for(ch = 0; ch < s->num_channels; ch++) {
-       for(i = 0; i < s->lpc_order; i++) {
-           s->lpc_coefs[ch][i] = get_sbits(&s->gb, cbits);
-       }
+        for(i = 0; i < s->lpc_order; i++) {
+            s->lpc_coefs[ch][i] = get_sbits(&s->gb, cbits);
+        }
     }
 }
 
@@ -1057,73 +1057,73 @@ static int decode_subframe(WmallDecodeCtx *s)
     if(s->seekable_tile) {
         clear_codec_buffers(s);
 
-       s->do_arith_coding    = get_bits1(&s->gb);
-       if(s->do_arith_coding) {
-           dprintf(s->avctx, "do_arith_coding == 1");
-           abort();
-       }
-       s->do_ac_filter       = get_bits1(&s->gb);
-       s->do_inter_ch_decorr = get_bits1(&s->gb);
-       s->do_mclms           = get_bits1(&s->gb);
-       
-       if(s->do_ac_filter)
-           decode_ac_filter(s);
-
-       if(s->do_mclms)
-           decode_mclms(s);
-
-       decode_cdlms(s);
-       s->movave_scaling = get_bits(&s->gb, 3);
-       s->quant_stepsize = get_bits(&s->gb, 8) + 1;
-
-           reset_codec(s);
+        s->do_arith_coding    = get_bits1(&s->gb);
+        if(s->do_arith_coding) {
+            dprintf(s->avctx, "do_arith_coding == 1");
+            abort();
+        }
+        s->do_ac_filter       = get_bits1(&s->gb);
+        s->do_inter_ch_decorr = get_bits1(&s->gb);
+        s->do_mclms           = get_bits1(&s->gb);
+
+        if(s->do_ac_filter)
+            decode_ac_filter(s);
+
+        if(s->do_mclms)
+            decode_mclms(s);
+
+        decode_cdlms(s);
+        s->movave_scaling = get_bits(&s->gb, 3);
+        s->quant_stepsize = get_bits(&s->gb, 8) + 1;
+
+            reset_codec(s);
     }
 
     rawpcm_tile = get_bits1(&s->gb);
 
     for(i = 0; i < s->num_channels; i++) {
-       s->is_channel_coded[i] = 1;
+        s->is_channel_coded[i] = 1;
     }
 
     if(!rawpcm_tile) {
 
-       for(i = 0; i < s->num_channels; i++) {
-           s->is_channel_coded[i] = get_bits1(&s->gb);
-       }
-
-       if(s->bV3RTM) {
-           // LPC
-           s->do_lpc = get_bits1(&s->gb);
-           if(s->do_lpc) {
-               decode_lpc(s);
-           }
-       } else {
-           s->do_lpc = 0;
-       }
+        for(i = 0; i < s->num_channels; i++) {
+            s->is_channel_coded[i] = get_bits1(&s->gb);
+        }
+
+        if(s->bV3RTM) {
+            // LPC
+            s->do_lpc = get_bits1(&s->gb);
+            if(s->do_lpc) {
+                decode_lpc(s);
+            }
+        } else {
+            s->do_lpc = 0;
+        }
     }
 
 
     if(get_bits1(&s->gb)) {
-       padding_zeroes = get_bits(&s->gb, 5);
+        padding_zeroes = get_bits(&s->gb, 5);
     } else {
-       padding_zeroes = 0;
+        padding_zeroes = 0;
     }
 
     if(rawpcm_tile) {
-       
-       int bits = s->bits_per_sample - padding_zeroes;
-       int j;
-       dprintf(s->avctx, "RAWPCM %d bits per sample. total %d bits, remain=%d\n", bits,
-               bits * s->num_channels * subframe_len, get_bits_count(&s->gb));
-       for(i = 0; i < s->num_channels; i++) {
-           for(j = 0; j < subframe_len; j++) {
-               s->channel_coeffs[i][j] = get_sbits(&s->gb, bits);
-//             dprintf(s->avctx, "PCM[%d][%d] = 0x%04x\n", i, j, s->channel_coeffs[i][j]);
-           }
-       }
+
+        int bits = s->bits_per_sample - padding_zeroes;
+        int j;
+        dprintf(s->avctx, "RAWPCM %d bits per sample. total %d bits, remain=%d\n", bits,
+                bits * s->num_channels * subframe_len, get_bits_count(&s->gb));
+        for(i = 0; i < s->num_channels; i++) {
+            for(j = 0; j < subframe_len; j++) {
+                s->channel_coeffs[i][j] = get_sbits(&s->gb, bits);
+//                dprintf(s->avctx, "PCM[%d][%d] = 0x%04x\n", i, j, s->channel_coeffs[i][j]);
+            }
+        }
     } else {
-       for(i = 0; i < s->num_channels; i++)
-           if(s->is_channel_coded[i]) {
+        for(i = 0; i < s->num_channels; i++)
+            if(s->is_channel_coded[i]) {
             decode_channel_residues(s, i, subframe_len);
             if (s->seekable_tile)
                 use_high_update_speed(s, i);
@@ -1241,7 +1241,7 @@ static int decode_frame(WmallDecodeCtx *s)
     } else {
 /*
         while (get_bits_count(gb) < s->num_saved_bits && get_bits1(gb) == 0) {
-           dprintf(s->avctx, "skip1\n");
+            dprintf(s->avctx, "skip1\n");
         }
 */
     }
@@ -1351,8 +1351,8 @@ static int decode_packet(AVCodecContext *avctx,
         /** parse packet header */
         init_get_bits(gb, buf, s->buf_bit_size);
         packet_sequence_number = get_bits(gb, 4);
-       int seekable_frame_in_packet = get_bits1(gb);
-       int spliced_packet = get_bits1(gb);
+        int seekable_frame_in_packet = get_bits1(gb);
+        int spliced_packet = get_bits1(gb);
 
         /** get number of bits that need to be added to the previous frame */
         num_bits_prev_frame = get_bits(gb, s->log2_frame_size);
@@ -1379,7 +1379,7 @@ static int decode_packet(AVCodecContext *avctx,
 
             /** decode the cross packet frame if it is valid */
             if (!s->packet_loss)
-               decode_frame(s);
+                decode_frame(s);
         } else if (s->num_saved_bits - s->frame_offset) {
             dprintf(avctx, "ignoring %x previously saved bits\n",
                     s->num_saved_bits - s->frame_offset);
@@ -1399,7 +1399,7 @@ static int decode_packet(AVCodecContext *avctx,
         s->buf_bit_size = (avpkt->size - s->next_packet_start) << 3;
         init_get_bits(gb, avpkt->data, s->buf_bit_size);
         skip_bits(gb, s->packet_offset);
-       
+
         if (s->len_prefix && remaining_bits(s, gb) > s->log2_frame_size &&
             (frame_size = show_bits(gb, s->log2_frame_size)) &&
             frame_size <= remaining_bits(s, gb)) {
@@ -1417,7 +1417,7 @@ static int decode_packet(AVCodecContext *avctx,
             s->packet_done = !decode_frame(s);
         } else {
             s->packet_done = 1;
-       }
+        }
     }
 
     if (s->packet_done && !s->packet_loss &&
@@ -1462,7 +1462,7 @@ AVCodec ff_wmalossless_decoder = {
     NULL,
     decode_end,
     decode_packet,
-    .capabilities = CODEC_CAP_SUBFRAMES,
+    .capabilities = CODEC_CAP_SUBFRAMES | CODEC_CAP_EXPERIMENTAL,
     .flush= flush,
     .long_name = NULL_IF_CONFIG_SMALL("Windows Media Audio 9 Lossless"),
 };