ffmpeg.c

   1 /*
   2  * Copyright (c) 2000-2003 Fabrice Bellard
   3  *
   4  * This file is part of FFmpeg.
   5  *
   6  * FFmpeg is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * FFmpeg is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with FFmpeg; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 /**
  22  * @file
  23  * multimedia converter based on the FFmpeg libraries
  24  */
  25
  26 #include "config.h"
  27 #include <ctype.h>
  28 #include <string.h>
  29 #include <math.h>
  30 #include <stdlib.h>
  31 #include <errno.h>
  32 #include <limits.h>
  33 #include <stdint.h>
  34
  35 #if HAVE_IO_H
  36 #include <io.h>
  37 #endif
  38 #if HAVE_UNISTD_H
  39 #include <unistd.h>
  40 #endif
  41
  42 #include "libavformat/avformat.h"
  43 #include "libavdevice/avdevice.h"
  44 #include "libswresample/swresample.h"
  45 #include "libavutil/opt.h"
  46 #include "libavutil/channel_layout.h"
  47 #include "libavutil/parseutils.h"
  48 #include "libavutil/samplefmt.h"
  49 #include "libavutil/fifo.h"
  50 #include "libavutil/internal.h"
  51 #include "libavutil/intreadwrite.h"
  52 #include "libavutil/dict.h"
  53 #include "libavutil/mathematics.h"
  54 #include "libavutil/pixdesc.h"
  55 #include "libavutil/avstring.h"
  56 #include "libavutil/libm.h"
  57 #include "libavutil/imgutils.h"
  58 #include "libavutil/timestamp.h"
  59 #include "libavutil/bprint.h"
  60 #include "libavutil/time.h"
  61 #include "libavutil/threadmessage.h"
  62 #include "libavcodec/mathops.h"
  63 #include "libavformat/os_support.h"
  64
  65 # include "libavfilter/avfilter.h"
  66 # include "libavfilter/buffersrc.h"
  67 # include "libavfilter/buffersink.h"
  68
  69 #if HAVE_SYS_RESOURCE_H
  70 #include <sys/time.h>
  71 #include <sys/types.h>
  72 #include <sys/resource.h>
  73 #elif HAVE_GETPROCESSTIMES
  74 #include <windows.h>
  75 #endif
  76 #if HAVE_GETPROCESSMEMORYINFO
  77 #include <windows.h>
  78 #include <psapi.h>
  79 #endif
  80 #if HAVE_SETCONSOLECTRLHANDLER
  81 #include <windows.h>
  82 #endif
  83
  84
  85 #if HAVE_SYS_SELECT_H
  86 #include <sys/select.h>
  87 #endif
  88
  89 #if HAVE_TERMIOS_H
  90 #include <fcntl.h>
  91 #include <sys/ioctl.h>
  92 #include <sys/time.h>
  93 #include <termios.h>
  94 #elif HAVE_KBHIT
  95 #include <conio.h>
  96 #endif
  97
  98 #if HAVE_PTHREADS
  99 #include <pthread.h>
 100 #endif
 101
 102 #include <time.h>
 103
 104 #include "ffmpeg.h"
 105 #include "cmdutils.h"
 106
 107 #include "libavutil/avassert.h"
 108
 109 const char program_name[] = "ffmpeg";
 110 const int program_birth_year = 2000;
 111
 112 static FILE *vstats_file;
 113
 114 const char *const forced_keyframes_const_names[] = {
 115     "n",
 116     "n_forced",
 117     "prev_forced_n",
 118     "prev_forced_t",
 119     "t",
 120     NULL
 121 };
 122
 123 static void do_video_stats(OutputStream *ost, int frame_size);
 124 static int64_t getutime(void);
 125 static int64_t getmaxrss(void);
 126
 127 static int run_as_daemon  = 0;
 128 static int nb_frames_dup = 0;
 129 static int nb_frames_drop = 0;
 130 static int64_t decode_error_stat[2];
 131
 132 static int current_time;
 133 AVIOContext *progress_avio = NULL;
 134
 135 static uint8_t *subtitle_out;
 136
 137 InputStream **input_streams = NULL;
 138 int        nb_input_streams = 0;
 139 InputFile   **input_files   = NULL;
 140 int        nb_input_files   = 0;
 141
 142 OutputStream **output_streams = NULL;
 143 int         nb_output_streams = 0;
 144 OutputFile   **output_files   = NULL;
 145 int         nb_output_files   = 0;
 146
 147 FilterGraph **filtergraphs;
 148 int        nb_filtergraphs;
 149
 150 #if HAVE_TERMIOS_H
 151
 152 /* init terminal so that we can grab keys */
 153 static struct termios oldtty;
 154 static int restore_tty;
 155 #endif
 156
 157 #if HAVE_PTHREADS
 158 static void free_input_threads(void);
 159 #endif
 160
 161 /* sub2video hack:
 162    Convert subtitles to video with alpha to insert them in filter graphs.
 163    This is a temporary solution until libavfilter gets real subtitles support.
 164  */
 165
 166 static int sub2video_get_blank_frame(InputStream *ist)
 167 {
 168     int ret;
 169     AVFrame *frame = ist->sub2video.frame;
 170
 171     av_frame_unref(frame);
 172     ist->sub2video.frame->width  = ist->dec_ctx->width  ? ist->dec_ctx->width  : ist->sub2video.w;
 173     ist->sub2video.frame->height = ist->dec_ctx->height ? ist->dec_ctx->height : ist->sub2video.h;
 174     ist->sub2video.frame->format = AV_PIX_FMT_RGB32;
 175     if ((ret = av_frame_get_buffer(frame, 32)) < 0)
 176         return ret;
 177     memset(frame->data[0], 0, frame->height * frame->linesize[0]);
 178     return 0;
 179 }
 180
 181 static void sub2video_copy_rect(uint8_t *dst, int dst_linesize, int w, int h,
 182                                 AVSubtitleRect *r)
 183 {
 184     uint32_t *pal, *dst2;
 185     uint8_t *src, *src2;
 186     int x, y;
 187
 188     if (r->type != SUBTITLE_BITMAP) {
 189         av_log(NULL, AV_LOG_WARNING, "sub2video: non-bitmap subtitle\n");
 190         return;
 191     }
 192     if (r->x < 0 || r->x + r->w > w || r->y < 0 || r->y + r->h > h) {
 193         av_log(NULL, AV_LOG_WARNING, "sub2video: rectangle (%d %d %d %d) overflowing %d %d\n",
 194             r->x, r->y, r->w, r->h, w, h
 195         );
 196         return;
 197     }
 198
 199     dst += r->y * dst_linesize + r->x * 4;
 200     src = r->data[0];
 201     pal = (uint32_t *)r->data[1];
 202     for (y = 0; y < r->h; y++) {
 203         dst2 = (uint32_t *)dst;
 204         src2 = src;
 205         for (x = 0; x < r->w; x++)
 206             *(dst2++) = pal[*(src2++)];
 207         dst += dst_linesize;
 208         src += r->linesize[0];
 209     }
 210 }
 211
 212 static void sub2video_push_ref(InputStream *ist, int64_t pts)
 213 {
 214     AVFrame *frame = ist->sub2video.frame;
 215     int i;
 216
 217     av_assert1(frame->data[0]);
 218     ist->sub2video.last_pts = frame->pts = pts;
 219     for (i = 0; i < ist->nb_filters; i++)
 220         av_buffersrc_add_frame_flags(ist->filters[i]->filter, frame,
 221                                      AV_BUFFERSRC_FLAG_KEEP_REF |
 222                                      AV_BUFFERSRC_FLAG_PUSH);
 223 }
 224
 225 static void sub2video_update(InputStream *ist, AVSubtitle *sub)
 226 {
 227     AVFrame *frame = ist->sub2video.frame;
 228     int8_t *dst;
 229     int     dst_linesize;
 230     int num_rects, i;
 231     int64_t pts, end_pts;
 232
 233     if (!frame)
 234         return;
 235     if (sub) {
 236         pts       = av_rescale_q(sub->pts + sub->start_display_time * 1000LL,
 237                                  AV_TIME_BASE_Q, ist->st->time_base);
 238         end_pts   = av_rescale_q(sub->pts + sub->end_display_time   * 1000LL,
 239                                  AV_TIME_BASE_Q, ist->st->time_base);
 240         num_rects = sub->num_rects;
 241     } else {
 242         pts       = ist->sub2video.end_pts;
 243         end_pts   = INT64_MAX;
 244         num_rects = 0;
 245     }
 246     if (sub2video_get_blank_frame(ist) < 0) {
 247         av_log(ist->dec_ctx, AV_LOG_ERROR,
 248                "Impossible to get a blank canvas.\n");
 249         return;
 250     }
 251     dst          = frame->data    [0];
 252     dst_linesize = frame->linesize[0];
 253     for (i = 0; i < num_rects; i++)
 254         sub2video_copy_rect(dst, dst_linesize, frame->width, frame->height, sub->rects[i]);
 255     sub2video_push_ref(ist, pts);
 256     ist->sub2video.end_pts = end_pts;
 257 }
 258
 259 static void sub2video_heartbeat(InputStream *ist, int64_t pts)
 260 {
 261     InputFile *infile = input_files[ist->file_index];
 262     int i, j, nb_reqs;
 263     int64_t pts2;
 264
 265     /* When a frame is read from a file, examine all sub2video streams in
 266        the same file and send the sub2video frame again. Otherwise, decoded
 267        video frames could be accumulating in the filter graph while a filter
 268        (possibly overlay) is desperately waiting for a subtitle frame. */
 269     for (i = 0; i < infile->nb_streams; i++) {
 270         InputStream *ist2 = input_streams[infile->ist_index + i];
 271         if (!ist2->sub2video.frame)
 272             continue;
 273         /* subtitles seem to be usually muxed ahead of other streams;
 274            if not, subtracting a larger time here is necessary */
 275         pts2 = av_rescale_q(pts, ist->st->time_base, ist2->st->time_base) - 1;
 276         /* do not send the heartbeat frame if the subtitle is already ahead */
 277         if (pts2 <= ist2->sub2video.last_pts)
 278             continue;
 279         if (pts2 >= ist2->sub2video.end_pts || !ist2->sub2video.frame->data[0])
 280             sub2video_update(ist2, NULL);
 281         for (j = 0, nb_reqs = 0; j < ist2->nb_filters; j++)
 282             nb_reqs += av_buffersrc_get_nb_failed_requests(ist2->filters[j]->filter);
 283         if (nb_reqs)
 284             sub2video_push_ref(ist2, pts2);
 285     }
 286 }
 287
 288 static void sub2video_flush(InputStream *ist)
 289 {
 290     int i;
 291
 292     if (ist->sub2video.end_pts < INT64_MAX)
 293         sub2video_update(ist, NULL);
 294     for (i = 0; i < ist->nb_filters; i++)
 295         av_buffersrc_add_frame(ist->filters[i]->filter, NULL);
 296 }
 297
 298 /* end of sub2video hack */
 299
 300 static void term_exit_sigsafe(void)
 301 {
 302 #if HAVE_TERMIOS_H
 303     if(restore_tty)
 304         tcsetattr (0, TCSANOW, &oldtty);
 305 #endif
 306 }
 307
 308 void term_exit(void)
 309 {
 310     av_log(NULL, AV_LOG_QUIET, "%s", "");
 311     term_exit_sigsafe();
 312 }
 313
 314 static volatile int received_sigterm = 0;
 315 static volatile int received_nb_signals = 0;
 316 static volatile int transcode_init_done = 0;
 317 static volatile int ffmpeg_exited = 0;
 318 static int main_return_code = 0;
 319
 320 static void
 321 sigterm_handler(int sig)
 322 {
 323     received_sigterm = sig;
 324     received_nb_signals++;
 325     term_exit_sigsafe();
 326     if(received_nb_signals > 3) {
 327         write(2/*STDERR_FILENO*/, "Received > 3 system signals, hard exiting\n",
 328                            strlen("Received > 3 system signals, hard exiting\n"));
 329
 330         exit(123);
 331     }
 332 }
 333
 334 #if HAVE_SETCONSOLECTRLHANDLER
 335 static BOOL WINAPI CtrlHandler(DWORD fdwCtrlType)
 336 {
 337     av_log(NULL, AV_LOG_DEBUG, "\nReceived windows signal %ld\n", fdwCtrlType);
 338
 339     switch (fdwCtrlType)
 340     {
 341     case CTRL_C_EVENT:
 342     case CTRL_BREAK_EVENT:
 343         sigterm_handler(SIGINT);
 344         return TRUE;
 345
 346     case CTRL_CLOSE_EVENT:
 347     case CTRL_LOGOFF_EVENT:
 348     case CTRL_SHUTDOWN_EVENT:
 349         sigterm_handler(SIGTERM);
 350         /* Basically, with these 3 events, when we return from this method the
 351            process is hard terminated, so stall as long as we need to
 352            to try and let the main thread(s) clean up and gracefully terminate
 353            (we have at most 5 seconds, but should be done far before that). */
 354         while (!ffmpeg_exited) {
 355             Sleep(0);
 356         }
 357         return TRUE;
 358
 359     default:
 360         av_log(NULL, AV_LOG_ERROR, "Received unknown windows signal %ld\n", fdwCtrlType);
 361         return FALSE;
 362     }
 363 }
 364 #endif
 365
 366 void term_init(void)
 367 {
 368 #if HAVE_TERMIOS_H
 369     if(!run_as_daemon){
 370         struct termios tty;
 371         if (tcgetattr (0, &tty) == 0) {
 372             oldtty = tty;
 373             restore_tty = 1;
 374
 375             tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
 376                              |INLCR|IGNCR|ICRNL|IXON);
 377             tty.c_oflag |= OPOST;
 378             tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);
 379             tty.c_cflag &= ~(CSIZE|PARENB);
 380             tty.c_cflag |= CS8;
 381             tty.c_cc[VMIN] = 1;
 382             tty.c_cc[VTIME] = 0;
 383
 384             tcsetattr (0, TCSANOW, &tty);
 385         }
 386         signal(SIGQUIT, sigterm_handler); /* Quit (POSIX).  */
 387     }
 388 #endif
 389
 390     signal(SIGINT , sigterm_handler); /* Interrupt (ANSI).    */
 391     signal(SIGTERM, sigterm_handler); /* Termination (ANSI).  */
 392 #ifdef SIGXCPU
 393     signal(SIGXCPU, sigterm_handler);
 394 #endif
 395 #if HAVE_SETCONSOLECTRLHANDLER
 396     SetConsoleCtrlHandler((PHANDLER_ROUTINE) CtrlHandler, TRUE);
 397 #endif
 398 }
 399
 400 /* read a key without blocking */
 401 static int read_key(void)
 402 {
 403     unsigned char ch;
 404 #if HAVE_TERMIOS_H
 405     int n = 1;
 406     struct timeval tv;
 407     fd_set rfds;
 408
 409     FD_ZERO(&rfds);
 410     FD_SET(0, &rfds);
 411     tv.tv_sec = 0;
 412     tv.tv_usec = 0;
 413     n = select(1, &rfds, NULL, NULL, &tv);
 414     if (n > 0) {
 415         n = read(0, &ch, 1);
 416         if (n == 1)
 417             return ch;
 418
 419         return n;
 420     }
 421 #elif HAVE_KBHIT
 422 #    if HAVE_PEEKNAMEDPIPE
 423     static int is_pipe;
 424     static HANDLE input_handle;
 425     DWORD dw, nchars;
 426     if(!input_handle){
 427         input_handle = GetStdHandle(STD_INPUT_HANDLE);
 428         is_pipe = !GetConsoleMode(input_handle, &dw);
 429     }
 430
 431     if (is_pipe) {
 432         /* When running under a GUI, you will end here. */
 433         if (!PeekNamedPipe(input_handle, NULL, 0, NULL, &nchars, NULL)) {
 434             // input pipe may have been closed by the program that ran ffmpeg
 435             return -1;
 436         }
 437         //Read it
 438         if(nchars != 0) {
 439             read(0, &ch, 1);
 440             return ch;
 441         }else{
 442             return -1;
 443         }
 444     }
 445 #    endif
 446     if(kbhit())
 447         return(getch());
 448 #endif
 449     return -1;
 450 }
 451
 452 static int decode_interrupt_cb(void *ctx)
 453 {
 454     return received_nb_signals > transcode_init_done;
 455 }
 456
 457 const AVIOInterruptCB int_cb = { decode_interrupt_cb, NULL };
 458
 459 static void ffmpeg_cleanup(int ret)
 460 {
 461     int i, j;
 462
 463     if (do_benchmark) {
 464         int maxrss = getmaxrss() / 1024;
 465         av_log(NULL, AV_LOG_INFO, "bench: maxrss=%ikB\n", maxrss);
 466     }
 467
 468     for (i = 0; i < nb_filtergraphs; i++) {
 469         FilterGraph *fg = filtergraphs[i];
 470         avfilter_graph_free(&fg->graph);
 471         for (j = 0; j < fg->nb_inputs; j++) {
 472             av_freep(&fg->inputs[j]->name);
 473             av_freep(&fg->inputs[j]);
 474         }
 475         av_freep(&fg->inputs);
 476         for (j = 0; j < fg->nb_outputs; j++) {
 477             av_freep(&fg->outputs[j]->name);
 478             av_freep(&fg->outputs[j]);
 479         }
 480         av_freep(&fg->outputs);
 481         av_freep(&fg->graph_desc);
 482
 483         av_freep(&filtergraphs[i]);
 484     }
 485     av_freep(&filtergraphs);
 486
 487     av_freep(&subtitle_out);
 488
 489     /* close files */
 490     for (i = 0; i < nb_output_files; i++) {
 491         OutputFile *of = output_files[i];
 492         AVFormatContext *s;
 493         if (!of)
 494             continue;
 495         s = of->ctx;
 496         if (s && s->oformat && !(s->oformat->flags & AVFMT_NOFILE))
 497             avio_closep(&s->pb);
 498         avformat_free_context(s);
 499         av_dict_free(&of->opts);
 500
 501         av_freep(&output_files[i]);
 502     }
 503     for (i = 0; i < nb_output_streams; i++) {
 504         OutputStream *ost = output_streams[i];
 505         AVBitStreamFilterContext *bsfc;
 506
 507         if (!ost)
 508             continue;
 509
 510         bsfc = ost->bitstream_filters;
 511         while (bsfc) {
 512             AVBitStreamFilterContext *next = bsfc->next;
 513             av_bitstream_filter_close(bsfc);
 514             bsfc = next;
 515         }
 516         ost->bitstream_filters = NULL;
 517         av_frame_free(&ost->filtered_frame);
 518         av_frame_free(&ost->last_frame);
 519
 520         av_parser_close(ost->parser);
 521
 522         av_freep(&ost->forced_keyframes);
 523         av_expr_free(ost->forced_keyframes_pexpr);
 524         av_freep(&ost->avfilter);
 525         av_freep(&ost->logfile_prefix);
 526
 527         av_freep(&ost->audio_channels_map);
 528         ost->audio_channels_mapped = 0;
 529
 530         av_dict_free(&ost->sws_dict);
 531
 532         avcodec_free_context(&ost->enc_ctx);
 533
 534         av_freep(&output_streams[i]);
 535     }
 536 #if HAVE_PTHREADS
 537     free_input_threads();
 538 #endif
 539     for (i = 0; i < nb_input_files; i++) {
 540         avformat_close_input(&input_files[i]->ctx);
 541         av_freep(&input_files[i]);
 542     }
 543     for (i = 0; i < nb_input_streams; i++) {
 544         InputStream *ist = input_streams[i];
 545
 546         av_frame_free(&ist->decoded_frame);
 547         av_frame_free(&ist->filter_frame);
 548         av_dict_free(&ist->decoder_opts);
 549         avsubtitle_free(&ist->prev_sub.subtitle);
 550         av_frame_free(&ist->sub2video.frame);
 551         av_freep(&ist->filters);
 552         av_freep(&ist->hwaccel_device);
 553
 554         avcodec_free_context(&ist->dec_ctx);
 555
 556         av_freep(&input_streams[i]);
 557     }
 558
 559     if (vstats_file) {
 560         if (fclose(vstats_file))
 561             av_log(NULL, AV_LOG_ERROR,
 562                    "Error closing vstats file, loss of information possible: %s\n",
 563                    av_err2str(AVERROR(errno)));
 564     }
 565     av_freep(&vstats_filename);
 566
 567     av_freep(&input_streams);
 568     av_freep(&input_files);
 569     av_freep(&output_streams);
 570     av_freep(&output_files);
 571
 572     uninit_opts();
 573
 574     avformat_network_deinit();
 575
 576     if (received_sigterm) {
 577         av_log(NULL, AV_LOG_INFO, "Exiting normally, received signal %d.\n",
 578                (int) received_sigterm);
 579     } else if (ret && transcode_init_done) {
 580         av_log(NULL, AV_LOG_INFO, "Conversion failed!\n");
 581     }
 582     term_exit();
 583     ffmpeg_exited = 1;
 584 }
 585
 586 void remove_avoptions(AVDictionary **a, AVDictionary *b)
 587 {
 588     AVDictionaryEntry *t = NULL;
 589
 590     while ((t = av_dict_get(b, "", t, AV_DICT_IGNORE_SUFFIX))) {
 591         av_dict_set(a, t->key, NULL, AV_DICT_MATCH_CASE);
 592     }
 593 }
 594
 595 void assert_avoptions(AVDictionary *m)
 596 {
 597     AVDictionaryEntry *t;
 598     if ((t = av_dict_get(m, "", NULL, AV_DICT_IGNORE_SUFFIX))) {
 599         av_log(NULL, AV_LOG_FATAL, "Option %s not found.\n", t->key);
 600         exit_program(1);
 601     }
 602 }
 603
 604 static void abort_codec_experimental(AVCodec *c, int encoder)
 605 {
 606     exit_program(1);
 607 }
 608
 609 static void update_benchmark(const char *fmt, ...)
 610 {
 611     if (do_benchmark_all) {
 612         int64_t t = getutime();
 613         va_list va;
 614         char buf[1024];
 615
 616         if (fmt) {
 617             va_start(va, fmt);
 618             vsnprintf(buf, sizeof(buf), fmt, va);
 619             va_end(va);
 620             av_log(NULL, AV_LOG_INFO, "bench: %8"PRIu64" %s \n", t - current_time, buf);
 621         }
 622         current_time = t;
 623     }
 624 }
 625
 626 static void close_all_output_streams(OutputStream *ost, OSTFinished this_stream, OSTFinished others)
 627 {
 628     int i;
 629     for (i = 0; i < nb_output_streams; i++) {
 630         OutputStream *ost2 = output_streams[i];
 631         ost2->finished |= ost == ost2 ? this_stream : others;
 632     }
 633 }
 634
 635 static void write_frame(AVFormatContext *s, AVPacket *pkt, OutputStream *ost)
 636 {
 637     AVBitStreamFilterContext *bsfc = ost->bitstream_filters;
 638     AVCodecContext          *avctx = ost->encoding_needed ? ost->enc_ctx : ost->st->codec;
 639     int ret;
 640
 641     if (!ost->st->codec->extradata_size && ost->enc_ctx->extradata_size) {
 642         ost->st->codec->extradata = av_mallocz(ost->enc_ctx->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
 643         if (ost->st->codec->extradata) {
 644             memcpy(ost->st->codec->extradata, ost->enc_ctx->extradata, ost->enc_ctx->extradata_size);
 645             ost->st->codec->extradata_size = ost->enc_ctx->extradata_size;
 646         }
 647     }
 648
 649     if ((avctx->codec_type == AVMEDIA_TYPE_VIDEO && video_sync_method == VSYNC_DROP) ||
 650         (avctx->codec_type == AVMEDIA_TYPE_AUDIO && audio_sync_method < 0))
 651         pkt->pts = pkt->dts = AV_NOPTS_VALUE;
 652
 653     /*
 654      * Audio encoders may split the packets --  #frames in != #packets out.
 655      * But there is no reordering, so we can limit the number of output packets
 656      * by simply dropping them here.
 657      * Counting encoded video frames needs to be done separately because of
 658      * reordering, see do_video_out()
 659      */
 660     if (!(avctx->codec_type == AVMEDIA_TYPE_VIDEO && avctx->codec)) {
 661         if (ost->frame_number >= ost->max_frames) {
 662             av_packet_unref(pkt);
 663             return;
 664         }
 665         ost->frame_number++;
 666     }
 667     if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
 668         int i;
 669         uint8_t *sd = av_packet_get_side_data(pkt, AV_PKT_DATA_QUALITY_STATS,
 670                                               NULL);
 671         ost->quality = sd ? AV_RL32(sd) : -1;
 672         ost->pict_type = sd ? sd[4] : AV_PICTURE_TYPE_NONE;
 673
 674         for (i = 0; i<FF_ARRAY_ELEMS(ost->error); i++) {
 675             if (sd && i < sd[5])
 676                 ost->error[i] = AV_RL64(sd + 8 + 8*i);
 677             else
 678                 ost->error[i] = -1;
 679         }
 680
 681         if (ost->frame_rate.num && ost->is_cfr) {
 682             if (pkt->duration > 0)
 683                 av_log(NULL, AV_LOG_WARNING, "Overriding packet duration by frame rate, this should not happen\n");
 684             pkt->duration = av_rescale_q(1, av_inv_q(ost->frame_rate),
 685                                          ost->st->time_base);
 686         }
 687     }
 688
 689     if (bsfc)
 690         av_packet_split_side_data(pkt);
 691
 692     if ((ret = av_apply_bitstream_filters(avctx, pkt, bsfc)) < 0) {
 693         print_error("", ret);
 694         if (exit_on_error)
 695             exit_program(1);
 696     }
 697     if (pkt->size == 0 && pkt->side_data_elems == 0)
 698         return;
 699     if (!ost->st->codecpar->extradata && avctx->extradata) {
 700         ost->st->codecpar->extradata = av_malloc(avctx->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
 701         if (!ost->st->codecpar->extradata) {
 702             av_log(NULL, AV_LOG_ERROR, "Could not allocate extradata buffer to copy parser data.\n");
 703             exit_program(1);
 704         }
 705         ost->st->codecpar->extradata_size = avctx->extradata_size;
 706         memcpy(ost->st->codecpar->extradata, avctx->extradata, avctx->extradata_size);
 707     }
 708
 709     if (!(s->oformat->flags & AVFMT_NOTIMESTAMPS)) {
 710         if (pkt->dts != AV_NOPTS_VALUE &&
 711             pkt->pts != AV_NOPTS_VALUE &&
 712             pkt->dts > pkt->pts) {
 713             av_log(s, AV_LOG_WARNING, "Invalid DTS: %"PRId64" PTS: %"PRId64" in output stream %d:%d, replacing by guess\n",
 714                    pkt->dts, pkt->pts,
 715                    ost->file_index, ost->st->index);
 716             pkt->pts =
 717             pkt->dts = pkt->pts + pkt->dts + ost->last_mux_dts + 1
 718                      - FFMIN3(pkt->pts, pkt->dts, ost->last_mux_dts + 1)
 719                      - FFMAX3(pkt->pts, pkt->dts, ost->last_mux_dts + 1);
 720         }
 721         if ((avctx->codec_type == AVMEDIA_TYPE_AUDIO || avctx->codec_type == AVMEDIA_TYPE_VIDEO) &&
 722             pkt->dts != AV_NOPTS_VALUE &&
 723             !(avctx->codec_id == AV_CODEC_ID_VP9 && ost->stream_copy) &&
 724             ost->last_mux_dts != AV_NOPTS_VALUE) {
 725             int64_t max = ost->last_mux_dts + !(s->oformat->flags & AVFMT_TS_NONSTRICT);
 726             if (pkt->dts < max) {
 727                 int loglevel = max - pkt->dts > 2 || avctx->codec_type == AVMEDIA_TYPE_VIDEO ? AV_LOG_WARNING : AV_LOG_DEBUG;
 728                 av_log(s, loglevel, "Non-monotonous DTS in output stream "
 729                        "%d:%d; previous: %"PRId64", current: %"PRId64"; ",
 730                        ost->file_index, ost->st->index, ost->last_mux_dts, pkt->dts);
 731                 if (exit_on_error) {
 732                     av_log(NULL, AV_LOG_FATAL, "aborting.\n");
 733                     exit_program(1);
 734                 }
 735                 av_log(s, loglevel, "changing to %"PRId64". This may result "
 736                        "in incorrect timestamps in the output file.\n",
 737                        max);
 738                 if (pkt->pts >= pkt->dts)
 739                     pkt->pts = FFMAX(pkt->pts, max);
 740                 pkt->dts = max;
 741             }
 742         }
 743     }
 744     ost->last_mux_dts = pkt->dts;
 745
 746     ost->data_size += pkt->size;
 747     ost->packets_written++;
 748
 749     pkt->stream_index = ost->index;
 750
 751     if (debug_ts) {
 752         av_log(NULL, AV_LOG_INFO, "muxer <- type:%s "
 753                 "pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s size:%d\n",
 754                 av_get_media_type_string(ost->enc_ctx->codec_type),
 755                 av_ts2str(pkt->pts), av_ts2timestr(pkt->pts, &ost->st->time_base),
 756                 av_ts2str(pkt->dts), av_ts2timestr(pkt->dts, &ost->st->time_base),
 757                 pkt->size
 758               );
 759     }
 760
 761     ret = av_interleaved_write_frame(s, pkt);
 762     if (ret < 0) {
 763         print_error("av_interleaved_write_frame()", ret);
 764         main_return_code = 1;
 765         close_all_output_streams(ost, MUXER_FINISHED | ENCODER_FINISHED, ENCODER_FINISHED);
 766     }
 767     av_packet_unref(pkt);
 768 }
 769
 770 static void close_output_stream(OutputStream *ost)
 771 {
 772     OutputFile *of = output_files[ost->file_index];
 773
 774     ost->finished |= ENCODER_FINISHED;
 775     if (of->shortest) {
 776         int64_t end = av_rescale_q(ost->sync_opts - ost->first_pts, ost->enc_ctx->time_base, AV_TIME_BASE_Q);
 777         of->recording_time = FFMIN(of->recording_time, end);
 778     }
 779 }
 780
 781 static int check_recording_time(OutputStream *ost)
 782 {
 783     OutputFile *of = output_files[ost->file_index];
 784
 785     if (of->recording_time != INT64_MAX &&
 786         av_compare_ts(ost->sync_opts - ost->first_pts, ost->enc_ctx->time_base, of->recording_time,
 787                       AV_TIME_BASE_Q) >= 0) {
 788         close_output_stream(ost);
 789         return 0;
 790     }
 791     return 1;
 792 }
 793
 794 static void do_audio_out(AVFormatContext *s, OutputStream *ost,
 795                          AVFrame *frame)
 796 {
 797     AVCodecContext *enc = ost->enc_ctx;
 798     AVPacket pkt;
 799     int got_packet = 0;
 800
 801     av_init_packet(&pkt);
 802     pkt.data = NULL;
 803     pkt.size = 0;
 804
 805     if (!check_recording_time(ost))
 806         return;
 807
 808     if (frame->pts == AV_NOPTS_VALUE || audio_sync_method < 0)
 809         frame->pts = ost->sync_opts;
 810     ost->sync_opts = frame->pts + frame->nb_samples;
 811     ost->samples_encoded += frame->nb_samples;
 812     ost->frames_encoded++;
 813
 814     av_assert0(pkt.size || !pkt.data);
 815     update_benchmark(NULL);
 816     if (debug_ts) {
 817         av_log(NULL, AV_LOG_INFO, "encoder <- type:audio "
 818                "frame_pts:%s frame_pts_time:%s time_base:%d/%d\n",
 819                av_ts2str(frame->pts), av_ts2timestr(frame->pts, &enc->time_base),
 820                enc->time_base.num, enc->time_base.den);
 821     }
 822
 823     if (avcodec_encode_audio2(enc, &pkt, frame, &got_packet) < 0) {
 824         av_log(NULL, AV_LOG_FATAL, "Audio encoding failed (avcodec_encode_audio2)\n");
 825         exit_program(1);
 826     }
 827     update_benchmark("encode_audio %d.%d", ost->file_index, ost->index);
 828
 829     if (got_packet) {
 830         av_packet_rescale_ts(&pkt, enc->time_base, ost->st->time_base);
 831
 832         if (debug_ts) {
 833             av_log(NULL, AV_LOG_INFO, "encoder -> type:audio "
 834                    "pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s\n",
 835                    av_ts2str(pkt.pts), av_ts2timestr(pkt.pts, &ost->st->time_base),
 836                    av_ts2str(pkt.dts), av_ts2timestr(pkt.dts, &ost->st->time_base));
 837         }
 838
 839         write_frame(s, &pkt, ost);
 840     }
 841 }
 842
 843 static void do_subtitle_out(AVFormatContext *s,
 844                             OutputStream *ost,
 845                             InputStream *ist,
 846                             AVSubtitle *sub)
 847 {
 848     int subtitle_out_max_size = 1024 * 1024;
 849     int subtitle_out_size, nb, i;
 850     AVCodecContext *enc;
 851     AVPacket pkt;
 852     int64_t pts;
 853
 854     if (sub->pts == AV_NOPTS_VALUE) {
 855         av_log(NULL, AV_LOG_ERROR, "Subtitle packets must have a pts\n");
 856         if (exit_on_error)
 857             exit_program(1);
 858         return;
 859     }
 860
 861     enc = ost->enc_ctx;
 862
 863     if (!subtitle_out) {
 864         subtitle_out = av_malloc(subtitle_out_max_size);
 865         if (!subtitle_out) {
 866             av_log(NULL, AV_LOG_FATAL, "Failed to allocate subtitle_out\n");
 867             exit_program(1);
 868         }
 869     }
 870
 871     /* Note: DVB subtitle need one packet to draw them and one other
 872        packet to clear them */
 873     /* XXX: signal it in the codec context ? */
 874     if (enc->codec_id == AV_CODEC_ID_DVB_SUBTITLE)
 875         nb = 2;
 876     else
 877         nb = 1;
 878
 879     /* shift timestamp to honor -ss and make check_recording_time() work with -t */
 880     pts = sub->pts;
 881     if (output_files[ost->file_index]->start_time != AV_NOPTS_VALUE)
 882         pts -= output_files[ost->file_index]->start_time;
 883     for (i = 0; i < nb; i++) {
 884         unsigned save_num_rects = sub->num_rects;
 885
 886         ost->sync_opts = av_rescale_q(pts, AV_TIME_BASE_Q, enc->time_base);
 887         if (!check_recording_time(ost))
 888             return;
 889
 890         sub->pts = pts;
 891         // start_display_time is required to be 0
 892         sub->pts               += av_rescale_q(sub->start_display_time, (AVRational){ 1, 1000 }, AV_TIME_BASE_Q);
 893         sub->end_display_time  -= sub->start_display_time;
 894         sub->start_display_time = 0;
 895         if (i == 1)
 896             sub->num_rects = 0;
 897
 898         ost->frames_encoded++;
 899
 900         subtitle_out_size = avcodec_encode_subtitle(enc, subtitle_out,
 901                                                     subtitle_out_max_size, sub);
 902         if (i == 1)
 903             sub->num_rects = save_num_rects;
 904         if (subtitle_out_size < 0) {
 905             av_log(NULL, AV_LOG_FATAL, "Subtitle encoding failed\n");
 906             exit_program(1);
 907         }
 908
 909         av_init_packet(&pkt);
 910         pkt.data = subtitle_out;
 911         pkt.size = subtitle_out_size;
 912         pkt.pts  = av_rescale_q(sub->pts, AV_TIME_BASE_Q, ost->st->time_base);
 913         pkt.duration = av_rescale_q(sub->end_display_time, (AVRational){ 1, 1000 }, ost->st->time_base);
 914         if (enc->codec_id == AV_CODEC_ID_DVB_SUBTITLE) {
 915             /* XXX: the pts correction is handled here. Maybe handling
 916                it in the codec would be better */
 917             if (i == 0)
 918                 pkt.pts += 90 * sub->start_display_time;
 919             else
 920                 pkt.pts += 90 * sub->end_display_time;
 921         }
 922         pkt.dts = pkt.pts;
 923         write_frame(s, &pkt, ost);
 924     }
 925 }
 926
 927 static void do_video_out(AVFormatContext *s,
 928                          OutputStream *ost,
 929                          AVFrame *next_picture,
 930                          double sync_ipts)
 931 {
 932     int ret, format_video_sync;
 933     AVPacket pkt;
 934     AVCodecContext *enc = ost->enc_ctx;
 935     AVCodecContext *mux_enc = ost->st->codec;
 936     int nb_frames, nb0_frames, i;
 937     double delta, delta0;
 938     double duration = 0;
 939     int frame_size = 0;
 940     InputStream *ist = NULL;
 941     AVFilterContext *filter = ost->filter->filter;
 942
 943     if (ost->source_index >= 0)
 944         ist = input_streams[ost->source_index];
 945
 946     if (filter->inputs[0]->frame_rate.num > 0 &&
 947         filter->inputs[0]->frame_rate.den > 0)
 948         duration = 1/(av_q2d(filter->inputs[0]->frame_rate) * av_q2d(enc->time_base));
 949
 950     if(ist && ist->st->start_time != AV_NOPTS_VALUE && ist->st->first_dts != AV_NOPTS_VALUE && ost->frame_rate.num)
 951         duration = FFMIN(duration, 1/(av_q2d(ost->frame_rate) * av_q2d(enc->time_base)));
 952
 953     if (!ost->filters_script &&
 954         !ost->filters &&
 955         next_picture &&
 956         ist &&
 957         lrintf(av_frame_get_pkt_duration(next_picture) * av_q2d(ist->st->time_base) / av_q2d(enc->time_base)) > 0) {
 958         duration = lrintf(av_frame_get_pkt_duration(next_picture) * av_q2d(ist->st->time_base) / av_q2d(enc->time_base));
 959     }
 960
 961     if (!next_picture) {
 962         //end, flushing
 963         nb0_frames = nb_frames = mid_pred(ost->last_nb0_frames[0],
 964                                           ost->last_nb0_frames[1],
 965                                           ost->last_nb0_frames[2]);
 966     } else {
 967         delta0 = sync_ipts - ost->sync_opts; // delta0 is the "drift" between the input frame (next_picture) and where it would fall in the output.
 968         delta  = delta0 + duration;
 969
 970         /* by default, we output a single frame */
 971         nb0_frames = 0; // tracks the number of times the PREVIOUS frame should be duplicated, mostly for variable framerate (VFR)
 972         nb_frames = 1;
 973
 974         format_video_sync = video_sync_method;
 975         if (format_video_sync == VSYNC_AUTO) {
 976             if(!strcmp(s->oformat->name, "avi")) {
 977                 format_video_sync = VSYNC_VFR;
 978             } else
 979                 format_video_sync = (s->oformat->flags & AVFMT_VARIABLE_FPS) ? ((s->oformat->flags & AVFMT_NOTIMESTAMPS) ? VSYNC_PASSTHROUGH : VSYNC_VFR) : VSYNC_CFR;
 980             if (   ist
 981                 && format_video_sync == VSYNC_CFR
 982                 && input_files[ist->file_index]->ctx->nb_streams == 1
 983                 && input_files[ist->file_index]->input_ts_offset == 0) {
 984                 format_video_sync = VSYNC_VSCFR;
 985             }
 986             if (format_video_sync == VSYNC_CFR && copy_ts) {
 987                 format_video_sync = VSYNC_VSCFR;
 988             }
 989         }
 990         ost->is_cfr = (format_video_sync == VSYNC_CFR || format_video_sync == VSYNC_VSCFR);
 991
 992         if (delta0 < 0 &&
 993             delta > 0 &&
 994             format_video_sync != VSYNC_PASSTHROUGH &&
 995             format_video_sync != VSYNC_DROP) {
 996             if (delta0 < -0.6) {
 997                 av_log(NULL, AV_LOG_WARNING, "Past duration %f too large\n", -delta0);
 998             } else
 999                 av_log(NULL, AV_LOG_DEBUG, "Clipping frame in rate conversion by %f\n", -delta0);
1000             sync_ipts = ost->sync_opts;
1001             duration += delta0;
1002             delta0 = 0;
1003         }
1004
1005         switch (format_video_sync) {
1006         case VSYNC_VSCFR:
1007             if (ost->frame_number == 0 && delta0 >= 0.5) {
1008                 av_log(NULL, AV_LOG_DEBUG, "Not duplicating %d initial frames\n", (int)lrintf(delta0));
1009                 delta = duration;
1010                 delta0 = 0;
1011                 ost->sync_opts = lrint(sync_ipts);
1012             }
1013         case VSYNC_CFR:
1014             // FIXME set to 0.5 after we fix some dts/pts bugs like in avidec.c
1015             if (frame_drop_threshold && delta < frame_drop_threshold && ost->frame_number) {
1016                 nb_frames = 0;
1017             } else if (delta < -1.1)
1018                 nb_frames = 0;
1019             else if (delta > 1.1) {
1020                 nb_frames = lrintf(delta);
1021                 if (delta0 > 1.1)
1022                     nb0_frames = lrintf(delta0 - 0.6);
1023             }
1024             break;
1025         case VSYNC_VFR:
1026             if (delta <= -0.6)
1027                 nb_frames = 0;
1028             else if (delta > 0.6)
1029                 ost->sync_opts = lrint(sync_ipts);
1030             break;
1031         case VSYNC_DROP:
1032         case VSYNC_PASSTHROUGH:
1033             ost->sync_opts = lrint(sync_ipts);
1034             break;
1035         default:
1036             av_assert0(0);
1037         }
1038     }
1039
1040     nb_frames = FFMIN(nb_frames, ost->max_frames - ost->frame_number);
1041     nb0_frames = FFMIN(nb0_frames, nb_frames);
1042
1043     memmove(ost->last_nb0_frames + 1,
1044             ost->last_nb0_frames,
1045             sizeof(ost->last_nb0_frames[0]) * (FF_ARRAY_ELEMS(ost->last_nb0_frames) - 1));
1046     ost->last_nb0_frames[0] = nb0_frames;
1047
1048     if (nb0_frames == 0 && ost->last_dropped) {
1049         nb_frames_drop++;
1050         av_log(NULL, AV_LOG_VERBOSE,
1051                "*** dropping frame %d from stream %d at ts %"PRId64"\n",
1052                ost->frame_number, ost->st->index, ost->last_frame->pts);
1053     }
1054     if (nb_frames > (nb0_frames && ost->last_dropped) + (nb_frames > nb0_frames)) {
1055         if (nb_frames > dts_error_threshold * 30) {
1056             av_log(NULL, AV_LOG_ERROR, "%d frame duplication too large, skipping\n", nb_frames - 1);
1057             nb_frames_drop++;
1058             return;
1059         }
1060         nb_frames_dup += nb_frames - (nb0_frames && ost->last_dropped) - (nb_frames > nb0_frames);
1061         av_log(NULL, AV_LOG_VERBOSE, "*** %d dup!\n", nb_frames - 1);
1062     }
1063     ost->last_dropped = nb_frames == nb0_frames && next_picture;
1064
1065   /* duplicates frame if needed */
1066   for (i = 0; i < nb_frames; i++) {
1067     AVFrame *in_picture;
1068     av_init_packet(&pkt);
1069     pkt.data = NULL;
1070     pkt.size = 0;
1071
1072     if (i < nb0_frames && ost->last_frame) {
1073         in_picture = ost->last_frame;
1074     } else
1075         in_picture = next_picture;
1076
1077     if (!in_picture)
1078         return;
1079
1080     in_picture->pts = ost->sync_opts;
1081
1082 #if 1
1083     if (!check_recording_time(ost))
1084 #else
1085     if (ost->frame_number >= ost->max_frames)
1086 #endif
1087         return;
1088
1089 #if FF_API_LAVF_FMT_RAWPICTURE
1090     if (s->oformat->flags & AVFMT_RAWPICTURE &&
1091         enc->codec->id == AV_CODEC_ID_RAWVIDEO) {
1092         /* raw pictures are written as AVPicture structure to
1093            avoid any copies. We support temporarily the older
1094            method. */
1095         if (in_picture->interlaced_frame)
1096             mux_enc->field_order = in_picture->top_field_first ? AV_FIELD_TB:AV_FIELD_BT;
1097         else
1098             mux_enc->field_order = AV_FIELD_PROGRESSIVE;
1099         pkt.data   = (uint8_t *)in_picture;
1100         pkt.size   =  sizeof(AVPicture);
1101         pkt.pts    = av_rescale_q(in_picture->pts, enc->time_base, ost->st->time_base);
1102         pkt.flags |= AV_PKT_FLAG_KEY;
1103
1104         write_frame(s, &pkt, ost);
1105     } else
1106 #endif
1107     {
1108         int got_packet, forced_keyframe = 0;
1109         double pts_time;
1110
1111         if (enc->flags & (AV_CODEC_FLAG_INTERLACED_DCT | AV_CODEC_FLAG_INTERLACED_ME) &&
1112             ost->top_field_first >= 0)
1113             in_picture->top_field_first = !!ost->top_field_first;
1114
1115         if (in_picture->interlaced_frame) {
1116             if (enc->codec->id == AV_CODEC_ID_MJPEG)
1117                 mux_enc->field_order = in_picture->top_field_first ? AV_FIELD_TT:AV_FIELD_BB;
1118             else
1119                 mux_enc->field_order = in_picture->top_field_first ? AV_FIELD_TB:AV_FIELD_BT;
1120         } else
1121             mux_enc->field_order = AV_FIELD_PROGRESSIVE;
1122
1123         in_picture->quality = enc->global_quality;
1124         in_picture->pict_type = 0;
1125
1126         pts_time = in_picture->pts != AV_NOPTS_VALUE ?
1127             in_picture->pts * av_q2d(enc->time_base) : NAN;
1128         if (ost->forced_kf_index < ost->forced_kf_count &&
1129             in_picture->pts >= ost->forced_kf_pts[ost->forced_kf_index]) {
1130             ost->forced_kf_index++;
1131             forced_keyframe = 1;
1132         } else if (ost->forced_keyframes_pexpr) {
1133             double res;
1134             ost->forced_keyframes_expr_const_values[FKF_T] = pts_time;
1135             res = av_expr_eval(ost->forced_keyframes_pexpr,
1136                                ost->forced_keyframes_expr_const_values, NULL);
1137             ff_dlog(NULL, "force_key_frame: n:%f n_forced:%f prev_forced_n:%f t:%f prev_forced_t:%f -> res:%f\n",
1138                     ost->forced_keyframes_expr_const_values[FKF_N],
1139                     ost->forced_keyframes_expr_const_values[FKF_N_FORCED],
1140                     ost->forced_keyframes_expr_const_values[FKF_PREV_FORCED_N],
1141                     ost->forced_keyframes_expr_const_values[FKF_T],
1142                     ost->forced_keyframes_expr_const_values[FKF_PREV_FORCED_T],
1143                     res);
1144             if (res) {
1145                 forced_keyframe = 1;
1146                 ost->forced_keyframes_expr_const_values[FKF_PREV_FORCED_N] =
1147                     ost->forced_keyframes_expr_const_values[FKF_N];
1148                 ost->forced_keyframes_expr_const_values[FKF_PREV_FORCED_T] =
1149                     ost->forced_keyframes_expr_const_values[FKF_T];
1150                 ost->forced_keyframes_expr_const_values[FKF_N_FORCED] += 1;
1151             }
1152
1153             ost->forced_keyframes_expr_const_values[FKF_N] += 1;
1154         } else if (   ost->forced_keyframes
1155                    && !strncmp(ost->forced_keyframes, "source", 6)
1156                    && in_picture->key_frame==1) {
1157             forced_keyframe = 1;
1158         }
1159
1160         if (forced_keyframe) {
1161             in_picture->pict_type = AV_PICTURE_TYPE_I;
1162             av_log(NULL, AV_LOG_DEBUG, "Forced keyframe at time %f\n", pts_time);
1163         }
1164
1165         update_benchmark(NULL);
1166         if (debug_ts) {
1167             av_log(NULL, AV_LOG_INFO, "encoder <- type:video "
1168                    "frame_pts:%s frame_pts_time:%s time_base:%d/%d\n",
1169                    av_ts2str(in_picture->pts), av_ts2timestr(in_picture->pts, &enc->time_base),
1170                    enc->time_base.num, enc->time_base.den);
1171         }
1172
1173         ost->frames_encoded++;
1174
1175         ret = avcodec_encode_video2(enc, &pkt, in_picture, &got_packet);
1176         update_benchmark("encode_video %d.%d", ost->file_index, ost->index);
1177         if (ret < 0) {
1178             av_log(NULL, AV_LOG_FATAL, "Video encoding failed\n");
1179             exit_program(1);
1180         }
1181
1182         if (got_packet) {
1183             if (debug_ts) {
1184                 av_log(NULL, AV_LOG_INFO, "encoder -> type:video "
1185                        "pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s\n",
1186                        av_ts2str(pkt.pts), av_ts2timestr(pkt.pts, &enc->time_base),
1187                        av_ts2str(pkt.dts), av_ts2timestr(pkt.dts, &enc->time_base));
1188             }
1189
1190             if (pkt.pts == AV_NOPTS_VALUE && !(enc->codec->capabilities & AV_CODEC_CAP_DELAY))
1191                 pkt.pts = ost->sync_opts;
1192
1193             av_packet_rescale_ts(&pkt, enc->time_base, ost->st->time_base);
1194
1195             if (debug_ts) {
1196                 av_log(NULL, AV_LOG_INFO, "encoder -> type:video "
1197                     "pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s\n",
1198                     av_ts2str(pkt.pts), av_ts2timestr(pkt.pts, &ost->st->time_base),
1199                     av_ts2str(pkt.dts), av_ts2timestr(pkt.dts, &ost->st->time_base));
1200             }
1201
1202             frame_size = pkt.size;
1203             write_frame(s, &pkt, ost);
1204
1205             /* if two pass, output log */
1206             if (ost->logfile && enc->stats_out) {
1207                 fprintf(ost->logfile, "%s", enc->stats_out);
1208             }
1209         }
1210     }
1211     ost->sync_opts++;
1212     /*
1213      * For video, number of frames in == number of packets out.
1214      * But there may be reordering, so we can't throw away frames on encoder
1215      * flush, we need to limit them here, before they go into encoder.
1216      */
1217     ost->frame_number++;
1218
1219     if (vstats_filename && frame_size)
1220         do_video_stats(ost, frame_size);
1221   }
1222
1223     if (!ost->last_frame)
1224         ost->last_frame = av_frame_alloc();
1225     av_frame_unref(ost->last_frame);
1226     if (next_picture && ost->last_frame)
1227         av_frame_ref(ost->last_frame, next_picture);
1228     else
1229         av_frame_free(&ost->last_frame);
1230 }
1231
1232 static double psnr(double d)
1233 {
1234     return -10.0 * log10(d);
1235 }
1236
1237 static void do_video_stats(OutputStream *ost, int frame_size)
1238 {
1239     AVCodecContext *enc;
1240     int frame_number;
1241     double ti1, bitrate, avg_bitrate;
1242
1243     /* this is executed just the first time do_video_stats is called */
1244     if (!vstats_file) {
1245         vstats_file = fopen(vstats_filename, "w");
1246         if (!vstats_file) {
1247             perror("fopen");
1248             exit_program(1);
1249         }
1250     }
1251
1252     enc = ost->enc_ctx;
1253     if (enc->codec_type == AVMEDIA_TYPE_VIDEO) {
1254         frame_number = ost->st->nb_frames;
1255         fprintf(vstats_file, "frame= %5d q= %2.1f ", frame_number,
1256                 ost->quality / (float)FF_QP2LAMBDA);
1257
1258         if (ost->error[0]>=0 && (enc->flags & AV_CODEC_FLAG_PSNR))
1259             fprintf(vstats_file, "PSNR= %6.2f ", psnr(ost->error[0] / (enc->width * enc->height * 255.0 * 255.0)));
1260
1261         fprintf(vstats_file,"f_size= %6d ", frame_size);
1262         /* compute pts value */
1263         ti1 = av_stream_get_end_pts(ost->st) * av_q2d(ost->st->time_base);
1264         if (ti1 < 0.01)
1265             ti1 = 0.01;
1266
1267         bitrate     = (frame_size * 8) / av_q2d(enc->time_base) / 1000.0;
1268         avg_bitrate = (double)(ost->data_size * 8) / ti1 / 1000.0;
1269         fprintf(vstats_file, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",
1270                (double)ost->data_size / 1024, ti1, bitrate, avg_bitrate);
1271         fprintf(vstats_file, "type= %c\n", av_get_picture_type_char(ost->pict_type));
1272     }
1273 }
1274
1275 static void finish_output_stream(OutputStream *ost)
1276 {
1277     OutputFile *of = output_files[ost->file_index];
1278     int i;
1279
1280     ost->finished = ENCODER_FINISHED | MUXER_FINISHED;
1281
1282     if (of->shortest) {
1283         for (i = 0; i < of->ctx->nb_streams; i++)
1284             output_streams[of->ost_index + i]->finished = ENCODER_FINISHED | MUXER_FINISHED;
1285     }
1286 }
1287
1288 /**
1289  * Get and encode new output from any of the filtergraphs, without causing
1290  * activity.
1291  *
1292  * @return  0 for success, <0 for severe errors
1293  */
1294 static int reap_filters(int flush)
1295 {
1296     AVFrame *filtered_frame = NULL;
1297     int i;
1298
1299     /* Reap all buffers present in the buffer sinks */
1300     for (i = 0; i < nb_output_streams; i++) {
1301         OutputStream *ost = output_streams[i];
1302         OutputFile    *of = output_files[ost->file_index];
1303         AVFilterContext *filter;
1304         AVCodecContext *enc = ost->enc_ctx;
1305         int ret = 0;
1306
1307         if (!ost->filter)
1308             continue;
1309         filter = ost->filter->filter;
1310
1311         if (!ost->filtered_frame && !(ost->filtered_frame = av_frame_alloc())) {
1312             return AVERROR(ENOMEM);
1313         }
1314         filtered_frame = ost->filtered_frame;
1315
1316         while (1) {
1317             double float_pts = AV_NOPTS_VALUE; // this is identical to filtered_frame.pts but with higher precision
1318             ret = av_buffersink_get_frame_flags(filter, filtered_frame,
1319                                                AV_BUFFERSINK_FLAG_NO_REQUEST);
1320             if (ret < 0) {
1321                 if (ret != AVERROR(EAGAIN) && ret != AVERROR_EOF) {
1322                     av_log(NULL, AV_LOG_WARNING,
1323                            "Error in av_buffersink_get_frame_flags(): %s\n", av_err2str(ret));
1324                 } else if (flush && ret == AVERROR_EOF) {
1325                     if (filter->inputs[0]->type == AVMEDIA_TYPE_VIDEO)
1326                         do_video_out(of->ctx, ost, NULL, AV_NOPTS_VALUE);
1327                 }
1328                 break;
1329             }
1330             if (ost->finished) {
1331                 av_frame_unref(filtered_frame);
1332                 continue;
1333             }
1334             if (filtered_frame->pts != AV_NOPTS_VALUE) {
1335                 int64_t start_time = (of->start_time == AV_NOPTS_VALUE) ? 0 : of->start_time;
1336                 AVRational tb = enc->time_base;
1337                 int extra_bits = av_clip(29 - av_log2(tb.den), 0, 16);
1338
1339                 tb.den <<= extra_bits;
1340                 float_pts =
1341                     av_rescale_q(filtered_frame->pts, filter->inputs[0]->time_base, tb) -
1342                     av_rescale_q(start_time, AV_TIME_BASE_Q, tb);
1343                 float_pts /= 1 << extra_bits;
1344                 // avoid exact midoints to reduce the chance of rounding differences, this can be removed in case the fps code is changed to work with integers
1345                 float_pts += FFSIGN(float_pts) * 1.0 / (1<<17);
1346
1347                 filtered_frame->pts =
1348                     av_rescale_q(filtered_frame->pts, filter->inputs[0]->time_base, enc->time_base) -
1349                     av_rescale_q(start_time, AV_TIME_BASE_Q, enc->time_base);
1350             }
1351             //if (ost->source_index >= 0)
1352             //    *filtered_frame= *input_streams[ost->source_index]->decoded_frame; //for me_threshold
1353
1354             switch (filter->inputs[0]->type) {
1355             case AVMEDIA_TYPE_VIDEO:
1356                 if (!ost->frame_aspect_ratio.num)
1357                     enc->sample_aspect_ratio = filtered_frame->sample_aspect_ratio;
1358
1359                 if (debug_ts) {
1360                     av_log(NULL, AV_LOG_INFO, "filter -> pts:%s pts_time:%s exact:%f time_base:%d/%d\n",
1361                             av_ts2str(filtered_frame->pts), av_ts2timestr(filtered_frame->pts, &enc->time_base),
1362                             float_pts,
1363                             enc->time_base.num, enc->time_base.den);
1364                 }
1365
1366                 do_video_out(of->ctx, ost, filtered_frame, float_pts);
1367                 break;
1368             case AVMEDIA_TYPE_AUDIO:
1369                 if (!(enc->codec->capabilities & AV_CODEC_CAP_PARAM_CHANGE) &&
1370                     enc->channels != av_frame_get_channels(filtered_frame)) {
1371                     av_log(NULL, AV_LOG_ERROR,
1372                            "Audio filter graph output is not normalized and encoder does not support parameter changes\n");
1373                     break;
1374                 }
1375                 do_audio_out(of->ctx, ost, filtered_frame);
1376                 break;
1377             default:
1378                 // TODO support subtitle filters
1379                 av_assert0(0);
1380             }
1381
1382             av_frame_unref(filtered_frame);
1383         }
1384     }
1385
1386     return 0;
1387 }
1388
1389 static void print_final_stats(int64_t total_size)
1390 {
1391     uint64_t video_size = 0, audio_size = 0, extra_size = 0, other_size = 0;
1392     uint64_t subtitle_size = 0;
1393     uint64_t data_size = 0;
1394     float percent = -1.0;
1395     int i, j;
1396     int pass1_used = 1;
1397
1398     for (i = 0; i < nb_output_streams; i++) {
1399         OutputStream *ost = output_streams[i];
1400         switch (ost->enc_ctx->codec_type) {
1401             case AVMEDIA_TYPE_VIDEO: video_size += ost->data_size; break;
1402             case AVMEDIA_TYPE_AUDIO: audio_size += ost->data_size; break;
1403             case AVMEDIA_TYPE_SUBTITLE: subtitle_size += ost->data_size; break;
1404             default:                 other_size += ost->data_size; break;
1405         }
1406         extra_size += ost->enc_ctx->extradata_size;
1407         data_size  += ost->data_size;
1408         if (   (ost->enc_ctx->flags & (AV_CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))
1409             != AV_CODEC_FLAG_PASS1)
1410             pass1_used = 0;
1411     }
1412
1413     if (data_size && total_size>0 && total_size >= data_size)
1414         percent = 100.0 * (total_size - data_size) / data_size;
1415
1416     av_log(NULL, AV_LOG_INFO, "video:%1.0fkB audio:%1.0fkB subtitle:%1.0fkB other streams:%1.0fkB global headers:%1.0fkB muxing overhead: ",
1417            video_size / 1024.0,
1418            audio_size / 1024.0,
1419            subtitle_size / 1024.0,
1420            other_size / 1024.0,
1421            extra_size / 1024.0);
1422     if (percent >= 0.0)
1423         av_log(NULL, AV_LOG_INFO, "%f%%", percent);
1424     else
1425         av_log(NULL, AV_LOG_INFO, "unknown");
1426     av_log(NULL, AV_LOG_INFO, "\n");
1427
1428     /* print verbose per-stream stats */
1429     for (i = 0; i < nb_input_files; i++) {
1430         InputFile *f = input_files[i];
1431         uint64_t total_packets = 0, total_size = 0;
1432
1433         av_log(NULL, AV_LOG_VERBOSE, "Input file #%d (%s):\n",
1434                i, f->ctx->filename);
1435
1436         for (j = 0; j < f->nb_streams; j++) {
1437             InputStream *ist = input_streams[f->ist_index + j];
1438             enum AVMediaType type = ist->dec_ctx->codec_type;
1439
1440             total_size    += ist->data_size;
1441             total_packets += ist->nb_packets;
1442
1443             av_log(NULL, AV_LOG_VERBOSE, "  Input stream #%d:%d (%s): ",
1444                    i, j, media_type_string(type));
1445             av_log(NULL, AV_LOG_VERBOSE, "%"PRIu64" packets read (%"PRIu64" bytes); ",
1446                    ist->nb_packets, ist->data_size);
1447
1448             if (ist->decoding_needed) {
1449                 av_log(NULL, AV_LOG_VERBOSE, "%"PRIu64" frames decoded",
1450                        ist->frames_decoded);
1451                 if (type == AVMEDIA_TYPE_AUDIO)
1452                     av_log(NULL, AV_LOG_VERBOSE, " (%"PRIu64" samples)", ist->samples_decoded);
1453                 av_log(NULL, AV_LOG_VERBOSE, "; ");
1454             }
1455
1456             av_log(NULL, AV_LOG_VERBOSE, "\n");
1457         }
1458
1459         av_log(NULL, AV_LOG_VERBOSE, "  Total: %"PRIu64" packets (%"PRIu64" bytes) demuxed\n",
1460                total_packets, total_size);
1461     }
1462
1463     for (i = 0; i < nb_output_files; i++) {
1464         OutputFile *of = output_files[i];
1465         uint64_t total_packets = 0, total_size = 0;
1466
1467         av_log(NULL, AV_LOG_VERBOSE, "Output file #%d (%s):\n",
1468                i, of->ctx->filename);
1469
1470         for (j = 0; j < of->ctx->nb_streams; j++) {
1471             OutputStream *ost = output_streams[of->ost_index + j];
1472             enum AVMediaType type = ost->enc_ctx->codec_type;
1473
1474             total_size    += ost->data_size;
1475             total_packets += ost->packets_written;
1476
1477             av_log(NULL, AV_LOG_VERBOSE, "  Output stream #%d:%d (%s): ",
1478                    i, j, media_type_string(type));
1479             if (ost->encoding_needed) {
1480                 av_log(NULL, AV_LOG_VERBOSE, "%"PRIu64" frames encoded",
1481                        ost->frames_encoded);
1482                 if (type == AVMEDIA_TYPE_AUDIO)
1483                     av_log(NULL, AV_LOG_VERBOSE, " (%"PRIu64" samples)", ost->samples_encoded);
1484                 av_log(NULL, AV_LOG_VERBOSE, "; ");
1485             }
1486
1487             av_log(NULL, AV_LOG_VERBOSE, "%"PRIu64" packets muxed (%"PRIu64" bytes); ",
1488                    ost->packets_written, ost->data_size);
1489
1490             av_log(NULL, AV_LOG_VERBOSE, "\n");
1491         }
1492
1493         av_log(NULL, AV_LOG_VERBOSE, "  Total: %"PRIu64" packets (%"PRIu64" bytes) muxed\n",
1494                total_packets, total_size);
1495     }
1496     if(video_size + data_size + audio_size + subtitle_size + extra_size == 0){
1497         av_log(NULL, AV_LOG_WARNING, "Output file is empty, nothing was encoded ");
1498         if (pass1_used) {
1499             av_log(NULL, AV_LOG_WARNING, "\n");
1500         } else {
1501             av_log(NULL, AV_LOG_WARNING, "(check -ss / -t / -frames parameters if used)\n");
1502         }
1503     }
1504 }
1505
1506 static void print_report(int is_last_report, int64_t timer_start, int64_t cur_time)
1507 {
1508     char buf[1024];
1509     AVBPrint buf_script;
1510     OutputStream *ost;
1511     AVFormatContext *oc;
1512     int64_t total_size;
1513     AVCodecContext *enc;
1514     int frame_number, vid, i;
1515     double bitrate;
1516     double speed;
1517     int64_t pts = INT64_MIN + 1;
1518     static int64_t last_time = -1;
1519     static int qp_histogram[52];
1520     int hours, mins, secs, us;
1521     int ret;
1522     float t;
1523
1524     if (!print_stats && !is_last_report && !progress_avio)
1525         return;
1526
1527     if (!is_last_report) {
1528         if (last_time == -1) {
1529             last_time = cur_time;
1530             return;
1531         }
1532         if ((cur_time - last_time) < 500000)
1533             return;
1534         last_time = cur_time;
1535     }
1536
1537     t = (cur_time-timer_start) / 1000000.0;
1538
1539
1540     oc = output_files[0]->ctx;
1541
1542     total_size = avio_size(oc->pb);
1543     if (total_size <= 0) // FIXME improve avio_size() so it works with non seekable output too
1544         total_size = avio_tell(oc->pb);
1545
1546     buf[0] = '\0';
1547     vid = 0;
1548     av_bprint_init(&buf_script, 0, 1);
1549     for (i = 0; i < nb_output_streams; i++) {
1550         float q = -1;
1551         ost = output_streams[i];
1552         enc = ost->enc_ctx;
1553         if (!ost->stream_copy)
1554             q = ost->quality / (float) FF_QP2LAMBDA;
1555
1556         if (vid && enc->codec_type == AVMEDIA_TYPE_VIDEO) {
1557             snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "q=%2.1f ", q);
1558             av_bprintf(&buf_script, "stream_%d_%d_q=%.1f\n",
1559                        ost->file_index, ost->index, q);
1560         }
1561         if (!vid && enc->codec_type == AVMEDIA_TYPE_VIDEO) {
1562             float fps;
1563
1564             frame_number = ost->frame_number;
1565             fps = t > 1 ? frame_number / t : 0;
1566             snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "frame=%5d fps=%3.*f q=%3.1f ",
1567                      frame_number, fps < 9.95, fps, q);
1568             av_bprintf(&buf_script, "frame=%d\n", frame_number);
1569             av_bprintf(&buf_script, "fps=%.1f\n", fps);
1570             av_bprintf(&buf_script, "stream_%d_%d_q=%.1f\n",
1571                        ost->file_index, ost->index, q);
1572             if (is_last_report)
1573                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "L");
1574             if (qp_hist) {
1575                 int j;
1576                 int qp = lrintf(q);
1577                 if (qp >= 0 && qp < FF_ARRAY_ELEMS(qp_histogram))
1578                     qp_histogram[qp]++;
1579                 for (j = 0; j < 32; j++)
1580                     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%X", av_log2(qp_histogram[j] + 1));
1581             }
1582
1583             if ((enc->flags & AV_CODEC_FLAG_PSNR) && (ost->pict_type != AV_PICTURE_TYPE_NONE || is_last_report)) {
1584                 int j;
1585                 double error, error_sum = 0;
1586                 double scale, scale_sum = 0;
1587                 double p;
1588                 char type[3] = { 'Y','U','V' };
1589                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "PSNR=");
1590                 for (j = 0; j < 3; j++) {
1591                     if (is_last_report) {
1592                         error = enc->error[j];
1593                         scale = enc->width * enc->height * 255.0 * 255.0 * frame_number;
1594                     } else {
1595                         error = ost->error[j];
1596                         scale = enc->width * enc->height * 255.0 * 255.0;
1597                     }
1598                     if (j)
1599                         scale /= 4;
1600                     error_sum += error;
1601                     scale_sum += scale;
1602                     p = psnr(error / scale);
1603                     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%c:%2.2f ", type[j], p);
1604                     av_bprintf(&buf_script, "stream_%d_%d_psnr_%c=%2.2f\n",
1605                                ost->file_index, ost->index, type[j] | 32, p);
1606                 }
1607                 p = psnr(error_sum / scale_sum);
1608                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "*:%2.2f ", psnr(error_sum / scale_sum));
1609                 av_bprintf(&buf_script, "stream_%d_%d_psnr_all=%2.2f\n",
1610                            ost->file_index, ost->index, p);
1611             }
1612             vid = 1;
1613         }
1614         /* compute min output value */
1615         if (av_stream_get_end_pts(ost->st) != AV_NOPTS_VALUE)
1616             pts = FFMAX(pts, av_rescale_q(av_stream_get_end_pts(ost->st),
1617                                           ost->st->time_base, AV_TIME_BASE_Q));
1618         if (is_last_report)
1619             nb_frames_drop += ost->last_dropped;
1620     }
1621
1622     secs = FFABS(pts) / AV_TIME_BASE;
1623     us = FFABS(pts) % AV_TIME_BASE;
1624     mins = secs / 60;
1625     secs %= 60;
1626     hours = mins / 60;
1627     mins %= 60;
1628
1629     bitrate = pts && total_size >= 0 ? total_size * 8 / (pts / 1000.0) : -1;
1630     speed = t != 0.0 ? (double)pts / AV_TIME_BASE / t : -1;
1631
1632     if (total_size < 0) snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1633                                  "size=N/A time=");
1634     else                snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1635                                  "size=%8.0fkB time=", total_size / 1024.0);
1636     if (pts < 0)
1637         snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "-");
1638     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1639              "%02d:%02d:%02d.%02d ", hours, mins, secs,
1640              (100 * us) / AV_TIME_BASE);
1641
1642     if (bitrate < 0) {
1643         snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),"bitrate=N/A");
1644         av_bprintf(&buf_script, "bitrate=N/A\n");
1645     }else{
1646         snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),"bitrate=%6.1fkbits/s", bitrate);
1647         av_bprintf(&buf_script, "bitrate=%6.1fkbits/s\n", bitrate);
1648     }
1649
1650     if (total_size < 0) av_bprintf(&buf_script, "total_size=N/A\n");
1651     else                av_bprintf(&buf_script, "total_size=%"PRId64"\n", total_size);
1652     av_bprintf(&buf_script, "out_time_ms=%"PRId64"\n", pts);
1653     av_bprintf(&buf_script, "out_time=%02d:%02d:%02d.%06d\n",
1654                hours, mins, secs, us);
1655
1656     if (nb_frames_dup || nb_frames_drop)
1657         snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), " dup=%d drop=%d",
1658                 nb_frames_dup, nb_frames_drop);
1659     av_bprintf(&buf_script, "dup_frames=%d\n", nb_frames_dup);
1660     av_bprintf(&buf_script, "drop_frames=%d\n", nb_frames_drop);
1661
1662     if (speed < 0) {
1663         snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf)," speed=N/A");
1664         av_bprintf(&buf_script, "speed=N/A\n");
1665     } else {
1666         snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf)," speed=%4.3gx", speed);
1667         av_bprintf(&buf_script, "speed=%4.3gx\n", speed);
1668     }
1669
1670     if (print_stats || is_last_report) {
1671         const char end = is_last_report ? '\n' : '\r';
1672         if (print_stats==1 && AV_LOG_INFO > av_log_get_level()) {
1673             fprintf(stderr, "%s    %c", buf, end);
1674         } else
1675             av_log(NULL, AV_LOG_INFO, "%s    %c", buf, end);
1676
1677     fflush(stderr);
1678     }
1679
1680     if (progress_avio) {
1681         av_bprintf(&buf_script, "progress=%s\n",
1682                    is_last_report ? "end" : "continue");
1683         avio_write(progress_avio, buf_script.str,
1684                    FFMIN(buf_script.len, buf_script.size - 1));
1685         avio_flush(progress_avio);
1686         av_bprint_finalize(&buf_script, NULL);
1687         if (is_last_report) {
1688             if ((ret = avio_closep(&progress_avio)) < 0)
1689                 av_log(NULL, AV_LOG_ERROR,
1690                        "Error closing progress log, loss of information possible: %s\n", av_err2str(ret));
1691         }
1692     }
1693
1694     if (is_last_report)
1695         print_final_stats(total_size);
1696 }
1697
1698 static void flush_encoders(void)
1699 {
1700     int i, ret;
1701
1702     for (i = 0; i < nb_output_streams; i++) {
1703         OutputStream   *ost = output_streams[i];
1704         AVCodecContext *enc = ost->enc_ctx;
1705         AVFormatContext *os = output_files[ost->file_index]->ctx;
1706         int stop_encoding = 0;
1707
1708         if (!ost->encoding_needed)
1709             continue;
1710
1711         if (enc->codec_type == AVMEDIA_TYPE_AUDIO && enc->frame_size <= 1)
1712             continue;
1713 #if FF_API_LAVF_FMT_RAWPICTURE
1714         if (enc->codec_type == AVMEDIA_TYPE_VIDEO && (os->oformat->flags & AVFMT_RAWPICTURE) && enc->codec->id == AV_CODEC_ID_RAWVIDEO)
1715             continue;
1716 #endif
1717
1718         for (;;) {
1719             int (*encode)(AVCodecContext*, AVPacket*, const AVFrame*, int*) = NULL;
1720             const char *desc;
1721
1722             switch (enc->codec_type) {
1723             case AVMEDIA_TYPE_AUDIO:
1724                 encode = avcodec_encode_audio2;
1725                 desc   = "audio";
1726                 break;
1727             case AVMEDIA_TYPE_VIDEO:
1728                 encode = avcodec_encode_video2;
1729                 desc   = "video";
1730                 break;
1731             default:
1732                 stop_encoding = 1;
1733             }
1734
1735             if (encode) {
1736                 AVPacket pkt;
1737                 int pkt_size;
1738                 int got_packet;
1739                 av_init_packet(&pkt);
1740                 pkt.data = NULL;
1741                 pkt.size = 0;
1742
1743                 update_benchmark(NULL);
1744                 ret = encode(enc, &pkt, NULL, &got_packet);
1745                 update_benchmark("flush_%s %d.%d", desc, ost->file_index, ost->index);
1746                 if (ret < 0) {
1747                     av_log(NULL, AV_LOG_FATAL, "%s encoding failed: %s\n",
1748                            desc,
1749                            av_err2str(ret));
1750                     exit_program(1);
1751                 }
1752                 if (ost->logfile && enc->stats_out) {
1753                     fprintf(ost->logfile, "%s", enc->stats_out);
1754                 }
1755                 if (!got_packet) {
1756                     stop_encoding = 1;
1757                     break;
1758                 }
1759                 if (ost->finished & MUXER_FINISHED) {
1760                     av_packet_unref(&pkt);
1761                     continue;
1762                 }
1763                 av_packet_rescale_ts(&pkt, enc->time_base, ost->st->time_base);
1764                 pkt_size = pkt.size;
1765                 write_frame(os, &pkt, ost);
1766                 if (ost->enc_ctx->codec_type == AVMEDIA_TYPE_VIDEO && vstats_filename) {
1767                     do_video_stats(ost, pkt_size);
1768                 }
1769             }
1770
1771             if (stop_encoding)
1772                 break;
1773         }
1774     }
1775 }
1776
1777 /*
1778  * Check whether a packet from ist should be written into ost at this time
1779  */
1780 static int check_output_constraints(InputStream *ist, OutputStream *ost)
1781 {
1782     OutputFile *of = output_files[ost->file_index];
1783     int ist_index  = input_files[ist->file_index]->ist_index + ist->st->index;
1784
1785     if (ost->source_index != ist_index)
1786         return 0;
1787
1788     if (ost->finished)
1789         return 0;
1790
1791     if (of->start_time != AV_NOPTS_VALUE && ist->pts < of->start_time)
1792         return 0;
1793
1794     return 1;
1795 }
1796
1797 static void do_streamcopy(InputStream *ist, OutputStream *ost, const AVPacket *pkt)
1798 {
1799     OutputFile *of = output_files[ost->file_index];
1800     InputFile   *f = input_files [ist->file_index];
1801     int64_t start_time = (of->start_time == AV_NOPTS_VALUE) ? 0 : of->start_time;
1802     int64_t ost_tb_start_time = av_rescale_q(start_time, AV_TIME_BASE_Q, ost->st->time_base);
1803     AVPicture pict;
1804     AVPacket opkt;
1805
1806     av_init_packet(&opkt);
1807
1808     if ((!ost->frame_number && !(pkt->flags & AV_PKT_FLAG_KEY)) &&
1809         !ost->copy_initial_nonkeyframes)
1810         return;
1811
1812     if (!ost->frame_number && !ost->copy_prior_start) {
1813         int64_t comp_start = start_time;
1814         if (copy_ts && f->start_time != AV_NOPTS_VALUE)
1815             comp_start = FFMAX(start_time, f->start_time + f->ts_offset);
1816         if (pkt->pts == AV_NOPTS_VALUE ?
1817             ist->pts < comp_start :
1818             pkt->pts < av_rescale_q(comp_start, AV_TIME_BASE_Q, ist->st->time_base))
1819             return;
1820     }
1821
1822     if (of->recording_time != INT64_MAX &&
1823         ist->pts >= of->recording_time + start_time) {
1824         close_output_stream(ost);
1825         return;
1826     }
1827
1828     if (f->recording_time != INT64_MAX) {
1829         start_time = f->ctx->start_time;
1830         if (f->start_time != AV_NOPTS_VALUE && copy_ts)
1831             start_time += f->start_time;
1832         if (ist->pts >= f->recording_time + start_time) {
1833             close_output_stream(ost);
1834             return;
1835         }
1836     }
1837
1838     /* force the input stream PTS */
1839     if (ost->enc_ctx->codec_type == AVMEDIA_TYPE_VIDEO)
1840         ost->sync_opts++;
1841
1842     if (pkt->pts != AV_NOPTS_VALUE)
1843         opkt.pts = av_rescale_q(pkt->pts, ist->st->time_base, ost->st->time_base) - ost_tb_start_time;
1844     else
1845         opkt.pts = AV_NOPTS_VALUE;
1846
1847     if (pkt->dts == AV_NOPTS_VALUE)
1848         opkt.dts = av_rescale_q(ist->dts, AV_TIME_BASE_Q, ost->st->time_base);
1849     else
1850         opkt.dts = av_rescale_q(pkt->dts, ist->st->time_base, ost->st->time_base);
1851     opkt.dts -= ost_tb_start_time;
1852
1853     if (ost->st->codec->codec_type == AVMEDIA_TYPE_AUDIO && pkt->dts != AV_NOPTS_VALUE) {
1854         int duration = av_get_audio_frame_duration(ist->dec_ctx, pkt->size);
1855         if(!duration)
1856             duration = ist->dec_ctx->frame_size;
1857         opkt.dts = opkt.pts = av_rescale_delta(ist->st->time_base, pkt->dts,
1858                                                (AVRational){1, ist->dec_ctx->sample_rate}, duration, &ist->filter_in_rescale_delta_last,
1859                                                ost->st->time_base) - ost_tb_start_time;
1860     }
1861
1862     opkt.duration = av_rescale_q(pkt->duration, ist->st->time_base, ost->st->time_base);
1863     opkt.flags    = pkt->flags;
1864     // FIXME remove the following 2 lines they shall be replaced by the bitstream filters
1865     if (  ost->st->codec->codec_id != AV_CODEC_ID_H264
1866        && ost->st->codec->codec_id != AV_CODEC_ID_MPEG1VIDEO
1867        && ost->st->codec->codec_id != AV_CODEC_ID_MPEG2VIDEO
1868        && ost->st->codec->codec_id != AV_CODEC_ID_VC1
1869        ) {
1870         int ret = av_parser_change(ost->parser, ost->st->codec,
1871                              &opkt.data, &opkt.size,
1872                              pkt->data, pkt->size,
1873                              pkt->flags & AV_PKT_FLAG_KEY);
1874         if (ret < 0) {
1875             av_log(NULL, AV_LOG_FATAL, "av_parser_change failed: %s\n",
1876                    av_err2str(ret));
1877             exit_program(1);
1878         }
1879         if (ret) {
1880             opkt.buf = av_buffer_create(opkt.data, opkt.size, av_buffer_default_free, NULL, 0);
1881             if (!opkt.buf)
1882                 exit_program(1);
1883         }
1884     } else {
1885         opkt.data = pkt->data;
1886         opkt.size = pkt->size;
1887     }
1888     av_copy_packet_side_data(&opkt, pkt);
1889
1890 #if FF_API_LAVF_FMT_RAWPICTURE
1891     if (ost->st->codec->codec_type == AVMEDIA_TYPE_VIDEO &&
1892         ost->st->codec->codec_id == AV_CODEC_ID_RAWVIDEO &&
1893         (of->ctx->oformat->flags & AVFMT_RAWPICTURE)) {
1894         /* store AVPicture in AVPacket, as expected by the output format */
1895         int ret = avpicture_fill(&pict, opkt.data, ost->st->codec->pix_fmt, ost->st->codec->width, ost->st->codec->height);
1896         if (ret < 0) {
1897             av_log(NULL, AV_LOG_FATAL, "avpicture_fill failed: %s\n",
1898                    av_err2str(ret));
1899             exit_program(1);
1900         }
1901         opkt.data = (uint8_t *)&pict;
1902         opkt.size = sizeof(AVPicture);
1903         opkt.flags |= AV_PKT_FLAG_KEY;
1904     }
1905 #endif
1906
1907     write_frame(of->ctx, &opkt, ost);
1908 }
1909
1910 int guess_input_channel_layout(InputStream *ist)
1911 {
1912     AVCodecContext *dec = ist->dec_ctx;
1913
1914     if (!dec->channel_layout) {
1915         char layout_name[256];
1916
1917         if (dec->channels > ist->guess_layout_max)
1918             return 0;
1919         dec->channel_layout = av_get_default_channel_layout(dec->channels);
1920         if (!dec->channel_layout)
1921             return 0;
1922         av_get_channel_layout_string(layout_name, sizeof(layout_name),
1923                                      dec->channels, dec->channel_layout);
1924         av_log(NULL, AV_LOG_WARNING, "Guessed Channel Layout for Input Stream "
1925                "#%d.%d : %s\n", ist->file_index, ist->st->index, layout_name);
1926     }
1927     return 1;
1928 }
1929
1930 static void check_decode_result(InputStream *ist, int *got_output, int ret)
1931 {
1932     if (*got_output || ret<0)
1933         decode_error_stat[ret<0] ++;
1934
1935     if (ret < 0 && exit_on_error)
1936         exit_program(1);
1937
1938     if (exit_on_error && *got_output && ist) {
1939         if (av_frame_get_decode_error_flags(ist->decoded_frame) || (ist->decoded_frame->flags & AV_FRAME_FLAG_CORRUPT)) {
1940             av_log(NULL, AV_LOG_FATAL, "%s: corrupt decoded frame in stream %d\n", input_files[ist->file_index]->ctx->filename, ist->st->index);
1941             exit_program(1);
1942         }
1943     }
1944 }
1945
1946 static int decode_audio(InputStream *ist, AVPacket *pkt, int *got_output)
1947 {
1948     AVFrame *decoded_frame, *f;
1949     AVCodecContext *avctx = ist->dec_ctx;
1950     int i, ret, err = 0, resample_changed;
1951     AVRational decoded_frame_tb;
1952
1953     if (!ist->decoded_frame && !(ist->decoded_frame = av_frame_alloc()))
1954         return AVERROR(ENOMEM);
1955     if (!ist->filter_frame && !(ist->filter_frame = av_frame_alloc()))
1956         return AVERROR(ENOMEM);
1957     decoded_frame = ist->decoded_frame;
1958
1959     update_benchmark(NULL);
1960     ret = avcodec_decode_audio4(avctx, decoded_frame, got_output, pkt);
1961     update_benchmark("decode_audio %d.%d", ist->file_index, ist->st->index);
1962
1963     if (ret >= 0 && avctx->sample_rate <= 0) {
1964         av_log(avctx, AV_LOG_ERROR, "Sample rate %d invalid\n", avctx->sample_rate);
1965         ret = AVERROR_INVALIDDATA;
1966     }
1967
1968     check_decode_result(ist, got_output, ret);
1969
1970     if (!*got_output || ret < 0)
1971         return ret;
1972
1973     ist->samples_decoded += decoded_frame->nb_samples;
1974     ist->frames_decoded++;
1975
1976 #if 1
1977     /* increment next_dts to use for the case where the input stream does not
1978        have timestamps or there are multiple frames in the packet */
1979     ist->next_pts += ((int64_t)AV_TIME_BASE * decoded_frame->nb_samples) /
1980                      avctx->sample_rate;
1981     ist->next_dts += ((int64_t)AV_TIME_BASE * decoded_frame->nb_samples) /
1982                      avctx->sample_rate;
1983 #endif
1984
1985     resample_changed = ist->resample_sample_fmt     != decoded_frame->format         ||
1986                        ist->resample_channels       != avctx->channels               ||
1987                        ist->resample_channel_layout != decoded_frame->channel_layout ||
1988                        ist->resample_sample_rate    != decoded_frame->sample_rate;
1989     if (resample_changed) {
1990         char layout1[64], layout2[64];
1991
1992         if (!guess_input_channel_layout(ist)) {
1993             av_log(NULL, AV_LOG_FATAL, "Unable to find default channel "
1994                    "layout for Input Stream #%d.%d\n", ist->file_index,
1995                    ist->st->index);
1996             exit_program(1);
1997         }
1998         decoded_frame->channel_layout = avctx->channel_layout;
1999
2000         av_get_channel_layout_string(layout1, sizeof(layout1), ist->resample_channels,
2001                                      ist->resample_channel_layout);
2002         av_get_channel_layout_string(layout2, sizeof(layout2), avctx->channels,
2003                                      decoded_frame->channel_layout);
2004
2005         av_log(NULL, AV_LOG_INFO,
2006                "Input stream #%d:%d frame changed from rate:%d fmt:%s ch:%d chl:%s to rate:%d fmt:%s ch:%d chl:%s\n",
2007                ist->file_index, ist->st->index,
2008                ist->resample_sample_rate,  av_get_sample_fmt_name(ist->resample_sample_fmt),
2009                ist->resample_channels, layout1,
2010                decoded_frame->sample_rate, av_get_sample_fmt_name(decoded_frame->format),
2011                avctx->channels, layout2);
2012
2013         ist->resample_sample_fmt     = decoded_frame->format;
2014         ist->resample_sample_rate    = decoded_frame->sample_rate;
2015         ist->resample_channel_layout = decoded_frame->channel_layout;
2016         ist->resample_channels       = avctx->channels;
2017
2018         for (i = 0; i < nb_filtergraphs; i++)
2019             if (ist_in_filtergraph(filtergraphs[i], ist)) {
2020                 FilterGraph *fg = filtergraphs[i];
2021                 if (configure_filtergraph(fg) < 0) {
2022                     av_log(NULL, AV_LOG_FATAL, "Error reinitializing filters!\n");
2023                     exit_program(1);
2024                 }
2025             }
2026     }
2027
2028     /* if the decoder provides a pts, use it instead of the last packet pts.
2029        the decoder could be delaying output by a packet or more. */
2030     if (decoded_frame->pts != AV_NOPTS_VALUE) {
2031         ist->dts = ist->next_dts = ist->pts = ist->next_pts = av_rescale_q(decoded_frame->pts, avctx->time_base, AV_TIME_BASE_Q);
2032         decoded_frame_tb   = avctx->time_base;
2033     } else if (decoded_frame->pkt_pts != AV_NOPTS_VALUE) {
2034         decoded_frame->pts = decoded_frame->pkt_pts;
2035         decoded_frame_tb   = ist->st->time_base;
2036     } else if (pkt->pts != AV_NOPTS_VALUE) {
2037         decoded_frame->pts = pkt->pts;
2038         decoded_frame_tb   = ist->st->time_base;
2039     }else {
2040         decoded_frame->pts = ist->dts;
2041         decoded_frame_tb   = AV_TIME_BASE_Q;
2042     }
2043     pkt->pts           = AV_NOPTS_VALUE;
2044     if (decoded_frame->pts != AV_NOPTS_VALUE)
2045         decoded_frame->pts = av_rescale_delta(decoded_frame_tb, decoded_frame->pts,
2046                                               (AVRational){1, avctx->sample_rate}, decoded_frame->nb_samples, &ist->filter_in_rescale_delta_last,
2047                                               (AVRational){1, avctx->sample_rate});
2048     ist->nb_samples = decoded_frame->nb_samples;
2049     for (i = 0; i < ist->nb_filters; i++) {
2050         if (i < ist->nb_filters - 1) {
2051             f = ist->filter_frame;
2052             err = av_frame_ref(f, decoded_frame);
2053             if (err < 0)
2054                 break;
2055         } else
2056             f = decoded_frame;
2057         err = av_buffersrc_add_frame_flags(ist->filters[i]->filter, f,
2058                                      AV_BUFFERSRC_FLAG_PUSH);
2059         if (err == AVERROR_EOF)
2060             err = 0; /* ignore */
2061         if (err < 0)
2062             break;
2063     }
2064     decoded_frame->pts = AV_NOPTS_VALUE;
2065
2066     av_frame_unref(ist->filter_frame);
2067     av_frame_unref(decoded_frame);
2068     return err < 0 ? err : ret;
2069 }
2070
2071 static int decode_video(InputStream *ist, AVPacket *pkt, int *got_output)
2072 {
2073     AVFrame *decoded_frame, *f;
2074     int i, ret = 0, err = 0, resample_changed;
2075     int64_t best_effort_timestamp;
2076     AVRational *frame_sample_aspect;
2077
2078     if (!ist->decoded_frame && !(ist->decoded_frame = av_frame_alloc()))
2079         return AVERROR(ENOMEM);
2080     if (!ist->filter_frame && !(ist->filter_frame = av_frame_alloc()))
2081         return AVERROR(ENOMEM);
2082     decoded_frame = ist->decoded_frame;
2083     pkt->dts  = av_rescale_q(ist->dts, AV_TIME_BASE_Q, ist->st->time_base);
2084
2085     update_benchmark(NULL);
2086     ret = avcodec_decode_video2(ist->dec_ctx,
2087                                 decoded_frame, got_output, pkt);
2088     update_benchmark("decode_video %d.%d", ist->file_index, ist->st->index);
2089
2090     // The following line may be required in some cases where there is no parser
2091     // or the parser does not has_b_frames correctly
2092     if (ist->st->codec->has_b_frames < ist->dec_ctx->has_b_frames) {
2093         if (ist->dec_ctx->codec_id == AV_CODEC_ID_H264) {
2094             ist->st->codec->has_b_frames = ist->dec_ctx->has_b_frames;
2095         } else
2096             av_log(ist->dec_ctx, AV_LOG_WARNING,
2097                    "has_b_frames is larger in decoder than demuxer %d > %d.\n"
2098                    "If you want to help, upload a sample "
2099                    "of this file to ftp://upload.ffmpeg.org/incoming/ "
2100                    "and contact the ffmpeg-devel mailing list. (ffmpeg-devel@ffmpeg.org)",
2101                    ist->dec_ctx->has_b_frames,
2102                    ist->st->codec->has_b_frames);
2103     }
2104
2105     check_decode_result(ist, got_output, ret);
2106
2107     if (*got_output && ret >= 0) {
2108         if (ist->dec_ctx->width  != decoded_frame->width ||
2109             ist->dec_ctx->height != decoded_frame->height ||
2110             ist->dec_ctx->pix_fmt != decoded_frame->format) {
2111             av_log(NULL, AV_LOG_DEBUG, "Frame parameters mismatch context %d,%d,%d != %d,%d,%d\n",
2112                 decoded_frame->width,
2113                 decoded_frame->height,
2114                 decoded_frame->format,
2115                 ist->dec_ctx->width,
2116                 ist->dec_ctx->height,
2117                 ist->dec_ctx->pix_fmt);
2118         }
2119     }
2120
2121     if (!*got_output || ret < 0)
2122         return ret;
2123
2124     if(ist->top_field_first>=0)
2125         decoded_frame->top_field_first = ist->top_field_first;
2126
2127     ist->frames_decoded++;
2128
2129     if (ist->hwaccel_retrieve_data && decoded_frame->format == ist->hwaccel_pix_fmt) {
2130         err = ist->hwaccel_retrieve_data(ist->dec_ctx, decoded_frame);
2131         if (err < 0)
2132             goto fail;
2133     }
2134     ist->hwaccel_retrieved_pix_fmt = decoded_frame->format;
2135
2136     best_effort_timestamp= av_frame_get_best_effort_timestamp(decoded_frame);
2137     if(best_effort_timestamp != AV_NOPTS_VALUE) {
2138         int64_t ts = av_rescale_q(decoded_frame->pts = best_effort_timestamp, ist->st->time_base, AV_TIME_BASE_Q);
2139
2140         if (ts != AV_NOPTS_VALUE)
2141             ist->next_pts = ist->pts = ts;
2142     }
2143
2144     if (debug_ts) {
2145         av_log(NULL, AV_LOG_INFO, "decoder -> ist_index:%d type:video "
2146                "frame_pts:%s frame_pts_time:%s best_effort_ts:%"PRId64" best_effort_ts_time:%s keyframe:%d frame_type:%d time_base:%d/%d\n",
2147                ist->st->index, av_ts2str(decoded_frame->pts),
2148                av_ts2timestr(decoded_frame->pts, &ist->st->time_base),
2149                best_effort_timestamp,
2150                av_ts2timestr(best_effort_timestamp, &ist->st->time_base),
2151                decoded_frame->key_frame, decoded_frame->pict_type,
2152                ist->st->time_base.num, ist->st->time_base.den);
2153     }
2154
2155     pkt->size = 0;
2156
2157     if (ist->st->sample_aspect_ratio.num)
2158         decoded_frame->sample_aspect_ratio = ist->st->sample_aspect_ratio;
2159
2160     resample_changed = ist->resample_width   != decoded_frame->width  ||
2161                        ist->resample_height  != decoded_frame->height ||
2162                        ist->resample_pix_fmt != decoded_frame->format;
2163     if (resample_changed) {
2164         av_log(NULL, AV_LOG_INFO,
2165                "Input stream #%d:%d frame changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s\n",
2166                ist->file_index, ist->st->index,
2167                ist->resample_width,  ist->resample_height,  av_get_pix_fmt_name(ist->resample_pix_fmt),
2168                decoded_frame->width, decoded_frame->height, av_get_pix_fmt_name(decoded_frame->format));
2169
2170         ist->resample_width   = decoded_frame->width;
2171         ist->resample_height  = decoded_frame->height;
2172         ist->resample_pix_fmt = decoded_frame->format;
2173
2174         for (i = 0; i < nb_filtergraphs; i++) {
2175             if (ist_in_filtergraph(filtergraphs[i], ist) && ist->reinit_filters &&
2176                 configure_filtergraph(filtergraphs[i]) < 0) {
2177                 av_log(NULL, AV_LOG_FATAL, "Error reinitializing filters!\n");
2178                 exit_program(1);
2179             }
2180         }
2181     }
2182
2183     frame_sample_aspect= av_opt_ptr(avcodec_get_frame_class(), decoded_frame, "sample_aspect_ratio");
2184     for (i = 0; i < ist->nb_filters; i++) {
2185         if (!frame_sample_aspect->num)
2186             *frame_sample_aspect = ist->st->sample_aspect_ratio;
2187
2188         if (i < ist->nb_filters - 1) {
2189             f = ist->filter_frame;
2190             err = av_frame_ref(f, decoded_frame);
2191             if (err < 0)
2192                 break;
2193         } else
2194             f = decoded_frame;
2195         ret = av_buffersrc_add_frame_flags(ist->filters[i]->filter, f, AV_BUFFERSRC_FLAG_PUSH);
2196         if (ret == AVERROR_EOF) {
2197             ret = 0; /* ignore */
2198         } else if (ret < 0) {
2199             av_log(NULL, AV_LOG_FATAL,
2200                    "Failed to inject frame into filter network: %s\n", av_err2str(ret));
2201             exit_program(1);
2202         }
2203     }
2204
2205 fail:
2206     av_frame_unref(ist->filter_frame);
2207     av_frame_unref(decoded_frame);
2208     return err < 0 ? err : ret;
2209 }
2210
2211 static int transcode_subtitles(InputStream *ist, AVPacket *pkt, int *got_output)
2212 {
2213     AVSubtitle subtitle;
2214     int i, ret = avcodec_decode_subtitle2(ist->dec_ctx,
2215                                           &subtitle, got_output, pkt);
2216
2217     check_decode_result(NULL, got_output, ret);
2218
2219     if (ret < 0 || !*got_output) {
2220         if (!pkt->size)
2221             sub2video_flush(ist);
2222         return ret;
2223     }
2224
2225     if (ist->fix_sub_duration) {
2226         int end = 1;
2227         if (ist->prev_sub.got_output) {
2228             end = av_rescale(subtitle.pts - ist->prev_sub.subtitle.pts,
2229                              1000, AV_TIME_BASE);
2230             if (end < ist->prev_sub.subtitle.end_display_time) {
2231                 av_log(ist->dec_ctx, AV_LOG_DEBUG,
2232                        "Subtitle duration reduced from %d to %d%s\n",
2233                        ist->prev_sub.subtitle.end_display_time, end,
2234                        end <= 0 ? ", dropping it" : "");
2235                 ist->prev_sub.subtitle.end_display_time = end;
2236             }
2237         }
2238         FFSWAP(int,        *got_output, ist->prev_sub.got_output);
2239         FFSWAP(int,        ret,         ist->prev_sub.ret);
2240         FFSWAP(AVSubtitle, subtitle,    ist->prev_sub.subtitle);
2241         if (end <= 0)
2242             goto out;
2243     }
2244
2245     if (!*got_output)
2246         return ret;
2247
2248     sub2video_update(ist, &subtitle);
2249
2250     if (!subtitle.num_rects)
2251         goto out;
2252
2253     ist->frames_decoded++;
2254
2255     for (i = 0; i < nb_output_streams; i++) {
2256         OutputStream *ost = output_streams[i];
2257
2258         if (!check_output_constraints(ist, ost) || !ost->encoding_needed
2259             || ost->enc->type != AVMEDIA_TYPE_SUBTITLE)
2260             continue;
2261
2262         do_subtitle_out(output_files[ost->file_index]->ctx, ost, ist, &subtitle);
2263     }
2264
2265 out:
2266     avsubtitle_free(&subtitle);
2267     return ret;
2268 }
2269
2270 static int send_filter_eof(InputStream *ist)
2271 {
2272     int i, ret;
2273     for (i = 0; i < ist->nb_filters; i++) {
2274         ret = av_buffersrc_add_frame(ist->filters[i]->filter, NULL);
2275         if (ret < 0)
2276             return ret;
2277     }
2278     return 0;
2279 }
2280
2281 /* pkt = NULL means EOF (needed to flush decoder buffers) */
2282 static int process_input_packet(InputStream *ist, const AVPacket *pkt, int no_eof)
2283 {
2284     int ret = 0, i;
2285     int got_output = 0;
2286
2287     AVPacket avpkt;
2288     if (!ist->saw_first_ts) {
2289         ist->dts = ist->st->avg_frame_rate.num ? - ist->dec_ctx->has_b_frames * AV_TIME_BASE / av_q2d(ist->st->avg_frame_rate) : 0;
2290         ist->pts = 0;
2291         if (pkt && pkt->pts != AV_NOPTS_VALUE && !ist->decoding_needed) {
2292             ist->dts += av_rescale_q(pkt->pts, ist->st->time_base, AV_TIME_BASE_Q);
2293             ist->pts = ist->dts; //unused but better to set it to a value thats not totally wrong
2294         }
2295         ist->saw_first_ts = 1;
2296     }
2297
2298     if (ist->next_dts == AV_NOPTS_VALUE)
2299         ist->next_dts = ist->dts;
2300     if (ist->next_pts == AV_NOPTS_VALUE)
2301         ist->next_pts = ist->pts;
2302
2303     if (!pkt) {
2304         /* EOF handling */
2305         av_init_packet(&avpkt);
2306         avpkt.data = NULL;
2307         avpkt.size = 0;
2308         goto handle_eof;
2309     } else {
2310         avpkt = *pkt;
2311     }
2312
2313     if (pkt->dts != AV_NOPTS_VALUE) {
2314         ist->next_dts = ist->dts = av_rescale_q(pkt->dts, ist->st->time_base, AV_TIME_BASE_Q);
2315         if (ist->dec_ctx->codec_type != AVMEDIA_TYPE_VIDEO || !ist->decoding_needed)
2316             ist->next_pts = ist->pts = ist->dts;
2317     }
2318
2319     // while we have more to decode or while the decoder did output something on EOF
2320     while (ist->decoding_needed && (avpkt.size > 0 || (!pkt && got_output))) {
2321         int duration;
2322     handle_eof:
2323
2324         ist->pts = ist->next_pts;
2325         ist->dts = ist->next_dts;
2326
2327         if (avpkt.size && avpkt.size != pkt->size &&
2328             !(ist->dec->capabilities & AV_CODEC_CAP_SUBFRAMES)) {
2329             av_log(NULL, ist->showed_multi_packet_warning ? AV_LOG_VERBOSE : AV_LOG_WARNING,
2330                    "Multiple frames in a packet from stream %d\n", pkt->stream_index);
2331             ist->showed_multi_packet_warning = 1;
2332         }
2333
2334         switch (ist->dec_ctx->codec_type) {
2335         case AVMEDIA_TYPE_AUDIO:
2336             ret = decode_audio    (ist, &avpkt, &got_output);
2337             break;
2338         case AVMEDIA_TYPE_VIDEO:
2339             ret = decode_video    (ist, &avpkt, &got_output);
2340             if (avpkt.duration) {
2341                 duration = av_rescale_q(avpkt.duration, ist->st->time_base, AV_TIME_BASE_Q);
2342             } else if(ist->dec_ctx->framerate.num != 0 && ist->dec_ctx->framerate.den != 0) {
2343                 int ticks= av_stream_get_parser(ist->st) ? av_stream_get_parser(ist->st)->repeat_pict+1 : ist->dec_ctx->ticks_per_frame;
2344                 duration = ((int64_t)AV_TIME_BASE *
2345                                 ist->dec_ctx->framerate.den * ticks) /
2346                                 ist->dec_ctx->framerate.num / ist->dec_ctx->ticks_per_frame;
2347             } else
2348                 duration = 0;
2349
2350             if(ist->dts != AV_NOPTS_VALUE && duration) {
2351                 ist->next_dts += duration;
2352             }else
2353                 ist->next_dts = AV_NOPTS_VALUE;
2354
2355             if (got_output)
2356                 ist->next_pts += duration; //FIXME the duration is not correct in some cases
2357             break;
2358         case AVMEDIA_TYPE_SUBTITLE:
2359             ret = transcode_subtitles(ist, &avpkt, &got_output);
2360             break;
2361         default:
2362             return -1;
2363         }
2364
2365         if (ret < 0) {
2366             av_log(NULL, AV_LOG_ERROR, "Error while decoding stream #%d:%d: %s\n",
2367                    ist->file_index, ist->st->index, av_err2str(ret));
2368             if (exit_on_error)
2369                 exit_program(1);
2370             break;
2371         }
2372
2373         avpkt.dts=
2374         avpkt.pts= AV_NOPTS_VALUE;
2375
2376         // touch data and size only if not EOF
2377         if (pkt) {
2378             if(ist->dec_ctx->codec_type != AVMEDIA_TYPE_AUDIO)
2379                 ret = avpkt.size;
2380             avpkt.data += ret;
2381             avpkt.size -= ret;
2382         }
2383         if (!got_output) {
2384             continue;
2385         }
2386         if (got_output && !pkt)
2387             break;
2388     }
2389
2390     /* after flushing, send an EOF on all the filter inputs attached to the stream */
2391     /* except when looping we need to flush but not to send an EOF */
2392     if (!pkt && ist->decoding_needed && !got_output && !no_eof) {
2393         int ret = send_filter_eof(ist);
2394         if (ret < 0) {
2395             av_log(NULL, AV_LOG_FATAL, "Error marking filters as finished\n");
2396             exit_program(1);
2397         }
2398     }
2399
2400     /* handle stream copy */
2401     if (!ist->decoding_needed) {
2402         ist->dts = ist->next_dts;
2403         switch (ist->dec_ctx->codec_type) {
2404         case AVMEDIA_TYPE_AUDIO:
2405             ist->next_dts += ((int64_t)AV_TIME_BASE * ist->dec_ctx->frame_size) /
2406                              ist->dec_ctx->sample_rate;
2407             break;
2408         case AVMEDIA_TYPE_VIDEO:
2409             if (ist->framerate.num) {
2410                 // TODO: Remove work-around for c99-to-c89 issue 7
2411                 AVRational time_base_q = AV_TIME_BASE_Q;
2412                 int64_t next_dts = av_rescale_q(ist->next_dts, time_base_q, av_inv_q(ist->framerate));
2413                 ist->next_dts = av_rescale_q(next_dts + 1, av_inv_q(ist->framerate), time_base_q);
2414             } else if (pkt->duration) {
2415                 ist->next_dts += av_rescale_q(pkt->duration, ist->st->time_base, AV_TIME_BASE_Q);
2416             } else if(ist->dec_ctx->framerate.num != 0) {
2417                 int ticks= av_stream_get_parser(ist->st) ? av_stream_get_parser(ist->st)->repeat_pict + 1 : ist->dec_ctx->ticks_per_frame;
2418                 ist->next_dts += ((int64_t)AV_TIME_BASE *
2419                                   ist->dec_ctx->framerate.den * ticks) /
2420                                   ist->dec_ctx->framerate.num / ist->dec_ctx->ticks_per_frame;
2421             }
2422             break;
2423         }
2424         ist->pts = ist->dts;
2425         ist->next_pts = ist->next_dts;
2426     }
2427     for (i = 0; pkt && i < nb_output_streams; i++) {
2428         OutputStream *ost = output_streams[i];
2429
2430         if (!check_output_constraints(ist, ost) || ost->encoding_needed)
2431             continue;
2432
2433         do_streamcopy(ist, ost, pkt);
2434     }
2435
2436     return got_output;
2437 }
2438
2439 static void print_sdp(void)
2440 {
2441     char sdp[16384];
2442     int i;
2443     int j;
2444     AVIOContext *sdp_pb;
2445     AVFormatContext **avc = av_malloc_array(nb_output_files, sizeof(*avc));
2446
2447     if (!avc)
2448         exit_program(1);
2449     for (i = 0, j = 0; i < nb_output_files; i++) {
2450         if (!strcmp(output_files[i]->ctx->oformat->name, "rtp")) {
2451             avc[j] = output_files[i]->ctx;
2452             j++;
2453         }
2454     }
2455
2456     if (!j)
2457         goto fail;
2458
2459     av_sdp_create(avc, j, sdp, sizeof(sdp));
2460
2461     if (!sdp_filename) {
2462         printf("SDP:\n%s\n", sdp);
2463         fflush(stdout);
2464     } else {
2465         if (avio_open2(&sdp_pb, sdp_filename, AVIO_FLAG_WRITE, &int_cb, NULL) < 0) {
2466             av_log(NULL, AV_LOG_ERROR, "Failed to open sdp file '%s'\n", sdp_filename);
2467         } else {
2468             avio_printf(sdp_pb, "SDP:\n%s", sdp);
2469             avio_closep(&sdp_pb);
2470             av_freep(&sdp_filename);
2471         }
2472     }
2473
2474 fail:
2475     av_freep(&avc);
2476 }
2477
2478 static const HWAccel *get_hwaccel(enum AVPixelFormat pix_fmt)
2479 {
2480     int i;
2481     for (i = 0; hwaccels[i].name; i++)
2482         if (hwaccels[i].pix_fmt == pix_fmt)
2483             return &hwaccels[i];
2484     return NULL;
2485 }
2486
2487 static enum AVPixelFormat get_format(AVCodecContext *s, const enum AVPixelFormat *pix_fmts)
2488 {
2489     InputStream *ist = s->opaque;
2490     const enum AVPixelFormat *p;
2491     int ret;
2492
2493     for (p = pix_fmts; *p != -1; p++) {
2494         const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(*p);
2495         const HWAccel *hwaccel;
2496
2497         if (!(desc->flags & AV_PIX_FMT_FLAG_HWACCEL))
2498             break;
2499
2500         hwaccel = get_hwaccel(*p);
2501         if (!hwaccel ||
2502             (ist->active_hwaccel_id && ist->active_hwaccel_id != hwaccel->id) ||
2503             (ist->hwaccel_id != HWACCEL_AUTO && ist->hwaccel_id != hwaccel->id))
2504             continue;
2505
2506         ret = hwaccel->init(s);
2507         if (ret < 0) {
2508             if (ist->hwaccel_id == hwaccel->id) {
2509                 av_log(NULL, AV_LOG_FATAL,
2510                        "%s hwaccel requested for input stream #%d:%d, "
2511                        "but cannot be initialized.\n", hwaccel->name,
2512                        ist->file_index, ist->st->index);
2513                 return AV_PIX_FMT_NONE;
2514             }
2515             continue;
2516         }
2517         ist->active_hwaccel_id = hwaccel->id;
2518         ist->hwaccel_pix_fmt   = *p;
2519         break;
2520     }
2521
2522     return *p;
2523 }
2524
2525 static int get_buffer(AVCodecContext *s, AVFrame *frame, int flags)
2526 {
2527     InputStream *ist = s->opaque;
2528
2529     if (ist->hwaccel_get_buffer && frame->format == ist->hwaccel_pix_fmt)
2530         return ist->hwaccel_get_buffer(s, frame, flags);
2531
2532     return avcodec_default_get_buffer2(s, frame, flags);
2533 }
2534
2535 static int init_input_stream(int ist_index, char *error, int error_len)
2536 {
2537     int ret;
2538     InputStream *ist = input_streams[ist_index];
2539
2540     if (ist->decoding_needed) {
2541         AVCodec *codec = ist->dec;
2542         if (!codec) {
2543             snprintf(error, error_len, "Decoder (codec %s) not found for input stream #%d:%d",
2544                     avcodec_get_name(ist->dec_ctx->codec_id), ist->file_index, ist->st->index);
2545             return AVERROR(EINVAL);
2546         }
2547
2548         ist->dec_ctx->opaque                = ist;
2549         ist->dec_ctx->get_format            = get_format;
2550         ist->dec_ctx->get_buffer2           = get_buffer;
2551         ist->dec_ctx->thread_safe_callbacks = 1;
2552
2553         av_opt_set_int(ist->dec_ctx, "refcounted_frames", 1, 0);
2554         if (ist->dec_ctx->codec_id == AV_CODEC_ID_DVB_SUBTITLE &&
2555            (ist->decoding_needed & DECODING_FOR_OST)) {
2556             av_dict_set(&ist->decoder_opts, "compute_edt", "1", AV_DICT_DONT_OVERWRITE);
2557             if (ist->decoding_needed & DECODING_FOR_FILTER)
2558                 av_log(NULL, AV_LOG_WARNING, "Warning using DVB subtitles for filtering and output at the same time is not fully supported, also see -compute_edt [0|1]\n");
2559         }
2560
2561         av_dict_set(&ist->decoder_opts, "sub_text_format", "ass", AV_DICT_DONT_OVERWRITE);
2562
2563         if (!av_dict_get(ist->decoder_opts, "threads", NULL, 0))
2564             av_dict_set(&ist->decoder_opts, "threads", "auto", 0);
2565         if ((ret = avcodec_open2(ist->dec_ctx, codec, &ist->decoder_opts)) < 0) {
2566             if (ret == AVERROR_EXPERIMENTAL)
2567                 abort_codec_experimental(codec, 0);
2568
2569             snprintf(error, error_len,
2570                      "Error while opening decoder for input stream "
2571                      "#%d:%d : %s",
2572                      ist->file_index, ist->st->index, av_err2str(ret));
2573             return ret;
2574         }
2575         assert_avoptions(ist->decoder_opts);
2576     }
2577
2578     ist->next_pts = AV_NOPTS_VALUE;
2579     ist->next_dts = AV_NOPTS_VALUE;
2580
2581     return 0;
2582 }
2583
2584 static InputStream *get_input_stream(OutputStream *ost)
2585 {
2586     if (ost->source_index >= 0)
2587         return input_streams[ost->source_index];
2588     return NULL;
2589 }
2590
2591 static int compare_int64(const void *a, const void *b)
2592 {
2593     return FFDIFFSIGN(*(const int64_t *)a, *(const int64_t *)b);
2594 }
2595
2596 static int init_output_stream(OutputStream *ost, char *error, int error_len)
2597 {
2598     int ret = 0;
2599
2600     if (ost->encoding_needed) {
2601         AVCodec      *codec = ost->enc;
2602         AVCodecContext *dec = NULL;
2603         InputStream *ist;
2604
2605         if ((ist = get_input_stream(ost)))
2606             dec = ist->dec_ctx;
2607         if (dec && dec->subtitle_header) {
2608             /* ASS code assumes this buffer is null terminated so add extra byte. */
2609             ost->enc_ctx->subtitle_header = av_mallocz(dec->subtitle_header_size + 1);
2610             if (!ost->enc_ctx->subtitle_header)
2611                 return AVERROR(ENOMEM);
2612             memcpy(ost->enc_ctx->subtitle_header, dec->subtitle_header, dec->subtitle_header_size);
2613             ost->enc_ctx->subtitle_header_size = dec->subtitle_header_size;
2614         }
2615         if (!av_dict_get(ost->encoder_opts, "threads", NULL, 0))
2616             av_dict_set(&ost->encoder_opts, "threads", "auto", 0);
2617         if (ost->enc->type == AVMEDIA_TYPE_AUDIO &&
2618             !codec->defaults &&
2619             !av_dict_get(ost->encoder_opts, "b", NULL, 0) &&
2620             !av_dict_get(ost->encoder_opts, "ab", NULL, 0))
2621             av_dict_set(&ost->encoder_opts, "b", "128000", 0);
2622
2623         if (ost->filter && ost->filter->filter->inputs[0]->hw_frames_ctx) {
2624             ost->enc_ctx->hw_frames_ctx = av_buffer_ref(ost->filter->filter->inputs[0]->hw_frames_ctx);
2625             if (!ost->enc_ctx->hw_frames_ctx)
2626                 return AVERROR(ENOMEM);
2627         }
2628
2629         if ((ret = avcodec_open2(ost->enc_ctx, codec, &ost->encoder_opts)) < 0) {
2630             if (ret == AVERROR_EXPERIMENTAL)
2631                 abort_codec_experimental(codec, 1);
2632             snprintf(error, error_len,
2633                      "Error while opening encoder for output stream #%d:%d - "
2634                      "maybe incorrect parameters such as bit_rate, rate, width or height",
2635                     ost->file_index, ost->index);
2636             return ret;
2637         }
2638         if (ost->enc->type == AVMEDIA_TYPE_AUDIO &&
2639             !(ost->enc->capabilities & AV_CODEC_CAP_VARIABLE_FRAME_SIZE))
2640             av_buffersink_set_frame_size(ost->filter->filter,
2641                                             ost->enc_ctx->frame_size);
2642         assert_avoptions(ost->encoder_opts);
2643         if (ost->enc_ctx->bit_rate && ost->enc_ctx->bit_rate < 1000)
2644             av_log(NULL, AV_LOG_WARNING, "The bitrate parameter is set too low."
2645                                          " It takes bits/s as argument, not kbits/s\n");
2646
2647         ret = avcodec_copy_context(ost->st->codec, ost->enc_ctx);
2648         if (ret < 0) {
2649             av_log(NULL, AV_LOG_FATAL,
2650                    "Error initializing the output stream codec context.\n");
2651             exit_program(1);
2652         }
2653
2654         if (ost->enc_ctx->nb_coded_side_data) {
2655             int i;
2656
2657             ost->st->side_data = av_realloc_array(NULL, ost->enc_ctx->nb_coded_side_data,
2658                                                   sizeof(*ost->st->side_data));
2659             if (!ost->st->side_data)
2660                 return AVERROR(ENOMEM);
2661
2662             for (i = 0; i < ost->enc_ctx->nb_coded_side_data; i++) {
2663                 const AVPacketSideData *sd_src = &ost->enc_ctx->coded_side_data[i];
2664                 AVPacketSideData *sd_dst = &ost->st->side_data[i];
2665
2666                 sd_dst->data = av_malloc(sd_src->size);
2667                 if (!sd_dst->data)
2668                     return AVERROR(ENOMEM);
2669                 memcpy(sd_dst->data, sd_src->data, sd_src->size);
2670                 sd_dst->size = sd_src->size;
2671                 sd_dst->type = sd_src->type;
2672                 ost->st->nb_side_data++;
2673             }
2674         }
2675
2676         // copy timebase while removing common factors
2677         ost->st->time_base = av_add_q(ost->enc_ctx->time_base, (AVRational){0, 1});
2678         ost->st->codec->codec= ost->enc_ctx->codec;
2679     } else if (ost->stream_copy) {
2680         ret = av_opt_set_dict(ost->st->codec, &ost->encoder_opts);
2681         if (ret < 0) {
2682            av_log(NULL, AV_LOG_FATAL,
2683                   "Error setting up codec context options.\n");
2684            return ret;
2685         }
2686         // copy timebase while removing common factors
2687         ost->st->time_base = av_add_q(ost->st->codec->time_base, (AVRational){0, 1});
2688     }
2689
2690     return ret;
2691 }
2692
2693 static void parse_forced_key_frames(char *kf, OutputStream *ost,
2694                                     AVCodecContext *avctx)
2695 {
2696     char *p;
2697     int n = 1, i, size, index = 0;
2698     int64_t t, *pts;
2699
2700     for (p = kf; *p; p++)
2701         if (*p == ',')
2702             n++;
2703     size = n;
2704     pts = av_malloc_array(size, sizeof(*pts));
2705     if (!pts) {
2706         av_log(NULL, AV_LOG_FATAL, "Could not allocate forced key frames array.\n");
2707         exit_program(1);
2708     }
2709
2710     p = kf;
2711     for (i = 0; i < n; i++) {
2712         char *next = strchr(p, ',');
2713
2714         if (next)
2715             *next++ = 0;
2716
2717         if (!memcmp(p, "chapters", 8)) {
2718
2719             AVFormatContext *avf = output_files[ost->file_index]->ctx;
2720             int j;
2721
2722             if (avf->nb_chapters > INT_MAX - size ||
2723                 !(pts = av_realloc_f(pts, size += avf->nb_chapters - 1,
2724                                      sizeof(*pts)))) {
2725                 av_log(NULL, AV_LOG_FATAL,
2726                        "Could not allocate forced key frames array.\n");
2727                 exit_program(1);
2728             }
2729             t = p[8] ? parse_time_or_die("force_key_frames", p + 8, 1) : 0;
2730             t = av_rescale_q(t, AV_TIME_BASE_Q, avctx->time_base);
2731
2732             for (j = 0; j < avf->nb_chapters; j++) {
2733                 AVChapter *c = avf->chapters[j];
2734                 av_assert1(index < size);
2735                 pts[index++] = av_rescale_q(c->start, c->time_base,
2736                                             avctx->time_base) + t;
2737             }
2738
2739         } else {
2740
2741             t = parse_time_or_die("force_key_frames", p, 1);
2742             av_assert1(index < size);
2743             pts[index++] = av_rescale_q(t, AV_TIME_BASE_Q, avctx->time_base);
2744
2745         }
2746
2747         p = next;
2748     }
2749
2750     av_assert0(index == size);
2751     qsort(pts, size, sizeof(*pts), compare_int64);
2752     ost->forced_kf_count = size;
2753     ost->forced_kf_pts   = pts;
2754 }
2755
2756 static void report_new_stream(int input_index, AVPacket *pkt)
2757 {
2758     InputFile *file = input_files[input_index];
2759     AVStream *st = file->ctx->streams[pkt->stream_index];
2760
2761     if (pkt->stream_index < file->nb_streams_warn)
2762         return;
2763     av_log(file->ctx, AV_LOG_WARNING,
2764            "New %s stream %d:%d at pos:%"PRId64" and DTS:%ss\n",
2765            av_get_media_type_string(st->codec->codec_type),
2766            input_index, pkt->stream_index,
2767            pkt->pos, av_ts2timestr(pkt->dts, &st->time_base));
2768     file->nb_streams_warn = pkt->stream_index + 1;
2769 }
2770
2771 static void set_encoder_id(OutputFile *of, OutputStream *ost)
2772 {
2773     AVDictionaryEntry *e;
2774
2775     uint8_t *encoder_string;
2776     int encoder_string_len;
2777     int format_flags = 0;
2778     int codec_flags = 0;
2779
2780     if (av_dict_get(ost->st->metadata, "encoder",  NULL, 0))
2781         return;
2782
2783     e = av_dict_get(of->opts, "fflags", NULL, 0);
2784     if (e) {
2785         const AVOption *o = av_opt_find(of->ctx, "fflags", NULL, 0, 0);
2786         if (!o)
2787             return;
2788         av_opt_eval_flags(of->ctx, o, e->value, &format_flags);
2789     }
2790     e = av_dict_get(ost->encoder_opts, "flags", NULL, 0);
2791     if (e) {
2792         const AVOption *o = av_opt_find(ost->enc_ctx, "flags", NULL, 0, 0);
2793         if (!o)
2794             return;
2795         av_opt_eval_flags(ost->enc_ctx, o, e->value, &codec_flags);
2796     }
2797
2798     encoder_string_len = sizeof(LIBAVCODEC_IDENT) + strlen(ost->enc->name) + 2;
2799     encoder_string     = av_mallocz(encoder_string_len);
2800     if (!encoder_string)
2801         exit_program(1);
2802
2803     if (!(format_flags & AVFMT_FLAG_BITEXACT) && !(codec_flags & AV_CODEC_FLAG_BITEXACT))
2804         av_strlcpy(encoder_string, LIBAVCODEC_IDENT " ", encoder_string_len);
2805     else
2806         av_strlcpy(encoder_string, "Lavc ", encoder_string_len);
2807     av_strlcat(encoder_string, ost->enc->name, encoder_string_len);
2808     av_dict_set(&ost->st->metadata, "encoder",  encoder_string,
2809                 AV_DICT_DONT_STRDUP_VAL | AV_DICT_DONT_OVERWRITE);
2810 }
2811
2812 static int transcode_init(void)
2813 {
2814     int ret = 0, i, j, k;
2815     AVFormatContext *oc;
2816     OutputStream *ost;
2817     InputStream *ist;
2818     char error[1024] = {0};
2819     int want_sdp = 1;
2820
2821     for (i = 0; i < nb_filtergraphs; i++) {
2822         FilterGraph *fg = filtergraphs[i];
2823         for (j = 0; j < fg->nb_outputs; j++) {
2824             OutputFilter *ofilter = fg->outputs[j];
2825             if (!ofilter->ost || ofilter->ost->source_index >= 0)
2826                 continue;
2827             if (fg->nb_inputs != 1)
2828                 continue;
2829             for (k = nb_input_streams-1; k >= 0 ; k--)
2830                 if (fg->inputs[0]->ist == input_streams[k])
2831                     break;
2832             ofilter->ost->source_index = k;
2833         }
2834     }
2835
2836     /* init framerate emulation */
2837     for (i = 0; i < nb_input_files; i++) {
2838         InputFile *ifile = input_files[i];
2839         if (ifile->rate_emu)
2840             for (j = 0; j < ifile->nb_streams; j++)
2841                 input_streams[j + ifile->ist_index]->start = av_gettime_relative();
2842     }
2843
2844     /* for each output stream, we compute the right encoding parameters */
2845     for (i = 0; i < nb_output_streams; i++) {
2846         AVCodecContext *enc_ctx;
2847         AVCodecContext *dec_ctx = NULL;
2848         ost = output_streams[i];
2849         oc  = output_files[ost->file_index]->ctx;
2850         ist = get_input_stream(ost);
2851
2852         if (ost->attachment_filename)
2853             continue;
2854
2855         enc_ctx = ost->stream_copy ? ost->st->codec : ost->enc_ctx;
2856
2857         if (ist) {
2858             dec_ctx = ist->dec_ctx;
2859
2860             ost->st->disposition          = ist->st->disposition;
2861             enc_ctx->chroma_sample_location = dec_ctx->chroma_sample_location;
2862         } else {
2863             for (j=0; j<oc->nb_streams; j++) {
2864                 AVStream *st = oc->streams[j];
2865                 if (st != ost->st && st->codec->codec_type == enc_ctx->codec_type)
2866                     break;
2867             }
2868             if (j == oc->nb_streams)
2869                 if (enc_ctx->codec_type == AVMEDIA_TYPE_AUDIO || enc_ctx->codec_type == AVMEDIA_TYPE_VIDEO)
2870                     ost->st->disposition = AV_DISPOSITION_DEFAULT;
2871         }
2872
2873         if (ost->stream_copy) {
2874             AVRational sar;
2875             uint64_t extra_size;
2876
2877             av_assert0(ist && !ost->filter);
2878
2879             extra_size = (uint64_t)dec_ctx->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE;
2880
2881             if (extra_size > INT_MAX) {
2882                 return AVERROR(EINVAL);
2883             }
2884
2885             /* if stream_copy is selected, no need to decode or encode */
2886             enc_ctx->codec_id   = dec_ctx->codec_id;
2887             enc_ctx->codec_type = dec_ctx->codec_type;
2888
2889             if (!enc_ctx->codec_tag) {
2890                 unsigned int codec_tag;
2891                 if (!oc->oformat->codec_tag ||
2892                      av_codec_get_id (oc->oformat->codec_tag, dec_ctx->codec_tag) == enc_ctx->codec_id ||
2893                      !av_codec_get_tag2(oc->oformat->codec_tag, dec_ctx->codec_id, &codec_tag))
2894                     enc_ctx->codec_tag = dec_ctx->codec_tag;
2895             }
2896
2897             enc_ctx->bit_rate       = dec_ctx->bit_rate;
2898             enc_ctx->rc_max_rate    = dec_ctx->rc_max_rate;
2899             enc_ctx->rc_buffer_size = dec_ctx->rc_buffer_size;
2900             enc_ctx->field_order    = dec_ctx->field_order;
2901             if (dec_ctx->extradata_size) {
2902                 enc_ctx->extradata      = av_mallocz(extra_size);
2903                 if (!enc_ctx->extradata) {
2904                     return AVERROR(ENOMEM);
2905                 }
2906                 memcpy(enc_ctx->extradata, dec_ctx->extradata, dec_ctx->extradata_size);
2907             }
2908             enc_ctx->extradata_size= dec_ctx->extradata_size;
2909             enc_ctx->bits_per_coded_sample  = dec_ctx->bits_per_coded_sample;
2910             enc_ctx->bits_per_raw_sample    = dec_ctx->bits_per_raw_sample;
2911
2912             if (!ost->frame_rate.num)
2913                 ost->frame_rate = ist->framerate;
2914             ost->st->avg_frame_rate = ost->frame_rate;
2915
2916             ret = avformat_transfer_internal_stream_timing_info(oc->oformat, ost->st, ist->st, copy_tb);
2917             if (ret < 0)
2918                 return ret;
2919
2920             if (ist->st->nb_side_data) {
2921                 ost->st->side_data = av_realloc_array(NULL, ist->st->nb_side_data,
2922                                                       sizeof(*ist->st->side_data));
2923                 if (!ost->st->side_data)
2924                     return AVERROR(ENOMEM);
2925
2926                 ost->st->nb_side_data = 0;
2927                 for (j = 0; j < ist->st->nb_side_data; j++) {
2928                     const AVPacketSideData *sd_src = &ist->st->side_data[j];
2929                     AVPacketSideData *sd_dst = &ost->st->side_data[ost->st->nb_side_data];
2930
2931                     if (ost->rotate_overridden && sd_src->type == AV_PKT_DATA_DISPLAYMATRIX)
2932                         continue;
2933
2934                     sd_dst->data = av_malloc(sd_src->size);
2935                     if (!sd_dst->data)
2936                         return AVERROR(ENOMEM);
2937                     memcpy(sd_dst->data, sd_src->data, sd_src->size);
2938                     sd_dst->size = sd_src->size;
2939                     sd_dst->type = sd_src->type;
2940                     ost->st->nb_side_data++;
2941                 }
2942             }
2943
2944             ost->parser = av_parser_init(enc_ctx->codec_id);
2945
2946             switch (enc_ctx->codec_type) {
2947             case AVMEDIA_TYPE_AUDIO:
2948                 if (audio_volume != 256) {
2949                     av_log(NULL, AV_LOG_FATAL, "-acodec copy and -vol are incompatible (frames are not decoded)\n");
2950                     exit_program(1);
2951                 }
2952                 enc_ctx->channel_layout     = dec_ctx->channel_layout;
2953                 enc_ctx->sample_rate        = dec_ctx->sample_rate;
2954                 enc_ctx->channels           = dec_ctx->channels;
2955                 enc_ctx->frame_size         = dec_ctx->frame_size;
2956                 enc_ctx->audio_service_type = dec_ctx->audio_service_type;
2957                 enc_ctx->block_align        = dec_ctx->block_align;
2958                 enc_ctx->initial_padding    = dec_ctx->delay;
2959                 enc_ctx->trailing_padding   = dec_ctx->trailing_padding;
2960                 enc_ctx->profile            = dec_ctx->profile;
2961 #if FF_API_AUDIOENC_DELAY
2962                 enc_ctx->delay              = dec_ctx->delay;
2963 #endif
2964                 if((enc_ctx->block_align == 1 || enc_ctx->block_align == 1152 || enc_ctx->block_align == 576) && enc_ctx->codec_id == AV_CODEC_ID_MP3)
2965                     enc_ctx->block_align= 0;
2966                 if(enc_ctx->codec_id == AV_CODEC_ID_AC3)
2967                     enc_ctx->block_align= 0;
2968                 break;
2969             case AVMEDIA_TYPE_VIDEO:
2970                 enc_ctx->pix_fmt            = dec_ctx->pix_fmt;
2971                 enc_ctx->colorspace         = dec_ctx->colorspace;
2972                 enc_ctx->color_range        = dec_ctx->color_range;
2973                 enc_ctx->color_primaries    = dec_ctx->color_primaries;
2974                 enc_ctx->color_trc          = dec_ctx->color_trc;
2975                 enc_ctx->width              = dec_ctx->width;
2976                 enc_ctx->height             = dec_ctx->height;
2977                 enc_ctx->has_b_frames       = dec_ctx->has_b_frames;
2978                 enc_ctx->profile            = dec_ctx->profile;
2979                 if (ost->frame_aspect_ratio.num) { // overridden by the -aspect cli option
2980                     sar =
2981                         av_mul_q(ost->frame_aspect_ratio,
2982                                  (AVRational){ enc_ctx->height, enc_ctx->width });
2983                     av_log(NULL, AV_LOG_WARNING, "Overriding aspect ratio "
2984                            "with stream copy may produce invalid files\n");
2985                 }
2986                 else if (ist->st->sample_aspect_ratio.num)
2987                     sar = ist->st->sample_aspect_ratio;
2988                 else
2989                     sar = dec_ctx->sample_aspect_ratio;
2990                 ost->st->sample_aspect_ratio = enc_ctx->sample_aspect_ratio = sar;
2991                 ost->st->avg_frame_rate = ist->st->avg_frame_rate;
2992                 ost->st->r_frame_rate = ist->st->r_frame_rate;
2993                 break;
2994             case AVMEDIA_TYPE_SUBTITLE:
2995                 enc_ctx->width  = dec_ctx->width;
2996                 enc_ctx->height = dec_ctx->height;
2997                 break;
2998             case AVMEDIA_TYPE_UNKNOWN:
2999             case AVMEDIA_TYPE_DATA:
3000             case AVMEDIA_TYPE_ATTACHMENT:
3001                 break;
3002             default:
3003                 abort();
3004             }
3005         } else {
3006             set_encoder_id(output_files[ost->file_index], ost);
3007
3008 #if CONFIG_LIBMFX
3009             if (qsv_transcode_init(ost))
3010                 exit_program(1);
3011 #endif
3012
3013 #if CONFIG_CUVID
3014             if (cuvid_transcode_init(ost))
3015                 exit_program(1);
3016 #endif
3017
3018             if ((enc_ctx->codec_type == AVMEDIA_TYPE_VIDEO ||
3019                  enc_ctx->codec_type == AVMEDIA_TYPE_AUDIO) &&
3020                  filtergraph_is_simple(ost->filter->graph)) {
3021                     FilterGraph *fg = ost->filter->graph;
3022                     if (configure_filtergraph(fg)) {
3023                         av_log(NULL, AV_LOG_FATAL, "Error opening filters!\n");
3024                         exit_program(1);
3025                     }
3026             }
3027
3028             if (enc_ctx->codec_type == AVMEDIA_TYPE_VIDEO) {
3029                 if (!ost->frame_rate.num)
3030                     ost->frame_rate = av_buffersink_get_frame_rate(ost->filter->filter);
3031                 if (ist && !ost->frame_rate.num)
3032                     ost->frame_rate = ist->framerate;
3033                 if (ist && !ost->frame_rate.num)
3034                     ost->frame_rate = ist->st->r_frame_rate;
3035                 if (ist && !ost->frame_rate.num) {
3036                     ost->frame_rate = (AVRational){25, 1};
3037                     av_log(NULL, AV_LOG_WARNING,
3038                            "No information "
3039                            "about the input framerate is available. Falling "
3040                            "back to a default value of 25fps for output stream #%d:%d. Use the -r option "
3041                            "if you want a different framerate.\n",
3042                            ost->file_index, ost->index);
3043                 }
3044 //                    ost->frame_rate = ist->st->avg_frame_rate.num ? ist->st->avg_frame_rate : (AVRational){25, 1};
3045                 if (ost->enc && ost->enc->supported_framerates && !ost->force_fps) {
3046                     int idx = av_find_nearest_q_idx(ost->frame_rate, ost->enc->supported_framerates);
3047                     ost->frame_rate = ost->enc->supported_framerates[idx];
3048                 }
3049                 // reduce frame rate for mpeg4 to be within the spec limits
3050                 if (enc_ctx->codec_id == AV_CODEC_ID_MPEG4) {
3051                     av_reduce(&ost->frame_rate.num, &ost->frame_rate.den,
3052                               ost->frame_rate.num, ost->frame_rate.den, 65535);
3053                 }
3054             }
3055
3056             switch (enc_ctx->codec_type) {
3057             case AVMEDIA_TYPE_AUDIO:
3058                 enc_ctx->sample_fmt     = ost->filter->filter->inputs[0]->format;
3059                 if (dec_ctx)
3060                     enc_ctx->bits_per_raw_sample = FFMIN(dec_ctx->bits_per_raw_sample,
3061                                                          av_get_bytes_per_sample(enc_ctx->sample_fmt) << 3);
3062                 enc_ctx->sample_rate    = ost->filter->filter->inputs[0]->sample_rate;
3063                 enc_ctx->channel_layout = ost->filter->filter->inputs[0]->channel_layout;
3064                 enc_ctx->channels       = avfilter_link_get_channels(ost->filter->filter->inputs[0]);
3065                 enc_ctx->time_base      = (AVRational){ 1, enc_ctx->sample_rate };
3066                 break;
3067             case AVMEDIA_TYPE_VIDEO:
3068                 enc_ctx->time_base = av_inv_q(ost->frame_rate);
3069                 if (!(enc_ctx->time_base.num && enc_ctx->time_base.den))
3070                     enc_ctx->time_base = ost->filter->filter->inputs[0]->time_base;
3071                 if (   av_q2d(enc_ctx->time_base) < 0.001 && video_sync_method != VSYNC_PASSTHROUGH
3072                    && (video_sync_method == VSYNC_CFR || video_sync_method == VSYNC_VSCFR || (video_sync_method == VSYNC_AUTO && !(oc->oformat->flags & AVFMT_VARIABLE_FPS)))){
3073                     av_log(oc, AV_LOG_WARNING, "Frame rate very high for a muxer not efficiently supporting it.\n"
3074                                                "Please consider specifying a lower framerate, a different muxer or -vsync 2\n");
3075                 }
3076                 for (j = 0; j < ost->forced_kf_count; j++)
3077                     ost->forced_kf_pts[j] = av_rescale_q(ost->forced_kf_pts[j],
3078                                                          AV_TIME_BASE_Q,
3079                                                          enc_ctx->time_base);
3080
3081                 enc_ctx->width  = ost->filter->filter->inputs[0]->w;
3082                 enc_ctx->height = ost->filter->filter->inputs[0]->h;
3083                 enc_ctx->sample_aspect_ratio = ost->st->sample_aspect_ratio =
3084                     ost->frame_aspect_ratio.num ? // overridden by the -aspect cli option
3085                     av_mul_q(ost->frame_aspect_ratio, (AVRational){ enc_ctx->height, enc_ctx->width }) :
3086                     ost->filter->filter->inputs[0]->sample_aspect_ratio;
3087                 if (!strncmp(ost->enc->name, "libx264", 7) &&
3088                     enc_ctx->pix_fmt == AV_PIX_FMT_NONE &&
3089                     ost->filter->filter->inputs[0]->format != AV_PIX_FMT_YUV420P)
3090                     av_log(NULL, AV_LOG_WARNING,
3091                            "No pixel format specified, %s for H.264 encoding chosen.\n"
3092                            "Use -pix_fmt yuv420p for compatibility with outdated media players.\n",
3093                            av_get_pix_fmt_name(ost->filter->filter->inputs[0]->format));
3094                 if (!strncmp(ost->enc->name, "mpeg2video", 10) &&
3095                     enc_ctx->pix_fmt == AV_PIX_FMT_NONE &&
3096                     ost->filter->filter->inputs[0]->format != AV_PIX_FMT_YUV420P)
3097                     av_log(NULL, AV_LOG_WARNING,
3098                            "No pixel format specified, %s for MPEG-2 encoding chosen.\n"
3099                            "Use -pix_fmt yuv420p for compatibility with outdated media players.\n",
3100                            av_get_pix_fmt_name(ost->filter->filter->inputs[0]->format));
3101                 enc_ctx->pix_fmt = ost->filter->filter->inputs[0]->format;
3102                 if (dec_ctx)
3103                     enc_ctx->bits_per_raw_sample = FFMIN(dec_ctx->bits_per_raw_sample,
3104                                                          av_pix_fmt_desc_get(enc_ctx->pix_fmt)->comp[0].depth);
3105
3106                 ost->st->avg_frame_rate = ost->frame_rate;
3107
3108                 if (!dec_ctx ||
3109                     enc_ctx->width   != dec_ctx->width  ||
3110                     enc_ctx->height  != dec_ctx->height ||
3111                     enc_ctx->pix_fmt != dec_ctx->pix_fmt) {
3112                     enc_ctx->bits_per_raw_sample = frame_bits_per_raw_sample;
3113                 }
3114
3115                 if (ost->forced_keyframes) {
3116                     if (!strncmp(ost->forced_keyframes, "expr:", 5)) {
3117                         ret = av_expr_parse(&ost->forced_keyframes_pexpr, ost->forced_keyframes+5,
3118                                             forced_keyframes_const_names, NULL, NULL, NULL, NULL, 0, NULL);
3119                         if (ret < 0) {
3120                             av_log(NULL, AV_LOG_ERROR,
3121                                    "Invalid force_key_frames expression '%s'\n", ost->forced_keyframes+5);
3122                             return ret;
3123                         }
3124                         ost->forced_keyframes_expr_const_values[FKF_N] = 0;
3125                         ost->forced_keyframes_expr_const_values[FKF_N_FORCED] = 0;
3126                         ost->forced_keyframes_expr_const_values[FKF_PREV_FORCED_N] = NAN;
3127                         ost->forced_keyframes_expr_const_values[FKF_PREV_FORCED_T] = NAN;
3128
3129                         // Don't parse the 'forced_keyframes' in case of 'keep-source-keyframes',
3130                         // parse it only for static kf timings
3131                     } else if(strncmp(ost->forced_keyframes, "source", 6)) {
3132                         parse_forced_key_frames(ost->forced_keyframes, ost, ost->enc_ctx);
3133                     }
3134                 }
3135                 break;
3136             case AVMEDIA_TYPE_SUBTITLE:
3137                 enc_ctx->time_base = (AVRational){1, 1000};
3138                 if (!enc_ctx->width) {
3139                     enc_ctx->width     = input_streams[ost->source_index]->st->codec->width;
3140                     enc_ctx->height    = input_streams[ost->source_index]->st->codec->height;
3141                 }
3142                 break;
3143             case AVMEDIA_TYPE_DATA:
3144                 break;
3145             default:
3146                 abort();
3147                 break;
3148             }
3149         }
3150
3151         if (ost->disposition) {
3152             static const AVOption opts[] = {
3153                 { "disposition"         , NULL, 0, AV_OPT_TYPE_FLAGS, { .i64 = 0 }, INT64_MIN, INT64_MAX, .unit = "flags" },
3154                 { "default"             , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_DEFAULT           },    .unit = "flags" },
3155                 { "dub"                 , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_DUB               },    .unit = "flags" },
3156                 { "original"            , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_ORIGINAL          },    .unit = "flags" },
3157                 { "comment"             , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_COMMENT           },    .unit = "flags" },
3158                 { "lyrics"              , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_LYRICS            },    .unit = "flags" },
3159                 { "karaoke"             , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_KARAOKE           },    .unit = "flags" },
3160                 { "forced"              , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_FORCED            },    .unit = "flags" },
3161                 { "hearing_impaired"    , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_HEARING_IMPAIRED  },    .unit = "flags" },
3162                 { "visual_impaired"     , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_VISUAL_IMPAIRED   },    .unit = "flags" },
3163                 { "clean_effects"       , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_CLEAN_EFFECTS     },    .unit = "flags" },
3164                 { "captions"            , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_CAPTIONS          },    .unit = "flags" },
3165                 { "descriptions"        , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_DESCRIPTIONS      },    .unit = "flags" },
3166                 { "metadata"            , NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_METADATA          },    .unit = "flags" },
3167                 { NULL },
3168             };
3169             static const AVClass class = {
3170                 .class_name = "",
3171                 .item_name  = av_default_item_name,
3172                 .option     = opts,
3173                 .version    = LIBAVUTIL_VERSION_INT,
3174             };
3175             const AVClass *pclass = &class;
3176
3177             ret = av_opt_eval_flags(&pclass, &opts[0], ost->disposition, &ost->st->disposition);
3178             if (ret < 0)
3179                 goto dump_format;
3180         }
3181     }
3182
3183     /* init input streams */
3184     for (i = 0; i < nb_input_streams; i++)
3185         if ((ret = init_input_stream(i, error, sizeof(error))) < 0) {
3186             for (i = 0; i < nb_output_streams; i++) {
3187                 ost = output_streams[i];
3188                 avcodec_close(ost->enc_ctx);
3189             }
3190             goto dump_format;
3191         }
3192
3193     /* open each encoder */
3194     for (i = 0; i < nb_output_streams; i++) {
3195         ret = init_output_stream(output_streams[i], error, sizeof(error));
3196         if (ret < 0)
3197             goto dump_format;
3198     }
3199
3200     /* discard unused programs */
3201     for (i = 0; i < nb_input_files; i++) {
3202         InputFile *ifile = input_files[i];
3203         for (j = 0; j < ifile->ctx->nb_programs; j++) {
3204             AVProgram *p = ifile->ctx->programs[j];
3205             int discard  = AVDISCARD_ALL;
3206
3207             for (k = 0; k < p->nb_stream_indexes; k++)
3208                 if (!input_streams[ifile->ist_index + p->stream_index[k]]->discard) {
3209                     discard = AVDISCARD_DEFAULT;
3210                     break;
3211                 }
3212             p->discard = discard;
3213         }
3214     }
3215
3216     /* open files and write file headers */
3217     for (i = 0; i < nb_output_files; i++) {
3218         oc = output_files[i]->ctx;
3219         oc->interrupt_callback = int_cb;
3220         if ((ret = avformat_write_header(oc, &output_files[i]->opts)) < 0) {
3221             snprintf(error, sizeof(error),
3222                      "Could not write header for output file #%d "
3223                      "(incorrect codec parameters ?): %s",
3224                      i, av_err2str(ret));
3225             ret = AVERROR(EINVAL);
3226             goto dump_format;
3227         }
3228 //         assert_avoptions(output_files[i]->opts);
3229         if (strcmp(oc->oformat->name, "rtp")) {
3230             want_sdp = 0;
3231         }
3232     }
3233
3234  dump_format:
3235     /* dump the file output parameters - cannot be done before in case
3236        of stream copy */
3237     for (i = 0; i < nb_output_files; i++) {
3238         av_dump_format(output_files[i]->ctx, i, output_files[i]->ctx->filename, 1);
3239     }
3240
3241     /* dump the stream mapping */
3242     av_log(NULL, AV_LOG_INFO, "Stream mapping:\n");
3243     for (i = 0; i < nb_input_streams; i++) {
3244         ist = input_streams[i];
3245
3246         for (j = 0; j < ist->nb_filters; j++) {
3247             if (!filtergraph_is_simple(ist->filters[j]->graph)) {
3248                 av_log(NULL, AV_LOG_INFO, "  Stream #%d:%d (%s) -> %s",
3249                        ist->file_index, ist->st->index, ist->dec ? ist->dec->name : "?",
3250                        ist->filters[j]->name);
3251                 if (nb_filtergraphs > 1)
3252                     av_log(NULL, AV_LOG_INFO, " (graph %d)", ist->filters[j]->graph->index);
3253                 av_log(NULL, AV_LOG_INFO, "\n");
3254             }
3255         }
3256     }
3257
3258     for (i = 0; i < nb_output_streams; i++) {
3259         ost = output_streams[i];
3260
3261         if (ost->attachment_filename) {
3262             /* an attached file */
3263             av_log(NULL, AV_LOG_INFO, "  File %s -> Stream #%d:%d\n",
3264                    ost->attachment_filename, ost->file_index, ost->index);
3265             continue;
3266         }
3267
3268         if (ost->filter && !filtergraph_is_simple(ost->filter->graph)) {
3269             /* output from a complex graph */
3270             av_log(NULL, AV_LOG_INFO, "  %s", ost->filter->name);
3271             if (nb_filtergraphs > 1)
3272                 av_log(NULL, AV_LOG_INFO, " (graph %d)", ost->filter->graph->index);
3273
3274             av_log(NULL, AV_LOG_INFO, " -> Stream #%d:%d (%s)\n", ost->file_index,
3275                    ost->index, ost->enc ? ost->enc->name : "?");
3276             continue;
3277         }
3278
3279         av_log(NULL, AV_LOG_INFO, "  Stream #%d:%d -> #%d:%d",
3280                input_streams[ost->source_index]->file_index,
3281                input_streams[ost->source_index]->st->index,
3282                ost->file_index,
3283                ost->index);
3284         if (ost->sync_ist != input_streams[ost->source_index])
3285             av_log(NULL, AV_LOG_INFO, " [sync #%d:%d]",
3286                    ost->sync_ist->file_index,
3287                    ost->sync_ist->st->index);
3288         if (ost->stream_copy)
3289             av_log(NULL, AV_LOG_INFO, " (copy)");
3290         else {
3291             const AVCodec *in_codec    = input_streams[ost->source_index]->dec;
3292             const AVCodec *out_codec   = ost->enc;
3293             const char *decoder_name   = "?";
3294             const char *in_codec_name  = "?";
3295             const char *encoder_name   = "?";
3296             const char *out_codec_name = "?";
3297             const AVCodecDescriptor *desc;
3298
3299             if (in_codec) {
3300                 decoder_name  = in_codec->name;
3301                 desc = avcodec_descriptor_get(in_codec->id);
3302                 if (desc)
3303                     in_codec_name = desc->name;
3304                 if (!strcmp(decoder_name, in_codec_name))
3305                     decoder_name = "native";
3306             }
3307
3308             if (out_codec) {
3309                 encoder_name   = out_codec->name;
3310                 desc = avcodec_descriptor_get(out_codec->id);
3311                 if (desc)
3312                     out_codec_name = desc->name;
3313                 if (!strcmp(encoder_name, out_codec_name))
3314                     encoder_name = "native";
3315             }
3316
3317             av_log(NULL, AV_LOG_INFO, " (%s (%s) -> %s (%s))",
3318                    in_codec_name, decoder_name,
3319                    out_codec_name, encoder_name);
3320         }
3321         av_log(NULL, AV_LOG_INFO, "\n");
3322     }
3323
3324     if (ret) {
3325         av_log(NULL, AV_LOG_ERROR, "%s\n", error);
3326         return ret;
3327     }
3328
3329     if (sdp_filename || want_sdp) {
3330         print_sdp();
3331     }
3332
3333     transcode_init_done = 1;
3334
3335     return 0;
3336 }
3337
3338 /* Return 1 if there remain streams where more output is wanted, 0 otherwise. */
3339 static int need_output(void)
3340 {
3341     int i;
3342
3343     for (i = 0; i < nb_output_streams; i++) {
3344         OutputStream *ost    = output_streams[i];
3345         OutputFile *of       = output_files[ost->file_index];
3346         AVFormatContext *os  = output_files[ost->file_index]->ctx;
3347
3348         if (ost->finished ||
3349             (os->pb && avio_tell(os->pb) >= of->limit_filesize))
3350             continue;
3351         if (ost->frame_number >= ost->max_frames) {
3352             int j;
3353             for (j = 0; j < of->ctx->nb_streams; j++)
3354                 close_output_stream(output_streams[of->ost_index + j]);
3355             continue;
3356         }
3357
3358         return 1;
3359     }
3360
3361     return 0;
3362 }
3363
3364 /**
3365  * Select the output stream to process.
3366  *
3367  * @return  selected output stream, or NULL if none available
3368  */
3369 static OutputStream *choose_output(void)
3370 {
3371     int i;
3372     int64_t opts_min = INT64_MAX;
3373     OutputStream *ost_min = NULL;
3374
3375     for (i = 0; i < nb_output_streams; i++) {
3376         OutputStream *ost = output_streams[i];
3377         int64_t opts = ost->st->cur_dts == AV_NOPTS_VALUE ? INT64_MIN :
3378                        av_rescale_q(ost->st->cur_dts, ost->st->time_base,
3379                                     AV_TIME_BASE_Q);
3380         if (ost->st->cur_dts == AV_NOPTS_VALUE)
3381             av_log(NULL, AV_LOG_DEBUG, "cur_dts is invalid (this is harmless if it occurs once at the start per stream)\n");
3382
3383         if (!ost->finished && opts < opts_min) {
3384             opts_min = opts;
3385             ost_min  = ost->unavailable ? NULL : ost;
3386         }
3387     }
3388     return ost_min;
3389 }
3390
3391 static void set_tty_echo(int on)
3392 {
3393 #if HAVE_TERMIOS_H
3394     struct termios tty;
3395     if (tcgetattr(0, &tty) == 0) {
3396         if (on) tty.c_lflag |= ECHO;
3397         else    tty.c_lflag &= ~ECHO;
3398         tcsetattr(0, TCSANOW, &tty);
3399     }
3400 #endif
3401 }
3402
3403 static int check_keyboard_interaction(int64_t cur_time)
3404 {
3405     int i, ret, key;
3406     static int64_t last_time;
3407     if (received_nb_signals)
3408         return AVERROR_EXIT;
3409     /* read_key() returns 0 on EOF */
3410     if(cur_time - last_time >= 100000 && !run_as_daemon){
3411         key =  read_key();
3412         last_time = cur_time;
3413     }else
3414         key = -1;
3415     if (key == 'q')
3416         return AVERROR_EXIT;
3417     if (key == '+') av_log_set_level(av_log_get_level()+10);
3418     if (key == '-') av_log_set_level(av_log_get_level()-10);
3419     if (key == 's') qp_hist     ^= 1;
3420     if (key == 'h'){
3421         if (do_hex_dump){
3422             do_hex_dump = do_pkt_dump = 0;
3423         } else if(do_pkt_dump){
3424             do_hex_dump = 1;
3425         } else
3426             do_pkt_dump = 1;
3427         av_log_set_level(AV_LOG_DEBUG);
3428     }
3429     if (key == 'c' || key == 'C'){
3430         char buf[4096], target[64], command[256], arg[256] = {0};
3431         double time;
3432         int k, n = 0;
3433         fprintf(stderr, "\nEnter command: <target>|all <time>|-1 <command>[ <argument>]\n");
3434         i = 0;
3435         set_tty_echo(1);
3436         while ((k = read_key()) != '\n' && k != '\r' && i < sizeof(buf)-1)
3437             if (k > 0)
3438                 buf[i++] = k;
3439         buf[i] = 0;
3440         set_tty_echo(0);
3441         fprintf(stderr, "\n");
3442         if (k > 0 &&
3443             (n = sscanf(buf, "%63[^ ] %lf %255[^ ] %255[^\n]", target, &time, command, arg)) >= 3) {
3444             av_log(NULL, AV_LOG_DEBUG, "Processing command target:%s time:%f command:%s arg:%s",
3445                    target, time, command, arg);
3446             for (i = 0; i < nb_filtergraphs; i++) {
3447                 FilterGraph *fg = filtergraphs[i];
3448                 if (fg->graph) {
3449                     if (time < 0) {
3450                         ret = avfilter_graph_send_command(fg->graph, target, command, arg, buf, sizeof(buf),
3451                                                           key == 'c' ? AVFILTER_CMD_FLAG_ONE : 0);
3452                         fprintf(stderr, "Command reply for stream %d: ret:%d res:\n%s", i, ret, buf);
3453                     } else if (key == 'c') {
3454                         fprintf(stderr, "Queuing commands only on filters supporting the specific command is unsupported\n");
3455                         ret = AVERROR_PATCHWELCOME;
3456                     } else {
3457                         ret = avfilter_graph_queue_command(fg->graph, target, command, arg, 0, time);
3458                         if (ret < 0)
3459                             fprintf(stderr, "Queuing command failed with error %s\n", av_err2str(ret));
3460                     }
3461                 }
3462             }
3463         } else {
3464             av_log(NULL, AV_LOG_ERROR,
3465                    "Parse error, at least 3 arguments were expected, "
3466                    "only %d given in string '%s'\n", n, buf);
3467         }
3468     }
3469     if (key == 'd' || key == 'D'){
3470         int debug=0;
3471         if(key == 'D') {
3472             debug = input_streams[0]->st->codec->debug<<1;
3473             if(!debug) debug = 1;
3474             while(debug & (FF_DEBUG_DCT_COEFF|FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) //unsupported, would just crash
3475                 debug += debug;
3476         }else{
3477             char buf[32];
3478             int k = 0;
3479             i = 0;
3480             set_tty_echo(1);
3481             while ((k = read_key()) != '\n' && k != '\r' && i < sizeof(buf)-1)
3482                 if (k > 0)
3483                     buf[i++] = k;
3484             buf[i] = 0;
3485             set_tty_echo(0);
3486             fprintf(stderr, "\n");
3487             if (k <= 0 || sscanf(buf, "%d", &debug)!=1)
3488                 fprintf(stderr,"error parsing debug value\n");
3489         }
3490         for(i=0;i<nb_input_streams;i++) {
3491             input_streams[i]->st->codec->debug = debug;
3492         }
3493         for(i=0;i<nb_output_streams;i++) {
3494             OutputStream *ost = output_streams[i];
3495             ost->enc_ctx->debug = debug;
3496         }
3497         if(debug) av_log_set_level(AV_LOG_DEBUG);
3498         fprintf(stderr,"debug=%d\n", debug);
3499     }
3500     if (key == '?'){
3501         fprintf(stderr, "key    function\n"
3502                         "?      show this help\n"
3503                         "+      increase verbosity\n"
3504                         "-      decrease verbosity\n"
3505                         "c      Send command to first matching filter supporting it\n"
3506                         "C      Send/Que command to all matching filters\n"
3507                         "D      cycle through available debug modes\n"
3508                         "h      dump packets/hex press to cycle through the 3 states\n"
3509                         "q      quit\n"
3510                         "s      Show QP histogram\n"
3511         );
3512     }
3513     return 0;
3514 }
3515
3516 #if HAVE_PTHREADS
3517 static void *input_thread(void *arg)
3518 {
3519     InputFile *f = arg;
3520     unsigned flags = f->non_blocking ? AV_THREAD_MESSAGE_NONBLOCK : 0;
3521     int ret = 0;
3522
3523     while (1) {
3524         AVPacket pkt;
3525         ret = av_read_frame(f->ctx, &pkt);
3526
3527         if (ret == AVERROR(EAGAIN)) {
3528             av_usleep(10000);
3529             continue;
3530         }
3531         if (ret < 0) {
3532             av_thread_message_queue_set_err_recv(f->in_thread_queue, ret);
3533             break;
3534         }
3535         ret = av_thread_message_queue_send(f->in_thread_queue, &pkt, flags);
3536         if (flags && ret == AVERROR(EAGAIN)) {
3537             flags = 0;
3538             ret = av_thread_message_queue_send(f->in_thread_queue, &pkt, flags);
3539             av_log(f->ctx, AV_LOG_WARNING,
3540                    "Thread message queue blocking; consider raising the "
3541                    "thread_queue_size option (current value: %d)\n",
3542                    f->thread_queue_size);
3543         }
3544         if (ret < 0) {
3545             if (ret != AVERROR_EOF)
3546                 av_log(f->ctx, AV_LOG_ERROR,
3547                        "Unable to send packet to main thread: %s\n",
3548                        av_err2str(ret));
3549             av_packet_unref(&pkt);
3550             av_thread_message_queue_set_err_recv(f->in_thread_queue, ret);
3551             break;
3552         }
3553     }
3554
3555     return NULL;
3556 }
3557
3558 static void free_input_threads(void)
3559 {
3560     int i;
3561
3562     for (i = 0; i < nb_input_files; i++) {
3563         InputFile *f = input_files[i];
3564         AVPacket pkt;
3565
3566         if (!f || !f->in_thread_queue)
3567             continue;
3568         av_thread_message_queue_set_err_send(f->in_thread_queue, AVERROR_EOF);
3569         while (av_thread_message_queue_recv(f->in_thread_queue, &pkt, 0) >= 0)
3570             av_packet_unref(&pkt);
3571
3572         pthread_join(f->thread, NULL);
3573         f->joined = 1;
3574         av_thread_message_queue_free(&f->in_thread_queue);
3575     }
3576 }
3577
3578 static int init_input_threads(void)
3579 {
3580     int i, ret;
3581
3582     if (nb_input_files == 1)
3583         return 0;
3584
3585     for (i = 0; i < nb_input_files; i++) {
3586         InputFile *f = input_files[i];
3587
3588         if (f->ctx->pb ? !f->ctx->pb->seekable :
3589             strcmp(f->ctx->iformat->name, "lavfi"))
3590             f->non_blocking = 1;
3591         ret = av_thread_message_queue_alloc(&f->in_thread_queue,
3592                                             f->thread_queue_size, sizeof(AVPacket));
3593         if (ret < 0)
3594             return ret;
3595
3596         if ((ret = pthread_create(&f->thread, NULL, input_thread, f))) {
3597             av_log(NULL, AV_LOG_ERROR, "pthread_create failed: %s. Try to increase `ulimit -v` or decrease `ulimit -s`.\n", strerror(ret));
3598             av_thread_message_queue_free(&f->in_thread_queue);
3599             return AVERROR(ret);
3600         }
3601     }
3602     return 0;
3603 }
3604
3605 static int get_input_packet_mt(InputFile *f, AVPacket *pkt)
3606 {
3607     return av_thread_message_queue_recv(f->in_thread_queue, pkt,
3608                                         f->non_blocking ?
3609                                         AV_THREAD_MESSAGE_NONBLOCK : 0);
3610 }
3611 #endif
3612
3613 static int get_input_packet(InputFile *f, AVPacket *pkt)
3614 {
3615     if (f->rate_emu) {
3616         int i;
3617         for (i = 0; i < f->nb_streams; i++) {
3618             InputStream *ist = input_streams[f->ist_index + i];
3619             int64_t pts = av_rescale(ist->dts, 1000000, AV_TIME_BASE);
3620             int64_t now = av_gettime_relative() - ist->start;
3621             if (pts > now)
3622                 return AVERROR(EAGAIN);
3623         }
3624     }
3625
3626 #if HAVE_PTHREADS
3627     if (nb_input_files > 1)
3628         return get_input_packet_mt(f, pkt);
3629 #endif
3630     return av_read_frame(f->ctx, pkt);
3631 }
3632
3633 static int got_eagain(void)
3634 {
3635     int i;
3636     for (i = 0; i < nb_output_streams; i++)
3637         if (output_streams[i]->unavailable)
3638             return 1;
3639     return 0;
3640 }
3641
3642 static void reset_eagain(void)
3643 {
3644     int i;
3645     for (i = 0; i < nb_input_files; i++)
3646         input_files[i]->eagain = 0;
3647     for (i = 0; i < nb_output_streams; i++)
3648         output_streams[i]->unavailable = 0;
3649 }
3650
3651 // set duration to max(tmp, duration) in a proper time base and return duration's time_base
3652 static AVRational duration_max(int64_t tmp, int64_t *duration, AVRational tmp_time_base,
3653                                 AVRational time_base)
3654 {
3655     int ret;
3656
3657     if (!*duration) {
3658         *duration = tmp;
3659         return tmp_time_base;
3660     }
3661
3662     ret = av_compare_ts(*duration, time_base, tmp, tmp_time_base);
3663     if (ret < 0) {
3664         *duration = tmp;
3665         return tmp_time_base;
3666     }
3667
3668     return time_base;
3669 }
3670
3671 static int seek_to_start(InputFile *ifile, AVFormatContext *is)
3672 {
3673     InputStream *ist;
3674     AVCodecContext *avctx;
3675     int i, ret, has_audio = 0;
3676     int64_t duration = 0;
3677
3678     ret = av_seek_frame(is, -1, is->start_time, 0);
3679     if (ret < 0)
3680         return ret;
3681
3682     for (i = 0; i < ifile->nb_streams; i++) {
3683         ist   = input_streams[ifile->ist_index + i];
3684         avctx = ist->dec_ctx;
3685
3686         // flush decoders
3687         if (ist->decoding_needed) {
3688             process_input_packet(ist, NULL, 1);
3689             avcodec_flush_buffers(avctx);
3690         }
3691
3692         /* duration is the length of the last frame in a stream
3693          * when audio stream is present we don't care about
3694          * last video frame length because it's not defined exactly */
3695         if (avctx->codec_type == AVMEDIA_TYPE_AUDIO && ist->nb_samples)
3696             has_audio = 1;
3697     }
3698
3699     for (i = 0; i < ifile->nb_streams; i++) {
3700         ist   = input_streams[ifile->ist_index + i];
3701         avctx = ist->dec_ctx;
3702
3703         if (has_audio) {
3704             if (avctx->codec_type == AVMEDIA_TYPE_AUDIO && ist->nb_samples) {
3705                 AVRational sample_rate = {1, avctx->sample_rate};
3706
3707                 duration = av_rescale_q(ist->nb_samples, sample_rate, ist->st->time_base);
3708             } else
3709                 continue;
3710         } else {
3711             if (ist->framerate.num) {
3712                 duration = av_rescale_q(1, ist->framerate, ist->st->time_base);
3713             } else if (ist->st->avg_frame_rate.num) {
3714                 duration = av_rescale_q(1, ist->st->avg_frame_rate, ist->st->time_base);
3715             } else duration = 1;
3716         }
3717         if (!ifile->duration)
3718             ifile->time_base = ist->st->time_base;
3719         /* the total duration of the stream, max_pts - min_pts is
3720          * the duration of the stream without the last frame */
3721         duration += ist->max_pts - ist->min_pts;
3722         ifile->time_base = duration_max(duration, &ifile->duration, ist->st->time_base,
3723                                         ifile->time_base);
3724     }
3725
3726     if (ifile->loop > 0)
3727         ifile->loop--;
3728
3729     return ret;
3730 }
3731
3732 /*
3733  * Return
3734  * - 0 -- one packet was read and processed
3735  * - AVERROR(EAGAIN) -- no packets were available for selected file,
3736  *   this function should be called again
3737  * - AVERROR_EOF -- this function should not be called again
3738  */
3739 static int process_input(int file_index)
3740 {
3741     InputFile *ifile = input_files[file_index];
3742     AVFormatContext *is;
3743     InputStream *ist;
3744     AVPacket pkt;
3745     int ret, i, j;
3746     int64_t duration;
3747     int64_t pkt_dts;
3748
3749     is  = ifile->ctx;
3750     ret = get_input_packet(ifile, &pkt);
3751
3752     if (ret == AVERROR(EAGAIN)) {
3753         ifile->eagain = 1;
3754         return ret;
3755     }
3756     if (ret < 0 && ifile->loop) {
3757         if ((ret = seek_to_start(ifile, is)) < 0)
3758             return ret;
3759         ret = get_input_packet(ifile, &pkt);
3760         if (ret == AVERROR(EAGAIN)) {
3761             ifile->eagain = 1;
3762             return ret;
3763         }
3764     }
3765     if (ret < 0) {
3766         if (ret != AVERROR_EOF) {
3767             print_error(is->filename, ret);
3768             if (exit_on_error)
3769                 exit_program(1);
3770         }
3771
3772         for (i = 0; i < ifile->nb_streams; i++) {
3773             ist = input_streams[ifile->ist_index + i];
3774             if (ist->decoding_needed) {
3775                 ret = process_input_packet(ist, NULL, 0);
3776                 if (ret>0)
3777                     return 0;
3778             }
3779
3780             /* mark all outputs that don't go through lavfi as finished */
3781             for (j = 0; j < nb_output_streams; j++) {
3782                 OutputStream *ost = output_streams[j];
3783
3784                 if (ost->source_index == ifile->ist_index + i &&
3785                     (ost->stream_copy || ost->enc->type == AVMEDIA_TYPE_SUBTITLE))
3786                     finish_output_stream(ost);
3787             }
3788         }
3789
3790         ifile->eof_reached = 1;
3791         return AVERROR(EAGAIN);
3792     }
3793
3794     reset_eagain();
3795
3796     if (do_pkt_dump) {
3797         av_pkt_dump_log2(NULL, AV_LOG_INFO, &pkt, do_hex_dump,
3798                          is->streams[pkt.stream_index]);
3799     }
3800     /* the following test is needed in case new streams appear
3801        dynamically in stream : we ignore them */
3802     if (pkt.stream_index >= ifile->nb_streams) {
3803         report_new_stream(file_index, &pkt);
3804         goto discard_packet;
3805     }
3806
3807     ist = input_streams[ifile->ist_index + pkt.stream_index];
3808
3809     ist->data_size += pkt.size;
3810     ist->nb_packets++;
3811
3812     if (ist->discard)
3813         goto discard_packet;
3814
3815     if (exit_on_error && (pkt.flags & AV_PKT_FLAG_CORRUPT)) {
3816         av_log(NULL, AV_LOG_FATAL, "%s: corrupt input packet in stream %d\n", is->filename, pkt.stream_index);
3817         exit_program(1);
3818     }
3819
3820     if (debug_ts) {
3821         av_log(NULL, AV_LOG_INFO, "demuxer -> ist_index:%d type:%s "
3822                "next_dts:%s next_dts_time:%s next_pts:%s next_pts_time:%s pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s off:%s off_time:%s\n",
3823                ifile->ist_index + pkt.stream_index, av_get_media_type_string(ist->dec_ctx->codec_type),
3824                av_ts2str(ist->next_dts), av_ts2timestr(ist->next_dts, &AV_TIME_BASE_Q),
3825                av_ts2str(ist->next_pts), av_ts2timestr(ist->next_pts, &AV_TIME_BASE_Q),
3826                av_ts2str(pkt.pts), av_ts2timestr(pkt.pts, &ist->st->time_base),
3827                av_ts2str(pkt.dts), av_ts2timestr(pkt.dts, &ist->st->time_base),
3828                av_ts2str(input_files[ist->file_index]->ts_offset),
3829                av_ts2timestr(input_files[ist->file_index]->ts_offset, &AV_TIME_BASE_Q));
3830     }
3831
3832     if(!ist->wrap_correction_done && is->start_time != AV_NOPTS_VALUE && ist->st->pts_wrap_bits < 64){
3833         int64_t stime, stime2;
3834         // Correcting starttime based on the enabled streams
3835         // FIXME this ideally should be done before the first use of starttime but we do not know which are the enabled streams at that point.
3836         //       so we instead do it here as part of discontinuity handling
3837         if (   ist->next_dts == AV_NOPTS_VALUE
3838             && ifile->ts_offset == -is->start_time
3839             && (is->iformat->flags & AVFMT_TS_DISCONT)) {
3840             int64_t new_start_time = INT64_MAX;
3841             for (i=0; i<is->nb_streams; i++) {
3842                 AVStream *st = is->streams[i];
3843                 if(st->discard == AVDISCARD_ALL || st->start_time == AV_NOPTS_VALUE)
3844                     continue;
3845                 new_start_time = FFMIN(new_start_time, av_rescale_q(st->start_time, st->time_base, AV_TIME_BASE_Q));
3846             }
3847             if (new_start_time > is->start_time) {
3848                 av_log(is, AV_LOG_VERBOSE, "Correcting start time by %"PRId64"\n", new_start_time - is->start_time);
3849                 ifile->ts_offset = -new_start_time;
3850             }
3851         }
3852
3853         stime = av_rescale_q(is->start_time, AV_TIME_BASE_Q, ist->st->time_base);
3854         stime2= stime + (1ULL<<ist->st->pts_wrap_bits);
3855         ist->wrap_correction_done = 1;
3856
3857         if(stime2 > stime && pkt.dts != AV_NOPTS_VALUE && pkt.dts > stime + (1LL<<(ist->st->pts_wrap_bits-1))) {
3858             pkt.dts -= 1ULL<<ist->st->pts_wrap_bits;
3859             ist->wrap_correction_done = 0;
3860         }
3861         if(stime2 > stime && pkt.pts != AV_NOPTS_VALUE && pkt.pts > stime + (1LL<<(ist->st->pts_wrap_bits-1))) {
3862             pkt.pts -= 1ULL<<ist->st->pts_wrap_bits;
3863             ist->wrap_correction_done = 0;
3864         }
3865     }
3866
3867     /* add the stream-global side data to the first packet */
3868     if (ist->nb_packets == 1) {
3869         if (ist->st->nb_side_data)
3870             av_packet_split_side_data(&pkt);
3871         for (i = 0; i < ist->st->nb_side_data; i++) {
3872             AVPacketSideData *src_sd = &ist->st->side_data[i];
3873             uint8_t *dst_data;
3874
3875             if (av_packet_get_side_data(&pkt, src_sd->type, NULL))
3876                 continue;
3877             if (ist->autorotate && src_sd->type == AV_PKT_DATA_DISPLAYMATRIX)
3878                 continue;
3879
3880             dst_data = av_packet_new_side_data(&pkt, src_sd->type, src_sd->size);
3881             if (!dst_data)
3882                 exit_program(1);
3883
3884             memcpy(dst_data, src_sd->data, src_sd->size);
3885         }
3886     }
3887
3888     if (pkt.dts != AV_NOPTS_VALUE)
3889         pkt.dts += av_rescale_q(ifile->ts_offset, AV_TIME_BASE_Q, ist->st->time_base);
3890     if (pkt.pts != AV_NOPTS_VALUE)
3891         pkt.pts += av_rescale_q(ifile->ts_offset, AV_TIME_BASE_Q, ist->st->time_base);
3892
3893     if (pkt.pts != AV_NOPTS_VALUE)
3894         pkt.pts *= ist->ts_scale;
3895     if (pkt.dts != AV_NOPTS_VALUE)
3896         pkt.dts *= ist->ts_scale;
3897
3898     pkt_dts = av_rescale_q_rnd(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q, AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX);
3899     if ((ist->dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO ||
3900          ist->dec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) &&
3901         pkt_dts != AV_NOPTS_VALUE && ist->next_dts == AV_NOPTS_VALUE && !copy_ts
3902         && (is->iformat->flags & AVFMT_TS_DISCONT) && ifile->last_ts != AV_NOPTS_VALUE) {
3903         int64_t delta   = pkt_dts - ifile->last_ts;
3904         if (delta < -1LL*dts_delta_threshold*AV_TIME_BASE ||
3905             delta >  1LL*dts_delta_threshold*AV_TIME_BASE){
3906             ifile->ts_offset -= delta;
3907             av_log(NULL, AV_LOG_DEBUG,
3908                    "Inter stream timestamp discontinuity %"PRId64", new offset= %"PRId64"\n",
3909                    delta, ifile->ts_offset);
3910             pkt.dts -= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
3911             if (pkt.pts != AV_NOPTS_VALUE)
3912                 pkt.pts -= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
3913         }
3914     }
3915
3916     duration = av_rescale_q(ifile->duration, ifile->time_base, ist->st->time_base);
3917     if (pkt.pts != AV_NOPTS_VALUE) {
3918         pkt.pts += duration;
3919         ist->max_pts = FFMAX(pkt.pts, ist->max_pts);
3920         ist->min_pts = FFMIN(pkt.pts, ist->min_pts);
3921     }
3922
3923     if (pkt.dts != AV_NOPTS_VALUE)
3924         pkt.dts += duration;
3925
3926     pkt_dts = av_rescale_q_rnd(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q, AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX);
3927     if ((ist->dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO ||
3928          ist->dec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) &&
3929          pkt_dts != AV_NOPTS_VALUE && ist->next_dts != AV_NOPTS_VALUE &&
3930         !copy_ts) {
3931         int64_t delta   = pkt_dts - ist->next_dts;
3932         if (is->iformat->flags & AVFMT_TS_DISCONT) {
3933             if (delta < -1LL*dts_delta_threshold*AV_TIME_BASE ||
3934                 delta >  1LL*dts_delta_threshold*AV_TIME_BASE ||
3935                 pkt_dts + AV_TIME_BASE/10 < FFMAX(ist->pts, ist->dts)) {
3936                 ifile->ts_offset -= delta;
3937                 av_log(NULL, AV_LOG_DEBUG,
3938                        "timestamp discontinuity %"PRId64", new offset= %"PRId64"\n",
3939                        delta, ifile->ts_offset);
3940                 pkt.dts -= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
3941                 if (pkt.pts != AV_NOPTS_VALUE)
3942                     pkt.pts -= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
3943             }
3944         } else {
3945             if ( delta < -1LL*dts_error_threshold*AV_TIME_BASE ||
3946                  delta >  1LL*dts_error_threshold*AV_TIME_BASE) {
3947                 av_log(NULL, AV_LOG_WARNING, "DTS %"PRId64", next:%"PRId64" st:%d invalid dropping\n", pkt.dts, ist->next_dts, pkt.stream_index);
3948                 pkt.dts = AV_NOPTS_VALUE;
3949             }
3950             if (pkt.pts != AV_NOPTS_VALUE){
3951                 int64_t pkt_pts = av_rescale_q(pkt.pts, ist->st->time_base, AV_TIME_BASE_Q);
3952                 delta   = pkt_pts - ist->next_dts;
3953                 if ( delta < -1LL*dts_error_threshold*AV_TIME_BASE ||
3954                      delta >  1LL*dts_error_threshold*AV_TIME_BASE) {
3955                     av_log(NULL, AV_LOG_WARNING, "PTS %"PRId64", next:%"PRId64" invalid dropping st:%d\n", pkt.pts, ist->next_dts, pkt.stream_index);
3956                     pkt.pts = AV_NOPTS_VALUE;
3957                 }
3958             }
3959         }
3960     }
3961
3962     if (pkt.dts != AV_NOPTS_VALUE)
3963         ifile->last_ts = av_rescale_q(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q);
3964
3965     if (debug_ts) {
3966         av_log(NULL, AV_LOG_INFO, "demuxer+ffmpeg -> ist_index:%d type:%s pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s off:%s off_time:%s\n",
3967                ifile->ist_index + pkt.stream_index, av_get_media_type_string(ist->dec_ctx->codec_type),
3968                av_ts2str(pkt.pts), av_ts2timestr(pkt.pts, &ist->st->time_base),
3969                av_ts2str(pkt.dts), av_ts2timestr(pkt.dts, &ist->st->time_base),
3970                av_ts2str(input_files[ist->file_index]->ts_offset),
3971                av_ts2timestr(input_files[ist->file_index]->ts_offset, &AV_TIME_BASE_Q));
3972     }
3973
3974     sub2video_heartbeat(ist, pkt.pts);
3975
3976     process_input_packet(ist, &pkt, 0);
3977
3978 discard_packet:
3979     av_packet_unref(&pkt);
3980
3981     return 0;
3982 }
3983
3984 /**
3985  * Perform a step of transcoding for the specified filter graph.
3986  *
3987  * @param[in]  graph     filter graph to consider
3988  * @param[out] best_ist  input stream where a frame would allow to continue
3989  * @return  0 for success, <0 for error
3990  */
3991 static int transcode_from_filter(FilterGraph *graph, InputStream **best_ist)
3992 {
3993     int i, ret;
3994     int nb_requests, nb_requests_max = 0;
3995     InputFilter *ifilter;
3996     InputStream *ist;
3997
3998     *best_ist = NULL;
3999     ret = avfilter_graph_request_oldest(graph->graph);
4000     if (ret >= 0)
4001         return reap_filters(0);
4002
4003     if (ret == AVERROR_EOF) {
4004         ret = reap_filters(1);
4005         for (i = 0; i < graph->nb_outputs; i++)
4006             close_output_stream(graph->outputs[i]->ost);
4007         return ret;
4008     }
4009     if (ret != AVERROR(EAGAIN))
4010         return ret;
4011
4012     for (i = 0; i < graph->nb_inputs; i++) {
4013         ifilter = graph->inputs[i];
4014         ist = ifilter->ist;
4015         if (input_files[ist->file_index]->eagain ||
4016             input_files[ist->file_index]->eof_reached)
4017             continue;
4018         nb_requests = av_buffersrc_get_nb_failed_requests(ifilter->filter);
4019         if (nb_requests > nb_requests_max) {
4020             nb_requests_max = nb_requests;
4021             *best_ist = ist;
4022         }
4023     }
4024
4025     if (!*best_ist)
4026         for (i = 0; i < graph->nb_outputs; i++)
4027             graph->outputs[i]->ost->unavailable = 1;
4028
4029     return 0;
4030 }
4031
4032 /**
4033  * Run a single step of transcoding.
4034  *
4035  * @return  0 for success, <0 for error
4036  */
4037 static int transcode_step(void)
4038 {
4039     OutputStream *ost;
4040     InputStream  *ist;
4041     int ret;
4042
4043     ost = choose_output();
4044     if (!ost) {
4045         if (got_eagain()) {
4046             reset_eagain();
4047             av_usleep(10000);
4048             return 0;
4049         }
4050         av_log(NULL, AV_LOG_VERBOSE, "No more inputs to read from, finishing.\n");
4051         return AVERROR_EOF;
4052     }
4053
4054     if (ost->filter) {
4055         if ((ret = transcode_from_filter(ost->filter->graph, &ist)) < 0)
4056             return ret;
4057         if (!ist)
4058             return 0;
4059     } else {
4060         av_assert0(ost->source_index >= 0);
4061         ist = input_streams[ost->source_index];
4062     }
4063
4064     ret = process_input(ist->file_index);
4065     if (ret == AVERROR(EAGAIN)) {
4066         if (input_files[ist->file_index]->eagain)
4067             ost->unavailable = 1;
4068         return 0;
4069     }
4070
4071     if (ret < 0)
4072         return ret == AVERROR_EOF ? 0 : ret;
4073
4074     return reap_filters(0);
4075 }
4076
4077 /*
4078  * The following code is the main loop of the file converter
4079  */
4080 static int transcode(void)
4081 {
4082     int ret, i;
4083     AVFormatContext *os;
4084     OutputStream *ost;
4085     InputStream *ist;
4086     int64_t timer_start;
4087     int64_t total_packets_written = 0;
4088
4089     ret = transcode_init();
4090     if (ret < 0)
4091         goto fail;
4092
4093     if (stdin_interaction) {
4094         av_log(NULL, AV_LOG_INFO, "Press [q] to stop, [?] for help\n");
4095     }
4096
4097     timer_start = av_gettime_relative();
4098
4099 #if HAVE_PTHREADS
4100     if ((ret = init_input_threads()) < 0)
4101         goto fail;
4102 #endif
4103
4104     while (!received_sigterm) {
4105         int64_t cur_time= av_gettime_relative();
4106
4107         /* if 'q' pressed, exits */
4108         if (stdin_interaction)
4109             if (check_keyboard_interaction(cur_time) < 0)
4110                 break;
4111
4112         /* check if there's any stream where output is still needed */
4113         if (!need_output()) {
4114             av_log(NULL, AV_LOG_VERBOSE, "No more output streams to write to, finishing.\n");
4115             break;
4116         }
4117
4118         ret = transcode_step();
4119         if (ret < 0 && ret != AVERROR_EOF) {
4120             char errbuf[128];
4121             av_strerror(ret, errbuf, sizeof(errbuf));
4122
4123             av_log(NULL, AV_LOG_ERROR, "Error while filtering: %s\n", errbuf);
4124             break;
4125         }
4126
4127         /* dump report by using the output first video and audio streams */
4128         print_report(0, timer_start, cur_time);
4129     }
4130 #if HAVE_PTHREADS
4131     free_input_threads();
4132 #endif
4133
4134     /* at the end of stream, we must flush the decoder buffers */
4135     for (i = 0; i < nb_input_streams; i++) {
4136         ist = input_streams[i];
4137         if (!input_files[ist->file_index]->eof_reached && ist->decoding_needed) {
4138             process_input_packet(ist, NULL, 0);
4139         }
4140     }
4141     flush_encoders();
4142
4143     term_exit();
4144
4145     /* write the trailer if needed and close file */
4146     for (i = 0; i < nb_output_files; i++) {
4147         os = output_files[i]->ctx;
4148         if ((ret = av_write_trailer(os)) < 0) {
4149             av_log(NULL, AV_LOG_ERROR, "Error writing trailer of %s: %s", os->filename, av_err2str(ret));
4150             if (exit_on_error)
4151                 exit_program(1);
4152         }
4153     }
4154
4155     /* dump report by using the first video and audio streams */
4156     print_report(1, timer_start, av_gettime_relative());
4157
4158     /* close each encoder */
4159     for (i = 0; i < nb_output_streams; i++) {
4160         ost = output_streams[i];
4161         if (ost->encoding_needed) {
4162             av_freep(&ost->enc_ctx->stats_in);
4163         }
4164         total_packets_written += ost->packets_written;
4165     }
4166
4167     if (!total_packets_written && (abort_on_flags & ABORT_ON_FLAG_EMPTY_OUTPUT)) {
4168         av_log(NULL, AV_LOG_FATAL, "Empty output\n");
4169         exit_program(1);
4170     }
4171
4172     /* close each decoder */
4173     for (i = 0; i < nb_input_streams; i++) {
4174         ist = input_streams[i];
4175         if (ist->decoding_needed) {
4176             avcodec_close(ist->dec_ctx);
4177             if (ist->hwaccel_uninit)
4178                 ist->hwaccel_uninit(ist->dec_ctx);
4179         }
4180     }
4181
4182     av_buffer_unref(&hw_device_ctx);
4183
4184     /* finished ! */
4185     ret = 0;
4186
4187  fail:
4188 #if HAVE_PTHREADS
4189     free_input_threads();
4190 #endif
4191
4192     if (output_streams) {
4193         for (i = 0; i < nb_output_streams; i++) {
4194             ost = output_streams[i];
4195             if (ost) {
4196                 if (ost->logfile) {
4197                     if (fclose(ost->logfile))
4198                         av_log(NULL, AV_LOG_ERROR,
4199                                "Error closing logfile, loss of information possible: %s\n",
4200                                av_err2str(AVERROR(errno)));
4201                     ost->logfile = NULL;
4202                 }
4203                 av_freep(&ost->forced_kf_pts);
4204                 av_freep(&ost->apad);
4205                 av_freep(&ost->disposition);
4206                 av_dict_free(&ost->encoder_opts);
4207                 av_dict_free(&ost->sws_dict);
4208                 av_dict_free(&ost->swr_opts);
4209                 av_dict_free(&ost->resample_opts);
4210             }
4211         }
4212     }
4213     return ret;
4214 }
4215
4216
4217 static int64_t getutime(void)
4218 {
4219 #if HAVE_GETRUSAGE
4220     struct rusage rusage;
4221
4222     getrusage(RUSAGE_SELF, &rusage);
4223     return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
4224 #elif HAVE_GETPROCESSTIMES
4225     HANDLE proc;
4226     FILETIME c, e, k, u;
4227     proc = GetCurrentProcess();
4228     GetProcessTimes(proc, &c, &e, &k, &u);
4229     return ((int64_t) u.dwHighDateTime << 32 | u.dwLowDateTime) / 10;
4230 #else
4231     return av_gettime_relative();
4232 #endif
4233 }
4234
4235 static int64_t getmaxrss(void)
4236 {
4237 #if HAVE_GETRUSAGE && HAVE_STRUCT_RUSAGE_RU_MAXRSS
4238     struct rusage rusage;
4239     getrusage(RUSAGE_SELF, &rusage);
4240     return (int64_t)rusage.ru_maxrss * 1024;
4241 #elif HAVE_GETPROCESSMEMORYINFO
4242     HANDLE proc;
4243     PROCESS_MEMORY_COUNTERS memcounters;
4244     proc = GetCurrentProcess();
4245     memcounters.cb = sizeof(memcounters);
4246     GetProcessMemoryInfo(proc, &memcounters, sizeof(memcounters));
4247     return memcounters.PeakPagefileUsage;
4248 #else
4249     return 0;
4250 #endif
4251 }
4252
4253 static void log_callback_null(void *ptr, int level, const char *fmt, va_list vl)
4254 {
4255 }
4256
4257 int main(int argc, char **argv)
4258 {
4259     int ret;
4260     int64_t ti;
4261
4262     init_dynload();
4263
4264     register_exit(ffmpeg_cleanup);
4265
4266     setvbuf(stderr,NULL,_IONBF,0); /* win32 runtime needs this */
4267
4268     av_log_set_flags(AV_LOG_SKIP_REPEATED);
4269     parse_loglevel(argc, argv, options);
4270
4271     if(argc>1 && !strcmp(argv[1], "-d")){
4272         run_as_daemon=1;
4273         av_log_set_callback(log_callback_null);
4274         argc--;
4275         argv++;
4276     }
4277
4278     avcodec_register_all();
4279 #if CONFIG_AVDEVICE
4280     avdevice_register_all();
4281 #endif
4282     avfilter_register_all();
4283     av_register_all();
4284     avformat_network_init();
4285
4286     show_banner(argc, argv, options);
4287
4288     term_init();
4289
4290     /* parse options and open all input/output files */
4291     ret = ffmpeg_parse_options(argc, argv);
4292     if (ret < 0)
4293         exit_program(1);
4294
4295     if (nb_output_files <= 0 && nb_input_files == 0) {
4296         show_usage();
4297         av_log(NULL, AV_LOG_WARNING, "Use -h to get full help or, even better, run 'man %s'\n", program_name);
4298         exit_program(1);
4299     }
4300
4301     /* file converter / grab */
4302     if (nb_output_files <= 0) {
4303         av_log(NULL, AV_LOG_FATAL, "At least one output file must be specified\n");
4304         exit_program(1);
4305     }
4306
4307 //     if (nb_input_files == 0) {
4308 //         av_log(NULL, AV_LOG_FATAL, "At least one input file must be specified\n");
4309 //         exit_program(1);
4310 //     }
4311
4312     current_time = ti = getutime();
4313     if (transcode() < 0)
4314         exit_program(1);
4315     ti = getutime() - ti;
4316     if (do_benchmark) {
4317         av_log(NULL, AV_LOG_INFO, "bench: utime=%0.3fs\n", ti / 1000000.0);
4318     }
4319     av_log(NULL, AV_LOG_DEBUG, "%"PRIu64" frames successfully decoded, %"PRIu64" decoding errors\n",
4320            decode_error_stat[0], decode_error_stat[1]);
4321     if ((decode_error_stat[0] + decode_error_stat[1]) * max_error_rate < decode_error_stat[1])
4322         exit_program(69);
4323
4324     exit_program(received_nb_signals ? 255 : main_return_code);
4325     return main_return_code;
4326 }