Statistics
| Branch: | Revision:

ffmpeg / libavformat / utils.c @ 118ccae0

History | View | Annotate | Download (119 KB)

1
/*
2
 * various utility functions for use within FFmpeg
3
 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
4
 *
5
 * This file is part of FFmpeg.
6
 *
7
 * FFmpeg 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.
11
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
 * Lesser General Public License for more details.
16
 *
17
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with FFmpeg; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21
#include "avformat.h"
22
#include "internal.h"
23
#include "libavcodec/internal.h"
24
#include "libavutil/opt.h"
25
#include "metadata.h"
26
#include "id3v2.h"
27
#include "libavutil/avstring.h"
28
#include "riff.h"
29
#include "audiointerleave.h"
30
#include <sys/time.h>
31
#include <time.h>
32
#include <strings.h>
33
#include <stdarg.h>
34
#if CONFIG_NETWORK
35
#include "network.h"
36
#endif
37

    
38
#undef NDEBUG
39
#include <assert.h>
40

    
41
/**
42
 * @file
43
 * various utility functions for use within FFmpeg
44
 */
45

    
46
unsigned avformat_version(void)
47
{
48
    return LIBAVFORMAT_VERSION_INT;
49
}
50

    
51
const char *avformat_configuration(void)
52
{
53
    return FFMPEG_CONFIGURATION;
54
}
55

    
56
const char *avformat_license(void)
57
{
58
#define LICENSE_PREFIX "libavformat license: "
59
    return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
60
}
61

    
62
/* fraction handling */
63

    
64
/**
65
 * f = val + (num / den) + 0.5.
66
 *
67
 * 'num' is normalized so that it is such as 0 <= num < den.
68
 *
69
 * @param f fractional number
70
 * @param val integer value
71
 * @param num must be >= 0
72
 * @param den must be >= 1
73
 */
74
static void av_frac_init(AVFrac *f, int64_t val, int64_t num, int64_t den)
75
{
76
    num += (den >> 1);
77
    if (num >= den) {
78
        val += num / den;
79
        num = num % den;
80
    }
81
    f->val = val;
82
    f->num = num;
83
    f->den = den;
84
}
85

    
86
/**
87
 * Fractional addition to f: f = f + (incr / f->den).
88
 *
89
 * @param f fractional number
90
 * @param incr increment, can be positive or negative
91
 */
92
static void av_frac_add(AVFrac *f, int64_t incr)
93
{
94
    int64_t num, den;
95

    
96
    num = f->num + incr;
97
    den = f->den;
98
    if (num < 0) {
99
        f->val += num / den;
100
        num = num % den;
101
        if (num < 0) {
102
            num += den;
103
            f->val--;
104
        }
105
    } else if (num >= den) {
106
        f->val += num / den;
107
        num = num % den;
108
    }
109
    f->num = num;
110
}
111

    
112
/** head of registered input format linked list */
113
#if !FF_API_FIRST_FORMAT
114
static
115
#endif
116
AVInputFormat *first_iformat = NULL;
117
/** head of registered output format linked list */
118
#if !FF_API_FIRST_FORMAT
119
static
120
#endif
121
AVOutputFormat *first_oformat = NULL;
122

    
123
AVInputFormat  *av_iformat_next(AVInputFormat  *f)
124
{
125
    if(f) return f->next;
126
    else  return first_iformat;
127
}
128

    
129
AVOutputFormat *av_oformat_next(AVOutputFormat *f)
130
{
131
    if(f) return f->next;
132
    else  return first_oformat;
133
}
134

    
135
void av_register_input_format(AVInputFormat *format)
136
{
137
    AVInputFormat **p;
138
    p = &first_iformat;
139
    while (*p != NULL) p = &(*p)->next;
140
    *p = format;
141
    format->next = NULL;
142
}
143

    
144
void av_register_output_format(AVOutputFormat *format)
145
{
146
    AVOutputFormat **p;
147
    p = &first_oformat;
148
    while (*p != NULL) p = &(*p)->next;
149
    *p = format;
150
    format->next = NULL;
151
}
152

    
153
int av_match_ext(const char *filename, const char *extensions)
154
{
155
    const char *ext, *p;
156
    char ext1[32], *q;
157

    
158
    if(!filename)
159
        return 0;
160

    
161
    ext = strrchr(filename, '.');
162
    if (ext) {
163
        ext++;
164
        p = extensions;
165
        for(;;) {
166
            q = ext1;
167
            while (*p != '\0' && *p != ',' && q-ext1<sizeof(ext1)-1)
168
                *q++ = *p++;
169
            *q = '\0';
170
            if (!strcasecmp(ext1, ext))
171
                return 1;
172
            if (*p == '\0')
173
                break;
174
            p++;
175
        }
176
    }
177
    return 0;
178
}
179

    
180
static int match_format(const char *name, const char *names)
181
{
182
    const char *p;
183
    int len, namelen;
184

    
185
    if (!name || !names)
186
        return 0;
187

    
188
    namelen = strlen(name);
189
    while ((p = strchr(names, ','))) {
190
        len = FFMAX(p - names, namelen);
191
        if (!strncasecmp(name, names, len))
192
            return 1;
193
        names = p+1;
194
    }
195
    return !strcasecmp(name, names);
196
}
197

    
198
#if FF_API_GUESS_FORMAT
199
AVOutputFormat *guess_format(const char *short_name, const char *filename,
200
                             const char *mime_type)
201
{
202
    return av_guess_format(short_name, filename, mime_type);
203
}
204
#endif
205

    
206
AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
207
                                const char *mime_type)
208
{
209
    AVOutputFormat *fmt = NULL, *fmt_found;
210
    int score_max, score;
211

    
212
    /* specific test for image sequences */
213
#if CONFIG_IMAGE2_MUXER
214
    if (!short_name && filename &&
215
        av_filename_number_test(filename) &&
216
        av_guess_image2_codec(filename) != CODEC_ID_NONE) {
217
        return av_guess_format("image2", NULL, NULL);
218
    }
219
#endif
220
    /* Find the proper file type. */
221
    fmt_found = NULL;
222
    score_max = 0;
223
    while ((fmt = av_oformat_next(fmt))) {
224
        score = 0;
225
        if (fmt->name && short_name && !strcmp(fmt->name, short_name))
226
            score += 100;
227
        if (fmt->mime_type && mime_type && !strcmp(fmt->mime_type, mime_type))
228
            score += 10;
229
        if (filename && fmt->extensions &&
230
            av_match_ext(filename, fmt->extensions)) {
231
            score += 5;
232
        }
233
        if (score > score_max) {
234
            score_max = score;
235
            fmt_found = fmt;
236
        }
237
    }
238
    return fmt_found;
239
}
240

    
241
#if FF_API_GUESS_FORMAT
242
AVOutputFormat *guess_stream_format(const char *short_name, const char *filename,
243
                             const char *mime_type)
244
{
245
    AVOutputFormat *fmt = av_guess_format(short_name, filename, mime_type);
246

    
247
    if (fmt) {
248
        AVOutputFormat *stream_fmt;
249
        char stream_format_name[64];
250

    
251
        snprintf(stream_format_name, sizeof(stream_format_name), "%s_stream", fmt->name);
252
        stream_fmt = av_guess_format(stream_format_name, NULL, NULL);
253

    
254
        if (stream_fmt)
255
            fmt = stream_fmt;
256
    }
257

    
258
    return fmt;
259
}
260
#endif
261

    
262
enum CodecID av_guess_codec(AVOutputFormat *fmt, const char *short_name,
263
                            const char *filename, const char *mime_type, enum AVMediaType type){
264
    if(type == AVMEDIA_TYPE_VIDEO){
265
        enum CodecID codec_id= CODEC_ID_NONE;
266

    
267
#if CONFIG_IMAGE2_MUXER
268
        if(!strcmp(fmt->name, "image2") || !strcmp(fmt->name, "image2pipe")){
269
            codec_id= av_guess_image2_codec(filename);
270
        }
271
#endif
272
        if(codec_id == CODEC_ID_NONE)
273
            codec_id= fmt->video_codec;
274
        return codec_id;
275
    }else if(type == AVMEDIA_TYPE_AUDIO)
276
        return fmt->audio_codec;
277
    else if (type == AVMEDIA_TYPE_SUBTITLE)
278
        return fmt->subtitle_codec;
279
    else
280
        return CODEC_ID_NONE;
281
}
282

    
283
AVInputFormat *av_find_input_format(const char *short_name)
284
{
285
    AVInputFormat *fmt = NULL;
286
    while ((fmt = av_iformat_next(fmt))) {
287
        if (match_format(short_name, fmt->name))
288
            return fmt;
289
    }
290
    return NULL;
291
}
292

    
293
#if FF_API_SYMVER && CONFIG_SHARED && HAVE_SYMVER
294
FF_SYMVER(void, av_destruct_packet_nofree, (AVPacket *pkt), "LIBAVFORMAT_52")
295
{
296
    av_destruct_packet_nofree(pkt);
297
}
298

    
299
FF_SYMVER(void, av_destruct_packet, (AVPacket *pkt), "LIBAVFORMAT_52")
300
{
301
    av_destruct_packet(pkt);
302
}
303

    
304
FF_SYMVER(int, av_new_packet, (AVPacket *pkt, int size), "LIBAVFORMAT_52")
305
{
306
    return av_new_packet(pkt, size);
307
}
308

    
309
FF_SYMVER(int, av_dup_packet, (AVPacket *pkt), "LIBAVFORMAT_52")
310
{
311
    return av_dup_packet(pkt);
312
}
313

    
314
FF_SYMVER(void, av_free_packet, (AVPacket *pkt), "LIBAVFORMAT_52")
315
{
316
    av_free_packet(pkt);
317
}
318

    
319
FF_SYMVER(void, av_init_packet, (AVPacket *pkt), "LIBAVFORMAT_52")
320
{
321
    av_log(NULL, AV_LOG_WARNING, "Diverting av_*_packet function calls to libavcodec. Recompile to improve performance\n");
322
    av_init_packet(pkt);
323
}
324
#endif
325

    
326
int av_get_packet(ByteIOContext *s, AVPacket *pkt, int size)
327
{
328
    int ret= av_new_packet(pkt, size);
329

    
330
    if(ret<0)
331
        return ret;
332

    
333
    pkt->pos= url_ftell(s);
334

    
335
    ret= get_buffer(s, pkt->data, size);
336
    if(ret<=0)
337
        av_free_packet(pkt);
338
    else
339
        av_shrink_packet(pkt, ret);
340

    
341
    return ret;
342
}
343

    
344
int av_append_packet(ByteIOContext *s, AVPacket *pkt, int size)
345
{
346
    int ret;
347
    int old_size;
348
    if (!pkt->size)
349
        return av_get_packet(s, pkt, size);
350
    old_size = pkt->size;
351
    ret = av_grow_packet(pkt, size);
352
    if (ret < 0)
353
        return ret;
354
    ret = get_buffer(s, pkt->data + old_size, size);
355
    av_shrink_packet(pkt, old_size + FFMAX(ret, 0));
356
    return ret;
357
}
358

    
359

    
360
int av_filename_number_test(const char *filename)
361
{
362
    char buf[1024];
363
    return filename && (av_get_frame_filename(buf, sizeof(buf), filename, 1)>=0);
364
}
365

    
366
AVInputFormat *av_probe_input_format2(AVProbeData *pd, int is_opened, int *score_max)
367
{
368
    AVProbeData lpd = *pd;
369
    AVInputFormat *fmt1 = NULL, *fmt;
370
    int score;
371

    
372
    if (lpd.buf_size > 10 && ff_id3v2_match(lpd.buf, ID3v2_DEFAULT_MAGIC)) {
373
        int id3len = ff_id3v2_tag_len(lpd.buf);
374
        if (lpd.buf_size > id3len + 16) {
375
            lpd.buf += id3len;
376
            lpd.buf_size -= id3len;
377
        }
378
    }
379

    
380
    fmt = NULL;
381
    while ((fmt1 = av_iformat_next(fmt1))) {
382
        if (!is_opened == !(fmt1->flags & AVFMT_NOFILE))
383
            continue;
384
        score = 0;
385
        if (fmt1->read_probe) {
386
            score = fmt1->read_probe(&lpd);
387
        } else if (fmt1->extensions) {
388
            if (av_match_ext(lpd.filename, fmt1->extensions)) {
389
                score = 50;
390
            }
391
        }
392
        if (score > *score_max) {
393
            *score_max = score;
394
            fmt = fmt1;
395
        }else if (score == *score_max)
396
            fmt = NULL;
397
    }
398
    return fmt;
399
}
400

    
401
AVInputFormat *av_probe_input_format(AVProbeData *pd, int is_opened){
402
    int score=0;
403
    return av_probe_input_format2(pd, is_opened, &score);
404
}
405

    
406
static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st, AVProbeData *pd, int score)
407
{
408
    static const struct {
409
        const char *name; enum CodecID id; enum AVMediaType type;
410
    } fmt_id_type[] = {
411
        { "aac"      , CODEC_ID_AAC       , AVMEDIA_TYPE_AUDIO },
412
        { "ac3"      , CODEC_ID_AC3       , AVMEDIA_TYPE_AUDIO },
413
        { "dts"      , CODEC_ID_DTS       , AVMEDIA_TYPE_AUDIO },
414
        { "eac3"     , CODEC_ID_EAC3      , AVMEDIA_TYPE_AUDIO },
415
        { "h264"     , CODEC_ID_H264      , AVMEDIA_TYPE_VIDEO },
416
        { "m4v"      , CODEC_ID_MPEG4     , AVMEDIA_TYPE_VIDEO },
417
        { "mp3"      , CODEC_ID_MP3       , AVMEDIA_TYPE_AUDIO },
418
        { "mpegvideo", CODEC_ID_MPEG2VIDEO, AVMEDIA_TYPE_VIDEO },
419
        { 0 }
420
    };
421
    AVInputFormat *fmt = av_probe_input_format2(pd, 1, &score);
422

    
423
    if (fmt) {
424
        int i;
425
        av_log(s, AV_LOG_DEBUG, "Probe with size=%d, packets=%d detected %s with score=%d\n",
426
               pd->buf_size, MAX_PROBE_PACKETS - st->probe_packets, fmt->name, score);
427
        for (i = 0; fmt_id_type[i].name; i++) {
428
            if (!strcmp(fmt->name, fmt_id_type[i].name)) {
429
                st->codec->codec_id   = fmt_id_type[i].id;
430
                st->codec->codec_type = fmt_id_type[i].type;
431
                break;
432
            }
433
        }
434
    }
435
    return !!fmt;
436
}
437

    
438
/************************************************************/
439
/* input media file */
440

    
441
/**
442
 * Open a media file from an IO stream. 'fmt' must be specified.
443
 */
444
int av_open_input_stream(AVFormatContext **ic_ptr,
445
                         ByteIOContext *pb, const char *filename,
446
                         AVInputFormat *fmt, AVFormatParameters *ap)
447
{
448
    int err;
449
    AVFormatContext *ic;
450
    AVFormatParameters default_ap;
451

    
452
    if(!ap){
453
        ap=&default_ap;
454
        memset(ap, 0, sizeof(default_ap));
455
    }
456

    
457
    if(!ap->prealloced_context)
458
        ic = avformat_alloc_context();
459
    else
460
        ic = *ic_ptr;
461
    if (!ic) {
462
        err = AVERROR(ENOMEM);
463
        goto fail;
464
    }
465
    ic->iformat = fmt;
466
    ic->pb = pb;
467
    ic->duration = AV_NOPTS_VALUE;
468
    ic->start_time = AV_NOPTS_VALUE;
469
    av_strlcpy(ic->filename, filename, sizeof(ic->filename));
470

    
471
    /* allocate private data */
472
    if (fmt->priv_data_size > 0) {
473
        ic->priv_data = av_mallocz(fmt->priv_data_size);
474
        if (!ic->priv_data) {
475
            err = AVERROR(ENOMEM);
476
            goto fail;
477
        }
478
    } else {
479
        ic->priv_data = NULL;
480
    }
481

    
482
    // e.g. AVFMT_NOFILE formats will not have a ByteIOContext
483
    if (ic->pb)
484
        ff_id3v2_read(ic, ID3v2_DEFAULT_MAGIC);
485

    
486
    if (ic->iformat->read_header) {
487
        err = ic->iformat->read_header(ic, ap);
488
        if (err < 0)
489
            goto fail;
490
    }
491

    
492
    if (pb && !ic->data_offset)
493
        ic->data_offset = url_ftell(ic->pb);
494

    
495
#if FF_API_OLD_METADATA
496
    ff_metadata_demux_compat(ic);
497
#endif
498

    
499
    ic->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
500

    
501
    *ic_ptr = ic;
502
    return 0;
503
 fail:
504
    if (ic) {
505
        int i;
506
        av_freep(&ic->priv_data);
507
        for(i=0;i<ic->nb_streams;i++) {
508
            AVStream *st = ic->streams[i];
509
            if (st) {
510
                av_free(st->priv_data);
511
                av_free(st->codec->extradata);
512
                av_free(st->codec);
513
                av_free(st->info);
514
            }
515
            av_free(st);
516
        }
517
    }
518
    av_free(ic);
519
    *ic_ptr = NULL;
520
    return err;
521
}
522

    
523
/** size of probe buffer, for guessing file type from file contents */
524
#define PROBE_BUF_MIN 2048
525
#define PROBE_BUF_MAX (1<<20)
526

    
527
int ff_probe_input_buffer(ByteIOContext **pb, AVInputFormat **fmt,
528
                          const char *filename, void *logctx,
529
                          unsigned int offset, unsigned int max_probe_size)
530
{
531
    AVProbeData pd = { filename ? filename : "", NULL, -offset };
532
    unsigned char *buf = NULL;
533
    int ret = 0, probe_size;
534

    
535
    if (!max_probe_size) {
536
        max_probe_size = PROBE_BUF_MAX;
537
    } else if (max_probe_size > PROBE_BUF_MAX) {
538
        max_probe_size = PROBE_BUF_MAX;
539
    } else if (max_probe_size < PROBE_BUF_MIN) {
540
        return AVERROR(EINVAL);
541
    }
542

    
543
    if (offset >= max_probe_size) {
544
        return AVERROR(EINVAL);
545
    }
546

    
547
    for(probe_size= PROBE_BUF_MIN; probe_size<=max_probe_size && !*fmt && ret >= 0;
548
        probe_size = FFMIN(probe_size<<1, FFMAX(max_probe_size, probe_size+1))) {
549
        int ret, score = probe_size < max_probe_size ? AVPROBE_SCORE_MAX/4 : 0;
550
        int buf_offset = (probe_size == PROBE_BUF_MIN) ? 0 : probe_size>>1;
551

    
552
        if (probe_size < offset) {
553
            continue;
554
        }
555

    
556
        /* read probe data */
557
        buf = av_realloc(buf, probe_size + AVPROBE_PADDING_SIZE);
558
        if ((ret = get_buffer(*pb, buf + buf_offset, probe_size - buf_offset)) < 0) {
559
            /* fail if error was not end of file, otherwise, lower score */
560
            if (ret != AVERROR_EOF) {
561
                av_free(buf);
562
                return ret;
563
            }
564
            score = 0;
565
            ret = 0;            /* error was end of file, nothing read */
566
        }
567
        pd.buf_size += ret;
568
        pd.buf = &buf[offset];
569

    
570
        memset(pd.buf + pd.buf_size, 0, AVPROBE_PADDING_SIZE);
571

    
572
        /* guess file format */
573
        *fmt = av_probe_input_format2(&pd, 1, &score);
574
        if(*fmt){
575
            if(score <= AVPROBE_SCORE_MAX/4){ //this can only be true in the last iteration
576
                av_log(logctx, AV_LOG_WARNING, "Format detected only with low score of %d, misdetection possible!\n", score);
577
            }else
578
                av_log(logctx, AV_LOG_DEBUG, "Probed with size=%d and score=%d\n", probe_size, score);
579
        }
580
    }
581

    
582
    if (!*fmt) {
583
        av_free(buf);
584
        return AVERROR_INVALIDDATA;
585
    }
586

    
587
    /* rewind. reuse probe buffer to avoid seeking */
588
    if ((ret = ff_rewind_with_probe_data(*pb, buf, pd.buf_size)) < 0)
589
        av_free(buf);
590

    
591
    return ret;
592
}
593

    
594
int av_open_input_file(AVFormatContext **ic_ptr, const char *filename,
595
                       AVInputFormat *fmt,
596
                       int buf_size,
597
                       AVFormatParameters *ap)
