Mercurial > mplayer.hg
annotate libmpdemux/demux_lavf.c @ 15815:488f31bd127e
support raw ac3 (in private pes packets without the usual dvd 4 bytes substream header). Patch by Matthias Scharzott
author | nicodvb |
---|---|
date | Sat, 25 Jun 2005 13:25:27 +0000 |
parents | e9865b828a89 |
children | dd23cdecf060 |
rev | line source |
---|---|
12164 | 1 /* |
2 Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at> | |
3 | |
4 This program is free software; you can redistribute it and/or modify | |
5 it under the terms of the GNU General Public License as published by | |
6 the Free Software Foundation; either version 2 of the License, or | |
7 (at your option) any later version. | |
8 | |
9 This program is distributed in the hope that it will be useful, | |
10 but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 GNU General Public License for more details. | |
13 | |
14 You should have received a copy of the GNU General Public License | |
15 along with this program; if not, write to the Free Software | |
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
17 */ | |
18 | |
19 // #include <stdio.h> | |
20 #include <stdlib.h> | |
21 // #include <unistd.h> | |
22 | |
23 #include "config.h" | |
24 #include "mp_msg.h" | |
25 // #include "help_mp.h" | |
26 | |
27 #include "stream.h" | |
28 #include "demuxer.h" | |
29 #include "stheader.h" | |
30 | |
31 #ifdef USE_LIBAVFORMAT | |
32 | |
33 #include "avformat.h" | |
34 #include "avi.h" | |
35 | |
36 #define PROBE_BUF_SIZE 2048 | |
37 | |
38 typedef struct lavf_priv_t{ | |
39 AVInputFormat *avif; | |
40 AVFormatContext *avfc; | |
41 ByteIOContext pb; | |
42 int audio_streams; | |
43 int video_streams; | |
12168 | 44 int64_t last_pts; |
12164 | 45 }lavf_priv_t; |
46 | |
47 extern void print_wave_header(WAVEFORMATEX *h); | |
48 extern void print_video_header(BITMAPINFOHEADER *h); | |
49 | |
15011 | 50 int64_t ff_gcd(int64_t a, int64_t b); |
51 | |
12164 | 52 static int mp_open(URLContext *h, const char *filename, int flags){ |
53 return 0; | |
54 } | |
55 | |
56 static int mp_read(URLContext *h, unsigned char *buf, int size){ | |
57 stream_t *stream = (stream_t*)h->priv_data; | |
12165
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
58 int ret; |
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
59 |
12164 | 60 if(stream_eof(stream)) //needed? |
61 return -1; | |
12165
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
62 ret=stream_read(stream, buf, size); |
12166 | 63 |
12165
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
64 mp_msg(MSGT_HEADER,MSGL_DBG2,"%d=mp_read(%p, %p, %d), eof:%d\n", ret, h, buf, size, stream->eof); |
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
65 return ret; |
12164 | 66 } |
67 | |
68 static int mp_write(URLContext *h, unsigned char *buf, int size){ | |
69 return -1; | |
70 } | |
71 | |
72 static offset_t mp_seek(URLContext *h, offset_t pos, int whence){ | |
73 stream_t *stream = (stream_t*)h->priv_data; | |
12165
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
74 |
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
75 mp_msg(MSGT_HEADER,MSGL_DBG2,"mp_seek(%p, %d, %d)\n", h, (int)pos, whence); |
12164 | 76 if(whence == SEEK_CUR) |
77 pos +=stream_tell(stream); | |
78 else if(whence == SEEK_END) | |
79 pos += stream->end_pos; | |
80 else if(whence != SEEK_SET) | |
81 return -1; | |
82 | |
12167 | 83 if(pos<stream->end_pos && stream->eof) |
12166 | 84 stream_reset(stream); |
12164 | 85 if(stream_seek(stream, pos)==0) |
86 return -1; | |
12166 | 87 |
12164 | 88 return pos; |
89 } | |
90 | |
91 static int mp_close(URLContext *h){ | |
92 return 0; | |
93 } | |
94 | |
95 static URLProtocol mp_protocol = { | |
96 "mp", | |
97 mp_open, | |
98 mp_read, | |
99 mp_write, | |
100 mp_seek, | |
101 mp_close, | |
102 }; | |
103 | |
104 int lavf_check_file(demuxer_t *demuxer){ | |
105 AVProbeData avpd; | |
106 uint8_t buf[PROBE_BUF_SIZE]; | |
107 lavf_priv_t *priv; | |
108 | |
109 if(!demuxer->priv) | |
110 demuxer->priv=calloc(sizeof(lavf_priv_t),1); | |
111 priv= demuxer->priv; | |
112 | |
113 av_register_all(); | |
114 | |
115 stream_read(demuxer->stream, buf, PROBE_BUF_SIZE); | |
116 avpd.filename= demuxer->stream->url; | |
117 avpd.buf= buf; | |
118 avpd.buf_size= PROBE_BUF_SIZE; | |
119 | |
120 priv->avif= av_probe_input_format(&avpd, 1); | |
121 if(!priv->avif){ | |
122 mp_msg(MSGT_HEADER,MSGL_V,"LAVF_check: no clue about this gibberish!\n"); | |
123 return 0; | |
124 }else | |
125 mp_msg(MSGT_HEADER,MSGL_V,"LAVF_check: %s\n", priv->avif->long_name); | |
126 | |
127 return 1; | |
128 } | |
129 | |
130 int demux_open_lavf(demuxer_t *demuxer){ | |
131 AVFormatContext *avfc; | |
132 AVFormatParameters ap; | |
133 lavf_priv_t *priv= demuxer->priv; | |
15011 | 134 int i,g; |
12164 | 135 char mp_filename[256]="mp:"; |
136 | |
137 memset(&ap, 0, sizeof(AVFormatParameters)); | |
138 | |
139 stream_seek(demuxer->stream, 0); | |
140 | |
141 register_protocol(&mp_protocol); | |
142 | |
12463 | 143 if(demuxer->stream->url) |
144 strncpy(mp_filename + 3, demuxer->stream->url, sizeof(mp_filename)-3); | |
145 else | |
146 strncpy(mp_filename + 3, "foobar.dummy", sizeof(mp_filename)-3); | |
12164 | 147 |
148 url_fopen(&priv->pb, mp_filename, URL_RDONLY); | |
149 | |
150 ((URLContext*)(priv->pb.opaque))->priv_data= demuxer->stream; | |
151 | |
152 if(av_open_input_stream(&avfc, &priv->pb, mp_filename, priv->avif, &ap)<0){ | |
153 mp_msg(MSGT_HEADER,MSGL_ERR,"LAVF_header: av_open_input_stream() failed\n"); | |
154 return 0; | |
155 } | |
156 | |
157 priv->avfc= avfc; | |
158 | |
159 if(av_find_stream_info(avfc) < 0){ | |
160 mp_msg(MSGT_HEADER,MSGL_ERR,"LAVF_header: av_find_stream_info() failed\n"); | |
161 return 0; | |
162 } | |
163 | |
12167 | 164 if(avfc->title [0]) demux_info_add(demuxer, "name" , avfc->title ); |
165 if(avfc->author [0]) demux_info_add(demuxer, "author" , avfc->author ); | |
166 if(avfc->copyright[0]) demux_info_add(demuxer, "copyright", avfc->copyright); | |
167 if(avfc->comment [0]) demux_info_add(demuxer, "comments" , avfc->comment ); | |
168 if(avfc->album [0]) demux_info_add(demuxer, "album" , avfc->album ); | |
169 // if(avfc->year ) demux_info_add(demuxer, "year" , avfc->year ); | |
170 // if(avfc->track ) demux_info_add(demuxer, "track" , avfc->track ); | |
171 if(avfc->genre [0]) demux_info_add(demuxer, "genre" , avfc->genre ); | |
12164 | 172 |
173 for(i=0; i<avfc->nb_streams; i++){ | |
174 AVStream *st= avfc->streams[i]; | |
175 AVCodecContext *codec= &st->codec; | |
176 | |
177 switch(codec->codec_type){ | |
178 case CODEC_TYPE_AUDIO:{ | |
179 WAVEFORMATEX *wf= calloc(sizeof(WAVEFORMATEX) + codec->extradata_size, 1); | |
180 sh_audio_t* sh_audio=new_sh_audio(demuxer, i); | |
181 priv->audio_streams++; | |
182 if(!codec->codec_tag) | |
183 codec->codec_tag= codec_get_wav_tag(codec->codec_id); | |
184 wf->wFormatTag= codec->codec_tag; | |
185 wf->nChannels= codec->channels; | |
186 wf->nSamplesPerSec= codec->sample_rate; | |
187 wf->nAvgBytesPerSec= codec->bit_rate/8; | |
188 wf->nBlockAlign= codec->block_align; | |
189 wf->wBitsPerSample= codec->bits_per_sample; | |
190 wf->cbSize= codec->extradata_size; | |
191 if(codec->extradata_size){ | |
192 memcpy( | |
193 wf + 1, | |
194 codec->extradata, | |
195 codec->extradata_size); | |
196 } | |
197 sh_audio->wf= wf; | |
15011 | 198 sh_audio->audio.dwSampleSize= codec->block_align; |
199 if(codec->frame_size && codec->sample_rate){ | |
200 sh_audio->audio.dwScale=codec->frame_size; | |
201 sh_audio->audio.dwRate= codec->sample_rate; | |
202 }else{ | |
203 sh_audio->audio.dwScale= codec->block_align ? codec->block_align*8 : 8; | |
204 sh_audio->audio.dwRate = codec->bit_rate; | |
205 } | |
206 g= ff_gcd(sh_audio->audio.dwScale, sh_audio->audio.dwRate); | |
207 sh_audio->audio.dwScale /= g; | |
208 sh_audio->audio.dwRate /= g; | |
209 // printf("sca:%d rat:%d fs:%d sr:%d ba:%d\n", sh_audio->audio.dwScale, sh_audio->audio.dwRate, codec->frame_size, codec->sample_rate, codec->block_align); | |
12164 | 210 sh_audio->ds= demuxer->audio; |
211 sh_audio->format= codec->codec_tag; | |
212 sh_audio->channels= codec->channels; | |
213 sh_audio->samplerate= codec->sample_rate; | |
15007 | 214 sh_audio->i_bps= codec->bit_rate/8; |
12164 | 215 if(verbose>=1) print_wave_header(sh_audio->wf); |
15004 | 216 if(demuxer->audio->id != i && demuxer->audio->id != -1) |
217 st->discard= AVDISCARD_ALL; | |
218 else{ | |
219 demuxer->audio->id = i; | |
220 demuxer->audio->sh= demuxer->a_streams[i]; | |
221 } | |
12164 | 222 break;} |
223 case CODEC_TYPE_VIDEO:{ | |
224 BITMAPINFOHEADER *bih=calloc(sizeof(BITMAPINFOHEADER) + codec->extradata_size,1); | |
225 sh_video_t* sh_video=new_sh_video(demuxer, i); | |
226 | |
227 priv->video_streams++; | |
228 if(!codec->codec_tag) | |
229 codec->codec_tag= codec_get_bmp_tag(codec->codec_id); | |
230 bih->biSize= sizeof(BITMAPINFOHEADER) + codec->extradata_size; | |
231 bih->biWidth= codec->width; | |
232 bih->biHeight= codec->height; | |
233 bih->biBitCount= codec->bits_per_sample; | |
234 bih->biSizeImage = bih->biWidth * bih->biHeight * bih->biBitCount/8; | |
235 bih->biCompression= codec->codec_tag; | |
236 sh_video->bih= bih; | |
237 sh_video->disp_w= codec->width; | |
238 sh_video->disp_h= codec->height; | |
15308 | 239 #if LIBAVFORMAT_BUILD >= 4624 |
240 sh_video->video.dwRate= codec->time_base.den; | |
241 sh_video->video.dwScale= codec->time_base.num; | |
242 #else | |
12164 | 243 sh_video->video.dwRate= codec->frame_rate; |
244 sh_video->video.dwScale= codec->frame_rate_base; | |
15308 | 245 #endif |
12164 | 246 sh_video->fps=(float)sh_video->video.dwRate/(float)sh_video->video.dwScale; |
247 sh_video->frametime=(float)sh_video->video.dwScale/(float)sh_video->video.dwRate; | |
248 sh_video->format = bih->biCompression; | |
12167 | 249 sh_video->aspect= codec->width * codec->sample_aspect_ratio.num |
250 / (float)(codec->height * codec->sample_aspect_ratio.den); | |
15007 | 251 sh_video->i_bps= codec->bit_rate/8; |
12167 | 252 mp_msg(MSGT_DEMUX,MSGL_DBG2,"aspect= %d*%d/(%d*%d)\n", |
253 codec->width, codec->sample_aspect_ratio.num, | |
254 codec->height, codec->sample_aspect_ratio.den); | |
255 | |
12164 | 256 sh_video->ds= demuxer->video; |
257 if(codec->extradata_size) | |
258 memcpy(sh_video->bih + 1, codec->extradata, codec->extradata_size); | |
259 if(verbose>=1) print_video_header(sh_video->bih); | |
260 /* short biPlanes; | |
261 int biXPelsPerMeter; | |
262 int biYPelsPerMeter; | |
263 int biClrUsed; | |
264 int biClrImportant;*/ | |
15004 | 265 if(demuxer->video->id != i && demuxer->video->id != -1) |
266 st->discard= AVDISCARD_ALL; | |
267 else{ | |
268 demuxer->video->id = i; | |
269 demuxer->video->sh= demuxer->v_streams[i]; | |
270 } | |
12164 | 271 break;} |
15004 | 272 default: |
273 st->discard= AVDISCARD_ALL; | |
12164 | 274 } |
275 } | |
276 | |
277 mp_msg(MSGT_HEADER,MSGL_V,"LAVF: %d audio and %d video streams found\n",priv->audio_streams,priv->video_streams); | |
13749 | 278 mp_msg(MSGT_HEADER,MSGL_V,"LAVF: build %d\n", LIBAVFORMAT_BUILD); |
12164 | 279 if(!priv->audio_streams) demuxer->audio->id=-2; // nosound |
280 // else if(best_audio > 0 && demuxer->audio->id == -1) demuxer->audio->id=best_audio; | |
281 if(!priv->video_streams){ | |
282 if(!priv->audio_streams){ | |
283 mp_msg(MSGT_HEADER,MSGL_ERR,"LAVF: no audio or video headers found - broken file?\n"); | |
284 return 0; | |
285 } | |
286 demuxer->video->id=-2; // audio-only | |
287 } //else if (best_video > 0 && demuxer->video->id == -1) demuxer->video->id = best_video; | |
288 | |
289 return 1; | |
290 } | |
291 | |
292 int demux_lavf_fill_buffer(demuxer_t *demux){ | |
293 lavf_priv_t *priv= demux->priv; | |
294 AVPacket pkt; | |
295 demux_packet_t *dp; | |
296 demux_stream_t *ds; | |
297 int id; | |
298 mp_msg(MSGT_DEMUX,MSGL_DBG2,"demux_lavf_fill_buffer()\n"); | |
299 | |
300 demux->filepos=stream_tell(demux->stream); | |
301 | |
302 if(stream_eof(demux->stream)){ | |
303 // demuxre->stream->eof=1; | |
304 return 0; | |
305 } | |
306 | |
307 if(av_read_frame(priv->avfc, &pkt) < 0) | |
308 return 0; | |
309 | |
310 id= pkt.stream_index; | |
311 | |
312 if(id==demux->audio->id){ | |
313 // audio | |
314 ds=demux->audio; | |
315 if(!ds->sh){ | |
316 ds->sh=demux->a_streams[id]; | |
317 mp_msg(MSGT_DEMUX,MSGL_V,"Auto-selected LAVF audio ID = %d\n",ds->id); | |
318 } | |
319 } else if(id==demux->video->id){ | |
320 // video | |
321 ds=demux->video; | |
322 if(!ds->sh){ | |
323 ds->sh=demux->v_streams[id]; | |
324 mp_msg(MSGT_DEMUX,MSGL_V,"Auto-selected LAVF video ID = %d\n",ds->id); | |
325 } | |
14611 | 326 } else { |
327 av_free_packet(&pkt); | |
328 return 1; | |
329 } | |
12164 | 330 |
331 if(0/*pkt.destruct == av_destruct_packet*/){ | |
332 //ok kids, dont try this at home :) | |
333 dp=(demux_packet_t*)malloc(sizeof(demux_packet_t)); | |
334 dp->len=pkt.size; | |
335 dp->next=NULL; | |
336 dp->refcount=1; | |
337 dp->master=NULL; | |
338 dp->buffer=pkt.data; | |
339 pkt.destruct= NULL; | |
340 }else{ | |
341 dp=new_demux_packet(pkt.size); | |
342 memcpy(dp->buffer, pkt.data, pkt.size); | |
343 av_free_packet(&pkt); | |
344 } | |
345 | |
13747 | 346 if(pkt.pts != AV_NOPTS_VALUE){ |
15308 | 347 #if LIBAVFORMAT_BUILD >= 4624 |
348 dp->pts=pkt.pts * av_q2d(priv->avfc->streams[id]->time_base); | |
349 priv->last_pts= dp->pts * AV_TIME_BASE; | |
350 #else | |
13747 | 351 priv->last_pts= pkt.pts; |
352 dp->pts=pkt.pts / (float)AV_TIME_BASE; | |
15308 | 353 #endif |
13747 | 354 } |
12164 | 355 dp->pos=demux->filepos; |
356 dp->flags= !!(pkt.flags&PKT_FLAG_KEY); | |
357 // append packet to DS stream: | |
358 ds_add_packet(ds,dp); | |
359 return 1; | |
360 } | |
361 | |
362 void demux_seek_lavf(demuxer_t *demuxer, float rel_seek_secs, int flags){ | |
12168 | 363 lavf_priv_t *priv = demuxer->priv; |
364 mp_msg(MSGT_DEMUX,MSGL_DBG2,"demux_seek_lavf(%p, %f, %d)\n", demuxer, rel_seek_secs, flags); | |
365 | |
13607 | 366 #if LIBAVFORMAT_BUILD < 4619 |
12168 | 367 av_seek_frame(priv->avfc, -1, priv->last_pts + rel_seek_secs*AV_TIME_BASE); |
13607 | 368 #else |
369 av_seek_frame(priv->avfc, -1, priv->last_pts + rel_seek_secs*AV_TIME_BASE, rel_seek_secs < 0 ? AVSEEK_FLAG_BACKWARD : 0); | |
370 #endif | |
12164 | 371 } |
372 | |
373 int demux_lavf_control(demuxer_t *demuxer, int cmd, void *arg) | |
374 { | |
375 lavf_priv_t *priv = demuxer->priv; | |
376 | |
377 switch (cmd) { | |
12168 | 378 case DEMUXER_CTRL_GET_TIME_LENGTH: |
379 if (priv->avfc->duration == 0) | |
12164 | 380 return DEMUXER_CTRL_DONTKNOW; |
381 | |
12168 | 382 *((unsigned long *)arg) = priv->avfc->duration / AV_TIME_BASE; |
12164 | 383 return DEMUXER_CTRL_OK; |
384 | |
385 case DEMUXER_CTRL_GET_PERCENT_POS: | |
12168 | 386 if (priv->avfc->duration == 0) |
12164 | 387 return DEMUXER_CTRL_DONTKNOW; |
388 | |
12168 | 389 *((int *)arg) = (int)(priv->last_pts*100 / priv->avfc->duration); |
390 return DEMUXER_CTRL_OK; | |
12164 | 391 |
392 default: | |
393 return DEMUXER_CTRL_NOTIMPL; | |
394 } | |
395 } | |
396 | |
397 void demux_close_lavf(demuxer_t *demuxer) | |
398 { | |
399 lavf_priv_t* priv = demuxer->priv; | |
400 if (priv){ | |
12304
434242b0706c
fix possible segfault on lavf demuxer patch by (adland <adland123 at yahoo dot com>)
michael
parents:
12168
diff
changeset
|
401 if(priv->avfc) |
434242b0706c
fix possible segfault on lavf demuxer patch by (adland <adland123 at yahoo dot com>)
michael
parents:
12168
diff
changeset
|
402 { |
434242b0706c
fix possible segfault on lavf demuxer patch by (adland <adland123 at yahoo dot com>)
michael
parents:
12168
diff
changeset
|
403 av_close_input_file(priv->avfc); priv->avfc= NULL; |
434242b0706c
fix possible segfault on lavf demuxer patch by (adland <adland123 at yahoo dot com>)
michael
parents:
12168
diff
changeset
|
404 } |
12164 | 405 free(priv); demuxer->priv= NULL; |
406 } | |
407 } | |
408 | |
409 #endif // USE_LIBAVFORMAT |