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