From 2b7d9a1f3fa759c82aaa3569612b40d5ccb1e319 Mon Sep 17 00:00:00 2001 From: Jun Zhao Date: Wed, 14 Jun 2017 10:22:10 +0800 Subject: [PATCH] lavc/put_bits: Add put_bits64() to support up to 64 bits. put_bits64() can write up to 64 bits into a bitstream. Reviewed-by: Mark Thompson Reviewed-by: Michael Niedermayer Signed-off-by: Jun Zhao Signed-off-by: Michael Niedermayer --- libavcodec/put_bits.h | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) diff --git a/libavcodec/put_bits.h b/libavcodec/put_bits.h index 9bd45cd8ba..b85e88f28c 100644 --- a/libavcodec/put_bits.h +++ b/libavcodec/put_bits.h @@ -242,6 +242,41 @@ static void av_unused put_bits32(PutBitContext *s, uint32_t value) s->bit_left = bit_left; } +/** + * Write up to 64 bits into a bitstream. + */ +static inline void put_bits64(PutBitContext *s, int n, uint64_t value) +{ + av_assert2((n == 64) || (n < 64 && value < (UINT64_C(1) << n))); + + if (n < 32) + put_bits(s, n, value); + else if (n == 32) + put_bits32(s, value); + else if (n < 64) { + uint32_t lo = value & 0xffffffff; + uint32_t hi = value >> 32; +#ifdef BITSTREAM_WRITER_LE + put_bits32(s, lo); + put_bits(s, n - 32, hi); +#else + put_bits(s, n - 32, hi); + put_bits32(s, lo); +#endif + } else { + uint32_t lo = value & 0xffffffff; + uint32_t hi = value >> 32; +#ifdef BITSTREAM_WRITER_LE + put_bits32(s, lo); + put_bits32(s, hi); +#else + put_bits32(s, hi); + put_bits32(s, lo); +#endif + + } +} + /** * Return the pointer to the byte where the bitstream writer will put * the next bit.