598
{
599
    int err;
600
    AVProbeData probe_data, *pd = &probe_data;
601
    ByteIOContext *pb = NULL;
602
    void *logctx= ap && ap->prealloced_context ? *ic_ptr : NULL;
603

    
604
    pd->filename = "";
605
    if (filename)
606
        pd->filename = filename;
607
    pd->buf = NULL;
608
    pd->buf_size = 0;
609

    
610
    if (!fmt) {
611
        /* guess format if no file can be opened */
612
        fmt = av_probe_input_format(pd, 0);
613
    }
614

    
615
    /* Do not open file if the format does not need it. XXX: specific
616
       hack needed to handle RTSP/TCP */
617
    if (!fmt || !(fmt->flags & AVFMT_NOFILE)) {
618
        /* if no file needed do not try to open one */
619
        if ((err=url_fopen(&pb, filename, URL_RDONLY)) < 0) {
620
            goto fail;
621
        }
622
        if (buf_size > 0) {
623
            url_setbufsize(pb, buf_size);
624
        }
625
        if (!fmt && (err = ff_probe_input_buffer(&pb, &fmt, filename, logctx, 0, logctx ? (*ic_ptr)->probesize : 0)) < 0) {
626
            goto fail;
627
        }
628
    }
629

    
630
    /* if still no format found, error */
631
    if (!fmt) {
632
        err = AVERROR_INVALIDDATA;
633
        goto fail;
634
    }
635

    
636
    /* check filename in case an image number is expected */
637
    if (fmt->flags & AVFMT_NEEDNUMBER) {
638
        if (!av_filename_number_test(filename)) {
639
            err = AVERROR_NUMEXPECTED;
640
            goto fail;
641
        }
642
    }
643
    err = av_open_input_stream(ic_ptr, pb, filename, fmt, ap);
644
    if (err)
645
        goto fail;
646
    return 0;
647
 fail:
648
    av_freep(&pd->buf);
649
    if (pb)
650
        url_fclose(pb);
651
    if (ap && ap->prealloced_context)
652
        av_free(*ic_ptr);
653
    *ic_ptr = NULL;
654
    return err;
655

    
656
}
657

    
658
/*******************************************************/
659

    
660
static AVPacket *add_to_pktbuf(AVPacketList **packet_buffer, AVPacket *pkt,
661
                               AVPacketList **plast_pktl){
662
    AVPacketList *pktl = av_mallocz(sizeof(AVPacketList));
663
    if (!pktl)
664
        return NULL;
665

    
666
    if (*packet_buffer)
667
        (*plast_pktl)->next = pktl;
668
    else
669
        *packet_buffer = pktl;
670

    
671
    /* add the packet in the buffered packet list */
672
    *plast_pktl = pktl;
673
    pktl->pkt= *pkt;
674
    return &pktl->pkt;
675
}
676

    
677
int av_read_packet(AVFormatContext *s, AVPacket *pkt)
678
{
679
    int ret, i;
680
    AVStream *st;
681

    
682
    for(;;){
683
        AVPacketList *pktl = s->raw_packet_buffer;
684

    
685
        if (pktl) {
686
            *pkt = pktl->pkt;
687
            if(s->streams[pkt->stream_index]->codec->codec_id != CODEC_ID_PROBE ||
688
               !s->streams[pkt->stream_index]->probe_packets ||
689
               s->raw_packet_buffer_remaining_size < pkt->size){
690
                AVProbeData *pd = &s->streams[pkt->stream_index]->probe_data;
691
                av_freep(&pd->buf);
692
                pd->buf_size = 0;
693
                s->raw_packet_buffer = pktl->next;
694
                s->raw_packet_buffer_remaining_size += pkt->size;
695
                av_free(pktl);
696
                return 0;
697
            }
698
        }
699

    
700
        av_init_packet(pkt);
701
        ret= s->iformat->read_packet(s, pkt);
702
        if (ret < 0) {
703
            if (!pktl || ret == AVERROR(EAGAIN))
704
                return ret;
705
            for (i = 0; i < s->nb_streams; i++)
706
                s->streams[i]->probe_packets = 0;
707
            continue;
708
        }
709
        st= s->streams[pkt->stream_index];
710

    
711
        switch(st->codec->codec_type){
712
        case AVMEDIA_TYPE_VIDEO:
713
            if(s->video_codec_id)   st->codec->codec_id= s->video_codec_id;
714
            break;
715
        case AVMEDIA_TYPE_AUDIO:
716
            if(s->audio_codec_id)   st->codec->codec_id= s->audio_codec_id;
717
            break;
718
        case AVMEDIA_TYPE_SUBTITLE:
719
            if(s->subtitle_codec_id)st->codec->codec_id= s->subtitle_codec_id;
720
            break;
721
        }
722

    
723
        if(!pktl && (st->codec->codec_id != CODEC_ID_PROBE ||
724
                     !st->probe_packets))
725
            return ret;
726

    
727
        add_to_pktbuf(&s->raw_packet_buffer, pkt, &s->raw_packet_buffer_end);
728
        s->raw_packet_buffer_remaining_size -= pkt->size;
729

    
730
        if(st->codec->codec_id == CODEC_ID_PROBE){
731
            AVProbeData *pd = &st->probe_data;
732
            av_log(s, AV_LOG_DEBUG, "probing stream %d\n", st->index);
733
            --st->probe_packets;
734

    
735
            pd->buf = av_realloc(pd->buf, pd->buf_size+pkt->size+AVPROBE_PADDING_SIZE);
736
            memcpy(pd->buf+pd->buf_size, pkt->data, pkt->size);
737
            pd->buf_size += pkt->size;
738
            memset(pd->buf+pd->buf_size, 0, AVPROBE_PADDING_SIZE);
739

    
740
            if(av_log2(pd->buf_size) != av_log2(pd->buf_size - pkt->size)){
741
                //FIXME we dont reduce score to 0 for the case of running out of buffer space in bytes
742
                set_codec_from_probe_data(s, st, pd, st->probe_packets > 0 ? AVPROBE_SCORE_MAX/4 : 0);
743
                if(st->codec->codec_id != CODEC_ID_PROBE){
744
                    pd->buf_size=0;
745
                    av_freep(&pd->buf);
746
                    av_log(s, AV_LOG_DEBUG, "probed stream %d\n", st->index);
747
                }
748
            }
749
        }
750
    }
751
}
752

    
753
/**********************************************************/
754

    
755
/**
756
 * Get the number of samples of an audio frame. Return -1 on error.
757
 */
758
static int get_audio_frame_size(AVCodecContext *enc, int size)
759
{
760
    int frame_size;
761

    
762
    if(enc->codec_id == CODEC_ID_VORBIS)
763
        return -1;
764

    
765
    if (enc->frame_size <= 1) {
766
        int bits_per_sample = av_get_bits_per_sample(enc->codec_id);
767

    
768
        if (bits_per_sample) {
769
            if (enc->channels == 0)
770
                return -1;
771
            frame_size = (size << 3) / (bits_per_sample * enc->channels);
772
        } else {
773
            /* used for example by ADPCM codecs */
774
            if (enc->bit_rate == 0)
775
                return -1;
776
            frame_size = ((int64_t)size * 8 * enc->sample_rate) / enc->bit_rate;
777
        }
778
    } else {
779
        frame_size = enc->frame_size;
780
    }
781
    return frame_size;
782
}
783

    
784

    
785
/**
786
 * Return the frame duration in seconds. Return 0 if not available.
787
 */
788
static void compute_frame_duration(int *pnum, int *pden, AVStream *st,
789
                                   AVCodecParserContext *pc, AVPacket *pkt)
790
{
791
    int frame_size;
792

    
793
    *pnum = 0;
794
    *pden = 0;
795
    switch(st->codec->codec_type) {
796
    case AVMEDIA_TYPE_VIDEO:
797
        if(st->time_base.num*1000LL > st->time_base.den){
798
            *pnum = st->time_base.num;
799
            *pden = st->time_base.den;
800
        }else if(st->codec->time_base.num*1000LL > st->codec->time_base.den){
801
            *pnum = st->codec->time_base.num;
802
            *pden = st->codec->time_base.den;
803
            if (pc && pc->repeat_pict) {
804
                *pnum = (*pnum) * (1 + pc->repeat_pict);
805
            }
806
            //If this codec can be interlaced or progressive then we need a parser to compute duration of a packet
807
            //Thus if we have no parser in such case leave duration undefined.
808
            if(st->codec->ticks_per_frame>1 && !pc){
809
                *pnum = *pden = 0;
810
            }
811
        }
812
        break;
813
    case AVMEDIA_TYPE_AUDIO:
814
        frame_size = get_audio_frame_size(st->codec, pkt->size);
815
        if (frame_size < 0)
816
            break;
817
        *pnum = frame_size;
818
        *pden = st->codec->sample_rate;
819
        break;
820
    default:
821
        break;
822
    }
823
}
824

    
825
static int is_intra_only(AVCodecContext *enc){
826
    if(enc->codec_type == AVMEDIA_TYPE_AUDIO){
827
        return 1;
828
    }else if(enc->codec_type == AVMEDIA_TYPE_VIDEO){
829
        switch(enc->codec_id){
830
        case CODEC_ID_MJPEG:
831
        case CODEC_ID_MJPEGB:
832
        case CODEC_ID_LJPEG:
833
        case CODEC_ID_RAWVIDEO:
834
        case CODEC_ID_DVVIDEO:
835
        case CODEC_ID_HUFFYUV:
836
        case CODEC_ID_FFVHUFF:
837
        case CODEC_ID_ASV1:
838
        case CODEC_ID_ASV2:
839
        case CODEC_ID_VCR1:
840
        case CODEC_ID_DNXHD:
841
        case CODEC_ID_JPEG2000:
842
            return 1;
843
        default: break;
844
        }
845
    }
846
    return 0;
847
}
848

    
849
static void update_initial_timestamps(AVFormatContext *s, int stream_index,
850
                                      int64_t dts, int64_t pts)
851
{
852
    AVStream *st= s->streams[stream_index];
853
    AVPacketList *pktl= s->packet_buffer;
854

    
855
    if(st->first_dts != AV_NOPTS_VALUE || dts == AV_NOPTS_VALUE || st->cur_dts == AV_NOPTS_VALUE)
856
        return;
857

    
858
    st->first_dts= dts - st->cur_dts;
859
    st->cur_dts= dts;
860

    
861
    for(; pktl; pktl= pktl->next){
862
        if(pktl->pkt.stream_index != stream_index)
863
            continue;
864
        //FIXME think more about this check
865
        if(pktl->pkt.pts != AV_NOPTS_VALUE && pktl->pkt.pts == pktl->pkt.dts)
866
            pktl->pkt.pts += st->first_dts;
867

    
868
        if(pktl->pkt.dts != AV_NOPTS_VALUE)
869
            pktl->pkt.dts += st->first_dts;
870

    
871
        if(st->start_time == AV_NOPTS_VALUE && pktl->pkt.pts != AV_NOPTS_VALUE)
872
            st->start_time= pktl->pkt.pts;
873
    }
874
    if (st->start_time == AV_NOPTS_VALUE)
875
        st->start_time = pts;
876
}
877

    
878
static void update_initial_durations(AVFormatContext *s, AVStream *st, AVPacket *pkt)
879
{
880
    AVPacketList *pktl= s->packet_buffer;
881
    int64_t cur_dts= 0;
882

    
883
    if(st->first_dts != AV_NOPTS_VALUE){
884
        cur_dts= st->first_dts;
885
        for(; pktl; pktl= pktl->next){
886
            if(pktl->pkt.stream_index == pkt->stream_index){
887
                if(pktl->pkt.pts != pktl->pkt.dts || pktl->pkt.dts != AV_NOPTS_VALUE || pktl->pkt.duration)
888
                    break;
889
                cur_dts -= pkt->duration;
890
            }
891
        }
892
        pktl= s->packet_buffer;
893
        st->first_dts = cur_dts;
894
    }else if(st->cur_dts)
895
        return;
896

    
897
    for(; pktl; pktl= pktl->next){
898
        if(pktl->pkt.stream_index != pkt->stream_index)
899
            continue;
900
        if(pktl->pkt.pts == pktl->pkt.dts && pktl->pkt.dts == AV_NOPTS_VALUE
901
           && !pktl->pkt.duration){
902
            pktl->pkt.dts= cur_dts;
903
            if(!st->codec->has_b_frames)
904
                pktl->pkt.pts= cur_dts;
905
            cur_dts += pkt->duration;
906
            pktl->pkt.duration= pkt->duration;
907
        }else
908
            break;
909
    }
910
    if(st->first_dts == AV_NOPTS_VALUE)
911
        st->cur_dts= cur_dts;
912
}
913

    
914
static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
915
                               AVCodecParserContext *pc, AVPacket *pkt)
916
{
917
    int num, den, presentation_delayed, delay, i;
918
    int64_t offset;
919

    
920
    if (s->flags & AVFMT_FLAG_NOFILLIN)
921
        return;
922

    
923
    if((s->flags & AVFMT_FLAG_IGNDTS) && pkt->pts != AV_NOPTS_VALUE)
924
        pkt->dts= AV_NOPTS_VALUE;
925

    
926
    if (st->codec->codec_id != CODEC_ID_H264 && pc && pc->pict_type == FF_B_TYPE)
927
        //FIXME Set low_delay = 0 when has_b_frames = 1
928
        st->codec->has_b_frames = 1;
929

    
930
    /* do we have a video B-frame ? */
931
    delay= st->codec->has_b_frames;
932
    presentation_delayed = 0;
933
    /* XXX: need has_b_frame, but cannot get it if the codec is
934
        not initialized */
935
    if (delay &&
936
        pc && pc->pict_type != FF_B_TYPE)
937
        presentation_delayed = 1;
938

    
939
    if(pkt->pts != AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE && pkt->dts > pkt->pts && st->pts_wrap_bits<63
940
       /*&& pkt->dts-(1LL<<st->pts_wrap_bits) < pkt->pts*/){
941
        pkt->dts -= 1LL<<st->pts_wrap_bits;
942
    }
943

    
944
    // some mpeg2 in mpeg-ps lack dts (issue171 / input_file.mpg)
945
    // we take the conservative approach and discard both
946
    // Note, if this is misbehaving for a H.264 file then possibly presentation_delayed is not set correctly.
947
    if(delay==1 && pkt->dts == pkt->pts && pkt->dts != AV_NOPTS_VALUE && presentation_delayed){
948
        av_log(s, AV_LOG_DEBUG, "invalid dts/pts combination\n");
949
        pkt->dts= pkt->pts= AV_NOPTS_VALUE;
950
    }
951

    
952
    if (pkt->duration == 0) {
953
        compute_frame_duration(&num, &den, st, pc, pkt);
954
        if (den && num) {
955
            pkt->duration = av_rescale_rnd(1, num * (int64_t)st->time_base.den, den * (int64_t)st->time_base.num, AV_ROUND_DOWN);
956

    
957
            if(pkt->duration != 0 && s->packet_buffer)
958
                update_initial_durations(s, st, pkt);
959
        }
960
    }
961

    
962
    /* correct timestamps with byte offset if demuxers only have timestamps
963
       on packet boundaries */
964
    if(pc && st->need_parsing == AVSTREAM_PARSE_TIMESTAMPS && pkt->size){
965
        /* this will estimate bitrate based on this frame's duration and size */
966
        offset = av_rescale(pc->offset, pkt->duration, pkt->size);
967
        if(pkt->pts != AV_NOPTS_VALUE)
968
            pkt->pts += offset;
969
        if(pkt->dts != AV_NOPTS_VALUE)
970
            pkt->dts += offset;
971
    }
972

    
973
    if (pc && pc->dts_sync_point >= 0) {
974
        // we have synchronization info from the parser
975
        int64_t den = st->codec->time_base.den * (int64_t) st->time_base.num;
976
        if (den > 0) {
977
            int64_t num = st->codec->time_base.num * (int64_t) st->time_base.den;
978
            if (pkt->dts != AV_NOPTS_VALUE) {
979
                // got DTS from the stream, update reference timestamp
980
                st->reference_dts = pkt->dts - pc->dts_ref_dts_delta * num / den;
981
                pkt->pts = pkt->dts + pc->pts_dts_delta * num / den;
982
            } else if (st->reference_dts != AV_NOPTS_VALUE) {
983
                // compute DTS based on reference timestamp
984
                pkt->dts = st->reference_dts + pc->dts_ref_dts_delta * num / den;
985
                pkt->pts = pkt->dts + pc->pts_dts_delta * num / den;
986
            }
987
            if (pc->dts_sync_point > 0)
988
                st->reference_dts = pkt->dts; // new reference
989
        }
990
    }
991

    
992
    /* This may be redundant, but it should not hurt. */
993
    if(pkt->dts != AV_NOPTS_VALUE && pkt->pts != AV_NOPTS_VALUE && pkt->pts > pkt->dts)
994
        presentation_delayed = 1;
995

    
996
//    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);
997
    /* interpolate PTS and DTS if they are not present */
998
    //We skip H264 currently because delay and has_b_frames are not reliably set
999
    if((delay==0 || (delay==1 && pc)) && st->codec->codec_id != CODEC_ID_H264){
1000
        if (presentation_delayed) {
1001
            /* DTS = decompression timestamp */
1002
            /* PTS = presentation timestamp */
1003
            if (pkt->dts == AV_NOPTS_VALUE)
1004
                pkt->dts = st->last_IP_pts;
1005
            update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts);
1006
            if (pkt->dts == AV_NOPTS_VALUE)
1007
                pkt->dts = st->cur_dts;
1008

    
1009
            /* this is tricky: the dts must be incremented by the duration
1010
            of the frame we are displaying, i.e. the last I- or P-frame */
1011
            if (st->last_IP_duration == 0)
1012
                st->last_IP_duration = pkt->duration;
1013
            if(pkt->dts != AV_NOPTS_VALUE)
1014
                st->cur_dts = pkt->dts + st->last_IP_duration;
1015
            st->last_IP_duration  = pkt->duration;
1016
            st->last_IP_pts= pkt->pts;
1017
            /* cannot compute PTS if not present (we can compute it only
1018
            by knowing the future */
1019
        } else if(pkt->pts != AV_NOPTS_VALUE || pkt->dts != AV_NOPTS_VALUE || pkt->duration){
1020
            if(pkt->pts != AV_NOPTS_VALUE && pkt->duration){
1021
                int64_t old_diff= FFABS(st->cur_dts - pkt->duration - pkt->pts);
1022
                int64_t new_diff= FFABS(st->cur_dts - pkt->pts);
1023
                if(old_diff < new_diff && old_diff < (pkt->duration>>3)){
1024
                    pkt->pts += pkt->duration;
1025
    //                av_log(NULL, AV_LOG_DEBUG, "id:%d old:%"PRId64" new:%"PRId64" dur:%d cur:%"PRId64" size:%d\n", pkt->stream_index, old_diff, new_diff, pkt->duration, st->cur_dts, pkt->size);
1026
                }
1027
            }
1028

    
1029
            /* presentation is not delayed : PTS and DTS are the same */
1030
            if(pkt->pts == AV_NOPTS_VALUE)
1031
                pkt->pts = pkt->dts;
1032
            update_initial_timestamps(s, pkt->stream_index, pkt->pts, pkt->pts);
1033
            if(pkt->pts == AV_NOPTS_VALUE)
1034
                pkt->pts = st->cur_dts;
1035
            pkt->dts = pkt->pts;
1036
            if(pkt->pts != AV_NOPTS_VALUE)
1037
                st->cur_dts = pkt->pts + pkt->duration;
1038
        }
1039
    }
