2 * various utility functions for use within Libav
3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
5 * This file is part of Libav.
7 * Libav is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * Libav 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 GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with Libav; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
25 #include "avio_internal.h"
27 #include "libavcodec/internal.h"
28 #include "libavcodec/bytestream.h"
29 #include "libavutil/opt.h"
30 #include "libavutil/dict.h"
31 #include "libavutil/pixdesc.h"
34 #include "libavutil/avassert.h"
35 #include "libavutil/avstring.h"
36 #include "libavutil/mathematics.h"
37 #include "libavutil/parseutils.h"
38 #include "libavutil/time.h"
40 #include "audiointerleave.h"
52 * various utility functions for use within Libav
55 unsigned avformat_version(void)
57 return LIBAVFORMAT_VERSION_INT;
60 const char *avformat_configuration(void)
62 return LIBAV_CONFIGURATION;
65 const char *avformat_license(void)
67 #define LICENSE_PREFIX "libavformat license: "
68 return LICENSE_PREFIX LIBAV_LICENSE + sizeof(LICENSE_PREFIX) - 1;
71 /* fraction handling */
74 * f = val + (num / den) + 0.5.
76 * 'num' is normalized so that it is such as 0 <= num < den.
78 * @param f fractional number
79 * @param val integer value
80 * @param num must be >= 0
81 * @param den must be >= 1
83 static void frac_init(AVFrac *f, int64_t val, int64_t num, int64_t den)
96 * Fractional addition to f: f = f + (incr / f->den).
98 * @param f fractional number
99 * @param incr increment, can be positive or negative
101 static void frac_add(AVFrac *f, int64_t incr)
114 } else if (num >= den) {
121 /** head of registered input format linked list */
122 static AVInputFormat *first_iformat = NULL;
123 /** head of registered output format linked list */
124 static AVOutputFormat *first_oformat = NULL;
126 AVInputFormat *av_iformat_next(AVInputFormat *f)
128 if(f) return f->next;
129 else return first_iformat;
132 AVOutputFormat *av_oformat_next(AVOutputFormat *f)
134 if(f) return f->next;
135 else return first_oformat;
138 void av_register_input_format(AVInputFormat *format)
142 while (*p != NULL) p = &(*p)->next;
147 void av_register_output_format(AVOutputFormat *format)
151 while (*p != NULL) p = &(*p)->next;
156 int av_match_ext(const char *filename, const char *extensions)
164 ext = strrchr(filename, '.');
170 while (*p != '\0' && *p != ',' && q-ext1<sizeof(ext1)-1)
173 if (!av_strcasecmp(ext1, ext))
183 static int match_format(const char *name, const char *names)
191 namelen = strlen(name);
192 while ((p = strchr(names, ','))) {
193 len = FFMAX(p - names, namelen);
194 if (!av_strncasecmp(name, names, len))
198 return !av_strcasecmp(name, names);
201 AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
202 const char *mime_type)
204 AVOutputFormat *fmt = NULL, *fmt_found;
205 int score_max, score;
207 /* specific test for image sequences */
208 #if CONFIG_IMAGE2_MUXER
209 if (!short_name && filename &&
210 av_filename_number_test(filename) &&
211 ff_guess_image2_codec(filename) != AV_CODEC_ID_NONE) {
212 return av_guess_format("image2", NULL, NULL);
215 /* Find the proper file type. */
218 while ((fmt = av_oformat_next(fmt))) {
220 if (fmt->name && short_name && !av_strcasecmp(fmt->name, short_name))
222 if (fmt->mime_type && mime_type && !strcmp(fmt->mime_type, mime_type))
224 if (filename && fmt->extensions &&
225 av_match_ext(filename, fmt->extensions)) {
228 if (score > score_max) {
236 enum AVCodecID av_guess_codec(AVOutputFormat *fmt, const char *short_name,
237 const char *filename, const char *mime_type, enum AVMediaType type){
238 if(type == AVMEDIA_TYPE_VIDEO){
239 enum AVCodecID codec_id= AV_CODEC_ID_NONE;
241 #if CONFIG_IMAGE2_MUXER
242 if(!strcmp(fmt->name, "image2") || !strcmp(fmt->name, "image2pipe")){
243 codec_id= ff_guess_image2_codec(filename);
246 if(codec_id == AV_CODEC_ID_NONE)
247 codec_id= fmt->video_codec;
249 }else if(type == AVMEDIA_TYPE_AUDIO)
250 return fmt->audio_codec;
251 else if (type == AVMEDIA_TYPE_SUBTITLE)
252 return fmt->subtitle_codec;
254 return AV_CODEC_ID_NONE;
257 AVInputFormat *av_find_input_format(const char *short_name)
259 AVInputFormat *fmt = NULL;
260 while ((fmt = av_iformat_next(fmt))) {
261 if (match_format(short_name, fmt->name))
268 int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
270 int ret= av_new_packet(pkt, size);
275 pkt->pos= avio_tell(s);
277 ret= avio_read(s, pkt->data, size);
281 av_shrink_packet(pkt, ret);
286 int av_append_packet(AVIOContext *s, AVPacket *pkt, int size)
291 return av_get_packet(s, pkt, size);
292 old_size = pkt->size;
293 ret = av_grow_packet(pkt, size);
296 ret = avio_read(s, pkt->data + old_size, size);
297 av_shrink_packet(pkt, old_size + FFMAX(ret, 0));
302 int av_filename_number_test(const char *filename)
305 return filename && (av_get_frame_filename(buf, sizeof(buf), filename, 1)>=0);
308 AVInputFormat *av_probe_input_format2(AVProbeData *pd, int is_opened, int *score_max)
310 AVProbeData lpd = *pd;
311 AVInputFormat *fmt1 = NULL, *fmt;
314 if (lpd.buf_size > 10 && ff_id3v2_match(lpd.buf, ID3v2_DEFAULT_MAGIC)) {
315 int id3len = ff_id3v2_tag_len(lpd.buf);
316 if (lpd.buf_size > id3len + 16) {
318 lpd.buf_size -= id3len;
324 while ((fmt1 = av_iformat_next(fmt1))) {
325 if (!is_opened == !(fmt1->flags & AVFMT_NOFILE))
328 if (fmt1->read_probe) {
329 score = fmt1->read_probe(&lpd);
330 } else if (fmt1->extensions) {
331 if (av_match_ext(lpd.filename, fmt1->extensions)) {
335 if (score > *score_max) {
338 }else if (score == *score_max)
342 /* a hack for files with huge id3v2 tags -- try to guess by file extension. */
343 if (!fmt && is_opened && *score_max < AVPROBE_SCORE_MAX/4) {
344 while ((fmt = av_iformat_next(fmt)))
345 if (fmt->extensions && av_match_ext(lpd.filename, fmt->extensions)) {
346 *score_max = AVPROBE_SCORE_MAX/4;
351 if (!fmt && id3 && *score_max < AVPROBE_SCORE_MAX/4-1) {
352 while ((fmt = av_iformat_next(fmt)))
353 if (fmt->extensions && av_match_ext("mp3", fmt->extensions)) {
354 *score_max = AVPROBE_SCORE_MAX/4-1;
362 AVInputFormat *av_probe_input_format(AVProbeData *pd, int is_opened){
364 return av_probe_input_format2(pd, is_opened, &score);
367 static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st, AVProbeData *pd, int score)
369 static const struct {
370 const char *name; enum AVCodecID id; enum AVMediaType type;
372 { "aac" , AV_CODEC_ID_AAC , AVMEDIA_TYPE_AUDIO },
373 { "ac3" , AV_CODEC_ID_AC3 , AVMEDIA_TYPE_AUDIO },
374 { "dts" , AV_CODEC_ID_DTS , AVMEDIA_TYPE_AUDIO },
375 { "eac3" , AV_CODEC_ID_EAC3 , AVMEDIA_TYPE_AUDIO },
376 { "h264" , AV_CODEC_ID_H264 , AVMEDIA_TYPE_VIDEO },
377 { "m4v" , AV_CODEC_ID_MPEG4 , AVMEDIA_TYPE_VIDEO },
378 { "mp3" , AV_CODEC_ID_MP3 , AVMEDIA_TYPE_AUDIO },
379 { "mpegvideo", AV_CODEC_ID_MPEG2VIDEO, AVMEDIA_TYPE_VIDEO },
382 AVInputFormat *fmt = av_probe_input_format2(pd, 1, &score);
386 av_log(s, AV_LOG_DEBUG, "Probe with size=%d, packets=%d detected %s with score=%d\n",
387 pd->buf_size, MAX_PROBE_PACKETS - st->probe_packets, fmt->name, score);
388 for (i = 0; fmt_id_type[i].name; i++) {
389 if (!strcmp(fmt->name, fmt_id_type[i].name)) {
390 st->codec->codec_id = fmt_id_type[i].id;
391 st->codec->codec_type = fmt_id_type[i].type;
399 /************************************************************/
400 /* input media file */
402 /** size of probe buffer, for guessing file type from file contents */
403 #define PROBE_BUF_MIN 2048
404 #define PROBE_BUF_MAX (1<<20)
406 int av_probe_input_buffer(AVIOContext *pb, AVInputFormat **fmt,
407 const char *filename, void *logctx,
408 unsigned int offset, unsigned int max_probe_size)
410 AVProbeData pd = { filename ? filename : "", NULL, -offset };
411 unsigned char *buf = NULL;
412 int ret = 0, probe_size;
414 if (!max_probe_size) {
415 max_probe_size = PROBE_BUF_MAX;
416 } else if (max_probe_size > PROBE_BUF_MAX) {
417 max_probe_size = PROBE_BUF_MAX;
418 } else if (max_probe_size < PROBE_BUF_MIN) {
419 return AVERROR(EINVAL);
422 if (offset >= max_probe_size) {
423 return AVERROR(EINVAL);
426 for(probe_size= PROBE_BUF_MIN; probe_size<=max_probe_size && !*fmt;
427 probe_size = FFMIN(probe_size<<1, FFMAX(max_probe_size, probe_size+1))) {
428 int score = probe_size < max_probe_size ? AVPROBE_SCORE_MAX/4 : 0;
429 int buf_offset = (probe_size == PROBE_BUF_MIN) ? 0 : probe_size>>1;
431 if (probe_size < offset) {
435 /* read probe data */
436 buf = av_realloc(buf, probe_size + AVPROBE_PADDING_SIZE);
437 if ((ret = avio_read(pb, buf + buf_offset, probe_size - buf_offset)) < 0) {
438 /* fail if error was not end of file, otherwise, lower score */
439 if (ret != AVERROR_EOF) {
444 ret = 0; /* error was end of file, nothing read */
447 pd.buf = &buf[offset];
449 memset(pd.buf + pd.buf_size, 0, AVPROBE_PADDING_SIZE);
451 /* guess file format */
452 *fmt = av_probe_input_format2(&pd, 1, &score);
454 if(score <= AVPROBE_SCORE_MAX/4){ //this can only be true in the last iteration
455 av_log(logctx, AV_LOG_WARNING, "Format detected only with low score of %d, misdetection possible!\n", score);
457 av_log(logctx, AV_LOG_DEBUG, "Probed with size=%d and score=%d\n", probe_size, score);
463 return AVERROR_INVALIDDATA;
466 /* rewind. reuse probe buffer to avoid seeking */
467 if ((ret = ffio_rewind_with_probe_data(pb, buf, pd.buf_size)) < 0)
473 /* open input file and probe the format if necessary */
474 static int init_input(AVFormatContext *s, const char *filename, AVDictionary **options)
477 AVProbeData pd = {filename, NULL, 0};
480 s->flags |= AVFMT_FLAG_CUSTOM_IO;
482 return av_probe_input_buffer(s->pb, &s->iformat, filename, s, 0, s->probesize);
483 else if (s->iformat->flags & AVFMT_NOFILE)
484 return AVERROR(EINVAL);
488 if ( (s->iformat && s->iformat->flags & AVFMT_NOFILE) ||
489 (!s->iformat && (s->iformat = av_probe_input_format(&pd, 0))))
492 if ((ret = avio_open2(&s->pb, filename, AVIO_FLAG_READ,
493 &s->interrupt_callback, options)) < 0)
497 return av_probe_input_buffer(s->pb, &s->iformat, filename, s, 0, s->probesize);
500 static AVPacket *add_to_pktbuf(AVPacketList **packet_buffer, AVPacket *pkt,
501 AVPacketList **plast_pktl){
502 AVPacketList *pktl = av_mallocz(sizeof(AVPacketList));
507 (*plast_pktl)->next = pktl;
509 *packet_buffer = pktl;
511 /* add the packet in the buffered packet list */
517 static void queue_attached_pictures(AVFormatContext *s)
520 for (i = 0; i < s->nb_streams; i++)
521 if (s->streams[i]->disposition & AV_DISPOSITION_ATTACHED_PIC &&
522 s->streams[i]->discard < AVDISCARD_ALL) {
523 AVPacket copy = s->streams[i]->attached_pic;
524 copy.destruct = NULL;
525 add_to_pktbuf(&s->raw_packet_buffer, ©, &s->raw_packet_buffer_end);
529 int avformat_open_input(AVFormatContext **ps, const char *filename, AVInputFormat *fmt, AVDictionary **options)
531 AVFormatContext *s = *ps;
533 AVDictionary *tmp = NULL;
534 ID3v2ExtraMeta *id3v2_extra_meta = NULL;
536 if (!s && !(s = avformat_alloc_context()))
537 return AVERROR(ENOMEM);
542 av_dict_copy(&tmp, *options, 0);
544 if ((ret = av_opt_set_dict(s, &tmp)) < 0)
547 if ((ret = init_input(s, filename, &tmp)) < 0)
550 /* check filename in case an image number is expected */
551 if (s->iformat->flags & AVFMT_NEEDNUMBER) {
552 if (!av_filename_number_test(filename)) {
553 ret = AVERROR(EINVAL);
558 s->duration = s->start_time = AV_NOPTS_VALUE;
559 av_strlcpy(s->filename, filename ? filename : "", sizeof(s->filename));
561 /* allocate private data */
562 if (s->iformat->priv_data_size > 0) {
563 if (!(s->priv_data = av_mallocz(s->iformat->priv_data_size))) {
564 ret = AVERROR(ENOMEM);
567 if (s->iformat->priv_class) {
568 *(const AVClass**)s->priv_data = s->iformat->priv_class;
569 av_opt_set_defaults(s->priv_data);
570 if ((ret = av_opt_set_dict(s->priv_data, &tmp)) < 0)
575 /* e.g. AVFMT_NOFILE formats will not have a AVIOContext */
577 ff_id3v2_read(s, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta);
579 if (s->iformat->read_header)
580 if ((ret = s->iformat->read_header(s)) < 0)
583 if (id3v2_extra_meta &&
584 (ret = ff_id3v2_parse_apic(s, &id3v2_extra_meta)) < 0)
586 ff_id3v2_free_extra_meta(&id3v2_extra_meta);
588 queue_attached_pictures(s);
590 if (s->pb && !s->data_offset)
591 s->data_offset = avio_tell(s->pb);
593 s->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
596 av_dict_free(options);
603 ff_id3v2_free_extra_meta(&id3v2_extra_meta);
605 if (s->pb && !(s->flags & AVFMT_FLAG_CUSTOM_IO))
607 avformat_free_context(s);
612 /*******************************************************/
614 static void probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
616 if(st->codec->codec_id == AV_CODEC_ID_PROBE){
617 AVProbeData *pd = &st->probe_data;
618 av_log(s, AV_LOG_DEBUG, "probing stream %d\n", st->index);
622 pd->buf = av_realloc(pd->buf, pd->buf_size+pkt->size+AVPROBE_PADDING_SIZE);
623 memcpy(pd->buf+pd->buf_size, pkt->data, pkt->size);
624 pd->buf_size += pkt->size;
625 memset(pd->buf+pd->buf_size, 0, AVPROBE_PADDING_SIZE);
627 st->probe_packets = 0;
629 av_log(s, AV_LOG_ERROR, "nothing to probe for stream %d\n",
635 if (!st->probe_packets ||
636 av_log2(pd->buf_size) != av_log2(pd->buf_size - pkt->size)) {
637 set_codec_from_probe_data(s, st, pd, st->probe_packets > 0 ? AVPROBE_SCORE_MAX/4 : 0);
638 if(st->codec->codec_id != AV_CODEC_ID_PROBE){
641 av_log(s, AV_LOG_DEBUG, "probed stream %d\n", st->index);
647 int ff_read_packet(AVFormatContext *s, AVPacket *pkt)
653 AVPacketList *pktl = s->raw_packet_buffer;
657 st = s->streams[pkt->stream_index];
658 if (st->codec->codec_id != AV_CODEC_ID_PROBE || !st->probe_packets ||
659 s->raw_packet_buffer_remaining_size < pkt->size) {
661 if (st->probe_packets) {
662 probe_codec(s, st, NULL);
664 pd = &st->probe_data;
667 s->raw_packet_buffer = pktl->next;
668 s->raw_packet_buffer_remaining_size += pkt->size;
675 ret= s->iformat->read_packet(s, pkt);
677 if (!pktl || ret == AVERROR(EAGAIN))
679 for (i = 0; i < s->nb_streams; i++) {
681 if (st->probe_packets) {
682 probe_codec(s, st, NULL);
688 if ((s->flags & AVFMT_FLAG_DISCARD_CORRUPT) &&
689 (pkt->flags & AV_PKT_FLAG_CORRUPT)) {
690 av_log(s, AV_LOG_WARNING,
691 "Dropped corrupted packet (stream = %d)\n",
697 st= s->streams[pkt->stream_index];
699 switch(st->codec->codec_type){
700 case AVMEDIA_TYPE_VIDEO:
701 if(s->video_codec_id) st->codec->codec_id= s->video_codec_id;
703 case AVMEDIA_TYPE_AUDIO:
704 if(s->audio_codec_id) st->codec->codec_id= s->audio_codec_id;
706 case AVMEDIA_TYPE_SUBTITLE:
707 if(s->subtitle_codec_id)st->codec->codec_id= s->subtitle_codec_id;
711 if(!pktl && (st->codec->codec_id != AV_CODEC_ID_PROBE ||
715 add_to_pktbuf(&s->raw_packet_buffer, pkt, &s->raw_packet_buffer_end);
716 s->raw_packet_buffer_remaining_size -= pkt->size;
718 probe_codec(s, st, pkt);
722 #if FF_API_READ_PACKET
723 int av_read_packet(AVFormatContext *s, AVPacket *pkt)
725 return ff_read_packet(s, pkt);
730 /**********************************************************/
733 * Get the number of samples of an audio frame. Return -1 on error.
735 static int get_audio_frame_size(AVCodecContext *enc, int size, int mux)
739 /* give frame_size priority if demuxing */
740 if (!mux && enc->frame_size > 1)
741 return enc->frame_size;
743 if ((frame_size = av_get_audio_frame_duration(enc, size)) > 0)
746 /* fallback to using frame_size if muxing */
747 if (enc->frame_size > 1)
748 return enc->frame_size;
755 * Return the frame duration in seconds. Return 0 if not available.
757 static void compute_frame_duration(int *pnum, int *pden, AVStream *st,
758 AVCodecParserContext *pc, AVPacket *pkt)
764 switch(st->codec->codec_type) {
765 case AVMEDIA_TYPE_VIDEO:
766 if (st->avg_frame_rate.num) {
767 *pnum = st->avg_frame_rate.den;
768 *pden = st->avg_frame_rate.num;
769 } else if(st->time_base.num*1000LL > st->time_base.den) {
770 *pnum = st->time_base.num;
771 *pden = st->time_base.den;
772 }else if(st->codec->time_base.num*1000LL > st->codec->time_base.den){
773 *pnum = st->codec->time_base.num;
774 *pden = st->codec->time_base.den;
775 if (pc && pc->repeat_pict) {
776 *pnum = (*pnum) * (1 + pc->repeat_pict);
778 //If this codec can be interlaced or progressive then we need a parser to compute duration of a packet
779 //Thus if we have no parser in such case leave duration undefined.
780 if(st->codec->ticks_per_frame>1 && !pc){
785 case AVMEDIA_TYPE_AUDIO:
786 frame_size = get_audio_frame_size(st->codec, pkt->size, 0);
787 if (frame_size <= 0 || st->codec->sample_rate <= 0)
790 *pden = st->codec->sample_rate;
797 static int is_intra_only(enum AVCodecID id)
799 const AVCodecDescriptor *d = avcodec_descriptor_get(id);
802 if (d->type == AVMEDIA_TYPE_VIDEO && !(d->props & AV_CODEC_PROP_INTRA_ONLY))
807 static void update_initial_timestamps(AVFormatContext *s, int stream_index,
808 int64_t dts, int64_t pts)
810 AVStream *st= s->streams[stream_index];
811 AVPacketList *pktl= s->packet_buffer;
813 if(st->first_dts != AV_NOPTS_VALUE || dts == AV_NOPTS_VALUE || st->cur_dts == AV_NOPTS_VALUE)
816 st->first_dts= dts - st->cur_dts;
819 for(; pktl; pktl= pktl->next){
820 if(pktl->pkt.stream_index != stream_index)
822 //FIXME think more about this check
823 if(pktl->pkt.pts != AV_NOPTS_VALUE && pktl->pkt.pts == pktl->pkt.dts)
824 pktl->pkt.pts += st->first_dts;
826 if(pktl->pkt.dts != AV_NOPTS_VALUE)
827 pktl->pkt.dts += st->first_dts;
829 if(st->start_time == AV_NOPTS_VALUE && pktl->pkt.pts != AV_NOPTS_VALUE)
830 st->start_time= pktl->pkt.pts;
832 if (st->start_time == AV_NOPTS_VALUE)
833 st->start_time = pts;
836 static void update_initial_durations(AVFormatContext *s, AVStream *st,
837 int stream_index, int duration)
839 AVPacketList *pktl= s->packet_buffer;
842 if(st->first_dts != AV_NOPTS_VALUE){
843 cur_dts= st->first_dts;
844 for(; pktl; pktl= pktl->next){
845 if(pktl->pkt.stream_index == stream_index){
846 if(pktl->pkt.pts != pktl->pkt.dts || pktl->pkt.dts != AV_NOPTS_VALUE || pktl->pkt.duration)
851 pktl= s->packet_buffer;
852 st->first_dts = cur_dts;
853 }else if(st->cur_dts)
856 for(; pktl; pktl= pktl->next){
857 if(pktl->pkt.stream_index != stream_index)
859 if(pktl->pkt.pts == pktl->pkt.dts && pktl->pkt.dts == AV_NOPTS_VALUE
860 && !pktl->pkt.duration){
861 pktl->pkt.dts= cur_dts;
862 if(!st->codec->has_b_frames)
863 pktl->pkt.pts= cur_dts;
865 if (st->codec->codec_type != AVMEDIA_TYPE_AUDIO)
866 pktl->pkt.duration = duration;
870 if(st->first_dts == AV_NOPTS_VALUE)
871 st->cur_dts= cur_dts;
874 static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
875 AVCodecParserContext *pc, AVPacket *pkt)
877 int num, den, presentation_delayed, delay, i;
880 if (s->flags & AVFMT_FLAG_NOFILLIN)
883 if((s->flags & AVFMT_FLAG_IGNDTS) && pkt->pts != AV_NOPTS_VALUE)
884 pkt->dts= AV_NOPTS_VALUE;
886 /* do we have a video B-frame ? */
887 delay= st->codec->has_b_frames;
888 presentation_delayed = 0;
890 /* XXX: need has_b_frame, but cannot get it if the codec is
893 pc && pc->pict_type != AV_PICTURE_TYPE_B)
894 presentation_delayed = 1;
896 if(pkt->pts != AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE && pkt->dts > pkt->pts && st->pts_wrap_bits<63
897 /*&& pkt->dts-(1LL<<st->pts_wrap_bits) < pkt->pts*/){
898 pkt->dts -= 1LL<<st->pts_wrap_bits;
901 // some mpeg2 in mpeg-ps lack dts (issue171 / input_file.mpg)
902 // we take the conservative approach and discard both
903 // Note, if this is misbehaving for a H.264 file then possibly presentation_delayed is not set correctly.
904 if(delay==1 && pkt->dts == pkt->pts && pkt->dts != AV_NOPTS_VALUE && presentation_delayed){
905 av_log(s, AV_LOG_DEBUG, "invalid dts/pts combination\n");
906 pkt->dts= pkt->pts= AV_NOPTS_VALUE;
909 if (pkt->duration == 0 && st->codec->codec_type != AVMEDIA_TYPE_AUDIO) {
910 compute_frame_duration(&num, &den, st, pc, pkt);
912 pkt->duration = av_rescale_rnd(1, num * (int64_t)st->time_base.den, den * (int64_t)st->time_base.num, AV_ROUND_DOWN);
914 if(pkt->duration != 0 && s->packet_buffer)
915 update_initial_durations(s, st, pkt->stream_index, pkt->duration);
919 /* correct timestamps with byte offset if demuxers only have timestamps
920 on packet boundaries */
921 if(pc && st->need_parsing == AVSTREAM_PARSE_TIMESTAMPS && pkt->size){
922 /* this will estimate bitrate based on this frame's duration and size */
923 offset = av_rescale(pc->offset, pkt->duration, pkt->size);
924 if(pkt->pts != AV_NOPTS_VALUE)
926 if(pkt->dts != AV_NOPTS_VALUE)
930 if (pc && pc->dts_sync_point >= 0) {
931 // we have synchronization info from the parser
932 int64_t den = st->codec->time_base.den * (int64_t) st->time_base.num;
934 int64_t num = st->codec->time_base.num * (int64_t) st->time_base.den;
935 if (pkt->dts != AV_NOPTS_VALUE) {
936 // got DTS from the stream, update reference timestamp
937 st->reference_dts = pkt->dts - pc->dts_ref_dts_delta * num / den;
938 pkt->pts = pkt->dts + pc->pts_dts_delta * num / den;
939 } else if (st->reference_dts != AV_NOPTS_VALUE) {
940 // compute DTS based on reference timestamp
941 pkt->dts = st->reference_dts + pc->dts_ref_dts_delta * num / den;
942 pkt->pts = pkt->dts + pc->pts_dts_delta * num / den;
944 if (pc->dts_sync_point > 0)
945 st->reference_dts = pkt->dts; // new reference
949 /* This may be redundant, but it should not hurt. */
950 if(pkt->dts != AV_NOPTS_VALUE && pkt->pts != AV_NOPTS_VALUE && pkt->pts > pkt->dts)
951 presentation_delayed = 1;
953 // av_log(NULL, AV_LOG_DEBUG, "IN delayed:%d pts:%"PRId64", dts:%"PRId64" cur_dts:%"PRId64" st:%d pc:%p\n", presentation_delayed, pkt->pts, pkt->dts, st->cur_dts, pkt->stream_index, pc);
954 /* interpolate PTS and DTS if they are not present */
955 //We skip H264 currently because delay and has_b_frames are not reliably set
956 if((delay==0 || (delay==1 && pc)) && st->codec->codec_id != AV_CODEC_ID_H264){
957 if (presentation_delayed) {
958 /* DTS = decompression timestamp */
959 /* PTS = presentation timestamp */
960 if (pkt->dts == AV_NOPTS_VALUE)
961 pkt->dts = st->last_IP_pts;
962 update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts);
963 if (pkt->dts == AV_NOPTS_VALUE)
964 pkt->dts = st->cur_dts;
966 /* this is tricky: the dts must be incremented by the duration
967 of the frame we are displaying, i.e. the last I- or P-frame */
968 if (st->last_IP_duration == 0)
969 st->last_IP_duration = pkt->duration;
970 if(pkt->dts != AV_NOPTS_VALUE)
971 st->cur_dts = pkt->dts + st->last_IP_duration;
972 st->last_IP_duration = pkt->duration;
973 st->last_IP_pts= pkt->pts;
974 /* cannot compute PTS if not present (we can compute it only
975 by knowing the future */
976 } else if (pkt->pts != AV_NOPTS_VALUE ||
977 pkt->dts != AV_NOPTS_VALUE ||
979 st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
980 int duration = pkt->duration;
981 if (!duration && st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
982 compute_frame_duration(&num, &den, st, pc, pkt);
984 duration = av_rescale_rnd(1, num * (int64_t)st->time_base.den,
985 den * (int64_t)st->time_base.num,
987 if (duration != 0 && s->packet_buffer) {
988 update_initial_durations(s, st, pkt->stream_index,
994 if (pkt->pts != AV_NOPTS_VALUE || pkt->dts != AV_NOPTS_VALUE ||
996 /* presentation is not delayed : PTS and DTS are the same */
997 if (pkt->pts == AV_NOPTS_VALUE)
999 update_initial_timestamps(s, pkt->stream_index, pkt->pts,
1001 if (pkt->pts == AV_NOPTS_VALUE)
1002 pkt->pts = st->cur_dts;
1003 pkt->dts = pkt->pts;
1004 if (pkt->pts != AV_NOPTS_VALUE)
1005 st->cur_dts = pkt->pts + duration;
1010 if(pkt->pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY){
1011 st->pts_buffer[0]= pkt->pts;
1012 for(i=0; i<delay && st->pts_buffer[i] > st->pts_buffer[i+1]; i++)
1013 FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i+1]);
1014 if(pkt->dts == AV_NOPTS_VALUE)
1015 pkt->dts= st->pts_buffer[0];
1016 if(st->codec->codec_id == AV_CODEC_ID_H264){ // we skipped it above so we try here
1017 update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts); // this should happen on the first packet
1019 if(pkt->dts > st->cur_dts)
1020 st->cur_dts = pkt->dts;
1023 // av_log(NULL, AV_LOG_ERROR, "OUTdelayed:%d/%d pts:%"PRId64", dts:%"PRId64" cur_dts:%"PRId64"\n", presentation_delayed, delay, pkt->pts, pkt->dts, st->cur_dts);
1026 if (is_intra_only(st->codec->codec_id))
1027 pkt->flags |= AV_PKT_FLAG_KEY;
1029 pkt->convergence_duration = pc->convergence_duration;
1032 static void free_packet_buffer(AVPacketList **pkt_buf, AVPacketList **pkt_buf_end)
1035 AVPacketList *pktl = *pkt_buf;
1036 *pkt_buf = pktl->next;
1037 av_free_packet(&pktl->pkt);
1040 *pkt_buf_end = NULL;
1044 * Parse a packet, add all split parts to parse_queue
1046 * @param pkt packet to parse, NULL when flushing the parser at end of stream
1048 static int parse_packet(AVFormatContext *s, AVPacket *pkt, int stream_index)
1050 AVPacket out_pkt = { 0 }, flush_pkt = { 0 };
1051 AVStream *st = s->streams[stream_index];
1052 uint8_t *data = pkt ? pkt->data : NULL;
1053 int size = pkt ? pkt->size : 0;
1054 int ret = 0, got_output = 0;
1057 av_init_packet(&flush_pkt);
1062 while (size > 0 || (pkt == &flush_pkt && got_output)) {
1065 av_init_packet(&out_pkt);
1066 len = av_parser_parse2(st->parser, st->codec,
1067 &out_pkt.data, &out_pkt.size, data, size,
1068 pkt->pts, pkt->dts, pkt->pos);
1070 pkt->pts = pkt->dts = AV_NOPTS_VALUE;
1071 /* increment read pointer */
1075 got_output = !!out_pkt.size;
1080 /* set the duration */
1081 out_pkt.duration = 0;
1082 if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
1083 if (st->codec->sample_rate > 0) {
1084 out_pkt.duration = av_rescale_q_rnd(st->parser->duration,
1085 (AVRational){ 1, st->codec->sample_rate },
1089 } else if (st->codec->time_base.num != 0 &&
1090 st->codec->time_base.den != 0) {
1091 out_pkt.duration = av_rescale_q_rnd(st->parser->duration,
1092 st->codec->time_base,
1097 out_pkt.stream_index = st->index;
1098 out_pkt.pts = st->parser->pts;
1099 out_pkt.dts = st->parser->dts;
1100 out_pkt.pos = st->parser->pos;
1102 if (st->parser->key_frame == 1 ||
1103 (st->parser->key_frame == -1 &&
1104 st->parser->pict_type == AV_PICTURE_TYPE_I))
1105 out_pkt.flags |= AV_PKT_FLAG_KEY;
1107 compute_pkt_fields(s, st, st->parser, &out_pkt);
1109 if ((s->iformat->flags & AVFMT_GENERIC_INDEX) &&
1110 out_pkt.flags & AV_PKT_FLAG_KEY) {
1111 ff_reduce_index(s, st->index);
1112 av_add_index_entry(st, st->parser->frame_offset, out_pkt.dts,
1113 0, 0, AVINDEX_KEYFRAME);
1116 if (out_pkt.data == pkt->data && out_pkt.size == pkt->size) {
1117 out_pkt.destruct = pkt->destruct;
1118 pkt->destruct = NULL;
1120 if ((ret = av_dup_packet(&out_pkt)) < 0)
1123 if (!add_to_pktbuf(&s->parse_queue, &out_pkt, &s->parse_queue_end)) {
1124 av_free_packet(&out_pkt);
1125 ret = AVERROR(ENOMEM);
1131 /* end of the stream => close and free the parser */
1132 if (pkt == &flush_pkt) {
1133 av_parser_close(st->parser);
1138 av_free_packet(pkt);
1142 static int read_from_packet_buffer(AVPacketList **pkt_buffer,
1143 AVPacketList **pkt_buffer_end,
1147 av_assert0(*pkt_buffer);
1150 *pkt_buffer = pktl->next;
1152 *pkt_buffer_end = NULL;
1157 static int read_frame_internal(AVFormatContext *s, AVPacket *pkt)
1159 int ret = 0, i, got_packet = 0;
1161 av_init_packet(pkt);
1163 while (!got_packet && !s->parse_queue) {
1167 /* read next packet */
1168 ret = ff_read_packet(s, &cur_pkt);
1170 if (ret == AVERROR(EAGAIN))
1172 /* flush the parsers */
1173 for(i = 0; i < s->nb_streams; i++) {
1175 if (st->parser && st->need_parsing)
1176 parse_packet(s, NULL, st->index);
1178 /* all remaining packets are now in parse_queue =>
1179 * really terminate parsing */
1183 st = s->streams[cur_pkt.stream_index];
1185 if (cur_pkt.pts != AV_NOPTS_VALUE &&
1186 cur_pkt.dts != AV_NOPTS_VALUE &&
1187 cur_pkt.pts < cur_pkt.dts) {
1188 av_log(s, AV_LOG_WARNING, "Invalid timestamps stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d\n",
1189 cur_pkt.stream_index,
1194 if (s->debug & FF_FDEBUG_TS)
1195 av_log(s, AV_LOG_DEBUG, "ff_read_packet stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d, duration=%d, flags=%d\n",
1196 cur_pkt.stream_index,
1203 if (st->need_parsing && !st->parser && !(s->flags & AVFMT_FLAG_NOPARSE)) {
1204 st->parser = av_parser_init(st->codec->codec_id);
1206 /* no parser available: just output the raw packets */
1207 st->need_parsing = AVSTREAM_PARSE_NONE;
1208 } else if(st->need_parsing == AVSTREAM_PARSE_HEADERS) {
1209 st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
1210 } else if(st->need_parsing == AVSTREAM_PARSE_FULL_ONCE) {
1211 st->parser->flags |= PARSER_FLAG_ONCE;
1215 if (!st->need_parsing || !st->parser) {
1216 /* no parsing needed: we just output the packet as is */
1218 compute_pkt_fields(s, st, NULL, pkt);
1219 if ((s->iformat->flags & AVFMT_GENERIC_INDEX) &&
1220 (pkt->flags & AV_PKT_FLAG_KEY) && pkt->dts != AV_NOPTS_VALUE) {
1221 ff_reduce_index(s, st->index);
1222 av_add_index_entry(st, pkt->pos, pkt->dts, 0, 0, AVINDEX_KEYFRAME);
1225 } else if (st->discard < AVDISCARD_ALL) {
1226 if ((ret = parse_packet(s, &cur_pkt, cur_pkt.stream_index)) < 0)
1230 av_free_packet(&cur_pkt);
1234 if (!got_packet && s->parse_queue)
1235 ret = read_from_packet_buffer(&s->parse_queue, &s->parse_queue_end, pkt);
1237 if(s->debug & FF_FDEBUG_TS)
1238 av_log(s, AV_LOG_DEBUG, "read_frame_internal stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d, duration=%d, flags=%d\n",
1249 int av_read_frame(AVFormatContext *s, AVPacket *pkt)
1251 const int genpts = s->flags & AVFMT_FLAG_GENPTS;
1255 return s->packet_buffer ? read_from_packet_buffer(&s->packet_buffer,
1256 &s->packet_buffer_end,
1258 read_frame_internal(s, pkt);
1262 AVPacketList *pktl = s->packet_buffer;
1265 AVPacket *next_pkt = &pktl->pkt;
1267 if (next_pkt->dts != AV_NOPTS_VALUE) {
1268 int wrap_bits = s->streams[next_pkt->stream_index]->pts_wrap_bits;
1269 while (pktl && next_pkt->pts == AV_NOPTS_VALUE) {
1270 if (pktl->pkt.stream_index == next_pkt->stream_index &&
1271 (av_compare_mod(next_pkt->dts, pktl->pkt.dts, 2LL << (wrap_bits - 1)) < 0) &&
1272 av_compare_mod(pktl->pkt.pts, pktl->pkt.dts, 2LL << (wrap_bits - 1))) { //not b frame
1273 next_pkt->pts = pktl->pkt.dts;
1277 pktl = s->packet_buffer;
1280 /* read packet from packet buffer, if there is data */
1281 if (!(next_pkt->pts == AV_NOPTS_VALUE &&
1282 next_pkt->dts != AV_NOPTS_VALUE && !eof))
1283 return read_from_packet_buffer(&s->packet_buffer,
1284 &s->packet_buffer_end, pkt);
1287 ret = read_frame_internal(s, pkt);
1289 if (pktl && ret != AVERROR(EAGAIN)) {
1296 if (av_dup_packet(add_to_pktbuf(&s->packet_buffer, pkt,
1297 &s->packet_buffer_end)) < 0)
1298 return AVERROR(ENOMEM);
1302 /* XXX: suppress the packet queue */
1303 static void flush_packet_queue(AVFormatContext *s)
1305 free_packet_buffer(&s->parse_queue, &s->parse_queue_end);
1306 free_packet_buffer(&s->packet_buffer, &s->packet_buffer_end);
1307 free_packet_buffer(&s->raw_packet_buffer, &s->raw_packet_buffer_end);
1309 s->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
1312 /*******************************************************/
1315 int av_find_default_stream_index(AVFormatContext *s)
1317 int first_audio_index = -1;
1321 if (s->nb_streams <= 0)
1323 for(i = 0; i < s->nb_streams; i++) {
1325 if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO &&
1326 !(st->disposition & AV_DISPOSITION_ATTACHED_PIC)) {
1329 if (first_audio_index < 0 && st->codec->codec_type == AVMEDIA_TYPE_AUDIO)
1330 first_audio_index = i;
1332 return first_audio_index >= 0 ? first_audio_index : 0;
1336 * Flush the frame reader.
1338 void ff_read_frame_flush(AVFormatContext *s)
1343 flush_packet_queue(s);
1345 /* for each stream, reset read state */
1346 for(i = 0; i < s->nb_streams; i++) {
1350 av_parser_close(st->parser);
1353 st->last_IP_pts = AV_NOPTS_VALUE;
1354 st->cur_dts = AV_NOPTS_VALUE; /* we set the current DTS to an unspecified origin */
1355 st->reference_dts = AV_NOPTS_VALUE;
1357 st->probe_packets = MAX_PROBE_PACKETS;
1359 for(j=0; j<MAX_REORDER_DELAY+1; j++)
1360 st->pts_buffer[j]= AV_NOPTS_VALUE;
1364 void ff_update_cur_dts(AVFormatContext *s, AVStream *ref_st, int64_t timestamp)
1368 for(i = 0; i < s->nb_streams; i++) {
1369 AVStream *st = s->streams[i];
1371 st->cur_dts = av_rescale(timestamp,
1372 st->time_base.den * (int64_t)ref_st->time_base.num,
1373 st->time_base.num * (int64_t)ref_st->time_base.den);
1377 void ff_reduce_index(AVFormatContext *s, int stream_index)
1379 AVStream *st= s->streams[stream_index];
1380 unsigned int max_entries= s->max_index_size / sizeof(AVIndexEntry);
1382 if((unsigned)st->nb_index_entries >= max_entries){
1384 for(i=0; 2*i<st->nb_index_entries; i++)
1385 st->index_entries[i]= st->index_entries[2*i];
1386 st->nb_index_entries= i;
1390 int ff_add_index_entry(AVIndexEntry **index_entries,
1391 int *nb_index_entries,
1392 unsigned int *index_entries_allocated_size,
1393 int64_t pos, int64_t timestamp, int size, int distance, int flags)
1395 AVIndexEntry *entries, *ie;
1398 if((unsigned)*nb_index_entries + 1 >= UINT_MAX / sizeof(AVIndexEntry))
1401 entries = av_fast_realloc(*index_entries,
1402 index_entries_allocated_size,
1403 (*nb_index_entries + 1) *
1404 sizeof(AVIndexEntry));
1408 *index_entries= entries;
1410 index= ff_index_search_timestamp(*index_entries, *nb_index_entries, timestamp, AVSEEK_FLAG_ANY);
1413 index= (*nb_index_entries)++;
1414 ie= &entries[index];
1415 assert(index==0 || ie[-1].timestamp < timestamp);
1417 ie= &entries[index];
1418 if(ie->timestamp != timestamp){
1419 if(ie->timestamp <= timestamp)
1421 memmove(entries + index + 1, entries + index, sizeof(AVIndexEntry)*(*nb_index_entries - index));
1422 (*nb_index_entries)++;
1423 }else if(ie->pos == pos && distance < ie->min_distance) //do not reduce the distance
1424 distance= ie->min_distance;
1428 ie->timestamp = timestamp;
1429 ie->min_distance= distance;
1436 int av_add_index_entry(AVStream *st,
1437 int64_t pos, int64_t timestamp, int size, int distance, int flags)
1439 return ff_add_index_entry(&st->index_entries, &st->nb_index_entries,
1440 &st->index_entries_allocated_size, pos,
1441 timestamp, size, distance, flags);
1444 int ff_index_search_timestamp(const AVIndexEntry *entries, int nb_entries,
1445 int64_t wanted_timestamp, int flags)
1453 //optimize appending index entries at the end
1454 if(b && entries[b-1].timestamp < wanted_timestamp)
1459 timestamp = entries[m].timestamp;
1460 if(timestamp >= wanted_timestamp)
1462 if(timestamp <= wanted_timestamp)
1465 m= (flags & AVSEEK_FLAG_BACKWARD) ? a : b;
1467 if(!(flags & AVSEEK_FLAG_ANY)){
1468 while(m>=0 && m<nb_entries && !(entries[m].flags & AVINDEX_KEYFRAME)){
1469 m += (flags & AVSEEK_FLAG_BACKWARD) ? -1 : 1;
1478 int av_index_search_timestamp(AVStream *st, int64_t wanted_timestamp,
1481 return ff_index_search_timestamp(st->index_entries, st->nb_index_entries,
1482 wanted_timestamp, flags);
1485 int ff_seek_frame_binary(AVFormatContext *s, int stream_index, int64_t target_ts, int flags)
1487 AVInputFormat *avif= s->iformat;
1488 int64_t av_uninit(pos_min), av_uninit(pos_max), pos, pos_limit;
1489 int64_t ts_min, ts_max, ts;
1494 if (stream_index < 0)
1497 av_dlog(s, "read_seek: %d %"PRId64"\n", stream_index, target_ts);
1500 ts_min= AV_NOPTS_VALUE;
1501 pos_limit= -1; //gcc falsely says it may be uninitialized
1503 st= s->streams[stream_index];
1504 if(st->index_entries){
1507 index= av_index_search_timestamp(st, target_ts, flags | AVSEEK_FLAG_BACKWARD); //FIXME whole func must be checked for non-keyframe entries in index case, especially read_timestamp()
1508 index= FFMAX(index, 0);
1509 e= &st->index_entries[index];
1511 if(e->timestamp <= target_ts || e->pos == e->min_distance){
1513 ts_min= e->timestamp;
1514 av_dlog(s, "using cached pos_min=0x%"PRIx64" dts_min=%"PRId64"\n",
1520 index= av_index_search_timestamp(st, target_ts, flags & ~AVSEEK_FLAG_BACKWARD);
1521 assert(index < st->nb_index_entries);
1523 e= &st->index_entries[index];
1524 assert(e->timestamp >= target_ts);
1526 ts_max= e->timestamp;
1527 pos_limit= pos_max - e->min_distance;
1528 av_dlog(s, "using cached pos_max=0x%"PRIx64" pos_limit=0x%"PRIx64" dts_max=%"PRId64"\n",
1529 pos_max,pos_limit, ts_max);
1533 pos= ff_gen_search(s, stream_index, target_ts, pos_min, pos_max, pos_limit, ts_min, ts_max, flags, &ts, avif->read_timestamp);
1538 if ((ret = avio_seek(s->pb, pos, SEEK_SET)) < 0)
1541 ff_update_cur_dts(s, st, ts);
1546 int64_t ff_gen_search(AVFormatContext *s, int stream_index, int64_t target_ts,
1547 int64_t pos_min, int64_t pos_max, int64_t pos_limit,
1548 int64_t ts_min, int64_t ts_max, int flags, int64_t *ts_ret,
1549 int64_t (*read_timestamp)(struct AVFormatContext *, int , int64_t *, int64_t ))
1552 int64_t start_pos, filesize;
1555 av_dlog(s, "gen_seek: %d %"PRId64"\n", stream_index, target_ts);
1557 if(ts_min == AV_NOPTS_VALUE){
1558 pos_min = s->data_offset;
1559 ts_min = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1560 if (ts_min == AV_NOPTS_VALUE)
1564 if(ts_max == AV_NOPTS_VALUE){
1566 filesize = avio_size(s->pb);
1567 pos_max = filesize - 1;
1570 ts_max = read_timestamp(s, stream_index, &pos_max, pos_max + step);
1572 }while(ts_max == AV_NOPTS_VALUE && pos_max >= step);
1573 if (ts_max == AV_NOPTS_VALUE)
1577 int64_t tmp_pos= pos_max + 1;
1578 int64_t tmp_ts= read_timestamp(s, stream_index, &tmp_pos, INT64_MAX);
1579 if(tmp_ts == AV_NOPTS_VALUE)
1583 if(tmp_pos >= filesize)
1589 if(ts_min > ts_max){
1591 }else if(ts_min == ts_max){
1596 while (pos_min < pos_limit) {
1597 av_dlog(s, "pos_min=0x%"PRIx64" pos_max=0x%"PRIx64" dts_min=%"PRId64" dts_max=%"PRId64"\n",
1598 pos_min, pos_max, ts_min, ts_max);
1599 assert(pos_limit <= pos_max);
1602 int64_t approximate_keyframe_distance= pos_max - pos_limit;
1603 // interpolate position (better than dichotomy)
1604 pos = av_rescale(target_ts - ts_min, pos_max - pos_min, ts_max - ts_min)
1605 + pos_min - approximate_keyframe_distance;
1606 }else if(no_change==1){
1607 // bisection, if interpolation failed to change min or max pos last time
1608 pos = (pos_min + pos_limit)>>1;
1610 /* linear search if bisection failed, can only happen if there
1611 are very few or no keyframes between min/max */
1616 else if(pos > pos_limit)
1620 ts = read_timestamp(s, stream_index, &pos, INT64_MAX); //may pass pos_limit instead of -1
1625 av_dlog(s, "%"PRId64" %"PRId64" %"PRId64" / %"PRId64" %"PRId64" %"PRId64" target:%"PRId64" limit:%"PRId64" start:%"PRId64" noc:%d\n",
1626 pos_min, pos, pos_max, ts_min, ts, ts_max, target_ts,
1627 pos_limit, start_pos, no_change);
1628 if(ts == AV_NOPTS_VALUE){
1629 av_log(s, AV_LOG_ERROR, "read_timestamp() failed in the middle\n");
1632 assert(ts != AV_NOPTS_VALUE);
1633 if (target_ts <= ts) {
1634 pos_limit = start_pos - 1;
1638 if (target_ts >= ts) {
1644 pos = (flags & AVSEEK_FLAG_BACKWARD) ? pos_min : pos_max;
1645 ts = (flags & AVSEEK_FLAG_BACKWARD) ? ts_min : ts_max;
1647 ts_min = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1649 ts_max = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1650 av_dlog(s, "pos=0x%"PRIx64" %"PRId64"<=%"PRId64"<=%"PRId64"\n",
1651 pos, ts_min, target_ts, ts_max);
1656 static int seek_frame_byte(AVFormatContext *s, int stream_index, int64_t pos, int flags){
1657 int64_t pos_min, pos_max;
1659 pos_min = s->data_offset;
1660 pos_max = avio_size(s->pb) - 1;
1662 if (pos < pos_min) pos= pos_min;
1663 else if(pos > pos_max) pos= pos_max;
1665 avio_seek(s->pb, pos, SEEK_SET);
1670 static int seek_frame_generic(AVFormatContext *s,
1671 int stream_index, int64_t timestamp, int flags)
1678 st = s->streams[stream_index];
1680 index = av_index_search_timestamp(st, timestamp, flags);
1682 if(index < 0 && st->nb_index_entries && timestamp < st->index_entries[0].timestamp)
1685 if(index < 0 || index==st->nb_index_entries-1){
1688 if(st->nb_index_entries){
1689 assert(st->index_entries);
1690 ie= &st->index_entries[st->nb_index_entries-1];
1691 if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
1693 ff_update_cur_dts(s, st, ie->timestamp);
1695 if ((ret = avio_seek(s->pb, s->data_offset, SEEK_SET)) < 0)
1701 read_status = av_read_frame(s, &pkt);
1702 } while (read_status == AVERROR(EAGAIN));
1703 if (read_status < 0)
1705 av_free_packet(&pkt);
1706 if(stream_index == pkt.stream_index){
1707 if((pkt.flags & AV_PKT_FLAG_KEY) && pkt.dts > timestamp)
1711 index = av_index_search_timestamp(st, timestamp, flags);
1716 ff_read_frame_flush(s);
1717 if (s->iformat->read_seek){
1718 if(s->iformat->read_seek(s, stream_index, timestamp, flags) >= 0)
1721 ie = &st->index_entries[index];
1722 if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
1724 ff_update_cur_dts(s, st, ie->timestamp);
1729 static int seek_frame_internal(AVFormatContext *s, int stream_index,
1730 int64_t timestamp, int flags)
1735 if (flags & AVSEEK_FLAG_BYTE) {
1736 if (s->iformat->flags & AVFMT_NO_BYTE_SEEK)
1738 ff_read_frame_flush(s);
1739 return seek_frame_byte(s, stream_index, timestamp, flags);
1742 if(stream_index < 0){
1743 stream_index= av_find_default_stream_index(s);
1744 if(stream_index < 0)
1747 st= s->streams[stream_index];
1748 /* timestamp for default must be expressed in AV_TIME_BASE units */
1749 timestamp = av_rescale(timestamp, st->time_base.den, AV_TIME_BASE * (int64_t)st->time_base.num);
1752 /* first, we try the format specific seek */
1753 if (s->iformat->read_seek) {
1754 ff_read_frame_flush(s);
1755 ret = s->iformat->read_seek(s, stream_index, timestamp, flags);
1762 if (s->iformat->read_timestamp && !(s->iformat->flags & AVFMT_NOBINSEARCH)) {
1763 ff_read_frame_flush(s);
1764 return ff_seek_frame_binary(s, stream_index, timestamp, flags);
1765 } else if (!(s->iformat->flags & AVFMT_NOGENSEARCH)) {
1766 ff_read_frame_flush(s);
1767 return seek_frame_generic(s, stream_index, timestamp, flags);
1773 int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
1775 int ret = seek_frame_internal(s, stream_index, timestamp, flags);
1778 queue_attached_pictures(s);
1783 int avformat_seek_file(AVFormatContext *s, int stream_index, int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
1785 if(min_ts > ts || max_ts < ts)
1788 if (s->iformat->read_seek2) {
1790 ff_read_frame_flush(s);
1791 ret = s->iformat->read_seek2(s, stream_index, min_ts, ts, max_ts, flags);
1794 queue_attached_pictures(s);
1798 if(s->iformat->read_timestamp){
1799 //try to seek via read_timestamp()
1802 //Fallback to old API if new is not implemented but old is
1803 //Note the old has somewat different sematics
1804 if(s->iformat->read_seek || 1)
1805 return av_seek_frame(s, stream_index, ts, flags | (ts - min_ts > (uint64_t)(max_ts - ts) ? AVSEEK_FLAG_BACKWARD : 0));
1807 // try some generic seek like seek_frame_generic() but with new ts semantics
1810 /*******************************************************/
1813 * Return TRUE if the stream has accurate duration in any stream.
1815 * @return TRUE if the stream has accurate duration for at least one component.
1817 static int has_duration(AVFormatContext *ic)
1822 for(i = 0;i < ic->nb_streams; i++) {
1823 st = ic->streams[i];
1824 if (st->duration != AV_NOPTS_VALUE)
1827 if (ic->duration != AV_NOPTS_VALUE)
1833 * Estimate the stream timings from the one of each components.
1835 * Also computes the global bitrate if possible.
1837 static void update_stream_timings(AVFormatContext *ic)
1839 int64_t start_time, start_time1, end_time, end_time1;
1840 int64_t duration, duration1, filesize;
1844 start_time = INT64_MAX;
1845 end_time = INT64_MIN;
1846 duration = INT64_MIN;
1847 for(i = 0;i < ic->nb_streams; i++) {
1848 st = ic->streams[i];
1849 if (st->start_time != AV_NOPTS_VALUE && st->time_base.den) {
1850 start_time1= av_rescale_q(st->start_time, st->time_base, AV_TIME_BASE_Q);
1851 start_time = FFMIN(start_time, start_time1);
1852 if (st->duration != AV_NOPTS_VALUE) {
1853 end_time1 = start_time1
1854 + av_rescale_q(st->duration, st->time_base, AV_TIME_BASE_Q);
1855 end_time = FFMAX(end_time, end_time1);
1858 if (st->duration != AV_NOPTS_VALUE) {
1859 duration1 = av_rescale_q(st->duration, st->time_base, AV_TIME_BASE_Q);
1860 duration = FFMAX(duration, duration1);
1863 if (start_time != INT64_MAX) {
1864 ic->start_time = start_time;
1865 if (end_time != INT64_MIN)
1866 duration = FFMAX(duration, end_time - start_time);
1868 if (duration != INT64_MIN) {
1869 ic->duration = duration;
1870 if (ic->pb && (filesize = avio_size(ic->pb)) > 0) {
1871 /* compute the bitrate */
1872 ic->bit_rate = (double)filesize * 8.0 * AV_TIME_BASE /
1873 (double)ic->duration;
1878 static void fill_all_stream_timings(AVFormatContext *ic)
1883 update_stream_timings(ic);
1884 for(i = 0;i < ic->nb_streams; i++) {
1885 st = ic->streams[i];
1886 if (st->start_time == AV_NOPTS_VALUE) {
1887 if(ic->start_time != AV_NOPTS_VALUE)
1888 st->start_time = av_rescale_q(ic->start_time, AV_TIME_BASE_Q, st->time_base);
1889 if(ic->duration != AV_NOPTS_VALUE)
1890 st->duration = av_rescale_q(ic->duration, AV_TIME_BASE_Q, st->time_base);
1895 static void estimate_timings_from_bit_rate(AVFormatContext *ic)
1897 int64_t filesize, duration;
1901 /* if bit_rate is already set, we believe it */
1902 if (ic->bit_rate <= 0) {
1904 for(i=0;i<ic->nb_streams;i++) {
1905 st = ic->streams[i];
1906 if (st->codec->bit_rate > 0)
1907 bit_rate += st->codec->bit_rate;
1909 ic->bit_rate = bit_rate;
1912 /* if duration is already set, we believe it */
1913 if (ic->duration == AV_NOPTS_VALUE &&
1914 ic->bit_rate != 0) {
1915 filesize = ic->pb ? avio_size(ic->pb) : 0;
1917 for(i = 0; i < ic->nb_streams; i++) {
1918 st = ic->streams[i];
1919 duration= av_rescale(8*filesize, st->time_base.den, ic->bit_rate*(int64_t)st->time_base.num);
1920 if (st->duration == AV_NOPTS_VALUE)
1921 st->duration = duration;
1927 #define DURATION_MAX_READ_SIZE 250000
1928 #define DURATION_MAX_RETRY 3
1930 /* only usable for MPEG-PS streams */
1931 static void estimate_timings_from_pts(AVFormatContext *ic, int64_t old_offset)
1933 AVPacket pkt1, *pkt = &pkt1;
1935 int read_size, i, ret;
1937 int64_t filesize, offset, duration;
1940 /* flush packet queue */
1941 flush_packet_queue(ic);
1943 for (i=0; i<ic->nb_streams; i++) {
1944 st = ic->streams[i];
1945 if (st->start_time == AV_NOPTS_VALUE && st->first_dts == AV_NOPTS_VALUE)
1946 av_log(st->codec, AV_LOG_WARNING, "start time is not set in estimate_timings_from_pts\n");
1949 av_parser_close(st->parser);
1954 /* estimate the end time (duration) */
1955 /* XXX: may need to support wrapping */
1956 filesize = ic->pb ? avio_size(ic->pb) : 0;
1957 end_time = AV_NOPTS_VALUE;
1959 offset = filesize - (DURATION_MAX_READ_SIZE<<retry);
1963 avio_seek(ic->pb, offset, SEEK_SET);
1966 if (read_size >= DURATION_MAX_READ_SIZE<<(FFMAX(retry-1,0)))
1970 ret = ff_read_packet(ic, pkt);
1971 } while(ret == AVERROR(EAGAIN));
1974 read_size += pkt->size;
1975 st = ic->streams[pkt->stream_index];
1976 if (pkt->pts != AV_NOPTS_VALUE &&
1977 (st->start_time != AV_NOPTS_VALUE ||
1978 st->first_dts != AV_NOPTS_VALUE)) {
1979 duration = end_time = pkt->pts;
1980 if (st->start_time != AV_NOPTS_VALUE)
1981 duration -= st->start_time;
1983 duration -= st->first_dts;
1985 duration += 1LL<<st->pts_wrap_bits;
1987 if (st->duration == AV_NOPTS_VALUE || st->duration < duration)
1988 st->duration = duration;
1991 av_free_packet(pkt);
1993 }while( end_time==AV_NOPTS_VALUE
1994 && filesize > (DURATION_MAX_READ_SIZE<<retry)
1995 && ++retry <= DURATION_MAX_RETRY);
1997 fill_all_stream_timings(ic);
1999 avio_seek(ic->pb, old_offset, SEEK_SET);
2000 for (i=0; i<ic->nb_streams; i++) {
2002 st->cur_dts= st->first_dts;
2003 st->last_IP_pts = AV_NOPTS_VALUE;
2004 st->reference_dts = AV_NOPTS_VALUE;
2008 static void estimate_timings(AVFormatContext *ic, int64_t old_offset)
2012 /* get the file size, if possible */
2013 if (ic->iformat->flags & AVFMT_NOFILE) {
2016 file_size = avio_size(ic->pb);
2017 file_size = FFMAX(0, file_size);
2020 if ((!strcmp(ic->iformat->name, "mpeg") ||
2021 !strcmp(ic->iformat->name, "mpegts")) &&
2022 file_size && ic->pb->seekable) {
2023 /* get accurate estimate from the PTSes */
2024 estimate_timings_from_pts(ic, old_offset);
2025 } else if (has_duration(ic)) {
2026 /* at least one component has timings - we use them for all
2028 fill_all_stream_timings(ic);
2030 av_log(ic, AV_LOG_WARNING, "Estimating duration from bitrate, this may be inaccurate\n");
2031 /* less precise: use bitrate info */
2032 estimate_timings_from_bit_rate(ic);
2034 update_stream_timings(ic);
2038 AVStream av_unused *st;
2039 for(i = 0;i < ic->nb_streams; i++) {
2040 st = ic->streams[i];
2041 av_dlog(ic, "%d: start_time: %0.3f duration: %0.3f\n", i,
2042 (double) st->start_time / AV_TIME_BASE,
2043 (double) st->duration / AV_TIME_BASE);
2045 av_dlog(ic, "stream: start_time: %0.3f duration: %0.3f bitrate=%d kb/s\n",
2046 (double) ic->start_time / AV_TIME_BASE,
2047 (double) ic->duration / AV_TIME_BASE,
2048 ic->bit_rate / 1000);
2052 static int has_codec_parameters(AVStream *st)
2054 AVCodecContext *avctx = st->codec;
2056 switch (avctx->codec_type) {
2057 case AVMEDIA_TYPE_AUDIO:
2058 val = avctx->sample_rate && avctx->channels;
2059 if (st->info->found_decoder >= 0 && avctx->sample_fmt == AV_SAMPLE_FMT_NONE)
2062 case AVMEDIA_TYPE_VIDEO:
2064 if (st->info->found_decoder >= 0 && avctx->pix_fmt == PIX_FMT_NONE)
2071 return avctx->codec_id != AV_CODEC_ID_NONE && val != 0;
2074 static int has_decode_delay_been_guessed(AVStream *st)
2076 return st->codec->codec_id != AV_CODEC_ID_H264 ||
2077 st->info->nb_decoded_frames >= 6;
2080 /* returns 1 or 0 if or if not decoded data was returned, or a negative error */
2081 static int try_decode_frame(AVStream *st, AVPacket *avpkt, AVDictionary **options)
2083 const AVCodec *codec;
2084 int got_picture = 1, ret = 0;
2086 AVPacket pkt = *avpkt;
2088 if (!avcodec_is_open(st->codec) && !st->info->found_decoder) {
2089 AVDictionary *thread_opt = NULL;
2091 codec = st->codec->codec ? st->codec->codec :
2092 avcodec_find_decoder(st->codec->codec_id);
2095 st->info->found_decoder = -1;
2099 /* force thread count to 1 since the h264 decoder will not extract SPS
2100 * and PPS to extradata during multi-threaded decoding */
2101 av_dict_set(options ? options : &thread_opt, "threads", "1", 0);
2102 ret = avcodec_open2(st->codec, codec, options ? options : &thread_opt);
2104 av_dict_free(&thread_opt);
2106 st->info->found_decoder = -1;
2109 st->info->found_decoder = 1;
2110 } else if (!st->info->found_decoder)
2111 st->info->found_decoder = 1;
2113 if (st->info->found_decoder < 0)
2116 while ((pkt.size > 0 || (!pkt.data && got_picture)) &&
2118 (!has_codec_parameters(st) ||
2119 !has_decode_delay_been_guessed(st) ||
2120 (!st->codec_info_nb_frames && st->codec->codec->capabilities & CODEC_CAP_CHANNEL_CONF))) {
2122 avcodec_get_frame_defaults(&picture);
2123 switch(st->codec->codec_type) {
2124 case AVMEDIA_TYPE_VIDEO:
2125 ret = avcodec_decode_video2(st->codec, &picture,
2126 &got_picture, &pkt);
2128 case AVMEDIA_TYPE_AUDIO:
2129 ret = avcodec_decode_audio4(st->codec, &picture, &got_picture, &pkt);
2136 st->info->nb_decoded_frames++;
2145 unsigned int ff_codec_get_tag(const AVCodecTag *tags, enum AVCodecID id)
2147 while (tags->id != AV_CODEC_ID_NONE) {
2155 enum AVCodecID ff_codec_get_id(const AVCodecTag *tags, unsigned int tag)
2158 for(i=0; tags[i].id != AV_CODEC_ID_NONE;i++) {
2159 if(tag == tags[i].tag)
2162 for(i=0; tags[i].id != AV_CODEC_ID_NONE; i++) {
2163 if (avpriv_toupper4(tag) == avpriv_toupper4(tags[i].tag))
2166 return AV_CODEC_ID_NONE;
2169 unsigned int av_codec_get_tag(const AVCodecTag * const *tags, enum AVCodecID id)
2172 for(i=0; tags && tags[i]; i++){
2173 int tag= ff_codec_get_tag(tags[i], id);
2179 enum AVCodecID av_codec_get_id(const AVCodecTag * const *tags, unsigned int tag)
2182 for(i=0; tags && tags[i]; i++){
2183 enum AVCodecID id= ff_codec_get_id(tags[i], tag);
2184 if(id!=AV_CODEC_ID_NONE) return id;
2186 return AV_CODEC_ID_NONE;
2189 static void compute_chapters_end(AVFormatContext *s)
2192 int64_t max_time = s->duration + ((s->start_time == AV_NOPTS_VALUE) ? 0 : s->start_time);
2194 for (i = 0; i < s->nb_chapters; i++)
2195 if (s->chapters[i]->end == AV_NOPTS_VALUE) {
2196 AVChapter *ch = s->chapters[i];
2197 int64_t end = max_time ? av_rescale_q(max_time, AV_TIME_BASE_Q, ch->time_base)
2200 for (j = 0; j < s->nb_chapters; j++) {
2201 AVChapter *ch1 = s->chapters[j];
2202 int64_t next_start = av_rescale_q(ch1->start, ch1->time_base, ch->time_base);
2203 if (j != i && next_start > ch->start && next_start < end)
2206 ch->end = (end == INT64_MAX) ? ch->start : end;
2210 static int get_std_framerate(int i){
2211 if(i<60*12) return i*1001;
2212 else return ((const int[]){24,30,60,12,15})[i-60*12]*1000*12;
2216 * Is the time base unreliable.
2217 * This is a heuristic to balance between quick acceptance of the values in
2218 * the headers vs. some extra checks.
2219 * Old DivX and Xvid often have nonsense timebases like 1fps or 2fps.
2220 * MPEG-2 commonly misuses field repeat flags to store different framerates.
2221 * And there are "variable" fps files this needs to detect as well.
2223 static int tb_unreliable(AVCodecContext *c){
2224 if( c->time_base.den >= 101L*c->time_base.num
2225 || c->time_base.den < 5L*c->time_base.num
2226 /* || c->codec_tag == AV_RL32("DIVX")
2227 || c->codec_tag == AV_RL32("XVID")*/
2228 || c->codec_id == AV_CODEC_ID_MPEG2VIDEO
2229 || c->codec_id == AV_CODEC_ID_H264
2235 int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
2237 int i, count, ret, read_size, j;
2239 AVPacket pkt1, *pkt;
2240 int64_t old_offset = avio_tell(ic->pb);
2241 int orig_nb_streams = ic->nb_streams; // new streams might appear, no options for those
2243 for(i=0;i<ic->nb_streams;i++) {
2244 const AVCodec *codec;
2245 AVDictionary *thread_opt = NULL;
2246 st = ic->streams[i];
2248 //only for the split stuff
2249 if (!st->parser && !(ic->flags & AVFMT_FLAG_NOPARSE)) {
2250 st->parser = av_parser_init(st->codec->codec_id);
2251 if(st->need_parsing == AVSTREAM_PARSE_HEADERS && st->parser){
2252 st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
2255 codec = st->codec->codec ? st->codec->codec :
2256 avcodec_find_decoder(st->codec->codec_id);
2258 /* force thread count to 1 since the h264 decoder will not extract SPS
2259 * and PPS to extradata during multi-threaded decoding */
2260 av_dict_set(options ? &options[i] : &thread_opt, "threads", "1", 0);
2262 /* Ensure that subtitle_header is properly set. */
2263 if (st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE
2264 && codec && !st->codec->codec)
2265 avcodec_open2(st->codec, codec, options ? &options[i]
2268 //try to just open decoders, in case this is enough to get parameters
2269 if (!has_codec_parameters(st)) {
2270 if (codec && !st->codec->codec)
2271 avcodec_open2(st->codec, codec, options ? &options[i]
2275 av_dict_free(&thread_opt);
2278 for (i=0; i<ic->nb_streams; i++) {
2279 #if FF_API_R_FRAME_RATE
2280 ic->streams[i]->info->last_dts = AV_NOPTS_VALUE;
2282 ic->streams[i]->info->fps_first_dts = AV_NOPTS_VALUE;
2283 ic->streams[i]->info->fps_last_dts = AV_NOPTS_VALUE;
2289 if (ff_check_interrupt(&ic->interrupt_callback)){
2291 av_log(ic, AV_LOG_DEBUG, "interrupted\n");
2295 /* check if one codec still needs to be handled */
2296 for(i=0;i<ic->nb_streams;i++) {
2297 int fps_analyze_framecount = 20;
2299 st = ic->streams[i];
2300 if (!has_codec_parameters(st))
2302 /* if the timebase is coarse (like the usual millisecond precision
2303 of mkv), we need to analyze more frames to reliably arrive at
2305 if (av_q2d(st->time_base) > 0.0005)
2306 fps_analyze_framecount *= 2;
2307 if (ic->fps_probe_size >= 0)
2308 fps_analyze_framecount = ic->fps_probe_size;
2309 /* variable fps and no guess at the real fps */
2310 if( tb_unreliable(st->codec) && !st->avg_frame_rate.num
2311 && st->codec_info_nb_frames < fps_analyze_framecount
2312 && st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
2314 if(st->parser && st->parser->parser->split && !st->codec->extradata)
2316 if (st->first_dts == AV_NOPTS_VALUE &&
2317 (st->codec->codec_type == AVMEDIA_TYPE_VIDEO ||
2318 st->codec->codec_type == AVMEDIA_TYPE_AUDIO))
2321 if (i == ic->nb_streams) {
2322 /* NOTE: if the format has no header, then we need to read
2323 some packets to get most of the streams, so we cannot
2325 if (!(ic->ctx_flags & AVFMTCTX_NOHEADER)) {
2326 /* if we found the info for all the codecs, we can stop */
2328 av_log(ic, AV_LOG_DEBUG, "All info found\n");
2332 /* we did not get all the codec info, but we read too much data */
2333 if (read_size >= ic->probesize) {
2335 av_log(ic, AV_LOG_DEBUG, "Probe buffer size limit %d reached\n", ic->probesize);
2339 /* NOTE: a new stream can be added there if no header in file
2340 (AVFMTCTX_NOHEADER) */
2341 ret = read_frame_internal(ic, &pkt1);
2342 if (ret == AVERROR(EAGAIN))
2347 AVPacket empty_pkt = { 0 };
2349 av_init_packet(&empty_pkt);
2351 ret = -1; /* we could not have all the codec parameters before EOF */
2352 for(i=0;i<ic->nb_streams;i++) {
2353 st = ic->streams[i];
2355 /* flush the decoders */
2356 if (st->info->found_decoder == 1) {
2358 err = try_decode_frame(st, &empty_pkt,
2359 (options && i < orig_nb_streams) ?
2360 &options[i] : NULL);
2361 } while (err > 0 && !has_codec_parameters(st));
2365 av_log(ic, AV_LOG_WARNING,
2366 "decoding for stream %d failed\n", st->index);
2367 } else if (!has_codec_parameters(st)) {
2369 avcodec_string(buf, sizeof(buf), st->codec, 0);
2370 av_log(ic, AV_LOG_WARNING,
2371 "Could not find codec parameters (%s)\n", buf);
2379 if (ic->flags & AVFMT_FLAG_NOBUFFER) {
2382 pkt = add_to_pktbuf(&ic->packet_buffer, &pkt1,
2383 &ic->packet_buffer_end);
2384 if ((ret = av_dup_packet(pkt)) < 0)
2385 goto find_stream_info_err;
2388 read_size += pkt->size;
2390 st = ic->streams[pkt->stream_index];
2391 if (pkt->dts != AV_NOPTS_VALUE && st->codec_info_nb_frames > 1) {
2392 /* check for non-increasing dts */
2393 if (st->info->fps_last_dts != AV_NOPTS_VALUE &&
2394 st->info->fps_last_dts >= pkt->dts) {
2395 av_log(ic, AV_LOG_WARNING, "Non-increasing DTS in stream %d: "
2396 "packet %d with DTS %"PRId64", packet %d with DTS "
2397 "%"PRId64"\n", st->index, st->info->fps_last_dts_idx,
2398 st->info->fps_last_dts, st->codec_info_nb_frames, pkt->dts);
2399 st->info->fps_first_dts = st->info->fps_last_dts = AV_NOPTS_VALUE;
2401 /* check for a discontinuity in dts - if the difference in dts
2402 * is more than 1000 times the average packet duration in the sequence,
2403 * we treat it as a discontinuity */
2404 if (st->info->fps_last_dts != AV_NOPTS_VALUE &&
2405 st->info->fps_last_dts_idx > st->info->fps_first_dts_idx &&
2406 (pkt->dts - st->info->fps_last_dts) / 1000 >
2407 (st->info->fps_last_dts - st->info->fps_first_dts) / (st->info->fps_last_dts_idx - st->info->fps_first_dts_idx)) {
2408 av_log(ic, AV_LOG_WARNING, "DTS discontinuity in stream %d: "
2409 "packet %d with DTS %"PRId64", packet %d with DTS "
2410 "%"PRId64"\n", st->index, st->info->fps_last_dts_idx,
2411 st->info->fps_last_dts, st->codec_info_nb_frames, pkt->dts);
2412 st->info->fps_first_dts = st->info->fps_last_dts = AV_NOPTS_VALUE;
2415 /* update stored dts values */
2416 if (st->info->fps_first_dts == AV_NOPTS_VALUE) {
2417 st->info->fps_first_dts = pkt->dts;
2418 st->info->fps_first_dts_idx = st->codec_info_nb_frames;
2420 st->info->fps_last_dts = pkt->dts;
2421 st->info->fps_last_dts_idx = st->codec_info_nb_frames;
2423 /* check max_analyze_duration */
2424 if (av_rescale_q(pkt->dts - st->info->fps_first_dts, st->time_base,
2425 AV_TIME_BASE_Q) >= ic->max_analyze_duration) {
2426 av_log(ic, AV_LOG_WARNING, "max_analyze_duration reached\n");
2430 #if FF_API_R_FRAME_RATE
2432 int64_t last = st->info->last_dts;
2434 if(pkt->dts != AV_NOPTS_VALUE && last != AV_NOPTS_VALUE && pkt->dts > last){
2435 int64_t duration= pkt->dts - last;
2436 double dur= duration * av_q2d(st->time_base);
2438 if (st->info->duration_count < 2)
2439 memset(st->info->duration_error, 0, sizeof(st->info->duration_error));
2440 for (i=1; i<FF_ARRAY_ELEMS(st->info->duration_error); i++) {
2441 int framerate= get_std_framerate(i);
2442 int ticks= lrintf(dur*framerate/(1001*12));
2443 double error = dur - (double)ticks*1001*12 / framerate;
2444 st->info->duration_error[i] += error*error;
2446 st->info->duration_count++;
2447 // ignore the first 4 values, they might have some random jitter
2448 if (st->info->duration_count > 3)
2449 st->info->duration_gcd = av_gcd(st->info->duration_gcd, duration);
2451 if (last == AV_NOPTS_VALUE || st->info->duration_count <= 1)
2452 st->info->last_dts = pkt->dts;
2455 if(st->parser && st->parser->parser->split && !st->codec->extradata){
2456 int i= st->parser->parser->split(st->codec, pkt->data, pkt->size);
2457 if (i > 0 && i < FF_MAX_EXTRADATA_SIZE) {
2458 st->codec->extradata_size= i;
2459 st->codec->extradata= av_malloc(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
2460 if (!st->codec->extradata)
2461 return AVERROR(ENOMEM);
2462 memcpy(st->codec->extradata, pkt->data, st->codec->extradata_size);
2463 memset(st->codec->extradata + i, 0, FF_INPUT_BUFFER_PADDING_SIZE);
2467 /* if still no information, we try to open the codec and to
2468 decompress the frame. We try to avoid that in most cases as
2469 it takes longer and uses more memory. For MPEG-4, we need to
2470 decompress for QuickTime.
2472 If CODEC_CAP_CHANNEL_CONF is set this will force decoding of at
2473 least one frame of codec data, this makes sure the codec initializes
2474 the channel configuration and does not only trust the values from the container.
2476 try_decode_frame(st, pkt, (options && i < orig_nb_streams ) ? &options[i] : NULL);
2478 st->codec_info_nb_frames++;
2482 // close codecs which were opened in try_decode_frame()
2483 for(i=0;i<ic->nb_streams;i++) {
2484 st = ic->streams[i];
2485 avcodec_close(st->codec);
2487 for(i=0;i<ic->nb_streams;i++) {
2488 st = ic->streams[i];
2489 if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
2490 /* estimate average framerate if not set by demuxer */
2491 if (!st->avg_frame_rate.num && st->info->fps_last_dts != st->info->fps_first_dts) {
2492 int64_t delta_dts = st->info->fps_last_dts - st->info->fps_first_dts;
2493 int delta_packets = st->info->fps_last_dts_idx - st->info->fps_first_dts_idx;
2495 double best_error = 0.01;
2497 av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
2498 delta_packets*(int64_t)st->time_base.den,
2499 delta_dts*(int64_t)st->time_base.num, 60000);
2501 /* round guessed framerate to a "standard" framerate if it's
2502 * within 1% of the original estimate*/
2503 for (j = 1; j < MAX_STD_TIMEBASES; j++) {
2504 AVRational std_fps = { get_std_framerate(j), 12*1001 };
2505 double error = fabs(av_q2d(st->avg_frame_rate) / av_q2d(std_fps) - 1);
2507 if (error < best_error) {
2509 best_fps = std_fps.num;
2513 av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
2514 best_fps, 12*1001, INT_MAX);
2517 #if FF_API_R_FRAME_RATE
2518 // the check for tb_unreliable() is not completely correct, since this is not about handling
2519 // a unreliable/inexact time base, but a time base that is finer than necessary, as e.g.
2520 // ipmovie.c produces.
2521 if (tb_unreliable(st->codec) && st->info->duration_count > 15 && st->info->duration_gcd > 1 && !st->r_frame_rate.num)
2522 av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den, st->time_base.den, st->time_base.num * st->info->duration_gcd, INT_MAX);
2523 if (st->info->duration_count && !st->r_frame_rate.num
2524 && tb_unreliable(st->codec)) {
2526 double best_error= 2*av_q2d(st->time_base);
2527 best_error = best_error*best_error*st->info->duration_count*1000*12*30;
2529 for (j=1; j<FF_ARRAY_ELEMS(st->info->duration_error); j++) {
2530 double error = st->info->duration_error[j] * get_std_framerate(j);
2531 if(error < best_error){
2533 num = get_std_framerate(j);
2536 // do not increase frame rate by more than 1 % in order to match a standard rate.
2537 if (num && (!st->r_frame_rate.num || (double)num/(12*1001) < 1.01 * av_q2d(st->r_frame_rate)))
2538 av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den, num, 12*1001, INT_MAX);
2541 }else if(st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
2542 if(!st->codec->bits_per_coded_sample)
2543 st->codec->bits_per_coded_sample= av_get_bits_per_sample(st->codec->codec_id);
2544 // set stream disposition based on audio service type
2545 switch (st->codec->audio_service_type) {
2546 case AV_AUDIO_SERVICE_TYPE_EFFECTS:
2547 st->disposition = AV_DISPOSITION_CLEAN_EFFECTS; break;
2548 case AV_AUDIO_SERVICE_TYPE_VISUALLY_IMPAIRED:
2549 st->disposition = AV_DISPOSITION_VISUAL_IMPAIRED; break;
2550 case AV_AUDIO_SERVICE_TYPE_HEARING_IMPAIRED:
2551 st->disposition = AV_DISPOSITION_HEARING_IMPAIRED; break;
2552 case AV_AUDIO_SERVICE_TYPE_COMMENTARY:
2553 st->disposition = AV_DISPOSITION_COMMENT; break;
2554 case AV_AUDIO_SERVICE_TYPE_KARAOKE:
2555 st->disposition = AV_DISPOSITION_KARAOKE; break;
2560 estimate_timings(ic, old_offset);
2562 compute_chapters_end(ic);
2564 find_stream_info_err:
2565 for (i=0; i < ic->nb_streams; i++) {
2566 if (ic->streams[i]->codec)
2567 ic->streams[i]->codec->thread_count = 0;
2568 av_freep(&ic->streams[i]->info);
2573 static AVProgram *find_program_from_stream(AVFormatContext *ic, int s)
2577 for (i = 0; i < ic->nb_programs; i++)
2578 for (j = 0; j < ic->programs[i]->nb_stream_indexes; j++)
2579 if (ic->programs[i]->stream_index[j] == s)
2580 return ic->programs[i];
2584 int av_find_best_stream(AVFormatContext *ic,
2585 enum AVMediaType type,
2586 int wanted_stream_nb,
2588 AVCodec **decoder_ret,
2591 int i, nb_streams = ic->nb_streams;
2592 int ret = AVERROR_STREAM_NOT_FOUND, best_count = -1;
2593 unsigned *program = NULL;
2594 AVCodec *decoder = NULL, *best_decoder = NULL;
2596 if (related_stream >= 0 && wanted_stream_nb < 0) {
2597 AVProgram *p = find_program_from_stream(ic, related_stream);
2599 program = p->stream_index;
2600 nb_streams = p->nb_stream_indexes;
2603 for (i = 0; i < nb_streams; i++) {
2604 int real_stream_index = program ? program[i] : i;
2605 AVStream *st = ic->streams[real_stream_index];
2606 AVCodecContext *avctx = st->codec;
2607 if (avctx->codec_type != type)
2609 if (wanted_stream_nb >= 0 && real_stream_index != wanted_stream_nb)
2611 if (st->disposition & (AV_DISPOSITION_HEARING_IMPAIRED|AV_DISPOSITION_VISUAL_IMPAIRED))
2614 decoder = avcodec_find_decoder(st->codec->codec_id);
2617 ret = AVERROR_DECODER_NOT_FOUND;
2621 if (best_count >= st->codec_info_nb_frames)
2623 best_count = st->codec_info_nb_frames;
2624 ret = real_stream_index;
2625 best_decoder = decoder;
2626 if (program && i == nb_streams - 1 && ret < 0) {
2628 nb_streams = ic->nb_streams;
2629 i = 0; /* no related stream found, try again with everything */
2633 *decoder_ret = best_decoder;
2637 /*******************************************************/
2639 int av_read_play(AVFormatContext *s)
2641 if (s->iformat->read_play)
2642 return s->iformat->read_play(s);
2644 return avio_pause(s->pb, 0);
2645 return AVERROR(ENOSYS);
2648 int av_read_pause(AVFormatContext *s)
2650 if (s->iformat->read_pause)
2651 return s->iformat->read_pause(s);
2653 return avio_pause(s->pb, 1);
2654 return AVERROR(ENOSYS);
2657 void avformat_free_context(AVFormatContext *s)
2663 if (s->iformat && s->iformat->priv_class && s->priv_data)
2664 av_opt_free(s->priv_data);
2666 for(i=0;i<s->nb_streams;i++) {
2667 /* free all data in a stream component */
2670 av_parser_close(st->parser);
2672 if (st->attached_pic.data)
2673 av_free_packet(&st->attached_pic);
2674 av_dict_free(&st->metadata);
2675 av_free(st->index_entries);
2676 av_free(st->codec->extradata);
2677 av_free(st->codec->subtitle_header);
2679 av_free(st->priv_data);
2683 for(i=s->nb_programs-1; i>=0; i--) {
2684 av_dict_free(&s->programs[i]->metadata);
2685 av_freep(&s->programs[i]->stream_index);
2686 av_freep(&s->programs[i]);
2688 av_freep(&s->programs);
2689 av_freep(&s->priv_data);
2690 while(s->nb_chapters--) {
2691 av_dict_free(&s->chapters[s->nb_chapters]->metadata);
2692 av_free(s->chapters[s->nb_chapters]);
2694 av_freep(&s->chapters);
2695 av_dict_free(&s->metadata);
2696 av_freep(&s->streams);
2700 #if FF_API_CLOSE_INPUT_FILE
2701 void av_close_input_file(AVFormatContext *s)
2703 avformat_close_input(&s);
2707 void avformat_close_input(AVFormatContext **ps)
2709 AVFormatContext *s = *ps;
2710 AVIOContext *pb = s->pb;
2712 if ((s->iformat && s->iformat->flags & AVFMT_NOFILE) ||
2713 (s->flags & AVFMT_FLAG_CUSTOM_IO))
2716 flush_packet_queue(s);
2719 if (s->iformat->read_close)
2720 s->iformat->read_close(s);
2723 avformat_free_context(s);
2730 AVStream *avformat_new_stream(AVFormatContext *s, AVCodec *c)
2736 if (s->nb_streams >= INT_MAX/sizeof(*streams))
2738 streams = av_realloc(s->streams, (s->nb_streams + 1) * sizeof(*streams));
2741 s->streams = streams;
2743 st = av_mallocz(sizeof(AVStream));
2746 if (!(st->info = av_mallocz(sizeof(*st->info)))) {
2751 st->codec = avcodec_alloc_context3(c);
2753 /* no default bitrate if decoding */
2754 st->codec->bit_rate = 0;
2756 st->index = s->nb_streams;
2757 st->start_time = AV_NOPTS_VALUE;
2758 st->duration = AV_NOPTS_VALUE;
2759 /* we set the current DTS to 0 so that formats without any timestamps
2760 but durations get some timestamps, formats with some unknown
2761 timestamps have their first few packets buffered and the
2762 timestamps corrected before they are returned to the user */
2764 st->first_dts = AV_NOPTS_VALUE;
2765 st->probe_packets = MAX_PROBE_PACKETS;
2767 /* default pts setting is MPEG-like */
2768 avpriv_set_pts_info(st, 33, 1, 90000);
2769 st->last_IP_pts = AV_NOPTS_VALUE;
2770 for(i=0; i<MAX_REORDER_DELAY+1; i++)
2771 st->pts_buffer[i]= AV_NOPTS_VALUE;
2772 st->reference_dts = AV_NOPTS_VALUE;
2774 st->sample_aspect_ratio = (AVRational){0,1};
2776 #if FF_API_R_FRAME_RATE
2777 st->info->last_dts = AV_NOPTS_VALUE;
2779 st->info->fps_first_dts = AV_NOPTS_VALUE;
2780 st->info->fps_last_dts = AV_NOPTS_VALUE;
2782 s->streams[s->nb_streams++] = st;
2786 AVProgram *av_new_program(AVFormatContext *ac, int id)
2788 AVProgram *program=NULL;
2791 av_dlog(ac, "new_program: id=0x%04x\n", id);
2793 for(i=0; i<ac->nb_programs; i++)
2794 if(ac->programs[i]->id == id)
2795 program = ac->programs[i];
2798 program = av_mallocz(sizeof(AVProgram));
2801 dynarray_add(&ac->programs, &ac->nb_programs, program);
2802 program->discard = AVDISCARD_NONE;
2809 AVChapter *avpriv_new_chapter(AVFormatContext *s, int id, AVRational time_base, int64_t start, int64_t end, const char *title)
2811 AVChapter *chapter = NULL;
2814 for(i=0; i<s->nb_chapters; i++)
2815 if(s->chapters[i]->id == id)
2816 chapter = s->chapters[i];
2819 chapter= av_mallocz(sizeof(AVChapter));
2822 dynarray_add(&s->chapters, &s->nb_chapters, chapter);
2824 av_dict_set(&chapter->metadata, "title", title, 0);
2826 chapter->time_base= time_base;
2827 chapter->start = start;
2833 /************************************************************/
2834 /* output media file */
2836 static int validate_codec_tag(AVFormatContext *s, AVStream *st)
2838 const AVCodecTag *avctag;
2840 enum AVCodecID id = AV_CODEC_ID_NONE;
2841 unsigned int tag = 0;
2844 * Check that tag + id is in the table
2845 * If neither is in the table -> OK
2846 * If tag is in the table with another id -> FAIL
2847 * If id is in the table with another tag -> FAIL unless strict < normal
2849 for (n = 0; s->oformat->codec_tag[n]; n++) {
2850 avctag = s->oformat->codec_tag[n];
2851 while (avctag->id != AV_CODEC_ID_NONE) {
2852 if (avpriv_toupper4(avctag->tag) == avpriv_toupper4(st->codec->codec_tag)) {
2854 if (id == st->codec->codec_id)
2857 if (avctag->id == st->codec->codec_id)
2862 if (id != AV_CODEC_ID_NONE)
2864 if (tag && (st->codec->strict_std_compliance >= FF_COMPLIANCE_NORMAL))
2869 int avformat_write_header(AVFormatContext *s, AVDictionary **options)
2873 AVDictionary *tmp = NULL;
2876 av_dict_copy(&tmp, *options, 0);
2877 if ((ret = av_opt_set_dict(s, &tmp)) < 0)
2880 // some sanity checks
2881 if (s->nb_streams == 0 && !(s->oformat->flags & AVFMT_NOSTREAMS)) {
2882 av_log(s, AV_LOG_ERROR, "no streams\n");
2883 ret = AVERROR(EINVAL);
2887 for(i=0;i<s->nb_streams;i++) {
2890 switch (st->codec->codec_type) {
2891 case AVMEDIA_TYPE_AUDIO:
2892 if(st->codec->sample_rate<=0){
2893 av_log(s, AV_LOG_ERROR, "sample rate not set\n");
2894 ret = AVERROR(EINVAL);
2897 if(!st->codec->block_align)
2898 st->codec->block_align = st->codec->channels *
2899 av_get_bits_per_sample(st->codec->codec_id) >> 3;
2901 case AVMEDIA_TYPE_VIDEO:
2902 if(st->codec->time_base.num<=0 || st->codec->time_base.den<=0){ //FIXME audio too?
2903 av_log(s, AV_LOG_ERROR, "time base not set\n");
2904 ret = AVERROR(EINVAL);
2907 if((st->codec->width<=0 || st->codec->height<=0) && !(s->oformat->flags & AVFMT_NODIMENSIONS)){
2908 av_log(s, AV_LOG_ERROR, "dimensions not set\n");
2909 ret = AVERROR(EINVAL);
2912 if(av_cmp_q(st->sample_aspect_ratio, st->codec->sample_aspect_ratio)){
2913 av_log(s, AV_LOG_ERROR, "Aspect ratio mismatch between muxer "
2914 "(%d/%d) and encoder layer (%d/%d)\n",
2915 st->sample_aspect_ratio.num, st->sample_aspect_ratio.den,
2916 st->codec->sample_aspect_ratio.num,
2917 st->codec->sample_aspect_ratio.den);
2918 ret = AVERROR(EINVAL);
2924 if(s->oformat->codec_tag){
2925 if(st->codec->codec_tag && st->codec->codec_id == AV_CODEC_ID_RAWVIDEO && av_codec_get_tag(s->oformat->codec_tag, st->codec->codec_id) == 0 && !validate_codec_tag(s, st)){
2926 //the current rawvideo encoding system ends up setting the wrong codec_tag for avi, we override it here
2927 st->codec->codec_tag= 0;
2929 if(st->codec->codec_tag){
2930 if (!validate_codec_tag(s, st)) {
2932 av_get_codec_tag_string(tagbuf, sizeof(tagbuf), st->codec->codec_tag);
2933 av_log(s, AV_LOG_ERROR,
2934 "Tag %s/0x%08x incompatible with output codec id '%d'\n",
2935 tagbuf, st->codec->codec_tag, st->codec->codec_id);
2936 ret = AVERROR_INVALIDDATA;
2940 st->codec->codec_tag= av_codec_get_tag(s->oformat->codec_tag, st->codec->codec_id);
2943 if(s->oformat->flags & AVFMT_GLOBALHEADER &&
2944 !(st->codec->flags & CODEC_FLAG_GLOBAL_HEADER))
2945 av_log(s, AV_LOG_WARNING, "Codec for stream %d does not use global headers but container format requires global headers\n", i);
2948 if (!s->priv_data && s->oformat->priv_data_size > 0) {
2949 s->priv_data = av_mallocz(s->oformat->priv_data_size);
2950 if (!s->priv_data) {
2951 ret = AVERROR(ENOMEM);
2954 if (s->oformat->priv_class) {
2955 *(const AVClass**)s->priv_data= s->oformat->priv_class;
2956 av_opt_set_defaults(s->priv_data);
2957 if ((ret = av_opt_set_dict(s->priv_data, &tmp)) < 0)
2962 /* set muxer identification string */
2963 if (s->nb_streams && !(s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT)) {
2964 av_dict_set(&s->metadata, "encoder", LIBAVFORMAT_IDENT, 0);
2967 if(s->oformat->write_header){
2968 ret = s->oformat->write_header(s);
2973 /* init PTS generation */
2974 for(i=0;i<s->nb_streams;i++) {
2975 int64_t den = AV_NOPTS_VALUE;
2978 switch (st->codec->codec_type) {
2979 case AVMEDIA_TYPE_AUDIO:
2980 den = (int64_t)st->time_base.num * st->codec->sample_rate;
2982 case AVMEDIA_TYPE_VIDEO:
2983 den = (int64_t)st->time_base.num * st->codec->time_base.den;
2988 if (den != AV_NOPTS_VALUE) {
2990 ret = AVERROR_INVALIDDATA;
2993 frac_init(&st->pts, 0, 0, den);
2998 av_dict_free(options);
3007 //FIXME merge with compute_pkt_fields
3008 static int compute_pkt_fields2(AVFormatContext *s, AVStream *st, AVPacket *pkt){
3009 int delay = FFMAX(st->codec->has_b_frames, !!st->codec->max_b_frames);
3010 int num, den, frame_size, i;
3012 av_dlog(s, "compute_pkt_fields2: pts:%"PRId64" dts:%"PRId64" cur_dts:%"PRId64" b:%d size:%d st:%d\n",
3013 pkt->pts, pkt->dts, st->cur_dts, delay, pkt->size, pkt->stream_index);
3015 /* if(pkt->pts == AV_NOPTS_VALUE && pkt->dts == AV_NOPTS_VALUE)
3016 return AVERROR(EINVAL);*/
3018 /* duration field */
3019 if (pkt->duration == 0) {
3020 compute_frame_duration(&num, &den, st, NULL, pkt);
3022 pkt->duration = av_rescale(1, num * (int64_t)st->time_base.den * st->codec->ticks_per_frame, den * (int64_t)st->time_base.num);
3026 if(pkt->pts == AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE && delay==0)
3029 //XXX/FIXME this is a temporary hack until all encoders output pts
3030 if((pkt->pts == 0 || pkt->pts == AV_NOPTS_VALUE) && pkt->dts == AV_NOPTS_VALUE && !delay){
3032 // pkt->pts= st->cur_dts;
3033 pkt->pts= st->pts.val;
3036 //calculate dts from pts
3037 if(pkt->pts != AV_NOPTS_VALUE && pkt->dts == AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY){
3038 st->pts_buffer[0]= pkt->pts;
3039 for(i=1; i<delay+1 && st->pts_buffer[i] == AV_NOPTS_VALUE; i++)
3040 st->pts_buffer[i]= pkt->pts + (i-delay-1) * pkt->duration;
3041 for(i=0; i<delay && st->pts_buffer[i] > st->pts_buffer[i+1]; i++)
3042 FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i+1]);
3044 pkt->dts= st->pts_buffer[0];
3047 if (st->cur_dts && st->cur_dts != AV_NOPTS_VALUE &&
3048 ((!(s->oformat->flags & AVFMT_TS_NONSTRICT) &&
3049 st->cur_dts >= pkt->dts) || st->cur_dts > pkt->dts)) {
3050 av_log(s, AV_LOG_ERROR,
3051 "Application provided invalid, non monotonically increasing dts to muxer in stream %d: %"PRId64" >= %"PRId64"\n",
3052 st->index, st->cur_dts, pkt->dts);
3053 return AVERROR(EINVAL);
3055 if(pkt->dts != AV_NOPTS_VALUE && pkt->pts != AV_NOPTS_VALUE && pkt->pts < pkt->dts){
3056 av_log(s, AV_LOG_ERROR, "pts < dts in stream %d\n", st->index);
3057 return AVERROR(EINVAL);
3060 // av_log(s, AV_LOG_DEBUG, "av_write_frame: pts2:%"PRId64" dts2:%"PRId64"\n", pkt->pts, pkt->dts);
3061 st->cur_dts= pkt->dts;
3062 st->pts.val= pkt->dts;
3065 switch (st->codec->codec_type) {
3066 case AVMEDIA_TYPE_AUDIO:
3067 frame_size = get_audio_frame_size(st->codec, pkt->size, 1);
3069 /* HACK/FIXME, we skip the initial 0 size packets as they are most
3070 likely equal to the encoder delay, but it would be better if we
3071 had the real timestamps from the encoder */
3072 if (frame_size >= 0 && (pkt->size || st->pts.num!=st->pts.den>>1 || st->pts.val)) {
3073 frac_add(&st->pts, (int64_t)st->time_base.den * frame_size);
3076 case AVMEDIA_TYPE_VIDEO:
3077 frac_add(&st->pts, (int64_t)st->time_base.den * st->codec->time_base.num);
3085 int av_write_frame(AVFormatContext *s, AVPacket *pkt)
3090 if (s->oformat->flags & AVFMT_ALLOW_FLUSH)
3091 return s->oformat->write_packet(s, pkt);
3095 ret = compute_pkt_fields2(s, s->streams[pkt->stream_index], pkt);
3097 if(ret<0 && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
3100 ret= s->oformat->write_packet(s, pkt);
3103 s->streams[pkt->stream_index]->nb_frames++;
3107 void ff_interleave_add_packet(AVFormatContext *s, AVPacket *pkt,
3108 int (*compare)(AVFormatContext *, AVPacket *, AVPacket *))
3110 AVPacketList **next_point, *this_pktl;
3112 this_pktl = av_mallocz(sizeof(AVPacketList));
3113 this_pktl->pkt= *pkt;
3114 pkt->destruct= NULL; // do not free original but only the copy
3115 av_dup_packet(&this_pktl->pkt); // duplicate the packet if it uses non-alloced memory
3117 if(s->streams[pkt->stream_index]->last_in_packet_buffer){
3118 next_point = &(s->streams[pkt->stream_index]->last_in_packet_buffer->next);
3120 next_point = &s->packet_buffer;
3123 if(compare(s, &s->packet_buffer_end->pkt, pkt)){
3124 while(!compare(s, &(*next_point)->pkt, pkt)){
3125 next_point= &(*next_point)->next;
3129 next_point = &(s->packet_buffer_end->next);
3132 assert(!*next_point);
3134 s->packet_buffer_end= this_pktl;
3137 this_pktl->next= *next_point;
3139 s->streams[pkt->stream_index]->last_in_packet_buffer=
3140 *next_point= this_pktl;
3143 static int ff_interleave_compare_dts(AVFormatContext *s, AVPacket *next, AVPacket *pkt)
3145 AVStream *st = s->streams[ pkt ->stream_index];
3146 AVStream *st2= s->streams[ next->stream_index];
3147 int comp = av_compare_ts(next->dts, st2->time_base, pkt->dts,
3151 return pkt->stream_index < next->stream_index;
3155 int ff_interleave_packet_per_dts(AVFormatContext *s, AVPacket *out,
3156 AVPacket *pkt, int flush)
3163 ff_interleave_add_packet(s, pkt, ff_interleave_compare_dts);
3166 for(i=0; i < s->nb_streams; i++)
3167 stream_count+= !!s->streams[i]->last_in_packet_buffer;
3169 if(stream_count && (s->nb_streams == stream_count || flush)){
3170 pktl= s->packet_buffer;
3173 s->packet_buffer= pktl->next;
3174 if(!s->packet_buffer)
3175 s->packet_buffer_end= NULL;
3177 if(s->streams[out->stream_index]->last_in_packet_buffer == pktl)
3178 s->streams[out->stream_index]->last_in_packet_buffer= NULL;
3182 av_init_packet(out);
3187 #if FF_API_INTERLEAVE_PACKET
3188 int av_interleave_packet_per_dts(AVFormatContext *s, AVPacket *out,
3189 AVPacket *pkt, int flush)
3191 return ff_interleave_packet_per_dts(s, out, pkt, flush);
3196 * Interleave an AVPacket correctly so it can be muxed.
3197 * @param out the interleaved packet will be output here
3198 * @param in the input packet
3199 * @param flush 1 if no further packets are available as input and all
3200 * remaining packets should be output
3201 * @return 1 if a packet was output, 0 if no packet could be output,
3202 * < 0 if an error occurred
3204 static int interleave_packet(AVFormatContext *s, AVPacket *out, AVPacket *in, int flush){
3205 if (s->oformat->interleave_packet) {
3206 int ret = s->oformat->interleave_packet(s, out, in, flush);
3211 return ff_interleave_packet_per_dts(s, out, in, flush);
3214 int av_interleaved_write_frame(AVFormatContext *s, AVPacket *pkt){
3218 AVStream *st= s->streams[ pkt->stream_index];
3220 //FIXME/XXX/HACK drop zero sized packets
3221 if(st->codec->codec_type == AVMEDIA_TYPE_AUDIO && pkt->size==0)
3224 av_dlog(s, "av_interleaved_write_frame size:%d dts:%"PRId64" pts:%"PRId64"\n",
3225 pkt->size, pkt->dts, pkt->pts);
3226 if((ret = compute_pkt_fields2(s, st, pkt)) < 0 && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
3229 if(pkt->dts == AV_NOPTS_VALUE && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
3230 return AVERROR(EINVAL);
3232 av_dlog(s, "av_interleaved_write_frame FLUSH\n");
3238 int ret= interleave_packet(s, &opkt, pkt, flush);
3239 if(ret<=0) //FIXME cleanup needed for ret<0 ?
3242 ret= s->oformat->write_packet(s, &opkt);
3244 s->streams[opkt.stream_index]->nb_frames++;
3246 av_free_packet(&opkt);
3254 int av_write_trailer(AVFormatContext *s)
3260 ret = interleave_packet(s, &pkt, NULL, 1);
3261 if (ret < 0) //FIXME cleanup needed for ret<0 ?
3266 ret = s->oformat->write_packet(s, &pkt);
3268 s->streams[pkt.stream_index]->nb_frames++;
3270 av_free_packet(&pkt);
3276 if (s->oformat->write_trailer)
3277 ret = s->oformat->write_trailer(s);
3279 if (!(s->oformat->flags & AVFMT_NOFILE))
3283 for (i = 0; i < s->nb_streams; i++) {
3284 av_freep(&s->streams[i]->priv_data);
3285 av_freep(&s->streams[i]->index_entries);
3287 if (s->oformat->priv_class)
3288 av_opt_free(s->priv_data);
3289 av_freep(&s->priv_data);
3293 void ff_program_add_stream_index(AVFormatContext *ac, int progid, unsigned int idx)
3296 AVProgram *program=NULL;
3299 if (idx >= ac->nb_streams) {
3300 av_log(ac, AV_LOG_ERROR, "stream index %d is not valid\n", idx);
3304 for(i=0; i<ac->nb_programs; i++){
3305 if(ac->programs[i]->id != progid)
3307 program = ac->programs[i];
3308 for(j=0; j<program->nb_stream_indexes; j++)
3309 if(program->stream_index[j] == idx)
3312 tmp = av_realloc(program->stream_index, sizeof(unsigned int)*(program->nb_stream_indexes+1));
3315 program->stream_index = tmp;
3316 program->stream_index[program->nb_stream_indexes++] = idx;
3321 static void print_fps(double d, const char *postfix){
3322 uint64_t v= lrintf(d*100);
3323 if (v% 100 ) av_log(NULL, AV_LOG_INFO, ", %3.2f %s", d, postfix);
3324 else if(v%(100*1000)) av_log(NULL, AV_LOG_INFO, ", %1.0f %s", d, postfix);
3325 else av_log(NULL, AV_LOG_INFO, ", %1.0fk %s", d/1000, postfix);
3328 static void dump_metadata(void *ctx, AVDictionary *m, const char *indent)
3330 if(m && !(av_dict_count(m) == 1 && av_dict_get(m, "language", NULL, 0))){
3331 AVDictionaryEntry *tag=NULL;
3333 av_log(ctx, AV_LOG_INFO, "%sMetadata:\n", indent);
3334 while((tag=av_dict_get(m, "", tag, AV_DICT_IGNORE_SUFFIX))) {
3335 if(strcmp("language", tag->key))
3336 av_log(ctx, AV_LOG_INFO, "%s %-16s: %s\n", indent, tag->key, tag->value);
3341 /* "user interface" functions */
3342 static void dump_stream_format(AVFormatContext *ic, int i, int index, int is_output)
3345 int flags = (is_output ? ic->oformat->flags : ic->iformat->flags);
3346 AVStream *st = ic->streams[i];
3347 int g = av_gcd(st->time_base.num, st->time_base.den);
3348 AVDictionaryEntry *lang = av_dict_get(st->metadata, "language", NULL, 0);
3349 avcodec_string(buf, sizeof(buf), st->codec, is_output);
3350 av_log(NULL, AV_LOG_INFO, " Stream #%d.%d", index, i);
3351 /* the pid is an important information, so we display it */
3352 /* XXX: add a generic system */
3353 if (flags & AVFMT_SHOW_IDS)
3354 av_log(NULL, AV_LOG_INFO, "[0x%x]", st->id);
3356 av_log(NULL, AV_LOG_INFO, "(%s)", lang->value);
3357 av_log(NULL, AV_LOG_DEBUG, ", %d, %d/%d", st->codec_info_nb_frames, st->time_base.num/g, st->time_base.den/g);
3358 av_log(NULL, AV_LOG_INFO, ": %s", buf);
3359 if (st->sample_aspect_ratio.num && // default
3360 av_cmp_q(st->sample_aspect_ratio, st->codec->sample_aspect_ratio)) {
3361 AVRational display_aspect_ratio;
3362 av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
3363 st->codec->width*st->sample_aspect_ratio.num,
3364 st->codec->height*st->sample_aspect_ratio.den,
3366 av_log(NULL, AV_LOG_INFO, ", PAR %d:%d DAR %d:%d",
3367 st->sample_aspect_ratio.num, st->sample_aspect_ratio.den,
3368 display_aspect_ratio.num, display_aspect_ratio.den);
3370 if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO){
3371 if(st->avg_frame_rate.den && st->avg_frame_rate.num)
3372 print_fps(av_q2d(st->avg_frame_rate), "fps");
3373 #if FF_API_R_FRAME_RATE
3374 if(st->r_frame_rate.den && st->r_frame_rate.num)
3375 print_fps(av_q2d(st->r_frame_rate), "tbr");
3377 if(st->time_base.den && st->time_base.num)
3378 print_fps(1/av_q2d(st->time_base), "tbn");
3379 if(st->codec->time_base.den && st->codec->time_base.num)
3380 print_fps(1/av_q2d(st->codec->time_base), "tbc");
3382 if (st->disposition & AV_DISPOSITION_DEFAULT)
3383 av_log(NULL, AV_LOG_INFO, " (default)");
3384 if (st->disposition & AV_DISPOSITION_DUB)
3385 av_log(NULL, AV_LOG_INFO, " (dub)");
3386 if (st->disposition & AV_DISPOSITION_ORIGINAL)
3387 av_log(NULL, AV_LOG_INFO, " (original)");
3388 if (st->disposition & AV_DISPOSITION_COMMENT)
3389 av_log(NULL, AV_LOG_INFO, " (comment)");
3390 if (st->disposition & AV_DISPOSITION_LYRICS)
3391 av_log(NULL, AV_LOG_INFO, " (lyrics)");
3392 if (st->disposition & AV_DISPOSITION_KARAOKE)
3393 av_log(NULL, AV_LOG_INFO, " (karaoke)");
3394 if (st->disposition & AV_DISPOSITION_FORCED)
3395 av_log(NULL, AV_LOG_INFO, " (forced)");
3396 if (st->disposition & AV_DISPOSITION_HEARING_IMPAIRED)
3397 av_log(NULL, AV_LOG_INFO, " (hearing impaired)");
3398 if (st->disposition & AV_DISPOSITION_VISUAL_IMPAIRED)
3399 av_log(NULL, AV_LOG_INFO, " (visual impaired)");
3400 if (st->disposition & AV_DISPOSITION_CLEAN_EFFECTS)
3401 av_log(NULL, AV_LOG_INFO, " (clean effects)");
3402 av_log(NULL, AV_LOG_INFO, "\n");
3403 dump_metadata(NULL, st->metadata, " ");
3406 void av_dump_format(AVFormatContext *ic,
3412 uint8_t *printed = ic->nb_streams ? av_mallocz(ic->nb_streams) : NULL;
3413 if (ic->nb_streams && !printed)
3416 av_log(NULL, AV_LOG_INFO, "%s #%d, %s, %s '%s':\n",
3417 is_output ? "Output" : "Input",
3419 is_output ? ic->oformat->name : ic->iformat->name,
3420 is_output ? "to" : "from", url);
3421 dump_metadata(NULL, ic->metadata, " ");
3423 av_log(NULL, AV_LOG_INFO, " Duration: ");
3424 if (ic->duration != AV_NOPTS_VALUE) {
3425 int hours, mins, secs, us;
3426 secs = ic->duration / AV_TIME_BASE;
3427 us = ic->duration % AV_TIME_BASE;
3432 av_log(NULL, AV_LOG_INFO, "%02d:%02d:%02d.%02d", hours, mins, secs,
3433 (100 * us) / AV_TIME_BASE);
3435 av_log(NULL, AV_LOG_INFO, "N/A");
3437 if (ic->start_time != AV_NOPTS_VALUE) {
3439 av_log(NULL, AV_LOG_INFO, ", start: ");
3440 secs = ic->start_time / AV_TIME_BASE;
3441 us = abs(ic->start_time % AV_TIME_BASE);
3442 av_log(NULL, AV_LOG_INFO, "%d.%06d",
3443 secs, (int)av_rescale(us, 1000000, AV_TIME_BASE));
3445 av_log(NULL, AV_LOG_INFO, ", bitrate: ");
3447 av_log(NULL, AV_LOG_INFO,"%d kb/s", ic->bit_rate / 1000);
3449 av_log(NULL, AV_LOG_INFO, "N/A");
3451 av_log(NULL, AV_LOG_INFO, "\n");
3453 for (i = 0; i < ic->nb_chapters; i++) {
3454 AVChapter *ch = ic->chapters[i];
3455 av_log(NULL, AV_LOG_INFO, " Chapter #%d.%d: ", index, i);
3456 av_log(NULL, AV_LOG_INFO, "start %f, ", ch->start * av_q2d(ch->time_base));
3457 av_log(NULL, AV_LOG_INFO, "end %f\n", ch->end * av_q2d(ch->time_base));
3459 dump_metadata(NULL, ch->metadata, " ");
3461 if(ic->nb_programs) {
3462 int j, k, total = 0;
3463 for(j=0; j<ic->nb_programs; j++) {
3464 AVDictionaryEntry *name = av_dict_get(ic->programs[j]->metadata,
3466 av_log(NULL, AV_LOG_INFO, " Program %d %s\n", ic->programs[j]->id,
3467 name ? name->value : "");
3468 dump_metadata(NULL, ic->programs[j]->metadata, " ");
3469 for(k=0; k<ic->programs[j]->nb_stream_indexes; k++) {
3470 dump_stream_format(ic, ic->programs[j]->stream_index[k], index, is_output);
3471 printed[ic->programs[j]->stream_index[k]] = 1;
3473 total += ic->programs[j]->nb_stream_indexes;
3475 if (total < ic->nb_streams)
3476 av_log(NULL, AV_LOG_INFO, " No Program\n");
3478 for(i=0;i<ic->nb_streams;i++)
3480 dump_stream_format(ic, i, index, is_output);
3485 #if FF_API_AV_GETTIME && CONFIG_SHARED && HAVE_SYMVER
3486 FF_SYMVER(int64_t, av_gettime, (void), "LIBAVFORMAT_54")
3488 return av_gettime();
3492 uint64_t ff_ntp_time(void)
3494 return (av_gettime() / 1000) * 1000 + NTP_OFFSET_US;
3497 int av_get_frame_filename(char *buf, int buf_size,
3498 const char *path, int number)
3501 char *q, buf1[20], c;
3502 int nd, len, percentd_found;
3514 while (isdigit(*p)) {
3515 nd = nd * 10 + *p++ - '0';
3518 } while (isdigit(c));
3527 snprintf(buf1, sizeof(buf1), "%0*d", nd, number);
3529 if ((q - buf + len) > buf_size - 1)
3531 memcpy(q, buf1, len);
3539 if ((q - buf) < buf_size - 1)
3543 if (!percentd_found)
3552 static void hex_dump_internal(void *avcl, FILE *f, int level, uint8_t *buf, int size)
3556 #define PRINT(...) do { if (!f) av_log(avcl, level, __VA_ARGS__); else fprintf(f, __VA_ARGS__); } while(0)
3558 for(i=0;i<size;i+=16) {
3565 PRINT(" %02x", buf[i+j]);
3570 for(j=0;j<len;j++) {
3572 if (c < ' ' || c > '~')
3581 void av_hex_dump(FILE *f, uint8_t *buf, int size)
3583 hex_dump_internal(NULL, f, 0, buf, size);
3586 void av_hex_dump_log(void *avcl, int level, uint8_t *buf, int size)
3588 hex_dump_internal(avcl, NULL, level, buf, size);
3591 static void pkt_dump_internal(void *avcl, FILE *f, int level, AVPacket *pkt, int dump_payload, AVRational time_base)
3594 #define PRINT(...) do { if (!f) av_log(avcl, level, __VA_ARGS__); else fprintf(f, __VA_ARGS__); } while(0)
3595 PRINT("stream #%d:\n", pkt->stream_index);
3596 PRINT(" keyframe=%d\n", ((pkt->flags & AV_PKT_FLAG_KEY) != 0));
3597 PRINT(" duration=%0.3f\n", pkt->duration * av_q2d(time_base));
3598 /* DTS is _always_ valid after av_read_frame() */
3600 if (pkt->dts == AV_NOPTS_VALUE)
3603 PRINT("%0.3f", pkt->dts * av_q2d(time_base));
3604 /* PTS may not be known if B-frames are present. */
3606 if (pkt->pts == AV_NOPTS_VALUE)
3609 PRINT("%0.3f", pkt->pts * av_q2d(time_base));
3611 PRINT(" size=%d\n", pkt->size);
3614 av_hex_dump(f, pkt->data, pkt->size);
3617 void av_pkt_dump2(FILE *f, AVPacket *pkt, int dump_payload, AVStream *st)
3619 pkt_dump_internal(NULL, f, 0, pkt, dump_payload, st->time_base);
3622 void av_pkt_dump_log2(void *avcl, int level, AVPacket *pkt, int dump_payload,