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.

1557 lines
50KB

  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. av_log(av_log_obj, AV_LOG_INFO, "\n");
  838. if (opt->unit && opt->type != AV_OPT_TYPE_CONST) {
  839. opt_list(obj, av_log_obj, opt->unit, req_flags, rej_flags);
  840. }
  841. }
  842. }
  843. int av_opt_show2(void *obj, void *av_log_obj, int req_flags, int rej_flags)
  844. {
  845. if (!obj)
  846. return -1;
  847. av_log(av_log_obj, AV_LOG_INFO, "%s AVOptions:\n", (*(AVClass**)obj)->class_name);
  848. opt_list(obj, av_log_obj, NULL, req_flags, rej_flags);
  849. return 0;
  850. }
  851. void av_opt_set_defaults(void *s)
  852. {
  853. #if FF_API_OLD_AVOPTIONS
  854. av_opt_set_defaults2(s, 0, 0);
  855. }
  856. void av_opt_set_defaults2(void *s, int mask, int flags)
  857. {
  858. #endif
  859. const AVClass *class = *(AVClass **)s;
  860. const AVOption *opt = NULL;
  861. while ((opt = av_opt_next(s, opt)) != NULL) {
  862. #if FF_API_OLD_AVOPTIONS
  863. if ((opt->flags & mask) != flags)
  864. continue;
  865. #endif
  866. switch (opt->type) {
  867. case AV_OPT_TYPE_CONST:
  868. /* Nothing to be done here */
  869. break;
  870. case AV_OPT_TYPE_FLAGS:
  871. case AV_OPT_TYPE_INT:
  872. case AV_OPT_TYPE_INT64:
  873. case AV_OPT_TYPE_DURATION:
  874. av_opt_set_int(s, opt->name, opt->default_val.i64, 0);
  875. break;
  876. case AV_OPT_TYPE_DOUBLE:
  877. case AV_OPT_TYPE_FLOAT: {
  878. double val;
  879. val = opt->default_val.dbl;
  880. av_opt_set_double(s, opt->name, val, 0);
  881. }
  882. break;
  883. case AV_OPT_TYPE_RATIONAL: {
  884. AVRational val;
  885. val = av_d2q(opt->default_val.dbl, INT_MAX);
  886. av_opt_set_q(s, opt->name, val, 0);
  887. }
  888. break;
  889. case AV_OPT_TYPE_COLOR:
  890. case AV_OPT_TYPE_STRING:
  891. case AV_OPT_TYPE_IMAGE_SIZE:
  892. case AV_OPT_TYPE_VIDEO_RATE:
  893. av_opt_set(s, opt->name, opt->default_val.str, 0);
  894. break;
  895. case AV_OPT_TYPE_PIXEL_FMT:
  896. #if LIBAVUTIL_VERSION_MAJOR < 53
  897. if (class->version && class->version < AV_VERSION_INT(52, 10, 100))
  898. av_opt_set(s, opt->name, opt->default_val.str, 0);
  899. else
  900. #endif
  901. av_opt_set_pixel_fmt(s, opt->name, opt->default_val.i64, 0);
  902. break;
  903. case AV_OPT_TYPE_SAMPLE_FMT:
  904. #if LIBAVUTIL_VERSION_MAJOR < 53
  905. if (class->version && class->version < AV_VERSION_INT(52, 10, 100))
  906. av_opt_set(s, opt->name, opt->default_val.str, 0);
  907. else
  908. #endif
  909. av_opt_set_sample_fmt(s, opt->name, opt->default_val.i64, 0);
  910. break;
  911. case AV_OPT_TYPE_BINARY:
  912. /* Cannot set default for binary */
  913. break;
  914. default:
  915. av_log(s, AV_LOG_DEBUG, "AVOption type %d of option %s not implemented yet\n", opt->type, opt->name);
  916. }
  917. }
  918. }
  919. /**
  920. * Store the value in the field in ctx that is named like key.
  921. * ctx must be an AVClass context, storing is done using AVOptions.
  922. *
  923. * @param buf the string to parse, buf will be updated to point at the
  924. * separator just after the parsed key/value pair
  925. * @param key_val_sep a 0-terminated list of characters used to
  926. * separate key from value
  927. * @param pairs_sep a 0-terminated list of characters used to separate
  928. * two pairs from each other
  929. * @return 0 if the key/value pair has been successfully parsed and
  930. * set, or a negative value corresponding to an AVERROR code in case
  931. * of error:
  932. * AVERROR(EINVAL) if the key/value pair cannot be parsed,
  933. * the error code issued by av_opt_set() if the key/value pair
  934. * cannot be set
  935. */
  936. static int parse_key_value_pair(void *ctx, const char **buf,
  937. const char *key_val_sep, const char *pairs_sep)
  938. {
  939. char *key = av_get_token(buf, key_val_sep);
  940. char *val;
  941. int ret;
  942. if (!key)
  943. return AVERROR(ENOMEM);
  944. if (*key && strspn(*buf, key_val_sep)) {
  945. (*buf)++;
  946. val = av_get_token(buf, pairs_sep);
  947. if (!val) {
  948. av_freep(&key);
  949. return AVERROR(ENOMEM);
  950. }
  951. } else {
  952. av_log(ctx, AV_LOG_ERROR, "Missing key or no key/value separator found after key '%s'\n", key);
  953. av_free(key);
  954. return AVERROR(EINVAL);
  955. }
  956. av_log(ctx, AV_LOG_DEBUG, "Setting entry with key '%s' to value '%s'\n", key, val);
  957. ret = av_opt_set(ctx, key, val, AV_OPT_SEARCH_CHILDREN);
  958. if (ret == AVERROR_OPTION_NOT_FOUND)
  959. av_log(ctx, AV_LOG_ERROR, "Key '%s' not found.\n", key);
  960. av_free(key);
  961. av_free(val);
  962. return ret;
  963. }
  964. int av_set_options_string(void *ctx, const char *opts,
  965. const char *key_val_sep, const char *pairs_sep)
  966. {
  967. int ret, count = 0;
  968. if (!opts)
  969. return 0;
  970. while (*opts) {
  971. if ((ret = parse_key_value_pair(ctx, &opts, key_val_sep, pairs_sep)) < 0)
  972. return ret;
  973. count++;
  974. if (*opts)
  975. opts++;
  976. }
  977. return count;
  978. }
  979. #define WHITESPACES " \n\t"
  980. static int is_key_char(char c)
  981. {
  982. return (unsigned)((c | 32) - 'a') < 26 ||
  983. (unsigned)(c - '0') < 10 ||
  984. c == '-' || c == '_' || c == '/' || c == '.';
  985. }
  986. /**
  987. * Read a key from a string.
  988. *
  989. * The key consists of is_key_char characters and must be terminated by a
  990. * character from the delim string; spaces are ignored.
  991. *
  992. * @return 0 for success (even with ellipsis), <0 for failure
  993. */
  994. static int get_key(const char **ropts, const char *delim, char **rkey)
  995. {
  996. const char *opts = *ropts;
  997. const char *key_start, *key_end;
  998. key_start = opts += strspn(opts, WHITESPACES);
  999. while (is_key_char(*opts))
  1000. opts++;
  1001. key_end = opts;
  1002. opts += strspn(opts, WHITESPACES);
  1003. if (!*opts || !strchr(delim, *opts))
  1004. return AVERROR(EINVAL);
  1005. opts++;
  1006. if (!(*rkey = av_malloc(key_end - key_start + 1)))
  1007. return AVERROR(ENOMEM);
  1008. memcpy(*rkey, key_start, key_end - key_start);
  1009. (*rkey)[key_end - key_start] = 0;
  1010. *ropts = opts;
  1011. return 0;
  1012. }
  1013. int av_opt_get_key_value(const char **ropts,
  1014. const char *key_val_sep, const char *pairs_sep,
  1015. unsigned flags,
  1016. char **rkey, char **rval)
  1017. {
  1018. int ret;
  1019. char *key = NULL, *val;
  1020. const char *opts = *ropts;
  1021. if ((ret = get_key(&opts, key_val_sep, &key)) < 0 &&
  1022. !(flags & AV_OPT_FLAG_IMPLICIT_KEY))
  1023. return AVERROR(EINVAL);
  1024. if (!(val = av_get_token(&opts, pairs_sep))) {
  1025. av_free(key);
  1026. return AVERROR(ENOMEM);
  1027. }
  1028. *ropts = opts;
  1029. *rkey = key;
  1030. *rval = val;
  1031. return 0;
  1032. }
  1033. int av_opt_set_from_string(void *ctx, const char *opts,
  1034. const char *const *shorthand,
  1035. const char *key_val_sep, const char *pairs_sep)
  1036. {
  1037. int ret, count = 0;
  1038. const char *dummy_shorthand = NULL;
  1039. char *av_uninit(parsed_key), *av_uninit(value);
  1040. const char *key;
  1041. if (!opts)
  1042. return 0;
  1043. if (!shorthand)
  1044. shorthand = &dummy_shorthand;
  1045. while (*opts) {
  1046. ret = av_opt_get_key_value(&opts, key_val_sep, pairs_sep,
  1047. *shorthand ? AV_OPT_FLAG_IMPLICIT_KEY : 0,
  1048. &parsed_key, &value);
  1049. if (ret < 0) {
  1050. if (ret == AVERROR(EINVAL))
  1051. av_log(ctx, AV_LOG_ERROR, "No option name near '%s'\n", opts);
  1052. else
  1053. av_log(ctx, AV_LOG_ERROR, "Unable to parse '%s': %s\n", opts,
  1054. av_err2str(ret));
  1055. return ret;
  1056. }
  1057. if (*opts)
  1058. opts++;
  1059. if (parsed_key) {
  1060. key = parsed_key;
  1061. while (*shorthand) /* discard all remaining shorthand */
  1062. shorthand++;
  1063. } else {
  1064. key = *(shorthand++);
  1065. }
  1066. av_log(ctx, AV_LOG_DEBUG, "Setting '%s' to value '%s'\n", key, value);
  1067. if ((ret = av_opt_set(ctx, key, value, 0)) < 0) {
  1068. if (ret == AVERROR_OPTION_NOT_FOUND)
  1069. av_log(ctx, AV_LOG_ERROR, "Option '%s' not found\n", key);
  1070. av_free(value);
  1071. av_free(parsed_key);
  1072. return ret;
  1073. }
  1074. av_free(value);
  1075. av_free(parsed_key);
  1076. count++;
  1077. }
  1078. return count;
  1079. }
  1080. void av_opt_free(void *obj)
  1081. {
  1082. const AVOption *o = NULL;
  1083. while ((o = av_opt_next(obj, o)))
  1084. if (o->type == AV_OPT_TYPE_STRING || o->type == AV_OPT_TYPE_BINARY)
  1085. av_freep((uint8_t *)obj + o->offset);
  1086. }
  1087. int av_opt_set_dict(void *obj, AVDictionary **options)
  1088. {
  1089. AVDictionaryEntry *t = NULL;
  1090. AVDictionary *tmp = NULL;
  1091. int ret = 0;
  1092. while ((t = av_dict_get(*options, "", t, AV_DICT_IGNORE_SUFFIX))) {
  1093. ret = av_opt_set(obj, t->key, t->value, 0);
  1094. if (ret == AVERROR_OPTION_NOT_FOUND)
  1095. av_dict_set(&tmp, t->key, t->value, 0);
  1096. else if (ret < 0) {
  1097. av_log(obj, AV_LOG_ERROR, "Error setting option %s to value %s.\n", t->key, t->value);
  1098. break;
  1099. }
  1100. ret = 0;
  1101. }
  1102. av_dict_free(options);
  1103. *options = tmp;
  1104. return ret;
  1105. }
  1106. const AVOption *av_opt_find(void *obj, const char *name, const char *unit,
  1107. int opt_flags, int search_flags)
  1108. {
  1109. return av_opt_find2(obj, name, unit, opt_flags, search_flags, NULL);
  1110. }
  1111. const AVOption *av_opt_find2(void *obj, const char *name, const char *unit,
  1112. int opt_flags, int search_flags, void **target_obj)
  1113. {
  1114. const AVClass *c;
  1115. const AVOption *o = NULL;
  1116. if(!obj)
  1117. return NULL;
  1118. c= *(AVClass**)obj;
  1119. if (search_flags & AV_OPT_SEARCH_CHILDREN) {
  1120. if (search_flags & AV_OPT_SEARCH_FAKE_OBJ) {
  1121. const AVClass *child = NULL;
  1122. while (child = av_opt_child_class_next(c, child))
  1123. if (o = av_opt_find2(&child, name, unit, opt_flags, search_flags, NULL))
  1124. return o;
  1125. } else {
  1126. void *child = NULL;
  1127. while (child = av_opt_child_next(obj, child))
  1128. if (o = av_opt_find2(child, name, unit, opt_flags, search_flags, target_obj))
  1129. return o;
  1130. }
  1131. }
  1132. while (o = av_opt_next(obj, o)) {
  1133. if (!strcmp(o->name, name) && (o->flags & opt_flags) == opt_flags &&
  1134. ((!unit && o->type != AV_OPT_TYPE_CONST) ||
  1135. (unit && o->type == AV_OPT_TYPE_CONST && o->unit && !strcmp(o->unit, unit)))) {
  1136. if (target_obj) {
  1137. if (!(search_flags & AV_OPT_SEARCH_FAKE_OBJ))
  1138. *target_obj = obj;
  1139. else
  1140. *target_obj = NULL;
  1141. }
  1142. return o;
  1143. }
  1144. }
  1145. return NULL;
  1146. }
  1147. void *av_opt_child_next(void *obj, void *prev)
  1148. {
  1149. const AVClass *c = *(AVClass**)obj;
  1150. if (c->child_next)
  1151. return c->child_next(obj, prev);
  1152. return NULL;
  1153. }
  1154. const AVClass *av_opt_child_class_next(const AVClass *parent, const AVClass *prev)
  1155. {
  1156. if (parent->child_class_next)
  1157. return parent->child_class_next(prev);
  1158. return NULL;
  1159. }
  1160. void *av_opt_ptr(const AVClass *class, void *obj, const char *name)
  1161. {
  1162. const AVOption *opt= av_opt_find2(&class, name, NULL, 0, AV_OPT_SEARCH_FAKE_OBJ, NULL);
  1163. if(!opt)
  1164. return NULL;
  1165. return (uint8_t*)obj + opt->offset;
  1166. }
  1167. int av_opt_query_ranges(AVOptionRanges **ranges_arg, void *obj, const char *key, int flags)
  1168. {
  1169. const AVClass *c = *(AVClass**)obj;
  1170. int (*callback)(AVOptionRanges **, void *obj, const char *key, int flags) = NULL;
  1171. if (c->version > (52 << 16 | 11 << 8))
  1172. callback = c->query_ranges;
  1173. if (!callback)
  1174. callback = av_opt_query_ranges_default;
  1175. return callback(ranges_arg, obj, key, flags);
  1176. }
  1177. int av_opt_query_ranges_default(AVOptionRanges **ranges_arg, void *obj, const char *key, int flags)
  1178. {
  1179. AVOptionRanges *ranges = av_mallocz(sizeof(*ranges));
  1180. AVOptionRange **range_array = av_mallocz(sizeof(void*));
  1181. AVOptionRange *range = av_mallocz(sizeof(*range));
  1182. const AVOption *field = av_opt_find(obj, key, NULL, 0, flags);
  1183. int ret;
  1184. *ranges_arg = NULL;
  1185. if (!ranges || !range || !range_array || !field) {
  1186. ret = AVERROR(ENOMEM);
  1187. goto fail;
  1188. }
  1189. ranges->range = range_array;
  1190. ranges->range[0] = range;
  1191. ranges->nb_ranges = 1;
  1192. range->is_range = 1;
  1193. range->value_min = field->min;
  1194. range->value_max = field->max;
  1195. switch (field->type) {
  1196. case AV_OPT_TYPE_INT:
  1197. case AV_OPT_TYPE_INT64:
  1198. case AV_OPT_TYPE_PIXEL_FMT:
  1199. case AV_OPT_TYPE_SAMPLE_FMT:
  1200. case AV_OPT_TYPE_FLOAT:
  1201. case AV_OPT_TYPE_DOUBLE:
  1202. case AV_OPT_TYPE_DURATION:
  1203. case AV_OPT_TYPE_COLOR:
  1204. break;
  1205. case AV_OPT_TYPE_STRING:
  1206. range->component_min = 0;
  1207. range->component_max = 0x10FFFF; // max unicode value
  1208. range->value_min = -1;
  1209. range->value_max = INT_MAX;
  1210. break;
  1211. case AV_OPT_TYPE_RATIONAL:
  1212. range->component_min = INT_MIN;
  1213. range->component_max = INT_MAX;
  1214. break;
  1215. case AV_OPT_TYPE_IMAGE_SIZE:
  1216. range->component_min = 0;
  1217. range->component_max = INT_MAX/128/8;
  1218. range->value_min = 0;
  1219. range->value_max = INT_MAX/8;
  1220. break;
  1221. case AV_OPT_TYPE_VIDEO_RATE:
  1222. range->component_min = 1;
  1223. range->component_max = INT_MAX;
  1224. range->value_min = 1;
  1225. range->value_max = INT_MAX;
  1226. break;
  1227. default:
  1228. ret = AVERROR(ENOSYS);
  1229. goto fail;
  1230. }
  1231. *ranges_arg = ranges;
  1232. return 0;
  1233. fail:
  1234. av_free(ranges);
  1235. av_free(range);
  1236. av_free(range_array);
  1237. return ret;
  1238. }
  1239. void av_opt_freep_ranges(AVOptionRanges **rangesp)
  1240. {
  1241. int i;
  1242. AVOptionRanges *ranges = *rangesp;
  1243. for (i = 0; i < ranges->nb_ranges; i++) {
  1244. AVOptionRange *range = ranges->range[i];
  1245. av_freep(&range->str);
  1246. av_freep(&ranges->range[i]);
  1247. }
  1248. av_freep(&ranges->range);
  1249. av_freep(rangesp);
  1250. }
  1251. #ifdef TEST
  1252. typedef struct TestContext
  1253. {
  1254. const AVClass *class;
  1255. int num;
  1256. int toggle;
  1257. char *string;
  1258. int flags;
  1259. AVRational rational;
  1260. AVRational video_rate;
  1261. int w, h;
  1262. enum AVPixelFormat pix_fmt;
  1263. enum AVSampleFormat sample_fmt;
  1264. int64_t duration;
  1265. uint8_t color[4];
  1266. } TestContext;
  1267. #define OFFSET(x) offsetof(TestContext, x)
  1268. #define TEST_FLAG_COOL 01
  1269. #define TEST_FLAG_LAME 02
  1270. #define TEST_FLAG_MU 04
  1271. static const AVOption test_options[]= {
  1272. {"num", "set num", OFFSET(num), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 100 },
  1273. {"toggle", "set toggle", OFFSET(toggle), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1 },
  1274. {"rational", "set rational", OFFSET(rational), AV_OPT_TYPE_RATIONAL, {.dbl = 0}, 0, 10 },
  1275. {"string", "set string", OFFSET(string), AV_OPT_TYPE_STRING, {.str = "default"}, CHAR_MIN, CHAR_MAX },
  1276. {"flags", "set flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64 = 0}, 0, INT_MAX, 0, "flags" },
  1277. {"cool", "set cool flag ", 0, AV_OPT_TYPE_CONST, {.i64 = TEST_FLAG_COOL}, INT_MIN, INT_MAX, 0, "flags" },
  1278. {"lame", "set lame flag ", 0, AV_OPT_TYPE_CONST, {.i64 = TEST_FLAG_LAME}, INT_MIN, INT_MAX, 0, "flags" },
  1279. {"mu", "set mu flag ", 0, AV_OPT_TYPE_CONST, {.i64 = TEST_FLAG_MU}, INT_MIN, INT_MAX, 0, "flags" },
  1280. {"size", "set size", OFFSET(w), AV_OPT_TYPE_IMAGE_SIZE,{0}, 0, 0 },
  1281. {"pix_fmt", "set pixfmt", OFFSET(pix_fmt), AV_OPT_TYPE_PIXEL_FMT, {.i64 = AV_PIX_FMT_NONE}, -1, AV_PIX_FMT_NB-1},
  1282. {"sample_fmt", "set samplefmt", OFFSET(sample_fmt), AV_OPT_TYPE_SAMPLE_FMT, {.i64 = AV_SAMPLE_FMT_NONE}, -1, AV_SAMPLE_FMT_NB-1},
  1283. {"video_rate", "set videorate", OFFSET(video_rate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, 0 },
  1284. {"duration", "set duration", OFFSET(duration), AV_OPT_TYPE_DURATION, {.i64 = 0}, 0, INT64_MAX},
  1285. {"color", "set color", OFFSET(color), AV_OPT_TYPE_COLOR, {.str = "pink"}, 0, 0},
  1286. {NULL},
  1287. };
  1288. static const char *test_get_name(void *ctx)
  1289. {
  1290. return "test";
  1291. }
  1292. static const AVClass test_class = {
  1293. "TestContext",
  1294. test_get_name,
  1295. test_options
  1296. };
  1297. int main(void)
  1298. {
  1299. int i;
  1300. printf("\nTesting av_set_options_string()\n");
  1301. {
  1302. TestContext test_ctx = { 0 };
  1303. const char *options[] = {
  1304. "",
  1305. ":",
  1306. "=",
  1307. "foo=:",
  1308. ":=foo",
  1309. "=foo",
  1310. "foo=",
  1311. "foo",
  1312. "foo=val",
  1313. "foo==val",
  1314. "toggle=:",
  1315. "string=:",
  1316. "toggle=1 : foo",
  1317. "toggle=100",
  1318. "toggle==1",
  1319. "flags=+mu-lame : num=42: toggle=0",
  1320. "num=42 : string=blahblah",
  1321. "rational=0 : rational=1/2 : rational=1/-1",
  1322. "rational=-1/0",
  1323. "size=1024x768",
  1324. "size=pal",
  1325. "size=bogus",
  1326. "pix_fmt=yuv420p",
  1327. "pix_fmt=2",
  1328. "pix_fmt=bogus",
  1329. "sample_fmt=s16",
  1330. "sample_fmt=2",
  1331. "sample_fmt=bogus",
  1332. "video_rate=pal",
  1333. "video_rate=25",
  1334. "video_rate=30000/1001",
  1335. "video_rate=30/1.001",
  1336. "video_rate=bogus",
  1337. "duration=bogus",
  1338. "duration=123.45",
  1339. "duration=1\\:23\\:45.67",
  1340. "color=blue",
  1341. "color=0x223300",
  1342. "color=0x42FF07AA",
  1343. };
  1344. test_ctx.class = &test_class;
  1345. av_opt_set_defaults(&test_ctx);
  1346. av_log_set_level(AV_LOG_DEBUG);
  1347. for (i=0; i < FF_ARRAY_ELEMS(options); i++) {
  1348. av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]);
  1349. if (av_set_options_string(&test_ctx, options[i], "=", ":") < 0)
  1350. av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]);
  1351. printf("\n");
  1352. }
  1353. av_freep(&test_ctx.string);
  1354. }
  1355. printf("\nTesting av_opt_set_from_string()\n");
  1356. {
  1357. TestContext test_ctx = { 0 };
  1358. const char *options[] = {
  1359. "",
  1360. "5",
  1361. "5:hello",
  1362. "5:hello:size=pal",
  1363. "5:size=pal:hello",
  1364. ":",
  1365. "=",
  1366. " 5 : hello : size = pal ",
  1367. "a_very_long_option_name_that_will_need_to_be_ellipsized_around_here=42"
  1368. };
  1369. const char *shorthand[] = { "num", "string", NULL };
  1370. test_ctx.class = &test_class;
  1371. av_opt_set_defaults(&test_ctx);
  1372. test_ctx.string = av_strdup("default");
  1373. av_log_set_level(AV_LOG_DEBUG);
  1374. for (i=0; i < FF_ARRAY_ELEMS(options); i++) {
  1375. av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]);
  1376. if (av_opt_set_from_string(&test_ctx, options[i], shorthand, "=", ":") < 0)
  1377. av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]);
  1378. printf("\n");
  1379. }
  1380. av_freep(&test_ctx.string);
  1381. }
  1382. return 0;
  1383. }
  1384. #endif