opt.c
上传用户:jylinhe
上传日期:2022-07-11
资源大小:334k
文件大小:16k
源码类别:

多媒体编程

开发平台:

Visual C++

  1. /*
  2.  * AVOptions
  3.  * Copyright (c) 2005 Michael Niedermayer <michaelni@gmx.at>
  4.  *
  5.  * This file is part of FFmpeg.
  6.  *
  7.  * FFmpeg is free software; you can redistribute it and/or
  8.  * modify it under the terms of the GNU Lesser General Public
  9.  * License as published by the Free Software Foundation; either
  10.  * version 2.1 of the License, or (at your option) any later version.
  11.  *
  12.  * FFmpeg is distributed in the hope that it will be useful,
  13.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15.  * Lesser General Public License for more details.
  16.  *
  17.  * You should have received a copy of the GNU Lesser General Public
  18.  * License along with FFmpeg; if not, write to the Free Software
  19.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20.  */
  21. /**
  22.  * @file opt.c
  23.  * AVOptions
  24.  * @author Michael Niedermayer <michaelni@gmx.at>
  25.  */
  26. #include "dsputil.h"
  27. //#include "internal.h"
  28. //#include "integer.h"
  29. #include "opt.h"
  30. //#include "eval.h"
  31. //FIXME order them and do a bin search
  32. const AVOption *av_find_opt(void *v, const char *name, const char *unit, int mask, int flags){
  33.     AVClass *c= *(AVClass**)v; //FIXME silly way of storing AVClass
  34.     const AVOption *o= c->option;
  35.     for(;o && o->name; o++){
  36.         if(!strcmp(o->name, name) && (!unit || (o->unit && !strcmp(o->unit, unit))) && (o->flags & mask) == flags )
  37.             return o;
  38.     }
  39.     return NULL;
  40. }
  41. const AVOption *av_next_option(void *obj, const AVOption *last){
  42.     if(last && last[1].name) return ++last;
  43.     else if(last)            return NULL;
  44.     else                     return (*(AVClass**)obj)->option;
  45. }
  46. static int av_set_number2(void *obj, const char *name, double num, int den, int64_t intnum, const AVOption **o_out){
  47.     const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);
  48.     void *dst;
  49.     if(o_out)
  50.         *o_out= o;
  51.     if(!o || o->offset<=0)
  52.         return AVERROR(ENOENT);
  53.     if(o->max*den < num*intnum || o->min*den > num*intnum) {
  54.         av_log(NULL, AV_LOG_ERROR, "Value %lf for parameter '%s' out of rangen", num, name);
  55.         return AVERROR(ERANGE);
  56.     }
  57.     dst= ((uint8_t*)obj) + o->offset;
  58.     switch(o->type){
  59.     case FF_OPT_TYPE_FLAGS:
  60.     case FF_OPT_TYPE_INT:   
  61. *(int       *)dst= (int)(llrint(num/den)*intnum);
  62. break;
  63.     case FF_OPT_TYPE_INT64: 
  64. *(int64_t   *)dst= llrint(num/den)*intnum; 
  65. break;
  66.     case FF_OPT_TYPE_FLOAT: 
  67. *(float     *)dst= (float)(num*intnum/den);
  68. break;
  69.     case FF_OPT_TYPE_DOUBLE:
  70. *(double    *)dst= num*intnum/den;         
  71. break;
  72.     case FF_OPT_TYPE_RATIONAL:
  73.         if((int)num == num) 
  74. {
  75. //*(AVRational*)dst= (AVRational){num*intnum, den};
  76. AVRational tmp;
  77. tmp.num = (int)(num*intnum);
  78. tmp.den = den;
  79. *(AVRational*)dst= tmp;
  80. }
  81.         else
  82. {
  83. *(AVRational*)dst= av_d2q(num*intnum/den, 1<<24);
  84. }
  85.         break;
  86.     default:
  87.         return AVERROR(EINVAL);
  88.     }
  89.     return 0;
  90. }
  91. static const AVOption *av_set_number(void *obj, const char *name, double num, int den, int64_t intnum){
  92.     const AVOption *o = NULL;
  93.     if (av_set_number2(obj, name, num, den, intnum, &o) < 0)
  94.         return NULL;
  95.     else
  96.         return o;
  97. }
  98. /*static const double const_values[]={
  99.     M_PI,
  100.     M_E,
  101.     FF_QP2LAMBDA,
  102.     0
  103. };*/
  104. /*
  105. static const char * const const_names[]={
  106.     "PI",
  107.     "E",
  108.     "QP2LAMBDA",
  109.     0
  110. };*/
  111. /*
  112. static int hexchar2int(char c) {
  113.     if (c >= '0' && c <= '9') return c - '0';
  114.     if (c >= 'a' && c <= 'f') return c - 'a' + 10;
  115.     if (c >= 'A' && c <= 'F') return c - 'A' + 10;
  116.     return -1;
  117. }*/
  118. /*
  119. int av_set_string3(void *obj, const char *name, const char *val, int alloc, const AVOption **o_out){
  120.     int ret;
  121.     const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);
  122.     if (o_out)
  123.         *o_out = o;
  124.     if(!o)
  125.         return AVERROR(ENOENT);
  126.     if(!val || o->offset<=0)
  127.         return AVERROR(EINVAL);
  128.     if(o->type == FF_OPT_TYPE_BINARY){
  129.         uint8_t **dst = (uint8_t **)(((uint8_t*)obj) + o->offset);
  130.         int *lendst = (int *)(dst + 1);
  131.         uint8_t *bin, *ptr;
  132.         int len = strlen(val);
  133.         av_freep(dst);
  134.         *lendst = 0;
  135.         if (len & 1) return AVERROR(EINVAL);
  136.         len /= 2;
  137.         ptr = bin = av_malloc(len);
  138.         while (*val) {
  139.             int a = hexchar2int(*val++);
  140.             int b = hexchar2int(*val++);
  141.             if (a < 0 || b < 0) {
  142.                 av_free(bin);
  143.                 return AVERROR(EINVAL);
  144.             }
  145.             *ptr++ = (a << 4) | b;
  146.         }
  147.         *dst = bin;
  148.         *lendst = len;
  149.         return 0;
  150.     }
  151.     if(o->type != FF_OPT_TYPE_STRING){
  152.         int notfirst=0;
  153.         for(;;){
  154.             int i;
  155.             char buf[256];
  156.             int cmd=0;
  157.             double d;
  158.             const char *error = NULL;
  159.             if(*val == '+' || *val == '-')
  160.                 cmd= *(val++);
  161.             for(i=0; i<sizeof(buf)-1 && val[i] && val[i]!='+' && val[i]!='-'; i++)
  162.                 buf[i]= val[i];
  163.             buf[i]=0;
  164.             d = ff_eval2(buf, const_values, const_names, NULL, NULL, NULL, NULL, NULL, &error);
  165.             if(isnan(d)) {
  166.                 const AVOption *o_named= av_find_opt(obj, buf, o->unit, 0, 0);
  167.                 if(o_named && o_named->type == FF_OPT_TYPE_CONST)
  168.                     d= o_named->default_val;
  169.                 else if(!strcmp(buf, "default")) d= o->default_val;
  170.                 else if(!strcmp(buf, "max"    )) d= o->max;
  171.                 else if(!strcmp(buf, "min"    )) d= o->min;
  172.                 else if(!strcmp(buf, "none"   )) d= 0;
  173.                 else if(!strcmp(buf, "all"    )) d= ~0;
  174.                 else {
  175.                     if (error)
  176.                         av_log(NULL, AV_LOG_ERROR, "Unable to parse option value "%s": %sn", val, error);
  177.                     return AVERROR(EINVAL);
  178.                 }
  179.             }
  180.             if(o->type == FF_OPT_TYPE_FLAGS){
  181.                 if     (cmd=='+') d= av_get_int(obj, name, NULL) | (int64_t)d;
  182.                 else if(cmd=='-') d= av_get_int(obj, name, NULL) &~(int64_t)d;
  183.             }else{
  184.                 if     (cmd=='+') d= notfirst*av_get_double(obj, name, NULL) + d;
  185.                 else if(cmd=='-') d= notfirst*av_get_double(obj, name, NULL) - d;
  186.             }
  187.             if ((ret = av_set_number2(obj, name, d, 1, 1, o_out)) < 0)
  188.                 return ret;
  189.             val+= i;
  190.             if(!*val)
  191.                 return 0;
  192.             notfirst=1;
  193.         }
  194.         return AVERROR(EINVAL);
  195.     }
  196.     if(alloc){
  197.         av_free(*(void**)(((uint8_t*)obj) + o->offset));
  198.         val= av_strdup(val);
  199.     }
  200.     memcpy(((uint8_t*)obj) + o->offset, &val, sizeof(val));
  201.     return 0;
  202. }
  203. #if LIBAVCODEC_VERSION_MAJOR < 53
  204. const AVOption *av_set_string2(void *obj, const char *name, const char *val, int alloc){
  205.     const AVOption *o;
  206.     if (av_set_string3(obj, name, val, alloc, &o) < 0)
  207.         return NULL;
  208.     return o;
  209. }
  210. const AVOption *av_set_string(void *obj, const char *name, const char *val){
  211.     const AVOption *o;
  212.     if (av_set_string3(obj, name, val, 0, &o) < 0)
  213.         return NULL;
  214.     return o;
  215. }
  216. #endif
  217. */
  218. const AVOption *av_set_double(void *obj, const char *name, double n){
  219.     return av_set_number(obj, name, n, 1, 1);
  220. }
  221. const AVOption *av_set_q(void *obj, const char *name, AVRational n){
  222.     return av_set_number(obj, name, n.num, n.den, 1);
  223. }
  224. const AVOption *av_set_int(void *obj, const char *name, int64_t n){
  225.     return av_set_number(obj, name, 1, 1, n);
  226. }
  227. /**
  228.  *
  229.  * @param buf a buffer which is used for returning non string values as strings, can be NULL
  230.  * @param buf_len allocated length in bytes of buf
  231.  */
  232. const char *av_get_string(void *obj, const char *name, const AVOption **o_out, char *buf, int buf_len){
  233.     const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);
  234.     void *dst;
  235.     uint8_t *bin;
  236.     int len, i;
  237.     if(!o || o->offset<=0)
  238.         return NULL;
  239.     if(o->type != FF_OPT_TYPE_STRING && (!buf || !buf_len))
  240.         return NULL;
  241.     dst= ((uint8_t*)obj) + o->offset;
  242.     if(o_out) *o_out= o;
  243.     switch(o->type){
  244.     case FF_OPT_TYPE_FLAGS:     snprintf(buf, buf_len, "0x%08X",*(int    *)dst);break;
  245.     case FF_OPT_TYPE_INT:       snprintf(buf, buf_len, "%d" , *(int    *)dst);break;
  246.     case FF_OPT_TYPE_INT64:     snprintf(buf, buf_len, "%"PRId64, *(int64_t*)dst);break;
  247.     case FF_OPT_TYPE_FLOAT:     snprintf(buf, buf_len, "%f" , *(float  *)dst);break;
  248.     case FF_OPT_TYPE_DOUBLE:    snprintf(buf, buf_len, "%f" , *(double *)dst);break;
  249.     case FF_OPT_TYPE_RATIONAL:  snprintf(buf, buf_len, "%d/%d", ((AVRational*)dst)->num, ((AVRational*)dst)->den);break;
  250.     case FF_OPT_TYPE_STRING:    return *(void**)dst;
  251.     case FF_OPT_TYPE_BINARY:
  252.         len = *(int*)(((uint8_t *)dst) + sizeof(uint8_t *));
  253.         if(len >= (buf_len + 1)/2) return NULL;
  254.         bin = *(uint8_t**)dst;
  255.         for(i = 0; i < len; i++) snprintf(buf + i*2, 3, "%02X", bin[i]);
  256.         break;
  257.     default: return NULL;
  258.     }
  259.     return buf;
  260. }
  261. static int av_get_number(void *obj, const char *name, const AVOption **o_out, double *num, int *den, int64_t *intnum){
  262.     const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);
  263.     void *dst;
  264.     if(!o || o->offset<=0)
  265.         goto error;
  266.     dst= ((uint8_t*)obj) + o->offset;
  267.     if(o_out) *o_out= o;
  268.     switch(o->type){
  269.     case FF_OPT_TYPE_FLAGS:     *intnum= *(unsigned int*)dst;return 0;
  270.     case FF_OPT_TYPE_INT:       *intnum= *(int    *)dst;return 0;
  271.     case FF_OPT_TYPE_INT64:     *intnum= *(int64_t*)dst;return 0;
  272.     case FF_OPT_TYPE_FLOAT:     *num=    *(float  *)dst;return 0;
  273.     case FF_OPT_TYPE_DOUBLE:    *num=    *(double *)dst;return 0;
  274.     case FF_OPT_TYPE_RATIONAL:  *intnum= ((AVRational*)dst)->num;
  275.                                 *den   = ((AVRational*)dst)->den;
  276.                                                         return 0;
  277.     }
  278. error:
  279.     *den=*intnum=0;
  280.     return -1;
  281. }
  282. double av_get_double(void *obj, const char *name, const AVOption **o_out){
  283.     int64_t intnum=1;
  284.     double num=1;
  285.     int den=1;
  286.     av_get_number(obj, name, o_out, &num, &den, &intnum);
  287.     return num*intnum/den;
  288. }
  289. AVRational av_get_q(void *obj, const char *name, const AVOption **o_out){
  290.     int64_t intnum=1;
  291.     double num=1;
  292.     int den=1;
  293.     av_get_number(obj, name, o_out, &num, &den, &intnum);
  294.     if(num == 1.0 && (int)intnum == intnum)
  295. {
  296.         //return (AVRational){intnum, den};
  297. AVRational tmp;
  298. tmp.num = (int)intnum;
  299. tmp.den = den;
  300. return tmp;
  301. }
  302.     else
  303. {
  304.         return av_d2q(num*intnum/den, 1<<24);
  305. }
  306. }
  307. int64_t av_get_int(void *obj, const char *name, const AVOption **o_out){
  308.     int64_t intnum=1;
  309.     double num=1;
  310.     int den=1;
  311.     av_get_number(obj, name, o_out, &num, &den, &intnum);
  312.     return (int64_t)(num*intnum/den);
  313. }
  314. static void opt_list(void *obj, void *av_log_obj, const char *unit)
  315. {
  316.     const AVOption *opt=NULL;
  317.     while((opt= av_next_option(obj, opt))){
  318.         if(!(opt->flags & (AV_OPT_FLAG_ENCODING_PARAM|AV_OPT_FLAG_DECODING_PARAM)))
  319.             continue;
  320.         /* Don't print CONST's on level one.
  321.          * Don't print anything but CONST's on level two.
  322.          * Only print items from the requested unit.
  323.          */
  324.         if (!unit && opt->type==FF_OPT_TYPE_CONST)
  325.             continue;
  326.         else if (unit && opt->type!=FF_OPT_TYPE_CONST)
  327.             continue;
  328.         else if (unit && opt->type==FF_OPT_TYPE_CONST && strcmp(unit, opt->unit))
  329.             continue;
  330.         else if (unit && opt->type == FF_OPT_TYPE_CONST)
  331.             av_log(av_log_obj, AV_LOG_INFO, "   %-15s ", opt->name);
  332.         else
  333.             av_log(av_log_obj, AV_LOG_INFO, "-%-17s ", opt->name);
  334.         switch( opt->type )
  335.         {
  336.             case FF_OPT_TYPE_FLAGS:
  337.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<flags>" );
  338.                 break;
  339.             case FF_OPT_TYPE_INT:
  340.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<int>" );
  341.                 break;
  342.             case FF_OPT_TYPE_INT64:
  343.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<int64>" );
  344.                 break;
  345.             case FF_OPT_TYPE_DOUBLE:
  346.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<double>" );
  347.                 break;
  348.             case FF_OPT_TYPE_FLOAT:
  349.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<float>" );
  350.                 break;
  351.             case FF_OPT_TYPE_STRING:
  352.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<string>" );
  353.                 break;
  354.             case FF_OPT_TYPE_RATIONAL:
  355.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<rational>" );
  356.                 break;
  357.             case FF_OPT_TYPE_BINARY:
  358.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "<binary>" );
  359.                 break;
  360.             case FF_OPT_TYPE_CONST:
  361.             default:
  362.                 av_log( av_log_obj, AV_LOG_INFO, "%-7s ", "" );
  363.                 break;
  364.         }
  365.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_ENCODING_PARAM) ? 'E' : '.');
  366.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_DECODING_PARAM) ? 'D' : '.');
  367.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_VIDEO_PARAM   ) ? 'V' : '.');
  368.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_AUDIO_PARAM   ) ? 'A' : '.');
  369.         av_log(av_log_obj, AV_LOG_INFO, "%c", (opt->flags & AV_OPT_FLAG_SUBTITLE_PARAM) ? 'S' : '.');
  370.         if(opt->help)
  371.             av_log(av_log_obj, AV_LOG_INFO, " %s", opt->help);
  372.         av_log(av_log_obj, AV_LOG_INFO, "n");
  373.         if (opt->unit && opt->type != FF_OPT_TYPE_CONST) {
  374.             opt_list(obj, av_log_obj, opt->unit);
  375.         }
  376.     }
  377. }
  378. int av_opt_show(void *obj, void *av_log_obj){
  379.     if(!obj)
  380.         return -1;
  381.     av_log(av_log_obj, AV_LOG_INFO, "%s AVOptions:n", (*(AVClass**)obj)->class_name);
  382.     opt_list(obj, av_log_obj, NULL);
  383.     return 0;
  384. }
  385. /** Set the values of the AVCodecContext or AVFormatContext structure.
  386.  * They are set to the defaults specified in the according AVOption options
  387.  * array default_val field.
  388.  *
  389.  * @param s AVCodecContext or AVFormatContext for which the defaults will be set
  390.  */
  391. void av_opt_set_defaults2(void *s, int mask, int flags)
  392. {
  393.     const AVOption *opt = NULL;
  394.     while ((opt = av_next_option(s, opt)) != NULL) {
  395.         if((opt->flags & mask) != flags)
  396.             continue;
  397.         switch(opt->type) {
  398.             case FF_OPT_TYPE_CONST:
  399.                 /* Nothing to be done here */
  400.             break;
  401.             case FF_OPT_TYPE_FLAGS:
  402.             case FF_OPT_TYPE_INT: {
  403.                 int val;
  404.                 val = (int)opt->default_val;
  405.                 av_set_int(s, opt->name, val);
  406.             }
  407.             break;
  408.             case FF_OPT_TYPE_INT64:
  409.                 if((double)(opt->default_val+0.6) == opt->default_val)
  410.                     av_log(s, AV_LOG_DEBUG, "loss of precission in default of %sn", opt->name);
  411.                 av_set_int(s, opt->name, (int64_t)opt->default_val);
  412.             break;
  413.             case FF_OPT_TYPE_FLOAT: {
  414.                 double val;
  415.                 val = opt->default_val;
  416.                 av_set_double(s, opt->name, val);
  417.             }
  418.             break;
  419.             case FF_OPT_TYPE_RATIONAL: {
  420.                 AVRational val;
  421.                 val = av_d2q(opt->default_val, INT_MAX);
  422.                 av_set_q(s, opt->name, val);
  423.             }
  424.             break;
  425.             case FF_OPT_TYPE_STRING:
  426.             case FF_OPT_TYPE_BINARY:
  427.                 /* Cannot set default for string as default_val is of type * double */
  428.             break;
  429.             default:
  430.                 av_log(s, AV_LOG_DEBUG, "AVOption type %d of option %s not implemented yetn", opt->type, opt->name);
  431.         }
  432.     }
  433. }
  434. void av_opt_set_defaults(void *s){
  435.     av_opt_set_defaults2(s, 0, 0);
  436. }