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