opt.c 15.1 KB
Newer Older
Michael Niedermayer's avatar
Michael Niedermayer committed
1 2 3 4
/*
 * AVOptions
 * Copyright (c) 2005 Michael Niedermayer <michaelni@gmx.at>
 *
5 6 7
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
Michael Niedermayer's avatar
Michael Niedermayer committed
8 9
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
Michael Niedermayer's avatar
Michael Niedermayer committed
11
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
Michael Niedermayer's avatar
Michael Niedermayer committed
13 14 15 16 17
 * 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
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
Michael Niedermayer's avatar
Michael Niedermayer committed
20
 */
21

Michael Niedermayer's avatar
Michael Niedermayer committed
22
/**
23
 * @file libavcodec/opt.c
Michael Niedermayer's avatar
Michael Niedermayer committed
24 25 26
 * AVOptions
 * @author Michael Niedermayer <michaelni@gmx.at>
 */
27

Michael Niedermayer's avatar
Michael Niedermayer committed
28
#include "avcodec.h"
29
#include "opt.h"
30
#include "eval.h"
Michael Niedermayer's avatar
Michael Niedermayer committed
31 32

//FIXME order them and do a bin search
33
const AVOption *av_find_opt(void *v, const char *name, const char *unit, int mask, int flags){
Michael Niedermayer's avatar
Michael Niedermayer committed
34
    AVClass *c= *(AVClass**)v; //FIXME silly way of storing AVClass
35
    const AVOption *o= c->option;
36

Michael Niedermayer's avatar
Michael Niedermayer committed
37
    for(;o && o->name; o++){
38
        if(!strcmp(o->name, name) && (!unit || (o->unit && !strcmp(o->unit, unit))) && (o->flags & mask) == flags )
Michael Niedermayer's avatar
Michael Niedermayer committed
39 40 41 42 43
            return o;
    }
    return NULL;
}

44
const AVOption *av_next_option(void *obj, const AVOption *last){
45 46 47 48 49
    if(last && last[1].name) return ++last;
    else if(last)            return NULL;
    else                     return (*(AVClass**)obj)->option;
}

50
static int av_set_number2(void *obj, const char *name, double num, int den, int64_t intnum, const AVOption **o_out){
51
    const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);
Michael Niedermayer's avatar
Michael Niedermayer committed
52
    void *dst;
53 54
    if(o_out)
        *o_out= o;
55
    if(!o || o->offset<=0)
56
        return AVERROR(ENOENT);
57

58
    if(o->max*den < num*intnum || o->min*den > num*intnum) {
59
        av_log(obj, AV_LOG_ERROR, "Value %lf for parameter '%s' out of range\n", num, name);
60
        return AVERROR(ERANGE);
61
    }
62

Michael Niedermayer's avatar
Michael Niedermayer committed
63 64 65
    dst= ((uint8_t*)obj) + o->offset;

    switch(o->type){
66
    case FF_OPT_TYPE_FLAGS:
67 68
    case FF_OPT_TYPE_INT:   *(int       *)dst= llrint(num/den)*intnum; break;
    case FF_OPT_TYPE_INT64: *(int64_t   *)dst= llrint(num/den)*intnum; break;
69 70
    case FF_OPT_TYPE_FLOAT: *(float     *)dst= num*intnum/den;         break;
    case FF_OPT_TYPE_DOUBLE:*(double    *)dst= num*intnum/den;         break;
Michael Niedermayer's avatar
Michael Niedermayer committed
71
    case FF_OPT_TYPE_RATIONAL:
72 73
        if((int)num == num) *(AVRational*)dst= (AVRational){num*intnum, den};
        else                *(AVRational*)dst= av_d2q(num*intnum/den, 1<<24);
74
        break;
Michael Niedermayer's avatar
Michael Niedermayer committed
75
    default:
76
        return AVERROR(EINVAL);
Michael Niedermayer's avatar
Michael Niedermayer committed
77
    }
78 79 80 81 82 83 84 85 86
    return 0;
}

static const AVOption *av_set_number(void *obj, const char *name, double num, int den, int64_t intnum){
    const AVOption *o = NULL;
    if (av_set_number2(obj, name, num, den, intnum, &o) < 0)
        return NULL;
    else
        return o;
Michael Niedermayer's avatar
Michael Niedermayer committed
87 88
}

