2 * Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at>
4 * This file is part of MPlayer.
6 * MPlayer is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * MPlayer is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License along
17 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
23 // #include <unistd.h>
33 #include "stream/stream.h"
37 #include "libvo/sub.h"
39 #include "libavformat/avformat.h"
40 #include "libavformat/avio.h"
41 #include "libavutil/avutil.h"
42 #include "libavcodec/opt.h"
44 #include "mp_taglists.h"
46 #define PROBE_BUF_SIZE (32*1024)
48 static unsigned int opt_probesize
= 0;
49 static unsigned int opt_analyzeduration
= 0;
50 static char *opt_format
;
51 static char *opt_cryptokey
;
53 static char *opt_avopt
= NULL
;
55 const m_option_t lavfdopts_conf
[] = {
56 {"probesize", &(opt_probesize
), CONF_TYPE_INT
, CONF_RANGE
, 32, INT_MAX
, NULL
},
57 {"format", &(opt_format
), CONF_TYPE_STRING
, 0, 0, 0, NULL
},
58 {"analyzeduration", &(opt_analyzeduration
), CONF_TYPE_INT
, CONF_RANGE
, 0, INT_MAX
, NULL
},
59 {"cryptokey", &(opt_cryptokey
), CONF_TYPE_STRING
, 0, 0, 0, NULL
},
60 {"o", &opt_avopt
, CONF_TYPE_STRING
, 0, 0, 0, NULL
},
61 {NULL
, NULL
, 0, 0, 0, 0, NULL
}
64 #define BIO_BUFFER_SIZE 32768
66 typedef struct lavf_priv_t
{
68 AVFormatContext
*avfc
;
70 uint8_t buffer
[BIO_BUFFER_SIZE
];
75 int astreams
[MAX_A_STREAMS
];
76 int vstreams
[MAX_V_STREAMS
];
77 int sstreams
[MAX_S_STREAMS
];
81 void print_wave_header(WAVEFORMATEX
*h
, int verbose_level
);
82 void print_video_header(BITMAPINFOHEADER
*h
, int verbose_level
);
84 static int mp_read(void *opaque
, uint8_t *buf
, int size
) {
85 stream_t
*stream
= opaque
;
88 if(stream_eof(stream
)) //needed?
90 ret
=stream_read(stream
, buf
, size
);
92 mp_msg(MSGT_HEADER
,MSGL_DBG2
,"%d=mp_read(%p, %p, %d), eof:%d\n", ret
, stream
, buf
, size
, stream
->eof
);
96 static int64_t mp_seek(void *opaque
, int64_t pos
, int whence
) {
97 stream_t
*stream
= opaque
;
99 mp_msg(MSGT_HEADER
,MSGL_DBG2
,"mp_seek(%p, %d, %d)\n", stream
, (int)pos
, whence
);
100 if(whence
== SEEK_CUR
)
101 pos
+=stream_tell(stream
);
102 else if(whence
== SEEK_END
&& stream
->end_pos
> 0)
103 pos
+= stream
->end_pos
;
104 else if(whence
== SEEK_SET
)
105 pos
+= stream
->start_pos
;
106 else if(whence
== AVSEEK_SIZE
&& stream
->end_pos
> 0)
107 return stream
->end_pos
- stream
->start_pos
;
113 if(pos
<stream
->end_pos
&& stream
->eof
)
114 stream_reset(stream
);
115 current_pos
= stream_tell(stream
);
116 if(stream_seek(stream
, pos
)==0) {
117 stream_reset(stream
);
118 stream_seek(stream
, current_pos
);
122 return pos
- stream
->start_pos
;
125 static void list_formats(void) {
127 mp_msg(MSGT_DEMUX
, MSGL_INFO
, "Available lavf input formats:\n");
128 for (fmt
= first_iformat
; fmt
; fmt
= fmt
->next
)
129 mp_msg(MSGT_DEMUX
, MSGL_INFO
, "%15s : %s\n", fmt
->name
, fmt
->long_name
);
132 static int lavf_check_file(demuxer_t
*demuxer
){
134 uint8_t buf
[PROBE_BUF_SIZE
];
138 demuxer
->priv
=calloc(sizeof(lavf_priv_t
),1);
144 if (strcmp(opt_format
, "help") == 0) {
148 priv
->avif
= av_find_input_format(opt_format
);
150 mp_msg(MSGT_DEMUX
,MSGL_FATAL
,"Unknown lavf format %s\n", opt_format
);
153 mp_msg(MSGT_DEMUX
,MSGL_INFO
,"Forced lavf %s demuxer\n", priv
->avif
->long_name
);
154 return DEMUXER_TYPE_LAVF
;
157 if(stream_read(demuxer
->stream
, buf
, PROBE_BUF_SIZE
)!=PROBE_BUF_SIZE
)
159 avpd
.filename
= demuxer
->stream
->url
;
161 avpd
.buf_size
= PROBE_BUF_SIZE
;
163 priv
->avif
= av_probe_input_format(&avpd
, 1);
165 mp_msg(MSGT_HEADER
,MSGL_V
,"LAVF_check: no clue about this gibberish!\n");
168 mp_msg(MSGT_HEADER
,MSGL_V
,"LAVF_check: %s\n", priv
->avif
->long_name
);
170 return DEMUXER_TYPE_LAVF
;
173 static const char * const preferred_list
[] = {
182 "mov,mp4,m4a,3gp,3g2,mj2",
188 static int lavf_check_preferred_file(demuxer_t
*demuxer
){
189 if (lavf_check_file(demuxer
)) {
190 const char * const *p
= preferred_list
;
191 lavf_priv_t
*priv
= demuxer
->priv
;
193 if (strcmp(*p
, priv
->avif
->name
) == 0)
194 return DEMUXER_TYPE_LAVF_PREFERRED
;
201 static uint8_t char2int(char c
) {
202 if (c
>= '0' && c
<= '9') return c
- '0';
203 if (c
>= 'a' && c
<= 'f') return c
- 'a' + 10;
204 if (c
>= 'A' && c
<= 'F') return c
- 'A' + 10;
208 static void parse_cryptokey(AVFormatContext
*avfc
, const char *str
) {
209 int len
= strlen(str
) / 2;
210 uint8_t *key
= av_mallocz(len
);
214 for (i
= 0; i
< len
; i
++, str
+= 2)
215 *key
++ = (char2int(str
[0]) << 4) | char2int(str
[1]);
218 static void handle_stream(demuxer_t
*demuxer
, AVFormatContext
*avfc
, int i
) {
219 lavf_priv_t
*priv
= demuxer
->priv
;
220 AVStream
*st
= avfc
->streams
[i
];
221 AVCodecContext
*codec
= st
->codec
;
222 AVMetadataTag
*lang
= av_metadata_get(st
->metadata
, "language", NULL
, 0);
225 switch(codec
->codec_type
){
226 case CODEC_TYPE_AUDIO
:{
229 sh_audio_t
* sh_audio
;
230 sh_audio
=new_sh_audio(demuxer
, i
);
231 mp_tmsg(MSGT_DEMUX
, MSGL_INFO
, "[%s] Audio stream found, -aid %d\n", "lavf", i
);
234 priv
->astreams
[priv
->audio_streams
] = i
;
235 priv
->audio_streams
++;
236 wf
= calloc(sizeof(WAVEFORMATEX
) + codec
->extradata_size
, 1);
237 // For some formats (like PCM) always trust CODEC_ID_* more than codec_tag
238 override_tag
= av_codec_get_tag(mp_wav_override_taglists
, codec
->codec_id
);
240 codec
->codec_tag
= override_tag
;
241 // mp4a tag is used for all mp4 files no matter what they actually contain
242 if(codec
->codec_tag
== MKTAG('m', 'p', '4', 'a'))
244 if(codec
->codec_id
== CODEC_ID_ADPCM_IMA_AMV
)
245 codec
->codec_tag
= MKTAG('A','M','V','A');
246 if(!codec
->codec_tag
)
247 codec
->codec_tag
= av_codec_get_tag(mp_wav_taglists
, codec
->codec_id
);
248 wf
->wFormatTag
= codec
->codec_tag
;
249 wf
->nChannels
= codec
->channels
;
250 wf
->nSamplesPerSec
= codec
->sample_rate
;
251 wf
->nAvgBytesPerSec
= codec
->bit_rate
/8;
252 wf
->nBlockAlign
= codec
->block_align
? codec
->block_align
: 1;
253 wf
->wBitsPerSample
= codec
->bits_per_coded_sample
;
254 wf
->cbSize
= codec
->extradata_size
;
255 if(codec
->extradata_size
)
256 memcpy(wf
+ 1, codec
->extradata
, codec
->extradata_size
);
258 sh_audio
->audio
.dwSampleSize
= codec
->block_align
;
259 if(codec
->frame_size
&& codec
->sample_rate
){
260 sh_audio
->audio
.dwScale
=codec
->frame_size
;
261 sh_audio
->audio
.dwRate
= codec
->sample_rate
;
263 sh_audio
->audio
.dwScale
= codec
->block_align
? codec
->block_align
*8 : 8;
264 sh_audio
->audio
.dwRate
= codec
->bit_rate
;
266 g
= av_gcd(sh_audio
->audio
.dwScale
, sh_audio
->audio
.dwRate
);
267 sh_audio
->audio
.dwScale
/= g
;
268 sh_audio
->audio
.dwRate
/= g
;
269 // 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);
270 sh_audio
->ds
= demuxer
->audio
;
271 sh_audio
->format
= codec
->codec_tag
;
272 sh_audio
->channels
= codec
->channels
;
273 sh_audio
->samplerate
= codec
->sample_rate
;
274 sh_audio
->i_bps
= codec
->bit_rate
/8;
275 switch (codec
->codec_id
) {
276 case CODEC_ID_PCM_S8
:
277 case CODEC_ID_PCM_U8
:
278 sh_audio
->samplesize
= 1;
280 case CODEC_ID_PCM_S16LE
:
281 case CODEC_ID_PCM_S16BE
:
282 case CODEC_ID_PCM_U16LE
:
283 case CODEC_ID_PCM_U16BE
:
284 sh_audio
->samplesize
= 2;
286 case CODEC_ID_PCM_ALAW
:
287 sh_audio
->format
= 0x6;
289 case CODEC_ID_PCM_MULAW
:
290 sh_audio
->format
= 0x7;
293 if (lang
&& lang
->value
) {
294 sh_audio
->lang
= strdup(lang
->value
);
295 mp_msg(MSGT_IDENTIFY
, MSGL_INFO
, "ID_AID_%d_LANG=%s\n", i
, sh_audio
->lang
);
297 if (st
->disposition
& AV_DISPOSITION_DEFAULT
)
298 sh_audio
->default_track
= 1;
299 if(mp_msg_test(MSGT_HEADER
,MSGL_V
) ) print_wave_header(sh_audio
->wf
, MSGL_V
);
300 // select the first audio stream
301 if (!demuxer
->audio
->sh
) {
302 demuxer
->audio
->id
= i
;
303 demuxer
->audio
->sh
= demuxer
->a_streams
[i
];
305 st
->discard
= AVDISCARD_ALL
;
308 case CODEC_TYPE_VIDEO
:{
309 sh_video_t
* sh_video
;
310 BITMAPINFOHEADER
*bih
;
311 sh_video
=new_sh_video(demuxer
, i
);
312 mp_tmsg(MSGT_DEMUX
, MSGL_INFO
, "[%s] Video stream found, -vid %d\n", "lavf", i
);
314 priv
->vstreams
[priv
->video_streams
] = i
;
315 priv
->video_streams
++;
316 bih
=calloc(sizeof(BITMAPINFOHEADER
) + codec
->extradata_size
,1);
318 if(codec
->codec_id
== CODEC_ID_RAWVIDEO
) {
319 switch (codec
->pix_fmt
) {
321 codec
->codec_tag
= MKTAG(24, 'B', 'G', 'R');
324 if(!codec
->codec_tag
)
325 codec
->codec_tag
= av_codec_get_tag(mp_bmp_taglists
, codec
->codec_id
);
326 bih
->biSize
= sizeof(BITMAPINFOHEADER
) + codec
->extradata_size
;
327 bih
->biWidth
= codec
->width
;
328 bih
->biHeight
= codec
->height
;
329 bih
->biBitCount
= codec
->bits_per_coded_sample
;
330 bih
->biSizeImage
= bih
->biWidth
* bih
->biHeight
* bih
->biBitCount
/8;
331 bih
->biCompression
= codec
->codec_tag
;
333 sh_video
->disp_w
= codec
->width
;
334 sh_video
->disp_h
= codec
->height
;
335 if (st
->time_base
.den
) { /* if container has time_base, use that */
336 sh_video
->video
.dwRate
= st
->time_base
.den
;
337 sh_video
->video
.dwScale
= st
->time_base
.num
;
339 sh_video
->video
.dwRate
= codec
->time_base
.den
;
340 sh_video
->video
.dwScale
= codec
->time_base
.num
;
342 sh_video
->fps
=av_q2d(st
->r_frame_rate
);
343 sh_video
->frametime
=1/av_q2d(st
->r_frame_rate
);
344 sh_video
->format
=bih
->biCompression
;
345 if(st
->sample_aspect_ratio
.num
)
346 sh_video
->aspect
= codec
->width
* st
->sample_aspect_ratio
.num
347 / (float)(codec
->height
* st
->sample_aspect_ratio
.den
);
349 sh_video
->aspect
=codec
->width
* codec
->sample_aspect_ratio
.num
350 / (float)(codec
->height
* codec
->sample_aspect_ratio
.den
);
351 sh_video
->i_bps
=codec
->bit_rate
/8;
352 mp_msg(MSGT_DEMUX
,MSGL_DBG2
,"aspect= %d*%d/(%d*%d)\n",
353 codec
->width
, codec
->sample_aspect_ratio
.num
,
354 codec
->height
, codec
->sample_aspect_ratio
.den
);
356 sh_video
->ds
= demuxer
->video
;
357 if(codec
->extradata_size
)
358 memcpy(sh_video
->bih
+ 1, codec
->extradata
, codec
->extradata_size
);
359 if( mp_msg_test(MSGT_HEADER
,MSGL_V
) ) print_video_header(sh_video
->bih
, MSGL_V
);
367 if(demuxer
->video
->id
!= i
&& demuxer
->video
->id
!= -1)
368 st
->discard
= AVDISCARD_ALL
;
370 demuxer
->video
->id
= i
;
371 demuxer
->video
->sh
= demuxer
->v_streams
[i
];
375 case CODEC_TYPE_SUBTITLE
:{
378 /* only support text subtitles for now */
379 if(codec
->codec_id
== CODEC_ID_TEXT
)
381 else if(codec
->codec_id
== CODEC_ID_MOV_TEXT
)
383 else if(codec
->codec_id
== CODEC_ID_SSA
)
385 else if(codec
->codec_id
== CODEC_ID_DVD_SUBTITLE
)
389 sh_sub
= new_sh_sub_sid(demuxer
, i
, priv
->sub_streams
);
390 mp_tmsg(MSGT_DEMUX
, MSGL_INFO
, "[%s] Subtitle stream found, -sid %d\n", "lavf", priv
->sub_streams
);
392 priv
->sstreams
[priv
->sub_streams
] = i
;
394 if (codec
->extradata_size
) {
395 sh_sub
->extradata
= malloc(codec
->extradata_size
);
396 memcpy(sh_sub
->extradata
, codec
->extradata
, codec
->extradata_size
);
397 sh_sub
->extradata_len
= codec
->extradata_size
;
399 if (lang
&& lang
->value
) {
400 sh_sub
->lang
= strdup(lang
->value
);
401 mp_msg(MSGT_IDENTIFY
, MSGL_INFO
, "ID_SID_%d_LANG=%s\n", priv
->sub_streams
, sh_sub
->lang
);
403 if (st
->disposition
& AV_DISPOSITION_DEFAULT
)
404 sh_sub
->default_track
= 1;
408 case CODEC_TYPE_ATTACHMENT
:{
409 if (st
->codec
->codec_id
== CODEC_ID_TTF
)
410 demuxer_add_attachment(demuxer
, st
->filename
,
411 "application/x-truetype-font",
412 codec
->extradata
, codec
->extradata_size
);
416 st
->discard
= AVDISCARD_ALL
;
420 static demuxer_t
* demux_open_lavf(demuxer_t
*demuxer
){
421 struct MPOpts
*opts
= demuxer
->opts
;
422 AVFormatContext
*avfc
;
423 AVFormatParameters ap
;
425 lavf_priv_t
*priv
= demuxer
->priv
;
427 char mp_filename
[256]="mp:";
429 memset(&ap
, 0, sizeof(AVFormatParameters
));
431 stream_seek(demuxer
->stream
, 0);
433 avfc
= av_alloc_format_context();
436 parse_cryptokey(avfc
, opt_cryptokey
);
437 if (opts
->user_correct_pts
!= 0)
438 avfc
->flags
|= AVFMT_FLAG_GENPTS
;
440 avfc
->flags
|= AVFMT_FLAG_IGNIDX
;
442 ap
.prealloced_context
= 1;
444 opt
= av_set_int(avfc
, "probesize", opt_probesize
);
445 if(!opt
) mp_msg(MSGT_HEADER
,MSGL_ERR
, "demux_lavf, couldn't set option probesize to %u\n", opt_probesize
);
447 if(opt_analyzeduration
) {
448 opt
= av_set_int(avfc
, "analyzeduration", opt_analyzeduration
* AV_TIME_BASE
);
449 if(!opt
) mp_msg(MSGT_HEADER
,MSGL_ERR
, "demux_lavf, couldn't set option analyzeduration to %u\n", opt_analyzeduration
);
453 if(parse_avopts(avfc
, opt_avopt
) < 0){
454 mp_msg(MSGT_HEADER
,MSGL_ERR
, "Your options /%s/ look like gibberish to me pal\n", opt_avopt
);
459 if(demuxer
->stream
->url
)
460 strncpy(mp_filename
+ 3, demuxer
->stream
->url
, sizeof(mp_filename
)-3);
462 strncpy(mp_filename
+ 3, "foobar.dummy", sizeof(mp_filename
)-3);
464 priv
->pb
= av_alloc_put_byte(priv
->buffer
, BIO_BUFFER_SIZE
, 0,
465 demuxer
->stream
, mp_read
, NULL
, mp_seek
);
466 priv
->pb
->is_streamed
= !demuxer
->stream
->end_pos
|| (demuxer
->stream
->flags
& STREAM_SEEK
) != STREAM_SEEK
;
468 if(av_open_input_stream(&avfc
, priv
->pb
, mp_filename
, priv
->avif
, &ap
)<0){
469 mp_msg(MSGT_HEADER
,MSGL_ERR
,"LAVF_header: av_open_input_stream() failed\n");
475 if(av_find_stream_info(avfc
) < 0){
476 mp_msg(MSGT_HEADER
,MSGL_ERR
,"LAVF_header: av_find_stream_info() failed\n");
480 if(avfc
->title
[0]) demux_info_add(demuxer
, "title" , avfc
->title
);
481 if(avfc
->author
[0]) demux_info_add(demuxer
, "author" , avfc
->author
);
482 if(avfc
->copyright
[0]) demux_info_add(demuxer
, "copyright", avfc
->copyright
);
483 if(avfc
->comment
[0]) demux_info_add(demuxer
, "comments" , avfc
->comment
);
484 if(avfc
->album
[0]) demux_info_add(demuxer
, "album" , avfc
->album
);
485 // if(avfc->year ) demux_info_add(demuxer, "year" , avfc->year );
486 // if(avfc->track ) demux_info_add(demuxer, "track" , avfc->track );
487 if(avfc
->genre
[0]) demux_info_add(demuxer
, "genre" , avfc
->genre
);
489 for(i
=0; i
< avfc
->nb_chapters
; i
++) {
490 AVChapter
*c
= avfc
->chapters
[i
];
491 uint64_t start
= av_rescale_q(c
->start
, c
->time_base
, (AVRational
){1,1000});
492 uint64_t end
= av_rescale_q(c
->end
, c
->time_base
, (AVRational
){1,1000});
493 demuxer_add_chapter(demuxer
, c
->title
, start
, end
);
496 if(avfc
->nb_programs
) {
497 int p
, start
=0, found
=0;
500 for(p
=0; p
<avfc
->nb_programs
; p
++) {
501 if(avfc
->programs
[p
]->id
== ts_prog
) {
508 mp_msg(MSGT_HEADER
,MSGL_ERR
,"DEMUX_LAVF: program %d doesn't seem to be present\n", ts_prog
);
514 AVProgram
*program
= avfc
->programs
[p
];
515 mp_msg(MSGT_HEADER
,MSGL_INFO
,"LAVF: Program %d %s\n", program
->id
, (program
->name
? program
->name
: ""));
516 for(i
=0; i
<program
->nb_stream_indexes
; i
++)
517 handle_stream(demuxer
, avfc
, program
->stream_index
[i
]);
518 if(!priv
->cur_program
&& (demuxer
->video
->sh
|| demuxer
->audio
->sh
))
519 priv
->cur_program
= program
->id
;
520 p
= (p
+ 1) % avfc
->nb_programs
;
523 for(i
=0; i
<avfc
->nb_streams
; i
++)
524 handle_stream(demuxer
, avfc
, i
);
526 mp_msg(MSGT_HEADER
,MSGL_V
,"LAVF: %d audio and %d video streams found\n",priv
->audio_streams
,priv
->video_streams
);
527 mp_msg(MSGT_HEADER
,MSGL_V
,"LAVF: build %d\n", LIBAVFORMAT_BUILD
);
528 if(!priv
->audio_streams
) demuxer
->audio
->id
=-2; // nosound
529 // else if(best_audio > 0 && demuxer->audio->id == -1) demuxer->audio->id=best_audio;
530 if(!priv
->video_streams
){
531 if(!priv
->audio_streams
){
532 mp_msg(MSGT_HEADER
,MSGL_ERR
,"LAVF: no audio or video headers found - broken file?\n");
535 demuxer
->video
->id
=-2; // audio-only
536 } //else if (best_video > 0 && demuxer->video->id == -1) demuxer->video->id = best_video;
538 demuxer
->accurate_seek
= true;
543 static int demux_lavf_fill_buffer(demuxer_t
*demux
, demux_stream_t
*dsds
){
544 lavf_priv_t
*priv
= demux
->priv
;
549 mp_msg(MSGT_DEMUX
,MSGL_DBG2
,"demux_lavf_fill_buffer()\n");
551 demux
->filepos
=stream_tell(demux
->stream
);
553 if(av_read_frame(priv
->avfc
, &pkt
) < 0)
556 id
= pkt
.stream_index
;
558 if(id
==demux
->audio
->id
){
562 ds
->sh
=demux
->a_streams
[id
];
563 mp_msg(MSGT_DEMUX
,MSGL_V
,"Auto-selected LAVF audio ID = %d\n",ds
->id
);
565 } else if(id
==demux
->video
->id
){
569 ds
->sh
=demux
->v_streams
[id
];
570 mp_msg(MSGT_DEMUX
,MSGL_V
,"Auto-selected LAVF video ID = %d\n",ds
->id
);
572 } else if(id
==demux
->sub
->id
){
577 av_free_packet(&pkt
);
581 if(0/*pkt.destruct == av_destruct_packet*/){
582 //ok kids, dont try this at home :)
583 dp
=malloc(sizeof(demux_packet_t
));
591 dp
=new_demux_packet(pkt
.size
);
592 memcpy(dp
->buffer
, pkt
.data
, pkt
.size
);
593 av_free_packet(&pkt
);
596 if(pkt
.pts
!= AV_NOPTS_VALUE
){
597 dp
->pts
=pkt
.pts
* av_q2d(priv
->avfc
->streams
[id
]->time_base
);
598 priv
->last_pts
= dp
->pts
* AV_TIME_BASE
;
599 if(pkt
.convergence_duration
)
600 dp
->endpts
= dp
->pts
+ pkt
.convergence_duration
* av_q2d(priv
->avfc
->streams
[id
]->time_base
);
602 dp
->pos
=demux
->filepos
;
603 dp
->flags
= !!(pkt
.flags
&PKT_FLAG_KEY
);
604 // append packet to DS stream:
605 ds_add_packet(ds
,dp
);
609 static void demux_seek_lavf(demuxer_t
*demuxer
, float rel_seek_secs
, float audio_delay
, int flags
){
610 lavf_priv_t
*priv
= demuxer
->priv
;
612 mp_msg(MSGT_DEMUX
,MSGL_DBG2
,"demux_seek_lavf(%p, %f, %f, %d)\n", demuxer
, rel_seek_secs
, audio_delay
, flags
);
614 if (flags
& SEEK_ABSOLUTE
) {
617 if (rel_seek_secs
< 0) avsflags
= AVSEEK_FLAG_BACKWARD
;
619 if (flags
& SEEK_FORWARD
)
621 else if (flags
& SEEK_BACKWARD
)
622 avsflags
= AVSEEK_FLAG_BACKWARD
;
623 if (flags
& SEEK_FACTOR
) {
624 if (priv
->avfc
->duration
== 0 || priv
->avfc
->duration
== AV_NOPTS_VALUE
)
626 priv
->last_pts
+= rel_seek_secs
* priv
->avfc
->duration
;
628 priv
->last_pts
+= rel_seek_secs
* AV_TIME_BASE
;
630 av_seek_frame(priv
->avfc
, -1, priv
->last_pts
, avsflags
);
633 static int demux_lavf_control(demuxer_t
*demuxer
, int cmd
, void *arg
)
635 lavf_priv_t
*priv
= demuxer
->priv
;
638 case DEMUXER_CTRL_CORRECT_PTS
:
639 return DEMUXER_CTRL_OK
;
640 case DEMUXER_CTRL_GET_TIME_LENGTH
:
641 if (priv
->avfc
->duration
== 0 || priv
->avfc
->duration
== AV_NOPTS_VALUE
)
642 return DEMUXER_CTRL_DONTKNOW
;
644 *((double *)arg
) = (double)priv
->avfc
->duration
/ AV_TIME_BASE
;
645 return DEMUXER_CTRL_OK
;
647 case DEMUXER_CTRL_GET_PERCENT_POS
:
648 if (priv
->avfc
->duration
== 0 || priv
->avfc
->duration
== AV_NOPTS_VALUE
)
649 return DEMUXER_CTRL_DONTKNOW
;
651 *((int *)arg
) = (int)((priv
->last_pts
- priv
->avfc
->start_time
)*100 / priv
->avfc
->duration
);
652 return DEMUXER_CTRL_OK
;
653 case DEMUXER_CTRL_SWITCH_AUDIO
:
654 case DEMUXER_CTRL_SWITCH_VIDEO
:
656 int id
= *((int*)arg
);
659 int nstreams
, *pstreams
;
662 if(cmd
== DEMUXER_CTRL_SWITCH_VIDEO
)
665 nstreams
= priv
->video_streams
;
666 pstreams
= priv
->vstreams
;
671 nstreams
= priv
->audio_streams
;
672 pstreams
= priv
->astreams
;
674 for(i
= 0; i
< nstreams
; i
++)
676 if(pstreams
[i
] == ds
->id
) //current stream id
683 if(id
== -2) { // no sound
685 } else if(id
== -1) { // next track
686 i
= (curridx
+ 2) % (nstreams
+ 1) - 1;
690 else // select track by id
692 for(i
= 0; i
< nstreams
; i
++)
694 if(pstreams
[i
] == id
)
702 return DEMUXER_CTRL_NOTIMPL
;
707 priv
->avfc
->streams
[ds
->id
]->discard
= AVDISCARD_ALL
;
708 *((int*)arg
) = ds
->id
= newid
;
710 priv
->avfc
->streams
[newid
]->discard
= AVDISCARD_NONE
;
711 return DEMUXER_CTRL_OK
;
714 case DEMUXER_CTRL_IDENTIFY_PROGRAM
:
716 demux_program_t
*prog
= arg
;
720 if(priv
->avfc
->nb_programs
< 2)
721 return DEMUXER_CTRL_NOTIMPL
;
723 if(prog
->progid
== -1)
726 while(p
<priv
->avfc
->nb_programs
&& priv
->avfc
->programs
[p
]->id
!= priv
->cur_program
)
728 p
= (p
+ 1) % priv
->avfc
->nb_programs
;
732 for(i
=0; i
<priv
->avfc
->nb_programs
; i
++)
733 if(priv
->avfc
->programs
[i
]->id
== prog
->progid
)
735 if(i
==priv
->avfc
->nb_programs
)
736 return DEMUXER_CTRL_NOTIMPL
;
739 prog
->vid
= prog
->aid
= prog
->sid
= -2; //no audio and no video by default
741 program
= priv
->avfc
->programs
[p
];
742 for(i
=0; i
<program
->nb_stream_indexes
; i
++)
744 switch(priv
->avfc
->streams
[program
->stream_index
[i
]]->codec
->codec_type
)
746 case CODEC_TYPE_VIDEO
:
748 prog
->vid
= program
->stream_index
[i
];
750 case CODEC_TYPE_AUDIO
:
752 prog
->aid
= program
->stream_index
[i
];
754 case CODEC_TYPE_SUBTITLE
:
755 if(prog
->sid
== -2 && priv
->avfc
->streams
[program
->stream_index
[i
]]->codec
->codec_id
== CODEC_ID_TEXT
)
756 prog
->sid
= program
->stream_index
[i
];
760 if(prog
->progid
== -1 && prog
->vid
== -2 && prog
->aid
== -2)
762 p
= (p
+ 1) % priv
->avfc
->nb_programs
;
765 priv
->cur_program
= prog
->progid
= program
->id
;
766 return DEMUXER_CTRL_OK
;
769 return DEMUXER_CTRL_NOTIMPL
;
773 static void demux_close_lavf(demuxer_t
*demuxer
)
775 lavf_priv_t
* priv
= demuxer
->priv
;
779 av_freep(&priv
->avfc
->key
);
780 av_close_input_stream(priv
->avfc
);
783 free(priv
); demuxer
->priv
= NULL
;
788 const demuxer_desc_t demuxer_desc_lavf
= {
789 "libavformat demuxer",
792 "Michael Niedermayer",
793 "supports many formats, requires libavformat",
795 0, // Check after other demuxer
797 demux_lavf_fill_buffer
,
804 const demuxer_desc_t demuxer_desc_lavf_preferred
= {
805 "libavformat preferred demuxer",
808 "Michael Niedermayer",
809 "supports many formats, requires libavformat",
810 DEMUXER_TYPE_LAVF_PREFERRED
,
812 lavf_check_preferred_file
,
813 demux_lavf_fill_buffer
,