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
22 #ifdef HAVE_AV_CONFIG_H
23 #undef HAVE_AV_CONFIG_H
36 #define BUFFER_SIZE 18726
37 #define HEADER_SIZE 14
40 #define CONVERT_LEVEL (1 << 26)
41 #define CONVERT_BIAS 0
43 #define CONVERT_LEVEL 1
44 #define CONVERT_BIAS 384
55 return (i
> 32767) ?
32767 : ((i
< -32768) ?
-32768 : i
);
59 convert2s16_2(sample_t
* _f
, int16_t * s16
)
62 int32_t *f
= (int32_t *) _f
;
64 for(i
= 0; i
< 256; i
++) {
65 s16
[2 * i
] = convert(f
[i
]);
66 s16
[2 * i
+ 1] = convert(f
[i
+ 256]);
71 convert2s16_4(sample_t
* _f
, int16_t * s16
)
74 int32_t *f
= (int32_t *) _f
;
76 for(i
= 0; i
< 256; i
++) {
77 s16
[4 * i
] = convert(f
[i
]);
78 s16
[4 * i
+ 1] = convert(f
[i
+ 256]);
79 s16
[4 * i
+ 2] = convert(f
[i
+ 512]);
80 s16
[4 * i
+ 3] = convert(f
[i
+ 768]);
85 convert2s16_5(sample_t
* _f
, int16_t * s16
)
88 int32_t *f
= (int32_t *) _f
;
90 for(i
= 0; i
< 256; i
++) {
91 s16
[5 * i
] = convert(f
[i
]);
92 s16
[5 * i
+ 1] = convert(f
[i
+ 256]);
93 s16
[5 * i
+ 2] = convert(f
[i
+ 512]);
94 s16
[5 * i
+ 3] = convert(f
[i
+ 768]);
95 s16
[5 * i
+ 4] = convert(f
[i
+ 1024]);
100 convert2s16_multi(sample_t
* _f
, int16_t * s16
, int flags
)
103 int32_t *f
= (int32_t *) _f
;
107 for(i
= 0; i
< 256; i
++) {
108 s16
[5 * i
] = s16
[5 * i
+ 1] = s16
[5 * i
+ 2] = s16
[5 * i
+ 3] =
110 s16
[5 * i
+ 4] = convert(f
[i
]);
116 convert2s16_2(_f
, s16
);
119 for(i
= 0; i
< 256; i
++) {
120 s16
[5 * i
] = convert(f
[i
]);
121 s16
[5 * i
+ 1] = convert(f
[i
+ 512]);
122 s16
[5 * i
+ 2] = s16
[5 * i
+ 3] = 0;
123 s16
[5 * i
+ 4] = convert(f
[i
+ 256]);
127 convert2s16_4(_f
, s16
);
130 convert2s16_5(_f
, s16
);
132 case DTS_MONO
| DTS_LFE
:
133 for(i
= 0; i
< 256; i
++) {
134 s16
[6 * i
] = s16
[6 * i
+ 1] = s16
[6 * i
+ 2] = s16
[6 * i
+ 3] =
136 s16
[6 * i
+ 4] = convert(f
[i
+ 256]);
137 s16
[6 * i
+ 5] = convert(f
[i
]);
140 case DTS_CHANNEL
| DTS_LFE
:
141 case DTS_STEREO
| DTS_LFE
:
142 case DTS_DOLBY
| DTS_LFE
:
143 for(i
= 0; i
< 256; i
++) {
144 s16
[6 * i
] = convert(f
[i
+ 256]);
145 s16
[6 * i
+ 1] = convert(f
[i
+ 512]);
146 s16
[6 * i
+ 2] = s16
[6 * i
+ 3] = s16
[6 * i
+ 4] = 0;
147 s16
[6 * i
+ 5] = convert(f
[i
]);
150 case DTS_3F
| DTS_LFE
:
151 for(i
= 0; i
< 256; i
++) {
152 s16
[6 * i
] = convert(f
[i
+ 256]);
153 s16
[6 * i
+ 1] = convert(f
[i
+ 768]);
154 s16
[6 * i
+ 2] = s16
[6 * i
+ 3] = 0;
155 s16
[6 * i
+ 4] = convert(f
[i
+ 512]);
156 s16
[6 * i
+ 5] = convert(f
[i
]);
159 case DTS_2F2R
| DTS_LFE
:
160 for(i
= 0; i
< 256; i
++) {
161 s16
[6 * i
] = convert(f
[i
+ 256]);
162 s16
[6 * i
+ 1] = convert(f
[i
+ 512]);
163 s16
[6 * i
+ 2] = convert(f
[i
+ 768]);
164 s16
[6 * i
+ 3] = convert(f
[i
+ 1024]);
166 s16
[6 * i
+ 5] = convert(f
[i
]);
169 case DTS_3F2R
| DTS_LFE
:
170 for(i
= 0; i
< 256; i
++) {
171 s16
[6 * i
] = convert(f
[i
+ 256]);
172 s16
[6 * i
+ 1] = convert(f
[i
+ 768]);
173 s16
[6 * i
+ 2] = convert(f
[i
+ 1024]);
174 s16
[6 * i
+ 3] = convert(f
[i
+ 1280]);
175 s16
[6 * i
+ 4] = convert(f
[i
+ 512]);
176 s16
[6 * i
+ 5] = convert(f
[i
]);
183 channels_multi(int flags
)
187 else if(flags
& 1) /* center channel */
189 else if((flags
& DTS_CHANNEL_MASK
) == DTS_2F2R
)
196 dts_decode_frame(AVCodecContext
* avctx
, void *data
, int *data_size
,
197 uint8_t * buff
, int buff_size
)
199 uint8_t *start
= buff
;
200 uint8_t *end
= buff
+ buff_size
;
201 static uint8_t buf
[BUFFER_SIZE
];
202 static uint8_t *bufptr
= buf
;
203 static uint8_t *bufpos
= buf
+ HEADER_SIZE
;
204 int16_t *out_samples
= data
;
205 static int sample_rate
;
206 static int frame_length
;
210 dts_state_t
*state
= avctx
->priv_data
;
223 if(len
> bufpos
- bufptr
)
224 len
= bufpos
- bufptr
;
225 memcpy(bufptr
, start
, len
);
230 if(bufpos
!= buf
+ HEADER_SIZE
)
233 length
= dts_syncinfo(state
, buf
, &flags
, &sample_rate
, &bit_rate
,
236 av_log(NULL
, AV_LOG_INFO
, "skip\n");
237 for(bufptr
= buf
; bufptr
< buf
+ HEADER_SIZE
- 1; bufptr
++)
238 bufptr
[0] = bufptr
[1];
241 bufpos
= buf
+ length
;
244 flags
= 2; /* ???????????? */
245 level
= CONVERT_LEVEL
;
248 flags
|= DTS_ADJUST_LEVEL
;
249 if(dts_frame(state
, buf
, &flags
, &level
, bias
)) {
250 av_log(avctx
, AV_LOG_ERROR
, "dts_frame() failed\n");
254 avctx
->sample_rate
= sample_rate
;
255 avctx
->channels
= channels_multi(flags
);
256 avctx
->bit_rate
= bit_rate
;
258 for(i
= 0; i
< dts_blocks_num(state
); i
++) {
261 if(dts_block(state
)) {
262 av_log(avctx
, AV_LOG_ERROR
, "dts_block() failed\n");
266 chans
= channels_multi(flags
);
267 convert2s16_multi(dts_samples(state
), out_samples
,
268 flags
& (DTS_CHANNEL_MASK
| DTS_LFE
));
270 out_samples
+= 256 * chans
;
271 *data_size
+= 256 * sizeof(int16_t) * chans
;
276 bufpos
= buf
+ HEADER_SIZE
;
281 dts_decode_init(AVCodecContext
* avctx
)
283 avctx
->priv_data
= dts_init(0);
284 if(avctx
->priv_data
== NULL
)
291 dts_decode_end(AVCodecContext
* s
)
296 AVCodec dts_decoder
= {
300 sizeof(dts_state_t
*),