Mercurial > mplayer.hg
comparison libmpdemux/demux_aac.c @ 15720:e999d58f94fd
added AAC ADTS demuxer
author | nicodvb |
---|---|
date | Mon, 13 Jun 2005 20:40:29 +0000 |
parents | |
children | e28cba8204c9 |
comparison
equal
deleted
inserted
replaced
15719:e040252b8370 | 15720:e999d58f94fd |
---|---|
1 #include <stdio.h> | |
2 #include <stdlib.h> | |
3 #include <string.h> | |
4 | |
5 #include "config.h" | |
6 #include "mp_msg.h" | |
7 #include "help_mp.h" | |
8 | |
9 #include "stream.h" | |
10 #include "demuxer.h" | |
11 #include "parse_es.h" | |
12 #include "stheader.h" | |
13 | |
14 #include "bswap.h" | |
15 #include "../unrarlib.h" | |
16 #include "ms_hdr.h" | |
17 | |
18 extern void resync_audio_stream(sh_audio_t *sh_audio); | |
19 | |
20 typedef struct { | |
21 uint8_t *buf; | |
22 uint64_t size; /// amount of time of data packets pushed to demuxer->audio (in bytes) | |
23 float time; /// amount of time elapsed based upon samples_per_frame/sample_rate (in milliseconds) | |
24 float last_pts; /// last pts seen | |
25 int bitrate; /// bitrate computed as size/time | |
26 } aac_priv_t; | |
27 | |
28 /// \param srate (out) sample rate | |
29 /// \param num (out) number of audio frames in this ADTS frame | |
30 /// \return size of the ADTS frame in bytes | |
31 /// aac_parse_frames needs a buffer at least 8 bytes long | |
32 int aac_parse_frame(uint8_t *buf, int *srate, int *num) | |
33 { | |
34 int i = 0, sr, fl = 0, id; | |
35 static int srates[] = {96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050, 16000, 12000, 11025, 8000, 0, 0, 0}; | |
36 | |
37 if((buf[i] != 0xFF) || ((buf[i+1] & 0xF6) != 0xF0)) | |
38 return 0; | |
39 | |
40 id = (buf[i+1] >> 3) & 0x01; //id=1 mpeg2, 0: mpeg4 | |
41 sr = (buf[i+2] >> 2) & 0x0F; | |
42 if(sr > 11) | |
43 return 0; | |
44 *srate = srates[sr]; | |
45 | |
46 fl = ((buf[i+3] & 0x03) << 11) | (buf[i+4] << 3) | ((buf[i+5] >> 5) & 0x07); | |
47 *num = (buf[i+6] & 0x02) + 1; | |
48 | |
49 return fl; | |
50 } | |
51 | |
52 static int demux_aac_init(demuxer_t *demuxer) | |
53 { | |
54 aac_priv_t *priv; | |
55 | |
56 priv = calloc(1, sizeof(aac_priv_t)); | |
57 if(!priv) | |
58 return 0; | |
59 | |
60 priv->buf = (uint8_t*) malloc(8); | |
61 if(!priv->buf) | |
62 return 0; | |
63 | |
64 demuxer->priv = priv; | |
65 return 1; | |
66 } | |
67 | |
68 void demux_close_aac(demuxer_t *demuxer) | |
69 { | |
70 aac_priv_t *priv = (aac_priv_t *) demuxer->priv; | |
71 | |
72 if(!priv) | |
73 return; | |
74 | |
75 if(priv->buf) | |
76 free(priv->buf); | |
77 | |
78 free(demuxer->priv); | |
79 | |
80 return; | |
81 } | |
82 | |
83 /// returns 1 if it finds 8 ADTS frames in 32768 bytes, 0 otherwise | |
84 int demux_aac_probe(demuxer_t *demuxer) | |
85 { | |
86 int cnt = 0, c, len, srate, num; | |
87 off_t init, probed; | |
88 aac_priv_t *priv; | |
89 | |
90 if(! demux_aac_init(demuxer)) | |
91 { | |
92 mp_msg(MSGT_DEMUX, MSGL_ERR, "COULDN'T INIT aac_demux, exit\n"); | |
93 return 0; | |
94 } | |
95 | |
96 priv = (aac_priv_t *) demuxer->priv; | |
97 | |
98 init = probed = stream_tell(demuxer->stream); | |
99 while(probed-init <= 32768 && cnt < 8) | |
100 { | |
101 c = 0; | |
102 while(c != 0xFF) | |
103 { | |
104 c = stream_read_char(demuxer->stream); | |
105 if(c < 0) | |
106 goto fail; | |
107 } | |
108 priv->buf[0] = 0xFF; | |
109 if(stream_read(demuxer->stream, &(priv->buf[1]), 7) < 7) | |
110 goto fail; | |
111 | |
112 len = aac_parse_frame(priv->buf, &srate, &num); | |
113 if(len > 0) | |
114 { | |
115 cnt++; | |
116 stream_skip(demuxer->stream, len - 8); | |
117 } | |
118 probed = stream_tell(demuxer->stream); | |
119 } | |
120 | |
121 stream_seek(demuxer->stream, init); | |
122 if(cnt < 8) | |
123 goto fail; | |
124 | |
125 mp_msg(MSGT_DEMUX, MSGL_V, "demux_aac_probe, INIT: %llu, PROBED: %llu, cnt: %d\n", init, probed, cnt); | |
126 return 1; | |
127 | |
128 fail: | |
129 mp_msg(MSGT_DEMUX, MSGL_V, "demux_aac_probe, failed to detect an AAC stream\n"); | |
130 demux_close_aac(demuxer); | |
131 return 0; | |
132 } | |
133 | |
134 int demux_aac_open(demuxer_t *demuxer) | |
135 { | |
136 sh_audio_t *sh; | |
137 | |
138 sh = new_sh_audio(demuxer, 0); | |
139 sh->ds = demuxer->audio; | |
140 sh->format = mmioFOURCC('M', 'P', '4', 'A'); | |
141 demuxer->audio->sh = sh; | |
142 | |
143 demuxer->filepos = stream_tell(demuxer->stream); | |
144 | |
145 return 1; | |
146 } | |
147 | |
148 int demux_aac_fill_buffer(demuxer_t *demuxer) | |
149 { | |
150 aac_priv_t *priv = (aac_priv_t *) demuxer->priv; | |
151 demux_packet_t *dp; | |
152 int c1, c2, len, srate, num; | |
153 float tm = 0; | |
154 | |
155 if(demuxer->stream->eof || (demuxer->movi_end && stream_tell(demuxer->stream) >= demuxer->movi_end)) | |
156 return 0; | |
157 | |
158 while(! demuxer->stream->eof) | |
159 { | |
160 c1 = c2 = 0; | |
161 while(c1 != 0xFF) | |
162 { | |
163 c1 = stream_read_char(demuxer->stream); | |
164 if(c1 < 0) | |
165 return 0; | |
166 } | |
167 c2 = stream_read_char(demuxer->stream); | |
168 if(c2 < 0) | |
169 return 0; | |
170 if((c2 & 0xF6) != 0xF0) | |
171 continue; | |
172 | |
173 priv->buf[0] = (unsigned char) c1; | |
174 priv->buf[1] = (unsigned char) c2; | |
175 if(stream_read(demuxer->stream, &(priv->buf[2]), 6) < 6) | |
176 return 0; | |
177 | |
178 len = aac_parse_frame(priv->buf, &srate, &num); | |
179 if(len > 0) | |
180 { | |
181 dp = new_demux_packet(len); | |
182 if(! dp) | |
183 { | |
184 mp_msg(MSGT_DEMUX, MSGL_ERR, "fill_buffer, NEW_ADD_PACKET(%d)FAILED\n", len); | |
185 return 0; | |
186 } | |
187 | |
188 | |
189 memcpy(dp->buffer, priv->buf, 8); | |
190 stream_read(demuxer->stream, &(dp->buffer[8]), len-8); | |
191 if(srate) | |
192 tm = (float) (num * 1024.0/srate); | |
193 priv->last_pts += tm; | |
194 dp->pts = priv->last_pts; | |
195 //fprintf(stderr, "\nPTS: %.3f\n", dp->pts); | |
196 ds_add_packet(demuxer->audio, dp); | |
197 priv->size += len; | |
198 priv->time += tm; | |
199 | |
200 priv->bitrate = (int) (priv->size / priv->time); | |
201 demuxer->filepos = stream_tell(demuxer->stream); | |
202 | |
203 return len; | |
204 } | |
205 else | |
206 stream_skip(demuxer->stream, -6); | |
207 } | |
208 | |
209 return 0; | |
210 } | |
211 | |
212 | |
213 //This is an almost verbatim copy of high_res_mp3_seek(), from demux_audio.c | |
214 void demux_aac_seek(demuxer_t *demuxer, float rel_seek_secs, int flags) | |
215 { | |
216 aac_priv_t *priv = (aac_priv_t *) demuxer->priv; | |
217 demux_stream_t *d_audio=demuxer->audio; | |
218 sh_audio_t *sh_audio=d_audio->sh; | |
219 float time; | |
220 | |
221 ds_free_packs(d_audio); | |
222 | |
223 time = (flags & 1) ? rel_seek_secs - priv->last_pts : rel_seek_secs; | |
224 if(time < 0) | |
225 { | |
226 stream_seek(demuxer->stream, demuxer->movi_start); | |
227 time = priv->last_pts + time; | |
228 priv->last_pts = 0; | |
229 } | |
230 | |
231 if(time > 0) | |
232 { | |
233 int len, nf, srate, num; | |
234 | |
235 nf = time * sh_audio->samplerate/1024; | |
236 | |
237 while(nf > 0) | |
238 { | |
239 if(stream_read(demuxer->stream,priv->buf, 8) < 8) | |
240 break; | |
241 len = aac_parse_frame(priv->buf, &srate, &num); | |
242 if(len <= 0) | |
243 { | |
244 stream_skip(demuxer->stream, -7); | |
245 continue; | |
246 } | |
247 stream_skip(demuxer->stream, len - 8); | |
248 priv->last_pts += (float) (num*1024.0/srate); | |
249 nf -= num; | |
250 } | |
251 } | |
252 sh_audio->delay = priv->last_pts - (ds_tell_pts(demuxer->audio)-sh_audio->a_in_buffer_len)/(float)priv->bitrate; | |
253 resync_audio_stream(sh_audio); | |
254 } | |
255 |