1040

    
1041
    if(pkt->pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY){
1042
        st->pts_buffer[0]= pkt->pts;
1043
        for(i=0; i<delay && st->pts_buffer[i] > st->pts_buffer[i+1]; i++)
1044
            FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i+1]);
1045
        if(pkt->dts == AV_NOPTS_VALUE)
1046
            pkt->dts= st->pts_buffer[0];
1047
        if(st->codec->codec_id == CODEC_ID_H264){ //we skiped it above so we try here
1048
            update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts); // this should happen on the first packet
1049
        }
1050
        if(pkt->dts > st->cur_dts)
1051
            st->cur_dts = pkt->dts;
1052
    }
1053

    
1054
//    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);
1055

    
1056
    /* update flags */
1057
    if(is_intra_only(st->codec))
1058
        pkt->flags |= AV_PKT_FLAG_KEY;
1059
    else if (pc) {
1060
        pkt->flags = 0;
1061
        /* keyframe computation */
1062
        if (pc->key_frame == 1)
1063
            pkt->flags |= AV_PKT_FLAG_KEY;
1064
        else if (pc->key_frame == -1 && pc->pict_type == FF_I_TYPE)
1065
            pkt->flags |= AV_PKT_FLAG_KEY;
1066
    }
1067
    if (pc)
1068
        pkt->convergence_duration = pc->convergence_duration;
1069
}
1070

    
1071

    
1072
static int av_read_frame_internal(AVFormatContext *s, AVPacket *pkt)
1073
{
1074
    AVStream *st;
1075
    int len, ret, i;
1076

    
1077
    av_init_packet(pkt);
1078

    
1079
    for(;;) {
1080
        /* select current input stream component */
1081
        st = s->cur_st;
1082
        if (st) {
1083
            if (!st->need_parsing || !st->parser) {
1084
                /* no parsing needed: we just output the packet as is */
1085
                /* raw data support */
1086
                *pkt = st->cur_pkt; st->cur_pkt.data= NULL;
1087
                compute_pkt_fields(s, st, NULL, pkt);
1088
                s->cur_st = NULL;
1089
                if ((s->iformat->flags & AVFMT_GENERIC_INDEX) &&
1090
                    (pkt->flags & AV_PKT_FLAG_KEY) && pkt->dts != AV_NOPTS_VALUE) {
1091
                    ff_reduce_index(s, st->index);
1092
                    av_add_index_entry(st, pkt->pos, pkt->dts, 0, 0, AVINDEX_KEYFRAME);
1093
                }
1094
                break;
1095
            } else if (st->cur_len > 0 && st->discard < AVDISCARD_ALL) {
1096
                len = av_parser_parse2(st->parser, st->codec, &pkt->data, &pkt->size,
1097
                                       st->cur_ptr, st->cur_len,
1098
                                       st->cur_pkt.pts, st->cur_pkt.dts,
1099
                                       st->cur_pkt.pos);
1100
                st->cur_pkt.pts = AV_NOPTS_VALUE;
1101
                st->cur_pkt.dts = AV_NOPTS_VALUE;
1102
                /* increment read pointer */
1103
                st->cur_ptr += len;
1104
                st->cur_len -= len;
1105

    
1106
                /* return packet if any */
1107
                if (pkt->size) {
1108
                got_packet:
1109
                    pkt->duration = 0;
1110
                    pkt->stream_index = st->index;
1111
                    pkt->pts = st->parser->pts;
1112
                    pkt->dts = st->parser->dts;
1113
                    pkt->pos = st->parser->pos;
1114
                    if(pkt->data == st->cur_pkt.data && pkt->size == st->cur_pkt.size){
1115
                        s->cur_st = NULL;
1116
                        pkt->destruct= st->cur_pkt.destruct;
1117
                        st->cur_pkt.destruct= NULL;
1118
                        st->cur_pkt.data    = NULL;
1119
                        assert(st->cur_len == 0);
1120
                    }else{
1121
                    pkt->destruct = NULL;
1122
                    }
1123
                    compute_pkt_fields(s, st, st->parser, pkt);
1124

    
1125
                    if((s->iformat->flags & AVFMT_GENERIC_INDEX) && pkt->flags & AV_PKT_FLAG_KEY){
1126
                        ff_reduce_index(s, st->index);
1127
                        av_add_index_entry(st, st->parser->frame_offset, pkt->dts,
1128
                                           0, 0, AVINDEX_KEYFRAME);
1129
                    }
1130

    
1131
                    break;
1132
                }
1133
            } else {
1134
                /* free packet */
1135
                av_free_packet(&st->cur_pkt);
1136
                s->cur_st = NULL;
1137
            }
1138
        } else {
1139
            AVPacket cur_pkt;
1140
            /* read next packet */
1141
            ret = av_read_packet(s, &cur_pkt);
1142
            if (ret < 0) {
1143
                if (ret == AVERROR(EAGAIN))
1144
                    return ret;
1145
                /* return the last frames, if any */
1146
                for(i = 0; i < s->nb_streams; i++) {
1147
                    st = s->streams[i];
1148
                    if (st->parser && st->need_parsing) {
1149
                        av_parser_parse2(st->parser, st->codec,
1150
                                        &pkt->data, &pkt->size,
1151
                                        NULL, 0,
1152
                                        AV_NOPTS_VALUE, AV_NOPTS_VALUE,
1153
                                        AV_NOPTS_VALUE);
1154
                        if (pkt->size)
1155
                            goto got_packet;
1156
                    }
1157
                }
1158
                /* no more packets: really terminate parsing */
1159
                return ret;
1160
            }
1161
            st = s->streams[cur_pkt.stream_index];
1162
            st->cur_pkt= cur_pkt;
1163

    
1164
            if(st->cur_pkt.pts != AV_NOPTS_VALUE &&
1165
               st->cur_pkt.dts != AV_NOPTS_VALUE &&
1166
               st->cur_pkt.pts < st->cur_pkt.dts){
1167
                av_log(s, AV_LOG_WARNING, "Invalid timestamps stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d\n",
1168
                    st->cur_pkt.stream_index,
1169
                    st->cur_pkt.pts,
1170
                    st->cur_pkt.dts,
1171
                    st->cur_pkt.size);
1172
//                av_free_packet(&st->cur_pkt);
1173
//                return -1;
1174
            }
1175

    
1176
            if(s->debug & FF_FDEBUG_TS)
1177
                av_log(s, AV_LOG_DEBUG, "av_read_packet stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d, duration=%d, flags=%d\n",
1178
                    st->cur_pkt.stream_index,
1179
                    st->cur_pkt.pts,
1180
                    st->cur_pkt.dts,
1181
                    st->cur_pkt.size,
1182
                    st->cur_pkt.duration,
1183
                    st->cur_pkt.flags);
1184

    
1185
            s->cur_st = st;
1186
            st->cur_ptr = st->cur_pkt.data;
1187
            st->cur_len = st->cur_pkt.size;
1188
            if (st->need_parsing && !st->parser && !(s->flags & AVFMT_FLAG_NOPARSE)) {
1189
                st->parser = av_parser_init(st->codec->codec_id);
1190
                if (!st->parser) {
1191
                    /* no parser available: just output the raw packets */
1192
                    st->need_parsing = AVSTREAM_PARSE_NONE;
1193
                }else if(st->need_parsing == AVSTREAM_PARSE_HEADERS){
1194
                    st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
1195
                }else if(st->need_parsing == AVSTREAM_PARSE_FULL_ONCE){
1196
                    st->parser->flags |= PARSER_FLAG_ONCE;
1197
                }
1198
            }
1199
        }
1200
    }
1201
    if(s->debug & FF_FDEBUG_TS)
1202
        av_log(s, AV_LOG_DEBUG, "av_read_frame_internal stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d, duration=%d, flags=%d\n",
1203
            pkt->stream_index,
1204
            pkt->pts,
1205
            pkt->dts,
1206
            pkt->size,
1207
            pkt->duration,
1208
            pkt->flags);
1209

    
1210
    return 0;
1211
}
1212

    
1213
int av_read_frame(AVFormatContext *s, AVPacket *pkt)
1214
{
1215
    AVPacketList *pktl;
1216
    int eof=0;
1217
    const int genpts= s->flags & AVFMT_FLAG_GENPTS;
1218

    
1219
    for(;;){
1220
        pktl = s->packet_buffer;
1221
        if (pktl) {
1222
            AVPacket *next_pkt= &pktl->pkt;
1223

    
1224
            if(genpts && next_pkt->dts != AV_NOPTS_VALUE){
1225
                int wrap_bits = s->streams[next_pkt->stream_index]->pts_wrap_bits;
1226
                while(pktl && next_pkt->pts == AV_NOPTS_VALUE){
1227
                    if(   pktl->pkt.stream_index == next_pkt->stream_index
1228
                       && (0 > av_compare_mod(next_pkt->dts, pktl->pkt.dts, 2LL << (wrap_bits - 1)))
1229
                       && av_compare_mod(pktl->pkt.pts, pktl->pkt.dts, 2LL << (wrap_bits - 1))) { //not b frame
1230
                        next_pkt->pts= pktl->pkt.dts;
1231
                    }
1232
                    pktl= pktl->next;
1233
                }
1234
                pktl = s->packet_buffer;
1235
            }
1236

    
1237
            if(   next_pkt->pts != AV_NOPTS_VALUE
1238
               || next_pkt->dts == AV_NOPTS_VALUE
1239
               || !genpts || eof){
1240
                /* read packet from packet buffer, if there is data */
1241
                *pkt = *next_pkt;
1242
                s->packet_buffer = pktl->next;
1243
                av_free(pktl);
1244
                return 0;
1245
            }
1246
        }
1247
        if(genpts){
1248
            int ret= av_read_frame_internal(s, pkt);
1249
            if(ret<0){
1250
                if(pktl && ret != AVERROR(EAGAIN)){
1251
                    eof=1;
1252
                    continue;
1253
                }else
1254
                    return ret;
1255
            }
1256

    
1257
            if(av_dup_packet(add_to_pktbuf(&s->packet_buffer, pkt,
1258
                                           &s->packet_buffer_end)) < 0)
1259
                return AVERROR(ENOMEM);
1260
        }else{
1261
            assert(!s->packet_buffer);
1262
            return av_read_frame_internal(s, pkt);
1263
        }
1264
    }
1265
}
1266

    
1267
/* XXX: suppress the packet queue */
1268
static void flush_packet_queue(AVFormatContext *s)
1269
{
1270
    AVPacketList *pktl;
1271

    
1272
    for(;;) {
1273
        pktl = s->packet_buffer;
1274
        if (!pktl)
1275
            break;
1276
        s->packet_buffer = pktl->next;
1277
        av_free_packet(&pktl->pkt);
1278
        av_free(pktl);
1279
    }
1280
    while(s->raw_packet_buffer){
1281
        pktl = s->raw_packet_buffer;
1282
        s->raw_packet_buffer = pktl->next;
1283
        av_free_packet(&pktl->pkt);
1284
        av_free(pktl);
1285
    }
1286
    s->packet_buffer_end=
1287
    s->raw_packet_buffer_end= NULL;
1288
    s->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
1289
}
1290

    
1291
/*******************************************************/
1292
/* seek support */
1293

    
1294
int av_find_default_stream_index(AVFormatContext *s)
1295
{
1296
    int first_audio_index = -1;
1297
    int i;
1298
    AVStream *st;
1299

    
1300
    if (s->nb_streams <= 0)
1301
        return -1;
1302
    for(i = 0; i < s->nb_streams; i++) {
1303
        st = s->streams[i];
1304
        if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
1305
            return i;
1306
        }
1307
        if (first_audio_index < 0 && st->codec->codec_type == AVMEDIA_TYPE_AUDIO)
1308
            first_audio_index = i;
1309
    }
1310
    return first_audio_index >= 0 ? first_audio_index : 0;
1311
}
1312

    
1313
/**
1314
 * Flush the frame reader.
1315
 */
1316
void ff_read_frame_flush(AVFormatContext *s)
1317
{
1318
    AVStream *st;
1319
    int i, j;
1320

    
1321
    flush_packet_queue(s);
1322

    
1323
    s->cur_st = NULL;
1324

    
1325
    /* for each stream, reset read state */
1326
    for(i = 0; i < s->nb_streams; i++) {
1327
        st = s->streams[i];
1328

    
1329
        if (st->parser) {
1330
            av_parser_close(st->parser);
1331
            st->parser = NULL;
1332
            av_free_packet(&st->cur_pkt);
1333
        }
1334
        st->last_IP_pts = AV_NOPTS_VALUE;
1335
        st->cur_dts = AV_NOPTS_VALUE; /* we set the current DTS to an unspecified origin */
1336
        st->reference_dts = AV_NOPTS_VALUE;
1337
        /* fail safe */
1338
        st->cur_ptr = NULL;
1339
        st->cur_len = 0;
1340

    
1341
        st->probe_packets = MAX_PROBE_PACKETS;
1342

    
1343
        for(j=0; j<MAX_REORDER_DELAY+1; j++)
1344
            st->pts_buffer[j]= AV_NOPTS_VALUE;
1345
    }
1346
}
1347

    
1348
void av_update_cur_dts(AVFormatContext *s, AVStream *ref_st, int64_t timestamp){
1349
    int i;
1350

    
1351
    for(i = 0; i < s->nb_streams; i++) {
1352
        AVStream *st = s->streams[i];
1353

    
1354
        st->cur_dts = av_rescale(timestamp,
1355
                                 st->time_base.den * (int64_t)ref_st->time_base.num,
1356
                                 st->time_base.num * (int64_t)ref_st->time_base.den);
1357
    }
1358
}
1359

    
1360
void ff_reduce_index(AVFormatContext *s, int stream_index)
1361
{
1362
    AVStream *st= s->streams[stream_index];
1363
    unsigned int max_entries= s->max_index_size / sizeof(AVIndexEntry);
1364

    
1365
    if((unsigned)st->nb_index_entries >= max_entries){
1366
        int i;
1367
        for(i=0; 2*i<st->nb_index_entries; i++)
1368
            st->index_entries[i]= st->index_entries[2*i];
1369
        st->nb_index_entries= i;
1370
    }
1371
}
1372

    
1373
int av_add_index_entry(AVStream *st,
1374
                            int64_t pos, int64_t timestamp, int size, int distance, int flags)
1375
{
1376
    AVIndexEntry *entries, *ie;
1377
    int index;
1378

    
1379
    if((unsigned)st->nb_index_entries + 1 >= UINT_MAX / sizeof(AVIndexEntry))
1380
        return -1;
1381

    
1382
    entries = av_fast_realloc(st->index_entries,
1383
                              &st->index_entries_allocated_size,
1384
                              (st->nb_index_entries + 1) *
1385
                              sizeof(AVIndexEntry));
1386
    if(!entries)
1387
        return -1;
1388

    
1389
    st->index_entries= entries;
1390

    
1391
    index= av_index_search_timestamp(st, timestamp, AVSEEK_FLAG_ANY);
1392

    
1393
    if(index<0){
1394
        index= st->nb_index_entries++;
1395
        ie= &entries[index];
1396
        assert(index==0 || ie[-1].timestamp < timestamp);
1397
    }else{
1398
        ie= &entries[index];
1399
        if(ie->timestamp != timestamp){
1400
            if(ie->timestamp <= timestamp)
1401
                return -1;
1402
            memmove(entries + index + 1, entries + index, sizeof(AVIndexEntry)*(st->nb_index_entries - index));
1403
            st->nb_index_entries++;
1404
        }else if(ie->pos == pos && distance < ie->min_distance) //do not reduce the distance
1405
            distance= ie->min_distance;
1406
    }
1407

    
1408
    ie->pos = pos;
1409
    ie->timestamp = timestamp;
1410
    ie->min_distance= distance;
1411
    ie->size= size;
1412
    ie->flags = flags;
1413

    
1414
    return index;
1415
}
1416

    
1417
int av_index_search_timestamp(AVStream *st, int64_t wanted_timestamp,
1418
                              int flags)
