Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
F
ffmpeg
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
ffmpeg
Commits
00c72a1e
Commit
00c72a1e
authored
Apr 10, 2016
by
Alexandra Hájková
Committed by
Diego Biurrun
Dec 19, 2016
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
mlp: Convert to the new bitstream reader
parent
fa64aea1
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
136 additions
and
132 deletions
+136
-132
mlp_parser.c
libavcodec/mlp_parser.c
+32
-31
mlp_parser.h
libavcodec/mlp_parser.h
+2
-2
mlpdec.c
libavcodec/mlpdec.c
+102
-99
No files found.
libavcodec/mlp_parser.c
View file @
00c72a1e
...
@@ -29,7 +29,8 @@
...
@@ -29,7 +29,8 @@
#include "libavutil/channel_layout.h"
#include "libavutil/channel_layout.h"
#include "libavutil/crc.h"
#include "libavutil/crc.h"
#include "libavutil/internal.h"
#include "libavutil/internal.h"
#include "get_bits.h"
#include "bitstream.h"
#include "parser.h"
#include "parser.h"
#include "mlp_parser.h"
#include "mlp_parser.h"
#include "mlp.h"
#include "mlp.h"
...
@@ -139,67 +140,67 @@ static int ff_mlp_get_major_sync_size(const uint8_t * buf, int bufsize)
...
@@ -139,67 +140,67 @@ static int ff_mlp_get_major_sync_size(const uint8_t * buf, int bufsize)
/** Read a major sync info header - contains high level information about
/** Read a major sync info header - contains high level information about
* the stream - sample rate, channel arrangement etc. Most of this
* the stream - sample rate, channel arrangement etc. Most of this
* information is not actually necessary for decoding, only for playback.
* information is not actually necessary for decoding, only for playback.
*
gb must be a freshly initialized GetBit
Context with no bits read.
*
bc must be a freshly-initialized Bitstream
Context with no bits read.
*/
*/
int
ff_mlp_read_major_sync
(
void
*
log
,
MLPHeaderInfo
*
mh
,
GetBitContext
*
gb
)
int
ff_mlp_read_major_sync
(
void
*
log
,
MLPHeaderInfo
*
mh
,
BitstreamContext
*
bc
)
{
{
int
ratebits
,
channel_arrangement
,
header_size
;
int
ratebits
,
channel_arrangement
,
header_size
;
uint16_t
checksum
;
uint16_t
checksum
;
assert
(
get_bits_count
(
gb
)
==
0
);
assert
(
bitstream_tell
(
bc
)
==
0
);
header_size
=
ff_mlp_get_major_sync_size
(
gb
->
buffer
,
gb
->
size_in_bits
>>
3
);
header_size
=
ff_mlp_get_major_sync_size
(
bc
->
buffer
,
bc
->
size_in_bits
>>
3
);
if
(
header_size
<
0
||
gb
->
size_in_bits
<
header_size
<<
3
)
{
if
(
header_size
<
0
||
bc
->
size_in_bits
<
header_size
<<
3
)
{
av_log
(
log
,
AV_LOG_ERROR
,
"packet too short, unable to read major sync
\n
"
);
av_log
(
log
,
AV_LOG_ERROR
,
"packet too short, unable to read major sync
\n
"
);
return
-
1
;
return
-
1
;
}
}
checksum
=
ff_mlp_checksum16
(
gb
->
buffer
,
header_size
-
2
);
checksum
=
ff_mlp_checksum16
(
bc
->
buffer
,
header_size
-
2
);
if
(
checksum
!=
AV_RL16
(
gb
->
buffer
+
header_size
-
2
))
{
if
(
checksum
!=
AV_RL16
(
bc
->
buffer
+
header_size
-
2
))
{
av_log
(
log
,
AV_LOG_ERROR
,
"major sync info header checksum error
\n
"
);
av_log
(
log
,
AV_LOG_ERROR
,
"major sync info header checksum error
\n
"
);
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
}
}
if
(
get_bits_long
(
gb
,
24
)
!=
0xf8726f
)
/* Sync words */
if
(
bitstream_read
(
bc
,
24
)
!=
0xf8726f
)
/* Sync words */
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
mh
->
stream_type
=
get_bits
(
gb
,
8
);
mh
->
stream_type
=
bitstream_read
(
bc
,
8
);
mh
->
header_size
=
header_size
;
mh
->
header_size
=
header_size
;
if
(
mh
->
stream_type
==
0xbb
)
{
if
(
mh
->
stream_type
==
0xbb
)
{
mh
->
group1_bits
=
mlp_quants
[
get_bits
(
gb
,
4
)];
mh
->
group1_bits
=
mlp_quants
[
bitstream_read
(
bc
,
4
)];
mh
->
group2_bits
=
mlp_quants
[
get_bits
(
gb
,
4
)];
mh
->
group2_bits
=
mlp_quants
[
bitstream_read
(
bc
,
4
)];
ratebits
=
get_bits
(
gb
,
4
);
ratebits
=
bitstream_read
(
bc
,
4
);
mh
->
group1_samplerate
=
mlp_samplerate
(
ratebits
);
mh
->
group1_samplerate
=
mlp_samplerate
(
ratebits
);
mh
->
group2_samplerate
=
mlp_samplerate
(
get_bits
(
gb
,
4
));
mh
->
group2_samplerate
=
mlp_samplerate
(
bitstream_read
(
bc
,
4
));
skip_bits
(
gb
,
11
);
bitstream_skip
(
bc
,
11
);
channel_arrangement
=
get_bits
(
gb
,
5
);
channel_arrangement
=
bitstream_read
(
bc
,
5
);
mh
->
channels_mlp
=
mlp_channels
[
channel_arrangement
];
mh
->
channels_mlp
=
mlp_channels
[
channel_arrangement
];
mh
->
channel_layout_mlp
=
mlp_layout
[
channel_arrangement
];
mh
->
channel_layout_mlp
=
mlp_layout
[
channel_arrangement
];
}
else
if
(
mh
->
stream_type
==
0xba
)
{
}
else
if
(
mh
->
stream_type
==
0xba
)
{
mh
->
group1_bits
=
24
;
// TODO: Is this information actually conveyed anywhere?
mh
->
group1_bits
=
24
;
// TODO: Is this information actually conveyed anywhere?
mh
->
group2_bits
=
0
;
mh
->
group2_bits
=
0
;
ratebits
=
get_bits
(
gb
,
4
);
ratebits
=
bitstream_read
(
bc
,
4
);
mh
->
group1_samplerate
=
mlp_samplerate
(
ratebits
);
mh
->
group1_samplerate
=
mlp_samplerate
(
ratebits
);
mh
->
group2_samplerate
=
0
;
mh
->
group2_samplerate
=
0
;
skip_bits
(
gb
,
4
);
bitstream_skip
(
bc
,
4
);
mh
->
channel_modifier_thd_stream0
=
get_bits
(
gb
,
2
);
mh
->
channel_modifier_thd_stream0
=
bitstream_read
(
bc
,
2
);
mh
->
channel_modifier_thd_stream1
=
get_bits
(
gb
,
2
);
mh
->
channel_modifier_thd_stream1
=
bitstream_read
(
bc
,
2
);
channel_arrangement
=
get_bits
(
gb
,
5
);
channel_arrangement
=
bitstream_read
(
bc
,
5
);
mh
->
channels_thd_stream1
=
truehd_channels
(
channel_arrangement
);
mh
->
channels_thd_stream1
=
truehd_channels
(
channel_arrangement
);
mh
->
channel_layout_thd_stream1
=
truehd_layout
(
channel_arrangement
);
mh
->
channel_layout_thd_stream1
=
truehd_layout
(
channel_arrangement
);
mh
->
channel_modifier_thd_stream2
=
get_bits
(
gb
,
2
);
mh
->
channel_modifier_thd_stream2
=
bitstream_read
(
bc
,
2
);
channel_arrangement
=
get_bits
(
gb
,
13
);
channel_arrangement
=
bitstream_read
(
bc
,
13
);
mh
->
channels_thd_stream2
=
truehd_channels
(
channel_arrangement
);
mh
->
channels_thd_stream2
=
truehd_channels
(
channel_arrangement
);
mh
->
channel_layout_thd_stream2
=
truehd_layout
(
channel_arrangement
);
mh
->
channel_layout_thd_stream2
=
truehd_layout
(
channel_arrangement
);
}
else
}
else
...
@@ -208,15 +209,15 @@ int ff_mlp_read_major_sync(void *log, MLPHeaderInfo *mh, GetBitContext *gb)
...
@@ -208,15 +209,15 @@ int ff_mlp_read_major_sync(void *log, MLPHeaderInfo *mh, GetBitContext *gb)
mh
->
access_unit_size
=
40
<<
(
ratebits
&
7
);
mh
->
access_unit_size
=
40
<<
(
ratebits
&
7
);
mh
->
access_unit_size_pow2
=
64
<<
(
ratebits
&
7
);
mh
->
access_unit_size_pow2
=
64
<<
(
ratebits
&
7
);
skip_bits_long
(
gb
,
48
);
bitstream_skip
(
bc
,
48
);
mh
->
is_vbr
=
get_bits1
(
gb
);
mh
->
is_vbr
=
bitstream_read_bit
(
bc
);
mh
->
peak_bitrate
=
(
get_bits
(
gb
,
15
)
*
mh
->
group1_samplerate
+
8
)
>>
4
;
mh
->
peak_bitrate
=
(
bitstream_read
(
bc
,
15
)
*
mh
->
group1_samplerate
+
8
)
>>
4
;
mh
->
num_substreams
=
get_bits
(
gb
,
4
);
mh
->
num_substreams
=
bitstream_read
(
bc
,
4
);
skip_bits_long
(
gb
,
4
+
(
header_size
-
17
)
*
8
);
bitstream_skip
(
bc
,
4
+
(
header_size
-
17
)
*
8
);
return
0
;
return
0
;
}
}
...
@@ -328,11 +329,11 @@ static int mlp_parse(AVCodecParserContext *s,
...
@@ -328,11 +329,11 @@ static int mlp_parse(AVCodecParserContext *s,
goto
lost_sync
;
goto
lost_sync
;
}
}
}
else
{
}
else
{
GetBitContext
gb
;
BitstreamContext
bc
;
MLPHeaderInfo
mh
;
MLPHeaderInfo
mh
;
init_get_bits
(
&
gb
,
buf
+
4
,
(
buf_size
-
4
)
<<
3
);
bitstream_init
(
&
bc
,
buf
+
4
,
(
buf_size
-
4
)
<<
3
);
if
(
ff_mlp_read_major_sync
(
avctx
,
&
mh
,
&
gb
)
<
0
)
if
(
ff_mlp_read_major_sync
(
avctx
,
&
mh
,
&
bc
)
<
0
)
goto
lost_sync
;
goto
lost_sync
;
avctx
->
bits_per_raw_sample
=
mh
.
group1_bits
;
avctx
->
bits_per_raw_sample
=
mh
.
group1_bits
;
...
...
libavcodec/mlp_parser.h
View file @
00c72a1e
...
@@ -27,7 +27,7 @@
...
@@ -27,7 +27,7 @@
#ifndef AVCODEC_MLP_PARSER_H
#ifndef AVCODEC_MLP_PARSER_H
#define AVCODEC_MLP_PARSER_H
#define AVCODEC_MLP_PARSER_H
#include "
get_bits
.h"
#include "
bitstream
.h"
typedef
struct
MLPHeaderInfo
typedef
struct
MLPHeaderInfo
{
{
...
@@ -61,6 +61,6 @@ typedef struct MLPHeaderInfo
...
@@ -61,6 +61,6 @@ typedef struct MLPHeaderInfo
}
MLPHeaderInfo
;
}
MLPHeaderInfo
;
int
ff_mlp_read_major_sync
(
void
*
log
,
MLPHeaderInfo
*
mh
,
GetBitContext
*
gb
);
int
ff_mlp_read_major_sync
(
void
*
log
,
MLPHeaderInfo
*
mh
,
BitstreamContext
*
bc
);
#endif
/* AVCODEC_MLP_PARSER_H */
#endif
/* AVCODEC_MLP_PARSER_H */
libavcodec/mlpdec.c
View file @
00c72a1e
...
@@ -26,13 +26,14 @@
...
@@ -26,13 +26,14 @@
#include <stdint.h>
#include <stdint.h>
#include "avcodec.h"
#include "libavutil/internal.h"
#include "libavutil/internal.h"
#include "libavutil/intreadwrite.h"
#include "libavutil/intreadwrite.h"
#include "libavutil/channel_layout.h"
#include "libavutil/channel_layout.h"
#include "get_bits.h"
#include "internal.h"
#include "libavutil/crc.h"
#include "libavutil/crc.h"
#include "avcodec.h"
#include "bitstream.h"
#include "internal.h"
#include "parser.h"
#include "parser.h"
#include "mlp_parser.h"
#include "mlp_parser.h"
#include "mlpdsp.h"
#include "mlpdsp.h"
...
@@ -233,7 +234,7 @@ static inline int32_t calculate_sign_huff(MLPDecodeContext *m,
...
@@ -233,7 +234,7 @@ static inline int32_t calculate_sign_huff(MLPDecodeContext *m,
/** Read a sample, consisting of either, both or neither of entropy-coded MSBs
/** Read a sample, consisting of either, both or neither of entropy-coded MSBs
* and plain LSBs. */
* and plain LSBs. */
static
inline
int
read_huff_channels
(
MLPDecodeContext
*
m
,
GetBitContext
*
gbp
,
static
inline
int
read_huff_channels
(
MLPDecodeContext
*
m
,
BitstreamContext
*
bc
,
unsigned
int
substr
,
unsigned
int
pos
)
unsigned
int
substr
,
unsigned
int
pos
)
{
{
SubStream
*
s
=
&
m
->
substream
[
substr
];
SubStream
*
s
=
&
m
->
substream
[
substr
];
...
@@ -241,7 +242,7 @@ static inline int read_huff_channels(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -241,7 +242,7 @@ static inline int read_huff_channels(MLPDecodeContext *m, GetBitContext *gbp,
for
(
mat
=
0
;
mat
<
s
->
num_primitive_matrices
;
mat
++
)
for
(
mat
=
0
;
mat
<
s
->
num_primitive_matrices
;
mat
++
)
if
(
s
->
lsb_bypass
[
mat
])
if
(
s
->
lsb_bypass
[
mat
])
m
->
bypassed_lsbs
[
pos
+
s
->
blockpos
][
mat
]
=
get_bits1
(
gbp
);
m
->
bypassed_lsbs
[
pos
+
s
->
blockpos
][
mat
]
=
bitstream_read_bit
(
bc
);
for
(
channel
=
s
->
min_channel
;
channel
<=
s
->
max_channel
;
channel
++
)
{
for
(
channel
=
s
->
min_channel
;
channel
<=
s
->
max_channel
;
channel
++
)
{
ChannelParams
*
cp
=
&
s
->
channel_params
[
channel
];
ChannelParams
*
cp
=
&
s
->
channel_params
[
channel
];
...
@@ -251,14 +252,15 @@ static inline int read_huff_channels(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -251,14 +252,15 @@ static inline int read_huff_channels(MLPDecodeContext *m, GetBitContext *gbp,
int
result
=
0
;
int
result
=
0
;
if
(
codebook
>
0
)
if
(
codebook
>
0
)
result
=
get_vlc2
(
gbp
,
huff_vlc
[
codebook
-
1
].
table
,
result
=
bitstream_read_vlc
(
bc
,
huff_vlc
[
codebook
-
1
].
table
,
VLC_BITS
,
(
9
+
VLC_BITS
-
1
)
/
VLC_BITS
);
VLC_BITS
,
(
9
+
VLC_BITS
-
1
)
/
VLC_BITS
);
if
(
result
<
0
)
if
(
result
<
0
)
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
if
(
lsb_bits
>
0
)
if
(
lsb_bits
>
0
)
result
=
(
result
<<
lsb_bits
)
+
get_bits
(
gbp
,
lsb_bits
);
result
=
(
result
<<
lsb_bits
)
+
bitstream_read
(
bc
,
lsb_bits
);
result
+=
cp
->
sign_huff_offset
;
result
+=
cp
->
sign_huff_offset
;
result
<<=
quant_step_size
;
result
<<=
quant_step_size
;
...
@@ -288,12 +290,12 @@ static av_cold int mlp_decode_init(AVCodecContext *avctx)
...
@@ -288,12 +290,12 @@ static av_cold int mlp_decode_init(AVCodecContext *avctx)
* information is not actually necessary for decoding, only for playback.
* information is not actually necessary for decoding, only for playback.
*/
*/
static
int
read_major_sync
(
MLPDecodeContext
*
m
,
GetBitContext
*
gb
)
static
int
read_major_sync
(
MLPDecodeContext
*
m
,
BitstreamContext
*
bc
)
{
{
MLPHeaderInfo
mh
;
MLPHeaderInfo
mh
;
int
substr
,
ret
;
int
substr
,
ret
;
if
((
ret
=
ff_mlp_read_major_sync
(
m
->
avctx
,
&
mh
,
gb
))
!=
0
)
if
((
ret
=
ff_mlp_read_major_sync
(
m
->
avctx
,
&
mh
,
bc
))
!=
0
)
return
ret
;
return
ret
;
if
(
mh
.
group1_bits
==
0
)
{
if
(
mh
.
group1_bits
==
0
)
{
...
@@ -441,7 +443,7 @@ static int read_major_sync(MLPDecodeContext *m, GetBitContext *gb)
...
@@ -441,7 +443,7 @@ static int read_major_sync(MLPDecodeContext *m, GetBitContext *gb)
* required to decode the audio that do not change very often. Generally
* required to decode the audio that do not change very often. Generally
* (always) present only in blocks following a major sync. */
* (always) present only in blocks following a major sync. */
static
int
read_restart_header
(
MLPDecodeContext
*
m
,
GetBitContext
*
gbp
,
static
int
read_restart_header
(
MLPDecodeContext
*
m
,
BitstreamContext
*
bc
,
const
uint8_t
*
buf
,
unsigned
int
substr
)
const
uint8_t
*
buf
,
unsigned
int
substr
)
{
{
SubStream
*
s
=
&
m
->
substream
[
substr
];
SubStream
*
s
=
&
m
->
substream
[
substr
];
...
@@ -449,13 +451,13 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -449,13 +451,13 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
int
sync_word
,
tmp
;
int
sync_word
,
tmp
;
uint8_t
checksum
;
uint8_t
checksum
;
uint8_t
lossless_check
;
uint8_t
lossless_check
;
int
start_count
=
get_bits_count
(
gbp
);
int
start_count
=
bitstream_tell
(
bc
);
int
min_channel
,
max_channel
,
max_matrix_channel
;
int
min_channel
,
max_channel
,
max_matrix_channel
;
const
int
std_max_matrix_channel
=
m
->
avctx
->
codec_id
==
AV_CODEC_ID_MLP
const
int
std_max_matrix_channel
=
m
->
avctx
->
codec_id
==
AV_CODEC_ID_MLP
?
MAX_MATRIX_CHANNEL_MLP
?
MAX_MATRIX_CHANNEL_MLP
:
MAX_MATRIX_CHANNEL_TRUEHD
;
:
MAX_MATRIX_CHANNEL_TRUEHD
;
sync_word
=
get_bits
(
gbp
,
13
);
sync_word
=
bitstream_read
(
bc
,
13
);
if
(
sync_word
!=
0x31ea
>>
1
)
{
if
(
sync_word
!=
0x31ea
>>
1
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
...
@@ -463,18 +465,18 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -463,18 +465,18 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
}
}
s
->
noise_type
=
get_bits1
(
gbp
);
s
->
noise_type
=
bitstream_read_bit
(
bc
);
if
(
m
->
avctx
->
codec_id
==
AV_CODEC_ID_MLP
&&
s
->
noise_type
)
{
if
(
m
->
avctx
->
codec_id
==
AV_CODEC_ID_MLP
&&
s
->
noise_type
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"MLP must have 0x31ea sync word.
\n
"
);
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"MLP must have 0x31ea sync word.
\n
"
);
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
}
}
skip_bits
(
gbp
,
16
);
/* Output timestamp */
bitstream_skip
(
bc
,
16
);
/* Output timestamp */
min_channel
=
get_bits
(
gbp
,
4
);
min_channel
=
bitstream_read
(
bc
,
4
);
max_channel
=
get_bits
(
gbp
,
4
);
max_channel
=
bitstream_read
(
bc
,
4
);
max_matrix_channel
=
get_bits
(
gbp
,
4
);
max_matrix_channel
=
bitstream_read
(
bc
,
4
);
if
(
max_matrix_channel
>
std_max_matrix_channel
)
{
if
(
max_matrix_channel
>
std_max_matrix_channel
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
...
@@ -518,13 +520,13 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -518,13 +520,13 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
m
->
max_decoded_substream
=
substr
;
m
->
max_decoded_substream
=
substr
;
}
}
s
->
noise_shift
=
get_bits
(
gbp
,
4
);
s
->
noise_shift
=
bitstream_read
(
bc
,
4
);
s
->
noisegen_seed
=
get_bits
(
gbp
,
23
);
s
->
noisegen_seed
=
bitstream_read
(
bc
,
23
);
skip_bits
(
gbp
,
19
);
bitstream_skip
(
bc
,
19
);
s
->
data_check_present
=
get_bits1
(
gbp
);
s
->
data_check_present
=
bitstream_read_bit
(
bc
);
lossless_check
=
get_bits
(
gbp
,
8
);
lossless_check
=
bitstream_read
(
bc
,
8
);
if
(
substr
==
m
->
max_decoded_substream
if
(
substr
==
m
->
max_decoded_substream
&&
s
->
lossless_check_data
!=
0xffffffff
)
{
&&
s
->
lossless_check_data
!=
0xffffffff
)
{
tmp
=
xor_32_to_8
(
s
->
lossless_check_data
);
tmp
=
xor_32_to_8
(
s
->
lossless_check_data
);
...
@@ -534,12 +536,12 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -534,12 +536,12 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
lossless_check
,
tmp
);
lossless_check
,
tmp
);
}
}
skip_bits
(
gbp
,
16
);
bitstream_skip
(
bc
,
16
);
memset
(
s
->
ch_assign
,
0
,
sizeof
(
s
->
ch_assign
));
memset
(
s
->
ch_assign
,
0
,
sizeof
(
s
->
ch_assign
));
for
(
ch
=
0
;
ch
<=
s
->
max_matrix_channel
;
ch
++
)
{
for
(
ch
=
0
;
ch
<=
s
->
max_matrix_channel
;
ch
++
)
{
int
ch_assign
=
get_bits
(
gbp
,
6
);
int
ch_assign
=
bitstream_read
(
bc
,
6
);
if
(
m
->
avctx
->
codec_id
==
AV_CODEC_ID_TRUEHD
)
{
if
(
m
->
avctx
->
codec_id
==
AV_CODEC_ID_TRUEHD
)
{
uint64_t
channel
=
thd_channel_layout_extract_channel
(
s
->
ch_layout
,
uint64_t
channel
=
thd_channel_layout_extract_channel
(
s
->
ch_layout
,
ch_assign
);
ch_assign
);
...
@@ -555,9 +557,9 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -555,9 +557,9 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
s
->
ch_assign
[
ch_assign
]
=
ch
;
s
->
ch_assign
[
ch_assign
]
=
ch
;
}
}
checksum
=
ff_mlp_restart_checksum
(
buf
,
get_bits_count
(
gbp
)
-
start_count
);
checksum
=
ff_mlp_restart_checksum
(
buf
,
bitstream_tell
(
bc
)
-
start_count
);
if
(
checksum
!=
get_bits
(
gbp
,
8
))
if
(
checksum
!=
bitstream_read
(
bc
,
8
))
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"restart header checksum error
\n
"
);
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"restart header checksum error
\n
"
);
/* Set default decoding parameters. */
/* Set default decoding parameters. */
...
@@ -597,7 +599,7 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -597,7 +599,7 @@ static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
/** Read parameters for one of the prediction filters. */
/** Read parameters for one of the prediction filters. */
static
int
read_filter_params
(
MLPDecodeContext
*
m
,
GetBitContext
*
gbp
,
static
int
read_filter_params
(
MLPDecodeContext
*
m
,
BitstreamContext
*
bc
,
unsigned
int
substr
,
unsigned
int
channel
,
unsigned
int
substr
,
unsigned
int
channel
,
unsigned
int
filter
)
unsigned
int
filter
)
{
{
...
@@ -615,7 +617,7 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -615,7 +617,7 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
}
}
order
=
get_bits
(
gbp
,
4
);
order
=
bitstream_read
(
bc
,
4
);
if
(
order
>
max_order
)
{
if
(
order
>
max_order
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"%cIR filter order %d is greater than maximum %d.
\n
"
,
"%cIR filter order %d is greater than maximum %d.
\n
"
,
...
@@ -628,10 +630,10 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -628,10 +630,10 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
int32_t
*
fcoeff
=
s
->
channel_params
[
channel
].
coeff
[
filter
];
int32_t
*
fcoeff
=
s
->
channel_params
[
channel
].
coeff
[
filter
];
int
coeff_bits
,
coeff_shift
;
int
coeff_bits
,
coeff_shift
;
fp
->
shift
=
get_bits
(
gbp
,
4
);
fp
->
shift
=
bitstream_read
(
bc
,
4
);
coeff_bits
=
get_bits
(
gbp
,
5
);
coeff_bits
=
bitstream_read
(
bc
,
5
);
coeff_shift
=
get_bits
(
gbp
,
3
);
coeff_shift
=
bitstream_read
(
bc
,
3
);
if
(
coeff_bits
<
1
||
coeff_bits
>
16
)
{
if
(
coeff_bits
<
1
||
coeff_bits
>
16
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"%cIR filter coeff_bits must be between 1 and 16.
\n
"
,
"%cIR filter coeff_bits must be between 1 and 16.
\n
"
,
...
@@ -646,9 +648,9 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -646,9 +648,9 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
}
}
for
(
i
=
0
;
i
<
order
;
i
++
)
for
(
i
=
0
;
i
<
order
;
i
++
)
fcoeff
[
i
]
=
get_sbits
(
gbp
,
coeff_bits
)
<<
coeff_shift
;
fcoeff
[
i
]
=
bitstream_read_signed
(
bc
,
coeff_bits
)
<<
coeff_shift
;
if
(
get_bits1
(
gbp
))
{
if
(
bitstream_read_bit
(
bc
))
{
int
state_bits
,
state_shift
;
int
state_bits
,
state_shift
;
if
(
filter
==
FIR
)
{
if
(
filter
==
FIR
)
{
...
@@ -657,13 +659,13 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -657,13 +659,13 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
}
}
state_bits
=
get_bits
(
gbp
,
4
);
state_bits
=
bitstream_read
(
bc
,
4
);
state_shift
=
get_bits
(
gbp
,
4
);
state_shift
=
bitstream_read
(
bc
,
4
);
/* TODO: Check validity of state data. */
/* TODO: Check validity of state data. */
for
(
i
=
0
;
i
<
order
;
i
++
)
for
(
i
=
0
;
i
<
order
;
i
++
)
fp
->
state
[
i
]
=
get_sbits
(
gbp
,
state_bits
)
<<
state_shift
;
fp
->
state
[
i
]
=
bitstream_read_signed
(
bc
,
state_bits
)
<<
state_shift
;
}
}
}
}
...
@@ -672,7 +674,8 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -672,7 +674,8 @@ static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
/** Read parameters for primitive matrices. */
/** Read parameters for primitive matrices. */
static
int
read_matrix_params
(
MLPDecodeContext
*
m
,
unsigned
int
substr
,
GetBitContext
*
gbp
)
static
int
read_matrix_params
(
MLPDecodeContext
*
m
,
unsigned
int
substr
,
BitstreamContext
*
bc
)
{
{
SubStream
*
s
=
&
m
->
substream
[
substr
];
SubStream
*
s
=
&
m
->
substream
[
substr
];
unsigned
int
mat
,
ch
;
unsigned
int
mat
,
ch
;
...
@@ -685,7 +688,7 @@ static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitCo
...
@@ -685,7 +688,7 @@ static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitCo
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
}
}
s
->
num_primitive_matrices
=
get_bits
(
gbp
,
4
);
s
->
num_primitive_matrices
=
bitstream_read
(
bc
,
4
);
if
(
s
->
num_primitive_matrices
>
max_primitive_matrices
)
{
if
(
s
->
num_primitive_matrices
>
max_primitive_matrices
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
...
@@ -696,9 +699,9 @@ static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitCo
...
@@ -696,9 +699,9 @@ static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitCo
for
(
mat
=
0
;
mat
<
s
->
num_primitive_matrices
;
mat
++
)
{
for
(
mat
=
0
;
mat
<
s
->
num_primitive_matrices
;
mat
++
)
{
int
frac_bits
,
max_chan
;
int
frac_bits
,
max_chan
;
s
->
matrix_out_ch
[
mat
]
=
get_bits
(
gbp
,
4
);
s
->
matrix_out_ch
[
mat
]
=
bitstream_read
(
bc
,
4
);
frac_bits
=
get_bits
(
gbp
,
4
);
frac_bits
=
bitstream_read
(
bc
,
4
);
s
->
lsb_bypass
[
mat
]
=
get_bits1
(
gbp
);
s
->
lsb_bypass
[
mat
]
=
bitstream_read_bit
(
bc
);
if
(
s
->
matrix_out_ch
[
mat
]
>
s
->
max_matrix_channel
)
{
if
(
s
->
matrix_out_ch
[
mat
]
>
s
->
max_matrix_channel
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
...
@@ -718,14 +721,14 @@ static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitCo
...
@@ -718,14 +721,14 @@ static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitCo
for
(
ch
=
0
;
ch
<=
max_chan
;
ch
++
)
{
for
(
ch
=
0
;
ch
<=
max_chan
;
ch
++
)
{
int
coeff_val
=
0
;
int
coeff_val
=
0
;
if
(
get_bits1
(
gbp
))
if
(
bitstream_read_bit
(
bc
))
coeff_val
=
get_sbits
(
gbp
,
frac_bits
+
2
);
coeff_val
=
bitstream_read_signed
(
bc
,
frac_bits
+
2
);
s
->
matrix_coeff
[
mat
][
ch
]
=
coeff_val
<<
(
14
-
frac_bits
);
s
->
matrix_coeff
[
mat
][
ch
]
=
coeff_val
<<
(
14
-
frac_bits
);
}
}
if
(
s
->
noise_type
)
if
(
s
->
noise_type
)
s
->
matrix_noise_shift
[
mat
]
=
get_bits
(
gbp
,
4
);
s
->
matrix_noise_shift
[
mat
]
=
bitstream_read
(
bc
,
4
);
else
else
s
->
matrix_noise_shift
[
mat
]
=
0
;
s
->
matrix_noise_shift
[
mat
]
=
0
;
}
}
...
@@ -736,7 +739,7 @@ static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitCo
...
@@ -736,7 +739,7 @@ static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitCo
/** Read channel parameters. */
/** Read channel parameters. */
static
int
read_channel_params
(
MLPDecodeContext
*
m
,
unsigned
int
substr
,
static
int
read_channel_params
(
MLPDecodeContext
*
m
,
unsigned
int
substr
,
GetBitContext
*
gbp
,
unsigned
int
ch
)
BitstreamContext
*
bc
,
unsigned
int
ch
)
{
{
SubStream
*
s
=
&
m
->
substream
[
substr
];
SubStream
*
s
=
&
m
->
substream
[
substr
];
ChannelParams
*
cp
=
&
s
->
channel_params
[
ch
];
ChannelParams
*
cp
=
&
s
->
channel_params
[
ch
];
...
@@ -745,13 +748,13 @@ static int read_channel_params(MLPDecodeContext *m, unsigned int substr,
...
@@ -745,13 +748,13 @@ static int read_channel_params(MLPDecodeContext *m, unsigned int substr,
int
ret
;
int
ret
;
if
(
s
->
param_presence_flags
&
PARAM_FIR
)
if
(
s
->
param_presence_flags
&
PARAM_FIR
)
if
(
get_bits1
(
gbp
))
if
(
bitstream_read_bit
(
bc
))
if
((
ret
=
read_filter_params
(
m
,
gbp
,
substr
,
ch
,
FIR
))
<
0
)
if
((
ret
=
read_filter_params
(
m
,
bc
,
substr
,
ch
,
FIR
))
<
0
)
return
ret
;
return
ret
;
if
(
s
->
param_presence_flags
&
PARAM_IIR
)
if
(
s
->
param_presence_flags
&
PARAM_IIR
)
if
(
get_bits1
(
gbp
))
if
(
bitstream_read_bit
(
bc
))
if
((
ret
=
read_filter_params
(
m
,
gbp
,
substr
,
ch
,
IIR
))
<
0
)
if
((
ret
=
read_filter_params
(
m
,
bc
,
substr
,
ch
,
IIR
))
<
0
)
return
ret
;
return
ret
;
if
(
fir
->
order
+
iir
->
order
>
8
)
{
if
(
fir
->
order
+
iir
->
order
>
8
)
{
...
@@ -774,11 +777,11 @@ static int read_channel_params(MLPDecodeContext *m, unsigned int substr,
...
@@ -774,11 +777,11 @@ static int read_channel_params(MLPDecodeContext *m, unsigned int substr,
fir
->
shift
=
iir
->
shift
;
fir
->
shift
=
iir
->
shift
;
if
(
s
->
param_presence_flags
&
PARAM_HUFFOFFSET
)
if
(
s
->
param_presence_flags
&
PARAM_HUFFOFFSET
)
if
(
get_bits1
(
gbp
))
if
(
bitstream_read_bit
(
bc
))
cp
->
huff_offset
=
get_sbits
(
gbp
,
15
);
cp
->
huff_offset
=
bitstream_read_signed
(
bc
,
15
);
cp
->
codebook
=
get_bits
(
gbp
,
2
);
cp
->
codebook
=
bitstream_read
(
bc
,
2
);
cp
->
huff_lsbs
=
get_bits
(
gbp
,
5
);
cp
->
huff_lsbs
=
bitstream_read
(
bc
,
5
);
if
(
cp
->
huff_lsbs
>
24
)
{
if
(
cp
->
huff_lsbs
>
24
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"Invalid huff_lsbs.
\n
"
);
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"Invalid huff_lsbs.
\n
"
);
...
@@ -793,7 +796,7 @@ static int read_channel_params(MLPDecodeContext *m, unsigned int substr,
...
@@ -793,7 +796,7 @@ static int read_channel_params(MLPDecodeContext *m, unsigned int substr,
/** Read decoding parameters that change more often than those in the restart
/** Read decoding parameters that change more often than those in the restart
* header. */
* header. */
static
int
read_decoding_params
(
MLPDecodeContext
*
m
,
GetBitContext
*
gbp
,
static
int
read_decoding_params
(
MLPDecodeContext
*
m
,
BitstreamContext
*
bc
,
unsigned
int
substr
)
unsigned
int
substr
)
{
{
SubStream
*
s
=
&
m
->
substream
[
substr
];
SubStream
*
s
=
&
m
->
substream
[
substr
];
...
@@ -801,12 +804,12 @@ static int read_decoding_params(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -801,12 +804,12 @@ static int read_decoding_params(MLPDecodeContext *m, GetBitContext *gbp,
int
ret
;
int
ret
;
if
(
s
->
param_presence_flags
&
PARAM_PRESENCE
)
if
(
s
->
param_presence_flags
&
PARAM_PRESENCE
)
if
(
get_bits1
(
gbp
))
if
(
bitstream_read_bit
(
bc
))
s
->
param_presence_flags
=
get_bits
(
gbp
,
8
);
s
->
param_presence_flags
=
bitstream_read
(
bc
,
8
);
if
(
s
->
param_presence_flags
&
PARAM_BLOCKSIZE
)
if
(
s
->
param_presence_flags
&
PARAM_BLOCKSIZE
)
if
(
get_bits1
(
gbp
))
{
if
(
bitstream_read_bit
(
bc
))
{
s
->
blocksize
=
get_bits
(
gbp
,
9
);
s
->
blocksize
=
bitstream_read
(
bc
,
9
);
if
(
s
->
blocksize
<
8
||
s
->
blocksize
>
m
->
access_unit_size
)
{
if
(
s
->
blocksize
<
8
||
s
->
blocksize
>
m
->
access_unit_size
)
{
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"Invalid blocksize."
);
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"Invalid blocksize."
);
s
->
blocksize
=
0
;
s
->
blocksize
=
0
;
...
@@ -815,14 +818,14 @@ static int read_decoding_params(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -815,14 +818,14 @@ static int read_decoding_params(MLPDecodeContext *m, GetBitContext *gbp,
}
}
if
(
s
->
param_presence_flags
&
PARAM_MATRIX
)
if
(
s
->
param_presence_flags
&
PARAM_MATRIX
)
if
(
get_bits1
(
gbp
))
if
(
bitstream_read_bit
(
bc
))
if
((
ret
=
read_matrix_params
(
m
,
substr
,
gbp
))
<
0
)
if
((
ret
=
read_matrix_params
(
m
,
substr
,
bc
))
<
0
)
return
ret
;
return
ret
;
if
(
s
->
param_presence_flags
&
PARAM_OUTSHIFT
)
if
(
s
->
param_presence_flags
&
PARAM_OUTSHIFT
)
if
(
get_bits1
(
gbp
))
{
if
(
bitstream_read_bit
(
bc
))
{
for
(
ch
=
0
;
ch
<=
s
->
max_matrix_channel
;
ch
++
)
for
(
ch
=
0
;
ch
<=
s
->
max_matrix_channel
;
ch
++
)
s
->
output_shift
[
ch
]
=
get_sbits
(
gbp
,
4
);
s
->
output_shift
[
ch
]
=
bitstream_read_signed
(
bc
,
4
);
if
(
substr
==
m
->
max_decoded_substream
)
if
(
substr
==
m
->
max_decoded_substream
)
m
->
dsp
.
mlp_pack_output
=
m
->
dsp
.
mlp_select_pack_output
(
s
->
ch_assign
,
m
->
dsp
.
mlp_pack_output
=
m
->
dsp
.
mlp_select_pack_output
(
s
->
ch_assign
,
s
->
output_shift
,
s
->
output_shift
,
...
@@ -831,18 +834,18 @@ static int read_decoding_params(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -831,18 +834,18 @@ static int read_decoding_params(MLPDecodeContext *m, GetBitContext *gbp,
}
}
if
(
s
->
param_presence_flags
&
PARAM_QUANTSTEP
)
if
(
s
->
param_presence_flags
&
PARAM_QUANTSTEP
)
if
(
get_bits1
(
gbp
))
if
(
bitstream_read_bit
(
bc
))
for
(
ch
=
0
;
ch
<=
s
->
max_channel
;
ch
++
)
{
for
(
ch
=
0
;
ch
<=
s
->
max_channel
;
ch
++
)
{
ChannelParams
*
cp
=
&
s
->
channel_params
[
ch
];
ChannelParams
*
cp
=
&
s
->
channel_params
[
ch
];
s
->
quant_step_size
[
ch
]
=
get_bits
(
gbp
,
4
);
s
->
quant_step_size
[
ch
]
=
bitstream_read
(
bc
,
4
);
cp
->
sign_huff_offset
=
calculate_sign_huff
(
m
,
substr
,
ch
);
cp
->
sign_huff_offset
=
calculate_sign_huff
(
m
,
substr
,
ch
);
}
}
for
(
ch
=
s
->
min_channel
;
ch
<=
s
->
max_channel
;
ch
++
)
for
(
ch
=
s
->
min_channel
;
ch
<=
s
->
max_channel
;
ch
++
)
if
(
get_bits1
(
gbp
))
if
(
bitstream_read_bit
(
bc
))
if
((
ret
=
read_channel_params
(
m
,
substr
,
gbp
,
ch
))
<
0
)
if
((
ret
=
read_channel_params
(
m
,
substr
,
bc
,
ch
))
<
0
)
return
ret
;
return
ret
;
return
0
;
return
0
;
...
@@ -880,7 +883,7 @@ static void filter_channel(MLPDecodeContext *m, unsigned int substr,
...
@@ -880,7 +883,7 @@ static void filter_channel(MLPDecodeContext *m, unsigned int substr,
/** Read a block of PCM residual data (or actual if no filtering active). */
/** Read a block of PCM residual data (or actual if no filtering active). */
static
int
read_block_data
(
MLPDecodeContext
*
m
,
GetBitContext
*
gbp
,
static
int
read_block_data
(
MLPDecodeContext
*
m
,
BitstreamContext
*
bc
,
unsigned
int
substr
)
unsigned
int
substr
)
{
{
SubStream
*
s
=
&
m
->
substream
[
substr
];
SubStream
*
s
=
&
m
->
substream
[
substr
];
...
@@ -888,8 +891,8 @@ static int read_block_data(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -888,8 +891,8 @@ static int read_block_data(MLPDecodeContext *m, GetBitContext *gbp,
int
ret
;
int
ret
;
if
(
s
->
data_check_present
)
{
if
(
s
->
data_check_present
)
{
expected_stream_pos
=
get_bits_count
(
gbp
);
expected_stream_pos
=
bitstream_tell
(
bc
);
expected_stream_pos
+=
get_bits
(
gbp
,
16
);
expected_stream_pos
+=
bitstream_read
(
bc
,
16
);
avpriv_request_sample
(
m
->
avctx
,
avpriv_request_sample
(
m
->
avctx
,
"Substreams with VLC block size check info"
);
"Substreams with VLC block size check info"
);
}
}
...
@@ -903,7 +906,7 @@ static int read_block_data(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -903,7 +906,7 @@ static int read_block_data(MLPDecodeContext *m, GetBitContext *gbp,
s
->
blocksize
*
sizeof
(
m
->
bypassed_lsbs
[
0
]));
s
->
blocksize
*
sizeof
(
m
->
bypassed_lsbs
[
0
]));
for
(
i
=
0
;
i
<
s
->
blocksize
;
i
++
)
for
(
i
=
0
;
i
<
s
->
blocksize
;
i
++
)
if
((
ret
=
read_huff_channels
(
m
,
gbp
,
substr
,
i
))
<
0
)
if
((
ret
=
read_huff_channels
(
m
,
bc
,
substr
,
i
))
<
0
)
return
ret
;
return
ret
;
for
(
ch
=
s
->
min_channel
;
ch
<=
s
->
max_channel
;
ch
++
)
for
(
ch
=
s
->
min_channel
;
ch
<=
s
->
max_channel
;
ch
++
)
...
@@ -912,9 +915,9 @@ static int read_block_data(MLPDecodeContext *m, GetBitContext *gbp,
...
@@ -912,9 +915,9 @@ static int read_block_data(MLPDecodeContext *m, GetBitContext *gbp,
s
->
blockpos
+=
s
->
blocksize
;
s
->
blockpos
+=
s
->
blocksize
;
if
(
s
->
data_check_present
)
{
if
(
s
->
data_check_present
)
{
if
(
get_bits_count
(
gbp
)
!=
expected_stream_pos
)
if
(
bitstream_tell
(
bc
)
!=
expected_stream_pos
)
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"block data length mismatch
\n
"
);
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"block data length mismatch
\n
"
);
skip_bits
(
gbp
,
8
);
bitstream_skip
(
bc
,
8
);
}
}
return
0
;
return
0
;
...
@@ -1074,7 +1077,7 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
...
@@ -1074,7 +1077,7 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
const
uint8_t
*
buf
=
avpkt
->
data
;
const
uint8_t
*
buf
=
avpkt
->
data
;
int
buf_size
=
avpkt
->
size
;
int
buf_size
=
avpkt
->
size
;
MLPDecodeContext
*
m
=
avctx
->
priv_data
;
MLPDecodeContext
*
m
=
avctx
->
priv_data
;
GetBitContext
gb
;
BitstreamContext
bc
;
unsigned
int
length
,
substr
;
unsigned
int
length
,
substr
;
unsigned
int
substream_start
;
unsigned
int
substream_start
;
unsigned
int
header_size
=
4
;
unsigned
int
header_size
=
4
;
...
@@ -1092,11 +1095,11 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
...
@@ -1092,11 +1095,11 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
if
(
length
<
4
||
length
>
buf_size
)
if
(
length
<
4
||
length
>
buf_size
)
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
init_get_bits
(
&
gb
,
(
buf
+
4
),
(
length
-
4
)
*
8
);
bitstream_init
(
&
bc
,
(
buf
+
4
),
(
length
-
4
)
*
8
);
m
->
is_major_sync_unit
=
0
;
m
->
is_major_sync_unit
=
0
;
if
(
show_bits_long
(
&
gb
,
31
)
==
(
0xf8726fba
>>
1
))
{
if
(
bitstream_peek
(
&
bc
,
31
)
==
(
0xf8726fba
>>
1
))
{
if
(
read_major_sync
(
m
,
&
gb
)
<
0
)
if
(
read_major_sync
(
m
,
&
bc
)
<
0
)
goto
error
;
goto
error
;
m
->
is_major_sync_unit
=
1
;
m
->
is_major_sync_unit
=
1
;
header_size
+=
m
->
major_sync_header_size
;
header_size
+=
m
->
major_sync_header_size
;
...
@@ -1114,12 +1117,12 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
...
@@ -1114,12 +1117,12 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
for
(
substr
=
0
;
substr
<
m
->
num_substreams
;
substr
++
)
{
for
(
substr
=
0
;
substr
<
m
->
num_substreams
;
substr
++
)
{
int
extraword_present
,
checkdata_present
,
end
,
nonrestart_substr
;
int
extraword_present
,
checkdata_present
,
end
,
nonrestart_substr
;
extraword_present
=
get_bits1
(
&
gb
);
extraword_present
=
bitstream_read_bit
(
&
bc
);
nonrestart_substr
=
get_bits1
(
&
gb
);
nonrestart_substr
=
bitstream_read_bit
(
&
bc
);
checkdata_present
=
get_bits1
(
&
gb
);
checkdata_present
=
bitstream_read_bit
(
&
bc
);
skip_bits1
(
&
gb
);
bitstream_skip
(
&
bc
,
1
);
end
=
get_bits
(
&
gb
,
12
)
*
2
;
end
=
bitstream_read
(
&
bc
,
12
)
*
2
;
substr_header_size
+=
2
;
substr_header_size
+=
2
;
...
@@ -1128,7 +1131,7 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
...
@@ -1128,7 +1131,7 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"There must be no extraword for MLP.
\n
"
);
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"There must be no extraword for MLP.
\n
"
);
goto
error
;
goto
error
;
}
}
skip_bits
(
&
gb
,
16
);
bitstream_skip
(
&
bc
,
16
);
substr_header_size
+=
2
;
substr_header_size
+=
2
;
}
}
...
@@ -1172,47 +1175,47 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
...
@@ -1172,47 +1175,47 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
for
(
substr
=
0
;
substr
<=
m
->
max_decoded_substream
;
substr
++
)
{
for
(
substr
=
0
;
substr
<=
m
->
max_decoded_substream
;
substr
++
)
{
SubStream
*
s
=
&
m
->
substream
[
substr
];
SubStream
*
s
=
&
m
->
substream
[
substr
];
init_get_bits
(
&
gb
,
buf
,
substream_data_len
[
substr
]
*
8
);
bitstream_init
(
&
bc
,
buf
,
substream_data_len
[
substr
]
*
8
);
m
->
matrix_changed
=
0
;
m
->
matrix_changed
=
0
;
memset
(
m
->
filter_changed
,
0
,
sizeof
(
m
->
filter_changed
));
memset
(
m
->
filter_changed
,
0
,
sizeof
(
m
->
filter_changed
));
s
->
blockpos
=
0
;
s
->
blockpos
=
0
;
do
{
do
{
if
(
get_bits1
(
&
gb
))
{
if
(
bitstream_read_bit
(
&
bc
))
{
if
(
get_bits1
(
&
gb
))
{
if
(
bitstream_read_bit
(
&
bc
))
{
/* A restart header should be present. */
/* A restart header should be present. */
if
(
read_restart_header
(
m
,
&
gb
,
buf
,
substr
)
<
0
)
if
(
read_restart_header
(
m
,
&
bc
,
buf
,
substr
)
<
0
)
goto
next_substr
;
goto
next_substr
;
s
->
restart_seen
=
1
;
s
->
restart_seen
=
1
;
}
}
if
(
!
s
->
restart_seen
)
if
(
!
s
->
restart_seen
)
goto
next_substr
;
goto
next_substr
;
if
(
read_decoding_params
(
m
,
&
gb
,
substr
)
<
0
)
if
(
read_decoding_params
(
m
,
&
bc
,
substr
)
<
0
)
goto
next_substr
;
goto
next_substr
;
}
}
if
(
!
s
->
restart_seen
)
if
(
!
s
->
restart_seen
)
goto
next_substr
;
goto
next_substr
;
if
((
ret
=
read_block_data
(
m
,
&
gb
,
substr
))
<
0
)
if
((
ret
=
read_block_data
(
m
,
&
bc
,
substr
))
<
0
)
return
ret
;
return
ret
;
if
(
get_bits_count
(
&
gb
)
>=
substream_data_len
[
substr
]
*
8
)
if
(
bitstream_tell
(
&
bc
)
>=
substream_data_len
[
substr
]
*
8
)
goto
substream_length_mismatch
;
goto
substream_length_mismatch
;
}
while
(
!
get_bits1
(
&
gb
));
}
while
(
!
bitstream_read_bit
(
&
bc
));
skip_bits
(
&
gb
,
(
-
get_bits_count
(
&
gb
))
&
15
);
bitstream_skip
(
&
bc
,
(
-
bitstream_tell
(
&
bc
))
&
15
);
if
(
substream_data_len
[
substr
]
*
8
-
get_bits_count
(
&
gb
)
>=
32
)
{
if
(
substream_data_len
[
substr
]
*
8
-
bitstream_tell
(
&
bc
)
>=
32
)
{
int
shorten_by
;
int
shorten_by
;
if
(
get_bits
(
&
gb
,
16
)
!=
0xD234
)
if
(
bitstream_read
(
&
bc
,
16
)
!=
0xD234
)
return
AVERROR_INVALIDDATA
;
return
AVERROR_INVALIDDATA
;
shorten_by
=
get_bits
(
&
gb
,
16
);
shorten_by
=
bitstream_read
(
&
bc
,
16
);
if
(
m
->
avctx
->
codec_id
==
AV_CODEC_ID_TRUEHD
&&
shorten_by
&
0x2000
)
if
(
m
->
avctx
->
codec_id
==
AV_CODEC_ID_TRUEHD
&&
shorten_by
&
0x2000
)
s
->
blockpos
-=
FFMIN
(
shorten_by
&
0x1FFF
,
s
->
blockpos
);
s
->
blockpos
-=
FFMIN
(
shorten_by
&
0x1FFF
,
s
->
blockpos
);
else
if
(
m
->
avctx
->
codec_id
==
AV_CODEC_ID_MLP
&&
shorten_by
!=
0xD234
)
else
if
(
m
->
avctx
->
codec_id
==
AV_CODEC_ID_MLP
&&
shorten_by
!=
0xD234
)
...
@@ -1225,19 +1228,19 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
...
@@ -1225,19 +1228,19 @@ static int read_access_unit(AVCodecContext *avctx, void* data,
if
(
substream_parity_present
[
substr
])
{
if
(
substream_parity_present
[
substr
])
{
uint8_t
parity
,
checksum
;
uint8_t
parity
,
checksum
;
if
(
substream_data_len
[
substr
]
*
8
-
get_bits_count
(
&
gb
)
!=
16
)
if
(
substream_data_len
[
substr
]
*
8
-
bitstream_tell
(
&
bc
)
!=
16
)
goto
substream_length_mismatch
;
goto
substream_length_mismatch
;
parity
=
ff_mlp_calculate_parity
(
buf
,
substream_data_len
[
substr
]
-
2
);
parity
=
ff_mlp_calculate_parity
(
buf
,
substream_data_len
[
substr
]
-
2
);
checksum
=
ff_mlp_checksum8
(
buf
,
substream_data_len
[
substr
]
-
2
);
checksum
=
ff_mlp_checksum8
(
buf
,
substream_data_len
[
substr
]
-
2
);
if
((
get_bits
(
&
gb
,
8
)
^
parity
)
!=
0xa9
)
if
((
bitstream_read
(
&
bc
,
8
)
^
parity
)
!=
0xa9
)
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"Substream %d parity check failed.
\n
"
,
substr
);
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"Substream %d parity check failed.
\n
"
,
substr
);
if
(
get_bits
(
&
gb
,
8
)
!=
checksum
)
if
(
bitstream_read
(
&
bc
,
8
)
!=
checksum
)
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"Substream %d checksum failed.
\n
"
,
substr
);
av_log
(
m
->
avctx
,
AV_LOG_ERROR
,
"Substream %d checksum failed.
\n
"
,
substr
);
}
}
if
(
substream_data_len
[
substr
]
*
8
!=
get_bits_count
(
&
gb
))
if
(
substream_data_len
[
substr
]
*
8
!=
bitstream_tell
(
&
bc
))
goto
substream_length_mismatch
;
goto
substream_length_mismatch
;
next_substr:
next_substr:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment