1629 lines · 52.2 KB · C
RawDownload
/* * Various utilities for command line tools * Copyright (c) 2000-2003 Fabrice Bellard * * This file is part of Libav. * * Libav is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * Libav is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with Libav; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include <string.h>#include <stdlib.h>#include <errno.h>#include <math.h> /* Include only the enabled headers since some compilers (namely, Sun   Studio) will not omit unused inline functions and create undefined   references to libraries that are not being built. */ #include "config.h"#include "libavformat/avformat.h"#include "libavfilter/avfilter.h"#include "libavdevice/avdevice.h"#include "libavresample/avresample.h"#include "libswscale/swscale.h"#include "libavutil/avassert.h"#include "libavutil/avstring.h"#include "libavutil/mathematics.h"#include "libavutil/imgutils.h"#include "libavutil/parseutils.h"#include "libavutil/pixdesc.h"#include "libavutil/eval.h"#include "libavutil/dict.h"#include "libavutil/opt.h"#include "cmdutils.h"#include "version.h"#if CONFIG_NETWORK#include "libavformat/network.h"#endif#if HAVE_SYS_RESOURCE_H#include <sys/time.h>#include <sys/resource.h>#endif struct SwsContext *sws_opts;AVDictionary *format_opts, *codec_opts, *resample_opts; static const int this_year = 2013; void init_opts(void){#if CONFIG_SWSCALE    sws_opts = sws_getContext(16, 16, 0, 16, 16, 0, SWS_BICUBIC,                              NULL, NULL, NULL);#endif} void uninit_opts(void){#if CONFIG_SWSCALE    sws_freeContext(sws_opts);    sws_opts = NULL;#endif    av_dict_free(&format_opts);    av_dict_free(&codec_opts);    av_dict_free(&resample_opts);} void log_callback_help(void *ptr, int level, const char *fmt, va_list vl){    vfprintf(stdout, fmt, vl);} static void (*program_exit)(int ret); void register_exit(void (*cb)(int ret)){    program_exit = cb;} void exit_program(int ret){    if (program_exit)        program_exit(ret);     exit(ret);} double parse_number_or_die(const char *context, const char *numstr, int type,                           double min, double max){    char *tail;    const char *error;    double d = av_strtod(numstr, &tail);    if (*tail)        error = "Expected number for %s but found: %s\n";    else if (d < min || d > max)        error = "The value for %s was %s which is not within %f - %f\n";    else if (type == OPT_INT64 && (int64_t)d != d)        error = "Expected int64 for %s but found %s\n";    else if (type == OPT_INT && (int)d != d)        error = "Expected int for %s but found %s\n";    else        return d;    av_log(NULL, AV_LOG_FATAL, error, context, numstr, min, max);    exit_program(1);    return 0;} int64_t parse_time_or_die(const char *context, const char *timestr,                          int is_duration){    int64_t us;    if (av_parse_time(&us, timestr, is_duration) < 0) {        av_log(NULL, AV_LOG_FATAL, "Invalid %s specification for %s: %s\n",               is_duration ? "duration" : "date", context, timestr);        exit_program(1);    }    return us;} void show_help_options(const OptionDef *options, const char *msg, int req_flags,                       int rej_flags, int alt_flags){    const OptionDef *po;    int first;     first = 1;    for (po = options; po->name != NULL; po++) {        char buf[64];         if (((po->flags & req_flags) != req_flags) ||            (alt_flags && !(po->flags & alt_flags)) ||            (po->flags & rej_flags))            continue;         if (first) {            printf("%s\n", msg);            first = 0;        }        av_strlcpy(buf, po->name, sizeof(buf));        if (po->argname) {            av_strlcat(buf, " ", sizeof(buf));            av_strlcat(buf, po->argname, sizeof(buf));        }        printf("-%-17s  %s\n", buf, po->help);    }    printf("\n");} void show_help_children(const AVClass *class, int flags){    const AVClass *child = NULL;    av_opt_show2(&class, NULL, flags, 0);    printf("\n");     while (child = av_opt_child_class_next(class, child))        show_help_children(child, flags);} static const OptionDef *find_option(const OptionDef *po, const char *name){    const char *p = strchr(name, ':');    int len = p ? p - name : strlen(name);     while (po->name != NULL) {        if (!strncmp(name, po->name, len) && strlen(po->name) == len)            break;        po++;    }    return po;} #if HAVE_COMMANDLINETOARGVW#include <windows.h>#include <shellapi.h>/* Will be leaked on exit */static char** win32_argv_utf8 = NULL;static int win32_argc = 0; /** * Prepare command line arguments for executable. * For Windows - perform wide-char to UTF-8 conversion. * Input arguments should be main() function arguments. * @param argc_ptr Arguments number (including executable) * @param argv_ptr Arguments list. */static void prepare_app_arguments(int *argc_ptr, char ***argv_ptr){    char *argstr_flat;    wchar_t **argv_w;    int i, buffsize = 0, offset = 0;     if (win32_argv_utf8) {        *argc_ptr = win32_argc;        *argv_ptr = win32_argv_utf8;        return;    }     win32_argc = 0;    argv_w = CommandLineToArgvW(GetCommandLineW(), &win32_argc);    if (win32_argc <= 0 || !argv_w)        return;     /* determine the UTF-8 buffer size (including NULL-termination symbols) */    for (i = 0; i < win32_argc; i++)        buffsize += WideCharToMultiByte(CP_UTF8, 0, argv_w[i], -1,                                        NULL, 0, NULL, NULL);     win32_argv_utf8 = av_mallocz(sizeof(char *) * (win32_argc + 1) + buffsize);    argstr_flat     = (char *)win32_argv_utf8 + sizeof(char *) * (win32_argc + 1);    if (win32_argv_utf8 == NULL) {        LocalFree(argv_w);        return;    }     for (i = 0; i < win32_argc; i++) {        win32_argv_utf8[i] = &argstr_flat[offset];        offset += WideCharToMultiByte(CP_UTF8, 0, argv_w[i], -1,                                      &argstr_flat[offset],                                      buffsize - offset, NULL, NULL);    }    win32_argv_utf8[i] = NULL;    LocalFree(argv_w);     *argc_ptr = win32_argc;    *argv_ptr = win32_argv_utf8;}#elsestatic inline void prepare_app_arguments(int *argc_ptr, char ***argv_ptr){    /* nothing to do */}#endif /* HAVE_COMMANDLINETOARGVW */ static int write_option(void *optctx, const OptionDef *po, const char *opt,                        const char *arg){    /* new-style options contain an offset into optctx, old-style address of     * a global var*/    void *dst = po->flags & (OPT_OFFSET | OPT_SPEC) ?                (uint8_t *)optctx + po->u.off : po->u.dst_ptr;    int *dstcount;     if (po->flags & OPT_SPEC) {        SpecifierOpt **so = dst;        char *p = strchr(opt, ':');         dstcount = (int *)(so + 1);        *so = grow_array(*so, sizeof(**so), dstcount, *dstcount + 1);        (*so)[*dstcount - 1].specifier = av_strdup(p ? p + 1 : "");        dst = &(*so)[*dstcount - 1].u;    }     if (po->flags & OPT_STRING) {        char *str;        str = av_strdup(arg);        av_freep(dst);        *(char **)dst = str;    } else if (po->flags & OPT_BOOL || po->flags & OPT_INT) {        *(int *)dst = parse_number_or_die(opt, arg, OPT_INT64, INT_MIN, INT_MAX);    } else if (po->flags & OPT_INT64) {        *(int64_t *)dst = parse_number_or_die(opt, arg, OPT_INT64, INT64_MIN, INT64_MAX);    } else if (po->flags & OPT_TIME) {        *(int64_t *)dst = parse_time_or_die(opt, arg, 1);    } else if (po->flags & OPT_FLOAT) {        *(float *)dst = parse_number_or_die(opt, arg, OPT_FLOAT, -INFINITY, INFINITY);    } else if (po->flags & OPT_DOUBLE) {        *(double *)dst = parse_number_or_die(opt, arg, OPT_DOUBLE, -INFINITY, INFINITY);    } else if (po->u.func_arg) {        int ret = po->u.func_arg(optctx, opt, arg);        if (ret < 0) {            av_log(NULL, AV_LOG_ERROR,                   "Failed to set value '%s' for option '%s'\n", arg, opt);            return ret;        }    }    if (po->flags & OPT_EXIT)        exit_program(0);     return 0;} int parse_option(void *optctx, const char *opt, const char *arg,                 const OptionDef *options){    const OptionDef *po;    int ret;     po = find_option(options, opt);    if (!po->name && opt[0] == 'n' && opt[1] == 'o') {        /* handle 'no' bool option */        po = find_option(options, opt + 2);        if ((po->name && (po->flags & OPT_BOOL)))            arg = "0";    } else if (po->flags & OPT_BOOL)        arg = "1";     if (!po->name)        po = find_option(options, "default");    if (!po->name) {        av_log(NULL, AV_LOG_ERROR, "Unrecognized option '%s'\n", opt);        return AVERROR(EINVAL);    }    if (po->flags & HAS_ARG && !arg) {        av_log(NULL, AV_LOG_ERROR, "Missing argument for option '%s'\n", opt);        return AVERROR(EINVAL);    }     ret = write_option(optctx, po, opt, arg);    if (ret < 0)        return ret;     return !!(po->flags & HAS_ARG);} void parse_options(void *optctx, int argc, char **argv, const OptionDef *options,                   void (*parse_arg_function)(void *, const char*)){    const char *opt;    int optindex, handleoptions = 1, ret;     /* perform system-dependent conversions for arguments list */    prepare_app_arguments(&argc, &argv);     /* parse options */    optindex = 1;    while (optindex < argc) {        opt = argv[optindex++];         if (handleoptions && opt[0] == '-' && opt[1] != '\0') {            if (opt[1] == '-' && opt[2] == '\0') {                handleoptions = 0;                continue;            }            opt++;             if ((ret = parse_option(optctx, opt, argv[optindex], options)) < 0)                exit_program(1);            optindex += ret;        } else {            if (parse_arg_function)                parse_arg_function(optctx, opt);        }    }} int parse_optgroup(void *optctx, OptionGroup *g){    int i, ret;     av_log(NULL, AV_LOG_DEBUG, "Parsing a group of options: %s %s.\n",           g->group_def->name, g->arg);     for (i = 0; i < g->nb_opts; i++) {        Option *o = &g->opts[i];         if (g->group_def->flags &&            !(g->group_def->flags & o->opt->flags)) {            av_log(NULL, AV_LOG_ERROR, "Option %s (%s) cannot be applied to "                   "%s %s -- you are trying to apply an input option to an "                   "output file or vice versa. Move this option before the "                   "file it belongs to.\n", o->key, o->opt->help,                   g->group_def->name, g->arg);            return AVERROR(EINVAL);        }         av_log(NULL, AV_LOG_DEBUG, "Applying option %s (%s) with argument %s.\n",               o->key, o->opt->help, o->val);         ret = write_option(optctx, o->opt, o->key, o->val);        if (ret < 0)            return ret;    }     av_log(NULL, AV_LOG_DEBUG, "Successfully parsed a group of options.\n");     return 0;} int locate_option(int argc, char **argv, const OptionDef *options,                  const char *optname){    const OptionDef *po;    int i;     for (i = 1; i < argc; i++) {        const char *cur_opt = argv[i];         if (*cur_opt++ != '-')            continue;         po = find_option(options, cur_opt);        if (!po->name && cur_opt[0] == 'n' && cur_opt[1] == 'o')            po = find_option(options, cur_opt + 2);         if ((!po->name && !strcmp(cur_opt, optname)) ||             (po->name && !strcmp(optname, po->name)))            return i;         if (!po || po->flags & HAS_ARG)            i++;    }    return 0;} void parse_loglevel(int argc, char **argv, const OptionDef *options){    int idx = locate_option(argc, argv, options, "loglevel");    if (!idx)        idx = locate_option(argc, argv, options, "v");    if (idx && argv[idx + 1])        opt_loglevel(NULL, "loglevel", argv[idx + 1]);} #define FLAGS (o->type == AV_OPT_TYPE_FLAGS) ? AV_DICT_APPEND : 0int opt_default(void *optctx, const char *opt, const char *arg){    const AVOption *o;    char opt_stripped[128];    const char *p;    const AVClass *cc = avcodec_get_class(), *fc = avformat_get_class();#if CONFIG_AVRESAMPLE    const AVClass *rc = avresample_get_class();#endif#if CONFIG_SWSCALE    const AVClass *sc = sws_get_class();#endif     if (!(p = strchr(opt, ':')))        p = opt + strlen(opt);    av_strlcpy(opt_stripped, opt, FFMIN(sizeof(opt_stripped), p - opt + 1));     if ((o = av_opt_find(&cc, opt_stripped, NULL, 0,                         AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)) ||        ((opt[0] == 'v' || opt[0] == 'a' || opt[0] == 's') &&         (o = av_opt_find(&cc, opt + 1, NULL, 0, AV_OPT_SEARCH_FAKE_OBJ))))        av_dict_set(&codec_opts, opt, arg, FLAGS);    else if ((o = av_opt_find(&fc, opt, NULL, 0,                              AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)))        av_dict_set(&format_opts, opt, arg, FLAGS);#if CONFIG_AVRESAMPLE    else if ((o = av_opt_find(&rc, opt, NULL, 0,                              AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)))        av_dict_set(&resample_opts, opt, arg, FLAGS);#endif#if CONFIG_SWSCALE    else if ((o = av_opt_find(&sc, opt, NULL, 0,                              AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {        // XXX we only support sws_flags, not arbitrary sws options        int ret = av_opt_set(sws_opts, opt, arg, 0);        if (ret < 0) {            av_log(NULL, AV_LOG_ERROR, "Error setting option %s.\n", opt);            return ret;        }    }#endif     if (o)        return 0;    return AVERROR_OPTION_NOT_FOUND;} /* * Check whether given option is a group separator. * * @return index of the group definition that matched or -1 if none */static int match_group_separator(const OptionGroupDef *groups, int nb_groups,                                 const char *opt){    int i;     for (i = 0; i < nb_groups; i++) {        const OptionGroupDef *p = &groups[i];        if (p->sep && !strcmp(p->sep, opt))            return i;    }     return -1;} /* * Finish parsing an option group. * * @param group_idx which group definition should this group belong to * @param arg argument of the group delimiting option */static void finish_group(OptionParseContext *octx, int group_idx,

Lines 1–500 of 1629.

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t