89
static const double const_values[]={
90 91 92 93 94 95
    M_PI,
    M_E,
    FF_QP2LAMBDA,
    0
};

96
static const char * const const_names[]={
97 98 99 100 101 102
    "PI",
    "E",
    "QP2LAMBDA",
    0
};

103 104 105 106 107 108 109
static int hexchar2int(char c) {
    if (c >= '0' && c <= '9') return c - '0';
    if (c >= 'a' && c <= 'f') return c - 'a' + 10;
    if (c >= 'A' && c <= 'F') return c - 'A' + 10;
    return -1;
}

110 111
int av_set_string3(void *obj, const char *name, const char *val, int alloc, const AVOption **o_out){
    int ret;
112
    const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);
113 114
    if (o_out)
        *o_out = o;
Michael Niedermayer's avatar
Michael Niedermayer committed
115
    if(!o)
116 117 118 119
        return AVERROR(ENOENT);
    if(!val || o->offset<=0)
        return AVERROR(EINVAL);

120 121 122 123 124 125 126
    if(o->type == FF_OPT_TYPE_BINARY){
        uint8_t **dst = (uint8_t **)(((uint8_t*)obj) + o->offset);
        int *lendst = (int *)(dst + 1);
        uint8_t *bin, *ptr;
        int len = strlen(val);
        av_freep(dst);
        *lendst = 0;
127
        if (len & 1) return AVERROR(EINVAL);
128 129 130 131 132 133 134
        len /= 2;
        ptr = bin = av_malloc(len);
        while (*val) {
            int a = hexchar2int(*val++);
            int b = hexchar2int(*val++);
            if (a < 0 || b < 0) {
                av_free(bin);
135
                return AVERROR(EINVAL);
136 137 138 139 140
            }
            *ptr++ = (a << 4) | b;
        }
        *dst = bin;
        *lendst = len;
141
        return 0;
142
    }
Michael Niedermayer's avatar
Michael Niedermayer committed
143
    if(o->type != FF_OPT_TYPE_STRING){
144
        int notfirst=0;
Michael Niedermayer's avatar
Michael Niedermayer committed
145 146
        for(;;){
            int i;
147
            char buf[256];
148 149
            int cmd=0;
            double d;
Luca Abeni's avatar
Luca Abeni committed
150
            const char *error = NULL;
Michael Niedermayer's avatar
Michael Niedermayer committed
151

152 153
            if(*val == '+' || *val == '-')
                cmd= *(val++);
154

155
            for(i=0; i<sizeof(buf)-1 && val[i] && val[i]!='+' && val[i]!='-'; i++)
Michael Niedermayer's avatar
Michael Niedermayer committed
156 157
                buf[i]= val[i];
            buf[i]=0;
158

159 160
            d = ff_eval2(buf, const_values, const_names, NULL, NULL, NULL, NULL, NULL, &error);
            if(isnan(d)) {
161
                const AVOption *o_named= av_find_opt(obj, buf, o->unit, 0, 0);
162
                if(o_named && o_named->type == FF_OPT_TYPE_CONST)
163 164 165 166
                    d= o_named->default_val;
                else if(!strcmp(buf, "default")) d= o->default_val;
                else if(!strcmp(buf, "max"    )) d= o->max;
                else if(!strcmp(buf, "min"    )) d= o->min;
167 168
                else if(!strcmp(buf, "none"   )) d= 0;
                else if(!strcmp(buf, "all"    )) d= ~0;
169
                else {
Stefano Sabatini's avatar
Stefano Sabatini committed
170
                    if (error)
171
                        av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\": %s\n", val, error);
172
                    return AVERROR(EINVAL);
173
                }
Michael Niedermayer's avatar
Michael Niedermayer committed
174
            }
175 176 177
            if(o->type == FF_OPT_TYPE_FLAGS){
                if     (cmd=='+') d= av_get_int(obj, name, NULL) | (int64_t)d;
                else if(cmd=='-') d= av_get_int(obj, name, NULL) &~(int64_t)d;
178
            }else{
179 180
                if     (cmd=='+') d= notfirst*av_get_double(obj, name, NULL) + d;
                else if(cmd=='-') d= notfirst*av_get_double(obj, name, NULL) - d;
181
            }
Michael Niedermayer's avatar
Michael Niedermayer committed
182

183 184
            if ((ret = av_set_number2(obj, name, d, 1, 1, o_out)) < 0)
                return ret;
185
            val+= i;
Michael Niedermayer's avatar
Michael Niedermayer committed
186
            if(!*val)
187
                return 0;
188
            notfirst=1;
Michael Niedermayer's avatar
Michael Niedermayer committed
189
        }
190
        return AVERROR(EINVAL);
Michael Niedermayer's avatar
Michael Niedermayer committed
191
    }
192

193
    if(alloc){
194
        av_free(*(void**)(((uint8_t*)obj) + o->offset));
195 196 197
        val= av_strdup(val);
    }

198
    memcpy(((uint8_t*)obj) + o->offset, &val, sizeof(val));
199 200 201
    return 0;
}

202
#if LIBAVCODEC_VERSION_MAJOR < 53
203 204 205 206
const AVOption *av_set_string2(void *obj, const char *name, const char *val, int alloc){
    const AVOption *o;
    if (av_set_string3(obj, name, val, alloc, &o) < 0)
        return NULL;
207
    return o;
Michael Niedermayer's avatar
Michael Niedermayer committed
208 209
}

210
const AVOption *av_set_string(void *obj, const char *name, const char *val){
211 212 213 214
    const AVOption *o;
    if (av_set_string3(obj, name, val, 0, &o) < 0)
        return NULL;
    return o;
215
}
216
#endif
217

218
const AVOption *av_set_double(void *obj, const char *name, double n){
Michael Niedermayer's avatar
Michael Niedermayer committed
219 220 221
    return av_set_number(obj, name, n, 1, 1);
}

222
const AVOption *av_set_q(void *obj, const char *name, AVRational n){
Michael Niedermayer's avatar
Michael Niedermayer committed
223 224 225
    return av_set_number(obj, name, n.num, n.den, 1);
}

226
const AVOption *av_set_int(void *obj, const char *name, int64_t n){
Michael Niedermayer's avatar
Michael Niedermayer committed
227 228 229
    return av_set_number(obj, name, 1, 1, n);
}

230
/**
231
 *
232 233 234
 * @param buf a buffer which is used for returning non string values as strings, can be NULL
 * @param buf_len allocated length in bytes of buf
 */
235
const char *av_get_string(void *obj, const char *name, const AVOption **o_out, char *buf, int buf_len){
236
    const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);
237
    void *dst;
238 239
    uint8_t *bin;
    int len, i;
Michael Niedermayer's avatar
Michael Niedermayer committed
240 241
    if(!o || o->offset<=0)
        return NULL;
242
    if(o->type != FF_OPT_TYPE_STRING && (!buf || !buf_len))
Michael Niedermayer's avatar
Michael Niedermayer committed
243 244
        return NULL;

245 246
    dst= ((uint8_t*)obj) + o->offset;
    if(o_out) *o_out= o;
247

248
    switch(o->type){
249
    case FF_OPT_TYPE_FLAGS:     snprintf(buf, buf_len, "0x%08X",*(int    *)dst);break;
250
    case FF_OPT_TYPE_INT:       snprintf(buf, buf_len, "%d" , *(int    *)dst);break;
251
    case FF_OPT_TYPE_INT64:     snprintf(buf, buf_len, "%"PRId64, *(int64_t*)dst);break;
252 253 254
    case FF_OPT_TYPE_FLOAT:     snprintf(buf, buf_len, "%f" , *(float  *)dst);break;
    case FF_OPT_TYPE_DOUBLE:    snprintf(buf, buf_len, "%f" , *(double *)dst);break;
    case FF_OPT_TYPE_RATIONAL:  snprintf(buf, buf_len, "%d/%d", ((AVRational*)dst)->num, ((AVRational*)dst)->den);break;
Michael Niedermayer's avatar
Michael Niedermayer committed
255
    case FF_OPT_TYPE_STRING:    return *(void**)dst;
256 257 258 259 260 261
    case FF_OPT_TYPE_BINARY:
        len = *(int*)(((uint8_t *)dst) + sizeof(uint8_t *));
        if(len >= (buf_len + 1)/2) return NULL;
        bin = *(uint8_t**)dst;
        for(i = 0; i < len; i++) snprintf(buf + i*2, 3, "%02X", bin[i]);
        break;
262 263 264
    default: return NULL;
    }
    return buf;