1419
{
1420
    AVIndexEntry *entries= st->index_entries;
1421
    int nb_entries= st->nb_index_entries;
1422
    int a, b, m;
1423
    int64_t timestamp;
1424

    
1425
    a = - 1;
1426
    b = nb_entries;
1427

    
1428
    //optimize appending index entries at the end
1429
    if(b && entries[b-1].timestamp < wanted_timestamp)
1430
        a= b-1;
1431

    
1432
    while (b - a > 1) {
1433
        m = (a + b) >> 1;
1434
        timestamp = entries[m].timestamp;
1435
        if(timestamp >= wanted_timestamp)
1436
            b = m;
1437
        if(timestamp <= wanted_timestamp)
1438
            a = m;
1439
    }
1440
    m= (flags & AVSEEK_FLAG_BACKWARD) ? a : b;
1441

    
1442
    if(!(flags & AVSEEK_FLAG_ANY)){
1443
        while(m>=0 && m<nb_entries && !(entries[m].flags & AVINDEX_KEYFRAME)){
1444
            m += (flags & AVSEEK_FLAG_BACKWARD) ? -1 : 1;
1445
        }
1446
    }
1447

    
1448
    if(m == nb_entries)
1449
        return -1;
1450
    return  m;
1451
}
1452

    
1453
#define DEBUG_SEEK
1454

    
1455
int av_seek_frame_binary(AVFormatContext *s, int stream_index, int64_t target_ts, int flags){
1456
    AVInputFormat *avif= s->iformat;
1457
    int64_t av_uninit(pos_min), av_uninit(pos_max), pos, pos_limit;
1458
    int64_t ts_min, ts_max, ts;
1459
    int index;
1460
    int64_t ret;
1461
    AVStream *st;
1462

    
1463
    if (stream_index < 0)
1464
        return -1;
1465

    
1466
#ifdef DEBUG_SEEK
1467
    av_log(s, AV_LOG_DEBUG, "read_seek: %d %"PRId64"\n", stream_index, target_ts);
1468
#endif
1469

    
1470
    ts_max=
1471
    ts_min= AV_NOPTS_VALUE;
1472
    pos_limit= -1; //gcc falsely says it may be uninitialized
1473

    
1474
    st= s->streams[stream_index];
1475
    if(st->index_entries){
1476
        AVIndexEntry *e;
1477

    
1478
        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()
1479
        index= FFMAX(index, 0);
1480
        e= &st->index_entries[index];
1481

    
1482
        if(e->timestamp <= target_ts || e->pos == e->min_distance){
1483
            pos_min= e->pos;
1484
            ts_min= e->timestamp;
1485
#ifdef DEBUG_SEEK
1486
            av_log(s, AV_LOG_DEBUG, "using cached pos_min=0x%"PRIx64" dts_min=%"PRId64"\n",
1487
                   pos_min,ts_min);
1488
#endif
1489
        }else{
1490
            assert(index==0);
1491
        }
1492

    
1493
        index= av_index_search_timestamp(st, target_ts, flags & ~AVSEEK_FLAG_BACKWARD);
1494
        assert(index < st->nb_index_entries);
1495
        if(index >= 0){
1496
            e= &st->index_entries[index];
1497
            assert(e->timestamp >= target_ts);
1498
            pos_max= e->pos;
1499
            ts_max= e->timestamp;
1500
            pos_limit= pos_max - e->min_distance;
1501
#ifdef DEBUG_SEEK
1502
            av_log(s, AV_LOG_DEBUG, "using cached pos_max=0x%"PRIx64" pos_limit=0x%"PRIx64" dts_max=%"PRId64"\n",
1503
                   pos_max,pos_limit, ts_max);
1504
#endif
1505
        }
1506
    }
1507

    
1508
    pos= av_gen_search(s, stream_index, target_ts, pos_min, pos_max, pos_limit, ts_min, ts_max, flags, &ts, avif->read_timestamp);
1509
    if(pos<0)
1510
        return -1;
1511

    
1512
    /* do the seek */
1513
    if ((ret = url_fseek(s->pb, pos, SEEK_SET)) < 0)
1514
        return ret;
1515

    
1516
    av_update_cur_dts(s, st, ts);
1517

    
1518
    return 0;
1519
}
1520

    
1521
int64_t av_gen_search(AVFormatContext *s, int stream_index, int64_t target_ts, int64_t pos_min, int64_t pos_max, int64_t pos_limit, int64_t ts_min, int64_t ts_max, int flags, int64_t *ts_ret, int64_t (*read_timestamp)(struct AVFormatContext *, int , int64_t *, int64_t )){
1522
    int64_t pos, ts;
1523
    int64_t start_pos, filesize;
1524
    int no_change;
1525

    
1526
#ifdef DEBUG_SEEK
1527
    av_log(s, AV_LOG_DEBUG, "gen_seek: %d %"PRId64"\n", stream_index, target_ts);
1528
#endif
1529

    
1530
    if(ts_min == AV_NOPTS_VALUE){
1531
        pos_min = s->data_offset;
1532
        ts_min = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1533
        if (ts_min == AV_NOPTS_VALUE)
1534
            return -1;
1535
    }
1536

    
1537
    if(ts_max == AV_NOPTS_VALUE){
1538
        int step= 1024;
1539
        filesize = url_fsize(s->pb);
1540
        pos_max = filesize - 1;
1541
        do{
1542
            pos_max -= step;
1543
            ts_max = read_timestamp(s, stream_index, &pos_max, pos_max + step);
1544
            step += step;
1545
        }while(ts_max == AV_NOPTS_VALUE && pos_max >= step);
1546
        if (ts_max == AV_NOPTS_VALUE)
1547
            return -1;
1548

    
1549
        for(;;){
1550
            int64_t tmp_pos= pos_max + 1;
1551
            int64_t tmp_ts= read_timestamp(s, stream_index, &tmp_pos, INT64_MAX);
1552
            if(tmp_ts == AV_NOPTS_VALUE)
1553
                break;
1554
            ts_max= tmp_ts;
1555
            pos_max= tmp_pos;
1556
            if(tmp_pos >= filesize)
1557
                break;
1558
        }
1559
        pos_limit= pos_max;
1560
    }
1561

    
1562
    if(ts_min > ts_max){
1563
        return -1;
1564
    }else if(ts_min == ts_max){
1565
        pos_limit= pos_min;
1566
    }
1567

    
1568
    no_change=0;
1569
    while (pos_min < pos_limit) {
1570
#ifdef DEBUG_SEEK
1571
        av_log(s, AV_LOG_DEBUG, "pos_min=0x%"PRIx64" pos_max=0x%"PRIx64" dts_min=%"PRId64" dts_max=%"PRId64"\n",
1572
               pos_min, pos_max,
1573
               ts_min, ts_max);
1574
#endif
1575
        assert(pos_limit <= pos_max);
1576

    
1577
        if(no_change==0){
1578
            int64_t approximate_keyframe_distance= pos_max - pos_limit;
1579
            // interpolate position (better than dichotomy)
1580
            pos = av_rescale(target_ts - ts_min, pos_max - pos_min, ts_max - ts_min)
1581
                + pos_min - approximate_keyframe_distance;
1582
        }else if(no_change==1){
1583
            // bisection, if interpolation failed to change min or max pos last time
1584
            pos = (pos_min + pos_limit)>>1;
1585
        }else{
1586
            /* linear search if bisection failed, can only happen if there
1587
               are very few or no keyframes between min/max */
1588
            pos=pos_min;
1589
        }
1590
        if(pos <= pos_min)
1591
            pos= pos_min + 1;
1592
        else if(pos > pos_limit)
1593
            pos= pos_limit;
1594
        start_pos= pos;
1595

    
1596
        ts = read_timestamp(s, stream_index, &pos, INT64_MAX); //may pass pos_limit instead of -1
1597
        if(pos == pos_max)
1598
            no_change++;
1599
        else
1600
            no_change=0;
1601
#ifdef DEBUG_SEEK
1602
        av_log(s, AV_LOG_DEBUG, "%"PRId64" %"PRId64" %"PRId64" / %"PRId64" %"PRId64" %"PRId64" target:%"PRId64" limit:%"PRId64" start:%"PRId64" noc:%d\n",
1603
               pos_min, pos, pos_max, ts_min, ts, ts_max, target_ts, pos_limit,
1604
               start_pos, no_change);
1605
#endif
1606
        if(ts == AV_NOPTS_VALUE){
1607
            av_log(s, AV_LOG_ERROR, "read_timestamp() failed in the middle\n");
1608
            return -1;
1609
        }
1610
        assert(ts != AV_NOPTS_VALUE);
1611
        if (target_ts <= ts) {
1612
            pos_limit = start_pos - 1;
1613
            pos_max = pos;
1614
            ts_max = ts;
1615
        }
1616
        if (target_ts >= ts) {
1617
            pos_min = pos;
1618
            ts_min = ts;
1619
        }
1620
    }
1621

    
1622
    pos = (flags & AVSEEK_FLAG_BACKWARD) ? pos_min : pos_max;
1623
    ts  = (flags & AVSEEK_FLAG_BACKWARD) ?  ts_min :  ts_max;
1624
#ifdef DEBUG_SEEK
1625
    pos_min = pos;
1626
    ts_min = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1627
    pos_min++;
1628
    ts_max = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1629
    av_log(s, AV_LOG_DEBUG, "pos=0x%"PRIx64" %"PRId64"<=%"PRId64"<=%"PRId64"\n",
1630
           pos, ts_min, target_ts, ts_max);
1631
#endif
1632
    *ts_ret= ts;
1633
    return pos;
1634
}
1635

    
1636
static int av_seek_frame_byte(AVFormatContext *s, int stream_index, int64_t pos, int flags){
1637
    int64_t pos_min, pos_max;
1638
#if 0
1639
    AVStream *st;
1640

1641
    if (stream_index < 0)
1642
        return -1;
1643

1644
    st= s->streams[stream_index];
1645
#endif
1646

    
1647
    pos_min = s->data_offset;
1648
    pos_max = url_fsize(s->pb) - 1;
1649

    
1650
    if     (pos < pos_min) pos= pos_min;
1651
    else if(pos > pos_max) pos= pos_max;
1652

    
1653
    url_fseek(s->pb, pos, SEEK_SET);
1654

    
1655
#if 0
1656
    av_update_cur_dts(s, st, ts);
1657
#endif
1658
    return 0;
1659
}
1660

    
1661
static int av_seek_frame_generic(AVFormatContext *s,
1662
                                 int stream_index, int64_t timestamp, int flags)
