3 * Copyright (c) 2012 Paul B Mahol
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 #include "libavutil/crc.h"
23 #include "libavutil/intreadwrite.h"
26 static const int64_t tak_channel_layouts
[] = {
34 AV_CH_FRONT_LEFT_OF_CENTER
,
35 AV_CH_FRONT_RIGHT_OF_CENTER
,
41 AV_CH_TOP_FRONT_CENTER
,
42 AV_CH_TOP_FRONT_RIGHT
,
44 AV_CH_TOP_BACK_CENTER
,
48 static const uint16_t frame_duration_type_quants
[] = {
49 3, 4, 6, 8, 4096, 8192, 16384, 512, 1024, 2048,
52 static int tak_get_nb_samples(int sample_rate
, enum TAKFrameSizeType type
)
54 int nb_samples
, max_nb_samples
;
56 if (type
<= TAK_FST_250ms
) {
57 nb_samples
= sample_rate
* frame_duration_type_quants
[type
] >>
58 TAK_FRAME_DURATION_QUANT_SHIFT
;
59 max_nb_samples
= 16384;
60 } else if (type
< FF_ARRAY_ELEMS(frame_duration_type_quants
)) {
61 nb_samples
= frame_duration_type_quants
[type
];
62 max_nb_samples
= sample_rate
*
63 frame_duration_type_quants
[TAK_FST_250ms
] >>
64 TAK_FRAME_DURATION_QUANT_SHIFT
;
66 return AVERROR_INVALIDDATA
;
69 if (nb_samples
<= 0 || nb_samples
> max_nb_samples
)
70 return AVERROR_INVALIDDATA
;
75 int ff_tak_check_crc(const uint8_t *buf
, unsigned int buf_size
)
80 return AVERROR_INVALIDDATA
;
83 CRC
= AV_RB24(buf
+ buf_size
);
84 crc
= av_crc(av_crc_get_table(AV_CRC_24_IEEE
), 0xCE04B7U
, buf
, buf_size
);
86 return AVERROR_INVALIDDATA
;
91 void avpriv_tak_parse_streaminfo(GetBitContext
*gb
, TAKStreamInfo
*s
)
93 uint64_t channel_mask
= 0;
96 s
->codec
= get_bits(gb
, TAK_ENCODER_CODEC_BITS
);
97 skip_bits(gb
, TAK_ENCODER_PROFILE_BITS
);
99 frame_type
= get_bits(gb
, TAK_SIZE_FRAME_DURATION_BITS
);
100 s
->samples
= get_bits64(gb
, TAK_SIZE_SAMPLES_NUM_BITS
);
102 s
->data_type
= get_bits(gb
, TAK_FORMAT_DATA_TYPE_BITS
);
103 s
->sample_rate
= get_bits(gb
, TAK_FORMAT_SAMPLE_RATE_BITS
) +
105 s
->bps
= get_bits(gb
, TAK_FORMAT_BPS_BITS
) +
107 s
->channels
= get_bits(gb
, TAK_FORMAT_CHANNEL_BITS
) +
111 skip_bits(gb
, TAK_FORMAT_VALID_BITS
);
113 for (i
= 0; i
< s
->channels
; i
++) {
114 int value
= get_bits(gb
, TAK_FORMAT_CH_LAYOUT_BITS
);
116 if (value
< FF_ARRAY_ELEMS(tak_channel_layouts
))
117 channel_mask
|= tak_channel_layouts
[value
];
122 s
->ch_layout
= channel_mask
;
123 s
->frame_samples
= tak_get_nb_samples(s
->sample_rate
, frame_type
);
126 int ff_tak_decode_frame_header(AVCodecContext
*avctx
, GetBitContext
*gb
,
127 TAKStreamInfo
*ti
, int log_level_offset
)
129 if (get_bits(gb
, TAK_FRAME_HEADER_SYNC_ID_BITS
) != TAK_FRAME_HEADER_SYNC_ID
) {
130 av_log(avctx
, AV_LOG_ERROR
+ log_level_offset
, "missing sync id\n");
131 return AVERROR_INVALIDDATA
;
134 ti
->flags
= get_bits(gb
, TAK_FRAME_HEADER_FLAGS_BITS
);
135 ti
->frame_num
= get_bits(gb
, TAK_FRAME_HEADER_NO_BITS
);
137 if (ti
->flags
& TAK_FRAME_FLAG_IS_LAST
) {
138 ti
->last_frame_samples
= get_bits(gb
, TAK_FRAME_HEADER_SAMPLE_COUNT_BITS
) + 1;
141 ti
->last_frame_samples
= 0;
144 if (ti
->flags
& TAK_FRAME_FLAG_HAS_INFO
) {
145 avpriv_tak_parse_streaminfo(gb
, ti
);
152 if (ti
->flags
& TAK_FRAME_FLAG_HAS_METADATA
)
153 return AVERROR_INVALIDDATA
;