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