1663
{
1664
    int index;
1665
    int64_t ret;
1666
    AVStream *st;
1667
    AVIndexEntry *ie;
1668

    
1669
    st = s->streams[stream_index];
1670

    
1671
    index = av_index_search_timestamp(st, timestamp, flags);
1672

    
1673
    if(index < 0 && st->nb_index_entries && timestamp < st->index_entries[0].timestamp)
1674
        return -1;
1675

    
1676
    if(index < 0 || index==st->nb_index_entries-1){
1677
        int i;
1678
        AVPacket pkt;
1679

    
1680
        if(st->nb_index_entries){
1681
            assert(st->index_entries);
1682
            ie= &st->index_entries[st->nb_index_entries-1];
1683
            if ((ret = url_fseek(s->pb, ie->pos, SEEK_SET)) < 0)
1684
                return ret;
1685
            av_update_cur_dts(s, st, ie->timestamp);
1686
        }else{
1687
            if ((ret = url_fseek(s->pb, s->data_offset, SEEK_SET)) < 0)
1688
                return ret;
1689
        }
1690
        for(i=0;; i++) {
1691
            int ret;
1692
            do{
1693
                ret = av_read_frame(s, &pkt);
1694
            }while(ret == AVERROR(EAGAIN));
1695
            if(ret<0)
1696
                break;
1697
            av_free_packet(&pkt);
1698
            if(stream_index == pkt.stream_index){
1699
                if((pkt.flags & AV_PKT_FLAG_KEY) && pkt.dts > timestamp)
1700
                    break;
1701
            }
1702
        }
1703
        index = av_index_search_timestamp(st, timestamp, flags);
1704
    }
1705
    if (index < 0)
1706
        return -1;
1707

    
1708
    ff_read_frame_flush(s);
1709
    if (s->iformat->read_seek){
1710
        if(s->iformat->read_seek(s, stream_index, timestamp, flags) >= 0)
1711
            return 0;
1712
    }
1713
    ie = &st->index_entries[index];
1714
    if ((ret = url_fseek(s->pb, ie->pos, SEEK_SET)) < 0)
1715
        return ret;
1716
    av_update_cur_dts(s, st, ie->timestamp);
1717

    
1718
    return 0;
1719
}
1720

    
1721
int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
1722
{
1723
    int ret;
1724
    AVStream *st;
1725

    
1726
    ff_read_frame_flush(s);
1727

    
1728
    if(flags & AVSEEK_FLAG_BYTE)
1729
        return av_seek_frame_byte(s, stream_index, timestamp, flags);
1730

    
1731
    if(stream_index < 0){
1732
        stream_index= av_find_default_stream_index(s);
1733
        if(stream_index < 0)
1734
            return -1;
1735

    
1736
        st= s->streams[stream_index];
1737
       /* timestamp for default must be expressed in AV_TIME_BASE units */
1738
        timestamp = av_rescale(timestamp, st->time_base.den, AV_TIME_BASE * (int64_t)st->time_base.num);
1739
    }
1740

    
1741
    /* first, we try the format specific seek */
1742
    if (s->iformat->read_seek)
1743
        ret = s->iformat->read_seek(s, stream_index, timestamp, flags);
1744
    else
1745
        ret = -1;
1746
    if (ret >= 0) {
1747
        return 0;
1748
    }
1749

    
1750
    if(s->iformat->read_timestamp)
1751
        return av_seek_frame_binary(s, stream_index, timestamp, flags);
1752
    else
1753
        return av_seek_frame_generic(s, stream_index, timestamp, flags);
1754
}
1755

    
1756
int avformat_seek_file(AVFormatContext *s, int stream_index, int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
1757
{
1758
    if(min_ts > ts || max_ts < ts)
1759
        return -1;
1760

    
1761
    ff_read_frame_flush(s);
1762

    
1763
    if (s->iformat->read_seek2)
1764
        return s->iformat->read_seek2(s, stream_index, min_ts, ts, max_ts, flags);
1765

    
1766
    if(s->iformat->read_timestamp){
1767
        //try to seek via read_timestamp()
1768
    }
1769

    
1770
    //Fallback to old API if new is not implemented but old is
1771
    //Note the old has somewat different sematics
1772
    if(s->iformat->read_seek || 1)
1773
        return av_seek_frame(s, stream_index, ts, flags | (ts - min_ts > (uint64_t)(max_ts - ts) ? AVSEEK_FLAG_BACKWARD : 0));
1774

    
1775
    // try some generic seek like av_seek_frame_generic() but with new ts semantics
1776
}
1777

    
1778
/*******************************************************/
1779

    
1780
/**
1781
 * Return TRUE if the stream has accurate duration in any stream.
1782
 *
1783
 * @return TRUE if the stream has accurate duration for at least one component.
1784
 */
1785
static int av_has_duration(AVFormatContext *ic)
1786
{
1787
    int i;
1788
    AVStream *st;
1789

    
1790
    for(i = 0;i < ic->nb_streams; i++) {
1791
        st = ic->streams[i];
1792
        if (st->duration != AV_NOPTS_VALUE)
1793
            return 1;
1794
    }
1795
    return 0;
1796
}
1797

    
1798
/**
1799
 * Estimate the stream timings from the one of each components.
1800
 *
1801
 * Also computes the global bitrate if possible.
1802
 */
1803
static void av_update_stream_timings(AVFormatContext *ic)
1804
{
1805
    int64_t start_time, start_time1, end_time, end_time1;
1806
    int64_t duration, duration1;
1807
    int i;
1808
    AVStream *st;
1809

    
1810
    start_time = INT64_MAX;
1811
    end_time = INT64_MIN;
1812
    duration = INT64_MIN;
1813
    for(i = 0;i < ic->nb_streams; i++) {
1814
        st = ic->streams[i];
1815
        if (st->start_time != AV_NOPTS_VALUE && st->time_base.den) {
1816
            start_time1= av_rescale_q(st->start_time, st->time_base, AV_TIME_BASE_Q);
1817
            if (start_time1 < start_time)
1818
                start_time = start_time1;
1819
            if (st->duration != AV_NOPTS_VALUE) {
1820
                end_time1 = start_time1
1821
                          + av_rescale_q(st->duration, st->time_base, AV_TIME_BASE_Q);
1822
                if (end_time1 > end_time)
1823
                    end_time = end_time1;
1824
            }
1825
        }
1826
        if (st->duration != AV_NOPTS_VALUE) {
1827
            duration1 = av_rescale_q(st->duration, st->time_base, AV_TIME_BASE_Q);
1828
            if (duration1 > duration)
1829
                duration = duration1;
1830
        }
1831
    }
1832
    if (start_time != INT64_MAX) {
1833
        ic->start_time = start_time;
1834
        if (end_time != INT64_MIN) {
1835
            if (end_time - start_time > duration)
1836
                duration = end_time - start_time;
1837
        }
1838
    }
1839
    if (duration != INT64_MIN) {
1840
        ic->duration = duration;
1841
        if (ic->file_size > 0) {
1842
            /* compute the bitrate */
1843
            ic->bit_rate = (double)ic->file_size * 8.0 * AV_TIME_BASE /
1844
                (double)ic->duration;
1845
        }
1846
    }
1847
}
1848

    
1849
static void fill_all_stream_timings(AVFormatContext *ic)
1850
{
1851
    int i;
1852
    AVStream *st;
1853

    
1854
    av_update_stream_timings(ic);
1855
    for(i = 0;i < ic->nb_streams; i++) {
1856
        st = ic->streams[i];
1857
        if (st->start_time == AV_NOPTS_VALUE) {
1858
            if(ic->start_time != AV_NOPTS_VALUE)
1859
                st->start_time = av_rescale_q(ic->start_time, AV_TIME_BASE_Q, st->time_base);
1860
            if(ic->duration != AV_NOPTS_VALUE)
1861
                st->duration = av_rescale_q(ic->duration, AV_TIME_BASE_Q, st->time_base);
1862
        }
1863
    }
1864
}
1865

    
1866
static void av_estimate_timings_from_bit_rate(AVFormatContext *ic)
1867
{
1868
    int64_t filesize, duration;
1869
    int bit_rate, i;
1870
    AVStream *st;
1871

    
1872
    /* if bit_rate is already set, we believe it */
1873
    if (ic->bit_rate == 0) {
1874
        bit_rate = 0;
1875
        for(i=0;i<ic->nb_streams;i++) {
1876
            st = ic->streams[i];
1877
            bit_rate += st->codec->bit_rate;
1878
        }
1879
        ic->bit_rate = bit_rate;
1880
    }
1881

    
1882
    /* if duration is already set, we believe it */
1883
    if (ic->duration == AV_NOPTS_VALUE &&
1884
        ic->bit_rate != 0 &&
1885
        ic->file_size != 0)  {
1886
        filesize = ic->file_size;
1887
        if (filesize > 0) {
1888
            for(i = 0; i < ic->nb_streams; i++) {
1889
                st = ic->streams[i];
1890
                duration= av_rescale(8*filesize, st->time_base.den, ic->bit_rate*(int64_t)st->time_base.num);
1891
                if (st->duration == AV_NOPTS_VALUE)
1892
                    st->duration = duration;
1893
            }
1894
        }
1895
    }
1896
}
1897

    
1898
#define DURATION_MAX_READ_SIZE 250000
1899
#define DURATION_MAX_RETRY 3
1900

    
1901
/* only usable for MPEG-PS streams */
1902
static void av_estimate_timings_from_pts(AVFormatContext *ic, int64_t old_offset)
1903
{
1904
    AVPacket pkt1, *pkt = &pkt1;
1905
    AVStream *st;
1906
    int read_size, i, ret;
1907
    int64_t end_time;
1908
    int64_t filesize, offset, duration;
1909
    int retry=0;
1910

    
1911
    ic->cur_st = NULL;
1912

    
1913
    /* flush packet queue */
1914
    flush_packet_queue(ic);
1915

    
1916
    for (i=0; i<ic->nb_streams; i++) {
1917
        st = ic->streams[i];
1918
        if (st->start_time == AV_NOPTS_VALUE && st->first_dts == AV_NOPTS_VALUE)
1919
            av_log(st->codec, AV_LOG_WARNING, "start time is not set in av_estimate_timings_from_pts\n");
1920

    
1921
        if (st->parser) {
1922
            av_parser_close(st->parser);
1923
            st->parser= NULL;
1924
            av_free_packet(&st->cur_pkt);
1925
        }
1926
    }
1927

    
1928
    /* estimate the end time (duration) */
1929
    /* XXX: may need to support wrapping */
1930
    filesize = ic->file_size;
1931
    end_time = AV_NOPTS_VALUE;
1932
    do{
1933
    offset = filesize - (DURATION_MAX_READ_SIZE<<retry);
1934
    if (offset < 0)
1935
        offset = 0;
1936

    
1937
    url_fseek(ic->pb, offset, SEEK_SET);
1938
    read_size = 0;
1939
    for(;;) {
1940
        if (read_size >= DURATION_MAX_READ_SIZE<<(FFMAX(retry-1,0)))
1941
            break;
1942

    
1943
        do{
1944
            ret = av_read_packet(ic, pkt);
1945
        }while(ret == AVERROR(EAGAIN));
1946
        if (ret != 0)
1947
            break;
1948
        read_size += pkt->size;
1949
        st = ic->streams[pkt->stream_index];
1950
        if (pkt->pts != AV_NOPTS_VALUE &&
1951
            (st->start_time != AV_NOPTS_VALUE ||
1952
             st->first_dts  != AV_NOPTS_VALUE)) {
1953
            duration = end_time = pkt->pts;
1954
            if (st->start_time != AV_NOPTS_VALUE)  duration -= st->start_time;
1955
            else                                   duration -= st->first_dts;
1956
            if (duration < 0)
1957
                duration += 1LL<<st->pts_wrap_bits;
1958
            if (duration > 0) {
1959
                if (st->duration == AV_NOPTS_VALUE ||
1960
                    st->duration < duration)
1961
                    st->duration = duration;
1962
            }
1963
        }
1964
        av_free_packet(pkt);
1965
    }
1966
    }while(   end_time==AV_NOPTS_VALUE
1967
           && filesize > (DURATION_MAX_READ_SIZE<<retry)
1968
           && ++retry <= DURATION_MAX_RETRY);
1969

    
1970
    fill_all_stream_timings(ic);
1971

    
1972
    url_fseek(ic->pb, old_offset, SEEK_SET);
1973
    for (i=0; i<ic->nb_streams; i++) {
1974
        st= ic->streams[i];
1975
        st->cur_dts= st->first_dts;
1976
        st->last_IP_pts = AV_NOPTS_VALUE;
1977
    }
1978
}
1979

    
1980
static void av_estimate_timings(AVFormatContext *ic, int64_t old_offset)
1981
{
1982
    int64_t file_size;
1983

    
1984
    /* get the file size, if possible */
1985
    if (ic->iformat->flags & AVFMT_NOFILE) {
1986
        file_size = 0;
1987
    } else {
1988
        file_size = url_fsize(ic->pb);
1989
        if (file_size < 0)
1990
            file_size = 0;
1991
    }
1992
    ic->file_size = file_size;
1993

    
1994
    if ((!strcmp(ic->iformat->name, "mpeg") ||
1995
         !strcmp(ic->iformat->name, "mpegts")) &&
1996
        file_size && !url_is_streamed(ic->pb)) {
1997
        /* get accurate estimate from the PTSes */
1998
        av_estimate_timings_from_pts(ic, old_offset);
1999
    } else if (av_has_duration(ic)) {
2000
        /* at least one component has timings - we use them for all
2001
           the components */
2002
        fill_all_stream_timings(ic);
2003
    } else {
2004
        av_log(ic, AV_LOG_WARNING, "Estimating duration from bitrate, this may be inaccurate\n");
2005
        /* less precise: use bitrate info */
2006
        av_estimate_timings_from_bit_rate(ic);
2007
    }
2008
    av_update_stream_timings(ic);
2009

    
2010
#if 0
2011
    {
2012
        int i;
2013
        AVStream *st;
2014
        for(i = 0;i < ic->nb_streams; i++) {
2015
            st = ic->streams[i];
2016
        printf("%d: start_time: %0.3f duration: %0.3f\n",
2017
               i, (double)st->start_time / AV_TIME_BASE,
2018
               (double)st->duration / AV_TIME_BASE);
2019
        }
2020
        printf("stream: start_time: %0.3f duration: %0.3f bitrate=%d kb/s\n",
2021
               (double)ic->start_time / AV_TIME_BASE,
2022
               (double)ic->duration / AV_TIME_BASE,
2023
               ic->bit_rate / 1000);
2024
    }
2025
#endif
2026
}
2027

    
2028
static int has_codec_parameters(AVCodecContext *enc)
2029
{
2030
    int val;
2031
    switch(enc->codec_type) {
2032
    case AVMEDIA_TYPE_AUDIO:
2033
        val = enc->sample_rate && enc->channels && enc->sample_fmt != AV_SAMPLE_FMT_NONE;
2034
        if(!enc->frame_size &&
2035
           (enc->codec_id == CODEC_ID_VORBIS ||
2036
            enc->codec_id == CODEC_ID_AAC ||
2037
            enc->codec_id == CODEC_ID_MP1 ||
2038
            enc->codec_id == CODEC_ID_MP2 ||
2039
            enc->codec_id == CODEC_ID_MP3 ||
2040
            enc->codec_id == CODEC_ID_SPEEX))
2041
            return 0;
2042
        break;
2043
    case AVMEDIA_TYPE_VIDEO:
2044
        val = enc->width && enc->pix_fmt != PIX_FMT_NONE;
2045
        break;
2046
    default:
2047
        val = 1;
2048
        break;
2049
    }
2050
    return enc->codec_id != CODEC_ID_NONE && val != 0;
2051
}
2052

    
2053
static int has_decode_delay_been_guessed(AVStream *st)
2054
{
2055
    return st->codec->codec_id != CODEC_ID_H264 ||
2056
        st->codec_info_nb_frames >= 6 + st->codec->has_b_frames;
2057
}
2058

    
2059
static int try_decode_frame(AVStream *st, AVPacket *avpkt)
2060
{
2061
    int16_t *samples;
2062
    AVCodec *codec;
2063
    int got_picture, data_size, ret=0;
2064
    AVFrame picture;
2065

    
2066
    if(!st->codec->codec){
2067
        codec = avcodec_find_decoder(st->codec->codec_id);
2068
        if (!codec)
2069
            return -1;
2070
        ret = avcodec_open(st->codec, codec);
2071
        if (ret < 0)
2072
            return ret;
2073
    }
2074

    
2075
    if(!has_codec_parameters(st->codec) || !has_decode_delay_been_guessed(st)){
2076
        switch(st->codec->codec_type) {
2077
        case AVMEDIA_TYPE_VIDEO:
2078
            avcodec_get_frame_defaults(&picture);
2079
            ret = avcodec_decode_video2(st->codec, &picture,
2080
                                        &got_picture, avpkt);
2081
            break;
2082
        case AVMEDIA_TYPE_AUDIO:
2083
            data_size = FFMAX(avpkt->size, AVCODEC_MAX_AUDIO_FRAME_SIZE);
2084
            samples = av_malloc(data_size);
2085
            if (!samples)
2086
                goto fail;
2087
            ret = avcodec_decode_audio3(st->codec, samples,
2088
                                        &data_size, avpkt);
2089
            av_free(samples);
2090
            break;
2091
        default:
2092
            break;
2093
        }
2094
    }
2095
 fail:
2096
    return ret;
2097
}
2098

    
2099
unsigned int ff_codec_get_tag(const AVCodecTag *tags, enum CodecID id)
2100
{
2101
    while (tags->id != CODEC_ID_NONE) {
2102
        if (tags->id == id)
2103
            return tags->tag;
2104
        tags++;
2105
    }
2106
    return 0;
2107
}
2108

    
2109
enum CodecID ff_codec_get_id(const AVCodecTag *tags, unsigned int tag)
2110
{
2111
    int i;
2112
    for(i=0; tags[i].id != CODEC_ID_NONE;i++) {
2113
        if(tag == tags[i].tag)
2114
            return tags[i].id;
2115
    }
2116
    for(i=0; tags[i].id != CODEC_ID_NONE; i++) {
2117
        if (ff_toupper4(tag) == ff_toupper4(tags[i].tag))
2118
            return tags[i].id;
2119
    }
2120
    return CODEC_ID_NONE;
2121
}
2122

    
2123
unsigned int av_codec_get_tag(const AVCodecTag * const *tags, enum CodecID id)
2124
{
2125
    int i;
2126
    for(i=0; tags && tags[i]; i++){
2127
        int tag= ff_codec_get_tag(tags[i], id);
2128
        if(tag) return tag;
2129
    }
2130
    return 0;
2131
}
2132

    
2133
enum CodecID av_codec_get_id(const AVCodecTag * const *tags, unsigned int tag)
2134
{
2135
    int i;
2136
    for(i=0; tags && tags[i]; i++){
2137
        enum CodecID id= ff_codec_get_id(tags[i], tag);
2138
        if(id!=CODEC_ID_NONE) return id;
2139
    }
2140
    return CODEC_ID_NONE;
2141
}
2142

    
2143
static void compute_chapters_end(AVFormatContext *s)
2144
{
2145
    unsigned int i;
2146

    
2147
    for (i=0; i+1<s->nb_chapters; i++)
2148
        if (s->chapters[i]->end == AV_NOPTS_VALUE) {
2149
            assert(s->chapters[i]->start <= s->chapters[i+1]->start);
2150
            assert(!av_cmp_q(s->chapters[i]->time_base, s->chapters[i+1]->time_base));
2151
            s->chapters[i]->end = s->chapters[i+1]->start;
2152
        }
2153

    
2154
    if (s->nb_chapters && s->chapters[i]->end == AV_NOPTS_VALUE) {
2155
        assert(s->start_time != AV_NOPTS_VALUE);
2156
        assert(s->duration > 0);
2157
        s->chapters[i]->end = av_rescale_q(s->start_time + s->duration,
2158
                                           AV_TIME_BASE_Q,
2159
                                           s->chapters[i]->time_base);
2160
    }
2161
}
2162

    
2163
static int get_std_framerate(int i){
2164
    if(i<60*12) return i*1001;
2165
    else        return ((const int[]){24,30,60,12,15})[i-60*12]*1000*12;
2166
}
2167

    
2168
/*
2169
 * Is the time base unreliable.
2170
 * This is a heuristic to balance between quick acceptance of the values in
2171
 * the headers vs. some extra checks.
2172
 * Old DivX and Xvid often have nonsense timebases like 1fps or 2fps.
2173
 * MPEG-2 commonly misuses field repeat flags to store different framerates.
2174
 * And there are "variable" fps files this needs to detect as well.
2175
 */
2176
static int tb_unreliable(AVCodecContext *c){
2177
    if(   c->time_base.den >= 101L*c->time_base.num
2178
       || c->time_base.den <    5L*c->time_base.num
2179
/*       || c->codec_tag == AV_RL32("DIVX")
2180
       || c->codec_tag == AV_RL32("XVID")*/
2181
       || c->codec_id == CODEC_ID_MPEG2VIDEO
2182
       || c->codec_id == CODEC_ID_H264
2183
       )
2184
        return 1;
2185
    return 0;
2186
}
2187

    
2188
int av_find_stream_info(AVFormatContext *ic)
2189
{
2190
    int i, count, ret, read_size, j;
2191
    AVStream *st;
2192
    AVPacket pkt1, *pkt;
2193
    int64_t old_offset = url_ftell(ic->pb);
2194

    
2195
    for(i=0;i<ic->nb_streams;i++) {
2196
        AVCodec *codec;
2197
        st = ic->streams[i];
2198
        if (st->codec->codec_id == CODEC_ID_AAC) {
2199
            st->codec->sample_rate = 0;
2200
            st->codec->frame_size = 0;
2201
            st->codec->channels = 0;
2202
        }
2203
        if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO ||
2204
            st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE) {
2205
/*            if(!st->time_base.num)
2206
                st->time_base= */
2207
            if(!st->codec->time_base.num)
2208
                st->codec->time_base= st->time_base;
2209
        }
2210
        //only for the split stuff
2211
        if (!st->parser && !(ic->flags & AVFMT_FLAG_NOPARSE)) {
2212
            st->parser = av_parser_init(st->codec->codec_id);
2213
            if(st->need_parsing == AVSTREAM_PARSE_HEADERS && st->parser){
2214
                st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
2215
            }
2216
        }
2217
        assert(!st->codec->codec);
2218
        codec = avcodec_find_decoder(st->codec->codec_id);
2219

    
2220
        /* Force decoding of at least one frame of codec data
2221
         * this makes sure the codec initializes the channel configuration
2222
         * and does not trust the values from the container.
2223
         */
2224
        if (codec && codec->capabilities & CODEC_CAP_CHANNEL_CONF)
2225
            st->codec->channels = 0;
2226

    
2227
        /* Ensure that subtitle_header is properly set. */
2228
        if (st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE
2229
            && codec && !st->codec->codec)
2230
            avcodec_open(st->codec, codec);
2231

    
2232
        //try to just open decoders, in case this is enough to get parameters
2233
        if(!has_codec_parameters(st->codec)){
2234
            if (codec && !st->codec->codec)
2235
                avcodec_open(st->codec, codec);
2236
        }
2237
    }
2238

    
2239
    for (i=0; i<ic->nb_streams; i++) {
2240
        ic->streams[i]->info->last_dts = AV_NOPTS_VALUE;
2241
    }
2242

    
2243
    count = 0;
2244
    read_size = 0;
2245
    for(;;) {
2246
        if(url_interrupt_cb()){
2247
            ret= AVERROR(EINTR);
2248
            av_log(ic, AV_LOG_DEBUG, "interrupted\n");
2249
            break;
2250
        }
2251

    
2252
        /* check if one codec still needs to be handled */
2253
        for(i=0;i<ic->nb_streams;i++) {
2254
            st = ic->streams[i];
2255
            if (!has_codec_parameters(st->codec))
2256
                break;
2257
            /* variable fps and no guess at the real fps */
2258
            if(   tb_unreliable(st->codec) && !(st->r_frame_rate.num && st->avg_frame_rate.num)
2259
               && st->info->duration_count<20 && st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
2260
                break;
2261
            if(st->parser && st->parser->parser->split && !st->codec->extradata)
2262
                break;
2263
            if(st->first_dts == AV_NOPTS_VALUE)
2264
                break;
2265
        }
2266
        if (i == ic->nb_streams) {
2267
            /* NOTE: if the format has no header, then we need to read
2268
               some packets to get most of the streams, so we cannot
2269
               stop here */
2270
            if (!(ic->ctx_flags & AVFMTCTX_NOHEADER)) {
2271
                /* if we found the info for all the codecs, we can stop */
2272
                ret = count;
2273
                av_log(ic, AV_LOG_DEBUG, "All info found\n");
2274
                break;
2275
            }
2276
        }
2277
        /* we did not get all the codec info, but we read too much data */
2278
        if (read_size >= ic->probesize) {
2279
            ret = count;
2280
            av_log(ic, AV_LOG_DEBUG, "Probe buffer size limit %d reached\n", ic->probesize);
2281
            break;
2282
        }
2283

    
2284
        /* NOTE: a new stream can be added there if no header in file
2285
           (AVFMTCTX_NOHEADER) */
2286
        ret = av_read_frame_internal(ic, &pkt1);
2287
        if (ret < 0 && ret != AVERROR(EAGAIN)) {
2288
            /* EOF or error */
2289
            ret = -1; /* we could not have all the codec parameters before EOF */
2290
            for(i=0;i<ic->nb_streams;i++) {
2291
                st = ic->streams[i];
2292
                if (!has_codec_parameters(st->codec)){
2293
                    char buf[256];
2294
                    avcodec_string(buf, sizeof(buf), st->codec, 0);
2295
                    av_log(ic, AV_LOG_WARNING, "Could not find codec parameters (%s)\n", buf);
2296
                } else {
2297
                    ret = 0;
2298
                }
2299
            }
2300
            break;
2301
        }
2302

    
2303
        if (ret == AVERROR(EAGAIN))
2304
            continue;
2305

    
2306
        pkt= add_to_pktbuf(&ic->packet_buffer, &pkt1, &ic->packet_buffer_end);
2307
        if ((ret = av_dup_packet(pkt)) < 0)
2308
            goto find_stream_info_err;
2309

    
2310
        read_size += pkt->size;
2311

    
2312
        st = ic->streams[pkt->stream_index];
2313
        if (st->codec_info_nb_frames>1) {
2314
            if (st->time_base.den > 0 && av_rescale_q(st->info->codec_info_duration, st->time_base, AV_TIME_BASE_Q) >= ic->max_analyze_duration) {
2315
                av_log(ic, AV_LOG_WARNING, "max_analyze_duration reached\n");
2316
                break;
2317
            }
2318
            st->info->codec_info_duration += pkt->duration;
2319
        }
2320
        {
2321
            int64_t last = st->info->last_dts;
2322
            int64_t duration= pkt->dts - last;
2323

    
2324
            if(pkt->dts != AV_NOPTS_VALUE && last != AV_NOPTS_VALUE && duration>0){
2325
                double dur= duration * av_q2d(st->time_base);
2326

    
2327
//                if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
2328
//                    av_log(NULL, AV_LOG_ERROR, "%f\n", dur);
2329
                if (st->info->duration_count < 2)
2330
                    memset(st->info->duration_error, 0, sizeof(st->info->duration_error));
2331
                for (i=1; i<FF_ARRAY_ELEMS(st->info->duration_error); i++) {
2332
                    int framerate= get_std_framerate(i);
2333
                    int ticks= lrintf(dur*framerate/(1001*12));
2334
                    double error= dur - ticks*1001*12/(double)framerate;
2335
                    st->info->duration_error[i] += error*error;
2336
                }
2337
                st->info->duration_count++;
2338
                // ignore the first 4 values, they might have some random jitter
2339
                if (st->info->duration_count > 3)
2340
                    st->info->duration_gcd = av_gcd(st->info->duration_gcd, duration);
2341
            }
2342
            if (last == AV_NOPTS_VALUE || st->info->duration_count <= 1)
2343
                st->info->last_dts = pkt->dts;
2344
        }
2345
        if(st->parser && st->parser->parser->split && !st->codec->extradata){
2346
            int i= st->parser->parser->split(st->codec, pkt->data, pkt->size);
2347
            if(i){
2348
                st->codec->extradata_size= i;
2349
                st->codec->extradata= av_malloc(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
2350
                memcpy(st->codec->extradata, pkt->data, st->codec->extradata_size);
2351
                memset(st->codec->extradata + i, 0, FF_INPUT_BUFFER_PADDING_SIZE);
2352
            }
2353
        }
2354

    
2355
        /* if still no information, we try to open the codec and to
2356
           decompress the frame. We try to avoid that in most cases as
2357
           it takes longer and uses more memory. For MPEG-4, we need to
2358
           decompress for QuickTime. */
2359
        if (!has_codec_parameters(st->codec) || !has_decode_delay_been_guessed(st))
2360
            try_decode_frame(st, pkt);
2361

    
2362
        st->codec_info_nb_frames++;
2363
        count++;
2364
    }
2365

    
2366
    // close codecs which were opened in try_decode_frame()
2367
    for(i=0;i<ic->nb_streams;i++) {
2368
        st = ic->streams[i];
2369
        if(st->codec->codec)
2370
            avcodec_close(st->codec);
2371
    }
2372
    for(i=0;i<ic->nb_streams;i++) {
2373
        st = ic->streams[i];
2374
        if (st->codec_info_nb_frames>2 && !st->avg_frame_rate.num && st->info->codec_info_duration)
2375
            av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
2376
                     (st->codec_info_nb_frames-2)*(int64_t)st->time_base.den,
2377
                      st->info->codec_info_duration*(int64_t)st->time_base.num, 60000);
2378
        if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
2379
            if(st->codec->codec_id == CODEC_ID_RAWVIDEO && !st->codec->codec_tag && !st->codec->bits_per_coded_sample)
2380
                st->codec->codec_tag= avcodec_pix_fmt_to_codec_tag(st->codec->pix_fmt);
2381

    
2382
            // the check for tb_unreliable() is not completely correct, since this is not about handling
2383
            // a unreliable/inexact time base, but a time base that is finer than necessary, as e.g.
2384
            // ipmovie.c produces.
2385
            if (tb_unreliable(st->codec) && st->info->duration_count > 15 && st->info->duration_gcd > 1 && !st->r_frame_rate.num)
2386
                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);
2387
            if (st->info->duration_count && !st->r_frame_rate.num
2388
               && tb_unreliable(st->codec) /*&&
2389
               //FIXME we should not special-case MPEG-2, but this needs testing with non-MPEG-2 ...
2390
               st->time_base.num*duration_sum[i]/st->info->duration_count*101LL > st->time_base.den*/){
2391
                int num = 0;
2392
                double best_error= 2*av_q2d(st->time_base);
2393
                best_error = best_error*best_error*st->info->duration_count*1000*12*30;
2394

    
2395
                for (j=1; j<FF_ARRAY_ELEMS(st->info->duration_error); j++) {
2396
                    double error = st->info->duration_error[j] * get_std_framerate(j);
2397
//                    if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
2398
//                        av_log(NULL, AV_LOG_ERROR, "%f %f\n", get_std_framerate(j) / 12.0/1001, error);
2399
                    if(error < best_error){
2400
                        best_error= error;
2401
                        num = get_std_framerate(j);
2402
                    }
2403
                }
2404
                // do not increase frame rate by more than 1 % in order to match a standard rate.
2405
                if (num && (!st->r_frame_rate.num || (double)num/(12*1001) < 1.01 * av_q2d(st->r_frame_rate)))
2406
                    av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den, num, 12*1001, INT_MAX);
2407
            }
2408

    
2409
            if (!st->r_frame_rate.num){
2410
                if(    st->codec->time_base.den * (int64_t)st->time_base.num
2411
                    <= st->codec->time_base.num * st->codec->ticks_per_frame * (int64_t)st->time_base.den){
2412
                    st->r_frame_rate.num = st->codec->time_base.den;
2413
                    st->r_frame_rate.den = st->codec->time_base.num * st->codec->ticks_per_frame;
2414
                }else{
2415
                    st->r_frame_rate.num = st->time_base.den;
2416
                    st->r_frame_rate.den = st->time_base.num;
2417
                }
2418
            }
2419
        }else if(st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
2420
            if(!st->codec->bits_per_coded_sample)
2421
                st->codec->bits_per_coded_sample= av_get_bits_per_sample(st->codec->codec_id);
2422
        }
