mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-23 12:43:46 +02:00
wavpack: Convert to the new bitstream reader
This commit is contained in:
parent
6668bc80b5
commit
e7f24c9ffc
@ -23,10 +23,10 @@
|
||||
|
||||
#define BITSTREAM_READER_LE
|
||||
#include "avcodec.h"
|
||||
#include "bitstream.h"
|
||||
#include "bytestream.h"
|
||||
#include "get_bits.h"
|
||||
#include "internal.h"
|
||||
#include "unary_legacy.h"
|
||||
#include "unary.h"
|
||||
|
||||
/**
|
||||
* @file
|
||||
@ -109,10 +109,10 @@ typedef struct WavpackFrameContext {
|
||||
int stereo, stereo_in;
|
||||
int joint;
|
||||
uint32_t CRC;
|
||||
GetBitContext gb;
|
||||
BitstreamContext bc;
|
||||
int got_extra_bits;
|
||||
uint32_t crc_extra_bits;
|
||||
GetBitContext gb_extra_bits;
|
||||
BitstreamContext bc_extra_bits;
|
||||
int data_size; // in bits
|
||||
int samples;
|
||||
int terms;
|
||||
@ -240,7 +240,7 @@ static av_always_inline int wp_log2(int32_t val)
|
||||
} \
|
||||
}
|
||||
|
||||
static av_always_inline int get_tail(GetBitContext *gb, int k)
|
||||
static av_always_inline int get_tail(BitstreamContext *bc, int k)
|
||||
{
|
||||
int p, e, res;
|
||||
|
||||
@ -248,9 +248,9 @@ static av_always_inline int get_tail(GetBitContext *gb, int k)
|
||||
return 0;
|
||||
p = av_log2(k);
|
||||
e = (1 << (p + 1)) - k - 1;
|
||||
res = get_bitsz(gb, p);
|
||||
res = bitstream_read(bc, p);
|
||||
if (res >= e)
|
||||
res = (res << 1) - e + get_bits1(gb);
|
||||
res = (res << 1) - e + bitstream_read_bit(bc);
|
||||
return res;
|
||||
}
|
||||
|
||||
@ -288,7 +288,7 @@ static void update_error_limit(WavpackFrameContext *ctx)
|
||||
}
|
||||
}
|
||||
|
||||
static int wv_get_value(WavpackFrameContext *ctx, GetBitContext *gb,
|
||||
static int wv_get_value(WavpackFrameContext *ctx, BitstreamContext *bc,
|
||||
int channel, int *last)
|
||||
{
|
||||
int t, t2;
|
||||
@ -306,13 +306,13 @@ static int wv_get_value(WavpackFrameContext *ctx, GetBitContext *gb,
|
||||
return 0;
|
||||
}
|
||||
} else {
|
||||
t = get_unary_0_33(gb);
|
||||
t = get_unary_0_33(bc);
|
||||
if (t >= 2) {
|
||||
if (get_bits_left(gb) < t - 1)
|
||||
if (bitstream_bits_left(bc) < t - 1)
|
||||
goto error;
|
||||
t = get_bits(gb, t - 1) | (1 << (t - 1));
|
||||
t = bitstream_read(bc, t - 1) | (1 << (t - 1));
|
||||
} else {
|
||||
if (get_bits_left(gb) < 0)
|
||||
if (bitstream_bits_left(bc) < 0)
|
||||
goto error;
|
||||
}
|
||||
ctx->zeroes = t;
|
||||
@ -329,19 +329,19 @@ static int wv_get_value(WavpackFrameContext *ctx, GetBitContext *gb,
|
||||
t = 0;
|
||||
ctx->zero = 0;
|
||||
} else {
|
||||
t = get_unary_0_33(gb);
|
||||
if (get_bits_left(gb) < 0)
|
||||
t = get_unary_0_33(bc);
|
||||
if (bitstream_bits_left(bc) < 0)
|
||||
goto error;
|
||||
if (t == 16) {
|
||||
t2 = get_unary_0_33(gb);
|
||||
t2 = get_unary_0_33(bc);
|
||||
if (t2 < 2) {
|
||||
if (get_bits_left(gb) < 0)
|
||||
if (bitstream_bits_left(bc) < 0)
|
||||
goto error;
|
||||
t += t2;
|
||||
} else {
|
||||
if (get_bits_left(gb) < t2 - 1)
|
||||
if (bitstream_bits_left(bc) < t2 - 1)
|
||||
goto error;
|
||||
t += get_bits(gb, t2 - 1) | (1 << (t2 - 1));
|
||||
t += bitstream_read(bc, t2 - 1) | (1 << (t2 - 1));
|
||||
}
|
||||
}
|
||||
|
||||
@ -381,15 +381,15 @@ static int wv_get_value(WavpackFrameContext *ctx, GetBitContext *gb,
|
||||
INC_MED(2);
|
||||
}
|
||||
if (!c->error_limit) {
|
||||
ret = base + get_tail(gb, add);
|
||||
if (get_bits_left(gb) <= 0)
|
||||
ret = base + get_tail(bc, add);
|
||||
if (bitstream_bits_left(bc) <= 0)
|
||||
goto error;
|
||||
} else {
|
||||
int mid = (base * 2 + add + 1) >> 1;
|
||||
while (add > c->error_limit) {
|
||||
if (get_bits_left(gb) <= 0)
|
||||
if (bitstream_bits_left(bc) <= 0)
|
||||
goto error;
|
||||
if (get_bits1(gb)) {
|
||||
if (bitstream_read_bit(bc)) {
|
||||
add -= (mid - base);
|
||||
base = mid;
|
||||
} else
|
||||
@ -398,7 +398,7 @@ static int wv_get_value(WavpackFrameContext *ctx, GetBitContext *gb,
|
||||
}
|
||||
ret = mid;
|
||||
}
|
||||
sign = get_bits1(gb);
|
||||
sign = bitstream_read_bit(bc);
|
||||
if (ctx->hybrid_bitrate)
|
||||
c->slow_level += wp_log2(ret) - LEVEL_DECAY(c->slow_level);
|
||||
return sign ? ~ret : ret;
|
||||
@ -417,8 +417,8 @@ static inline int wv_get_value_integer(WavpackFrameContext *s, uint32_t *crc,
|
||||
S <<= s->extra_bits;
|
||||
|
||||
if (s->got_extra_bits &&
|
||||
get_bits_left(&s->gb_extra_bits) >= s->extra_bits) {
|
||||
S |= get_bits(&s->gb_extra_bits, s->extra_bits);
|
||||
bitstream_bits_left(&s->bc_extra_bits) >= s->extra_bits) {
|
||||
S |= bitstream_read(&s->bc_extra_bits, s->extra_bits);
|
||||
*crc = *crc * 9 + (S & 0xffff) * 3 + ((unsigned)S >> 16);
|
||||
}
|
||||
}
|
||||
@ -444,7 +444,7 @@ static float wv_get_value_float(WavpackFrameContext *s, uint32_t *crc, int S)
|
||||
|
||||
if (s->got_extra_bits) {
|
||||
const int max_bits = 1 + 23 + 8 + 1;
|
||||
const int left_bits = get_bits_left(&s->gb_extra_bits);
|
||||
const int left_bits = bitstream_bits_left(&s->bc_extra_bits);
|
||||
|
||||
if (left_bits + 8 * AV_INPUT_BUFFER_PADDING_SIZE < max_bits)
|
||||
return 0.0;
|
||||
@ -456,8 +456,8 @@ static float wv_get_value_float(WavpackFrameContext *s, uint32_t *crc, int S)
|
||||
if (sign)
|
||||
S = -S;
|
||||
if (S >= 0x1000000) {
|
||||
if (s->got_extra_bits && get_bits1(&s->gb_extra_bits))
|
||||
S = get_bits(&s->gb_extra_bits, 23);
|
||||
if (s->got_extra_bits && bitstream_read_bit(&s->bc_extra_bits))
|
||||
S = bitstream_read(&s->bc_extra_bits, 23);
|
||||
else
|
||||
S = 0;
|
||||
exp = 255;
|
||||
@ -473,11 +473,11 @@ static float wv_get_value_float(WavpackFrameContext *s, uint32_t *crc, int S)
|
||||
if ((s->float_flag & WV_FLT_SHIFT_ONES) ||
|
||||
(s->got_extra_bits &&
|
||||
(s->float_flag & WV_FLT_SHIFT_SAME) &&
|
||||
get_bits1(&s->gb_extra_bits))) {
|
||||
bitstream_read_bit(&s->bc_extra_bits))) {
|
||||
S |= (1 << shift) - 1;
|
||||
} else if (s->got_extra_bits &&
|
||||
(s->float_flag & WV_FLT_SHIFT_SENT)) {
|
||||
S |= get_bits(&s->gb_extra_bits, shift);
|
||||
S |= bitstream_read(&s->bc_extra_bits, shift);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
@ -488,14 +488,14 @@ static float wv_get_value_float(WavpackFrameContext *s, uint32_t *crc, int S)
|
||||
sign = 0;
|
||||
exp = 0;
|
||||
if (s->got_extra_bits && (s->float_flag & WV_FLT_ZERO_SENT)) {
|
||||
if (get_bits1(&s->gb_extra_bits)) {
|
||||
S = get_bits(&s->gb_extra_bits, 23);
|
||||
if (bitstream_read_bit(&s->bc_extra_bits)) {
|
||||
S = bitstream_read(&s->bc_extra_bits, 23);
|
||||
if (s->float_max_exp >= 25)
|
||||
exp = get_bits(&s->gb_extra_bits, 8);
|
||||
sign = get_bits1(&s->gb_extra_bits);
|
||||
exp = bitstream_read(&s->bc_extra_bits, 8);
|
||||
sign = bitstream_read_bit(&s->bc_extra_bits);
|
||||
} else {
|
||||
if (s->float_flag & WV_FLT_ZERO_SIGN)
|
||||
sign = get_bits1(&s->gb_extra_bits);
|
||||
sign = bitstream_read_bit(&s->bc_extra_bits);
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -527,7 +527,7 @@ static inline int wv_check_crc(WavpackFrameContext *s, uint32_t crc,
|
||||
return 0;
|
||||
}
|
||||
|
||||
static inline int wv_unpack_stereo(WavpackFrameContext *s, GetBitContext *gb,
|
||||
static inline int wv_unpack_stereo(WavpackFrameContext *s, BitstreamContext *bc,
|
||||
void *dst_l, void *dst_r, const int type)
|
||||
{
|
||||
int i, j, count = 0;
|
||||
@ -545,10 +545,10 @@ static inline int wv_unpack_stereo(WavpackFrameContext *s, GetBitContext *gb,
|
||||
|
||||
s->one = s->zero = s->zeroes = 0;
|
||||
do {
|
||||
L = wv_get_value(s, gb, 0, &last);
|
||||
L = wv_get_value(s, bc, 0, &last);
|
||||
if (last)
|
||||
break;
|
||||
R = wv_get_value(s, gb, 1, &last);
|
||||
R = wv_get_value(s, bc, 1, &last);
|
||||
if (last)
|
||||
break;
|
||||
for (i = 0; i < s->terms; i++) {
|
||||
@ -645,7 +645,7 @@ static inline int wv_unpack_stereo(WavpackFrameContext *s, GetBitContext *gb,
|
||||
return 0;
|
||||
}
|
||||
|
||||
static inline int wv_unpack_mono(WavpackFrameContext *s, GetBitContext *gb,
|
||||
static inline int wv_unpack_mono(WavpackFrameContext *s, BitstreamContext *bc,
|
||||
void *dst, const int type)
|
||||
{
|
||||
int i, j, count = 0;
|
||||
@ -660,7 +660,7 @@ static inline int wv_unpack_mono(WavpackFrameContext *s, GetBitContext *gb,
|
||||
|
||||
s->one = s->zero = s->zeroes = 0;
|
||||
do {
|
||||
T = wv_get_value(s, gb, 0, &last);
|
||||
T = wv_get_value(s, bc, 0, &last);
|
||||
S = 0;
|
||||
if (last)
|
||||
break;
|
||||
@ -990,7 +990,7 @@ static int wavpack_decode_block(AVCodecContext *avctx, int block_no,
|
||||
case WP_ID_DATA:
|
||||
s->sc.offset = bytestream2_tell(&gb);
|
||||
s->sc.size = size * 8;
|
||||
init_get_bits(&s->gb, gb.buffer, size * 8);
|
||||
bitstream_init8(&s->bc, gb.buffer, size);
|
||||
s->data_size = size * 8;
|
||||
bytestream2_skip(&gb, size);
|
||||
got_bs = 1;
|
||||
@ -1004,8 +1004,8 @@ static int wavpack_decode_block(AVCodecContext *avctx, int block_no,
|
||||
}
|
||||
s->extra_sc.offset = bytestream2_tell(&gb);
|
||||
s->extra_sc.size = size * 8;
|
||||
init_get_bits(&s->gb_extra_bits, gb.buffer, size * 8);
|
||||
s->crc_extra_bits = get_bits_long(&s->gb_extra_bits, 32);
|
||||
bitstream_init8(&s->bc_extra_bits, gb.buffer, size);
|
||||
s->crc_extra_bits = bitstream_read(&s->bc_extra_bits, 32);
|
||||
bytestream2_skip(&gb, size);
|
||||
s->got_extra_bits = 1;
|
||||
break;
|
||||
@ -1084,7 +1084,7 @@ static int wavpack_decode_block(AVCodecContext *avctx, int block_no,
|
||||
return AVERROR_INVALIDDATA;
|
||||
}
|
||||
if (s->got_extra_bits && avctx->sample_fmt != AV_SAMPLE_FMT_FLTP) {
|
||||
const int size = get_bits_left(&s->gb_extra_bits);
|
||||
const int size = bitstream_bits_left(&s->bc_extra_bits);
|
||||
const int wanted = s->samples * s->extra_bits << s->stereo_in;
|
||||
if (size < wanted) {
|
||||
av_log(avctx, AV_LOG_ERROR, "Too small EXTRABITS\n");
|
||||
@ -1134,11 +1134,11 @@ static int wavpack_decode_block(AVCodecContext *avctx, int block_no,
|
||||
wc->ch_offset += 1 + s->stereo;
|
||||
|
||||
if (s->stereo_in) {
|
||||
ret = wv_unpack_stereo(s, &s->gb, samples_l, samples_r, avctx->sample_fmt);
|
||||
ret = wv_unpack_stereo(s, &s->bc, samples_l, samples_r, avctx->sample_fmt);
|
||||
if (ret < 0)
|
||||
return ret;
|
||||
} else {
|
||||
ret = wv_unpack_mono(s, &s->gb, samples_l, avctx->sample_fmt);
|
||||
ret = wv_unpack_mono(s, &s->bc, samples_l, avctx->sample_fmt);
|
||||
if (ret < 0)
|
||||
return ret;
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user