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