2423
    }
2424

    
2425
    av_estimate_timings(ic, old_offset);
2426

    
2427
    compute_chapters_end(ic);
2428

    
2429
#if 0
2430
    /* correct DTS for B-frame streams with no timestamps */
2431
    for(i=0;i<ic->nb_streams;i++) {
2432
        st = ic->streams[i];
2433
        if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
2434
            if(b-frames){
2435
                ppktl = &ic->packet_buffer;
2436
                while(ppkt1){
2437
                    if(ppkt1->stream_index != i)
2438
                        continue;
2439
                    if(ppkt1->pkt->dts < 0)
2440
                        break;
2441
                    if(ppkt1->pkt->pts != AV_NOPTS_VALUE)
2442
                        break;
2443
                    ppkt1->pkt->dts -= delta;
2444
                    ppkt1= ppkt1->next;
2445
                }
2446
                if(ppkt1)
2447
                    continue;
2448
                st->cur_dts -= delta;
2449
            }
2450
        }
2451
    }
2452
#endif
2453

    
2454
 find_stream_info_err:
2455
    for (i=0; i < ic->nb_streams; i++)
2456
        av_freep(&ic->streams[i]->info);
2457
    return ret;
2458
}
2459

    
2460
/*******************************************************/
2461

    
2462
int av_read_play(AVFormatContext *s)
2463
{
2464
    if (s->iformat->read_play)
2465
        return s->iformat->read_play(s);
2466
    if (s->pb)
2467
        return av_url_read_fpause(s->pb, 0);
2468
    return AVERROR(ENOSYS);
2469
}
2470

    
2471
int av_read_pause(AVFormatContext *s)
2472
{
2473
    if (s->iformat->read_pause)
2474
        return s->iformat->read_pause(s);
2475
    if (s->pb)
2476
        return av_url_read_fpause(s->pb, 1);
2477
    return AVERROR(ENOSYS);
2478
}
2479

    
2480
void av_close_input_stream(AVFormatContext *s)
2481
{
2482
    int i;
2483
    AVStream *st;
2484

    
2485
    if (s->iformat->read_close)
2486
        s->iformat->read_close(s);
2487
    for(i=0;i<s->nb_streams;i++) {
2488
        /* free all data in a stream component */
2489
        st = s->streams[i];
2490
        if (st->parser) {
2491
            av_parser_close(st->parser);
2492
            av_free_packet(&st->cur_pkt);
2493
        }
2494
        av_metadata_free(&st->metadata);
2495
        av_free(st->index_entries);
2496
        av_free(st->codec->extradata);
2497
        av_free(st->codec->subtitle_header);
2498
        av_free(st->codec);
2499
#if FF_API_OLD_METADATA
2500
        av_free(st->filename);
2501
#endif
2502
        av_free(st->priv_data);
2503
        av_free(st->info);
2504
        av_free(st);
2505
    }
2506
    for(i=s->nb_programs-1; i>=0; i--) {
2507
#if FF_API_OLD_METADATA
2508
        av_freep(&s->programs[i]->provider_name);
2509
        av_freep(&s->programs[i]->name);
2510
#endif
2511
        av_metadata_free(&s->programs[i]->metadata);
2512
        av_freep(&s->programs[i]->stream_index);
2513
        av_freep(&s->programs[i]);
2514
    }
2515
    av_freep(&s->programs);
2516
    flush_packet_queue(s);
2517
    av_freep(&s->priv_data);
2518
    while(s->nb_chapters--) {
2519
#if FF_API_OLD_METADATA
2520
        av_free(s->chapters[s->nb_chapters]->title);
2521
#endif
2522
        av_metadata_free(&s->chapters[s->nb_chapters]->metadata);
2523
        av_free(s->chapters[s->nb_chapters]);
2524
    }
2525
    av_freep(&s->chapters);
2526
    av_metadata_free(&s->metadata);
2527
    av_freep(&s->key);
2528
    av_free(s);
2529
}
2530

    
2531
void av_close_input_file(AVFormatContext *s)
2532
{
2533
    ByteIOContext *pb = s->iformat->flags & AVFMT_NOFILE ? NULL : s->pb;
2534
    av_close_input_stream(s);
2535
    if (pb)
2536
        url_fclose(pb);
2537
}
2538

    
2539
AVStream *av_new_stream(AVFormatContext *s, int id)
2540
{
2541
    AVStream *st;
2542
    int i;
2543

    
2544
#if FF_API_MAX_STREAMS
2545
    if (s->nb_streams >= MAX_STREAMS){
2546
        av_log(s, AV_LOG_ERROR, "Too many streams\n");
2547
        return NULL;
2548
    }
2549
#else
2550
    AVStream **streams;
2551

    
2552
    if (s->nb_streams >= INT_MAX/sizeof(*streams))
2553
        return NULL;
2554
    streams = av_realloc(s->streams, (s->nb_streams + 1) * sizeof(*streams));
2555
    if (!streams)
2556
        return NULL;
2557
    s->streams = streams;
2558
#endif
2559

    
2560
    st = av_mallocz(sizeof(AVStream));
2561
    if (!st)
2562
        return NULL;
2563
    if (!(st->info = av_mallocz(sizeof(*st->info)))) {
2564
        av_free(st);
2565
        return NULL;
2566
    }
2567

    
2568
    st->codec= avcodec_alloc_context();
2569
    if (s->iformat) {
2570
        /* no default bitrate if decoding */
2571
        st->codec->bit_rate = 0;
2572
    }
2573
    st->index = s->nb_streams;
2574
    st->id = id;
2575
    st->start_time = AV_NOPTS_VALUE;
2576
    st->duration = AV_NOPTS_VALUE;
2577
        /* we set the current DTS to 0 so that formats without any timestamps
2578
           but durations get some timestamps, formats with some unknown
2579
           timestamps have their first few packets buffered and the
2580
           timestamps corrected before they are returned to the user */
2581
    st->cur_dts = 0;
2582
    st->first_dts = AV_NOPTS_VALUE;
2583
    st->probe_packets = MAX_PROBE_PACKETS;
2584

    
2585
    /* default pts setting is MPEG-like */
2586
    av_set_pts_info(st, 33, 1, 90000);
2587
    st->last_IP_pts = AV_NOPTS_VALUE;
2588
    for(i=0; i<MAX_REORDER_DELAY+1; i++)
2589
        st->pts_buffer[i]= AV_NOPTS_VALUE;
2590
    st->reference_dts = AV_NOPTS_VALUE;
2591

    
2592
    st->sample_aspect_ratio = (AVRational){0,1};
2593

    
2594
    s->streams[s->nb_streams++] = st;
2595
    return st;
2596
}
2597

    
2598
AVProgram *av_new_program(AVFormatContext *ac, int id)
2599
{
2600
    AVProgram *program=NULL;
2601
    int i;
2602

    
2603
#ifdef DEBUG_SI
2604
    av_log(ac, AV_LOG_DEBUG, "new_program: id=0x%04x\n", id);
2605
#endif
2606

    
2607
    for(i=0; i<ac->nb_programs; i++)
2608
        if(ac->programs[i]->id == id)
2609
            program = ac->programs[i];
2610

    
2611
    if(!program){
2612
        program = av_mallocz(sizeof(AVProgram));
2613
        if (!program)
2614
            return NULL;
2615
        dynarray_add(&ac->programs, &ac->nb_programs, program);
2616
        program->discard = AVDISCARD_NONE;
2617
    }
2618
    program->id = id;
2619

    
2620
    return program;
2621
}
2622

    
2623
AVChapter *ff_new_chapter(AVFormatContext *s, int id, AVRational time_base, int64_t start, int64_t end, const char *title)
2624
{
2625
    AVChapter *chapter = NULL;
2626
    int i;
2627

    
2628
    for(i=0; i<s->nb_chapters; i++)
2629
        if(s->chapters[i]->id == id)
2630
            chapter = s->chapters[i];
2631

    
2632
    if(!chapter){
2633
        chapter= av_mallocz(sizeof(AVChapter));
2634
        if(!chapter)
2635
            return NULL;
2636
        dynarray_add(&s->chapters, &s->nb_chapters, chapter);
2637
    }
2638
#if FF_API_OLD_METADATA
2639
    av_free(chapter->title);
2640
#endif
2641
    av_metadata_set2(&chapter->metadata, "title", title, 0);
2642
    chapter->id    = id;
2643
    chapter->time_base= time_base;
2644
    chapter->start = start;
2645
    chapter->end   = end;
2646

    
2647
    return chapter;
2648
}
2649

    
2650
/************************************************************/
2651
/* output media file */
2652

    
2653
int av_set_parameters(AVFormatContext *s, AVFormatParameters *ap)
2654
{
2655
    int ret;
2656

    
2657
    if (s->oformat->priv_data_size > 0) {
2658
        s->priv_data = av_mallocz(s->oformat->priv_data_size);
2659
        if (!s->priv_data)
2660
            return AVERROR(ENOMEM);
2661
    } else
2662
        s->priv_data = NULL;
2663

    
2664
    if (s->oformat->set_parameters) {
2665
        ret = s->oformat->set_parameters(s, ap);
2666
        if (ret < 0)
2667
            return ret;
2668
    }
2669
    return 0;
2670
}
2671

    
2672
static int validate_codec_tag(AVFormatContext *s, AVStream *st)
2673
{
2674
    const AVCodecTag *avctag;
2675
    int n;
2676
    enum CodecID id = CODEC_ID_NONE;
2677
    unsigned int tag = 0;
2678

    
2679
    /**
2680
     * Check that tag + id is in the table
2681
     * If neither is in the table -> OK
2682
     * If tag is in the table with another id -> FAIL
2683
     * If id is in the table with another tag -> FAIL unless strict < normal
2684
     */
2685
    for (n = 0; s->oformat->codec_tag[n]; n++) {
2686
        avctag = s->oformat->codec_tag[n];
2687
        while (avctag->id != CODEC_ID_NONE) {
2688
            if (ff_toupper4(avctag->tag) == ff_toupper4(st->codec->codec_tag)) {
2689
                id = avctag->id;
2690
                if (id == st->codec->codec_id)
2691
                    return 1;
2692
            }
2693
            if (avctag->id == st->codec->codec_id)
2694
                tag = avctag->tag;
2695
            avctag++;
2696
        }
2697
    }
2698
    if (id != CODEC_ID_NONE)
2699
        return 0;
2700
    if (tag && (st->codec->strict_std_compliance >= FF_COMPLIANCE_NORMAL))
2701
        return 0;
2702
    return 1;
2703
}
2704

    
2705
int av_write_header(AVFormatContext *s)
2706
{
2707
    int ret, i;
2708
    AVStream *st;
2709

    
2710
    // some sanity checks
2711
    if (s->nb_streams == 0) {
2712
        av_log(s, AV_LOG_ERROR, "no streams\n");
2713
        return AVERROR(EINVAL);
2714
    }
2715

    
2716
    for(i=0;i<s->nb_streams;i++) {
2717
        st = s->streams[i];
2718

    
2719
        switch (st->codec->codec_type) {
2720
        case AVMEDIA_TYPE_AUDIO:
2721
            if(st->codec->sample_rate<=0){
2722
                av_log(s, AV_LOG_ERROR, "sample rate not set\n");
2723
                return AVERROR(EINVAL);
2724
            }
2725
            if(!st->codec->block_align)
2726
                st->codec->block_align = st->codec->channels *
2727
                    av_get_bits_per_sample(st->codec->codec_id) >> 3;
2728
            break;
2729
        case AVMEDIA_TYPE_VIDEO:
2730
            if(st->codec->time_base.num<=0 || st->codec->time_base.den<=0){ //FIXME audio too?
2731
                av_log(s, AV_LOG_ERROR, "time base not set\n");
2732
                return AVERROR(EINVAL);
2733
            }
2734
            if((st->codec->width<=0 || st->codec->height<=0) && !(s->oformat->flags & AVFMT_NODIMENSIONS)){
2735
                av_log(s, AV_LOG_ERROR, "dimensions not set\n");
2736
                return AVERROR(EINVAL);
2737
            }
2738
            if(av_cmp_q(st->sample_aspect_ratio, st->codec->sample_aspect_ratio)){
2739
                av_log(s, AV_LOG_ERROR, "Aspect ratio mismatch between encoder and muxer layer\n");
2740
                return AVERROR(EINVAL);
2741
            }
2742
            break;
2743
        }
2744

    
2745
        if(s->oformat->codec_tag){
2746
            if(st->codec->codec_tag && st->codec->codec_id == CODEC_ID_RAWVIDEO && av_codec_get_tag(s->oformat->codec_tag, st->codec->codec_id) == 0 && !validate_codec_tag(s, st)){
2747
                //the current rawvideo encoding system ends up setting the wrong codec_tag for avi, we override it here
2748
                st->codec->codec_tag= 0;
2749
            }
2750
            if(st->codec->codec_tag){
2751
                if (!validate_codec_tag(s, st)) {
2752
                    char tagbuf[32];
2753
                    av_get_codec_tag_string(tagbuf, sizeof(tagbuf), st->codec->codec_tag);
2754
                    av_log(s, AV_LOG_ERROR,
2755
                           "Tag %s/0x%08x incompatible with output codec id '%d'\n",
2756
                           tagbuf, st->codec->codec_tag, st->codec->codec_id);
2757
                    return AVERROR_INVALIDDATA;
2758
                }
2759
            }else
2760
                st->codec->codec_tag= av_codec_get_tag(s->oformat->codec_tag, st->codec->codec_id);
2761
        }
2762

    
2763
        if(s->oformat->flags & AVFMT_GLOBALHEADER &&
2764
            !(st->codec->flags & CODEC_FLAG_GLOBAL_HEADER))
2765
          av_log(s, AV_LOG_WARNING, "Codec for stream %d does not use global headers but container format requires global headers\n", i);
2766
    }
2767

    
2768
    if (!s->priv_data && s->oformat->priv_data_size > 0) {
2769
        s->priv_data = av_mallocz(s->oformat->priv_data_size);
2770
        if (!s->priv_data)
2771
            return AVERROR(ENOMEM);
2772
    }
2773

    
2774
#if FF_API_OLD_METADATA
2775
    ff_metadata_mux_compat(s);
2776
#endif
2777

    
2778
    /* set muxer identification string */
2779
    if (!(s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT)) {
2780
        av_metadata_set2(&s->metadata, "encoder", LIBAVFORMAT_IDENT, 0);
2781
    }
2782

    
2783
    if(s->oformat->write_header){
2784
        ret = s->oformat->write_header(s);
2785
        if (ret < 0)
2786
            return ret;
2787
    }
2788

    
2789
    /* init PTS generation */
2790
    for(i=0;i<s->nb_streams;i++) {
2791
        int64_t den = AV_NOPTS_VALUE;
2792
        st = s->streams[i];
2793

    
2794
        switch (st->codec->codec_type) {
2795
        case AVMEDIA_TYPE_AUDIO:
2796
            den = (int64_t)st->time_base.num * st->codec->sample_rate;
2797
            break;
2798
        case AVMEDIA_TYPE_VIDEO:
2799
            den = (int64_t)st->time_base.num * st->codec->time_base.den;
2800
            break;
2801
        default:
2802
            break;
2803
        }
2804
        if (den != AV_NOPTS_VALUE) {
2805
            if (den <= 0)
2806
                return AVERROR_INVALIDDATA;
2807
            av_frac_init(&st->pts, 0, 0, den);
2808
        }
2809
    }
2810
    return 0;
2811
}
2812

    
2813
//FIXME merge with compute_pkt_fields
2814
static int compute_pkt_fields2(AVFormatContext *s, AVStream *st, AVPacket *pkt){
2815
    int delay = FFMAX(st->codec->has_b_frames, !!st->codec->max_b_frames);
2816
    int num, den, frame_size, i;
2817

    
2818
//    av_log(s, AV_LOG_DEBUG, "av_write_frame: pts:%"PRId64" dts:%"PRId64" cur_dts:%"PRId64" b:%d size:%d st:%d\n", pkt->pts, pkt->dts, st->cur_dts, delay, pkt->size, pkt->stream_index);
2819

    
2820
/*    if(pkt->pts == AV_NOPTS_VALUE && pkt->dts == AV_NOPTS_VALUE)
2821
        return -1;*/
2822

    
2823
    /* duration field */
2824
    if (pkt->duration == 0) {
2825
        compute_frame_duration(&num, &den, st, NULL, pkt);
2826
        if (den && num) {
2827
            pkt->duration = av_rescale(1, num * (int64_t)st->time_base.den * st->codec->ticks_per_frame, den * (int64_t)st->time_base.num);
2828
        }
2829
    }
2830

    
2831
    if(pkt->pts == AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE && delay==0)
2832
        pkt->pts= pkt->dts;
2833

    
2834
    //XXX/FIXME this is a temporary hack until all encoders output pts
2835
    if((pkt->pts == 0 || pkt->pts == AV_NOPTS_VALUE) && pkt->dts == AV_NOPTS_VALUE && !delay){
2836
        pkt->dts=
2837
//        pkt->pts= st->cur_dts;
2838
        pkt->pts= st->pts.val;
2839
    }
2840

    
2841
    //calculate dts from pts
2842
    if(pkt->pts != AV_NOPTS_VALUE && pkt->dts == AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY){
2843
        st->pts_buffer[0]= pkt->pts;
2844
        for(i=1; i<delay+1 && st->pts_buffer[i] == AV_NOPTS_VALUE; i++)
2845
            st->pts_buffer[i]= pkt->pts + (i-delay-1) * pkt->duration;
2846
        for(i=0; i<delay && st->pts_buffer[i] > st->pts_buffer[i+1]; i++)
2847
            FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i+1]);
2848

    
2849
        pkt->dts= st->pts_buffer[0];
2850
    }
