]> git.sesse.net Git - ffmpeg/blob - cmdutils.c
avcodec/h264: Check *log2_weight_denom
[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     fseek(f, 0, SEEK_END);
1868     *size = ftell(f);
1869     fseek(f, 0, SEEK_SET);
1870     if (*size == (size_t)-1) {
1871         ret = AVERROR(errno);
1872         av_log(NULL, AV_LOG_ERROR, "IO error: %s\n", strerror(errno));
1873         fclose(f);
1874         return ret;
1875     }
1876     *bufptr = av_malloc(*size + 1);
1877     if (!*bufptr) {
1878         av_log(NULL, AV_LOG_ERROR, "Could not allocate file buffer\n");
1879         fclose(f);
1880         return AVERROR(ENOMEM);
1881     }
1882     ret = fread(*bufptr, 1, *size, f);
1883     if (ret < *size) {
1884         av_free(*bufptr);
1885         if (ferror(f)) {
1886             ret = AVERROR(errno);
1887             av_log(NULL, AV_LOG_ERROR, "Error while reading file '%s': %s\n",
1888                    filename, strerror(errno));
1889         } else
1890             ret = AVERROR_EOF;
1891     } else {
1892         ret = 0;
1893         (*bufptr)[(*size)++] = '\0';
1894     }
1895
1896     fclose(f);
1897     return ret;
1898 }
1899
1900 FILE *get_preset_file(char *filename, size_t filename_size,
1901                       const char *preset_name, int is_path,
1902                       const char *codec_name)
1903 {
1904     FILE *f = NULL;
1905     int i;
1906     const char *base[3] = { getenv("FFMPEG_DATADIR"),
1907                             getenv("HOME"),
1908                             FFMPEG_DATADIR, };
1909
1910     if (is_path) {
1911         av_strlcpy(filename, preset_name, filename_size);
1912         f = fopen(filename, "r");
1913     } else {
1914 #ifdef _WIN32
1915         char datadir[MAX_PATH], *ls;
1916         base[2] = NULL;
1917
1918         if (GetModuleFileNameA(GetModuleHandleA(NULL), datadir, sizeof(datadir) - 1))
1919         {
1920             for (ls = datadir; ls < datadir + strlen(datadir); ls++)
1921                 if (*ls == '\\') *ls = '/';
1922
1923             if (ls = strrchr(datadir, '/'))
1924             {
1925                 *ls = 0;
1926                 strncat(datadir, "/ffpresets",  sizeof(datadir) - 1 - strlen(datadir));
1927                 base[2] = datadir;
1928             }
1929         }
1930 #endif
1931         for (i = 0; i < 3 && !f; i++) {
1932             if (!base[i])
1933                 continue;
1934             snprintf(filename, filename_size, "%s%s/%s.ffpreset", base[i],
1935                      i != 1 ? "" : "/.ffmpeg", preset_name);
1936             f = fopen(filename, "r");
1937             if (!f && codec_name) {
1938                 snprintf(filename, filename_size,
1939                          "%s%s/%s-%s.ffpreset",
1940                          base[i], i != 1 ? "" : "/.ffmpeg", codec_name,
1941                          preset_name);
1942                 f = fopen(filename, "r");
1943             }
1944         }
1945     }
1946
1947     return f;
1948 }
1949
1950 int check_stream_specifier(AVFormatContext *s, AVStream *st, const char *spec)
1951 {
1952     int ret = avformat_match_stream_specifier(s, st, spec);
1953     if (ret < 0)
1954         av_log(s, AV_LOG_ERROR, "Invalid stream specifier: %s.\n", spec);
1955     return ret;
1956 }
1957
1958 AVDictionary *filter_codec_opts(AVDictionary *opts, enum AVCodecID codec_id,
1959                                 AVFormatContext *s, AVStream *st, AVCodec *codec)
1960 {
1961     AVDictionary    *ret = NULL;
1962     AVDictionaryEntry *t = NULL;
1963     int            flags = s->oformat ? AV_OPT_FLAG_ENCODING_PARAM
1964                                       : AV_OPT_FLAG_DECODING_PARAM;
1965     char          prefix = 0;
1966     const AVClass    *cc = avcodec_get_class();
1967
1968     if (!codec)
1969         codec            = s->oformat ? avcodec_find_encoder(codec_id)
1970                                       : avcodec_find_decoder(codec_id);
1971
1972     switch (st->codec->codec_type) {
1973     case AVMEDIA_TYPE_VIDEO:
1974         prefix  = 'v';
1975         flags  |= AV_OPT_FLAG_VIDEO_PARAM;
1976         break;
1977     case AVMEDIA_TYPE_AUDIO:
1978         prefix  = 'a';
1979         flags  |= AV_OPT_FLAG_AUDIO_PARAM;
1980         break;
1981     case AVMEDIA_TYPE_SUBTITLE:
1982         prefix  = 's';
1983         flags  |= AV_OPT_FLAG_SUBTITLE_PARAM;
1984         break;
1985     }
1986
1987     while (t = av_dict_get(opts, "", t, AV_DICT_IGNORE_SUFFIX)) {
1988         char *p = strchr(t->key, ':');
1989
1990         /* check stream specification in opt name */
1991         if (p)
1992             switch (check_stream_specifier(s, st, p + 1)) {
1993             case  1: *p = 0; break;
1994             case  0:         continue;
1995             default:         exit_program(1);
1996             }
1997
1998         if (av_opt_find(&cc, t->key, NULL, flags, AV_OPT_SEARCH_FAKE_OBJ) ||
1999             !codec ||
2000             (codec->priv_class &&
2001              av_opt_find(&codec->priv_class, t->key, NULL, flags,
2002                          AV_OPT_SEARCH_FAKE_OBJ)))
2003             av_dict_set(&ret, t->key, t->value, 0);
2004         else if (t->key[0] == prefix &&
2005                  av_opt_find(&cc, t->key + 1, NULL, flags,
2006                              AV_OPT_SEARCH_FAKE_OBJ))
2007             av_dict_set(&ret, t->key + 1, t->value, 0);
2008
2009         if (p)
2010             *p = ':';
2011     }
2012     return ret;
2013 }
2014
2015 AVDictionary **setup_find_stream_info_opts(AVFormatContext *s,
2016                                            AVDictionary *codec_opts)
2017 {
2018     int i;
2019     AVDictionary **opts;
2020
2021     if (!s->nb_streams)
2022         return NULL;
2023     opts = av_mallocz_array(s->nb_streams, sizeof(*opts));
2024     if (!opts) {
2025         av_log(NULL, AV_LOG_ERROR,
2026                "Could not alloc memory for stream options.\n");
2027         return NULL;
2028     }
2029     for (i = 0; i < s->nb_streams; i++)
2030         opts[i] = filter_codec_opts(codec_opts, s->streams[i]->codec->codec_id,
2031                                     s, s->streams[i], NULL);
2032     return opts;
2033 }
2034
2035 void *grow_array(void *array, int elem_size, int *size, int new_size)
2036 {
2037     if (new_size >= INT_MAX / elem_size) {
2038         av_log(NULL, AV_LOG_ERROR, "Array too big.\n");
2039         exit_program(1);
2040     }
2041     if (*size < new_size) {
2042         uint8_t *tmp = av_realloc(array, new_size*elem_size);
2043         if (!tmp) {
2044             av_log(NULL, AV_LOG_ERROR, "Could not alloc buffer.\n");
2045             exit_program(1);
2046         }
2047         memset(tmp + *size*elem_size, 0, (new_size-*size) * elem_size);
2048         *size = new_size;
2049         return tmp;
2050     }
2051     return array;
2052 }
2053
2054 #if CONFIG_AVDEVICE
2055 static int print_device_sources(AVInputFormat *fmt, AVDictionary *opts)
2056 {
2057     int ret, i;
2058     AVFormatContext *dev = NULL;
2059     AVDeviceInfoList *device_list = NULL;
2060     AVDictionary *tmp_opts = NULL;
2061
2062     if (!fmt || !fmt->priv_class  || !AV_IS_INPUT_DEVICE(fmt->priv_class->category))
2063         return AVERROR(EINVAL);
2064
2065     printf("Audo-detected sources for %s:\n", fmt->name);
2066     if (!fmt->get_device_list) {
2067         ret = AVERROR(ENOSYS);
2068         printf("Cannot list sources. Not implemented.\n");
2069         goto fail;
2070     }
2071
2072     /* TODO: avformat_open_input calls read_header callback which is not necessary.
2073              Function like avformat_alloc_output_context2 for input could be helpful here. */
2074     av_dict_copy(&tmp_opts, opts, 0);
2075     if ((ret = avformat_open_input(&dev, NULL, fmt, &tmp_opts)) < 0) {
2076         printf("Cannot open device: %s.\n", fmt->name);
2077         goto fail;
2078     }
2079
2080     if ((ret = avdevice_list_devices(dev, &device_list)) < 0) {
2081         printf("Cannot list sources.\n");
2082         goto fail;
2083     }
2084
2085     for (i = 0; i < device_list->nb_devices; i++) {
2086         printf("%s %s [%s]\n", device_list->default_device == i ? "*" : " ",
2087                device_list->devices[i]->device_name, device_list->devices[i]->device_description);
2088     }
2089
2090   fail:
2091     av_dict_free(&tmp_opts);
2092     avdevice_free_list_devices(&device_list);
2093     avformat_close_input(&dev);
2094     return ret;
2095 }
2096
2097 static int print_device_sinks(AVOutputFormat *fmt, AVDictionary *opts)
2098 {
2099     int ret, i;
2100     AVFormatContext *dev = NULL;
2101     AVDeviceInfoList *device_list = NULL;
2102     AVDictionary *tmp_opts = NULL;
2103
2104     if (!fmt || !fmt->priv_class  || !AV_IS_OUTPUT_DEVICE(fmt->priv_class->category))
2105         return AVERROR(EINVAL);
2106
2107     printf("Audo-detected sinks for %s:\n", fmt->name);
2108     if (!fmt->get_device_list) {
2109         ret = AVERROR(ENOSYS);
2110         printf("Cannot list sinks. Not implemented.\n");
2111         goto fail;
2112     }
2113
2114     if ((ret = avformat_alloc_output_context2(&dev, fmt, NULL, NULL)) < 0) {
2115         printf("Cannot open device: %s.\n", fmt->name);
2116         goto fail;
2117     }
2118     av_dict_copy(&tmp_opts, opts, 0);
2119     av_opt_set_dict2(dev, &tmp_opts, AV_OPT_SEARCH_CHILDREN);
2120
2121     if ((ret = avdevice_list_devices(dev, &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     av_dict_free(&tmp_opts);
2133     avdevice_free_list_devices(&device_list);
2134     avformat_free_context(dev);
2135     return ret;
2136 }
2137
2138 static int show_sinks_sources_parse_arg(const char *arg, char **dev, AVDictionary **opts)
2139 {
2140     int ret;
2141     if (arg) {
2142         char *opts_str = NULL;
2143         av_assert0(dev && opts);
2144         *dev = av_strdup(arg);
2145         if (!*dev)
2146             return AVERROR(ENOMEM);
2147         if ((opts_str = strchr(*dev, ','))) {
2148             *(opts_str++) = '\0';
2149             if (opts_str[0] && ((ret = av_dict_parse_string(opts, opts_str, "=", ":", 0)) < 0)) {
2150                 av_freep(dev);
2151                 return ret;
2152             }
2153         }
2154     } else
2155         printf("\nDevice name is not provided.\n"
2156                 "You can pass devicename[,opt1=val1[,opt2=val2...]] as an argument.\n\n");
2157     return 0;
2158 }
2159
2160 int show_sources(void *optctx, const char *opt, const char *arg)
2161 {
2162     AVInputFormat *fmt = NULL;
2163     char *dev = NULL;
2164     AVDictionary *opts = NULL;
2165     int ret = 0;
2166     int error_level = av_log_get_level();
2167
2168     av_log_set_level(AV_LOG_ERROR);
2169
2170     if ((ret = show_sinks_sources_parse_arg(arg, &dev, &opts)) < 0)
2171         goto fail;
2172
2173     do {
2174         fmt = av_input_audio_device_next(fmt);
2175         if (fmt) {
2176             if (!strcmp(fmt->name, "lavfi"))
2177                 continue; //it's pointless to probe lavfi
2178             if (dev && strcmp(fmt->name, dev))
2179                 continue;
2180             print_device_sources(fmt, opts);
2181         }
2182     } while (fmt);
2183     do {
2184         fmt = av_input_video_device_next(fmt);
2185         if (fmt) {
2186             if (dev && strcmp(fmt->name, dev))
2187                 continue;
2188             print_device_sources(fmt, opts);
2189         }
2190     } while (fmt);
2191   fail:
2192     av_dict_free(&opts);
2193     av_free(dev);
2194     av_log_set_level(error_level);
2195     return ret;
2196 }
2197
2198 int show_sinks(void *optctx, const char *opt, const char *arg)
2199 {
2200     AVOutputFormat *fmt = NULL;
2201     char *dev = NULL;
2202     AVDictionary *opts = NULL;
2203     int ret = 0;
2204     int error_level = av_log_get_level();
2205
2206     av_log_set_level(AV_LOG_ERROR);
2207
2208     if ((ret = show_sinks_sources_parse_arg(arg, &dev, &opts)) < 0)
2209         goto fail;
2210
2211     do {
2212         fmt = av_output_audio_device_next(fmt);
2213         if (fmt) {
2214             if (dev && strcmp(fmt->name, dev))
2215                 continue;
2216             print_device_sinks(fmt, opts);
2217         }
2218     } while (fmt);
2219     do {
2220         fmt = av_output_video_device_next(fmt);
2221         if (fmt) {
2222             if (dev && strcmp(fmt->name, dev))
2223                 continue;
2224             print_device_sinks(fmt, opts);
2225         }
2226     } while (fmt);
2227   fail:
2228     av_dict_free(&opts);
2229     av_free(dev);
2230     av_log_set_level(error_level);
2231     return ret;
2232 }
2233 #endif