FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
tak.c
Go to the documentation of this file.
1 /*
2  * TAK common code
3  * Copyright (c) 2012 Paul B Mahol
4  *
5  * This file is part of FFmpeg.
6  *
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.
11  *
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.
16  *
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
20  */
21 
22 #include "libavutil/crc.h"
23 #include "libavutil/intreadwrite.h"
24 
25 #define BITSTREAM_READER_LE
26 #include "tak.h"
27 
28 static const int64_t tak_channel_layouts[] = {
29  0,
48 };
49 
50 static const uint16_t frame_duration_type_quants[] = {
51  3, 4, 6, 8, 4096, 8192, 16384, 512, 1024, 2048,
52 };
53 
55 {
56  int nb_samples, max_nb_samples;
57 
58  if (type <= TAK_FST_250ms) {
59  nb_samples = sample_rate * frame_duration_type_quants[type] >>
61  max_nb_samples = 16384;
62  } else if (type < FF_ARRAY_ELEMS(frame_duration_type_quants)) {
63  nb_samples = frame_duration_type_quants[type];
64  max_nb_samples = sample_rate *
67  } else {
68  return AVERROR_INVALIDDATA;
69  }
70 
71  if (nb_samples <= 0 || nb_samples > max_nb_samples)
72  return AVERROR_INVALIDDATA;
73 
74  return nb_samples;
75 }
76 
77 int ff_tak_check_crc(const uint8_t *buf, unsigned int buf_size)
78 {
79  uint32_t crc, CRC;
80 
81  if (buf_size < 4)
82  return AVERROR_INVALIDDATA;
83  buf_size -= 3;
84 
85  CRC = AV_RB24(buf + buf_size);
86  crc = av_crc(av_crc_get_table(AV_CRC_24_IEEE), 0xCE04B7U, buf, buf_size);
87  if (CRC != crc)
88  return AVERROR_INVALIDDATA;
89 
90  return 0;
91 }
92 
94 {
95  uint64_t channel_mask = 0;
96  int frame_type, i;
97 
100 
101  frame_type = get_bits(gb, TAK_SIZE_FRAME_DURATION_BITS);
103 
107  s->bps = get_bits(gb, TAK_FORMAT_BPS_BITS) +
108  TAK_BPS_MIN;
111 
112  if (get_bits1(gb)) {
114  if (get_bits1(gb)) {
115  for (i = 0; i < s->channels; i++) {
117 
118  if (value < FF_ARRAY_ELEMS(tak_channel_layouts))
119  channel_mask |= tak_channel_layouts[value];
120  }
121  }
122  }
123 
124  s->ch_layout = channel_mask;
125  s->frame_samples = tak_get_nb_samples(s->sample_rate, frame_type);
126 }
127 
129 {
130  GetBitContext gb;
131  int ret = init_get_bits8(&gb, buf, size);
132 
133  if (ret < 0)
134  return AVERROR_INVALIDDATA;
135 
136  ff_tak_parse_streaminfo(s, &gb);
137 
138  return 0;
139 }
140 
142  TAKStreamInfo *ti, int log_level_offset)
143 {
145  av_log(avctx, AV_LOG_ERROR + log_level_offset, "missing sync id\n");
146  return AVERROR_INVALIDDATA;
147  }
148 
151 
152  if (ti->flags & TAK_FRAME_FLAG_IS_LAST) {
154  skip_bits(gb, 2);
155  } else {
156  ti->last_frame_samples = 0;
157  }
158 
159  if (ti->flags & TAK_FRAME_FLAG_HAS_INFO) {
160  ff_tak_parse_streaminfo(ti, gb);
161 
162  if (get_bits(gb, 6))
163  skip_bits(gb, 25);
164  align_get_bits(gb);
165  }
166 
168  return AVERROR_INVALIDDATA;
169 
170  skip_bits(gb, 24);
171 
172  return 0;
173 }
static const uint16_t frame_duration_type_quants[]
Definition: tak.c:50
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
int ff_tak_decode_frame_header(AVCodecContext *avctx, GetBitContext *gb, TAKStreamInfo *ti, int log_level_offset)
Validate and decode a frame header.
Definition: tak.c:141
#define AV_CH_TOP_FRONT_RIGHT
int channels
Definition: tak.h:133
#define TAK_FORMAT_BPS_BITS
Definition: tak.h:37
#define TAK_BPS_MIN
Definition: tak.h:50
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:381
#define AV_CH_TOP_FRONT_LEFT
#define TAK_FRAME_HEADER_NO_BITS
Definition: tak.h:55
static const int64_t tak_channel_layouts[]
Definition: tak.c:28
#define AV_CH_TOP_FRONT_CENTER
#define TAK_FRAME_FLAG_IS_LAST
Definition: tak.h:60
uint64_t ch_layout
Definition: tak.h:138
#define TAK_SIZE_FRAME_DURATION_BITS
Definition: tak.h:41
#define TAK_FORMAT_DATA_TYPE_BITS
Definition: tak.h:35
void ff_tak_parse_streaminfo(TAKStreamInfo *s, GetBitContext *gb)
Definition: tak.c:93
uint8_t
#define AV_CH_TOP_BACK_LEFT
int flags
Definition: tak.h:129
int ff_tak_check_crc(const uint8_t *buf, unsigned int buf_size)
Definition: tak.c:77
#define AV_CH_TOP_BACK_CENTER
#define AV_CH_LOW_FREQUENCY
#define TAK_FRAME_HEADER_FLAGS_BITS
Definition: tak.h:51
#define TAK_FRAME_FLAG_HAS_INFO
Definition: tak.h:61
Public header for CRC hash function implementation.
#define AV_CH_BACK_LEFT
ptrdiff_t size
Definition: opengl_enc.c:101
int data_type
Definition: tak.h:131
#define av_log(a,...)
#define TAK_FRAME_HEADER_SYNC_ID
Definition: tak.h:52
static uint64_t get_bits64(GetBitContext *s, int n)
Read 0-64 bits.
Definition: get_bits.h:557
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
int last_frame_samples
Definition: tak.h:137
#define TAK_SIZE_SAMPLES_NUM_BITS
Definition: tak.h:42
int bps
Definition: tak.h:134
int64_t samples
Definition: tak.h:139
#define AV_CH_TOP_CENTER
GLsizei GLboolean const GLfloat * value
Definition: opengl_enc.c:109
uint32_t av_crc(const AVCRC *ctx, uint32_t crc, const uint8_t *buffer, size_t length)
Calculate the CRC of a block.
Definition: crc.c:392
#define s(width, name)
Definition: cbs_vp9.c:257
#define AV_CH_FRONT_LEFT_OF_CENTER
#define AV_CH_FRONT_CENTER
#define TAK_CHANNELS_MIN
Definition: tak.h:49
#define FF_ARRAY_ELEMS(a)
#define AV_CH_FRONT_RIGHT_OF_CENTER
sample_rate
#define TAK_FRAME_FLAG_HAS_METADATA
Definition: tak.h:62
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_WB32 unsigned int_TMPL AV_RB24
Definition: bytestream.h:87
static int init_get_bits8(GetBitContext *s, const uint8_t *buffer, int byte_size)
Initialize GetBitContext.
Definition: get_bits.h:650
main external API structure.
Definition: avcodec.h:1533
#define AV_CH_FRONT_LEFT
enum TAKCodecType codec
Definition: tak.h:130
void * buf
Definition: avisynth_c.h:690
GLint GLenum type
Definition: opengl_enc.c:105
static unsigned int get_bits1(GetBitContext *s)
Definition: get_bits.h:487
#define AV_CH_TOP_BACK_RIGHT
static void skip_bits(GetBitContext *s, int n)
Definition: get_bits.h:460
#define TAK_SAMPLE_RATE_MIN
Definition: tak.h:48
int frame_samples
Definition: tak.h:136
TAK (Tom's lossless Audio Kompressor) decoder/demuxer common functions.
int sample_rate
Definition: tak.h:132
#define TAK_ENCODER_CODEC_BITS
Definition: tak.h:45
int frame_num
Definition: tak.h:135
#define AV_CH_BACK_CENTER
#define AV_CH_SIDE_RIGHT
TAKFrameSizeType
Definition: tak.h:115
const AVCRC * av_crc_get_table(AVCRCId crc_id)
Get an initialized standard CRC table.
Definition: crc.c:374
#define TAK_ENCODER_PROFILE_BITS
Definition: tak.h:46
#define TAK_FORMAT_CH_LAYOUT_BITS
Definition: tak.h:40
#define TAK_FRAME_HEADER_SAMPLE_COUNT_BITS
Definition: tak.h:54
#define TAK_FRAME_HEADER_SYNC_ID_BITS
Definition: tak.h:53
#define TAK_FRAME_DURATION_QUANT_SHIFT
Definition: tak.h:56
#define TAK_FORMAT_VALID_BITS
Definition: tak.h:39
static const uint8_t * align_get_bits(GetBitContext *s)
Definition: get_bits.h:658
#define AV_CH_FRONT_RIGHT
int avpriv_tak_parse_streaminfo(TAKStreamInfo *s, const uint8_t *buf, int size)
Parse the Streaminfo metadata block.
Definition: tak.c:128
#define AV_CH_SIDE_LEFT
#define TAK_FORMAT_CHANNEL_BITS
Definition: tak.h:38
static int tak_get_nb_samples(int sample_rate, enum TAKFrameSizeType type)
Definition: tak.c:54
#define TAK_FORMAT_SAMPLE_RATE_BITS
Definition: tak.h:36
#define AV_CH_BACK_RIGHT