Michael Niedermayer's avatar
Michael Niedermayer committed
265 266
}

267
static int av_get_number(void *obj, const char *name, const AVOption **o_out, double *num, int *den, int64_t *intnum){
268
    const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);
Michael Niedermayer's avatar
Michael Niedermayer committed
269 270
    void *dst;
    if(!o || o->offset<=0)
271
        goto error;
Michael Niedermayer's avatar
Michael Niedermayer committed
272 273 274

    dst= ((uint8_t*)obj) + o->offset;

275 276
    if(o_out) *o_out= o;

Michael Niedermayer's avatar
Michael Niedermayer committed
277
    switch(o->type){
278
    case FF_OPT_TYPE_FLAGS:     *intnum= *(unsigned int*)dst;return 0;
279 280 281 282
    case FF_OPT_TYPE_INT:       *intnum= *(int    *)dst;return 0;
    case FF_OPT_TYPE_INT64:     *intnum= *(int64_t*)dst;return 0;
    case FF_OPT_TYPE_FLOAT:     *num=    *(float  *)dst;return 0;
    case FF_OPT_TYPE_DOUBLE:    *num=    *(double *)dst;return 0;
283
    case FF_OPT_TYPE_RATIONAL:  *intnum= ((AVRational*)dst)->num;
284 285
                                *den   = ((AVRational*)dst)->den;
                                                        return 0;
Michael Niedermayer's avatar
Michael Niedermayer committed
286
    }
287 288 289 290 291
error:
    *den=*intnum=0;
    return -1;
}

292
double av_get_double(void *obj, const char *name, const AVOption **o_out){
293 294 295 296 297 298 299 300
    int64_t intnum=1;
    double num=1;
    int den=1;

    av_get_number(obj, name, o_out, &num, &den, &intnum);
    return num*intnum/den;
}

301
AVRational av_get_q(void *obj, const char *name, const AVOption **o_out){
302 303 304 305 306 307 308 309 310 311 312
    int64_t intnum=1;
    double num=1;
    int den=1;

    av_get_number(obj, name, o_out, &num, &den, &intnum);
    if(num == 1.0 && (int)intnum == intnum)
        return (AVRational){intnum, den};
    else
        return av_d2q(num*intnum/den, 1<<24);
}

313
int64_t av_get_int(void *obj, const char *name, const AVOption **o_out){
314 315 316 317 318 319 320 321
    int64_t intnum=1;
    double num=1;
    int den=1;

    av_get_number(obj, name, o_out, &num, &den, &intnum);
    return num*intnum/den;
}