2851

    
2852
    if(st->cur_dts && st->cur_dts != AV_NOPTS_VALUE && st->cur_dts >= pkt->dts){
2853
        av_log(s, AV_LOG_ERROR,
2854
               "st:%d error, non monotone timestamps %"PRId64" >= %"PRId64"\n",
2855
               st->index, st->cur_dts, pkt->dts);
2856
        return -1;
2857
    }
2858
    if(pkt->dts != AV_NOPTS_VALUE && pkt->pts != AV_NOPTS_VALUE && pkt->pts < pkt->dts){
2859
        av_log(s, AV_LOG_ERROR, "st:%d error, pts < dts\n", st->index);
2860
        return -1;
2861
    }
2862

    
2863
//    av_log(s, AV_LOG_DEBUG, "av_write_frame: pts2:%"PRId64" dts2:%"PRId64"\n", pkt->pts, pkt->dts);
2864
    st->cur_dts= pkt->dts;
2865
    st->pts.val= pkt->dts;
2866

    
2867
    /* update pts */
2868
    switch (st->codec->codec_type) {
2869
    case AVMEDIA_TYPE_AUDIO:
2870
        frame_size = get_audio_frame_size(st->codec, pkt->size);
2871

    
2872
        /* HACK/FIXME, we skip the initial 0 size packets as they are most
2873
           likely equal to the encoder delay, but it would be better if we
2874
           had the real timestamps from the encoder */
2875
        if (frame_size >= 0 && (pkt->size || st->pts.num!=st->pts.den>>1 || st->pts.val)) {
2876
            av_frac_add(&st->pts, (int64_t)st->time_base.den * frame_size);
2877
        }
2878
        break;
2879
    case AVMEDIA_TYPE_VIDEO:
2880
        av_frac_add(&st->pts, (int64_t)st->time_base.den * st->codec->time_base.num);
2881
        break;
2882
    default:
2883
        break;
2884
    }
2885
    return 0;
