libavutil/opt.c

   1 /*
   2  * AVOptions
   3  * Copyright (c) 2005 Michael Niedermayer <michaelni@gmx.at>
   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 /**
  23  * @file
  24  * AVOptions
  25  * @author Michael Niedermayer <michaelni@gmx.at>
  26  */
  27
  28 #include "avutil.h"
  29 #include "avstring.h"
  30 #include "opt.h"
  31 #include "eval.h"
  32 #include "dict.h"
  33 #include "log.h"
  34
  35 #if FF_API_FIND_OPT
  36 //FIXME order them and do a bin search
  37 const AVOption *av_find_opt(void *v, const char *name, const char *unit, int mask, int flags)
  38 {
  39     const AVOption *o = NULL;
  40
  41     while ((o = av_next_option(v, o))) {
  42         if (!strcmp(o->name, name) && (!unit || (o->unit && !strcmp(o->unit, unit))) && (o->flags & mask) == flags)
  43             return o;
  44     }
  45     return NULL;
  46 }
  47 #endif
  48
  49 const AVOption *av_next_option(void *obj, const AVOption *last)
  50 {
  51     if (last && last[1].name) return ++last;
  52     else if (last || !(*(AVClass**)obj)->option->name) return NULL;
  53     else                      return (*(AVClass**)obj)->option;
  54 }
  55
  56 static int av_set_number2(void *obj, const char *name, double num, int den, int64_t intnum, const AVOption **o_out)
  57 {
  58     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
  59     void *dst;
  60     if (o_out)
  61         *o_out= o;
  62     if (!o)
  63         return AVERROR_OPTION_NOT_FOUND;
  64
  65     if (o->max*den < num*intnum || o->min*den > num*intnum) {
  66         av_log(obj, AV_LOG_ERROR, "Value %lf for parameter '%s' out of range\n", num, name);
  67         return AVERROR(ERANGE);
  68     }
  69
  70     dst= ((uint8_t*)obj) + o->offset;
  71
  72     switch (o->type) {
  73     case FF_OPT_TYPE_FLAGS:
  74     case FF_OPT_TYPE_INT:   *(int       *)dst= llrint(num/den)*intnum; break;
  75     case FF_OPT_TYPE_INT64: *(int64_t   *)dst= llrint(num/den)*intnum; break;
  76     case FF_OPT_TYPE_FLOAT: *(float     *)dst= num*intnum/den;         break;
  77     case FF_OPT_TYPE_DOUBLE:*(double    *)dst= num*intnum/den;         break;
  78     case FF_OPT_TYPE_RATIONAL:
  79         if ((int)num == num) *(AVRational*)dst= (AVRational){num*intnum, den};
  80         else                 *(AVRational*)dst= av_d2q(num*intnum/den, 1<<24);
  81         break;
  82     default:
  83         return AVERROR(EINVAL);
  84     }
  85     return 0;
  86 }
  87
  88 static const AVOption *av_set_number(void *obj, const char *name, double num, int den, int64_t intnum)
  89 {
  90     const AVOption *o = NULL;
  91     if (av_set_number2(obj, name, num, den, intnum, &o) < 0)
  92         return NULL;
  93     else
  94         return o;
  95 }
  96
  97 static const double const_values[] = {
  98     M_PI,
  99     M_E,
 100     FF_QP2LAMBDA,
 101     0
 102 };
 103
 104 static const char * const const_names[] = {
 105     "PI",
 106     "E",
 107     "QP2LAMBDA",
 108     0
 109 };
 110
 111 static int hexchar2int(char c) {
 112     if (c >= '0' && c <= '9') return c - '0';
 113     if (c >= 'a' && c <= 'f') return c - 'a' + 10;
 114     if (c >= 'A' && c <= 'F') return c - 'A' + 10;
 115     return -1;
 116 }
 117
 118 int av_set_string3(void *obj, const char *name, const char *val, int alloc, const AVOption **o_out)
 119 {
 120     int ret;
 121     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
 122     if (o_out)
 123         *o_out = o;
 124     if (!o)
 125         return AVERROR_OPTION_NOT_FOUND;
 126     if (!val && o->type != FF_OPT_TYPE_STRING)
 127         return AVERROR(EINVAL);
 128
 129     if (o->type == FF_OPT_TYPE_BINARY) {
 130         uint8_t **dst = (uint8_t **)(((uint8_t*)obj) + o->offset);
 131         int *lendst = (int *)(dst + 1);
 132         uint8_t *bin, *ptr;
 133         int len = strlen(val);
 134         av_freep(dst);
 135         *lendst = 0;
 136         if (len & 1) return AVERROR(EINVAL);
 137         len /= 2;
 138         ptr = bin = av_malloc(len);
 139         while (*val) {
 140             int a = hexchar2int(*val++);
 141             int b = hexchar2int(*val++);
 142             if (a < 0 || b < 0) {
 143                 av_free(bin);
 144                 return AVERROR(EINVAL);
 145             }
 146             *ptr++ = (a << 4) | b;
 147         }
 148         *dst = bin;
 149         *lendst = len;
 150         return 0;
 151     }
 152     if (o->type != FF_OPT_TYPE_STRING) {
 153         int notfirst=0;
 154         for (;;) {
 155             int i;
 156             char buf[256];
 157             int cmd=0;
 158             double d;
 159
 160             if (*val == '+' || *val == '-')
 161                 cmd= *(val++);
 162
 163             for (i=0; i<sizeof(buf)-1 && val[i] && val[i]!='+' && val[i]!='-'; i++)
 164                 buf[i]= val[i];
 165             buf[i]=0;
 166
 167             {
 168                 const AVOption *o_named = av_opt_find(obj, buf, o->unit, 0, 0);
 169                 if (o_named && o_named->type == FF_OPT_TYPE_CONST)
 170                     d= o_named->default_val.dbl;
 171                 else if (!strcmp(buf, "default")) d= o->default_val.dbl;
 172                 else if (!strcmp(buf, "max"    )) d= o->max;
 173                 else if (!strcmp(buf, "min"    )) d= o->min;
 174                 else if (!strcmp(buf, "none"   )) d= 0;
 175                 else if (!strcmp(buf, "all"    )) d= ~0;
 176                 else {
 177                     int res = av_expr_parse_and_eval(&d, buf, const_names, const_values, NULL, NULL, NULL, NULL, NULL, 0, obj);
 178                     if (res < 0) {
 179                         av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\"\n", val);
 180                         return res;
 181                     }
 182                 }
 183             }
 184             if (o->type == FF_OPT_TYPE_FLAGS) {
 185                 if      (cmd=='+') d= av_get_int(obj, name, NULL) | (int64_t)d;
 186                 else if (cmd=='-') d= av_get_int(obj, name, NULL) &~(int64_t)d;
 187             } else {
 188                 if      (cmd=='+') d= notfirst*av_get_double(obj, name, NULL) + d;
 189                 else if (cmd=='-') d= notfirst*av_get_double(obj, name, NULL) - d;
 190             }
 191
 192             if ((ret = av_set_number2(obj, name, d, 1, 1, o_out)) < 0)
 193                 return ret;
 194             val+= i;
 195             if (!*val)
 196                 return 0;
 197             notfirst=1;
 198         }
 199     }
 200
 201     if (alloc) {
 202         av_free(*(void**)(((uint8_t*)obj) + o->offset));
 203         val= av_strdup(val);
 204     }
 205
 206     memcpy(((uint8_t*)obj) + o->offset, &val, sizeof(val));
 207     return 0;
 208 }
 209
 210 const AVOption *av_set_double(void *obj, const char *name, double n)
 211 {
 212     return av_set_number(obj, name, n, 1, 1);
 213 }
 214
 215 const AVOption *av_set_q(void *obj, const char *name, AVRational n)
 216 {
 217     return av_set_number(obj, name, n.num, n.den, 1);
 218 }
 219
 220 const AVOption *av_set_int(void *obj, const char *name, int64_t n)
 221 {
 222     return av_set_number(obj, name, 1, 1, n);
 223 }
 224
 225 /**
 226  *
 227  * @param buf a buffer which is used for returning non string values as strings, can be NULL
 228  * @param buf_len allocated length in bytes of buf
 229  */
 230 const char *av_get_string(void *obj, const char *name, const AVOption **o_out, char *buf, int buf_len)
 231 {
 232     const AVOption *o = av_opt_find(obj, name, NULL, 0, AV_OPT_SEARCH_CHILDREN);
 233     void *dst;
 234     uint8_t *bin;
 235     int len, i;
 236     if (!o)
 237         return NULL;
 238     if (o->type != FF_OPT_TYPE_STRING && (!buf || !buf_len))
 239         return NULL;
 240
 241     dst= ((uint8_t*)obj) + o->offset;
 242     if (o_out) *o_out= o;
 243
 244     switch (o->type) {
 245     case FF_OPT_TYPE_FLAGS:     snprintf(buf, buf_len, "0x%08X",*(int    *)dst);break;
 246     case FF_OPT_TYPE_INT:       snprintf(buf, buf_len, "%d" , *(int    *)dst);break;
 247     case FF_OPT_TYPE_INT64:     snprintf(buf, buf_len, "%"PRId64, *(int64_t*)dst);break;
 248     case FF_OPT_TYPE_FLOAT:     snprintf(buf, buf_len, "%f" , *(float  *)dst);break;
 249     case FF_OPT_TYPE_DOUBLE:    snprintf(buf, buf_len, "%f" , *(double *)dst);break;
 250     case FF_OPT_TYPE_RATIONAL:  snprintf(buf, buf_len, "%d/%d", ((AVRational*)dst)->num, ((AVRational*)dst)->den);break;
 251     case FF_OPT_TYPE_STRING:    return *(void**)dst;
 252     case FF_OPT_TYPE_BINARY:
 253         len = *(int*)(((uint8_t *)dst) + sizeof(uint8_t *));
 254         if (len >= (buf_len + 1)/2) return NULL;
 255         bin = *(uint8_t**)dst;
 256         for (i = 0; i < len; i++) snprintf(buf + i*2, 3, "%02X", bin[i]);
 257         break;
 258     default: return NULL;
 259     }
 260     return buf;
 261 }
 262
 263 static int get_number(void *obj, const char *name, const AVOption **o_out, double *num, int *den, int64_t *intnum)
 264 {
 265     const AVOption *o = av_opt_find(obj, name, NULL, 0, AV_OPT_SEARCH_CHILDREN);
 266     void *dst;
 267     if (!o || (o->offset<=0 && o->type != FF_OPT_TYPE_CONST))
 268         goto error;
 269
 270     dst= ((uint8_t*)obj) + o->offset;
 271
 272     if (o_out) *o_out= o;
 273
 274     switch (o->type) {
 275     case FF_OPT_TYPE_FLAGS:     *intnum= *(unsigned int*)dst;return 0;
 276     case FF_OPT_TYPE_INT:       *intnum= *(int    *)dst;return 0;
 277     case FF_OPT_TYPE_INT64:     *intnum= *(int64_t*)dst;return 0;
 278     case FF_OPT_TYPE_FLOAT:     *num=    *(float  *)dst;return 0;
 279     case FF_OPT_TYPE_DOUBLE:    *num=    *(double *)dst;return 0;
 280     case FF_OPT_TYPE_RATIONAL:  *intnum= ((AVRational*)dst)->num;
 281                                 *den   = ((AVRational*)dst)->den;
 282                                                         return 0;
 283     case FF_OPT_TYPE_CONST:     *intnum= o->default_val.dbl;return 0;
 284     }
 285 error:
 286     *den=*intnum=0;
 287     return -1;
 288 }
 289
 290 double av_get_double(void *obj, const char *name, const AVOption **o_out)
 291 {
 292     int64_t intnum=1;
 293     double num=1;
 294     int den=1;
 295
 296     if (get_number(obj, name, o_out, &num, &den, &intnum) < 0)
 297         return NAN;
 298     return num*intnum/den;
 299 }
 300
 301 AVRational av_get_q(void *obj, const char *name, const AVOption **o_out)
 302 {
 303     int64_t intnum=1;
 304     double num=1;
 305     int den=1;
 306
 307     if (get_number(obj, name, o_out, &num, &den, &intnum) < 0)
 308         return (AVRational){0, 0};
 309     if (num == 1.0 && (int)intnum == intnum)
 310         return (AVRational){intnum, den};
 311     else
 312         return av_d2q(num*intnum/den, 1<<24);
 313 }
 314
 315 int64_t av_get_int(void *obj, const char *name, const AVOption **o_out)
 316 {
 317     int64_t intnum=1;
 318     double num=1;
 319     int den=1;
 320
 321     if (get_number(obj, name, o_out, &num, &den, &intnum) < 0)
 322         return -1;
 323     return num*intnum/den;
 324 }
 325
 326 int av_opt_flag_is_set(void *obj, const char *field_name, const char *flag_name)
 327 {
 328     const AVOption *field = av_find_opt(obj, field_name, NULL, 0, 0);
 329     const AVOption *flag  = av_find_opt(obj, flag_name,  NULL, 0, 0);
 330
 331     if (!field || !flag || flag->type != FF_OPT_TYPE_CONST)
 332         return 0;
 333     return av_get_int(obj, field_name, NULL) & (int) flag->default_val.dbl;
 334 }
 335
 336 static void opt_list(void *obj, void *av_log_obj, const char *unit,
 337                      int req_flags, int rej_flags)
 338 {
 339     const AVOption *opt=NULL;
 340
 341     while ((opt= av_next_option(obj, opt))) {
 342         if (!(opt->flags & req_flags) || (opt->flags & rej_flags))
 343             continue;
 344
 345         /* Don't print CONST's on level one.
 346          * Don't print anything but CONST's on level two.
 347          * Only print items from the requested unit.
 348          */
 349         if (!unit && opt->type==FF_OPT_TYPE_CONST)
 350             continue;
 351         else if (unit && opt->type!=FF_OPT_TYPE_CONST)
 352             continue;
 353         else if (unit && opt->type==FF_OPT_TYPE_CONST && strcmp(unit, opt->unit))
 354             continue;
 355         else if (unit && opt->type == FF_OPT_TYPE_CONST)
 356             av_log(av_log_obj, AV_LOG_INFO, "   %-15s ", opt->name);
 357         else
 358             av_log(av_log_obj, AV_LOG_INFO, "-%-17s ", opt->name);
 359
 360         switch (opt->type) {
 361             case FF_OPT_TYPE_FLAGS:
 362                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<flags>");
 363                 break;
 364             case FF_OPT_TYPE_INT:
 365                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<int>");
 366                 break;
 367             case FF_OPT_TYPE_INT64:
 368                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<int64>");
 369                 break;
 370             case FF_OPT_TYPE_DOUBLE:
 371                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<double>");
 372                 break;
 373             case FF_OPT_TYPE_FLOAT:
 374                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<float>");
 375                 break;
 376             case FF_OPT_TYPE_STRING:
 377                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<string>");
 378                 break;
 379             case FF_OPT_TYPE_RATIONAL:
 380                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<rational>");
 381                 break;
 382             case FF_OPT_TYPE_BINARY:
 383                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<binary>");
 384                 break;
 385             case FF_OPT_TYPE_CONST:
 386             default:
 387                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "");
 388                 break;
 389         }
 390         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_ENCODING_PARAM) ? 'E' : '.');
 391         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_DECODING_PARAM) ? 'D' : '.');
 392         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_VIDEO_PARAM   ) ? 'V' : '.');
 393         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_AUDIO_PARAM   ) ? 'A' : '.');
 394         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_SUBTITLE_PARAM) ? 'S' : '.');
 395
 396         if (opt->help)
 397             av_log(av_log_obj, AV_LOG_INFO, " %s", opt->help);
 398         av_log(av_log_obj, AV_LOG_INFO, "\n");
 399         if (opt->unit && opt->type != FF_OPT_TYPE_CONST) {
 400             opt_list(obj, av_log_obj, opt->unit, req_flags, rej_flags);
 401         }
 402     }
 403 }
 404
 405 int av_opt_show2(void *obj, void *av_log_obj, int req_flags, int rej_flags)
 406 {
 407     if (!obj)
 408         return -1;
 409
 410     av_log(av_log_obj, AV_LOG_INFO, "%s AVOptions:\n", (*(AVClass**)obj)->class_name);
 411
 412     opt_list(obj, av_log_obj, NULL, req_flags, rej_flags);
 413
 414     return 0;
 415 }
 416
 417 void av_opt_set_defaults(void *s)
 418 {
 419 #if FF_API_OLD_AVOPTIONS
 420     av_opt_set_defaults2(s, 0, 0);
 421 }
 422
 423 void av_opt_set_defaults2(void *s, int mask, int flags)
 424 {
 425 #endif
 426     const AVOption *opt = NULL;
 427     while ((opt = av_next_option(s, opt)) != NULL) {
 428 #if FF_API_OLD_AVOPTIONS
 429         if ((opt->flags & mask) != flags)
 430             continue;
 431 #endif
 432         switch (opt->type) {
 433             case FF_OPT_TYPE_CONST:
 434                 /* Nothing to be done here */
 435             break;
 436             case FF_OPT_TYPE_FLAGS:
 437             case FF_OPT_TYPE_INT: {
 438                 int val;
 439                 val = opt->default_val.dbl;
 440                 av_set_int(s, opt->name, val);
 441             }
 442             break;
 443             case FF_OPT_TYPE_INT64:
 444                 if ((double)(opt->default_val.dbl+0.6) == opt->default_val.dbl)
 445                     av_log(s, AV_LOG_DEBUG, "loss of precision in default of %s\n", opt->name);
 446                 av_set_int(s, opt->name, opt->default_val.dbl);
 447             break;
 448             case FF_OPT_TYPE_DOUBLE:
 449             case FF_OPT_TYPE_FLOAT: {
 450                 double val;
 451                 val = opt->default_val.dbl;
 452                 av_set_double(s, opt->name, val);
 453             }
 454             break;
 455             case FF_OPT_TYPE_RATIONAL: {
 456                 AVRational val;
 457                 val = av_d2q(opt->default_val.dbl, INT_MAX);
 458                 av_set_q(s, opt->name, val);
 459             }
 460             break;
 461             case FF_OPT_TYPE_STRING:
 462                 av_set_string3(s, opt->name, opt->default_val.str, 1, NULL);
 463                 break;
 464             case FF_OPT_TYPE_BINARY:
 465                 /* Cannot set default for binary */
 466             break;
 467             default:
 468                 av_log(s, AV_LOG_DEBUG, "AVOption type %d of option %s not implemented yet\n", opt->type, opt->name);
 469         }
 470     }
 471 }
 472
 473 /**
 474  * Store the value in the field in ctx that is named like key.
 475  * ctx must be an AVClass context, storing is done using AVOptions.
 476  *
 477  * @param buf the string to parse, buf will be updated to point at the
 478  * separator just after the parsed key/value pair
 479  * @param key_val_sep a 0-terminated list of characters used to
 480  * separate key from value
 481  * @param pairs_sep a 0-terminated list of characters used to separate
 482  * two pairs from each other
 483  * @return 0 if the key/value pair has been successfully parsed and
 484  * set, or a negative value corresponding to an AVERROR code in case
 485  * of error:
 486  * AVERROR(EINVAL) if the key/value pair cannot be parsed,
 487  * the error code issued by av_set_string3() if the key/value pair
 488  * cannot be set
 489  */
 490 static int parse_key_value_pair(void *ctx, const char **buf,
 491                                 const char *key_val_sep, const char *pairs_sep)
 492 {
 493     char *key = av_get_token(buf, key_val_sep);
 494     char *val;
 495     int ret;
 496
 497     if (*key && strspn(*buf, key_val_sep)) {
 498         (*buf)++;
 499         val = av_get_token(buf, pairs_sep);
 500     } else {
 501         av_log(ctx, AV_LOG_ERROR, "Missing key or no key/value separator found after key '%s'\n", key);
 502         av_free(key);
 503         return AVERROR(EINVAL);
 504     }
 505
 506     av_log(ctx, AV_LOG_DEBUG, "Setting value '%s' for key '%s'\n", val, key);
 507
 508     ret = av_set_string3(ctx, key, val, 1, NULL);
 509     if (ret == AVERROR_OPTION_NOT_FOUND)
 510         av_log(ctx, AV_LOG_ERROR, "Key '%s' not found.\n", key);
 511
 512     av_free(key);
 513     av_free(val);
 514     return ret;
 515 }
 516
 517 int av_set_options_string(void *ctx, const char *opts,
 518                           const char *key_val_sep, const char *pairs_sep)
 519 {
 520     int ret, count = 0;
 521
 522     if (!opts)
 523         return 0;
 524     while (*opts) {
 525         if ((ret = parse_key_value_pair(ctx, &opts, key_val_sep, pairs_sep)) < 0)
 526             return ret;
 527         count++;
 528
 529         if (*opts)
 530             opts++;
 531     }
 532
 533     return count;
 534 }
 535
 536 void av_opt_free(void *obj)
 537 {
 538     const AVOption *o = NULL;
 539     while ((o = av_next_option(obj, o)))
 540         if (o->type == FF_OPT_TYPE_STRING || o->type == FF_OPT_TYPE_BINARY)
 541             av_freep((uint8_t *)obj + o->offset);
 542 }
 543
 544 int av_opt_set_dict(void *obj, AVDictionary **options)
 545 {
 546     AVDictionaryEntry *t = NULL;
 547     AVDictionary    *tmp = NULL;
 548     int ret = 0;
 549
 550     while ((t = av_dict_get(*options, "", t, AV_DICT_IGNORE_SUFFIX))) {
 551         ret = av_set_string3(obj, t->key, t->value, 1, NULL);
 552         if (ret == AVERROR_OPTION_NOT_FOUND)
 553             av_dict_set(&tmp, t->key, t->value, 0);
 554         else if (ret < 0) {
 555             av_log(obj, AV_LOG_ERROR, "Error setting option %s to value %s.\n", t->key, t->value);
 556             break;
 557         }
 558         ret = 0;
 559     }
 560     av_dict_free(options);
 561     *options = tmp;
 562     return ret;
 563 }
 564
 565 const AVOption *av_opt_find(void *obj, const char *name, const char *unit,
 566                             int opt_flags, int search_flags)
 567 {
 568     AVClass *c = *(AVClass**)obj;
 569     const AVOption *o = NULL;
 570
 571     if (c->opt_find && search_flags & AV_OPT_SEARCH_CHILDREN &&
 572         (o = c->opt_find(obj, name, unit, opt_flags, search_flags)))
 573         return o;
 574
 575     while (o = av_next_option(obj, o)) {
 576         if (!strcmp(o->name, name) && (o->flags & opt_flags) == opt_flags &&
 577             ((!unit && o->type != FF_OPT_TYPE_CONST) ||
 578              (unit  && o->unit && !strcmp(o->unit, unit))))
 579             return o;
 580     }
 581     return NULL;
 582 }
 583
 584 #ifdef TEST
 585
 586 #undef printf
 587
 588 typedef struct TestContext
 589 {
 590     const AVClass *class;
 591     int num;
 592     int toggle;
 593     char *string;
 594     int flags;
 595     AVRational rational;
 596 } TestContext;
 597
 598 #define OFFSET(x) offsetof(TestContext, x)
 599
 600 #define TEST_FLAG_COOL 01
 601 #define TEST_FLAG_LAME 02
 602 #define TEST_FLAG_MU   04
 603
 604 static const AVOption test_options[]= {
 605 {"num",      "set num",        OFFSET(num),      FF_OPT_TYPE_INT,      {0},              0,        100                 },
 606 {"toggle",   "set toggle",     OFFSET(toggle),   FF_OPT_TYPE_INT,      {0},              0,        1                   },
 607 {"rational", "set rational",   OFFSET(rational), FF_OPT_TYPE_RATIONAL, {0},              0,        10                  },
 608 {"string",   "set string",     OFFSET(string),   FF_OPT_TYPE_STRING,   {0},              CHAR_MIN, CHAR_MAX            },
 609 {"flags",    "set flags",      OFFSET(flags),    FF_OPT_TYPE_FLAGS,    {0},              0,        INT_MAX, 0, "flags" },
 610 {"cool",     "set cool flag ", 0,                FF_OPT_TYPE_CONST,    {TEST_FLAG_COOL}, INT_MIN,  INT_MAX, 0, "flags" },
 611 {"lame",     "set lame flag ", 0,                FF_OPT_TYPE_CONST,    {TEST_FLAG_LAME}, INT_MIN,  INT_MAX, 0, "flags" },
 612 {"mu",       "set mu flag ",   0,                FF_OPT_TYPE_CONST,    {TEST_FLAG_MU},   INT_MIN,  INT_MAX, 0, "flags" },
 613 {NULL},
 614 };
 615
 616 static const char *test_get_name(void *ctx)
 617 {
 618     return "test";
 619 }
 620
 621 static const AVClass test_class = {
 622     "TestContext",
 623     test_get_name,
 624     test_options
 625 };
 626
 627 int main(void)
 628 {
 629     int i;
 630
 631     printf("\nTesting av_set_options_string()\n");
 632     {
 633         TestContext test_ctx;
 634         const char *options[] = {
 635             "",
 636             ":",
 637             "=",
 638             "foo=:",
 639             ":=foo",
 640             "=foo",
 641             "foo=",
 642             "foo",
 643             "foo=val",
 644             "foo==val",
 645             "toggle=:",
 646             "string=:",
 647             "toggle=1 : foo",
 648             "toggle=100",
 649             "toggle==1",
 650             "flags=+mu-lame : num=42: toggle=0",
 651             "num=42 : string=blahblah",
 652             "rational=0 : rational=1/2 : rational=1/-1",
 653             "rational=-1/0",
 654         };
 655
 656         test_ctx.class = &test_class;
 657         av_opt_set_defaults(&test_ctx);
 658         test_ctx.string = av_strdup("default");
 659
 660         av_log_set_level(AV_LOG_DEBUG);
 661
 662         for (i=0; i < FF_ARRAY_ELEMS(options); i++) {
 663             av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]);
 664             if (av_set_options_string(&test_ctx, options[i], "=", ":") < 0)
 665                 av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]);
 666             printf("\n");
 667         }
 668     }
 669
 670     return 0;
 671 }
 672
 673 #endif