]> git.sesse.net Git - ffmpeg/blob - cmdutils.c
doc: Consistently use 'frame rate' everywhere
[ffmpeg] / cmdutils.c
1 /*
2  * Various utilities for command line tools
3  * Copyright (c) 2000-2003 Fabrice Bellard
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include <string.h>
23 #include <stdlib.h>
24 #include <errno.h>
25 #include <math.h>
26
27 /* Include only the enabled headers since some compilers (namely, Sun
28    Studio) will not omit unused inline functions and create undefined
29    references to libraries that are not being built. */
30
31 #include "config.h"
32 #include "compat/va_copy.h"
33 #include "libavformat/avformat.h"
34 #include "libavfilter/avfilter.h"
35 #include "libavdevice/avdevice.h"
36 #include "libavresample/avresample.h"
37 #include "libswscale/swscale.h"
38 #include "libswresample/swresample.h"
39 #include "libpostproc/postprocess.h"
40 #include "libavutil/avassert.h"
41 #include "libavutil/avstring.h"
42 #include "libavutil/bprint.h"
43 #include "libavutil/mathematics.h"
44 #include "libavutil/imgutils.h"
45 #include "libavutil/parseutils.h"
46 #include "libavutil/pixdesc.h"
47 #include "libavutil/eval.h"
48 #include "libavutil/dict.h"
49 #include "libavutil/opt.h"
50 #include "cmdutils.h"
51 #include "version.h"
52 #if CONFIG_NETWORK
53 #include "libavformat/network.h"
54 #endif
55 #if HAVE_SYS_RESOURCE_H
56 #include <sys/time.h>
57 #include <sys/resource.h>
58 #endif
59
60 static int init_report(const char *env);
61
62 struct SwsContext *sws_opts;
63 AVDictionary *swr_opts;
64 AVDictionary *format_opts, *codec_opts, *resample_opts;
65
66 const int this_year = 2013;
67
68 static FILE *report_file;
69
70 void init_opts(void)
71 {
72
73     if(CONFIG_SWSCALE)
74         sws_opts = sws_getContext(16, 16, 0, 16, 16, 0, SWS_BICUBIC,
75                               NULL, NULL, NULL);
76 }
77
78 void uninit_opts(void)
79 {
80 #if CONFIG_SWSCALE
81     sws_freeContext(sws_opts);
82     sws_opts = NULL;
83 #endif
84
85     av_dict_free(&swr_opts);
86     av_dict_free(&format_opts);
87     av_dict_free(&codec_opts);
88     av_dict_free(&resample_opts);
89 }
90
91 void log_callback_help(void *ptr, int level, const char *fmt, va_list vl)
92 {
93     vfprintf(stdout, fmt, vl);
94 }
95
96 static void log_callback_report(void *ptr, int level, const char *fmt, va_list vl)
97 {
98     va_list vl2;
99     char line[1024];
100     static int print_prefix = 1;
101
102     va_copy(vl2, vl);
103     av_log_default_callback(ptr, level, fmt, vl);
104     av_log_format_line(ptr, level, fmt, vl2, line, sizeof(line), &print_prefix);
105     va_end(vl2);
106     fputs(line, report_file);
107     fflush(report_file);
108 }
109
110 double parse_number_or_die(const char *context, const char *numstr, int type,
111                            double min, double max)
112 {
113     char *tail;
114     const char *error;
115     double d = av_strtod(numstr, &tail);
116     if (*tail)
117         error = "Expected number for %s but found: %s\n";
118     else if (d < min || d > max)
119         error = "The value for %s was %s which is not within %f - %f\n";
120     else if (type == OPT_INT64 && (int64_t)d != d)
121         error = "Expected int64 for %s but found %s\n";
122     else if (type == OPT_INT && (int)d != d)
123         error = "Expected int for %s but found %s\n";
124     else
125         return d;
126     av_log(NULL, AV_LOG_FATAL, error, context, numstr, min, max);
127     exit(1);
128     return 0;
129 }
130
131 int64_t parse_time_or_die(const char *context, const char *timestr,
132                           int is_duration)
133 {
134     int64_t us;
135     if (av_parse_time(&us, timestr, is_duration) < 0) {
136         av_log(NULL, AV_LOG_FATAL, "Invalid %s specification for %s: %s\n",
137                is_duration ? "duration" : "date", context, timestr);
138         exit(1);
139     }
140     return us;
141 }
142
143 void show_help_options(const OptionDef *options, const char *msg, int req_flags,
144                        int rej_flags, int alt_flags)
145 {
146     const OptionDef *po;
147     int first;
148
149     first = 1;
150     for (po = options; po->name != NULL; po++) {
151         char buf[64];
152
153         if (((po->flags & req_flags) != req_flags) ||
154             (alt_flags && !(po->flags & alt_flags)) ||
155             (po->flags & rej_flags))
156             continue;
157
158         if (first) {
159             printf("%s\n", msg);
160             first = 0;
161         }
162         av_strlcpy(buf, po->name, sizeof(buf));
163         if (po->argname) {
164             av_strlcat(buf, " ", sizeof(buf));
165             av_strlcat(buf, po->argname, sizeof(buf));
166         }
167         printf("-%-17s  %s\n", buf, po->help);
168     }
169     printf("\n");
170 }
171
172 void show_help_children(const AVClass *class, int flags)
173 {
174     const AVClass *child = NULL;
175     if (class->option) {
176         av_opt_show2(&class, NULL, flags, 0);
177         printf("\n");
178     }
179
180     while (child = av_opt_child_class_next(class, child))
181         show_help_children(child, flags);
182 }
183
184 static const OptionDef *find_option(const OptionDef *po, const char *name)
185 {
186     const char *p = strchr(name, ':');
187     int len = p ? p - name : strlen(name);
188
189     while (po->name != NULL) {
190         if (!strncmp(name, po->name, len) && strlen(po->name) == len)
191             break;
192         po++;
193     }
194     return po;
195 }
196
197 #if HAVE_COMMANDLINETOARGVW
198 #include <windows.h>
199 #include <shellapi.h>
200 /* Will be leaked on exit */
201 static char** win32_argv_utf8 = NULL;
202 static int win32_argc = 0;
203
204 /**
205  * Prepare command line arguments for executable.
206  * For Windows - perform wide-char to UTF-8 conversion.
207  * Input arguments should be main() function arguments.
208  * @param argc_ptr Arguments number (including executable)
209  * @param argv_ptr Arguments list.
210  */
211 static void prepare_app_arguments(int *argc_ptr, char ***argv_ptr)
212 {
213     char *argstr_flat;
214     wchar_t **argv_w;
215     int i, buffsize = 0, offset = 0;
216
217     if (win32_argv_utf8) {
218         *argc_ptr = win32_argc;
219         *argv_ptr = win32_argv_utf8;
220         return;
221     }
222
223     win32_argc = 0;
224     argv_w = CommandLineToArgvW(GetCommandLineW(), &win32_argc);
225     if (win32_argc <= 0 || !argv_w)
226         return;
227
228     /* determine the UTF-8 buffer size (including NULL-termination symbols) */
229     for (i = 0; i < win32_argc; i++)
230         buffsize += WideCharToMultiByte(CP_UTF8, 0, argv_w[i], -1,
231                                         NULL, 0, NULL, NULL);
232
233     win32_argv_utf8 = av_mallocz(sizeof(char *) * (win32_argc + 1) + buffsize);
234     argstr_flat     = (char *)win32_argv_utf8 + sizeof(char *) * (win32_argc + 1);
235     if (win32_argv_utf8 == NULL) {
236         LocalFree(argv_w);
237         return;
238     }
239
240     for (i = 0; i < win32_argc; i++) {
241         win32_argv_utf8[i] = &argstr_flat[offset];
242         offset += WideCharToMultiByte(CP_UTF8, 0, argv_w[i], -1,
243                                       &argstr_flat[offset],
244                                       buffsize - offset, NULL, NULL);
245     }
246     win32_argv_utf8[i] = NULL;
247     LocalFree(argv_w);
248
249     *argc_ptr = win32_argc;
250     *argv_ptr = win32_argv_utf8;
251 }
252 #else
253 static inline void prepare_app_arguments(int *argc_ptr, char ***argv_ptr)
254 {
255     /* nothing to do */
256 }
257 #endif /* HAVE_COMMANDLINETOARGVW */
258
259 static int write_option(void *optctx, const OptionDef *po, const char *opt,
260                         const char *arg)
261 {
262     /* new-style options contain an offset into optctx, old-style address of
263      * a global var*/
264     void *dst = po->flags & (OPT_OFFSET | OPT_SPEC) ?
265                 (uint8_t *)optctx + po->u.off : po->u.dst_ptr;
266     int *dstcount;
267
268     if (po->flags & OPT_SPEC) {
269         SpecifierOpt **so = dst;
270         char *p = strchr(opt, ':');
271
272         dstcount = (int *)(so + 1);
273         *so = grow_array(*so, sizeof(**so), dstcount, *dstcount + 1);
274         (*so)[*dstcount - 1].specifier = av_strdup(p ? p + 1 : "");
275         dst = &(*so)[*dstcount - 1].u;
276     }
277
278     if (po->flags & OPT_STRING) {
279         char *str;
280         str = av_strdup(arg);
281         av_freep(dst);
282         *(char **)dst = str;
283     } else if (po->flags & OPT_BOOL || po->flags & OPT_INT) {
284         *(int *)dst = parse_number_or_die(opt, arg, OPT_INT64, INT_MIN, INT_MAX);
285     } else if (po->flags & OPT_INT64) {
286         *(int64_t *)dst = parse_number_or_die(opt, arg, OPT_INT64, INT64_MIN, INT64_MAX);
287     } else if (po->flags & OPT_TIME) {
288         *(int64_t *)dst = parse_time_or_die(opt, arg, 1);
289     } else if (po->flags & OPT_FLOAT) {
290         *(float *)dst = parse_number_or_die(opt, arg, OPT_FLOAT, -INFINITY, INFINITY);
291     } else if (po->flags & OPT_DOUBLE) {
292         *(double *)dst = parse_number_or_die(opt, arg, OPT_DOUBLE, -INFINITY, INFINITY);
293     } else if (po->u.func_arg) {
294         int ret = po->u.func_arg(optctx, opt, arg);
295         if (ret < 0) {
296             av_log(NULL, AV_LOG_ERROR,
297                    "Failed to set value '%s' for option '%s': %s\n",
298                    arg, opt, av_err2str(ret));
299             return ret;
300         }
301     }
302     if (po->flags & OPT_EXIT)
303         exit(0);
304
305     return 0;
306 }
307
308 int parse_option(void *optctx, const char *opt, const char *arg,
309                  const OptionDef *options)
310 {
311     const OptionDef *po;
312     int ret;
313
314     po = find_option(options, opt);
315     if (!po->name && opt[0] == 'n' && opt[1] == 'o') {
316         /* handle 'no' bool option */
317         po = find_option(options, opt + 2);
318         if ((po->name && (po->flags & OPT_BOOL)))
319             arg = "0";
320     } else if (po->flags & OPT_BOOL)
321         arg = "1";
322
323     if (!po->name)
324         po = find_option(options, "default");
325     if (!po->name) {
326         av_log(NULL, AV_LOG_ERROR, "Unrecognized option '%s'\n", opt);
327         return AVERROR(EINVAL);
328     }
329     if (po->flags & HAS_ARG && !arg) {
330         av_log(NULL, AV_LOG_ERROR, "Missing argument for option '%s'\n", opt);
331         return AVERROR(EINVAL);
332     }
333
334     ret = write_option(optctx, po, opt, arg);
335     if (ret < 0)
336         return ret;
337
338     return !!(po->flags & HAS_ARG);
339 }
340
341 void parse_options(void *optctx, int argc, char **argv, const OptionDef *options,
342                    void (*parse_arg_function)(void *, const char*))
343 {
344     const char *opt;
345     int optindex, handleoptions = 1, ret;
346
347     /* perform system-dependent conversions for arguments list */
348     prepare_app_arguments(&argc, &argv);
349
350     /* parse options */
351     optindex = 1;
352     while (optindex < argc) {
353         opt = argv[optindex++];
354
355         if (handleoptions && opt[0] == '-' && opt[1] != '\0') {
356             if (opt[1] == '-' && opt[2] == '\0') {
357                 handleoptions = 0;
358                 continue;
359             }
360             opt++;
361
362             if ((ret = parse_option(optctx, opt, argv[optindex], options)) < 0)
363                 exit(1);
364             optindex += ret;
365         } else {
366             if (parse_arg_function)
367                 parse_arg_function(optctx, opt);
368         }
369     }
370 }
371
372 int parse_optgroup(void *optctx, OptionGroup *g)
373 {
374     int i, ret;
375
376     av_log(NULL, AV_LOG_DEBUG, "Parsing a group of options: %s %s.\n",
377            g->group_def->name, g->arg);
378
379     for (i = 0; i < g->nb_opts; i++) {
380         Option *o = &g->opts[i];
381
382         if (g->group_def->flags &&
383             !(g->group_def->flags & o->opt->flags)) {
384             av_log(NULL, AV_LOG_ERROR, "Option %s (%s) cannot be applied to "
385                    "%s %s -- you are trying to apply an input option to an "
386                    "output file or vice versa. Move this option before the "
387                    "file it belongs to.\n", o->key, o->opt->help,
388                    g->group_def->name, g->arg);
389             return AVERROR(EINVAL);
390         }
391
392         av_log(NULL, AV_LOG_DEBUG, "Applying option %s (%s) with argument %s.\n",
393                o->key, o->opt->help, o->val);
394
395         ret = write_option(optctx, o->opt, o->key, o->val);
396         if (ret < 0)
397             return ret;
398     }
399
400     av_log(NULL, AV_LOG_DEBUG, "Successfully parsed a group of options.\n");
401
402     return 0;
403 }
404
405 int locate_option(int argc, char **argv, const OptionDef *options,
406                   const char *optname)
407 {
408     const OptionDef *po;
409     int i;
410
411     for (i = 1; i < argc; i++) {
412         const char *cur_opt = argv[i];
413
414         if (*cur_opt++ != '-')
415             continue;
416
417         po = find_option(options, cur_opt);
418         if (!po->name && cur_opt[0] == 'n' && cur_opt[1] == 'o')
419             po = find_option(options, cur_opt + 2);
420
421         if ((!po->name && !strcmp(cur_opt, optname)) ||
422              (po->name && !strcmp(optname, po->name)))
423             return i;
424
425         if (po->flags & HAS_ARG)
426             i++;
427     }
428     return 0;
429 }
430
431 static void dump_argument(const char *a)
432 {
433     const unsigned char *p;
434
435     for (p = a; *p; p++)
436         if (!((*p >= '+' && *p <= ':') || (*p >= '@' && *p <= 'Z') ||
437               *p == '_' || (*p >= 'a' && *p <= 'z')))
438             break;
439     if (!*p) {
440         fputs(a, report_file);
441         return;
442     }
443     fputc('"', report_file);
444     for (p = a; *p; p++) {
445         if (*p == '\\' || *p == '"' || *p == '$' || *p == '`')
446             fprintf(report_file, "\\%c", *p);
447         else if (*p < ' ' || *p > '~')
448             fprintf(report_file, "\\x%02x", *p);
449         else
450             fputc(*p, report_file);
451     }
452     fputc('"', report_file);
453 }
454
455 void parse_loglevel(int argc, char **argv, const OptionDef *options)
456 {
457     int idx = locate_option(argc, argv, options, "loglevel");
458     const char *env;
459     if (!idx)
460         idx = locate_option(argc, argv, options, "v");
461     if (idx && argv[idx + 1])
462         opt_loglevel(NULL, "loglevel", argv[idx + 1]);
463     idx = locate_option(argc, argv, options, "report");
464     if ((env = getenv("FFREPORT")) || idx) {
465         init_report(env);
466         if (report_file) {
467             int i;
468             fprintf(report_file, "Command line:\n");
469             for (i = 0; i < argc; i++) {
470                 dump_argument(argv[i]);
471                 fputc(i < argc - 1 ? ' ' : '\n', report_file);
472             }
473             fflush(report_file);
474         }
475     }
476 }
477
478 #define FLAGS (o->type == AV_OPT_TYPE_FLAGS) ? AV_DICT_APPEND : 0
479 int opt_default(void *optctx, const char *opt, const char *arg)
480 {
481     const AVOption *o;
482     int consumed = 0;
483     char opt_stripped[128];
484     const char *p;
485     const AVClass *cc = avcodec_get_class(), *fc = avformat_get_class();
486 #if CONFIG_AVRESAMPLE
487     const AVClass *rc = avresample_get_class();
488 #endif
489     const AVClass *sc, *swr_class;
490
491     if (!strcmp(opt, "debug") || !strcmp(opt, "fdebug"))
492         av_log_set_level(AV_LOG_DEBUG);
493
494     if (!(p = strchr(opt, ':')))
495         p = opt + strlen(opt);
496     av_strlcpy(opt_stripped, opt, FFMIN(sizeof(opt_stripped), p - opt + 1));
497
498     if ((o = av_opt_find(&cc, opt_stripped, NULL, 0,
499                          AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)) ||
500         ((opt[0] == 'v' || opt[0] == 'a' || opt[0] == 's') &&
501          (o = av_opt_find(&cc, opt + 1, NULL, 0, AV_OPT_SEARCH_FAKE_OBJ)))) {
502         av_dict_set(&codec_opts, opt, arg, FLAGS);
503         consumed = 1;
504     }
505     if ((o = av_opt_find(&fc, opt, NULL, 0,
506                          AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {
507         av_dict_set(&format_opts, opt, arg, FLAGS);
508         if (consumed)
509             av_log(NULL, AV_LOG_VERBOSE, "Routing option %s to both codec and muxer layer\n", opt);
510         consumed = 1;
511     }
512 #if CONFIG_SWSCALE
513     sc = sws_get_class();
514     if (!consumed && av_opt_find(&sc, opt, NULL, 0,
515                          AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)) {
516         // XXX we only support sws_flags, not arbitrary sws options
517         int ret = av_opt_set(sws_opts, opt, arg, 0);
518         if (ret < 0) {
519             av_log(NULL, AV_LOG_ERROR, "Error setting option %s.\n", opt);
520             return ret;
521         }
522         consumed = 1;
523     }
524 #endif
525 #if CONFIG_SWRESAMPLE
526     swr_class = swr_get_class();
527     if (!consumed && (o=av_opt_find(&swr_class, opt, NULL, 0,
528                                     AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {
529         struct SwrContext *swr = swr_alloc();
530         int ret = av_opt_set(swr, opt, arg, 0);
531         swr_free(&swr);
532         if (ret < 0) {
533             av_log(NULL, AV_LOG_ERROR, "Error setting option %s.\n", opt);
534             return ret;
535         }
536         av_dict_set(&swr_opts, opt, arg, FLAGS);
537         consumed = 1;
538     }
539 #endif
540 #if CONFIG_AVRESAMPLE
541     if ((o=av_opt_find(&rc, opt, NULL, 0,
542                        AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {
543         av_dict_set(&resample_opts, opt, arg, FLAGS);
544         consumed = 1;
545     }
546 #endif
547
548     if (consumed)
549         return 0;
550     av_log(NULL, AV_LOG_ERROR, "Could not find option '%s' in any of the FFmpeg subsystems "
551            "(codec, format, scaler, resampler contexts)\n", opt);
552     return AVERROR_OPTION_NOT_FOUND;
553 }
554
555 /*
556  * Check whether given option is a group separator.
557  *
558  * @return index of the group definition that matched or -1 if none
559  */
560 static int match_group_separator(const OptionGroupDef *groups, int nb_groups,
561                                  const char *opt)
562 {
563     int i;
564
565     for (i = 0; i < nb_groups; i++) {
566         const OptionGroupDef *p = &groups[i];
567         if (p->sep && !strcmp(p->sep, opt))
568             return i;
569     }
570
571     return -1;
572 }
573
574 /*
575  * Finish parsing an option group.
576  *
577  * @param group_idx which group definition should this group belong to
578  * @param arg argument of the group delimiting option
579  */
580 static void finish_group(OptionParseContext *octx, int group_idx,
581                          const char *arg)
582 {
583     OptionGroupList *l = &octx->groups[group_idx];
584     OptionGroup *g;
585
586     GROW_ARRAY(l->groups, l->nb_groups);
587     g = &l->groups[l->nb_groups - 1];
588
589     *g             = octx->cur_group;
590     g->arg         = arg;
591     g->group_def   = l->group_def;
592 #if CONFIG_SWSCALE
593     g->sws_opts    = sws_opts;
594 #endif
595     g->swr_opts    = swr_opts;
596     g->codec_opts  = codec_opts;
597     g->format_opts = format_opts;
598     g->resample_opts = resample_opts;
599
600     codec_opts  = NULL;
601     format_opts = NULL;
602     resample_opts = NULL;
603 #if CONFIG_SWSCALE
604     sws_opts    = NULL;
605 #endif
606     swr_opts    = NULL;
607     init_opts();
608
609     memset(&octx->cur_group, 0, sizeof(octx->cur_group));
610 }
611
612 /*
613  * Add an option instance to currently parsed group.
614  */
615 static void add_opt(OptionParseContext *octx, const OptionDef *opt,
616                     const char *key, const char *val)
617 {
618     int global = !(opt->flags & (OPT_PERFILE | OPT_SPEC | OPT_OFFSET));
619     OptionGroup *g = global ? &octx->global_opts : &octx->cur_group;
620
621     GROW_ARRAY(g->opts, g->nb_opts);
622     g->opts[g->nb_opts - 1].opt = opt;
623     g->opts[g->nb_opts - 1].key = key;
624     g->opts[g->nb_opts - 1].val = val;
625 }
626
627 static void init_parse_context(OptionParseContext *octx,
628                                const OptionGroupDef *groups, int nb_groups)
629 {
630     static const OptionGroupDef global_group = { "global" };
631     int i;
632
633     memset(octx, 0, sizeof(*octx));
634
635     octx->nb_groups = nb_groups;
636     octx->groups    = av_mallocz(sizeof(*octx->groups) * octx->nb_groups);
637     if (!octx->groups)
638         exit(1);
639
640     for (i = 0; i < octx->nb_groups; i++)
641         octx->groups[i].group_def = &groups[i];
642
643     octx->global_opts.group_def = &global_group;
644     octx->global_opts.arg       = "";
645
646     init_opts();
647 }
648
649 void uninit_parse_context(OptionParseContext *octx)
650 {
651     int i, j;
652
653     for (i = 0; i < octx->nb_groups; i++) {
654         OptionGroupList *l = &octx->groups[i];
655
656         for (j = 0; j < l->nb_groups; j++) {
657             av_freep(&l->groups[j].opts);
658             av_dict_free(&l->groups[j].codec_opts);
659             av_dict_free(&l->groups[j].format_opts);
660             av_dict_free(&l->groups[j].resample_opts);
661 #if CONFIG_SWSCALE
662             sws_freeContext(l->groups[j].sws_opts);
663 #endif
664             av_dict_free(&l->groups[j].swr_opts);
665         }
666         av_freep(&l->groups);
667     }
668     av_freep(&octx->groups);
669
670     av_freep(&octx->cur_group.opts);
671     av_freep(&octx->global_opts.opts);
672
673     uninit_opts();
674 }
675
676 int split_commandline(OptionParseContext *octx, int argc, char *argv[],
677                       const OptionDef *options,
678                       const OptionGroupDef *groups, int nb_groups)
679 {
680     int optindex = 1;
681     int dashdash = -2;
682
683     /* perform system-dependent conversions for arguments list */
684     prepare_app_arguments(&argc, &argv);
685
686     init_parse_context(octx, groups, nb_groups);
687     av_log(NULL, AV_LOG_DEBUG, "Splitting the commandline.\n");
688
689     while (optindex < argc) {
690         const char *opt = argv[optindex++], *arg;
691         const OptionDef *po;
692         int ret;
693
694         av_log(NULL, AV_LOG_DEBUG, "Reading option '%s' ...", opt);
695
696         if (opt[0] == '-' && opt[1] == '-' && !opt[2]) {
697             dashdash = optindex;
698             continue;
699         }
700         /* unnamed group separators, e.g. output filename */
701         if (opt[0] != '-' || !opt[1] || dashdash+1 == optindex) {
702             finish_group(octx, 0, opt);
703             av_log(NULL, AV_LOG_DEBUG, " matched as %s.\n", groups[0].name);
704             continue;
705         }
706         opt++;
707
708 #define GET_ARG(arg)                                                           \
709 do {                                                                           \
710     arg = argv[optindex++];                                                    \
711     if (!arg) {                                                                \
712         av_log(NULL, AV_LOG_ERROR, "Missing argument for option '%s'.\n", opt);\
713         return AVERROR(EINVAL);                                                \
714     }                                                                          \
715 } while (0)
716
717         /* named group separators, e.g. -i */
718         if ((ret = match_group_separator(groups, nb_groups, opt)) >= 0) {
719             GET_ARG(arg);
720             finish_group(octx, ret, arg);
721             av_log(NULL, AV_LOG_DEBUG, " matched as %s with argument '%s'.\n",
722                    groups[ret].name, arg);
723             continue;
724         }
725
726         /* normal options */
727         po = find_option(options, opt);
728         if (po->name) {
729             if (po->flags & OPT_EXIT) {
730                 /* optional argument, e.g. -h */
731                 arg = argv[optindex++];
732             } else if (po->flags & HAS_ARG) {
733                 GET_ARG(arg);
734             } else {
735                 arg = "1";
736             }
737
738             add_opt(octx, po, opt, arg);
739             av_log(NULL, AV_LOG_DEBUG, " matched as option '%s' (%s) with "
740                    "argument '%s'.\n", po->name, po->help, arg);
741             continue;
742         }
743
744         /* AVOptions */
745         if (argv[optindex]) {
746             ret = opt_default(NULL, opt, argv[optindex]);
747             if (ret >= 0) {
748                 av_log(NULL, AV_LOG_DEBUG, " matched as AVOption '%s' with "
749                        "argument '%s'.\n", opt, argv[optindex]);
750                 optindex++;
751                 continue;
752             } else if (ret != AVERROR_OPTION_NOT_FOUND) {
753                 av_log(NULL, AV_LOG_ERROR, "Error parsing option '%s' "
754                        "with argument '%s'.\n", opt, argv[optindex]);
755                 return ret;
756             }
757         }
758
759         /* boolean -nofoo options */
760         if (opt[0] == 'n' && opt[1] == 'o' &&
761             (po = find_option(options, opt + 2)) &&
762             po->name && po->flags & OPT_BOOL) {
763             add_opt(octx, po, opt, "0");
764             av_log(NULL, AV_LOG_DEBUG, " matched as option '%s' (%s) with "
765                    "argument 0.\n", po->name, po->help);
766             continue;
767         }
768
769         av_log(NULL, AV_LOG_ERROR, "Unrecognized option '%s'.\n", opt);
770         return AVERROR_OPTION_NOT_FOUND;
771     }
772
773     if (octx->cur_group.nb_opts || codec_opts || format_opts || resample_opts)
774         av_log(NULL, AV_LOG_WARNING, "Trailing options were found on the "
775                "commandline.\n");
776
777     av_log(NULL, AV_LOG_DEBUG, "Finished splitting the commandline.\n");
778
779     return 0;
780 }
781
782 int opt_loglevel(void *optctx, const char *opt, const char *arg)
783 {
784     const struct { const char *name; int level; } log_levels[] = {
785         { "quiet"  , AV_LOG_QUIET   },
786         { "panic"  , AV_LOG_PANIC   },
787         { "fatal"  , AV_LOG_FATAL   },
788         { "error"  , AV_LOG_ERROR   },
789         { "warning", AV_LOG_WARNING },
790         { "info"   , AV_LOG_INFO    },
791         { "verbose", AV_LOG_VERBOSE },
792         { "debug"  , AV_LOG_DEBUG   },
793     };
794     char *tail;
795     int level;
796     int i;
797
798     tail = strstr(arg, "repeat");
799     av_log_set_flags(tail ? 0 : AV_LOG_SKIP_REPEATED);
800     if (tail == arg)
801         arg += 6 + (arg[6]=='+');
802     if(tail && !*arg)
803         return 0;
804
805     for (i = 0; i < FF_ARRAY_ELEMS(log_levels); i++) {
806         if (!strcmp(log_levels[i].name, arg)) {
807             av_log_set_level(log_levels[i].level);
808             return 0;
809         }
810     }
811
812     level = strtol(arg, &tail, 10);
813     if (*tail) {
814         av_log(NULL, AV_LOG_FATAL, "Invalid loglevel \"%s\". "
815                "Possible levels are numbers or:\n", arg);
816         for (i = 0; i < FF_ARRAY_ELEMS(log_levels); i++)
817             av_log(NULL, AV_LOG_FATAL, "\"%s\"\n", log_levels[i].name);
818         exit(1);
819     }
820     av_log_set_level(level);
821     return 0;
822 }
823
824 static void expand_filename_template(AVBPrint *bp, const char *template,
825                                      struct tm *tm)
826 {
827     int c;
828
829     while ((c = *(template++))) {
830         if (c == '%') {
831             if (!(c = *(template++)))
832                 break;
833             switch (c) {
834             case 'p':
835                 av_bprintf(bp, "%s", program_name);
836                 break;
837             case 't':
838                 av_bprintf(bp, "%04d%02d%02d-%02d%02d%02d",
839                            tm->tm_year + 1900, tm->tm_mon + 1, tm->tm_mday,
840                            tm->tm_hour, tm->tm_min, tm->tm_sec);
841                 break;
842             case '%':
843                 av_bprint_chars(bp, c, 1);
844                 break;
845             }
846         } else {
847             av_bprint_chars(bp, c, 1);
848         }
849     }
850 }
851
852 static int init_report(const char *env)
853 {
854     char *filename_template = NULL;
855     char *key, *val;
856     int ret, count = 0;
857     time_t now;
858     struct tm *tm;
859     AVBPrint filename;
860
861     if (report_file) /* already opened */
862         return 0;
863     time(&now);
864     tm = localtime(&now);
865
866     while (env && *env) {
867         if ((ret = av_opt_get_key_value(&env, "=", ":", 0, &key, &val)) < 0) {
868             if (count)
869                 av_log(NULL, AV_LOG_ERROR,
870                        "Failed to parse FFREPORT environment variable: %s\n",
871                        av_err2str(ret));
872             break;
873         }
874         if (*env)
875             env++;
876         count++;
877         if (!strcmp(key, "file")) {
878             av_free(filename_template);
879             filename_template = val;
880             val = NULL;
881         } else {
882             av_log(NULL, AV_LOG_ERROR, "Unknown key '%s' in FFREPORT\n", key);
883         }
884         av_free(val);
885         av_free(key);
886     }
887
888     av_bprint_init(&filename, 0, 1);
889     expand_filename_template(&filename,
890                              av_x_if_null(filename_template, "%p-%t.log"), tm);
891     av_free(filename_template);
892     if (!av_bprint_is_complete(&filename)) {
893         av_log(NULL, AV_LOG_ERROR, "Out of memory building report file name\n");
894         return AVERROR(ENOMEM);
895     }
896
897     report_file = fopen(filename.str, "w");
898     if (!report_file) {
899         av_log(NULL, AV_LOG_ERROR, "Failed to open report \"%s\": %s\n",
900                filename.str, strerror(errno));
901         return AVERROR(errno);
902     }
903     av_log_set_callback(log_callback_report);
904     av_log(NULL, AV_LOG_INFO,
905            "%s started on %04d-%02d-%02d at %02d:%02d:%02d\n"
906            "Report written to \"%s\"\n",
907            program_name,
908            tm->tm_year + 1900, tm->tm_mon + 1, tm->tm_mday,
909            tm->tm_hour, tm->tm_min, tm->tm_sec,
910            filename.str);
911     av_log_set_level(FFMAX(av_log_get_level(), AV_LOG_VERBOSE));
912     av_bprint_finalize(&filename, NULL);
913     return 0;
914 }
915
916 int opt_report(const char *opt)
917 {
918     return init_report(NULL);
919 }
920
921 int opt_max_alloc(void *optctx, const char *opt, const char *arg)
922 {
923     char *tail;
924     size_t max;
925
926     max = strtol(arg, &tail, 10);
927     if (*tail) {
928         av_log(NULL, AV_LOG_FATAL, "Invalid max_alloc \"%s\".\n", arg);
929         exit(1);
930     }
931     av_max_alloc(max);
932     return 0;
933 }
934
935 int opt_cpuflags(void *optctx, const char *opt, const char *arg)
936 {
937     int ret;
938     unsigned flags = av_get_cpu_flags();
939
940     if ((ret = av_parse_cpu_caps(&flags, arg)) < 0)
941         return ret;
942
943     av_force_cpu_flags(flags);
944     return 0;
945 }
946
947 int opt_timelimit(void *optctx, const char *opt, const char *arg)
948 {
949 #if HAVE_SETRLIMIT
950     int lim = parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX);
951     struct rlimit rl = { lim, lim + 1 };
952     if (setrlimit(RLIMIT_CPU, &rl))
953         perror("setrlimit");
954 #else
955     av_log(NULL, AV_LOG_WARNING, "-%s not implemented on this OS\n", opt);
956 #endif
957     return 0;
958 }
959
960 void print_error(const char *filename, int err)
961 {
962     char errbuf[128];
963     const char *errbuf_ptr = errbuf;
964
965     if (av_strerror(err, errbuf, sizeof(errbuf)) < 0)
966         errbuf_ptr = strerror(AVUNERROR(err));
967     av_log(NULL, AV_LOG_ERROR, "%s: %s\n", filename, errbuf_ptr);
968 }
969
970 static int warned_cfg = 0;
971
972 #define INDENT        1
973 #define SHOW_VERSION  2
974 #define SHOW_CONFIG   4
975 #define SHOW_COPYRIGHT 8
976
977 #define PRINT_LIB_INFO(libname, LIBNAME, flags, level)                  \
978     if (CONFIG_##LIBNAME) {                                             \
979         const char *indent = flags & INDENT? "  " : "";                 \
980         if (flags & SHOW_VERSION) {                                     \
981             unsigned int version = libname##_version();                 \
982             av_log(NULL, level,                                         \
983                    "%slib%-11s %2d.%3d.%3d / %2d.%3d.%3d\n",            \
984                    indent, #libname,                                    \
985                    LIB##LIBNAME##_VERSION_MAJOR,                        \
986                    LIB##LIBNAME##_VERSION_MINOR,                        \
987                    LIB##LIBNAME##_VERSION_MICRO,                        \
988                    version >> 16, version >> 8 & 0xff, version & 0xff); \
989         }                                                               \
990         if (flags & SHOW_CONFIG) {                                      \
991             const char *cfg = libname##_configuration();                \
992             if (strcmp(FFMPEG_CONFIGURATION, cfg)) {                    \
993                 if (!warned_cfg) {                                      \
994                     av_log(NULL, level,                                 \
995                             "%sWARNING: library configuration mismatch\n", \
996                             indent);                                    \
997                     warned_cfg = 1;                                     \
998                 }                                                       \
999                 av_log(NULL, level, "%s%-11s configuration: %s\n",      \
1000                         indent, #libname, cfg);                         \
1001             }                                                           \
1002         }                                                               \
1003     }                                                                   \
1004
1005 static void print_all_libs_info(int flags, int level)
1006 {
1007     PRINT_LIB_INFO(avutil,   AVUTIL,   flags, level);
1008     PRINT_LIB_INFO(avcodec,  AVCODEC,  flags, level);
1009     PRINT_LIB_INFO(avformat, AVFORMAT, flags, level);
1010     PRINT_LIB_INFO(avdevice, AVDEVICE, flags, level);
1011     PRINT_LIB_INFO(avfilter, AVFILTER, flags, level);
1012     PRINT_LIB_INFO(avresample, AVRESAMPLE, flags, level);
1013     PRINT_LIB_INFO(swscale,  SWSCALE,  flags, level);
1014     PRINT_LIB_INFO(swresample,SWRESAMPLE,  flags, level);
1015     PRINT_LIB_INFO(postproc, POSTPROC, flags, level);
1016 }
1017
1018 static void print_program_info(int flags, int level)
1019 {
1020     const char *indent = flags & INDENT? "  " : "";
1021
1022     av_log(NULL, level, "%s version " FFMPEG_VERSION, program_name);
1023     if (flags & SHOW_COPYRIGHT)
1024         av_log(NULL, level, " Copyright (c) %d-%d the FFmpeg developers",
1025                program_birth_year, this_year);
1026     av_log(NULL, level, "\n");
1027     av_log(NULL, level, "%sbuilt on %s %s with %s\n",
1028            indent, __DATE__, __TIME__, CC_IDENT);
1029
1030     av_log(NULL, level, "%sconfiguration: " FFMPEG_CONFIGURATION "\n", indent);
1031 }
1032
1033 void show_banner(int argc, char **argv, const OptionDef *options)
1034 {
1035     int idx = locate_option(argc, argv, options, "version");
1036     if (idx)
1037         return;
1038
1039     print_program_info (INDENT|SHOW_COPYRIGHT, AV_LOG_INFO);
1040     print_all_libs_info(INDENT|SHOW_CONFIG,  AV_LOG_INFO);
1041     print_all_libs_info(INDENT|SHOW_VERSION, AV_LOG_INFO);
1042 }
1043
1044 int show_version(void *optctx, const char *opt, const char *arg)
1045 {
1046     av_log_set_callback(log_callback_help);
1047     print_program_info (0           , AV_LOG_INFO);
1048     print_all_libs_info(SHOW_VERSION, AV_LOG_INFO);
1049
1050     return 0;
1051 }
1052
1053 int show_license(void *optctx, const char *opt, const char *arg)
1054 {
1055 #if CONFIG_NONFREE
1056     printf(
1057     "This version of %s has nonfree parts compiled in.\n"
1058     "Therefore it is not legally redistributable.\n",
1059     program_name );
1060 #elif CONFIG_GPLV3
1061     printf(
1062     "%s is free software; you can redistribute it and/or modify\n"
1063     "it under the terms of the GNU General Public License as published by\n"
1064     "the Free Software Foundation; either version 3 of the License, or\n"
1065     "(at your option) any later version.\n"
1066     "\n"
1067     "%s is distributed in the hope that it will be useful,\n"
1068     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1069     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
1070     "GNU General Public License for more details.\n"
1071     "\n"
1072     "You should have received a copy of the GNU General Public License\n"
1073     "along with %s.  If not, see <http://www.gnu.org/licenses/>.\n",
1074     program_name, program_name, program_name );
1075 #elif CONFIG_GPL
1076     printf(
1077     "%s is free software; you can redistribute it and/or modify\n"
1078     "it under the terms of the GNU General Public License as published by\n"
1079     "the Free Software Foundation; either version 2 of the License, or\n"
1080     "(at your option) any later version.\n"
1081     "\n"
1082     "%s is distributed in the hope that it will be useful,\n"
1083     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1084     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
1085     "GNU General Public License for more details.\n"
1086     "\n"
1087     "You should have received a copy of the GNU General Public License\n"
1088     "along with %s; if not, write to the Free Software\n"
1089     "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n",
1090     program_name, program_name, program_name );
1091 #elif CONFIG_LGPLV3
1092     printf(
1093     "%s is free software; you can redistribute it and/or modify\n"
1094     "it under the terms of the GNU Lesser General Public License as published by\n"
1095     "the Free Software Foundation; either version 3 of the License, or\n"
1096     "(at your option) any later version.\n"
1097     "\n"
1098     "%s is distributed in the hope that it will be useful,\n"
1099     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1100     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
1101     "GNU Lesser General Public License for more details.\n"
1102     "\n"
1103     "You should have received a copy of the GNU Lesser General Public License\n"
1104     "along with %s.  If not, see <http://www.gnu.org/licenses/>.\n",
1105     program_name, program_name, program_name );
1106 #else
1107     printf(
1108     "%s is free software; you can redistribute it and/or\n"
1109     "modify it under the terms of the GNU Lesser General Public\n"
1110     "License as published by the Free Software Foundation; either\n"
1111     "version 2.1 of the License, or (at your option) any later version.\n"
1112     "\n"
1113     "%s is distributed in the hope that it will be useful,\n"
1114     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1115     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n"
1116     "Lesser General Public License for more details.\n"
1117     "\n"
1118     "You should have received a copy of the GNU Lesser General Public\n"
1119     "License along with %s; if not, write to the Free Software\n"
1120     "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n",
1121     program_name, program_name, program_name );
1122 #endif
1123
1124     return 0;
1125 }
1126
1127 int show_formats(void *optctx, const char *opt, const char *arg)
1128 {
1129     AVInputFormat *ifmt  = NULL;
1130     AVOutputFormat *ofmt = NULL;
1131     const char *last_name;
1132
1133     printf("File formats:\n"
1134            " D. = Demuxing supported\n"
1135            " .E = Muxing supported\n"
1136            " --\n");
1137     last_name = "000";
1138     for (;;) {
1139         int decode = 0;
1140         int encode = 0;
1141         const char *name      = NULL;
1142         const char *long_name = NULL;
1143
1144         while ((ofmt = av_oformat_next(ofmt))) {
1145             if ((name == NULL || strcmp(ofmt->name, name) < 0) &&
1146                 strcmp(ofmt->name, last_name) > 0) {
1147                 name      = ofmt->name;
1148                 long_name = ofmt->long_name;
1149                 encode    = 1;
1150             }
1151         }
1152         while ((ifmt = av_iformat_next(ifmt))) {
1153             if ((name == NULL || strcmp(ifmt->name, name) < 0) &&
1154                 strcmp(ifmt->name, last_name) > 0) {
1155                 name      = ifmt->name;
1156                 long_name = ifmt->long_name;
1157                 encode    = 0;
1158             }
1159             if (name && strcmp(ifmt->name, name) == 0)
1160                 decode = 1;
1161         }
1162         if (name == NULL)
1163             break;
1164         last_name = name;
1165
1166         printf(" %s%s %-15s %s\n",
1167                decode ? "D" : " ",
1168                encode ? "E" : " ",
1169                name,
1170             long_name ? long_name:" ");
1171     }
1172     return 0;
1173 }
1174
1175 #define PRINT_CODEC_SUPPORTED(codec, field, type, list_name, term, get_name) \
1176     if (codec->field) {                                                      \
1177         const type *p = codec->field;                                        \
1178                                                                              \
1179         printf("    Supported " list_name ":");                              \
1180         while (*p != term) {                                                 \
1181             get_name(*p);                                                    \
1182             printf(" %s", name);                                             \
1183             p++;                                                             \
1184         }                                                                    \
1185         printf("\n");                                                        \
1186     }                                                                        \
1187
1188 static void print_codec(const AVCodec *c)
1189 {
1190     int encoder = av_codec_is_encoder(c);
1191
1192     printf("%s %s [%s]:\n", encoder ? "Encoder" : "Decoder", c->name,
1193            c->long_name ? c->long_name : "");
1194
1195     if (c->type == AVMEDIA_TYPE_VIDEO) {
1196         printf("    Threading capabilities: ");
1197         switch (c->capabilities & (CODEC_CAP_FRAME_THREADS |
1198                                    CODEC_CAP_SLICE_THREADS)) {
1199         case CODEC_CAP_FRAME_THREADS |
1200              CODEC_CAP_SLICE_THREADS: printf("frame and slice"); break;
1201         case CODEC_CAP_FRAME_THREADS: printf("frame");           break;
1202         case CODEC_CAP_SLICE_THREADS: printf("slice");           break;
1203         default:                      printf("no");              break;
1204         }
1205         printf("\n");
1206     }
1207
1208     if (c->supported_framerates) {
1209         const AVRational *fps = c->supported_framerates;
1210
1211         printf("    Supported framerates:");
1212         while (fps->num) {
1213             printf(" %d/%d", fps->num, fps->den);
1214             fps++;
1215         }
1216         printf("\n");
1217     }
1218     PRINT_CODEC_SUPPORTED(c, pix_fmts, enum AVPixelFormat, "pixel formats",
1219                           AV_PIX_FMT_NONE, GET_PIX_FMT_NAME);
1220     PRINT_CODEC_SUPPORTED(c, supported_samplerates, int, "sample rates", 0,
1221                           GET_SAMPLE_RATE_NAME);
1222     PRINT_CODEC_SUPPORTED(c, sample_fmts, enum AVSampleFormat, "sample formats",
1223                           AV_SAMPLE_FMT_NONE, GET_SAMPLE_FMT_NAME);
1224     PRINT_CODEC_SUPPORTED(c, channel_layouts, uint64_t, "channel layouts",
1225                           0, GET_CH_LAYOUT_DESC);
1226
1227     if (c->priv_class) {
1228         show_help_children(c->priv_class,
1229                            AV_OPT_FLAG_ENCODING_PARAM |
1230                            AV_OPT_FLAG_DECODING_PARAM);
1231     }
1232 }
1233
1234 static char get_media_type_char(enum AVMediaType type)
1235 {
1236     switch (type) {
1237         case AVMEDIA_TYPE_VIDEO:    return 'V';
1238         case AVMEDIA_TYPE_AUDIO:    return 'A';
1239         case AVMEDIA_TYPE_DATA:     return 'D';
1240         case AVMEDIA_TYPE_SUBTITLE: return 'S';
1241         case AVMEDIA_TYPE_ATTACHMENT:return 'T';
1242         default:                    return '?';
1243     }
1244 }
1245
1246 static const AVCodec *next_codec_for_id(enum AVCodecID id, const AVCodec *prev,
1247                                         int encoder)
1248 {
1249     while ((prev = av_codec_next(prev))) {
1250         if (prev->id == id &&
1251             (encoder ? av_codec_is_encoder(prev) : av_codec_is_decoder(prev)))
1252             return prev;
1253     }
1254     return NULL;
1255 }
1256
1257 static int compare_codec_desc(const void *a, const void *b)
1258 {
1259     const AVCodecDescriptor * const *da = a;
1260     const AVCodecDescriptor * const *db = b;
1261
1262     return (*da)->type != (*db)->type ? (*da)->type - (*db)->type :
1263            strcmp((*da)->name, (*db)->name);
1264 }
1265
1266 static unsigned get_codecs_sorted(const AVCodecDescriptor ***rcodecs)
1267 {
1268     const AVCodecDescriptor *desc = NULL;
1269     const AVCodecDescriptor **codecs;
1270     unsigned nb_codecs = 0, i = 0;
1271
1272     while ((desc = avcodec_descriptor_next(desc)))
1273         nb_codecs++;
1274     if (!(codecs = av_calloc(nb_codecs, sizeof(*codecs)))) {
1275         av_log(NULL, AV_LOG_ERROR, "Out of memory\n");
1276         exit(1);
1277     }
1278     desc = NULL;
1279     while ((desc = avcodec_descriptor_next(desc)))
1280         codecs[i++] = desc;
1281     av_assert0(i == nb_codecs);
1282     qsort(codecs, nb_codecs, sizeof(*codecs), compare_codec_desc);
1283     *rcodecs = codecs;
1284     return nb_codecs;
1285 }
1286
1287 static void print_codecs_for_id(enum AVCodecID id, int encoder)
1288 {
1289     const AVCodec *codec = NULL;
1290
1291     printf(" (%s: ", encoder ? "encoders" : "decoders");
1292
1293     while ((codec = next_codec_for_id(id, codec, encoder)))
1294         printf("%s ", codec->name);
1295
1296     printf(")");
1297 }
1298
1299 int show_codecs(void *optctx, const char *opt, const char *arg)
1300 {
1301     const AVCodecDescriptor **codecs;
1302     unsigned i, nb_codecs = get_codecs_sorted(&codecs);
1303
1304     printf("Codecs:\n"
1305            " D..... = Decoding supported\n"
1306            " .E.... = Encoding supported\n"
1307            " ..V... = Video codec\n"
1308            " ..A... = Audio codec\n"
1309            " ..S... = Subtitle codec\n"
1310            " ...I.. = Intra frame-only codec\n"
1311            " ....L. = Lossy compression\n"
1312            " .....S = Lossless compression\n"
1313            " -------\n");
1314     for (i = 0; i < nb_codecs; i++) {
1315         const AVCodecDescriptor *desc = codecs[i];
1316         const AVCodec *codec = NULL;
1317
1318         printf(" ");
1319         printf(avcodec_find_decoder(desc->id) ? "D" : ".");
1320         printf(avcodec_find_encoder(desc->id) ? "E" : ".");
1321
1322         printf("%c", get_media_type_char(desc->type));
1323         printf((desc->props & AV_CODEC_PROP_INTRA_ONLY) ? "I" : ".");
1324         printf((desc->props & AV_CODEC_PROP_LOSSY)      ? "L" : ".");
1325         printf((desc->props & AV_CODEC_PROP_LOSSLESS)   ? "S" : ".");
1326
1327         printf(" %-20s %s", desc->name, desc->long_name ? desc->long_name : "");
1328
1329         /* print decoders/encoders when there's more than one or their
1330          * names are different from codec name */
1331         while ((codec = next_codec_for_id(desc->id, codec, 0))) {
1332             if (strcmp(codec->name, desc->name)) {
1333                 print_codecs_for_id(desc->id, 0);
1334                 break;
1335             }
1336         }
1337         codec = NULL;
1338         while ((codec = next_codec_for_id(desc->id, codec, 1))) {
1339             if (strcmp(codec->name, desc->name)) {
1340                 print_codecs_for_id(desc->id, 1);
1341                 break;
1342             }
1343         }
1344
1345         printf("\n");
1346     }
1347     av_free(codecs);
1348     return 0;
1349 }
1350
1351 static void print_codecs(int encoder)
1352 {
1353     const AVCodecDescriptor **codecs;
1354     unsigned i, nb_codecs = get_codecs_sorted(&codecs);
1355
1356     printf("%s:\n"
1357            " V..... = Video\n"
1358            " A..... = Audio\n"
1359            " S..... = Subtitle\n"
1360            " .F.... = Frame-level multithreading\n"
1361            " ..S... = Slice-level multithreading\n"
1362            " ...X.. = Codec is experimental\n"
1363            " ....B. = Supports draw_horiz_band\n"
1364            " .....D = Supports direct rendering method 1\n"
1365            " ------\n",
1366            encoder ? "Encoders" : "Decoders");
1367     for (i = 0; i < nb_codecs; i++) {
1368         const AVCodecDescriptor *desc = codecs[i];
1369         const AVCodec *codec = NULL;
1370
1371         while ((codec = next_codec_for_id(desc->id, codec, encoder))) {
1372             printf(" %c", get_media_type_char(desc->type));
1373             printf((codec->capabilities & CODEC_CAP_FRAME_THREADS) ? "F" : ".");
1374             printf((codec->capabilities & CODEC_CAP_SLICE_THREADS) ? "S" : ".");
1375             printf((codec->capabilities & CODEC_CAP_EXPERIMENTAL)  ? "X" : ".");
1376             printf((codec->capabilities & CODEC_CAP_DRAW_HORIZ_BAND)?"B" : ".");
1377             printf((codec->capabilities & CODEC_CAP_DR1)           ? "D" : ".");
1378
1379             printf(" %-20s %s", codec->name, codec->long_name ? codec->long_name : "");
1380             if (strcmp(codec->name, desc->name))
1381                 printf(" (codec %s)", desc->name);
1382
1383             printf("\n");
1384         }
1385     }
1386     av_free(codecs);
1387 }
1388
1389 int show_decoders(void *optctx, const char *opt, const char *arg)
1390 {
1391     print_codecs(0);
1392     return 0;
1393 }
1394
1395 int show_encoders(void *optctx, const char *opt, const char *arg)
1396 {
1397     print_codecs(1);
1398     return 0;
1399 }
1400
1401 int show_bsfs(void *optctx, const char *opt, const char *arg)
1402 {
1403     AVBitStreamFilter *bsf = NULL;
1404
1405     printf("Bitstream filters:\n");
1406     while ((bsf = av_bitstream_filter_next(bsf)))
1407         printf("%s\n", bsf->name);
1408     printf("\n");
1409     return 0;
1410 }
1411
1412 int show_protocols(void *optctx, const char *opt, const char *arg)
1413 {
1414     void *opaque = NULL;
1415     const char *name;
1416
1417     printf("Supported file protocols:\n"
1418            "Input:\n");
1419     while ((name = avio_enum_protocols(&opaque, 0)))
1420         printf("%s\n", name);
1421     printf("Output:\n");
1422     while ((name = avio_enum_protocols(&opaque, 1)))
1423         printf("%s\n", name);
1424     return 0;
1425 }
1426
1427 int show_filters(void *optctx, const char *opt, const char *arg)
1428 {
1429     AVFilter av_unused(**filter) = NULL;
1430     char descr[64], *descr_cur;
1431     int i, j;
1432     const AVFilterPad *pad;
1433
1434     printf("Filters:\n");
1435 #if CONFIG_AVFILTER
1436     while ((filter = av_filter_next(filter)) && *filter) {
1437         descr_cur = descr;
1438         for (i = 0; i < 2; i++) {
1439             if (i) {
1440                 *(descr_cur++) = '-';
1441                 *(descr_cur++) = '>';
1442             }
1443             pad = i ? (*filter)->outputs : (*filter)->inputs;
1444             for (j = 0; pad && pad[j].name; j++) {
1445                 if (descr_cur >= descr + sizeof(descr) - 4)
1446                     break;
1447                 *(descr_cur++) = get_media_type_char(pad[j].type);
1448             }
1449             if (!j)
1450                 *(descr_cur++) = '|';
1451         }
1452         *descr_cur = 0;
1453         printf("%-16s %-10s %s\n", (*filter)->name, descr, (*filter)->description);
1454     }
1455 #endif
1456     return 0;
1457 }
1458
1459 int show_pix_fmts(void *optctx, const char *opt, const char *arg)
1460 {
1461     const AVPixFmtDescriptor *pix_desc = NULL;
1462
1463     printf("Pixel formats:\n"
1464            "I.... = Supported Input  format for conversion\n"
1465            ".O... = Supported Output format for conversion\n"
1466            "..H.. = Hardware accelerated format\n"
1467            "...P. = Paletted format\n"
1468            "....B = Bitstream format\n"
1469            "FLAGS NAME            NB_COMPONENTS BITS_PER_PIXEL\n"
1470            "-----\n");
1471
1472 #if !CONFIG_SWSCALE
1473 #   define sws_isSupportedInput(x)  0
1474 #   define sws_isSupportedOutput(x) 0
1475 #endif
1476
1477     while ((pix_desc = av_pix_fmt_desc_next(pix_desc))) {
1478         enum AVPixelFormat pix_fmt = av_pix_fmt_desc_get_id(pix_desc);
1479         printf("%c%c%c%c%c %-16s       %d            %2d\n",
1480                sws_isSupportedInput (pix_fmt)      ? 'I' : '.',
1481                sws_isSupportedOutput(pix_fmt)      ? 'O' : '.',
1482                pix_desc->flags & PIX_FMT_HWACCEL   ? 'H' : '.',
1483                pix_desc->flags & PIX_FMT_PAL       ? 'P' : '.',
1484                pix_desc->flags & PIX_FMT_BITSTREAM ? 'B' : '.',
1485                pix_desc->name,
1486                pix_desc->nb_components,
1487                av_get_bits_per_pixel(pix_desc));
1488     }
1489     return 0;
1490 }
1491
1492 int show_layouts(void *optctx, const char *opt, const char *arg)
1493 {
1494     int i = 0;
1495     uint64_t layout, j;
1496     const char *name, *descr;
1497
1498     printf("Individual channels:\n"
1499            "NAME        DESCRIPTION\n");
1500     for (i = 0; i < 63; i++) {
1501         name = av_get_channel_name((uint64_t)1 << i);
1502         if (!name)
1503             continue;
1504         descr = av_get_channel_description((uint64_t)1 << i);
1505         printf("%-12s%s\n", name, descr);
1506     }
1507     printf("\nStandard channel layouts:\n"
1508            "NAME        DECOMPOSITION\n");
1509     for (i = 0; !av_get_standard_channel_layout(i, &layout, &name); i++) {
1510         if (name) {
1511             printf("%-12s", name);
1512             for (j = 1; j; j <<= 1)
1513                 if ((layout & j))
1514                     printf("%s%s", (layout & (j - 1)) ? "+" : "", av_get_channel_name(j));
1515             printf("\n");
1516         }
1517     }
1518     return 0;
1519 }
1520
1521 int show_sample_fmts(void *optctx, const char *opt, const char *arg)
1522 {
1523     int i;
1524     char fmt_str[128];
1525     for (i = -1; i < AV_SAMPLE_FMT_NB; i++)
1526         printf("%s\n", av_get_sample_fmt_string(fmt_str, sizeof(fmt_str), i));
1527     return 0;
1528 }
1529
1530 static void show_help_codec(const char *name, int encoder)
1531 {
1532     const AVCodecDescriptor *desc;
1533     const AVCodec *codec;
1534
1535     if (!name) {
1536         av_log(NULL, AV_LOG_ERROR, "No codec name specified.\n");
1537         return;
1538     }
1539
1540     codec = encoder ? avcodec_find_encoder_by_name(name) :
1541                       avcodec_find_decoder_by_name(name);
1542
1543     if (codec)
1544         print_codec(codec);
1545     else if ((desc = avcodec_descriptor_get_by_name(name))) {
1546         int printed = 0;
1547
1548         while ((codec = next_codec_for_id(desc->id, codec, encoder))) {
1549             printed = 1;
1550             print_codec(codec);
1551         }
1552
1553         if (!printed) {
1554             av_log(NULL, AV_LOG_ERROR, "Codec '%s' is known to FFmpeg, "
1555                    "but no %s for it are available. FFmpeg might need to be "
1556                    "recompiled with additional external libraries.\n",
1557                    name, encoder ? "encoders" : "decoders");
1558         }
1559     } else {
1560         av_log(NULL, AV_LOG_ERROR, "Codec '%s' is not recognized by FFmpeg.\n",
1561                name);
1562     }
1563 }
1564
1565 static void show_help_demuxer(const char *name)
1566 {
1567     const AVInputFormat *fmt = av_find_input_format(name);
1568
1569     if (!fmt) {
1570         av_log(NULL, AV_LOG_ERROR, "Unknown format '%s'.\n", name);
1571         return;
1572     }
1573
1574     printf("Demuxer %s [%s]:\n", fmt->name, fmt->long_name);
1575
1576     if (fmt->extensions)
1577         printf("    Common extensions: %s.\n", fmt->extensions);
1578
1579     if (fmt->priv_class)
1580         show_help_children(fmt->priv_class, AV_OPT_FLAG_DECODING_PARAM);
1581 }
1582
1583 static void show_help_muxer(const char *name)
1584 {
1585     const AVCodecDescriptor *desc;
1586     const AVOutputFormat *fmt = av_guess_format(name, NULL, NULL);
1587
1588     if (!fmt) {
1589         av_log(NULL, AV_LOG_ERROR, "Unknown format '%s'.\n", name);
1590         return;
1591     }
1592
1593     printf("Muxer %s [%s]:\n", fmt->name, fmt->long_name);
1594
1595     if (fmt->extensions)
1596         printf("    Common extensions: %s.\n", fmt->extensions);
1597     if (fmt->mime_type)
1598         printf("    Mime type: %s.\n", fmt->mime_type);
1599     if (fmt->video_codec != AV_CODEC_ID_NONE &&
1600         (desc = avcodec_descriptor_get(fmt->video_codec))) {
1601         printf("    Default video codec: %s.\n", desc->name);
1602     }
1603     if (fmt->audio_codec != AV_CODEC_ID_NONE &&
1604         (desc = avcodec_descriptor_get(fmt->audio_codec))) {
1605         printf("    Default audio codec: %s.\n", desc->name);
1606     }
1607     if (fmt->subtitle_codec != AV_CODEC_ID_NONE &&
1608         (desc = avcodec_descriptor_get(fmt->subtitle_codec))) {
1609         printf("    Default subtitle codec: %s.\n", desc->name);
1610     }
1611
1612     if (fmt->priv_class)
1613         show_help_children(fmt->priv_class, AV_OPT_FLAG_ENCODING_PARAM);
1614 }
1615
1616 static void show_help_filter(const char *name)
1617 {
1618     const AVFilter *filter;
1619
1620     if (!name) {
1621         av_log(NULL, AV_LOG_ERROR, "No filter name specified.\n");
1622         return;
1623     }
1624     filter = avfilter_get_by_name(name);
1625     if (!filter) {
1626         av_log(NULL, AV_LOG_ERROR, "Filter '%s' not found.\n", name);
1627         return;
1628     }
1629     printf("Filter %s\n", filter->name);
1630     if (filter->description)
1631         printf("  %s\n", filter->description);
1632     if (filter->priv_class)
1633         show_help_children(filter->priv_class, AV_OPT_FLAG_FILTERING_PARAM);
1634     else
1635         printf("No AVOption available\n");
1636 }
1637
1638 int show_help(void *optctx, const char *opt, const char *arg)
1639 {
1640     char *topic, *par;
1641     av_log_set_callback(log_callback_help);
1642
1643     topic = av_strdup(arg ? arg : "");
1644     par = strchr(topic, '=');
1645     if (par)
1646         *par++ = 0;
1647
1648     if (!*topic) {
1649         show_help_default(topic, par);
1650     } else if (!strcmp(topic, "decoder")) {
1651         show_help_codec(par, 0);
1652     } else if (!strcmp(topic, "encoder")) {
1653         show_help_codec(par, 1);
1654     } else if (!strcmp(topic, "demuxer")) {
1655         show_help_demuxer(par);
1656     } else if (!strcmp(topic, "muxer")) {
1657         show_help_muxer(par);
1658     } else if (!strcmp(topic, "filter")) {
1659         show_help_filter(par);
1660     } else {
1661         show_help_default(topic, par);
1662     }
1663
1664     av_freep(&topic);
1665     return 0;
1666 }
1667
1668 int read_yesno(void)
1669 {
1670     int c = getchar();
1671     int yesno = (av_toupper(c) == 'Y');
1672
1673     while (c != '\n' && c != EOF)
1674         c = getchar();
1675
1676     return yesno;
1677 }
1678
1679 int cmdutils_read_file(const char *filename, char **bufptr, size_t *size)
1680 {
1681     int ret;
1682     FILE *f = fopen(filename, "rb");
1683
1684     if (!f) {
1685         av_log(NULL, AV_LOG_ERROR, "Cannot read file '%s': %s\n", filename,
1686                strerror(errno));
1687         return AVERROR(errno);
1688     }
1689     fseek(f, 0, SEEK_END);
1690     *size = ftell(f);
1691     fseek(f, 0, SEEK_SET);
1692     if (*size == (size_t)-1) {
1693         av_log(NULL, AV_LOG_ERROR, "IO error: %s\n", strerror(errno));
1694         fclose(f);
1695         return AVERROR(errno);
1696     }
1697     *bufptr = av_malloc(*size + 1);
1698     if (!*bufptr) {
1699         av_log(NULL, AV_LOG_ERROR, "Could not allocate file buffer\n");
1700         fclose(f);
1701         return AVERROR(ENOMEM);
1702     }
1703     ret = fread(*bufptr, 1, *size, f);
1704     if (ret < *size) {
1705         av_free(*bufptr);
1706         if (ferror(f)) {
1707             av_log(NULL, AV_LOG_ERROR, "Error while reading file '%s': %s\n",
1708                    filename, strerror(errno));
1709             ret = AVERROR(errno);
1710         } else
1711             ret = AVERROR_EOF;
1712     } else {
1713         ret = 0;
1714         (*bufptr)[(*size)++] = '\0';
1715     }
1716
1717     fclose(f);
1718     return ret;
1719 }
1720
1721 FILE *get_preset_file(char *filename, size_t filename_size,
1722                       const char *preset_name, int is_path,
1723                       const char *codec_name)
1724 {
1725     FILE *f = NULL;
1726     int i;
1727     const char *base[3] = { getenv("FFMPEG_DATADIR"),
1728                             getenv("HOME"),
1729                             FFMPEG_DATADIR, };
1730
1731     if (is_path) {
1732         av_strlcpy(filename, preset_name, filename_size);
1733         f = fopen(filename, "r");
1734     } else {
1735 #ifdef _WIN32
1736         char datadir[MAX_PATH], *ls;
1737         base[2] = NULL;
1738
1739         if (GetModuleFileNameA(GetModuleHandleA(NULL), datadir, sizeof(datadir) - 1))
1740         {
1741             for (ls = datadir; ls < datadir + strlen(datadir); ls++)
1742                 if (*ls == '\\') *ls = '/';
1743
1744             if (ls = strrchr(datadir, '/'))
1745             {
1746                 *ls = 0;
1747                 strncat(datadir, "/ffpresets",  sizeof(datadir) - 1 - strlen(datadir));
1748                 base[2] = datadir;
1749             }
1750         }
1751 #endif
1752         for (i = 0; i < 3 && !f; i++) {
1753             if (!base[i])
1754                 continue;
1755             snprintf(filename, filename_size, "%s%s/%s.ffpreset", base[i],
1756                      i != 1 ? "" : "/.ffmpeg", preset_name);
1757             f = fopen(filename, "r");
1758             if (!f && codec_name) {
1759                 snprintf(filename, filename_size,
1760                          "%s%s/%s-%s.ffpreset",
1761                          base[i], i != 1 ? "" : "/.ffmpeg", codec_name,
1762                          preset_name);
1763                 f = fopen(filename, "r");
1764             }
1765         }
1766     }
1767
1768     return f;
1769 }
1770
1771 int check_stream_specifier(AVFormatContext *s, AVStream *st, const char *spec)
1772 {
1773     int ret = avformat_match_stream_specifier(s, st, spec);
1774     if (ret < 0)
1775         av_log(s, AV_LOG_ERROR, "Invalid stream specifier: %s.\n", spec);
1776     return ret;
1777 }
1778
1779 AVDictionary *filter_codec_opts(AVDictionary *opts, enum AVCodecID codec_id,
1780                                 AVFormatContext *s, AVStream *st, AVCodec *codec)
1781 {
1782     AVDictionary    *ret = NULL;
1783     AVDictionaryEntry *t = NULL;
1784     int            flags = s->oformat ? AV_OPT_FLAG_ENCODING_PARAM
1785                                       : AV_OPT_FLAG_DECODING_PARAM;
1786     char          prefix = 0;
1787     const AVClass    *cc = avcodec_get_class();
1788
1789     if (!codec)
1790         codec            = s->oformat ? avcodec_find_encoder(codec_id)
1791                                       : avcodec_find_decoder(codec_id);
1792
1793     switch (st->codec->codec_type) {
1794     case AVMEDIA_TYPE_VIDEO:
1795         prefix  = 'v';
1796         flags  |= AV_OPT_FLAG_VIDEO_PARAM;
1797         break;
1798     case AVMEDIA_TYPE_AUDIO:
1799         prefix  = 'a';
1800         flags  |= AV_OPT_FLAG_AUDIO_PARAM;
1801         break;
1802     case AVMEDIA_TYPE_SUBTITLE:
1803         prefix  = 's';
1804         flags  |= AV_OPT_FLAG_SUBTITLE_PARAM;
1805         break;
1806     }
1807
1808     while (t = av_dict_get(opts, "", t, AV_DICT_IGNORE_SUFFIX)) {
1809         char *p = strchr(t->key, ':');
1810
1811         /* check stream specification in opt name */
1812         if (p)
1813             switch (check_stream_specifier(s, st, p + 1)) {
1814             case  1: *p = 0; break;
1815             case  0:         continue;
1816             default:         return NULL;
1817             }
1818
1819         if (av_opt_find(&cc, t->key, NULL, flags, AV_OPT_SEARCH_FAKE_OBJ) ||
1820             (codec && codec->priv_class &&
1821              av_opt_find(&codec->priv_class, t->key, NULL, flags,
1822                          AV_OPT_SEARCH_FAKE_OBJ)))
1823             av_dict_set(&ret, t->key, t->value, 0);
1824         else if (t->key[0] == prefix &&
1825                  av_opt_find(&cc, t->key + 1, NULL, flags,
1826                              AV_OPT_SEARCH_FAKE_OBJ))
1827             av_dict_set(&ret, t->key + 1, t->value, 0);
1828
1829         if (p)
1830             *p = ':';
1831     }
1832     return ret;
1833 }
1834
1835 AVDictionary **setup_find_stream_info_opts(AVFormatContext *s,
1836                                            AVDictionary *codec_opts)
1837 {
1838     int i;
1839     AVDictionary **opts;
1840
1841     if (!s->nb_streams)
1842         return NULL;
1843     opts = av_mallocz(s->nb_streams * sizeof(*opts));
1844     if (!opts) {
1845         av_log(NULL, AV_LOG_ERROR,
1846                "Could not alloc memory for stream options.\n");
1847         return NULL;
1848     }
1849     for (i = 0; i < s->nb_streams; i++)
1850         opts[i] = filter_codec_opts(codec_opts, s->streams[i]->codec->codec_id,
1851                                     s, s->streams[i], NULL);
1852     return opts;
1853 }
1854
1855 void *grow_array(void *array, int elem_size, int *size, int new_size)
1856 {
1857     if (new_size >= INT_MAX / elem_size) {
1858         av_log(NULL, AV_LOG_ERROR, "Array too big.\n");
1859         exit(1);
1860     }
1861     if (*size < new_size) {
1862         uint8_t *tmp = av_realloc(array, new_size*elem_size);
1863         if (!tmp) {
1864             av_log(NULL, AV_LOG_ERROR, "Could not alloc buffer.\n");
1865             exit(1);
1866         }
1867         memset(tmp + *size*elem_size, 0, (new_size-*size) * elem_size);
1868         *size = new_size;
1869         return tmp;
1870     }
1871     return array;
1872 }