falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30330 Commits
32 Branches
303MB
Tree: dac66da63d
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'dac66da63d'
${ noResults }
FFmpeg/libavutil/opt.c

762 lines
24KB
Raw Blame History 

  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. const AVOption *av_next_option(void *obj, const AVOption *last)

  51. {

  52.     if (last && last[1].name) return ++last;

  53.     else if (last)            return NULL;

  54.     else                      return (*(AVClass**)obj)->option;

  55. }

  56. 

  57. static int write_number(void *obj, const AVOption *o, void *dst, double num, int den, int64_t intnum)

  58. {

  59.     if (o->max*den < num*intnum || o->min*den > num*intnum) {

  60.         av_log(obj, AV_LOG_ERROR, "Value %lf for parameter '%s' out of range\n", num, o->name);

  61.         return AVERROR(ERANGE);

  62.     }

  63. 

  64.     switch (o->type) {

  65.     case FF_OPT_TYPE_FLAGS:

  66.     case FF_OPT_TYPE_INT:   *(int       *)dst= llrint(num/den)*intnum; break;

  67.     case FF_OPT_TYPE_INT64: *(int64_t   *)dst= llrint(num/den)*intnum; break;

  68.     case FF_OPT_TYPE_FLOAT: *(float     *)dst= num*intnum/den;         break;

  69.     case FF_OPT_TYPE_DOUBLE:*(double    *)dst= num*intnum/den;         break;

  70.     case FF_OPT_TYPE_RATIONAL:

  71.         if ((int)num == num) *(AVRational*)dst= (AVRational){num*intnum, den};

  72.         else                 *(AVRational*)dst= av_d2q(num*intnum/den, 1<<24);

  73.         break;

  74.     default:

  75.         return AVERROR(EINVAL);

  76.     }

  77.     return 0;

  78. }

  79. 

  80. static const double const_values[] = {

  81.     M_PI,

  82.     M_E,

  83.     FF_QP2LAMBDA,

  84.     0

  85. };

  86. 

  87. static const char * const const_names[] = {

  88.     "PI",

  89.     "E",

  90.     "QP2LAMBDA",

  91.     0

  92. };

  93. 

  94. static int hexchar2int(char c) {

  95.     if (c >= '0' && c <= '9') return c - '0';

  96.     if (c >= 'a' && c <= 'f') return c - 'a' + 10;

  97.     if (c >= 'A' && c <= 'F') return c - 'A' + 10;

  98.     return -1;

  99. }

  100. 

  101. static int set_string_binary(void *obj, const AVOption *o, const char *val, uint8_t **dst)

  102. {

  103.     int *lendst = (int *)(dst + 1);

  104.     uint8_t *bin, *ptr;

  105.     int len = strlen(val);

  106. 

  107.     av_freep(dst);

  108.     *lendst = 0;

  109. 

  110.     if (len & 1)

  111.         return AVERROR(EINVAL);

  112.     len /= 2;

  113. 

  114.     ptr = bin = av_malloc(len);

  115.     while (*val) {

  116.         int a = hexchar2int(*val++);

  117.         int b = hexchar2int(*val++);

  118.         if (a < 0 || b < 0) {

  119.             av_free(bin);

  120.             return AVERROR(EINVAL);

  121.         }

  122.         *ptr++ = (a << 4) | b;

  123.     }

  124.     *dst = bin;

  125.     *lendst = len;

  126. 

  127.     return 0;

  128. }

  129. 

  130. static int set_string(void *obj, const AVOption *o, const char *val, uint8_t **dst)

  131. {

  132.     av_freep(dst);

  133.     *dst = av_strdup(val);

  134.     return 0;

  135. }

  136. 

  137. static int set_string_number(void *obj, const AVOption *o, const char *val, void *dst)

  138. {

  139.     int ret = 0, notfirst = 0;

  140.     for (;;) {

  141.         int i;

  142.         char buf[256];

  143.         int cmd = 0;

  144.         double d;

  145. 

  146.         if (*val == '+' || *val == '-')

  147.             cmd = *(val++);

  148. 

  149.         for (i = 0; i < sizeof(buf) - 1 && val[i] && val[i] != '+' && val[i] != '-'; i++)

  150.             buf[i] = val[i];

  151.         buf[i] = 0;

  152. 

  153.         {

  154.             const AVOption *o_named = av_opt_find(obj, buf, o->unit, 0, 0);

  155.             if (o_named && o_named->type == FF_OPT_TYPE_CONST)

  156.                 d = o_named->default_val.dbl;

  157.             else if (!strcmp(buf, "default")) d = o->default_val.dbl;

  158.             else if (!strcmp(buf, "max"    )) d = o->max;

  159.             else if (!strcmp(buf, "min"    )) d = o->min;

  160.             else if (!strcmp(buf, "none"   )) d = 0;

  161.             else if (!strcmp(buf, "all"    )) d = ~0;

  162.             else {

  163.                 int res = av_expr_parse_and_eval(&d, buf, const_names, const_values, NULL, NULL, NULL, NULL, NULL, 0, obj);

  164.                 if (res < 0) {

  165.                     av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\"\n", val);

  166.                     return res;

  167.                 }

  168.             }

  169.         }

  170.         if (o->type == FF_OPT_TYPE_FLAGS) {

  171.             if      (cmd == '+') d = av_get_int(obj, o->name, NULL) | (int64_t)d;

  172.             else if (cmd == '-') d = av_get_int(obj, o->name, NULL) &~(int64_t)d;

  173.         } else {

  174.             if      (cmd == '+') d = notfirst*av_get_double(obj, o->name, NULL) + d;

  175.             else if (cmd == '-') d = notfirst*av_get_double(obj, o->name, NULL) - d;

  176.         }

  177. 

  178.         if ((ret = write_number(obj, o, dst, d, 1, 1)) < 0)

  179.             return ret;

  180.         val += i;

  181.         if (!*val)

  182.             return 0;

  183.         notfirst = 1;

  184.     }

  185. 

  186.     return 0;

  187. }

  188. 

  189. #if FF_API_OLD_AVOPTIONS

  190. int av_set_string3(void *obj, const char *name, const char *val, int alloc, const AVOption **o_out)

  191. {

  192.     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);

  193.     if (o_out)

  194.         *o_out = o;

  195.     return av_opt_set(obj, name, val, 0);

  196. }

  197. #endif

  198. 

  199. int av_opt_set(void *obj, const char *name, const char *val, int search_flags)

  200. {

  201.     void *dst, *target_obj;

  202.     const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);

  203.     if (!o || !target_obj)

  204.         return AVERROR_OPTION_NOT_FOUND;

  205.     if (!val)

  206.         return AVERROR(EINVAL);

  207. 

  208.     dst = ((uint8_t*)target_obj) + o->offset;

  209.     switch (o->type) {

  210.     case FF_OPT_TYPE_STRING:   return set_string(obj, o, val, dst);

  211.     case FF_OPT_TYPE_BINARY:   return set_string_binary(obj, o, val, dst);

  212.     case FF_OPT_TYPE_FLAGS:

  213.     case FF_OPT_TYPE_INT:

  214.     case FF_OPT_TYPE_INT64:

  215.     case FF_OPT_TYPE_FLOAT:

  216.     case FF_OPT_TYPE_DOUBLE:

  217.     case FF_OPT_TYPE_RATIONAL: return set_string_number(obj, o, val, dst);

  218.     }

  219. 

  220.     av_log(obj, AV_LOG_ERROR, "Invalid option type.\n");

  221.     return AVERROR(EINVAL);

  222. }

  223. 

  224. static int set_number(void *obj, const char *name, double num, int den, int64_t intnum,

  225.                                   int search_flags)

  226. {

  227.     void *dst, *target_obj;

  228.     const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);

  229. 

  230.     if (!o || !target_obj)

  231.         return AVERROR_OPTION_NOT_FOUND;

  232. 

  233.     dst = ((uint8_t*)target_obj) + o->offset;

  234.     return write_number(obj, o, dst, num, den, intnum);

  235. }

  236. 

  237. #if FF_API_OLD_AVOPTIONS

  238. const AVOption *av_set_double(void *obj, const char *name, double n)

  239. {

  240.     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);

  241.     if (set_number(obj, name, n, 1, 1, 0) < 0)

  242.         return NULL;

  243.     return o;

  244. }

  245. 

  246. const AVOption *av_set_q(void *obj, const char *name, AVRational n)

  247. {

  248.     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);

  249.     if (set_number(obj, name, n.num, n.den, 1, 0) < 0)

  250.         return NULL;

  251.     return o;

  252. }

  253. 

  254. const AVOption *av_set_int(void *obj, const char *name, int64_t n)

  255. {

  256.     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);

  257.     if (set_number(obj, name, 1, 1, n, 0) < 0)

  258.         return NULL;

  259.     return o;

  260. }

  261. #endif

  262. 

  263. int av_opt_set_int(void *obj, const char *name, int64_t val, int search_flags)

  264. {

  265.     return set_number(obj, name, 1, 1, val, search_flags);

  266. }

  267. 

  268. int av_opt_set_double(void *obj, const char *name, double val, int search_flags)

  269. {

  270.     return set_number(obj, name, val, 1, 1, search_flags);

  271. }

  272. 

  273. int av_opt_set_q(void *obj, const char *name, AVRational val, int search_flags)

  274. {

  275.     return set_number(obj, name, val.num, val.den, 1, search_flags);

  276. }

  277. 

  278. /**

  279.  *

  280.  * @param buf a buffer which is used for returning non string values as strings, can be NULL

  281.  * @param buf_len allocated length in bytes of buf

  282.  */

  283. const char *av_get_string(void *obj, const char *name, const AVOption **o_out, char *buf, int buf_len)

  284. {

  285.     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);

  286.     void *dst;

  287.     uint8_t *bin;

  288.     int len, i;

  289.     if (!o)

  290.         return NULL;

  291.     if (o->type != FF_OPT_TYPE_STRING && (!buf || !buf_len))

  292.         return NULL;

  293. 

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

  295.     if (o_out) *o_out= o;

  296. 

  297.     switch (o->type) {

  298.     case FF_OPT_TYPE_FLAGS:     snprintf(buf, buf_len, "0x%08X",*(int    *)dst);break;

  299.     case FF_OPT_TYPE_INT:       snprintf(buf, buf_len, "%d" , *(int    *)dst);break;

  300.     case FF_OPT_TYPE_INT64:     snprintf(buf, buf_len, "%"PRId64, *(int64_t*)dst);break;

  301.     case FF_OPT_TYPE_FLOAT:     snprintf(buf, buf_len, "%f" , *(float  *)dst);break;

  302.     case FF_OPT_TYPE_DOUBLE:    snprintf(buf, buf_len, "%f" , *(double *)dst);break;

  303.     case FF_OPT_TYPE_RATIONAL:  snprintf(buf, buf_len, "%d/%d", ((AVRational*)dst)->num, ((AVRational*)dst)->den);break;

  304.     case FF_OPT_TYPE_STRING:    return *(void**)dst;

  305.     case FF_OPT_TYPE_BINARY:

  306.         len = *(int*)(((uint8_t *)dst) + sizeof(uint8_t *));

  307.         if (len >= (buf_len + 1)/2) return NULL;

  308.         bin = *(uint8_t**)dst;

  309.         for (i = 0; i < len; i++) snprintf(buf + i*2, 3, "%02X", bin[i]);

  310.         break;

  311.     default: return NULL;

  312.     }

  313.     return buf;

  314. }

  315. 

  316. static int get_number(void *obj, const char *name, const AVOption **o_out, double *num, int *den, int64_t *intnum)

  317. {

  318.     const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);

  319.     void *dst;

  320.     if (!o)

  321.         goto error;

  322. 

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

  324. 

  325.     if (o_out) *o_out= o;

  326. 

  327.     switch (o->type) {

  328.     case FF_OPT_TYPE_FLAGS:     *intnum= *(unsigned int*)dst;return 0;

  329.     case FF_OPT_TYPE_INT:       *intnum= *(int    *)dst;return 0;

  330.     case FF_OPT_TYPE_INT64:     *intnum= *(int64_t*)dst;return 0;

  331.     case FF_OPT_TYPE_FLOAT:     *num=    *(float  *)dst;return 0;

  332.     case FF_OPT_TYPE_DOUBLE:    *num=    *(double *)dst;return 0;

  333.     case FF_OPT_TYPE_RATIONAL:  *intnum= ((AVRational*)dst)->num;

  334.                                 *den   = ((AVRational*)dst)->den;

  335.                                                         return 0;

  336.     }

  337. error:

  338.     *den=*intnum=0;

  339.     return -1;

  340. }

  341. 

  342. double av_get_double(void *obj, const char *name, const AVOption **o_out)

  343. {

  344.     int64_t intnum=1;

  345.     double num=1;

  346.     int den=1;

  347. 

  348.     if (get_number(obj, name, o_out, &num, &den, &intnum) < 0)

  349.         return NAN;

  350.     return num*intnum/den;

  351. }

  352. 

  353. AVRational av_get_q(void *obj, const char *name, const AVOption **o_out)

  354. {

  355.     int64_t intnum=1;

  356.     double num=1;

  357.     int den=1;

  358. 

  359.     if (get_number(obj, name, o_out, &num, &den, &intnum) < 0)

  360.         return (AVRational){0, 0};

  361.     if (num == 1.0 && (int)intnum == intnum)

  362.         return (AVRational){intnum, den};

  363.     else

  364.         return av_d2q(num*intnum/den, 1<<24);

  365. }

  366. 

  367. int64_t av_get_int(void *obj, const char *name, const AVOption **o_out)

  368. {

  369.     int64_t intnum=1;

  370.     double num=1;

  371.     int den=1;

  372. 

  373.     if (get_number(obj, name, o_out, &num, &den, &intnum) < 0)

  374.         return -1;

  375.     return num*intnum/den;

  376. }

  377. 

  378. int av_opt_flag_is_set(void *obj, const char *field_name, const char *flag_name)

  379. {

  380.     const AVOption *field = av_find_opt(obj, field_name, NULL, 0, 0);

  381.     const AVOption *flag  = av_find_opt(obj, flag_name,  NULL, 0, 0);

  382. 

  383.     if (!field || !flag || flag->type != FF_OPT_TYPE_CONST)

  384.         return 0;

  385.     return av_get_int(obj, field_name, NULL) & (int) flag->default_val.dbl;

  386. }

  387. 

  388. static void opt_list(void *obj, void *av_log_obj, const char *unit,

  389.                      int req_flags, int rej_flags)

  390. {

  391.     const AVOption *opt=NULL;

  392. 

  393.     while ((opt= av_next_option(obj, opt))) {

  394.         if (!(opt->flags & req_flags) || (opt->flags & rej_flags))

  395.             continue;

  396. 

  397.         /* Don't print CONST's on level one.

  398.          * Don't print anything but CONST's on level two.

  399.          * Only print items from the requested unit.

  400.          */

  401.         if (!unit && opt->type==FF_OPT_TYPE_CONST)

  402.             continue;

  403.         else if (unit && opt->type!=FF_OPT_TYPE_CONST)

  404.             continue;

  405.         else if (unit && opt->type==FF_OPT_TYPE_CONST && strcmp(unit, opt->unit))

  406.             continue;

  407.         else if (unit && opt->type == FF_OPT_TYPE_CONST)

  408.             av_log(av_log_obj, AV_LOG_INFO, "   %-15s ", opt->name);

  409.         else

  410.             av_log(av_log_obj, AV_LOG_INFO, "-%-17s ", opt->name);

  411. 

  412.         switch (opt->type) {

  413.             case FF_OPT_TYPE_FLAGS:

  414.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<flags>");

  415.                 break;

  416.             case FF_OPT_TYPE_INT:

  417.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<int>");

  418.                 break;

  419.             case FF_OPT_TYPE_INT64:

  420.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<int64>");

  421.                 break;

  422.             case FF_OPT_TYPE_DOUBLE:

  423.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<double>");

  424.                 break;

  425.             case FF_OPT_TYPE_FLOAT:

  426.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<float>");

  427.                 break;

  428.             case FF_OPT_TYPE_STRING:

  429.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<string>");

  430.                 break;

  431.             case FF_OPT_TYPE_RATIONAL:

  432.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<rational>");

  433.                 break;

  434.             case FF_OPT_TYPE_BINARY:

  435.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "<binary>");

  436.                 break;

  437.             case FF_OPT_TYPE_CONST:

  438.             default:

  439.                 av_log(av_log_obj, AV_LOG_INFO, "%-7s ", "");

  440.                 break;

  441.         }

  442.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_ENCODING_PARAM) ? 'E' : '.');

  443.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_DECODING_PARAM) ? 'D' : '.');

  444.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_VIDEO_PARAM   ) ? 'V' : '.');

  445.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_AUDIO_PARAM   ) ? 'A' : '.');

  446.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_SUBTITLE_PARAM) ? 'S' : '.');

  447. 

  448.         if (opt->help)

  449.             av_log(av_log_obj, AV_LOG_INFO, " %s", opt->help);

  450.         av_log(av_log_obj, AV_LOG_INFO, "\n");

  451.         if (opt->unit && opt->type != FF_OPT_TYPE_CONST) {

  452.             opt_list(obj, av_log_obj, opt->unit, req_flags, rej_flags);

  453.         }

  454.     }

  455. }

  456. 

  457. int av_opt_show2(void *obj, void *av_log_obj, int req_flags, int rej_flags)

  458. {

  459.     if (!obj)

  460.         return -1;

  461. 

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

  463. 

  464.     opt_list(obj, av_log_obj, NULL, req_flags, rej_flags);

  465. 

  466.     return 0;

  467. }

  468. 

  469. void av_opt_set_defaults(void *s)

  470. {

  471. #if FF_API_OLD_AVOPTIONS

  472.     av_opt_set_defaults2(s, 0, 0);

  473. }

  474. 

  475. void av_opt_set_defaults2(void *s, int mask, int flags)

  476. {

  477. #endif

  478.     const AVOption *opt = NULL;

  479.     while ((opt = av_next_option(s, opt)) != NULL) {

  480. #if FF_API_OLD_AVOPTIONS

  481.         if ((opt->flags & mask) != flags)

  482.             continue;

  483. #endif

  484.         switch (opt->type) {

  485.             case FF_OPT_TYPE_CONST:

  486.                 /* Nothing to be done here */

  487.             break;

  488.             case FF_OPT_TYPE_FLAGS:

  489.             case FF_OPT_TYPE_INT: {

  490.                 int val;

  491.                 val = opt->default_val.dbl;

  492.                 av_set_int(s, opt->name, val);

  493.             }

  494.             break;

  495.             case FF_OPT_TYPE_INT64:

  496.                 if ((double)(opt->default_val.dbl+0.6) == opt->default_val.dbl)

  497.                     av_log(s, AV_LOG_DEBUG, "loss of precision in default of %s\n", opt->name);

  498.                 av_set_int(s, opt->name, opt->default_val.dbl);

  499.             break;

  500.             case FF_OPT_TYPE_DOUBLE:

  501.             case FF_OPT_TYPE_FLOAT: {

  502.                 double val;

  503.                 val = opt->default_val.dbl;

  504.                 av_set_double(s, opt->name, val);

  505.             }

  506.             break;

  507.             case FF_OPT_TYPE_RATIONAL: {

  508.                 AVRational val;

  509.                 val = av_d2q(opt->default_val.dbl, INT_MAX);

  510.                 av_set_q(s, opt->name, val);

  511.             }

  512.             break;

  513.             case FF_OPT_TYPE_STRING:

  514.                 av_set_string3(s, opt->name, opt->default_val.str, 1, NULL);

  515.                 break;

  516.             case FF_OPT_TYPE_BINARY:

  517.                 /* Cannot set default for binary */

  518.             break;

  519.             default:

  520.                 av_log(s, AV_LOG_DEBUG, "AVOption type %d of option %s not implemented yet\n", opt->type, opt->name);

  521.         }

  522.     }

  523. }

  524. 

  525. /**

  526.  * Store the value in the field in ctx that is named like key.

  527.  * ctx must be an AVClass context, storing is done using AVOptions.

  528.  *

  529.  * @param buf the string to parse, buf will be updated to point at the

  530.  * separator just after the parsed key/value pair

  531.  * @param key_val_sep a 0-terminated list of characters used to

  532.  * separate key from value

  533.  * @param pairs_sep a 0-terminated list of characters used to separate

  534.  * two pairs from each other

  535.  * @return 0 if the key/value pair has been successfully parsed and

  536.  * set, or a negative value corresponding to an AVERROR code in case

  537.  * of error:

  538.  * AVERROR(EINVAL) if the key/value pair cannot be parsed,

  539.  * the error code issued by av_set_string3() if the key/value pair

  540.  * cannot be set

  541.  */

  542. static int parse_key_value_pair(void *ctx, const char **buf,

  543.                                 const char *key_val_sep, const char *pairs_sep)

  544. {

  545.     char *key = av_get_token(buf, key_val_sep);

  546.     char *val;

  547.     int ret;

  548. 

  549.     if (*key && strspn(*buf, key_val_sep)) {

  550.         (*buf)++;

  551.         val = av_get_token(buf, pairs_sep);

  552.     } else {

  553.         av_log(ctx, AV_LOG_ERROR, "Missing key or no key/value separator found after key '%s'\n", key);

  554.         av_free(key);

  555.         return AVERROR(EINVAL);

  556.     }

  557. 

  558.     av_log(ctx, AV_LOG_DEBUG, "Setting value '%s' for key '%s'\n", val, key);

  559. 

  560.     ret = av_set_string3(ctx, key, val, 1, NULL);

  561.     if (ret == AVERROR_OPTION_NOT_FOUND)

  562.         av_log(ctx, AV_LOG_ERROR, "Key '%s' not found.\n", key);

  563. 

  564.     av_free(key);

  565.     av_free(val);

  566.     return ret;

  567. }

  568. 

  569. int av_set_options_string(void *ctx, const char *opts,

  570.                           const char *key_val_sep, const char *pairs_sep)

  571. {

  572.     int ret, count = 0;

  573. 

  574.     while (*opts) {

  575.         if ((ret = parse_key_value_pair(ctx, &opts, key_val_sep, pairs_sep)) < 0)

  576.             return ret;

  577.         count++;

  578. 

  579.         if (*opts)

  580.             opts++;

  581.     }

  582. 

  583.     return count;

  584. }

  585. 

  586. void av_opt_free(void *obj)

  587. {

  588.     const AVOption *o = NULL;

  589.     while ((o = av_next_option(obj, o)))

  590.         if (o->type == FF_OPT_TYPE_STRING || o->type == FF_OPT_TYPE_BINARY)

  591.             av_freep((uint8_t *)obj + o->offset);

  592. }

  593. 

  594. int av_opt_set_dict(void *obj, AVDictionary **options)

  595. {

  596.     AVDictionaryEntry *t = NULL;

  597.     AVDictionary    *tmp = NULL;

  598.     int ret = 0;

  599. 

  600.     while ((t = av_dict_get(*options, "", t, AV_DICT_IGNORE_SUFFIX))) {

  601.         ret = av_set_string3(obj, t->key, t->value, 1, NULL);

  602.         if (ret == AVERROR_OPTION_NOT_FOUND)

  603.             av_dict_set(&tmp, t->key, t->value, 0);

  604.         else if (ret < 0) {

  605.             av_log(obj, AV_LOG_ERROR, "Error setting option %s to value %s.\n", t->key, t->value);

  606.             break;

  607.         }

  608.         ret = 0;

  609.     }

  610.     av_dict_free(options);

  611.     *options = tmp;

  612.     return ret;

  613. }

  614. 

  615. const AVOption *av_opt_find(void *obj, const char *name, const char *unit,

  616.                             int opt_flags, int search_flags)

  617. {

  618.     return av_opt_find2(obj, name, unit, opt_flags, search_flags, NULL);

  619. }

  620. 

  621. const AVOption *av_opt_find2(void *obj, const char *name, const char *unit,

  622.                              int opt_flags, int search_flags, void **target_obj)

  623. {

  624.     const AVClass  *c = *(AVClass**)obj;

  625.     const AVOption *o = NULL;

  626. 

  627.     if (search_flags & AV_OPT_SEARCH_CHILDREN) {

  628.         if (search_flags & AV_OPT_SEARCH_FAKE_OBJ) {

  629.             const AVClass *child = NULL;

  630.             while (child = av_opt_child_class_next(c, child))

  631.                 if (o = av_opt_find2(&child, name, unit, opt_flags, search_flags, NULL))

  632.                     return o;

  633.         } else {

  634.             void *child = NULL;

  635.             while (child = av_opt_child_next(obj, child))

  636.                 if (o = av_opt_find2(child, name, unit, opt_flags, search_flags, target_obj))

  637.                     return o;

  638.         }

  639.     }

  640. 

  641.     while (o = av_next_option(obj, o)) {

  642.         if (!strcmp(o->name, name) && (o->flags & opt_flags) == opt_flags &&

  643.             ((!unit && o->type != FF_OPT_TYPE_CONST) ||

  644.              (unit  && o->unit && !strcmp(o->unit, unit)))) {

  645.             if (target_obj) {

  646.                 if (!(search_flags & AV_OPT_SEARCH_FAKE_OBJ))

  647.                     *target_obj = obj;

  648.                 else

  649.                     *target_obj = NULL;

  650.             }

  651.             return o;

  652.         }

  653.     }

  654.     return NULL;

  655. }

  656. 

  657. void *av_opt_child_next(void *obj, void *prev)

  658. {

  659.     const AVClass *c = *(AVClass**)obj;

  660.     if (c->child_next)

  661.         return c->child_next(obj, prev);

  662.     return NULL;

  663. }

  664. 

  665. const AVClass *av_opt_child_class_next(const AVClass *parent, const AVClass *prev)

  666. {

  667.     if (parent->child_class_next)

  668.         return parent->child_class_next(prev);

  669.     return NULL;

  670. }

  671. 

  672. #ifdef TEST

  673. 

  674. #undef printf

  675. 

  676. typedef struct TestContext

  677. {

  678.     const AVClass *class;

  679.     int num;

  680.     int toggle;

  681.     char *string;

  682.     int flags;

  683.     AVRational rational;

  684. } TestContext;

  685. 

  686. #define OFFSET(x) offsetof(TestContext, x)

  687. 

  688. #define TEST_FLAG_COOL 01

  689. #define TEST_FLAG_LAME 02

  690. #define TEST_FLAG_MU   04

  691. 

  692. static const AVOption test_options[]= {

  693. {"num",      "set num",        OFFSET(num),      FF_OPT_TYPE_INT,      {0},              0,        100                 },

  694. {"toggle",   "set toggle",     OFFSET(toggle),   FF_OPT_TYPE_INT,      {0},              0,        1                   },

  695. {"rational", "set rational",   OFFSET(rational), FF_OPT_TYPE_RATIONAL, {0},              0,        10                  },

  696. {"string",   "set string",     OFFSET(string),   FF_OPT_TYPE_STRING,   {0},              CHAR_MIN, CHAR_MAX            },

  697. {"flags",    "set flags",      OFFSET(flags),    FF_OPT_TYPE_FLAGS,    {0},              0,        INT_MAX, 0, "flags" },

  698. {"cool",     "set cool flag ", 0,                FF_OPT_TYPE_CONST,    {TEST_FLAG_COOL}, INT_MIN,  INT_MAX, 0, "flags" },

  699. {"lame",     "set lame flag ", 0,                FF_OPT_TYPE_CONST,    {TEST_FLAG_LAME}, INT_MIN,  INT_MAX, 0, "flags" },

  700. {"mu",       "set mu flag ",   0,                FF_OPT_TYPE_CONST,    {TEST_FLAG_MU},   INT_MIN,  INT_MAX, 0, "flags" },

  701. {NULL},

  702. };

  703. 

  704. static const char *test_get_name(void *ctx)

  705. {

  706.     return "test";

  707. }

  708. 

  709. static const AVClass test_class = {

  710.     "TestContext",

  711.     test_get_name,

  712.     test_options

  713. };

  714. 

  715. int main(void)

  716. {

  717.     int i;

  718. 

  719.     printf("\nTesting av_set_options_string()\n");

  720.     {

  721.         TestContext test_ctx;

  722.         const char *options[] = {

  723.             "",

  724.             ":",

  725.             "=",

  726.             "foo=:",

  727.             ":=foo",

  728.             "=foo",

  729.             "foo=",

  730.             "foo",

  731.             "foo=val",

  732.             "foo==val",

  733.             "toggle=:",

  734.             "string=:",

  735.             "toggle=1 : foo",

  736.             "toggle=100",

  737.             "toggle==1",

  738.             "flags=+mu-lame : num=42: toggle=0",

  739.             "num=42 : string=blahblah",

  740.             "rational=0 : rational=1/2 : rational=1/-1",

  741.             "rational=-1/0",

  742.         };

  743. 

  744.         test_ctx.class = &test_class;

  745.         av_opt_set_defaults(&test_ctx);

  746.         test_ctx.string = av_strdup("default");

  747. 

  748.         av_log_set_level(AV_LOG_DEBUG);

  749. 

  750.         for (i=0; i < FF_ARRAY_ELEMS(options); i++) {

  751.             av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]);

  752.             if (av_set_options_string(&test_ctx, options[i], "=", ":") < 0)

  753.                 av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]);

  754.             printf("\n");

  755.         }

  756.     }

  757. 

  758.     return 0;

  759. }

  760. 

  761. #endif