322
static void opt_list(void *obj, void *av_log_obj, const char *unit)
323
{
324
    const AVOption *opt=NULL;
325

326 327 328
    while((opt= av_next_option(obj, opt))){
        if(!(opt->flags & (AV_OPT_FLAG_ENCODING_PARAM|AV_OPT_FLAG_DECODING_PARAM)))
            continue;
329

330 331 332 333 334 335 336 337 338 339 340 341 342 343
        /* Don't print CONST's on level one.
         * Don't print anything but CONST's on level two.
         * Only print items from the requested unit.
         */
        if (!unit && opt->type==FF_OPT_TYPE_CONST)
            continue;
        else if (unit && opt->type!=FF_OPT_TYPE_CONST)
            continue;
        else if (unit && opt->type==FF_OPT_TYPE_CONST && strcmp(unit, opt->unit))
            continue;
        else if (unit && opt->type == FF_OPT_TYPE_CONST)
            av_log(av_log_obj, AV_LOG_INFO, "   %-15s ", opt->name);
        else
            av_log(av_log_obj, AV_LOG_INFO, "-%-17s ", opt->name);
344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367

        switch( opt->type )
        {
            case FF_OPT_TYPE_FLAGS:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<flags>" );
                break;
            case FF_OPT_TYPE_INT:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<int>" );
                break;
            case FF_OPT_TYPE_INT64:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<int64>" );
                break;
            case FF_OPT_TYPE_DOUBLE:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<double>" );
                break;
            case FF_OPT_TYPE_FLOAT:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<float>" );
                break;
            case FF_OPT_TYPE_STRING:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<string>" );
                break;
            case FF_OPT_TYPE_RATIONAL:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<rational>" );
                break;
368 369 370
            case FF_OPT_TYPE_BINARY:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<binary>" );
                break;
371 372 373 374 375
            case FF_OPT_TYPE_CONST:
            default:
                av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "" );
                break;
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
376 377 378 379 380
        av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_ENCODING_PARAM) ? 'E' : '.');
        av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_DECODING_PARAM) ? 'D' : '.');
        av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_VIDEO_PARAM   ) ? 'V' : '.');
        av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_AUDIO_PARAM   ) ? 'A' : '.');
        av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_SUBTITLE_PARAM) ? 'S' : '.');
381

Michael Niedermayer's avatar
Michael Niedermayer committed
382 383 384
        if(opt->help)
            av_log(av_log_obj, AV_LOG_INFO, " %s", opt->help);
        av_log(av_log_obj, AV_LOG_INFO, "\n");
385 386 387
        if (opt->unit && opt->type != FF_OPT_TYPE_CONST) {
            opt_list(obj, av_log_obj, opt->unit);
        }
388
    }
389 390 391 392 393 394 395 396 397 398
}

int av_opt_show(void *obj, void *av_log_obj){
    if(!obj)
        return -1;

    av_log(av_log_obj, AV_LOG_INFO, "%s AVOptions:\n", (*(AVClass**)obj)->class_name);

    opt_list(obj, av_log_obj, NULL);

399
    return 0;
Michael Niedermayer's avatar
Michael Niedermayer committed
400
}
401

402 403 404 405 406 407
/** Set the values of the AVCodecContext or AVFormatContext structure.
 * They are set to the defaults specified in the according AVOption options
 * array default_val field.
 *
 * @param s AVCodecContext or AVFormatContext for which the defaults will be set
 */
408
void av_opt_set_defaults2(void *s, int mask, int flags)
409
{
410
    const AVOption *opt = NULL;
411
    while ((opt = av_next_option(s, opt)) != NULL) {
412 413
        if((opt->flags & mask) != flags)
            continue;
414 415 416 417 418 419 420 421 422 423 424
        switch(opt->type) {
            case FF_OPT_TYPE_CONST:
                /* Nothing to be done here */
            break;
            case FF_OPT_TYPE_FLAGS:
            case FF_OPT_TYPE_INT: {
                int val;
                val = opt->default_val;
                av_set_int(s, opt->name, val);
            }
            break;
425 426
            case FF_OPT_TYPE_INT64:
                if((double)(opt->default_val+0.6) == opt->default_val)
427
                    av_log(s, AV_LOG_DEBUG, "loss of precision in default of %s\n", opt->name);
428 429
                av_set_int(s, opt->name, opt->default_val);
            break;
430 431 432 433 434 435 436 437 438 439 440 441 442
            case FF_OPT_TYPE_FLOAT: {
                double val;
                val = opt->default_val;
                av_set_double(s, opt->name, val);
            }
            break;
            case FF_OPT_TYPE_RATIONAL: {
                AVRational val;
                val = av_d2q(opt->default_val, INT_MAX);
                av_set_q(s, opt->name, val);
            }
            break;
            case FF_OPT_TYPE_STRING:
443
            case FF_OPT_TYPE_BINARY:
444 445 446 447 448 449 450 451
                /* Cannot set default for string as default_val is of type * double */
            break;
            default:
                av_log(s, AV_LOG_DEBUG, "AVOption type %d of option %s not implemented yet\n", opt->type, opt->name);
        }
    }
}

452 453 454 455
void av_opt_set_defaults(void *s){
    av_opt_set_defaults2(s, 0, 0);
}