2 * dtsdec.c : free DTS Coherent Acoustics stream decoder.
3 * Copyright (C) 2004 Benjamin Zores <ben@geexbox.org>
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 2 of the License, or
10 * (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
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
28 #define BUFFER_SIZE 18726
29 #define HEADER_SIZE 14
32 #define CONVERT_LEVEL (1 << 26)
33 #define CONVERT_BIAS 0
35 #define CONVERT_LEVEL 1
36 #define CONVERT_BIAS 384
39 typedef struct DTSContext {
41 uint8_t buf[BUFFER_SIZE];
54 return (i > 32767) ? 32767 : ((i < -32768) ? -32768 : i);
58 convert2s16_2(sample_t * _f, int16_t * s16)
61 int32_t *f = (int32_t *) _f;
63 for(i = 0; i < 256; i++) {
64 s16[2 * i] = convert(f[i]);
65 s16[2 * i + 1] = convert(f[i + 256]);
70 convert2s16_4(sample_t * _f, int16_t * s16)
73 int32_t *f = (int32_t *) _f;
75 for(i = 0; i < 256; i++) {
76 s16[4 * i] = convert(f[i]);
77 s16[4 * i + 1] = convert(f[i + 256]);
78 s16[4 * i + 2] = convert(f[i + 512]);
79 s16[4 * i + 3] = convert(f[i + 768]);
84 convert2s16_5(sample_t * _f, int16_t * s16)
87 int32_t *f = (int32_t *) _f;
89 for(i = 0; i < 256; i++) {
90 s16[5 * i] = convert(f[i]);
91 s16[5 * i + 1] = convert(f[i + 256]);
92 s16[5 * i + 2] = convert(f[i + 512]);
93 s16[5 * i + 3] = convert(f[i + 768]);
94 s16[5 * i + 4] = convert(f[i + 1024]);
99 convert2s16_multi(sample_t * _f, int16_t * s16, int flags)
102 int32_t *f = (int32_t *) _f;
106 for(i = 0; i < 256; i++) {
107 s16[5 * i] = s16[5 * i + 1] = s16[5 * i + 2] = s16[5 * i + 3] =
109 s16[5 * i + 4] = convert(f[i]);
115 convert2s16_2(_f, s16);
118 for(i = 0; i < 256; i++) {
119 s16[5 * i] = convert(f[i]);
120 s16[5 * i + 1] = convert(f[i + 512]);
121 s16[5 * i + 2] = s16[5 * i + 3] = 0;
122 s16[5 * i + 4] = convert(f[i + 256]);
126 convert2s16_4(_f, s16);
129 convert2s16_5(_f, s16);
131 case DTS_MONO | DTS_LFE:
132 for(i = 0; i < 256; i++) {
133 s16[6 * i] = s16[6 * i + 1] = s16[6 * i + 2] = s16[6 * i + 3] =
135 s16[6 * i + 4] = convert(f[i + 256]);
136 s16[6 * i + 5] = convert(f[i]);
139 case DTS_CHANNEL | DTS_LFE:
140 case DTS_STEREO | DTS_LFE:
141 case DTS_DOLBY | DTS_LFE:
142 for(i = 0; i < 256; i++) {
143 s16[6 * i] = convert(f[i + 256]);
144 s16[6 * i + 1] = convert(f[i + 512]);
145 s16[6 * i + 2] = s16[6 * i + 3] = s16[6 * i + 4] = 0;
146 s16[6 * i + 5] = convert(f[i]);
149 case DTS_3F | DTS_LFE:
150 for(i = 0; i < 256; i++) {
151 s16[6 * i] = convert(f[i + 256]);
152 s16[6 * i + 1] = convert(f[i + 768]);
153 s16[6 * i + 2] = s16[6 * i + 3] = 0;
154 s16[6 * i + 4] = convert(f[i + 512]);
155 s16[6 * i + 5] = convert(f[i]);
158 case DTS_2F2R | DTS_LFE:
159 for(i = 0; i < 256; i++) {
160 s16[6 * i] = convert(f[i + 256]);
161 s16[6 * i + 1] = convert(f[i + 512]);
162 s16[6 * i + 2] = convert(f[i + 768]);
163 s16[6 * i + 3] = convert(f[i + 1024]);
165 s16[6 * i + 5] = convert(f[i]);
168 case DTS_3F2R | DTS_LFE:
169 for(i = 0; i < 256; i++) {
170 s16[6 * i] = convert(f[i + 256]);
171 s16[6 * i + 1] = convert(f[i + 768]);
172 s16[6 * i + 2] = convert(f[i + 1024]);
173 s16[6 * i + 3] = convert(f[i + 1280]);
174 s16[6 * i + 4] = convert(f[i + 512]);
175 s16[6 * i + 5] = convert(f[i]);
182 channels_multi(int flags)
186 else if(flags & 1) /* center channel */
188 else if((flags & DTS_CHANNEL_MASK) == DTS_2F2R)
195 dts_decode_frame(AVCodecContext * avctx, void *data, int *data_size,
196 uint8_t * buff, int buff_size)
198 DTSContext *s = avctx->priv_data;
199 uint8_t *start = buff;
200 uint8_t *end = buff + buff_size;
201 int16_t *out_samples = data;
219 if(len > s->bufpos - s->bufptr)
220 len = s->bufpos - s->bufptr;
221 memcpy(s->bufptr, start, len);
224 if(s->bufptr != s->bufpos)
226 if(s->bufpos != s->buf + HEADER_SIZE)
229 length = dts_syncinfo(s->state, s->buf, &flags, &sample_rate,
230 &bit_rate, &frame_length);
232 av_log(NULL, AV_LOG_INFO, "skip\n");
233 for(s->bufptr = s->buf; s->bufptr < s->buf + HEADER_SIZE - 1; s->bufptr++)
234 s->bufptr[0] = s->bufptr[1];
237 s->bufpos = s->buf + length;
240 flags = 2; /* ???????????? */
241 level = CONVERT_LEVEL;
244 flags |= DTS_ADJUST_LEVEL;
245 if(dts_frame(s->state, s->buf, &flags, &level, bias)) {
246 av_log(avctx, AV_LOG_ERROR, "dts_frame() failed\n");
250 avctx->sample_rate = sample_rate;
251 avctx->channels = channels_multi(flags);
252 avctx->bit_rate = bit_rate;
254 for(i = 0; i < dts_blocks_num(s->state); i++) {
257 if(dts_block(s->state)) {
258 av_log(avctx, AV_LOG_ERROR, "dts_block() failed\n");
262 chans = channels_multi(flags);
263 convert2s16_multi(dts_samples(s->state), out_samples,
264 flags & (DTS_CHANNEL_MASK | DTS_LFE));
266 out_samples += 256 * chans;
267 *data_size += 256 * sizeof(int16_t) * chans;
272 s->bufpos = s->buf + HEADER_SIZE;
277 dts_decode_init(AVCodecContext * avctx)
279 DTSContext *s = avctx->priv_data;
281 s->bufpos = s->buf + HEADER_SIZE;
282 s->state = dts_init(0);
290 dts_decode_end(AVCodecContext * avctx)
292 DTSContext *s = avctx->priv_data;
297 AVCodec dts_decoder = {