avoptions: Fix av_opt_flag_is_set
[ffmpeg.git] / libavutil / opt.c
1 /*
2  * AVOptions
3  * Copyright (c) 2005 Michael Niedermayer <michaelni@gmx.at>
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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  * Libav 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 Libav; 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     AVClass *c= *(AVClass**)v; //FIXME silly way of storing AVClass
40     const AVOption *o= c->option;
41
42     for (; o && o->name; o++) {
43         if (!strcmp(o->name, name) && (!unit || (o->unit && !strcmp(o->unit, unit))) && (o->flags & mask) == flags)
44             return o;
45     }
46     return NULL;
47 }
48 #endif
49
50 #if FF_API_OLD_AVOPTIONS
51 const AVOption *av_next_option(void *obj, const AVOption *last)
52 {
53     return av_opt_next(obj, last);
54 }
55 #endif
56
57 const AVOption *av_opt_next(void *obj, const AVOption *last)
58 {
59     if (last && last[1].name) return ++last;
60     else if (last)            return NULL;
61     else                      return (*(AVClass**)obj)->option;
62 }
63
64 static int read_number(const AVOption *o, void *dst, double *num, int *den, int64_t *intnum)
65 {
66     switch (o->type) {
67     case AV_OPT_TYPE_FLAGS:     *intnum = *(unsigned int*)dst;return 0;
68     case AV_OPT_TYPE_INT:       *intnum = *(int         *)dst;return 0;
69     case AV_OPT_TYPE_INT64:     *intnum = *(int64_t     *)dst;return 0;
70     case AV_OPT_TYPE_FLOAT:     *num    = *(float       *)dst;return 0;
71     case AV_OPT_TYPE_DOUBLE:    *num    = *(double      *)dst;return 0;
72     case AV_OPT_TYPE_RATIONAL:  *intnum = ((AVRational*)dst)->num;
73                                 *den    = ((AVRational*)dst)->den;
74                                                         return 0;
75     }
76     return AVERROR(EINVAL);
77 }
78
79 static int write_number(void *obj, const AVOption *o, void *dst, double num, int den, int64_t intnum)
80 {
81     if (o->max*den < num*intnum || o->min*den > num*intnum) {
82         av_log(obj, AV_LOG_ERROR, "Value %lf for parameter '%s' out of range\n", num, o->name);
83         return AVERROR(ERANGE);
84     }
85
86     switch (o->type) {
87     case AV_OPT_TYPE_FLAGS:
88     case AV_OPT_TYPE_INT:   *(int       *)dst= llrint(num/den)*intnum; break;
89     case AV_OPT_TYPE_INT64: *(int64_t   *)dst= llrint(num/den)*intnum; break;
90     case AV_OPT_TYPE_FLOAT: *(float     *)dst= num*intnum/den;         break;
91     case AV_OPT_TYPE_DOUBLE:*(double    *)dst= num*intnum/den;         break;
92     case AV_OPT_TYPE_RATIONAL:
93         if ((int)num == num) *(AVRational*)dst= (AVRational){num*intnum, den};
94         else                 *(AVRational*)dst= av_d2q(num*intnum/den, 1<<24);
95         break;
96     default:
97         return AVERROR(EINVAL);
98     }
99     return 0;
100 }
101
102 static const double const_values[] = {
103     M_PI,
104     M_E,
105     FF_QP2LAMBDA,
106     0
107 };
108
109 static const char * const const_names[] = {
110     "PI",
111     "E",
112     "QP2LAMBDA",
113     0
114 };
115
116 static int hexchar2int(char c) {
117     if (c >= '0' && c <= '9') return c - '0';
118     if (c >= 'a' && c <= 'f') return c - 'a' + 10;
119     if (c >= 'A' && c <= 'F') return c - 'A' + 10;
120     return -1;
121 }
122
123 static int set_string_binary(void *obj, const AVOption *o, const char *val, uint8_t **dst)
124 {
125     int *lendst = (int *)(dst + 1);
126     uint8_t *bin, *ptr;
127     int len = strlen(val);
128
129     av_freep(dst);
130     *lendst = 0;
131
132     if (len & 1)
133         return AVERROR(EINVAL);
134     len /= 2;
135
136     ptr = bin = av_malloc(len);
137     while (*val) {
138         int a = hexchar2int(*val++);
139         int b = hexchar2int(*val++);
140         if (a < 0 || b < 0) {
141             av_free(bin);
142             return AVERROR(EINVAL);
143         }
144         *ptr++ = (a << 4) | b;
145     }
146     *dst = bin;
147     *lendst = len;
148
149     return 0;
150 }
151
152 static int set_string(void *obj, const AVOption *o, const char *val, uint8_t **dst)
153 {
154     av_freep(dst);
155     *dst = av_strdup(val);
156     return 0;
157 }
158
159 static int set_string_number(void *obj, const AVOption *o, const char *val, void *dst)
160 {
161     int ret = 0, notfirst = 0;
162     for (;;) {
163         int i, den = 1;
164         char buf[256];
165         int cmd = 0;
166         double d, num = 1;
167         int64_t intnum = 1;
168
169         if (*val == '+' || *val == '-')
170             cmd = *(val++);
171
172         for (i = 0; i < sizeof(buf) - 1 && val[i] && val[i] != '+' && val[i] != '-'; i++)
173             buf[i] = val[i];
174         buf[i] = 0;
175
176         {
177             const AVOption *o_named = av_opt_find(obj, buf, o->unit, 0, 0);
178             if (o_named && o_named->type == AV_OPT_TYPE_CONST)
179                 d = o_named->default_val.dbl;
180             else if (!strcmp(buf, "default")) d = o->default_val.dbl;
181             else if (!strcmp(buf, "max"    )) d = o->max;
182             else if (!strcmp(buf, "min"    )) d = o->min;
183             else if (!strcmp(buf, "none"   )) d = 0;
184             else if (!strcmp(buf, "all"    )) d = ~0;
185             else {
186                 int res = av_expr_parse_and_eval(&d, buf, const_names, const_values, NULL, NULL, NULL, NULL, NULL, 0, obj);
187                 if (res < 0) {
188                     av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\"\n", val);
189                     return res;
190                 }
191             }
192         }
193         if (o->type == AV_OPT_TYPE_FLAGS) {
194             read_number(o, dst, NULL, NULL, &intnum);
195             if      (cmd == '+') d = intnum | (int64_t)d;
196             else if (cmd == '-') d = intnum &~(int64_t)d;
197         } else {
198             read_number(o, dst, &num, &den, &intnum);
199             if      (cmd == '+') d = notfirst*num*intnum/den + d;
200             else if (cmd == '-') d = notfirst*num*intnum/den - d;
201         }
202
203         if ((ret = write_number(obj, o, dst, d, 1, 1)) < 0)
204             return ret;
205         val += i;
206         if (!*val)
207             return 0;
208         notfirst = 1;
209     }
210
211     return 0;
212 }
213
214 #if FF_API_OLD_AVOPTIONS
215 int av_set_string3(void *obj, const char *name, const char *val, int alloc, const AVOption **o_out)
216 {
217     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
218     if (o_out)
219         *o_out = o;
220     return av_opt_set(obj, name, val, 0);
221 }
222 #endif
223
224 int av_opt_set(void *obj, const char *name, const char *val, int search_flags)
225 {
226     void *dst, *target_obj;
227     const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
228     if (!o || !target_obj)
229         return AVERROR_OPTION_NOT_FOUND;
230     if (!val)
231         return AVERROR(EINVAL);
232
233     dst = ((uint8_t*)target_obj) + o->offset;
234     switch (o->type) {
235     case AV_OPT_TYPE_STRING:   return set_string(obj, o, val, dst);
236     case AV_OPT_TYPE_BINARY:   return set_string_binary(obj, o, val, dst);
237     case AV_OPT_TYPE_FLAGS:
238     case AV_OPT_TYPE_INT:
239     case AV_OPT_TYPE_INT64:
240     case AV_OPT_TYPE_FLOAT:
241     case AV_OPT_TYPE_DOUBLE:
242     case AV_OPT_TYPE_RATIONAL: return set_string_number(obj, o, val, dst);
243     }
244
245     av_log(obj, AV_LOG_ERROR, "Invalid option type.\n");
246     return AVERROR(EINVAL);
247 }
248
249 #define OPT_EVAL_NUMBER(name, opttype, vartype)\
250     int av_opt_eval_ ## name(void *obj, const AVOption *o, const char *val, vartype *name ## _out)\
251     {\
252         if (!o || o->type != opttype)\
253             return AVERROR(EINVAL);\
254         return set_string_number(obj, o, val, name ## _out);\
255     }
256
257 OPT_EVAL_NUMBER(flags,  AV_OPT_TYPE_FLAGS,    int)
258 OPT_EVAL_NUMBER(int,    AV_OPT_TYPE_INT,      int)
259 OPT_EVAL_NUMBER(int64,  AV_OPT_TYPE_INT64,    int64_t)
260 OPT_EVAL_NUMBER(float,  AV_OPT_TYPE_FLOAT,    float)
261 OPT_EVAL_NUMBER(double, AV_OPT_TYPE_DOUBLE,   double)
262 OPT_EVAL_NUMBER(q,      AV_OPT_TYPE_RATIONAL, AVRational)
263
264 static int set_number(void *obj, const char *name, double num, int den, int64_t intnum,
265                                   int search_flags)
266 {
267     void *dst, *target_obj;
268     const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
269
270     if (!o || !target_obj)
271         return AVERROR_OPTION_NOT_FOUND;
272
273     dst = ((uint8_t*)target_obj) + o->offset;
274     return write_number(obj, o, dst, num, den, intnum);
275 }
276
277 #if FF_API_OLD_AVOPTIONS
278 const AVOption *av_set_double(void *obj, const char *name, double n)
279 {
280     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
281     if (set_number(obj, name, n, 1, 1, 0) < 0)
282         return NULL;
283     return o;
284 }
285
286 const AVOption *av_set_q(void *obj, const char *name, AVRational n)
287 {
288     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
289     if (set_number(obj, name, n.num, n.den, 1, 0) < 0)
290         return NULL;
291     return o;
292 }
293
294 const AVOption *av_set_int(void *obj, const char *name, int64_t n)
295 {
296     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
297     if (set_number(obj, name, 1, 1, n, 0) < 0)
298         return NULL;
299     return o;
300 }
301 #endif
302
303 int av_opt_set_int(void *obj, const char *name, int64_t val, int search_flags)
304 {
305     return set_number(obj, name, 1, 1, val, search_flags);
306 }
307
308 int av_opt_set_double(void *obj, const char *name, double val, int search_flags)
309 {
310     return set_number(obj, name, val, 1, 1, search_flags);
311 }
312
313 int av_opt_set_q(void *obj, const char *name, AVRational val, int search_flags)
314 {
315     return set_number(obj, name, val.num, val.den, 1, search_flags);
316 }
317
318 #if FF_API_OLD_AVOPTIONS
319 /**
320  *
321  * @param buf a buffer which is used for returning non string values as strings, can be NULL
322  * @param buf_len allocated length in bytes of buf
323  */
324 const char *av_get_string(void *obj, const char *name, const AVOption **o_out, char *buf, int buf_len)
325 {
326     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
327     void *dst;
328     uint8_t *bin;
329     int len, i;
330     if (!o)
331         return NULL;
332     if (o->type != AV_OPT_TYPE_STRING && (!buf || !buf_len))
333         return NULL;
334
335     dst= ((uint8_t*)obj) + o->offset;
336     if (o_out) *o_out= o;
337
338     switch (o->type) {
339     case AV_OPT_TYPE_FLAGS:     snprintf(buf, buf_len, "0x%08X",*(int    *)dst);break;
340     case AV_OPT_TYPE_INT:       snprintf(buf, buf_len, "%d" , *(int    *)dst);break;
341     case AV_OPT_TYPE_INT64:     snprintf(buf, buf_len, "%"PRId64, *(int64_t*)dst);break;
342     case AV_OPT_TYPE_FLOAT:     snprintf(buf, buf_len, "%f" , *(float  *)dst);break;
343     case AV_OPT_TYPE_DOUBLE:    snprintf(buf, buf_len, "%f" , *(double *)dst);break;
344     case AV_OPT_TYPE_RATIONAL:  snprintf(buf, buf_len, "%d/%d", ((AVRational*)dst)->num, ((AVRational*)dst)->den);break;
345     case AV_OPT_TYPE_STRING:    return *(void**)dst;
346     case AV_OPT_TYPE_BINARY:
347         len = *(int*)(((uint8_t *)dst) + sizeof(uint8_t *));
348         if (len >= (buf_len + 1)/2) return NULL;
349         bin = *(uint8_t**)dst;
350         for (i = 0; i < len; i++) snprintf(buf + i*2, 3, "%02X", bin[i]);
351         break;
352     default: return NULL;
353     }
354     return buf;
355 }
356 #endif
357
358 int av_opt_get(void *obj, const char *name, int search_flags, uint8_t **out_val)
359 {
360     void *dst, *target_obj;
361     const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
362     uint8_t *bin, buf[128];
363     int len, i, ret;
364
365     if (!o || !target_obj)
366         return AVERROR_OPTION_NOT_FOUND;
367
368     dst = (uint8_t*)target_obj + o->offset;
369
370     buf[0] = 0;
371     switch (o->type) {
372     case AV_OPT_TYPE_FLAGS:     ret = snprintf(buf, sizeof(buf), "0x%08X",  *(int    *)dst);break;
373     case AV_OPT_TYPE_INT:       ret = snprintf(buf, sizeof(buf), "%d" ,     *(int    *)dst);break;
374     case AV_OPT_TYPE_INT64:     ret = snprintf(buf, sizeof(buf), "%"PRId64, *(int64_t*)dst);break;
375     case AV_OPT_TYPE_FLOAT:     ret = snprintf(buf, sizeof(buf), "%f" ,     *(float  *)dst);break;
376     case AV_OPT_TYPE_DOUBLE:    ret = snprintf(buf, sizeof(buf), "%f" ,     *(double *)dst);break;
377     case AV_OPT_TYPE_RATIONAL:  ret = snprintf(buf, sizeof(buf), "%d/%d",   ((AVRational*)dst)->num, ((AVRational*)dst)->den);break;
378     case AV_OPT_TYPE_STRING:
379         if (*(uint8_t**)dst)
380             *out_val = av_strdup(*(uint8_t**)dst);
381         else
382             *out_val = av_strdup("");
383         return 0;
384     case AV_OPT_TYPE_BINARY:
385         len = *(int*)(((uint8_t *)dst) + sizeof(uint8_t *));
386         if ((uint64_t)len*2 + 1 > INT_MAX)
387             return AVERROR(EINVAL);
388         if (!(*out_val = av_malloc(len*2 + 1)))
389             return AVERROR(ENOMEM);
390         bin = *(uint8_t**)dst;
391         for (i = 0; i < len; i++)
392             snprintf(*out_val + i*2, 3, "%02X", bin[i]);
393         return 0;
394     default:
395         return AVERROR(EINVAL);
396     }
397
398     if (ret >= sizeof(buf))
399         return AVERROR(EINVAL);
400     *out_val = av_strdup(buf);
401     return 0;
402 }
403
404 static int get_number(void *obj, const char *name, const AVOption **o_out, double *num, int *den, int64_t *intnum,
405                       int search_flags)
406 {
407     void *dst, *target_obj;
408     const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
409     if (!o || !target_obj)
410         goto error;
411
412     dst = ((uint8_t*)target_obj) + o->offset;
413
414     if (o_out) *o_out= o;
415
416     return read_number(o, dst, num, den, intnum);
417
418 error:
419     *den=*intnum=0;
420     return -1;
421 }
422
423 #if FF_API_OLD_AVOPTIONS
424 double av_get_double(void *obj, const char *name, const AVOption **o_out)
425 {
426     int64_t intnum=1;
427     double num=1;
428     int den=1;
429
430     if (get_number(obj, name, o_out, &num, &den, &intnum, 0) < 0)
431         return NAN;
432     return num*intnum/den;
433 }
434
435 AVRational av_get_q(void *obj, const char *name, const AVOption **o_out)
436 {
437     int64_t intnum=1;
438     double num=1;
439     int den=1;
440
441     if (get_number(obj, name, o_out, &num, &den, &intnum, 0) < 0)
442         return (AVRational){0, 0};
443     if (num == 1.0 && (int)intnum == intnum)
444         return (AVRational){intnum, den};
445     else
446         return av_d2q(num*intnum/den, 1<<24);
447 }
448
449 int64_t av_get_int(void *obj, const char *name, const AVOption **o_out)
450 {
451     int64_t intnum=1;
452     double num=1;
453     int den=1;
454
455     if (get_number(obj, name, o_out, &num, &den, &intnum, 0) < 0)
456         return -1;
457     return num*intnum/den;
458 }
459 #endif
460
461 int av_opt_get_int(void *obj, const char *name, int search_flags, int64_t *out_val)
462 {
463     int64_t intnum = 1;
464     double     num = 1;
465     int   ret, den = 1;
466
467     if ((ret = get_number(obj, name, NULL, &num, &den, &intnum, search_flags)) < 0)
468         return ret;
469     *out_val = num*intnum/den;
470     return 0;
471 }
472
473 int av_opt_get_double(void *obj, const char *name, int search_flags, double *out_val)
474 {
475     int64_t intnum = 1;
476     double     num = 1;
477     int   ret, den = 1;
478
479     if ((ret = get_number(obj, name, NULL, &num, &den, &intnum, search_flags)) < 0)
480         return ret;
481     *out_val = num*intnum/den;
482     return 0;
483 }
484
485 int av_opt_get_q(void *obj, const char *name, int search_flags, AVRational *out_val)
486 {
487     int64_t intnum = 1;
488     double     num = 1;
489     int   ret, den = 1;
490
491     if ((ret = get_number(obj, name, NULL, &num, &den, &intnum, search_flags)) < 0)
492         return ret;
493
494     if (num == 1.0 && (int)intnum == intnum)
495         *out_val = (AVRational){intnum, den};
496     else
497         *out_val = av_d2q(num*intnum/den, 1<<24);
498     return 0;
499 }
500
501 int av_opt_flag_is_set(void *obj, const char *field_name, const char *flag_name)
502 {
503     const AVOption *field = av_opt_find(obj, field_name, NULL, 0, 0);
504     const AVOption *flag  = av_opt_find(obj, flag_name,
505                                         field ? field->unit : NULL, 0, 0);
506     int64_t res;
507
508     if (!field || !flag || flag->type != AV_OPT_TYPE_CONST ||
509         av_opt_get_int(obj, field_name, 0, &res) < 0)
510         return 0;
511     return res & (int) flag->default_val.dbl;
512 }
513
514 static void opt_list(void *obj, void *av_log_obj, const char *unit,
515                      int req_flags, int rej_flags)
516 {
517     const AVOption *opt=NULL;
518
519     while ((opt = av_opt_next(obj, opt))) {
520         if (!(opt->flags & req_flags) || (opt->flags & rej_flags))
521             continue;
522
523         /* Don't print CONST's on level one.
524          * Don't print anything but CONST's on level two.
525          * Only print items from the requested unit.
526          */
527         if (!unit && opt->type==AV_OPT_TYPE_CONST)
528             continue;
529         else if (unit && opt->type!=AV_OPT_TYPE_CONST)
530             continue;
531         else if (unit && opt->type==AV_OPT_TYPE_CONST && strcmp(unit, opt->unit))
532             continue;
533         else if (unit && opt->type == AV_OPT_TYPE_CONST)
534             av_log(av_log_obj, AV_LOG_INFO, "   %-15s ", opt->name);
535         else
536             av_log(av_log_obj, AV_LOG_INFO, "-%-17s ", opt->name);
537
538         switch (opt->type) {
539             case AV_OPT_TYPE_FLAGS:
540                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<flags>");
541                 break;
542             case AV_OPT_TYPE_INT:
543                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<int>");
544                 break;
545             case AV_OPT_TYPE_INT64:
546                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<int64>");
547                 break;
548             case AV_OPT_TYPE_DOUBLE:
549                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<double>");
550                 break;
551             case AV_OPT_TYPE_FLOAT:
552                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<float>");
553                 break;
554             case AV_OPT_TYPE_STRING:
555                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<string>");
556                 break;
557             case AV_OPT_TYPE_RATIONAL:
558                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<rational>");
559                 break;
560             case AV_OPT_TYPE_BINARY:
561                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<binary>");
562                 break;
563             case AV_OPT_TYPE_CONST:
564             default:
565                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "");
566                 break;
567         }
568         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_ENCODING_PARAM) ? 'E' : '.');
569         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_DECODING_PARAM) ? 'D' : '.');
570         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_VIDEO_PARAM   ) ? 'V' : '.');
571         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_AUDIO_PARAM   ) ? 'A' : '.');
572         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_SUBTITLE_PARAM) ? 'S' : '.');
573
574         if (opt->help)
575             av_log(av_log_obj, AV_LOG_INFO, " %s", opt->help);
576         av_log(av_log_obj, AV_LOG_INFO, "\n");
577         if (opt->unit && opt->type != AV_OPT_TYPE_CONST) {
578             opt_list(obj, av_log_obj, opt->unit, req_flags, rej_flags);
579         }
580     }
581 }
582
583 int av_opt_show2(void *obj, void *av_log_obj, int req_flags, int rej_flags)
584 {
585     if (!obj)
586         return -1;
587
588     av_log(av_log_obj, AV_LOG_INFO, "%s AVOptions:\n", (*(AVClass**)obj)->class_name);
589
590     opt_list(obj, av_log_obj, NULL, req_flags, rej_flags);
591
592     return 0;
593 }
594
595 void av_opt_set_defaults(void *s)
596 {
597 #if FF_API_OLD_AVOPTIONS
598     av_opt_set_defaults2(s, 0, 0);
599 }
600
601 void av_opt_set_defaults2(void *s, int mask, int flags)
602 {
603 #endif
604     const AVOption *opt = NULL;
605     while ((opt = av_opt_next(s, opt)) != NULL) {
606 #if FF_API_OLD_AVOPTIONS
607         if ((opt->flags & mask) != flags)
608             continue;
609 #endif
610         switch (opt->type) {
611             case AV_OPT_TYPE_CONST:
612                 /* Nothing to be done here */
613             break;
614             case AV_OPT_TYPE_FLAGS:
615             case AV_OPT_TYPE_INT: {
616                 int val;
617                 val = opt->default_val.dbl;
618                 av_opt_set_int(s, opt->name, val, 0);
619             }
620             break;
621             case AV_OPT_TYPE_INT64:
622                 if ((double)(opt->default_val.dbl+0.6) == opt->default_val.dbl)
623                     av_log(s, AV_LOG_DEBUG, "loss of precision in default of %s\n", opt->name);
624                 av_opt_set_int(s, opt->name, opt->default_val.dbl, 0);
625             break;
626             case AV_OPT_TYPE_DOUBLE:
627             case AV_OPT_TYPE_FLOAT: {
628                 double val;
629                 val = opt->default_val.dbl;
630                 av_opt_set_double(s, opt->name, val, 0);
631             }
632             break;
633             case AV_OPT_TYPE_RATIONAL: {
634                 AVRational val;
635                 val = av_d2q(opt->default_val.dbl, INT_MAX);
636                 av_opt_set_q(s, opt->name, val, 0);
637             }
638             break;
639             case AV_OPT_TYPE_STRING:
640                 av_opt_set(s, opt->name, opt->default_val.str, 0);
641                 break;
642             case AV_OPT_TYPE_BINARY:
643                 /* Cannot set default for binary */
644             break;
645             default:
646                 av_log(s, AV_LOG_DEBUG, "AVOption type %d of option %s not implemented yet\n", opt->type, opt->name);
647         }
648     }
649 }
650
651 /**
652  * Store the value in the field in ctx that is named like key.
653  * ctx must be an AVClass context, storing is done using AVOptions.
654  *
655  * @param buf the string to parse, buf will be updated to point at the
656  * separator just after the parsed key/value pair
657  * @param key_val_sep a 0-terminated list of characters used to
658  * separate key from value
659  * @param pairs_sep a 0-terminated list of characters used to separate
660  * two pairs from each other
661  * @return 0 if the key/value pair has been successfully parsed and
662  * set, or a negative value corresponding to an AVERROR code in case
663  * of error:
664  * AVERROR(EINVAL) if the key/value pair cannot be parsed,
665  * the error code issued by av_opt_set() if the key/value pair
666  * cannot be set
667  */
668 static int parse_key_value_pair(void *ctx, const char **buf,
669                                 const char *key_val_sep, const char *pairs_sep)
670 {
671     char *key = av_get_token(buf, key_val_sep);
672     char *val;
673     int ret;
674
675     if (*key && strspn(*buf, key_val_sep)) {
676         (*buf)++;
677         val = av_get_token(buf, pairs_sep);
678     } else {
679         av_log(ctx, AV_LOG_ERROR, "Missing key or no key/value separator found after key '%s'\n", key);
680         av_free(key);
681         return AVERROR(EINVAL);
682     }
683
684     av_log(ctx, AV_LOG_DEBUG, "Setting value '%s' for key '%s'\n", val, key);
685
686     ret = av_opt_set(ctx, key, val, 0);
687     if (ret == AVERROR_OPTION_NOT_FOUND)
688         av_log(ctx, AV_LOG_ERROR, "Key '%s' not found.\n", key);
689
690     av_free(key);
691     av_free(val);
692     return ret;
693 }
694
695 int av_set_options_string(void *ctx, const char *opts,
696                           const char *key_val_sep, const char *pairs_sep)
697 {
698     int ret, count = 0;
699
700     if (!opts)
701         return 0;
702
703     while (*opts) {
704         if ((ret = parse_key_value_pair(ctx, &opts, key_val_sep, pairs_sep)) < 0)
705             return ret;
706         count++;
707
708         if (*opts)
709             opts++;
710     }
711
712     return count;
713 }
714
715 void av_opt_free(void *obj)
716 {
717     const AVOption *o = NULL;
718     while ((o = av_opt_next(obj, o)))
719         if (o->type == AV_OPT_TYPE_STRING || o->type == AV_OPT_TYPE_BINARY)
720             av_freep((uint8_t *)obj + o->offset);
721 }
722
723 int av_opt_set_dict(void *obj, AVDictionary **options)
724 {
725     AVDictionaryEntry *t = NULL;
726     AVDictionary    *tmp = NULL;
727     int ret = 0;
728
729     while ((t = av_dict_get(*options, "", t, AV_DICT_IGNORE_SUFFIX))) {
730         ret = av_opt_set(obj, t->key, t->value, 0);
731         if (ret == AVERROR_OPTION_NOT_FOUND)
732             av_dict_set(&tmp, t->key, t->value, 0);
733         else if (ret < 0) {
734             av_log(obj, AV_LOG_ERROR, "Error setting option %s to value %s.\n", t->key, t->value);
735             break;
736         }
737         ret = 0;
738     }
739     av_dict_free(options);
740     *options = tmp;
741     return ret;
742 }
743
744 const AVOption *av_opt_find(void *obj, const char *name, const char *unit,
745                             int opt_flags, int search_flags)
746 {
747     return av_opt_find2(obj, name, unit, opt_flags, search_flags, NULL);
748 }
749
750 const AVOption *av_opt_find2(void *obj, const char *name, const char *unit,
751                              int opt_flags, int search_flags, void **target_obj)
752 {
753     const AVClass  *c = *(AVClass**)obj;
754     const AVOption *o = NULL;
755
756     if (search_flags & AV_OPT_SEARCH_CHILDREN) {
757         if (search_flags & AV_OPT_SEARCH_FAKE_OBJ) {
758             const AVClass *child = NULL;
759             while (child = av_opt_child_class_next(c, child))
760                 if (o = av_opt_find2(&child, name, unit, opt_flags, search_flags, NULL))
761                     return o;
762         } else {
763             void *child = NULL;
764             while (child = av_opt_child_next(obj, child))
765                 if (o = av_opt_find2(child, name, unit, opt_flags, search_flags, target_obj))
766                     return o;
767         }
768     }
769
770     while (o = av_opt_next(obj, o)) {
771         if (!strcmp(o->name, name) && (o->flags & opt_flags) == opt_flags &&
772             ((!unit && o->type != AV_OPT_TYPE_CONST) ||
773              (unit  && o->unit && !strcmp(o->unit, unit)))) {
774             if (target_obj) {
775                 if (!(search_flags & AV_OPT_SEARCH_FAKE_OBJ))
776                     *target_obj = obj;
777                 else
778                     *target_obj = NULL;
779             }
780             return o;
781         }
782     }
783     return NULL;
784 }
785
786 void *av_opt_child_next(void *obj, void *prev)
787 {
788     const AVClass *c = *(AVClass**)obj;
789     if (c->child_next)
790         return c->child_next(obj, prev);
791     return NULL;
792 }
793
794 const AVClass *av_opt_child_class_next(const AVClass *parent, const AVClass *prev)
795 {
796     if (parent->child_class_next)
797         return parent->child_class_next(prev);
798     return NULL;
799 }
800
801 #ifdef TEST
802
803 #undef printf
804
805 typedef struct TestContext
806 {
807     const AVClass *class;
808     int num;
809     int toggle;
810     char *string;
811     int flags;
812     AVRational rational;
813 } TestContext;
814
815 #define OFFSET(x) offsetof(TestContext, x)
816
817 #define TEST_FLAG_COOL 01
818 #define TEST_FLAG_LAME 02
819 #define TEST_FLAG_MU   04
820
821 static const AVOption test_options[]= {
822 {"num",      "set num",        OFFSET(num),      AV_OPT_TYPE_INT,      {0},              0,        100                 },
823 {"toggle",   "set toggle",     OFFSET(toggle),   AV_OPT_TYPE_INT,      {0},              0,        1                   },
824 {"rational", "set rational",   OFFSET(rational), AV_OPT_TYPE_RATIONAL, {0},              0,        10                  },
825 {"string",   "set string",     OFFSET(string),   AV_OPT_TYPE_STRING,   {0},              CHAR_MIN, CHAR_MAX            },
826 {"flags",    "set flags",      OFFSET(flags),    AV_OPT_TYPE_FLAGS,    {0},              0,        INT_MAX, 0, "flags" },
827 {"cool",     "set cool flag ", 0,                AV_OPT_TYPE_CONST,    {TEST_FLAG_COOL}, INT_MIN,  INT_MAX, 0, "flags" },
828 {"lame",     "set lame flag ", 0,                AV_OPT_TYPE_CONST,    {TEST_FLAG_LAME}, INT_MIN,  INT_MAX, 0, "flags" },
829 {"mu",       "set mu flag ",   0,                AV_OPT_TYPE_CONST,    {TEST_FLAG_MU},   INT_MIN,  INT_MAX, 0, "flags" },
830 {NULL},
831 };
832
833 static const char *test_get_name(void *ctx)
834 {
835     return "test";
836 }
837
838 static const AVClass test_class = {
839     "TestContext",
840     test_get_name,
841     test_options
842 };
843
844 int main(void)
845 {
846     int i;
847
848     printf("\nTesting av_set_options_string()\n");
849     {
850         TestContext test_ctx;
851         const char *options[] = {
852             "",
853             ":",
854             "=",
855             "foo=:",
856             ":=foo",
857             "=foo",
858             "foo=",
859             "foo",
860             "foo=val",
861             "foo==val",
862             "toggle=:",
863             "string=:",
864             "toggle=1 : foo",
865             "toggle=100",
866             "toggle==1",
867             "flags=+mu-lame : num=42: toggle=0",
868             "num=42 : string=blahblah",
869             "rational=0 : rational=1/2 : rational=1/-1",
870             "rational=-1/0",
871         };
872
873         test_ctx.class = &test_class;
874         av_opt_set_defaults(&test_ctx);
875         test_ctx.string = av_strdup("default");
876
877         av_log_set_level(AV_LOG_DEBUG);
878
879         for (i=0; i < FF_ARRAY_ELEMS(options); i++) {
880             av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]);
881             if (av_set_options_string(&test_ctx, options[i], "=", ":") < 0)
882                 av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]);
883             printf("\n");
884         }
885     }
886
887     return 0;
888 }
889
890 #endif