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.

1591 lines
51KB

  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. * @file
  23. * AVOptions
  24. * @author Michael Niedermayer <michaelni@gmx.at>
  25. */
  26. #include "avutil.h"
  27. #include "avstring.h"
  28. #include "common.h"
  29. #include "opt.h"
  30. #include "eval.h"
  31. #include "dict.h"
  32. #include "log.h"
  33. #include "parseutils.h"
  34. #include "pixdesc.h"
  35. #include "mathematics.h"
  36. #include "samplefmt.h"
  37. #include <float.h>
  38. #if FF_API_FIND_OPT
  39. //FIXME order them and do a bin search
  40. const AVOption *av_find_opt(void *v, const char *name, const char *unit, int mask, int flags)
  41. {
  42. const AVOption *o = NULL;
  43. while ((o = av_next_option(v, o))) {
  44. if (!strcmp(o->name, name) && (!unit || (o->unit && !strcmp(o->unit, unit))) && (o->flags & mask) == flags)
  45. return o;
  46. }
  47. return NULL;
  48. }
  49. #endif
  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. const AVOption *av_opt_next(void *obj, const AVOption *last)
  57. {
  58. AVClass *class = *(AVClass**)obj;
  59. if (!last && class && class->option && class->option[0].name)
  60. return class->option;
  61. if (last && last[1].name)
  62. return ++last;
  63. return NULL;
  64. }
  65. static int read_number(const AVOption *o, void *dst, double *num, int *den, int64_t *intnum)
  66. {
  67. switch (o->type) {
  68. case AV_OPT_TYPE_FLAGS: *intnum = *(unsigned int*)dst;return 0;
  69. case AV_OPT_TYPE_PIXEL_FMT:
  70. case AV_OPT_TYPE_SAMPLE_FMT:
  71. case AV_OPT_TYPE_INT: *intnum = *(int *)dst;return 0;
  72. case AV_OPT_TYPE_DURATION:
  73. case AV_OPT_TYPE_INT64: *intnum = *(int64_t *)dst;return 0;
  74. case AV_OPT_TYPE_FLOAT: *num = *(float *)dst;return 0;
  75. case AV_OPT_TYPE_DOUBLE: *num = *(double *)dst;return 0;
  76. case AV_OPT_TYPE_RATIONAL: *intnum = ((AVRational*)dst)->num;
  77. *den = ((AVRational*)dst)->den;
  78. return 0;
  79. case AV_OPT_TYPE_CONST: *num = o->default_val.dbl; return 0;
  80. }
  81. return AVERROR(EINVAL);
  82. }
  83. static int write_number(void *obj, const AVOption *o, void *dst, double num, int den, int64_t intnum)
  84. {
  85. if (o->max*den < num*intnum || o->min*den > num*intnum) {
  86. av_log(obj, AV_LOG_ERROR, "Value %f for parameter '%s' out of range [%g - %g]\n",
  87. num*intnum/den, o->name, o->min, o->max);
  88. return AVERROR(ERANGE);
  89. }
  90. switch (o->type) {
  91. case AV_OPT_TYPE_FLAGS:
  92. case AV_OPT_TYPE_PIXEL_FMT:
  93. case AV_OPT_TYPE_SAMPLE_FMT:
  94. case AV_OPT_TYPE_INT: *(int *)dst= llrint(num/den)*intnum; break;
  95. case AV_OPT_TYPE_DURATION:
  96. case AV_OPT_TYPE_INT64: *(int64_t *)dst= llrint(num/den)*intnum; break;
  97. case AV_OPT_TYPE_FLOAT: *(float *)dst= num*intnum/den; break;
  98. case AV_OPT_TYPE_DOUBLE:*(double *)dst= num*intnum/den; break;
  99. case AV_OPT_TYPE_RATIONAL:
  100. if ((int)num == num) *(AVRational*)dst= (AVRational){num*intnum, den};
  101. else *(AVRational*)dst= av_d2q(num*intnum/den, 1<<24);
  102. break;
  103. default:
  104. return AVERROR(EINVAL);
  105. }
  106. return 0;
  107. }
  108. static const double const_values[] = {
  109. M_PI,
  110. M_E,
  111. FF_QP2LAMBDA,
  112. 0
  113. };
  114. static const char * const const_names[] = {
  115. "PI",
  116. "E",
  117. "QP2LAMBDA",
  118. 0
  119. };
  120. static int hexchar2int(char c) {
  121. if (c >= '0' && c <= '9') return c - '0';
  122. if (c >= 'a' && c <= 'f') return c - 'a' + 10;
  123. if (c >= 'A' && c <= 'F') return c - 'A' + 10;
  124. return -1;
  125. }
  126. static int set_string_binary(void *obj, const AVOption *o, const char *val, uint8_t **dst)
  127. {
  128. int *lendst = (int *)(dst + 1);
  129. uint8_t *bin, *ptr;
  130. int len = strlen(val);
  131. av_freep(dst);
  132. *lendst = 0;
  133. if (len & 1)
  134. return AVERROR(EINVAL);
  135. len /= 2;
  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. return 0;
  149. }
  150. static int set_string(void *obj, const AVOption *o, const char *val, uint8_t **dst)
  151. {
  152. av_freep(dst);
  153. *dst = av_strdup(val);
  154. return 0;
  155. }
  156. #define DEFAULT_NUMVAL(opt) ((opt->type == AV_OPT_TYPE_INT64 || \
  157. opt->type == AV_OPT_TYPE_CONST || \
  158. opt->type == AV_OPT_TYPE_FLAGS || \
  159. opt->type == AV_OPT_TYPE_INT) ? \
  160. opt->default_val.i64 : opt->default_val.dbl)
  161. static int set_string_number(void *obj, void *target_obj, const AVOption *o, const char *val, void *dst)
  162. {
  163. int ret = 0, notfirst = 0;
  164. for (;;) {
  165. int i, den = 1;
  166. char buf[256];
  167. int cmd = 0;
  168. double d, num = 1;
  169. int64_t intnum = 1;
  170. i = 0;
  171. if (*val == '+' || *val == '-') {
  172. if (o->type == AV_OPT_TYPE_FLAGS)
  173. cmd = *(val++);
  174. else if (!notfirst)
  175. buf[i++] = *val;
  176. }
  177. for (; i < sizeof(buf) - 1 && val[i] && val[i] != '+' && val[i] != '-'; i++)
  178. buf[i] = val[i];
  179. buf[i] = 0;
  180. {
  181. const AVOption *o_named = av_opt_find(target_obj, buf, o->unit, 0, 0);
  182. if (o_named && o_named->type == AV_OPT_TYPE_CONST)
  183. d = DEFAULT_NUMVAL(o_named);
  184. else if (!strcmp(buf, "default")) d = DEFAULT_NUMVAL(o);
  185. else if (!strcmp(buf, "max" )) d = o->max;
  186. else if (!strcmp(buf, "min" )) d = o->min;
  187. else if (!strcmp(buf, "none" )) d = 0;
  188. else if (!strcmp(buf, "all" )) d = ~0;
  189. else {
  190. int res = av_expr_parse_and_eval(&d, buf, const_names, const_values, NULL, NULL, NULL, NULL, NULL, 0, obj);
  191. if (res < 0) {
  192. av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\"\n", val);
  193. return res;
  194. }
  195. }
  196. }
  197. if (o->type == AV_OPT_TYPE_FLAGS) {
  198. read_number(o, dst, NULL, NULL, &intnum);
  199. if (cmd == '+') d = intnum | (int64_t)d;
  200. else if (cmd == '-') d = intnum &~(int64_t)d;
  201. } else {
  202. read_number(o, dst, &num, &den, &intnum);
  203. if (cmd == '+') d = notfirst*num*intnum/den + d;
  204. else if (cmd == '-') d = notfirst*num*intnum/den - d;
  205. }
  206. if ((ret = write_number(obj, o, dst, d, 1, 1)) < 0)
  207. return ret;
  208. val += i;
  209. if (!*val)
  210. return 0;
  211. notfirst = 1;
  212. }
  213. return 0;
  214. }
  215. #if FF_API_OLD_AVOPTIONS
  216. int av_set_string3(void *obj, const char *name, const char *val, int alloc, const AVOption **o_out)
  217. {
  218. const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
  219. if (o_out)
  220. *o_out = o;
  221. return av_opt_set(obj, name, val, 0);
  222. }
  223. #endif
  224. int av_opt_set(void *obj, const char *name, const char *val, int search_flags)
  225. {
  226. int ret;
  227. void *dst, *target_obj;
  228. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  229. if (!o || !target_obj)
  230. return AVERROR_OPTION_NOT_FOUND;
  231. if (!val && (o->type != AV_OPT_TYPE_STRING &&
  232. o->type != AV_OPT_TYPE_PIXEL_FMT && o->type != AV_OPT_TYPE_SAMPLE_FMT &&
  233. o->type != AV_OPT_TYPE_IMAGE_SIZE && o->type != AV_OPT_TYPE_VIDEO_RATE &&
  234. o->type != AV_OPT_TYPE_DURATION && o->type != AV_OPT_TYPE_COLOR))
  235. return AVERROR(EINVAL);
  236. dst = ((uint8_t*)target_obj) + o->offset;
  237. switch (o->type) {
  238. case AV_OPT_TYPE_STRING: return set_string(obj, o, val, dst);
  239. case AV_OPT_TYPE_BINARY: return set_string_binary(obj, o, val, dst);
  240. case AV_OPT_TYPE_FLAGS:
  241. case AV_OPT_TYPE_INT:
  242. case AV_OPT_TYPE_INT64:
  243. case AV_OPT_TYPE_FLOAT:
  244. case AV_OPT_TYPE_DOUBLE:
  245. case AV_OPT_TYPE_RATIONAL: return set_string_number(obj, target_obj, o, val, dst);
  246. case AV_OPT_TYPE_IMAGE_SIZE:
  247. if (!val || !strcmp(val, "none")) {
  248. *(int *)dst = *((int *)dst + 1) = 0;
  249. return 0;
  250. }
  251. ret = av_parse_video_size(dst, ((int *)dst) + 1, val);
  252. if (ret < 0)
  253. av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\" as image size\n", val);
  254. return ret;
  255. case AV_OPT_TYPE_VIDEO_RATE:
  256. if (!val) {
  257. ret = AVERROR(EINVAL);
  258. } else {
  259. ret = av_parse_video_rate(dst, val);
  260. }
  261. if (ret < 0)
  262. av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\" as video rate\n", val);
  263. return ret;
  264. case AV_OPT_TYPE_PIXEL_FMT:
  265. if (!val || !strcmp(val, "none")) {
  266. ret = AV_PIX_FMT_NONE;
  267. } else {
  268. ret = av_get_pix_fmt(val);
  269. if (ret == AV_PIX_FMT_NONE) {
  270. char *tail;
  271. ret = strtol(val, &tail, 0);
  272. if (*tail || (unsigned)ret >= AV_PIX_FMT_NB) {
  273. av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\" as pixel format\n", val);
  274. return AVERROR(EINVAL);
  275. }
  276. }
  277. }
  278. *(enum AVPixelFormat *)dst = ret;
  279. return 0;
  280. case AV_OPT_TYPE_SAMPLE_FMT:
  281. if (!val || !strcmp(val, "none")) {
  282. ret = AV_SAMPLE_FMT_NONE;
  283. } else {
  284. ret = av_get_sample_fmt(val);
  285. if (ret == AV_SAMPLE_FMT_NONE) {
  286. char *tail;
  287. ret = strtol(val, &tail, 0);
  288. if (*tail || (unsigned)ret >= AV_SAMPLE_FMT_NB) {
  289. av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\" as sample format\n", val);
  290. return AVERROR(EINVAL);
  291. }
  292. }
  293. }
  294. *(enum AVSampleFormat *)dst = ret;
  295. return 0;
  296. case AV_OPT_TYPE_DURATION:
  297. if (!val) {
  298. *(int64_t *)dst = 0;
  299. return 0;
  300. } else {
  301. if ((ret = av_parse_time(dst, val, 1)) < 0)
  302. av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\" as duration\n", val);
  303. return ret;
  304. }
  305. break;
  306. case AV_OPT_TYPE_COLOR:
  307. if (!val) {
  308. return 0;
  309. } else {
  310. ret = av_parse_color(dst, val, -1, obj);
  311. if (ret < 0)
  312. av_log(obj, AV_LOG_ERROR, "Unable to parse option value \"%s\" as color\n", val);
  313. return ret;
  314. }
  315. }
  316. av_log(obj, AV_LOG_ERROR, "Invalid option type.\n");
  317. return AVERROR(EINVAL);
  318. }
  319. #define OPT_EVAL_NUMBER(name, opttype, vartype)\
  320. int av_opt_eval_ ## name(void *obj, const AVOption *o, const char *val, vartype *name ## _out)\
  321. {\
  322. if (!o || o->type != opttype)\
  323. return AVERROR(EINVAL);\
  324. return set_string_number(obj, obj, o, val, name ## _out);\
  325. }
  326. OPT_EVAL_NUMBER(flags, AV_OPT_TYPE_FLAGS, int)
  327. OPT_EVAL_NUMBER(int, AV_OPT_TYPE_INT, int)
  328. OPT_EVAL_NUMBER(int64, AV_OPT_TYPE_INT64, int64_t)
  329. OPT_EVAL_NUMBER(float, AV_OPT_TYPE_FLOAT, float)
  330. OPT_EVAL_NUMBER(double, AV_OPT_TYPE_DOUBLE, double)
  331. OPT_EVAL_NUMBER(q, AV_OPT_TYPE_RATIONAL, AVRational)
  332. static int set_number(void *obj, const char *name, double num, int den, int64_t intnum,
  333. int search_flags)
  334. {
  335. void *dst, *target_obj;
  336. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  337. if (!o || !target_obj)
  338. return AVERROR_OPTION_NOT_FOUND;
  339. dst = ((uint8_t*)target_obj) + o->offset;
  340. return write_number(obj, o, dst, num, den, intnum);
  341. }
  342. #if FF_API_OLD_AVOPTIONS
  343. const AVOption *av_set_double(void *obj, const char *name, double n)
  344. {
  345. const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
  346. if (set_number(obj, name, n, 1, 1, 0) < 0)
  347. return NULL;
  348. return o;
  349. }
  350. const AVOption *av_set_q(void *obj, const char *name, AVRational n)
  351. {
  352. const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
  353. if (set_number(obj, name, n.num, n.den, 1, 0) < 0)
  354. return NULL;
  355. return o;
  356. }
  357. const AVOption *av_set_int(void *obj, const char *name, int64_t n)
  358. {
  359. const AVOption *o = av_opt_find(obj, name, NULL, 0, 0);
  360. if (set_number(obj, name, 1, 1, n, 0) < 0)
  361. return NULL;
  362. return o;
  363. }
  364. #endif
  365. int av_opt_set_int(void *obj, const char *name, int64_t val, int search_flags)
  366. {
  367. return set_number(obj, name, 1, 1, val, search_flags);
  368. }
  369. int av_opt_set_double(void *obj, const char *name, double val, int search_flags)
  370. {
  371. return set_number(obj, name, val, 1, 1, search_flags);
  372. }
  373. int av_opt_set_q(void *obj, const char *name, AVRational val, int search_flags)
  374. {
  375. return set_number(obj, name, val.num, val.den, 1, search_flags);
  376. }
  377. int av_opt_set_bin(void *obj, const char *name, const uint8_t *val, int len, int search_flags)
  378. {
  379. void *target_obj;
  380. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  381. uint8_t *ptr;
  382. uint8_t **dst;
  383. int *lendst;
  384. if (!o || !target_obj)
  385. return AVERROR_OPTION_NOT_FOUND;
  386. if (o->type != AV_OPT_TYPE_BINARY)
  387. return AVERROR(EINVAL);
  388. ptr = len ? av_malloc(len) : NULL;
  389. if (len && !ptr)
  390. return AVERROR(ENOMEM);
  391. dst = (uint8_t **)(((uint8_t *)target_obj) + o->offset);
  392. lendst = (int *)(dst + 1);
  393. av_free(*dst);
  394. *dst = ptr;
  395. *lendst = len;
  396. if (len)
  397. memcpy(ptr, val, len);
  398. return 0;
  399. }
  400. int av_opt_set_image_size(void *obj, const char *name, int w, int h, int search_flags)
  401. {
  402. void *target_obj;
  403. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  404. if (!o || !target_obj)
  405. return AVERROR_OPTION_NOT_FOUND;
  406. if (o->type != AV_OPT_TYPE_IMAGE_SIZE) {
  407. av_log(obj, AV_LOG_ERROR,
  408. "The value set by option '%s' is not an image size.\n", o->name);
  409. return AVERROR(EINVAL);
  410. }
  411. if (w<0 || h<0) {
  412. av_log(obj, AV_LOG_ERROR,
  413. "Invalid negative size value %dx%d for size '%s'\n", w, h, o->name);
  414. return AVERROR(EINVAL);
  415. }
  416. *(int *)(((uint8_t *)target_obj) + o->offset) = w;
  417. *(int *)(((uint8_t *)target_obj+sizeof(int)) + o->offset) = h;
  418. return 0;
  419. }
  420. int av_opt_set_video_rate(void *obj, const char *name, AVRational val, int search_flags)
  421. {
  422. void *target_obj;
  423. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  424. if (!o || !target_obj)
  425. return AVERROR_OPTION_NOT_FOUND;
  426. if (o->type != AV_OPT_TYPE_VIDEO_RATE) {
  427. av_log(obj, AV_LOG_ERROR,
  428. "The value set by option '%s' is not a video rate.\n", o->name);
  429. return AVERROR(EINVAL);
  430. }
  431. if (val.num <= 0 || val.den <= 0)
  432. return AVERROR(EINVAL);
  433. return set_number(obj, name, val.num, val.den, 1, search_flags);
  434. }
  435. static int set_format(void *obj, const char *name, int fmt, int search_flags,
  436. enum AVOptionType type, const char *desc, int nb_fmts)
  437. {
  438. void *target_obj;
  439. const AVOption *o = av_opt_find2(obj, name, NULL, 0,
  440. search_flags, &target_obj);
  441. int min, max;
  442. const AVClass *class = *(AVClass **)obj;
  443. if (!o || !target_obj)
  444. return AVERROR_OPTION_NOT_FOUND;
  445. if (o->type != type) {
  446. av_log(obj, AV_LOG_ERROR,
  447. "The value set by option '%s' is not a %s format", name, desc);
  448. return AVERROR(EINVAL);
  449. }
  450. #if LIBAVUTIL_VERSION_MAJOR < 53
  451. if (class->version && class->version < AV_VERSION_INT(52, 11, 100)) {
  452. min = -1;
  453. max = nb_fmts-1;
  454. } else
  455. #endif
  456. {
  457. min = FFMIN(o->min, -1);
  458. max = FFMAX(o->max, nb_fmts-1);
  459. }
  460. if (fmt < min || fmt > max) {
  461. av_log(obj, AV_LOG_ERROR,
  462. "Value %d for parameter '%s' out of %s format range [%d - %d]\n",
  463. fmt, name, desc, min, max);
  464. return AVERROR(ERANGE);
  465. }
  466. *(int *)(((uint8_t *)target_obj) + o->offset) = fmt;
  467. return 0;
  468. }
  469. int av_opt_set_pixel_fmt(void *obj, const char *name, enum AVPixelFormat fmt, int search_flags)
  470. {
  471. return set_format(obj, name, fmt, search_flags, AV_OPT_TYPE_PIXEL_FMT, "pixel", AV_PIX_FMT_NB);
  472. }
  473. int av_opt_set_sample_fmt(void *obj, const char *name, enum AVSampleFormat fmt, int search_flags)
  474. {
  475. return set_format(obj, name, fmt, search_flags, AV_OPT_TYPE_SAMPLE_FMT, "sample", AV_SAMPLE_FMT_NB);
  476. }
  477. #if FF_API_OLD_AVOPTIONS
  478. /**
  479. *
  480. * @param buf a buffer which is used for returning non string values as strings, can be NULL
  481. * @param buf_len allocated length in bytes of buf
  482. */
  483. const char *av_get_string(void *obj, const char *name, const AVOption **o_out, char *buf, int buf_len)
  484. {
  485. const AVOption *o = av_opt_find(obj, name, NULL, 0, AV_OPT_SEARCH_CHILDREN);
  486. void *dst;
  487. uint8_t *bin;
  488. int len, i;
  489. if (!o)
  490. return NULL;
  491. if (o->type != AV_OPT_TYPE_STRING && (!buf || !buf_len))
  492. return NULL;
  493. dst= ((uint8_t*)obj) + o->offset;
  494. if (o_out) *o_out= o;
  495. switch (o->type) {
  496. case AV_OPT_TYPE_FLAGS: snprintf(buf, buf_len, "0x%08X",*(int *)dst);break;
  497. case AV_OPT_TYPE_INT: snprintf(buf, buf_len, "%d" , *(int *)dst);break;
  498. case AV_OPT_TYPE_INT64: snprintf(buf, buf_len, "%"PRId64, *(int64_t*)dst);break;
  499. case AV_OPT_TYPE_FLOAT: snprintf(buf, buf_len, "%f" , *(float *)dst);break;
  500. case AV_OPT_TYPE_DOUBLE: snprintf(buf, buf_len, "%f" , *(double *)dst);break;
  501. case AV_OPT_TYPE_RATIONAL: snprintf(buf, buf_len, "%d/%d", ((AVRational*)dst)->num, ((AVRational*)dst)->den);break;
  502. case AV_OPT_TYPE_CONST: snprintf(buf, buf_len, "%f" , o->default_val.dbl);break;
  503. case AV_OPT_TYPE_STRING: return *(void**)dst;
  504. case AV_OPT_TYPE_BINARY:
  505. len = *(int*)(((uint8_t *)dst) + sizeof(uint8_t *));
  506. if (len >= (buf_len + 1)/2) return NULL;
  507. bin = *(uint8_t**)dst;
  508. for (i = 0; i < len; i++) snprintf(buf + i*2, 3, "%02X", bin[i]);
  509. break;
  510. default: return NULL;
  511. }
  512. return buf;
  513. }
  514. #endif
  515. int av_opt_get(void *obj, const char *name, int search_flags, uint8_t **out_val)
  516. {
  517. void *dst, *target_obj;
  518. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  519. uint8_t *bin, buf[128];
  520. int len, i, ret;
  521. int64_t i64;
  522. if (!o || !target_obj || (o->offset<=0 && o->type != AV_OPT_TYPE_CONST))
  523. return AVERROR_OPTION_NOT_FOUND;
  524. dst = (uint8_t*)target_obj + o->offset;
  525. buf[0] = 0;
  526. switch (o->type) {
  527. case AV_OPT_TYPE_FLAGS: ret = snprintf(buf, sizeof(buf), "0x%08X", *(int *)dst);break;
  528. case AV_OPT_TYPE_INT: ret = snprintf(buf, sizeof(buf), "%d" , *(int *)dst);break;
  529. case AV_OPT_TYPE_INT64: ret = snprintf(buf, sizeof(buf), "%"PRId64, *(int64_t*)dst);break;
  530. case AV_OPT_TYPE_FLOAT: ret = snprintf(buf, sizeof(buf), "%f" , *(float *)dst);break;
  531. case AV_OPT_TYPE_DOUBLE: ret = snprintf(buf, sizeof(buf), "%f" , *(double *)dst);break;
  532. case AV_OPT_TYPE_VIDEO_RATE:
  533. case AV_OPT_TYPE_RATIONAL: ret = snprintf(buf, sizeof(buf), "%d/%d", ((AVRational*)dst)->num, ((AVRational*)dst)->den);break;
  534. case AV_OPT_TYPE_CONST: ret = snprintf(buf, sizeof(buf), "%f" , o->default_val.dbl);break;
  535. case AV_OPT_TYPE_STRING:
  536. if (*(uint8_t**)dst)
  537. *out_val = av_strdup(*(uint8_t**)dst);
  538. else
  539. *out_val = av_strdup("");
  540. return 0;
  541. case AV_OPT_TYPE_BINARY:
  542. len = *(int*)(((uint8_t *)dst) + sizeof(uint8_t *));
  543. if ((uint64_t)len*2 + 1 > INT_MAX)
  544. return AVERROR(EINVAL);
  545. if (!(*out_val = av_malloc(len*2 + 1)))
  546. return AVERROR(ENOMEM);
  547. bin = *(uint8_t**)dst;
  548. for (i = 0; i < len; i++)
  549. snprintf(*out_val + i*2, 3, "%02X", bin[i]);
  550. return 0;
  551. case AV_OPT_TYPE_IMAGE_SIZE:
  552. ret = snprintf(buf, sizeof(buf), "%dx%d", ((int *)dst)[0], ((int *)dst)[1]);
  553. break;
  554. case AV_OPT_TYPE_PIXEL_FMT:
  555. ret = snprintf(buf, sizeof(buf), "%s", (char *)av_x_if_null(av_get_pix_fmt_name(*(enum AVPixelFormat *)dst), "none"));
  556. break;
  557. case AV_OPT_TYPE_SAMPLE_FMT:
  558. ret = snprintf(buf, sizeof(buf), "%s", (char *)av_x_if_null(av_get_sample_fmt_name(*(enum AVSampleFormat *)dst), "none"));
  559. break;
  560. case AV_OPT_TYPE_DURATION:
  561. i64 = *(int64_t *)dst;
  562. ret = snprintf(buf, sizeof(buf), "%"PRIi64"d:%02d:%02d.%06d",
  563. i64 / 3600000000, (int)((i64 / 60000000) % 60),
  564. (int)((i64 / 1000000) % 60), (int)(i64 % 1000000));
  565. break;
  566. case AV_OPT_TYPE_COLOR:
  567. ret = snprintf(buf, sizeof(buf), "0x%02x%02x%02x%02x", ((int *)dst)[0], ((int *)dst)[1], ((int *)dst)[2], ((int *)dst)[3]);
  568. break;
  569. default:
  570. return AVERROR(EINVAL);
  571. }
  572. if (ret >= sizeof(buf))
  573. return AVERROR(EINVAL);
  574. *out_val = av_strdup(buf);
  575. return 0;
  576. }
  577. static int get_number(void *obj, const char *name, const AVOption **o_out, double *num, int *den, int64_t *intnum,
  578. int search_flags)
  579. {
  580. void *dst, *target_obj;
  581. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  582. if (!o || !target_obj)
  583. goto error;
  584. dst = ((uint8_t*)target_obj) + o->offset;
  585. if (o_out) *o_out= o;
  586. return read_number(o, dst, num, den, intnum);
  587. error:
  588. *den=*intnum=0;
  589. return -1;
  590. }
  591. #if FF_API_OLD_AVOPTIONS
  592. double av_get_double(void *obj, const char *name, const AVOption **o_out)
  593. {
  594. int64_t intnum=1;
  595. double num=1;
  596. int den=1;
  597. if (get_number(obj, name, o_out, &num, &den, &intnum, 0) < 0)
  598. return NAN;
  599. return num*intnum/den;
  600. }
  601. AVRational av_get_q(void *obj, const char *name, const AVOption **o_out)
  602. {
  603. int64_t intnum=1;
  604. double num=1;
  605. int den=1;
  606. if (get_number(obj, name, o_out, &num, &den, &intnum, 0) < 0)
  607. return (AVRational){0, 0};
  608. if (num == 1.0 && (int)intnum == intnum)
  609. return (AVRational){intnum, den};
  610. else
  611. return av_d2q(num*intnum/den, 1<<24);
  612. }
  613. int64_t av_get_int(void *obj, const char *name, const AVOption **o_out)
  614. {
  615. int64_t intnum=1;
  616. double num=1;
  617. int den=1;
  618. if (get_number(obj, name, o_out, &num, &den, &intnum, 0) < 0)
  619. return -1;
  620. return num*intnum/den;
  621. }
  622. #endif
  623. int av_opt_get_int(void *obj, const char *name, int search_flags, int64_t *out_val)
  624. {
  625. int64_t intnum = 1;
  626. double num = 1;
  627. int ret, den = 1;
  628. if ((ret = get_number(obj, name, NULL, &num, &den, &intnum, search_flags)) < 0)
  629. return ret;
  630. *out_val = num*intnum/den;
  631. return 0;
  632. }
  633. int av_opt_get_double(void *obj, const char *name, int search_flags, double *out_val)
  634. {
  635. int64_t intnum = 1;
  636. double num = 1;
  637. int ret, den = 1;
  638. if ((ret = get_number(obj, name, NULL, &num, &den, &intnum, search_flags)) < 0)
  639. return ret;
  640. *out_val = num*intnum/den;
  641. return 0;
  642. }
  643. int av_opt_get_q(void *obj, const char *name, int search_flags, AVRational *out_val)
  644. {
  645. int64_t intnum = 1;
  646. double num = 1;
  647. int ret, den = 1;
  648. if ((ret = get_number(obj, name, NULL, &num, &den, &intnum, search_flags)) < 0)
  649. return ret;
  650. if (num == 1.0 && (int)intnum == intnum)
  651. *out_val = (AVRational){intnum, den};
  652. else
  653. *out_val = av_d2q(num*intnum/den, 1<<24);
  654. return 0;
  655. }
  656. int av_opt_get_image_size(void *obj, const char *name, int search_flags, int *w_out, int *h_out)
  657. {
  658. void *dst, *target_obj;
  659. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  660. if (!o || !target_obj)
  661. return AVERROR_OPTION_NOT_FOUND;
  662. if (o->type != AV_OPT_TYPE_IMAGE_SIZE) {
  663. av_log(obj, AV_LOG_ERROR,
  664. "The value for option '%s' is not an image size.\n", name);
  665. return AVERROR(EINVAL);
  666. }
  667. dst = ((uint8_t*)target_obj) + o->offset;
  668. if (w_out) *w_out = *(int *)dst;
  669. if (h_out) *h_out = *((int *)dst+1);
  670. return 0;
  671. }
  672. int av_opt_get_video_rate(void *obj, const char *name, int search_flags, AVRational *out_val)
  673. {
  674. int64_t intnum = 1;
  675. double num = 1;
  676. int ret, den = 1;
  677. if ((ret = get_number(obj, name, NULL, &num, &den, &intnum, search_flags)) < 0)
  678. return ret;
  679. if (num == 1.0 && (int)intnum == intnum)
  680. *out_val = (AVRational){intnum, den};
  681. else
  682. *out_val = av_d2q(num*intnum/den, 1<<24);
  683. return 0;
  684. }
  685. static int get_format(void *obj, const char *name, int search_flags, int *out_fmt,
  686. enum AVOptionType type, const char *desc)
  687. {
  688. void *dst, *target_obj;
  689. const AVOption *o = av_opt_find2(obj, name, NULL, 0, search_flags, &target_obj);
  690. if (!o || !target_obj)
  691. return AVERROR_OPTION_NOT_FOUND;
  692. if (o->type != type) {
  693. av_log(obj, AV_LOG_ERROR,
  694. "The value for option '%s' is not a %s format.\n", desc, name);
  695. return AVERROR(EINVAL);
  696. }
  697. dst = ((uint8_t*)target_obj) + o->offset;
  698. *out_fmt = *(int *)dst;
  699. return 0;
  700. }
  701. int av_opt_get_pixel_fmt(void *obj, const char *name, int search_flags, enum AVPixelFormat *out_fmt)
  702. {
  703. return get_format(obj, name, search_flags, out_fmt, AV_OPT_TYPE_PIXEL_FMT, "pixel");
  704. }
  705. int av_opt_get_sample_fmt(void *obj, const char *name, int search_flags, enum AVSampleFormat *out_fmt)
  706. {
  707. return get_format(obj, name, search_flags, out_fmt, AV_OPT_TYPE_SAMPLE_FMT, "sample");
  708. }
  709. int av_opt_flag_is_set(void *obj, const char *field_name, const char *flag_name)
  710. {
  711. const AVOption *field = av_opt_find(obj, field_name, NULL, 0, 0);
  712. const AVOption *flag = av_opt_find(obj, flag_name,
  713. field ? field->unit : NULL, 0, 0);
  714. int64_t res;
  715. if (!field || !flag || flag->type != AV_OPT_TYPE_CONST ||
  716. av_opt_get_int(obj, field_name, 0, &res) < 0)
  717. return 0;
  718. return res & flag->default_val.i64;
  719. }
  720. static void log_value(void *av_log_obj, int level, double d)
  721. {
  722. if (d == INT_MAX) {
  723. av_log(av_log_obj, level, "INT_MAX");
  724. } else if (d == INT_MIN) {
  725. av_log(av_log_obj, level, "INT_MIN");
  726. } else if (d == (double)INT64_MAX) {
  727. av_log(av_log_obj, level, "I64_MAX");
  728. } else if (d == INT64_MIN) {
  729. av_log(av_log_obj, level, "I64_MIN");
  730. } else if (d == FLT_MAX) {
  731. av_log(av_log_obj, level, "FLT_MAX");
  732. } else if (d == FLT_MIN) {
  733. av_log(av_log_obj, level, "FLT_MIN");
  734. } else {
  735. av_log(av_log_obj, level, "%g", d);
  736. }
  737. }
  738. static void opt_list(void *obj, void *av_log_obj, const char *unit,
  739. int req_flags, int rej_flags)
  740. {
  741. const AVOption *opt=NULL;
  742. AVOptionRanges *r;
  743. int i;
  744. while ((opt = av_opt_next(obj, opt))) {
  745. if (!(opt->flags & req_flags) || (opt->flags & rej_flags))
  746. continue;
  747. /* Don't print CONST's on level one.
  748. * Don't print anything but CONST's on level two.
  749. * Only print items from the requested unit.
  750. */
  751. if (!unit && opt->type==AV_OPT_TYPE_CONST)
  752. continue;
  753. else if (unit && opt->type!=AV_OPT_TYPE_CONST)
  754. continue;
  755. else if (unit && opt->type==AV_OPT_TYPE_CONST && strcmp(unit, opt->unit))
  756. continue;
  757. else if (unit && opt->type == AV_OPT_TYPE_CONST)
  758. av_log(av_log_obj, AV_LOG_INFO, " %-15s ", opt->name);
  759. else
  760. av_log(av_log_obj, AV_LOG_INFO, " %s%-17s ",
  761. (opt->flags & AV_OPT_FLAG_FILTERING_PARAM) ? "" : "-",
  762. opt->name);
  763. switch (opt->type) {
  764. case AV_OPT_TYPE_FLAGS:
  765. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<flags>");
  766. break;
  767. case AV_OPT_TYPE_INT:
  768. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<int>");
  769. break;
  770. case AV_OPT_TYPE_INT64:
  771. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<int64>");
  772. break;
  773. case AV_OPT_TYPE_DOUBLE:
  774. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<double>");
  775. break;
  776. case AV_OPT_TYPE_FLOAT:
  777. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<float>");
  778. break;
  779. case AV_OPT_TYPE_STRING:
  780. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<string>");
  781. break;
  782. case AV_OPT_TYPE_RATIONAL:
  783. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<rational>");
  784. break;
  785. case AV_OPT_TYPE_BINARY:
  786. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<binary>");
  787. break;
  788. case AV_OPT_TYPE_IMAGE_SIZE:
  789. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<image_size>");
  790. break;
  791. case AV_OPT_TYPE_VIDEO_RATE:
  792. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<video_rate>");
  793. break;
  794. case AV_OPT_TYPE_PIXEL_FMT:
  795. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<pix_fmt>");
  796. break;
  797. case AV_OPT_TYPE_SAMPLE_FMT:
  798. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<sample_fmt>");
  799. break;
  800. case AV_OPT_TYPE_DURATION:
  801. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<duration>");
  802. break;
  803. case AV_OPT_TYPE_COLOR:
  804. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "<color>");
  805. break;
  806. case AV_OPT_TYPE_CONST:
  807. default:
  808. av_log(av_log_obj, AV_LOG_INFO, "%-12s ", "");
  809. break;
  810. }
  811. av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_ENCODING_PARAM) ? 'E' : '.');
  812. av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_DECODING_PARAM) ? 'D' : '.');
  813. av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_FILTERING_PARAM)? 'F' : '.');
  814. av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_VIDEO_PARAM ) ? 'V' : '.');
  815. av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_AUDIO_PARAM ) ? 'A' : '.');
  816. av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_SUBTITLE_PARAM) ? 'S' : '.');
  817. if (opt->help)
  818. av_log(av_log_obj, AV_LOG_INFO, " %s", opt->help);
  819. if (av_opt_query_ranges(&r, obj, opt->name, AV_OPT_SEARCH_FAKE_OBJ) >= 0) {
  820. switch (opt->type) {
  821. case AV_OPT_TYPE_INT:
  822. case AV_OPT_TYPE_INT64:
  823. case AV_OPT_TYPE_DOUBLE:
  824. case AV_OPT_TYPE_FLOAT:
  825. case AV_OPT_TYPE_RATIONAL:
  826. for (i = 0; i < r->nb_ranges; i++) {
  827. av_log(av_log_obj, AV_LOG_INFO, " (from ");
  828. log_value(av_log_obj, AV_LOG_INFO, r->range[i]->value_min);
  829. av_log(av_log_obj, AV_LOG_INFO, " to ");
  830. log_value(av_log_obj, AV_LOG_INFO, r->range[i]->value_max);
  831. av_log(av_log_obj, AV_LOG_INFO, ")");
  832. }
  833. break;
  834. }
  835. av_opt_freep_ranges(&r);
  836. }
  837. if (opt->type != AV_OPT_TYPE_CONST &&
  838. opt->type != AV_OPT_TYPE_BINARY &&
  839. !((opt->type == AV_OPT_TYPE_COLOR ||
  840. opt->type == AV_OPT_TYPE_IMAGE_SIZE ||
  841. opt->type == AV_OPT_TYPE_STRING ||
  842. opt->type == AV_OPT_TYPE_VIDEO_RATE) &&
  843. !opt->default_val.str)) {
  844. av_log(av_log_obj, AV_LOG_INFO, " (default ");
  845. switch (opt->type) {
  846. case AV_OPT_TYPE_FLAGS:
  847. av_log(av_log_obj, AV_LOG_INFO, "%0llX", opt->default_val.i64);
  848. break;
  849. case AV_OPT_TYPE_DURATION:
  850. case AV_OPT_TYPE_INT:
  851. case AV_OPT_TYPE_INT64:
  852. log_value(av_log_obj, AV_LOG_INFO, opt->default_val.i64);
  853. break;
  854. case AV_OPT_TYPE_DOUBLE:
  855. case AV_OPT_TYPE_FLOAT:
  856. log_value(av_log_obj, AV_LOG_INFO, opt->default_val.dbl);
  857. break;
  858. case AV_OPT_TYPE_RATIONAL: {
  859. AVRational q = av_d2q(opt->default_val.dbl, INT_MAX);
  860. av_log(av_log_obj, AV_LOG_INFO, "%d/%d", q.num, q.den); }
  861. break;
  862. case AV_OPT_TYPE_COLOR:
  863. case AV_OPT_TYPE_IMAGE_SIZE:
  864. case AV_OPT_TYPE_STRING:
  865. case AV_OPT_TYPE_VIDEO_RATE:
  866. av_log(av_log_obj, AV_LOG_INFO, "\"%s\"", opt->default_val.str);
  867. }
  868. av_log(av_log_obj, AV_LOG_INFO, ")");
  869. }
  870. av_log(av_log_obj, AV_LOG_INFO, "\n");
  871. if (opt->unit && opt->type != AV_OPT_TYPE_CONST) {
  872. opt_list(obj, av_log_obj, opt->unit, req_flags, rej_flags);
  873. }
  874. }
  875. }
  876. int av_opt_show2(void *obj, void *av_log_obj, int req_flags, int rej_flags)
  877. {
  878. if (!obj)
  879. return -1;
  880. av_log(av_log_obj, AV_LOG_INFO, "%s AVOptions:\n", (*(AVClass**)obj)->class_name);
  881. opt_list(obj, av_log_obj, NULL, req_flags, rej_flags);
  882. return 0;
  883. }
  884. void av_opt_set_defaults(void *s)
  885. {
  886. #if FF_API_OLD_AVOPTIONS
  887. av_opt_set_defaults2(s, 0, 0);
  888. }
  889. void av_opt_set_defaults2(void *s, int mask, int flags)
  890. {
  891. #endif
  892. const AVClass *class = *(AVClass **)s;
  893. const AVOption *opt = NULL;
  894. while ((opt = av_opt_next(s, opt)) != NULL) {
  895. #if FF_API_OLD_AVOPTIONS
  896. if ((opt->flags & mask) != flags)
  897. continue;
  898. #endif
  899. switch (opt->type) {
  900. case AV_OPT_TYPE_CONST:
  901. /* Nothing to be done here */
  902. break;
  903. case AV_OPT_TYPE_FLAGS:
  904. case AV_OPT_TYPE_INT:
  905. case AV_OPT_TYPE_INT64:
  906. case AV_OPT_TYPE_DURATION:
  907. av_opt_set_int(s, opt->name, opt->default_val.i64, 0);
  908. break;
  909. case AV_OPT_TYPE_DOUBLE:
  910. case AV_OPT_TYPE_FLOAT: {
  911. double val;
  912. val = opt->default_val.dbl;
  913. av_opt_set_double(s, opt->name, val, 0);
  914. }
  915. break;
  916. case AV_OPT_TYPE_RATIONAL: {
  917. AVRational val;
  918. val = av_d2q(opt->default_val.dbl, INT_MAX);
  919. av_opt_set_q(s, opt->name, val, 0);
  920. }
  921. break;
  922. case AV_OPT_TYPE_COLOR:
  923. case AV_OPT_TYPE_STRING:
  924. case AV_OPT_TYPE_IMAGE_SIZE:
  925. case AV_OPT_TYPE_VIDEO_RATE:
  926. av_opt_set(s, opt->name, opt->default_val.str, 0);
  927. break;
  928. case AV_OPT_TYPE_PIXEL_FMT:
  929. #if LIBAVUTIL_VERSION_MAJOR < 53
  930. if (class->version && class->version < AV_VERSION_INT(52, 10, 100))
  931. av_opt_set(s, opt->name, opt->default_val.str, 0);
  932. else
  933. #endif
  934. av_opt_set_pixel_fmt(s, opt->name, opt->default_val.i64, 0);
  935. break;
  936. case AV_OPT_TYPE_SAMPLE_FMT:
  937. #if LIBAVUTIL_VERSION_MAJOR < 53
  938. if (class->version && class->version < AV_VERSION_INT(52, 10, 100))
  939. av_opt_set(s, opt->name, opt->default_val.str, 0);
  940. else
  941. #endif
  942. av_opt_set_sample_fmt(s, opt->name, opt->default_val.i64, 0);
  943. break;
  944. case AV_OPT_TYPE_BINARY:
  945. /* Cannot set default for binary */
  946. break;
  947. default:
  948. av_log(s, AV_LOG_DEBUG, "AVOption type %d of option %s not implemented yet\n", opt->type, opt->name);
  949. }
  950. }
  951. }
  952. /**
  953. * Store the value in the field in ctx that is named like key.
  954. * ctx must be an AVClass context, storing is done using AVOptions.
  955. *
  956. * @param buf the string to parse, buf will be updated to point at the
  957. * separator just after the parsed key/value pair
  958. * @param key_val_sep a 0-terminated list of characters used to
  959. * separate key from value
  960. * @param pairs_sep a 0-terminated list of characters used to separate
  961. * two pairs from each other
  962. * @return 0 if the key/value pair has been successfully parsed and
  963. * set, or a negative value corresponding to an AVERROR code in case
  964. * of error:
  965. * AVERROR(EINVAL) if the key/value pair cannot be parsed,
  966. * the error code issued by av_opt_set() if the key/value pair
  967. * cannot be set
  968. */
  969. static int parse_key_value_pair(void *ctx, const char **buf,
  970. const char *key_val_sep, const char *pairs_sep)
  971. {
  972. char *key = av_get_token(buf, key_val_sep);
  973. char *val;
  974. int ret;
  975. if (!key)
  976. return AVERROR(ENOMEM);
  977. if (*key && strspn(*buf, key_val_sep)) {
  978. (*buf)++;
  979. val = av_get_token(buf, pairs_sep);
  980. if (!val) {
  981. av_freep(&key);
  982. return AVERROR(ENOMEM);
  983. }
  984. } else {
  985. av_log(ctx, AV_LOG_ERROR, "Missing key or no key/value separator found after key '%s'\n", key);
  986. av_free(key);
  987. return AVERROR(EINVAL);
  988. }
  989. av_log(ctx, AV_LOG_DEBUG, "Setting entry with key '%s' to value '%s'\n", key, val);
  990. ret = av_opt_set(ctx, key, val, AV_OPT_SEARCH_CHILDREN);
  991. if (ret == AVERROR_OPTION_NOT_FOUND)
  992. av_log(ctx, AV_LOG_ERROR, "Key '%s' not found.\n", key);
  993. av_free(key);
  994. av_free(val);
  995. return ret;
  996. }
  997. int av_set_options_string(void *ctx, const char *opts,
  998. const char *key_val_sep, const char *pairs_sep)
  999. {
  1000. int ret, count = 0;
  1001. if (!opts)
  1002. return 0;
  1003. while (*opts) {
  1004. if ((ret = parse_key_value_pair(ctx, &opts, key_val_sep, pairs_sep)) < 0)
  1005. return ret;
  1006. count++;
  1007. if (*opts)
  1008. opts++;
  1009. }
  1010. return count;
  1011. }
  1012. #define WHITESPACES " \n\t"
  1013. static int is_key_char(char c)
  1014. {
  1015. return (unsigned)((c | 32) - 'a') < 26 ||
  1016. (unsigned)(c - '0') < 10 ||
  1017. c == '-' || c == '_' || c == '/' || c == '.';
  1018. }
  1019. /**
  1020. * Read a key from a string.
  1021. *
  1022. * The key consists of is_key_char characters and must be terminated by a
  1023. * character from the delim string; spaces are ignored.
  1024. *
  1025. * @return 0 for success (even with ellipsis), <0 for failure
  1026. */
  1027. static int get_key(const char **ropts, const char *delim, char **rkey)
  1028. {
  1029. const char *opts = *ropts;
  1030. const char *key_start, *key_end;
  1031. key_start = opts += strspn(opts, WHITESPACES);
  1032. while (is_key_char(*opts))
  1033. opts++;
  1034. key_end = opts;
  1035. opts += strspn(opts, WHITESPACES);
  1036. if (!*opts || !strchr(delim, *opts))
  1037. return AVERROR(EINVAL);
  1038. opts++;
  1039. if (!(*rkey = av_malloc(key_end - key_start + 1)))
  1040. return AVERROR(ENOMEM);
  1041. memcpy(*rkey, key_start, key_end - key_start);
  1042. (*rkey)[key_end - key_start] = 0;
  1043. *ropts = opts;
  1044. return 0;
  1045. }
  1046. int av_opt_get_key_value(const char **ropts,
  1047. const char *key_val_sep, const char *pairs_sep,
  1048. unsigned flags,
  1049. char **rkey, char **rval)
  1050. {
  1051. int ret;
  1052. char *key = NULL, *val;
  1053. const char *opts = *ropts;
  1054. if ((ret = get_key(&opts, key_val_sep, &key)) < 0 &&
  1055. !(flags & AV_OPT_FLAG_IMPLICIT_KEY))
  1056. return AVERROR(EINVAL);
  1057. if (!(val = av_get_token(&opts, pairs_sep))) {
  1058. av_free(key);
  1059. return AVERROR(ENOMEM);
  1060. }
  1061. *ropts = opts;
  1062. *rkey = key;
  1063. *rval = val;
  1064. return 0;
  1065. }
  1066. int av_opt_set_from_string(void *ctx, const char *opts,
  1067. const char *const *shorthand,
  1068. const char *key_val_sep, const char *pairs_sep)
  1069. {
  1070. int ret, count = 0;
  1071. const char *dummy_shorthand = NULL;
  1072. char *av_uninit(parsed_key), *av_uninit(value);
  1073. const char *key;
  1074. if (!opts)
  1075. return 0;
  1076. if (!shorthand)
  1077. shorthand = &dummy_shorthand;
  1078. while (*opts) {
  1079. ret = av_opt_get_key_value(&opts, key_val_sep, pairs_sep,
  1080. *shorthand ? AV_OPT_FLAG_IMPLICIT_KEY : 0,
  1081. &parsed_key, &value);
  1082. if (ret < 0) {
  1083. if (ret == AVERROR(EINVAL))
  1084. av_log(ctx, AV_LOG_ERROR, "No option name near '%s'\n", opts);
  1085. else
  1086. av_log(ctx, AV_LOG_ERROR, "Unable to parse '%s': %s\n", opts,
  1087. av_err2str(ret));
  1088. return ret;
  1089. }
  1090. if (*opts)
  1091. opts++;
  1092. if (parsed_key) {
  1093. key = parsed_key;
  1094. while (*shorthand) /* discard all remaining shorthand */
  1095. shorthand++;
  1096. } else {
  1097. key = *(shorthand++);
  1098. }
  1099. av_log(ctx, AV_LOG_DEBUG, "Setting '%s' to value '%s'\n", key, value);
  1100. if ((ret = av_opt_set(ctx, key, value, 0)) < 0) {
  1101. if (ret == AVERROR_OPTION_NOT_FOUND)
  1102. av_log(ctx, AV_LOG_ERROR, "Option '%s' not found\n", key);
  1103. av_free(value);
  1104. av_free(parsed_key);
  1105. return ret;
  1106. }
  1107. av_free(value);
  1108. av_free(parsed_key);
  1109. count++;
  1110. }
  1111. return count;
  1112. }
  1113. void av_opt_free(void *obj)
  1114. {
  1115. const AVOption *o = NULL;
  1116. while ((o = av_opt_next(obj, o)))
  1117. if (o->type == AV_OPT_TYPE_STRING || o->type == AV_OPT_TYPE_BINARY)
  1118. av_freep((uint8_t *)obj + o->offset);
  1119. }
  1120. int av_opt_set_dict(void *obj, AVDictionary **options)
  1121. {
  1122. AVDictionaryEntry *t = NULL;
  1123. AVDictionary *tmp = NULL;
  1124. int ret = 0;
  1125. while ((t = av_dict_get(*options, "", t, AV_DICT_IGNORE_SUFFIX))) {
  1126. ret = av_opt_set(obj, t->key, t->value, 0);
  1127. if (ret == AVERROR_OPTION_NOT_FOUND)
  1128. av_dict_set(&tmp, t->key, t->value, 0);
  1129. else if (ret < 0) {
  1130. av_log(obj, AV_LOG_ERROR, "Error setting option %s to value %s.\n", t->key, t->value);
  1131. break;
  1132. }
  1133. ret = 0;
  1134. }
  1135. av_dict_free(options);
  1136. *options = tmp;
  1137. return ret;
  1138. }
  1139. const AVOption *av_opt_find(void *obj, const char *name, const char *unit,
  1140. int opt_flags, int search_flags)
  1141. {
  1142. return av_opt_find2(obj, name, unit, opt_flags, search_flags, NULL);
  1143. }
  1144. const AVOption *av_opt_find2(void *obj, const char *name, const char *unit,
  1145. int opt_flags, int search_flags, void **target_obj)
  1146. {
  1147. const AVClass *c;
  1148. const AVOption *o = NULL;
  1149. if(!obj)
  1150. return NULL;
  1151. c= *(AVClass**)obj;
  1152. if (search_flags & AV_OPT_SEARCH_CHILDREN) {
  1153. if (search_flags & AV_OPT_SEARCH_FAKE_OBJ) {
  1154. const AVClass *child = NULL;
  1155. while (child = av_opt_child_class_next(c, child))
  1156. if (o = av_opt_find2(&child, name, unit, opt_flags, search_flags, NULL))
  1157. return o;
  1158. } else {
  1159. void *child = NULL;
  1160. while (child = av_opt_child_next(obj, child))
  1161. if (o = av_opt_find2(child, name, unit, opt_flags, search_flags, target_obj))
  1162. return o;
  1163. }
  1164. }
  1165. while (o = av_opt_next(obj, o)) {
  1166. if (!strcmp(o->name, name) && (o->flags & opt_flags) == opt_flags &&
  1167. ((!unit && o->type != AV_OPT_TYPE_CONST) ||
  1168. (unit && o->type == AV_OPT_TYPE_CONST && o->unit && !strcmp(o->unit, unit)))) {
  1169. if (target_obj) {
  1170. if (!(search_flags & AV_OPT_SEARCH_FAKE_OBJ))
  1171. *target_obj = obj;
  1172. else
  1173. *target_obj = NULL;
  1174. }
  1175. return o;
  1176. }
  1177. }
  1178. return NULL;
  1179. }
  1180. void *av_opt_child_next(void *obj, void *prev)
  1181. {
  1182. const AVClass *c = *(AVClass**)obj;
  1183. if (c->child_next)
  1184. return c->child_next(obj, prev);
  1185. return NULL;
  1186. }
  1187. const AVClass *av_opt_child_class_next(const AVClass *parent, const AVClass *prev)
  1188. {
  1189. if (parent->child_class_next)
  1190. return parent->child_class_next(prev);
  1191. return NULL;
  1192. }
  1193. void *av_opt_ptr(const AVClass *class, void *obj, const char *name)
  1194. {
  1195. const AVOption *opt= av_opt_find2(&class, name, NULL, 0, AV_OPT_SEARCH_FAKE_OBJ, NULL);
  1196. if(!opt)
  1197. return NULL;
  1198. return (uint8_t*)obj + opt->offset;
  1199. }
  1200. int av_opt_query_ranges(AVOptionRanges **ranges_arg, void *obj, const char *key, int flags)
  1201. {
  1202. const AVClass *c = *(AVClass**)obj;
  1203. int (*callback)(AVOptionRanges **, void *obj, const char *key, int flags) = NULL;
  1204. if (c->version > (52 << 16 | 11 << 8))
  1205. callback = c->query_ranges;
  1206. if (!callback)
  1207. callback = av_opt_query_ranges_default;
  1208. return callback(ranges_arg, obj, key, flags);
  1209. }
  1210. int av_opt_query_ranges_default(AVOptionRanges **ranges_arg, void *obj, const char *key, int flags)
  1211. {
  1212. AVOptionRanges *ranges = av_mallocz(sizeof(*ranges));
  1213. AVOptionRange **range_array = av_mallocz(sizeof(void*));
  1214. AVOptionRange *range = av_mallocz(sizeof(*range));
  1215. const AVOption *field = av_opt_find(obj, key, NULL, 0, flags);
  1216. int ret;
  1217. *ranges_arg = NULL;
  1218. if (!ranges || !range || !range_array || !field) {
  1219. ret = AVERROR(ENOMEM);
  1220. goto fail;
  1221. }
  1222. ranges->range = range_array;
  1223. ranges->range[0] = range;
  1224. ranges->nb_ranges = 1;
  1225. range->is_range = 1;
  1226. range->value_min = field->min;
  1227. range->value_max = field->max;
  1228. switch (field->type) {
  1229. case AV_OPT_TYPE_INT:
  1230. case AV_OPT_TYPE_INT64:
  1231. case AV_OPT_TYPE_PIXEL_FMT:
  1232. case AV_OPT_TYPE_SAMPLE_FMT:
  1233. case AV_OPT_TYPE_FLOAT:
  1234. case AV_OPT_TYPE_DOUBLE:
  1235. case AV_OPT_TYPE_DURATION:
  1236. case AV_OPT_TYPE_COLOR:
  1237. break;
  1238. case AV_OPT_TYPE_STRING:
  1239. range->component_min = 0;
  1240. range->component_max = 0x10FFFF; // max unicode value
  1241. range->value_min = -1;
  1242. range->value_max = INT_MAX;
  1243. break;
  1244. case AV_OPT_TYPE_RATIONAL:
  1245. range->component_min = INT_MIN;
  1246. range->component_max = INT_MAX;
  1247. break;
  1248. case AV_OPT_TYPE_IMAGE_SIZE:
  1249. range->component_min = 0;
  1250. range->component_max = INT_MAX/128/8;
  1251. range->value_min = 0;
  1252. range->value_max = INT_MAX/8;
  1253. break;
  1254. case AV_OPT_TYPE_VIDEO_RATE:
  1255. range->component_min = 1;
  1256. range->component_max = INT_MAX;
  1257. range->value_min = 1;
  1258. range->value_max = INT_MAX;
  1259. break;
  1260. default:
  1261. ret = AVERROR(ENOSYS);
  1262. goto fail;
  1263. }
  1264. *ranges_arg = ranges;
  1265. return 0;
  1266. fail:
  1267. av_free(ranges);
  1268. av_free(range);
  1269. av_free(range_array);
  1270. return ret;
  1271. }
  1272. void av_opt_freep_ranges(AVOptionRanges **rangesp)
  1273. {
  1274. int i;
  1275. AVOptionRanges *ranges = *rangesp;
  1276. for (i = 0; i < ranges->nb_ranges; i++) {
  1277. AVOptionRange *range = ranges->range[i];
  1278. av_freep(&range->str);
  1279. av_freep(&ranges->range[i]);
  1280. }
  1281. av_freep(&ranges->range);
  1282. av_freep(rangesp);
  1283. }
  1284. #ifdef TEST
  1285. typedef struct TestContext
  1286. {
  1287. const AVClass *class;
  1288. int num;
  1289. int toggle;
  1290. char *string;
  1291. int flags;
  1292. AVRational rational;
  1293. AVRational video_rate;
  1294. int w, h;
  1295. enum AVPixelFormat pix_fmt;
  1296. enum AVSampleFormat sample_fmt;
  1297. int64_t duration;
  1298. uint8_t color[4];
  1299. } TestContext;
  1300. #define OFFSET(x) offsetof(TestContext, x)
  1301. #define TEST_FLAG_COOL 01
  1302. #define TEST_FLAG_LAME 02
  1303. #define TEST_FLAG_MU 04
  1304. static const AVOption test_options[]= {
  1305. {"num", "set num", OFFSET(num), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 100 },
  1306. {"toggle", "set toggle", OFFSET(toggle), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1 },
  1307. {"rational", "set rational", OFFSET(rational), AV_OPT_TYPE_RATIONAL, {.dbl = 0}, 0, 10 },
  1308. {"string", "set string", OFFSET(string), AV_OPT_TYPE_STRING, {.str = "default"}, CHAR_MIN, CHAR_MAX },
  1309. {"flags", "set flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64 = 0}, 0, INT_MAX, 0, "flags" },
  1310. {"cool", "set cool flag ", 0, AV_OPT_TYPE_CONST, {.i64 = TEST_FLAG_COOL}, INT_MIN, INT_MAX, 0, "flags" },
  1311. {"lame", "set lame flag ", 0, AV_OPT_TYPE_CONST, {.i64 = TEST_FLAG_LAME}, INT_MIN, INT_MAX, 0, "flags" },
  1312. {"mu", "set mu flag ", 0, AV_OPT_TYPE_CONST, {.i64 = TEST_FLAG_MU}, INT_MIN, INT_MAX, 0, "flags" },
  1313. {"size", "set size", OFFSET(w), AV_OPT_TYPE_IMAGE_SIZE,{0}, 0, 0 },
  1314. {"pix_fmt", "set pixfmt", OFFSET(pix_fmt), AV_OPT_TYPE_PIXEL_FMT, {.i64 = AV_PIX_FMT_NONE}, -1, AV_PIX_FMT_NB-1},
  1315. {"sample_fmt", "set samplefmt", OFFSET(sample_fmt), AV_OPT_TYPE_SAMPLE_FMT, {.i64 = AV_SAMPLE_FMT_NONE}, -1, AV_SAMPLE_FMT_NB-1},
  1316. {"video_rate", "set videorate", OFFSET(video_rate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, 0 },
  1317. {"duration", "set duration", OFFSET(duration), AV_OPT_TYPE_DURATION, {.i64 = 0}, 0, INT64_MAX},
  1318. {"color", "set color", OFFSET(color), AV_OPT_TYPE_COLOR, {.str = "pink"}, 0, 0},
  1319. {NULL},
  1320. };
  1321. static const char *test_get_name(void *ctx)
  1322. {
  1323. return "test";
  1324. }
  1325. static const AVClass test_class = {
  1326. "TestContext",
  1327. test_get_name,
  1328. test_options
  1329. };
  1330. int main(void)
  1331. {
  1332. int i;
  1333. printf("\nTesting av_set_options_string()\n");
  1334. {
  1335. TestContext test_ctx = { 0 };
  1336. const char *options[] = {
  1337. "",
  1338. ":",
  1339. "=",
  1340. "foo=:",
  1341. ":=foo",
  1342. "=foo",
  1343. "foo=",
  1344. "foo",
  1345. "foo=val",
  1346. "foo==val",
  1347. "toggle=:",
  1348. "string=:",
  1349. "toggle=1 : foo",
  1350. "toggle=100",
  1351. "toggle==1",
  1352. "flags=+mu-lame : num=42: toggle=0",
  1353. "num=42 : string=blahblah",
  1354. "rational=0 : rational=1/2 : rational=1/-1",
  1355. "rational=-1/0",
  1356. "size=1024x768",
  1357. "size=pal",
  1358. "size=bogus",
  1359. "pix_fmt=yuv420p",
  1360. "pix_fmt=2",
  1361. "pix_fmt=bogus",
  1362. "sample_fmt=s16",
  1363. "sample_fmt=2",
  1364. "sample_fmt=bogus",
  1365. "video_rate=pal",
  1366. "video_rate=25",
  1367. "video_rate=30000/1001",
  1368. "video_rate=30/1.001",
  1369. "video_rate=bogus",
  1370. "duration=bogus",
  1371. "duration=123.45",
  1372. "duration=1\\:23\\:45.67",
  1373. "color=blue",
  1374. "color=0x223300",
  1375. "color=0x42FF07AA",
  1376. };
  1377. test_ctx.class = &test_class;
  1378. av_opt_set_defaults(&test_ctx);
  1379. av_log_set_level(AV_LOG_DEBUG);
  1380. for (i=0; i < FF_ARRAY_ELEMS(options); i++) {
  1381. av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]);
  1382. if (av_set_options_string(&test_ctx, options[i], "=", ":") < 0)
  1383. av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]);
  1384. printf("\n");
  1385. }
  1386. av_freep(&test_ctx.string);
  1387. }
  1388. printf("\nTesting av_opt_set_from_string()\n");
  1389. {
  1390. TestContext test_ctx = { 0 };
  1391. const char *options[] = {
  1392. "",
  1393. "5",
  1394. "5:hello",
  1395. "5:hello:size=pal",
  1396. "5:size=pal:hello",
  1397. ":",
  1398. "=",
  1399. " 5 : hello : size = pal ",
  1400. "a_very_long_option_name_that_will_need_to_be_ellipsized_around_here=42"
  1401. };
  1402. const char *shorthand[] = { "num", "string", NULL };
  1403. test_ctx.class = &test_class;
  1404. av_opt_set_defaults(&test_ctx);
  1405. test_ctx.string = av_strdup("default");
  1406. av_log_set_level(AV_LOG_DEBUG);
  1407. for (i=0; i < FF_ARRAY_ELEMS(options); i++) {
  1408. av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]);
  1409. if (av_opt_set_from_string(&test_ctx, options[i], shorthand, "=", ":") < 0)
  1410. av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]);
  1411. printf("\n");
  1412. }
  1413. av_freep(&test_ctx.string);
  1414. }
  1415. return 0;
  1416. }
  1417. #endif