2886
}
2887

    
2888
int av_write_frame(AVFormatContext *s, AVPacket *pkt)
2889
{
2890
    int ret = compute_pkt_fields2(s, s->streams[pkt->stream_index], pkt);
2891

    
2892
    if(ret<0 && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
2893
        return ret;
2894

    
2895
    ret= s->oformat->write_packet(s, pkt);
2896
    if(!ret)
2897
        ret= url_ferror(s->pb);
2898
    return ret;
2899
}
2900

    
2901
void ff_interleave_add_packet(AVFormatContext *s, AVPacket *pkt,
2902
                              int (*compare)(AVFormatContext *, AVPacket *, AVPacket *))
2903
{
2904
    AVPacketList **next_point, *this_pktl;
2905

    
2906
    this_pktl = av_mallocz(sizeof(AVPacketList));
2907
    this_pktl->pkt= *pkt;
2908
    pkt->destruct= NULL;             // do not free original but only the copy
2909
    av_dup_packet(&this_pktl->pkt);  // duplicate the packet if it uses non-alloced memory
2910

    
2911
    if(s->streams[pkt->stream_index]->last_in_packet_buffer){
2912
        next_point = &(s->streams[pkt->stream_index]->last_in_packet_buffer->next);
2913
    }else
2914
        next_point = &s->packet_buffer;
2915

    
2916
    if(*next_point){
2917
        if(compare(s, &s->packet_buffer_end->pkt, pkt)){
2918
            while(!compare(s, &(*next_point)->pkt, pkt)){
2919
                next_point= &(*next_point)->next;
2920
            }
2921
            goto next_non_null;
2922
        }else{
2923
            next_point = &(s->packet_buffer_end->next);
2924
        }
2925
    }
2926
    assert(!*next_point);
2927

    
2928
    s->packet_buffer_end= this_pktl;
2929
next_non_null:
2930

    
2931
    this_pktl->next= *next_point;
2932

    
2933
    s->streams[pkt->stream_index]->last_in_packet_buffer=
2934
    *next_point= this_pktl;
2935
}
2936

    
2937
int ff_interleave_compare_dts(AVFormatContext *s, AVPacket *next, AVPacket *pkt)
2938
{
2939
    AVStream *st = s->streams[ pkt ->stream_index];
2940
    AVStream *st2= s->streams[ next->stream_index];
2941
    int64_t a= st2->time_base.num * (int64_t)st ->time_base.den;
2942
    int64_t b= st ->time_base.num * (int64_t)st2->time_base.den;
2943
    return av_rescale_rnd(pkt->dts, b, a, AV_ROUND_DOWN) < next->dts;
2944
}
2945

    
2946
int av_interleave_packet_per_dts(AVFormatContext *s, AVPacket *out, AVPacket *pkt, int flush){
2947
    AVPacketList *pktl;
2948
    int stream_count=0;
2949
    int i;
2950

    
2951
    if(pkt){
2952
        ff_interleave_add_packet(s, pkt, ff_interleave_compare_dts);
2953
    }
2954

    
2955
    for(i=0; i < s->nb_streams; i++)
2956
        stream_count+= !!s->streams[i]->last_in_packet_buffer;
2957

    
2958
    if(stream_count && (s->nb_streams == stream_count || flush)){
2959
        pktl= s->packet_buffer;
2960
        *out= pktl->pkt;
2961

    
2962
        s->packet_buffer= pktl->next;
2963
        if(!s->packet_buffer)
2964
            s->packet_buffer_end= NULL;
2965

    
2966
        if(s->streams[out->stream_index]->last_in_packet_buffer == pktl)
2967
            s->streams[out->stream_index]->last_in_packet_buffer= NULL;
2968
        av_freep(&pktl);
2969
        return 1;
2970
    }else{
2971
        av_init_packet(out);
2972
        return 0;
2973
    }
2974
}
2975

    
2976
/**
2977
 * Interleave an AVPacket correctly so it can be muxed.
2978
 * @param out the interleaved packet will be output here
2979
 * @param in the input packet
2980
 * @param flush 1 if no further packets are available as input and all
2981
 *              remaining packets should be output
2982
 * @return 1 if a packet was output, 0 if no packet could be output,
2983
 *         < 0 if an error occurred
2984
 */
2985
static int av_interleave_packet(AVFormatContext *s, AVPacket *out, AVPacket *in, int flush){
2986
    if(s->oformat->interleave_packet)
2987
        return s->oformat->interleave_packet(s, out, in, flush);
2988
    else
2989
        return av_interleave_packet_per_dts(s, out, in, flush);
2990
}
2991

    
2992
int av_interleaved_write_frame(AVFormatContext *s, AVPacket *pkt){
2993
    AVStream *st= s->streams[ pkt->stream_index];
2994

    
2995
    //FIXME/XXX/HACK drop zero sized packets
2996
    if(st->codec->codec_type == AVMEDIA_TYPE_AUDIO && pkt->size==0)
2997
        return 0;
2998

    
2999
//av_log(NULL, AV_LOG_DEBUG, "av_interleaved_write_frame %d %"PRId64" %"PRId64"\n", pkt->size, pkt->dts, pkt->pts);
3000
    if(compute_pkt_fields2(s, st, pkt) < 0 && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
3001
        return -1;
3002

    
3003
    if(pkt->dts == AV_NOPTS_VALUE && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
3004
        return -1;
3005

    
3006
    for(;;){
3007
        AVPacket opkt;
3008
        int ret= av_interleave_packet(s, &opkt, pkt, 0);
3009
        if(ret<=0) //FIXME cleanup needed for ret<0 ?
3010
            return ret;
3011

    
3012
        ret= s->oformat->write_packet(s, &opkt);
3013

    
3014
        av_free_packet(&opkt);
3015
        pkt= NULL;
3016

    
3017
        if(ret<0)
3018
            return ret;
3019
        if(url_ferror(s->pb))
3020
            return url_ferror(s->pb);
3021
    }
3022
}
3023

    
3024
int av_write_trailer(AVFormatContext *s)
3025
{
3026
    int ret, i;
3027

    
3028
    for(;;){
3029
        AVPacket pkt;
3030
        ret= av_interleave_packet(s, &pkt, NULL, 1);
3031
        if(ret<0) //FIXME cleanup needed for ret<0 ?
3032
            goto fail;
3033
        if(!ret)
3034
            break;
3035

    
3036
        ret= s->oformat->write_packet(s, &pkt);
3037

    
3038
        av_free_packet(&pkt);
3039

    
3040
        if(ret<0)
3041
            goto fail;
3042
        if(url_ferror(s->pb))
3043
            goto fail;
3044
    }
3045

    
3046
    if(s->oformat->write_trailer)
3047
        ret = s->oformat->write_trailer(s);
3048
fail:
3049
    if(ret == 0)
3050
       ret=url_ferror(s->pb);
3051
    for(i=0;i<s->nb_streams;i++) {
3052
        av_freep(&s->streams[i]->priv_data);
3053
        av_freep(&s->streams[i]->index_entries);
3054
    }
3055
    av_freep(&s->priv_data);
3056
    return ret;
3057
}
3058

    
3059
void ff_program_add_stream_index(AVFormatContext *ac, int progid, unsigned int idx)
3060
{
3061
    int i, j;
3062
    AVProgram *program=NULL;
3063
    void *tmp;
3064

    
3065
    if (idx >= ac->nb_streams) {
3066
        av_log(ac, AV_LOG_ERROR, "stream index %d is not valid\n", idx);
3067
        return;
3068
    }
3069

    
3070
    for(i=0; i<ac->nb_programs; i++){
3071
        if(ac->programs[i]->id != progid)
3072
            continue;
3073
        program = ac->programs[i];
3074
        for(j=0; j<program->nb_stream_indexes; j++)
3075
            if(program->stream_index[j] == idx)
3076
                return;
3077

    
3078
        tmp = av_realloc(program->stream_index, sizeof(unsigned int)*(program->nb_stream_indexes+1));
3079
        if(!tmp)
3080
            return;
3081
        program->stream_index = tmp;
3082
        program->stream_index[program->nb_stream_indexes++] = idx;
3083
        return;
3084
    }
3085
}
3086

    
3087
static void print_fps(double d, const char *postfix){
3088
    uint64_t v= lrintf(d*100);
3089
    if     (v% 100      ) av_log(NULL, AV_LOG_INFO, ", %3.2f %s", d, postfix);
3090
    else if(v%(100*1000)) av_log(NULL, AV_LOG_INFO, ", %1.0f %s", d, postfix);
3091
    else                  av_log(NULL, AV_LOG_INFO, ", %1.0fk %s", d/1000, postfix);
3092
}
3093

    
3094
static void dump_metadata(void *ctx, AVMetadata *m, const char *indent)
3095
{
3096
    if(m && !(m->count == 1 && av_metadata_get(m, "language", NULL, 0))){
3097
        AVMetadataTag *tag=NULL;
3098

    
3099
        av_log(ctx, AV_LOG_INFO, "%sMetadata:\n", indent);
3100
        while((tag=av_metadata_get(m, "", tag, AV_METADATA_IGNORE_SUFFIX))) {
3101
            if(strcmp("language", tag->key))
3102
                av_log(ctx, AV_LOG_INFO, "%s  %-16s: %s\n", indent, tag->key, tag->value);
3103
        }
3104
    }
3105
}
3106

    
3107
/* "user interface" functions */
3108
static void dump_stream_format(AVFormatContext *ic, int i, int index, int is_output)
3109
{
3110
    char buf[256];
3111
    int flags = (is_output ? ic->oformat->flags : ic->iformat->flags);
3112
    AVStream *st = ic->streams[i];
3113
    int g = av_gcd(st->time_base.num, st->time_base.den);
3114
    AVMetadataTag *lang = av_metadata_get(st->metadata, "language", NULL, 0);
3115
    avcodec_string(buf, sizeof(buf), st->codec, is_output);
3116
    av_log(NULL, AV_LOG_INFO, "    Stream #%d.%d", index, i);
3117
    /* the pid is an important information, so we display it */
3118
    /* XXX: add a generic system */
3119
    if (flags & AVFMT_SHOW_IDS)
3120
        av_log(NULL, AV_LOG_INFO, "[0x%x]", st->id);
3121
    if (lang)
3122
        av_log(NULL, AV_LOG_INFO, "(%s)", lang->value);
3123
    av_log(NULL, AV_LOG_DEBUG, ", %d, %d/%d", st->codec_info_nb_frames, st->time_base.num/g, st->time_base.den/g);
3124
    av_log(NULL, AV_LOG_INFO, ": %s", buf);
3125
    if (st->sample_aspect_ratio.num && // default
3126
        av_cmp_q(st->sample_aspect_ratio, st->codec->sample_aspect_ratio)) {
3127
        AVRational display_aspect_ratio;
3128
        av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
3129
                  st->codec->width*st->sample_aspect_ratio.num,
3130
                  st->codec->height*st->sample_aspect_ratio.den,
3131
                  1024*1024);
3132
        av_log(NULL, AV_LOG_INFO, ", PAR %d:%d DAR %d:%d",
3133
                 st->sample_aspect_ratio.num, st->sample_aspect_ratio.den,
3134
                 display_aspect_ratio.num, display_aspect_ratio.den);
3135
    }
3136
    if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO){
3137
        if(st->avg_frame_rate.den && st->avg_frame_rate.num)
3138
            print_fps(av_q2d(st->avg_frame_rate), "fps");
3139
        if(st->r_frame_rate.den && st->r_frame_rate.num)
3140
            print_fps(av_q2d(st->r_frame_rate), "tbr");
3141
        if(st->time_base.den && st->time_base.num)
3142
            print_fps(1/av_q2d(st->time_base), "tbn");
3143
        if(st->codec->time_base.den && st->codec->time_base.num)
3144
            print_fps(1/av_q2d(st->codec->time_base), "tbc");
3145
    }
3146
    av_log(NULL, AV_LOG_INFO, "\n");
3147
    dump_metadata(NULL, st->metadata, "    ");
3148
}
3149

    
3150
void dump_format(AVFormatContext *ic,
3151
                 int index,
3152
                 const char *url,
3153
                 int is_output)
3154
{
3155
    int i;
3156
    uint8_t *printed = av_mallocz(ic->nb_streams);
3157
    if (ic->nb_streams && !printed)
3158
        return;
3159

    
3160
    av_log(NULL, AV_LOG_INFO, "%s #%d, %s, %s '%s':\n",
3161
            is_output ? "Output" : "Input",
3162
            index,
3163
            is_output ? ic->oformat->name : ic->iformat->name,
3164
            is_output ? "to" : "from", url);
3165
    dump_metadata(NULL, ic->metadata, "  ");
3166
    if (!is_output) {
3167
        av_log(NULL, AV_LOG_INFO, "  Duration: ");
3168
        if (ic->duration != AV_NOPTS_VALUE) {
3169
            int hours, mins, secs, us;
3170
            secs = ic->duration / AV_TIME_BASE;
3171
            us = ic->duration % AV_TIME_BASE;
3172
            mins = secs / 60;
3173
            secs %= 60;
3174
            hours = mins / 60;
3175
            mins %= 60;
3176
            av_log(NULL, AV_LOG_INFO, "%02d:%02d:%02d.%02d", hours, mins, secs,
3177
                   (100 * us) / AV_TIME_BASE);
3178
        } else {
3179
            av_log(NULL, AV_LOG_INFO, "N/A");
3180
        }
3181
        if (ic->start_time != AV_NOPTS_VALUE) {
3182
            int secs, us;
3183
            av_log(NULL, AV_LOG_INFO, ", start: ");
3184
            secs = ic->start_time / AV_TIME_BASE;
3185
            us = abs(ic->start_time % AV_TIME_BASE);
3186
            av_log(NULL, AV_LOG_INFO, "%d.%06d",
3187
                   secs, (int)av_rescale(us, 1000000, AV_TIME_BASE));
3188
        }
3189
        av_log(NULL, AV_LOG_INFO, ", bitrate: ");
3190
        if (ic->bit_rate) {
3191
            av_log(NULL, AV_LOG_INFO,"%d kb/s", ic->bit_rate / 1000);
3192
        } else {
3193
            av_log(NULL, AV_LOG_INFO, "N/A");
3194
        }
3195
        av_log(NULL, AV_LOG_INFO, "\n");
3196
    }
3197
    for (i = 0; i < ic->nb_chapters; i++) {
3198
        AVChapter *ch = ic->chapters[i];
3199
        av_log(NULL, AV_LOG_INFO, "    Chapter #%d.%d: ", index, i);
3200
        av_log(NULL, AV_LOG_INFO, "start %f, ", ch->start * av_q2d(ch->time_base));
3201
        av_log(NULL, AV_LOG_INFO, "end %f\n",   ch->end   * av_q2d(ch->time_base));
3202

    
3203
        dump_metadata(NULL, ch->metadata, "    ");
3204
    }
3205
    if(ic->nb_programs) {
3206
        int j, k, total = 0;
3207
        for(j=0; j<ic->nb_programs; j++) {
3208
            AVMetadataTag *name = av_metadata_get(ic->programs[j]->metadata,
3209
                                                  "name", NULL, 0);
3210
            av_log(NULL, AV_LOG_INFO, "  Program %d %s\n", ic->programs[j]->id,
3211
                   name ? name->value : "");
3212
            dump_metadata(NULL, ic->programs[j]->metadata, "    ");
3213
            for(k=0; k<ic->programs[j]->nb_stream_indexes; k++) {
3214
                dump_stream_format(ic, ic->programs[j]->stream_index[k], index, is_output);
3215
                printed[ic->programs[j]->stream_index[k]] = 1;
3216
            }
3217
            total += ic->programs[j]->nb_stream_indexes;
3218
        }
3219
        if (total < ic->nb_streams)
3220
            av_log(NULL, AV_LOG_INFO, "  No Program\n");
3221
    }
3222
    for(i=0;i<ic->nb_streams;i++)
3223
        if (!printed[i])
3224
            dump_stream_format(ic, i, index, is_output);
3225

    
3226
    av_free(printed);
3227
}
3228

    
3229
#if FF_API_PARSE_FRAME_PARAM
3230
#include "libavcore/parseutils.h"
3231

    
3232
int parse_image_size(int *width_ptr, int *height_ptr, const char *str)
3233
{
3234
    return av_parse_video_size(width_ptr, height_ptr, str);
3235
}
3236

    
3237
int parse_frame_rate(int *frame_rate_num, int *frame_rate_den, const char *arg)
3238
{
3239
    AVRational frame_rate;
3240
    int ret = av_parse_video_rate(&frame_rate, arg);
3241
    *frame_rate_num= frame_rate.num;
3242
    *frame_rate_den= frame_rate.den;
3243
    return ret;
3244
}
3245
#endif
3246

    
3247
int64_t av_gettime(void)
3248
{
3249
    struct timeval tv;
3250
    gettimeofday(&tv,NULL);
3251
    return (int64_t)tv.tv_sec * 1000000 + tv.tv_usec;
3252
}
3253

    
3254
uint64_t ff_ntp_time(void)
3255
{
3256
  return (av_gettime() / 1000) * 1000 + NTP_OFFSET_US;
3257
}
3258

    
3259
int64_t parse_date(const char *datestr, int duration)
3260
{
3261
    const char *p;
3262
    int64_t t;
3263
    struct tm dt;
3264
    int i;
3265
    static const char * const date_fmt[] = {
3266
        "%Y-%m-%d",
3267
        "%Y%m%d",
3268
    };
3269
    static const char * const time_fmt[] = {
3270
        "%H:%M:%S",
3271
        "%H%M%S",
3272
    };
3273
    const char *q;
3274
    int is_utc, len;
3275
    char lastch;
3276
    int negative = 0;
3277

    
3278
#undef time
3279
    time_t now = time(0);
3280

    
3281
    len = strlen(datestr);
3282
    if (len > 0)
3283
        lastch = datestr[len - 1];
3284
    else
3285
        lastch = '\0';
3286
    is_utc = (lastch == 'z' || lastch == 'Z');
3287

    
3288
    memset(&dt, 0, sizeof(dt));
3289

    
3290
    p = datestr;
3291
    q = NULL;
3292
    if (!duration) {
3293
        if (!strncasecmp(datestr, "now", len))
3294
            return (int64_t) now * 1000000;
3295

    
3296
        /* parse the year-month-day part */
3297
        for (i = 0; i < FF_ARRAY_ELEMS(date_fmt); i++) {
3298
            q = small_strptime(p, date_fmt[i], &dt);
3299
            if (q) {
3300
                break;
3301
            }
3302
        }
3303

    
3304
        /* if the year-month-day part is missing, then take the
3305
         * current year-month-day time */
3306
        if (!q) {
3307
            if (is_utc) {
3308
                dt = *gmtime(&now);
3309
            } else {
3310
                dt = *localtime(&now);
3311
            }
3312
            dt.tm_hour = dt.tm_min = dt.tm_sec = 0;
3313
        } else {
3314
            p = q;
3315
        }
3316

    
3317
        if (*p == 'T' || *p == 't' || *p == ' ')
3318
            p++;
3319

    
3320
        /* parse the hour-minute-second part */
3321
        for (i = 0; i < FF_ARRAY_ELEMS(time_fmt); i++) {
3322
            q = small_strptime(p, time_fmt[i], &dt);
3323
            if (q) {
3324
                break;
3325
            }
3326
        }
3327
    } else {
3328
        /* parse datestr as a duration */
3329
        if (p[0] == '-') {
3330
            negative = 1;
3331
            ++p;
3332
        }
3333
        /* parse datestr as HH:MM:SS */
3334
        q = small_strptime(p, time_fmt[0], &dt);
3335
        if (!q) {
3336
            /* parse datestr as S+ */
3337
            dt.tm_sec = strtol(p, (char **)&q, 10);
3338
            if (q == p)
3339
                /* the parsing didn't succeed */
3340
                return INT64_MIN;
3341
            dt.tm_min = 0;
3342
            dt.tm_hour = 0;
3343
        }
3344
    }
3345

    
3346
    /* Now we have all the fields that we can get */
3347
    if (!q) {
3348
        return INT64_MIN;
3349
    }
3350

    
3351
    if (duration) {
3352
        t = dt.tm_hour * 3600 + dt.tm_min * 60 + dt.tm_sec;
3353
    } else {
3354
        dt.tm_isdst = -1;       /* unknown */
3355
        if (is_utc) {
3356
            t = mktimegm(&dt);
3357
        } else {
3358
            t = mktime(&dt);
3359
        }
3360
    }
3361

    
3362
    t *= 1000000;
3363

    
3364
    /* parse the .m... part */
3365
    if (*q == '.') {
3366
        int val, n;
3367
        q++;
3368
        for (val = 0, n = 100000; n >= 1; n /= 10, q++) {
3369
            if (!isdigit(*q))
3370
                break;
3371
            val += n * (*q - '0');
3372
        }
3373
        t += val;
3374
    }
3375
    return negative ? -t : t;
3376
}
3377

    
3378
int find_info_tag(char *arg, int arg_size, const char *tag1, const char *info)
3379
{
3380
    const char *p;
3381
    char tag[128], *q;
3382

    
3383
    p = info;
3384
    if (*p == '?')
3385
        p++;
3386
    for(;;) {
3387
        q = tag;
3388
        while (*p != '\0' && *p != '=' && *p != '&') {
3389
            if ((q - tag) < sizeof(tag) - 1)
3390
                *q++ = *p;
3391
            p++;
3392
        }
3393
        *q = '\0';
3394
        q = arg;
3395
        if (*p == '=') {
3396
            p++;
3397
            while (*p != '&' && *p != '\0') {
3398
                if ((q - arg) < arg_size - 1) {
3399
                    if (*p == '+')
3400
                        *q++ = ' ';
3401
                    else
3402
                        *q++ = *p;
3403
                }
3404
                p++;
3405
            }
3406
        }
3407
        *q = '\0';
3408
        if (!strcmp(tag, tag1))
3409
            return 1;
3410
        if (*p != '&')
3411
            break;
3412
        p++;
3413
    }
3414
    return 0;
3415
}
3416

    
3417
int av_get_frame_filename(char *buf, int buf_size,
3418
                          const char *path, int number)
3419
{
3420
    const char *p;
3421
    char *q, buf1[20], c;
3422
    int nd, len, percentd_found;
3423

    
3424
    q = buf;
3425
    p = path;
3426
    percentd_found = 0;
3427
    for(;;) {
3428
        c = *p++;
3429
        if (c == '\0')
3430
            break;
3431
        if (c == '%') {
3432
            do {
3433
                nd = 0;
3434
                while (isdigit(*p)) {
3435
                    nd = nd * 10 + *p++ - '0';
3436
                }
3437
                c = *p++;
3438
            } while (isdigit(c));
3439

    
3440
            switch(c) {
3441
            case '%':
3442
                goto addchar;
3443
            case 'd':
3444
                if (percentd_found)
3445
                    goto fail;
3446
                percentd_found = 1;
3447
                snprintf(buf1, sizeof(buf1), "%0*d", nd, number);
3448
                len = strlen(buf1);
3449
                if ((q - buf + len) > buf_size - 1)
3450
                    goto fail;
3451
                memcpy(q, buf1, len);
3452
                q += len;
3453
                break;
3454
            default:
3455
                goto fail;
3456
            }
3457
        } else {
3458
        addchar:
3459
            if ((q - buf) < buf_size - 1)
3460
                *q++ = c;
3461
        }
3462
    }
3463
    if (!percentd_found)
3464
        goto fail;
3465
    *q = '\0';
3466
    return 0;
3467
 fail:
3468
    *q = '\0';
3469
    return -1;
3470
}
3471

    
3472
static void hex_dump_internal(void *avcl, FILE *f, int level, uint8_t *buf, int size)
3473
{
3474
    int len, i, j, c;
3475
#undef fprintf
3476
#define PRINT(...) do { if (!f) av_log(avcl, level, __VA_ARGS__); else fprintf(f, __VA_ARGS__); } while(0)
3477

    
3478
    for(i=0;i<size;i+=16) {
3479
        len = size - i;
3480
        if (len > 16)
3481
            len = 16;
3482
        PRINT("%08x ", i);
3483
        for(j=0;j<16;j++) {
3484
            if (j < len)
3485
                PRINT(" %02x", buf[i+j]);
3486
            else
3487
                PRINT("   ");
3488
        }
3489
        PRINT(" ");
3490
        for(j=0;j<len;j++) {
3491
            c = buf[i+j];
3492
            if (c < ' ' || c > '~')
3493
                c = '.';
3494
            PRINT("%c", c);
3495
        }
3496
        PRINT("\n");
3497
    }
3498
#undef PRINT
3499
}
3500

    
3501
void av_hex_dump(FILE *f, uint8_t *buf, int size)
3502
{
3503
    hex_dump_internal(NULL, f, 0, buf, size);
3504
}
3505

    
3506
void av_hex_dump_log(void *avcl, int level, uint8_t *buf, int size)
3507
{
3508
    hex_dump_internal(avcl, NULL, level, buf, size);
3509
}
3510

    
3511
 //FIXME needs to know the time_base
3512
static void pkt_dump_internal(void *avcl, FILE *f, int level, AVPacket *pkt, int dump_payload)
3513
{
3514
#undef fprintf
3515
#define PRINT(...) do { if (!f) av_log(avcl, level, __VA_ARGS__); else fprintf(f, __VA_ARGS__); } while(0)
3516
    PRINT("stream #%d:\n", pkt->stream_index);
3517
    PRINT("  keyframe=%d\n", ((pkt->flags & AV_PKT_FLAG_KEY) != 0));
3518
    PRINT("  duration=%0.3f\n", (double)pkt->duration / AV_TIME_BASE);
3519
    /* DTS is _always_ valid after av_read_frame() */
3520
    PRINT("  dts=");
3521
    if (pkt->dts == AV_NOPTS_VALUE)
3522
        PRINT("N/A");
3523
    else
3524
        PRINT("%0.3f", (double)pkt->dts / AV_TIME_BASE);
3525
    /* PTS may not be known if B-frames are present. */
3526
    PRINT("  pts=");
3527
    if (pkt->pts == AV_NOPTS_VALUE)
3528
        PRINT("N/A");
3529
    else
3530
        PRINT("%0.3f", (double)pkt->pts / AV_TIME_BASE);
3531
    PRINT("\n");
3532
    PRINT("  size=%d\n", pkt->size);
3533
#undef PRINT
3534
    if (dump_payload)
3535
        av_hex_dump(f, pkt->data, pkt->size);
3536
}
3537

    
3538
void av_pkt_dump(FILE *f, AVPacket *pkt, int dump_payload)
3539
{
3540
    pkt_dump_internal(NULL, f, 0, pkt, dump_payload);
3541
}
3542

    
3543
void av_pkt_dump_log(void *avcl, int level, AVPacket *pkt, int dump_payload)
3544
{
3545
    pkt_dump_internal(avcl, NULL, level, pkt, dump_payload);
3546
}
3547

    
3548
#if FF_API_URL_SPLIT
3549
attribute_deprecated
3550
void ff_url_split(char *proto, int proto_size,
3551
                  char *authorization, int authorization_size,
3552
                  char *hostname, int hostname_size,
3553
                  int *port_ptr,
3554
                  char *path, int path_size,
3555
                  const char *url)
3556
{
3557
    av_url_split(proto, proto_size,
3558
                 authorization, authorization_size,
3559
                 hostname, hostname_size,
3560
                 port_ptr,
3561
                 path, path_size,
3562
                 url);
3563
}
3564
#endif
3565

    
3566
void av_url_split(char *proto, int proto_size,
3567
                  char *authorization, int authorization_size,
3568
                  char *hostname, int hostname_size,
3569
                  int *port_ptr,
3570
                  char *path, int path_size,
3571
                  const char *url)
3572
{
3573
    const char *p, *ls, *at, *col, *brk;
3574

    
3575
    if (port_ptr)               *port_ptr = -1;
3576
    if (proto_size > 0)         proto[0] = 0;
3577
    if (authorization_size > 0) authorization[0] = 0;
3578
    if (hostname_size > 0)      hostname[0] = 0;
3579
    if (path_size > 0)          path[0] = 0;
3580

    
3581
    /* parse protocol */
3582
    if ((p = strchr(url, ':'))) {
3583
        av_strlcpy(proto, url, FFMIN(proto_size, p + 1 - url));
3584
        p++; /* skip ':' */
3585
        if (*p == '/') p++;
3586
        if (*p == '/') p++;
3587
    } else {
3588
        /* no protocol means plain filename */
3589
        av_strlcpy(path, url, path_size);
3590
        return;
3591
    }
3592

    
3593
    /* separate path from hostname */
3594
    ls = strchr(p, '/');
3595
    if(!ls)
3596
        ls = strchr(p, '?');
3597
    if(ls)
3598
        av_strlcpy(path, ls, path_size);
3599
    else
3600
        ls = &p[strlen(p)]; // XXX
3601

    
3602
    /* the rest is hostname, use that to parse auth/port */
3603
    if (ls != p) {
3604
        /* authorization (user[:pass]@hostname) */
3605
        if ((at = strchr(p, '@')) && at < ls) {
3606
            av_strlcpy(authorization, p,
3607
                       FFMIN(authorization_size, at + 1 - p));
3608
            p = at + 1; /* skip '@' */
3609
        }
3610

    
3611
        if (*p == '[' && (brk = strchr(p, ']')) && brk < ls) {
3612
            /* [host]:port */
3613
            av_strlcpy(hostname, p + 1,
3614
                       FFMIN(hostname_size, brk - p));
3615
            if (brk[1] == ':' && port_ptr)
3616
                *port_ptr = atoi(brk + 2);
3617
        } else if ((col = strchr(p, ':')) && col < ls) {
3618
            av_strlcpy(hostname, p,
3619
                       FFMIN(col + 1 - p, hostname_size));
3620
            if (port_ptr) *port_ptr = atoi(col + 1);
3621
        } else
3622
            av_strlcpy(hostname, p,
3623
                       FFMIN(ls + 1 - p, hostname_size));
3624
    }
3625
}
3626

    
3627
char *ff_data_to_hex(char *buff, const uint8_t *src, int s, int lowercase)
3628
{
3629
    int i;
3630
    static const char hex_table_uc[16] = { '0', '1', '2', '3',
3631
                                           '4', '5', '6', '7',
3632
                                           '8', '9', 'A', 'B',
3633
                                           'C', 'D', 'E', 'F' };
3634
    static const char hex_table_lc[16] = { '0', '1', '2', '3',
3635
                                           '4', '5', '6', '7',
3636
                                           '8', '9', 'a', 'b',
3637
                                           'c', 'd', 'e', 'f' };
3638
    const char *hex_table = lowercase ? hex_table_lc : hex_table_uc;
3639

    
3640
    for(i = 0; i < s; i++) {
3641
        buff[i * 2]     = hex_table[src[i] >> 4];
3642
        buff[i * 2 + 1] = hex_table[src[i] & 0xF];
3643
    }
3644

    
3645
    return buff;
3646
}
3647

    
3648
int ff_hex_to_data(uint8_t *data, const char *p)
3649
{
3650
    int c, len, v;
3651

    
3652
    len = 0;
3653
    v = 1;
3654
    for (;;) {
3655
        p += strspn(p, SPACE_CHARS);
3656
        if (*p == '\0')
3657
            break;
3658
        c = toupper((unsigned char) *p++);
3659
        if (c >= '0' && c <= '9')
3660
            c = c - '0';
3661
        else if (c >= 'A' && c <= 'F')
3662
            c = c - 'A' + 10;
3663
        else
3664
            break;
3665
        v = (v << 4) | c;
3666
        if (v & 0x100) {
3667
            if (data)
3668
                data[len] = v;
3669
            len++;
3670
            v = 1;
3671
        }
3672
    }
3673
    return len;
3674
}
3675

    
3676
void av_set_pts_info(AVStream *s, int pts_wrap_bits,
3677
                     unsigned int pts_num, unsigned int pts_den)
3678
{
3679
    s->pts_wrap_bits = pts_wrap_bits;
3680

    
3681
    if(av_reduce(&s->time_base.num, &s->time_base.den, pts_num, pts_den, INT_MAX)){
3682
        if(s->time_base.num != pts_num)
3683
            av_log(NULL, AV_LOG_DEBUG, "st:%d removing common factor %d from timebase\n", s->index, pts_num/s->time_base.num);
3684
    }else
3685
        av_log(NULL, AV_LOG_WARNING, "st:%d has too large timebase, reducing\n", s->index);
3686

    
3687
    if(!s->time_base.num || !s->time_base.den)
3688
        s->time_base.num= s->time_base.den= 0;
3689
}
3690

    
3691
int ff_url_join(char *str, int size, const char *proto,
3692
                const char *authorization, const char *hostname,
3693
                int port, const char *fmt, ...)
3694
{
3695
#if CONFIG_NETWORK
3696
    struct addrinfo hints, *ai;
3697
#endif
3698

    
3699
    str[0] = '\0';
3700
    if (proto)
3701
        av_strlcatf(str, size, "%s://", proto);
3702
    if (authorization && authorization[0])
3703
        av_strlcatf(str, size, "%s@", authorization);
3704
#if CONFIG_NETWORK && defined(AF_INET6)
3705
    /* Determine if hostname is a numerical IPv6 address,
3706
     * properly escape it within [] in that case. */
3707
    memset(&hints, 0, sizeof(hints));
3708
    hints.ai_flags = AI_NUMERICHOST;
3709
    if (!getaddrinfo(hostname, NULL, &hints, &ai)) {
3710
        if (ai->ai_family == AF_INET6) {
3711
            av_strlcat(str, "[", size);
3712
            av_strlcat(str, hostname, size);
3713
            av_strlcat(str, "]", size);
3714
        } else {
3715
            av_strlcat(str, hostname, size);
3716
        }
3717
        freeaddrinfo(ai);
3718
    } else
3719
#endif
3720
        /* Not an IPv6 address, just output the plain string. */
3721
        av_strlcat(str, hostname, size);
3722

    
3723
    if (port >= 0)
3724
        av_strlcatf(str, size, ":%d", port);
3725
    if (fmt) {
3726
        va_list vl;
3727
        int len = strlen(str);
3728

    
3729
        va_start(vl, fmt);
3730
        vsnprintf(str + len, size > len ? size - len : 0, fmt, vl);
3731
        va_end(vl);
3732
    }
3733
    return strlen(str);
3734
}
3735

    
3736
int ff_write_chained(AVFormatContext *dst, int dst_stream, AVPacket *pkt,
3737
                     AVFormatContext *src)
3738
{
3739
    AVPacket local_pkt;
3740

    
3741
    local_pkt = *pkt;
3742
    local_pkt.stream_index = dst_stream;
3743
    if (pkt->pts != AV_NOPTS_VALUE)
3744
        local_pkt.pts = av_rescale_q(pkt->pts,
3745
                                     src->streams[pkt->stream_index]->time_base,
3746
                                     dst->streams[dst_stream]->time_base);
3747
    if (pkt->dts != AV_NOPTS_VALUE)
3748
        local_pkt.dts = av_rescale_q(pkt->dts,
3749
                                     src->streams[pkt->stream_index]->time_base,
3750
                                     dst->streams[dst_stream]->time_base);
3751
    return av_write_frame(dst, &local_pkt);
3752
}
3753

    
3754
void ff_parse_key_value(const char *str, ff_parse_key_val_cb callback_get_buf,
3755
                        void *context)
3756
{
3757
    const char *ptr = str;
3758

    
3759
    /* Parse key=value pairs. */
3760
    for (;;) {
3761
        const char *key;
3762
        char *dest = NULL, *dest_end;
3763
        int key_len, dest_len = 0;
3764

    
3765
        /* Skip whitespace and potential commas. */
3766
        while (*ptr && (isspace(*ptr) || *ptr == ','))
3767
            ptr++;
3768
        if (!*ptr)
3769
            break;
3770

    
3771
        key = ptr;
3772

    
3773
        if (!(ptr = strchr(key, '=')))
3774
            break;
3775
        ptr++;
3776
        key_len = ptr - key;
3777

    
3778
        callback_get_buf(context, key, key_len, &dest, &dest_len);
3779
        dest_end = dest + dest_len - 1;
3780

    
3781
        if (*ptr == '\"') {
3782
            ptr++;
3783
            while (*ptr && *ptr != '\"') {
3784
                if (*ptr == '\\') {
3785
                    if (!ptr[1])
3786
                        break;
3787
                    if (dest && dest < dest_end)
3788
                        *dest++ = ptr[1];
3789
                    ptr += 2;
3790
                } else {
3791
                    if (dest && dest < dest_end)
3792
                        *dest++ = *ptr;
3793
                    ptr++;
3794
                }
3795
            }
3796
            if (*ptr == '\"')
3797
                ptr++;
3798
        } else {
3799
            for (; *ptr && !(isspace(*ptr) || *ptr == ','); ptr++)
3800
                if (dest && dest < dest_end)
3801
                    *dest++ = *ptr;
3802
        }
3803
        if (dest)
3804
            *dest = 0;
3805
    }
3806
}
3807