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

444 lines
15KB
Raw Blame History 

  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. /**

  23.  * @file opt.c

  24.  * AVOptions

  25.  * @author Michael Niedermayer <michaelni@gmx.at>

  26.  */

  27. 

  28. #include "avcodec.h"

  29. #include "opt.h"

  30. #include "eval.h"

  31. 

  32. //FIXME order them and do a bin search

  33. const AVOption *av_find_opt(void *v, const char *name, const char *unit, int mask, int flags){

  34.     AVClass *c= *(AVClass**)v; //FIXME silly way of storing AVClass

  35.     const AVOption *o= c->option;

  36. 

  37.     for(;o && o->name; o++){

  38.         if(!strcmp(o->name, name) && (!unit || (o->unit && !strcmp(o->unit, unit))) && (o->flags & mask) == flags )

  39.             return o;

  40.     }

  41.     return NULL;

  42. }

  43. 

  44. const AVOption *av_next_option(void *obj, const AVOption *last){

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

  46.     else if(last)            return NULL;

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

  48. }

  49. 

  50. static const AVOption *av_set_number(void *obj, const char *name, double num, int den, int64_t intnum){

  51.     const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);

  52.     void *dst;

  53.     if(!o || o->offset<=0)

  54.         return NULL;

  55. 

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

  57.         av_log(NULL, AV_LOG_ERROR, "Value %lf for parameter '%s' out of range.\n", num, name);

  58.         return NULL;

  59.     }

  60. 

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

  62. 

  63.     switch(o->type){

  64.     case FF_OPT_TYPE_FLAGS:

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

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

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

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

  69.     case FF_OPT_TYPE_RATIONAL:

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

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

  72.         break;

  73.     default:

  74.         return NULL;

  75.     }

  76.     return o;

  77. }

  78. 

  79. static const AVOption *set_all_opt(void *v, const char *unit, double d){

  80.     AVClass *c= *(AVClass**)v; //FIXME silly way of storing AVClass

  81.     const AVOption *o= c->option;

  82.     const AVOption *ret=NULL;

  83. 

  84.     for(;o && o->name; o++){

  85.         if(o->type != FF_OPT_TYPE_CONST && o->unit && !strcmp(o->unit, unit)){

  86.             double tmp= d;

  87.             if(o->type == FF_OPT_TYPE_FLAGS)

  88.                 tmp= av_get_int(v, o->name, NULL) | (int64_t)d;

  89. 

  90.             av_set_number(v, o->name, tmp, 1, 1);

  91.             ret= o;

  92.         }

  93.     }

  94.     return ret;

  95. }

  96. 

  97. static const double const_values[]={

  98.     M_PI,

  99.     M_E,

  100.     FF_QP2LAMBDA,

  101.     0

  102. };

  103. 

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

  105.     "PI",

  106.     "E",

  107.     "QP2LAMBDA",

  108.     0

  109. };

  110. 

  111. static int hexchar2int(char c) {

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

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

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

  115.     return -1;

  116. }

  117. 

  118. const AVOption *av_set_string2(void *obj, const char *name, const char *val, int alloc){

  119.     const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);

  120.     if(o && o->offset==0 && o->type == FF_OPT_TYPE_CONST && o->unit){

  121.         return set_all_opt(obj, o->unit, o->default_val);

  122.     }

  123.     if(!o || !val || o->offset<=0)

  124.         return NULL;

  125.     if(o->type == FF_OPT_TYPE_BINARY){

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

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

  128.         uint8_t *bin, *ptr;

  129.         int len = strlen(val);

  130.         av_freep(dst);

  131.         *lendst = 0;

  132.         if (len & 1) return NULL;

  133.         len /= 2;

  134.         ptr = bin = av_malloc(len);

  135.         while (*val) {

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

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

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

  139.                 av_free(bin);

  140.                 return NULL;

  141.             }

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

  143.         }

  144.         *dst = bin;

  145.         *lendst = len;

  146.         return o;

  147.     }

  148.     if(o->type != FF_OPT_TYPE_STRING){

  149.         int notfirst=0;

  150.         for(;;){

  151.             int i;

  152.             char buf[256];

  153.             int cmd=0;

  154.             double d;

  155.             const char *error = NULL;

  156. 

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

  158.                 cmd= *(val++);

  159. 

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

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

  162.             buf[i]=0;

  163. 

  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\": %s\n", val, error);

  177.                     return NULL;

  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. 

  188.             if (!av_set_number(obj, name, d, 1, 1))

  189.                 return NULL;

  190.             val+= i;

  191.             if(!*val)

  192.                 return o;

  193.             notfirst=1;

  194.         }

  195.         return NULL;

  196.     }

  197. 

  198.     if(alloc){

  199.         av_free(*(void**)(((uint8_t*)obj) + o->offset));

  200.         val= av_strdup(val);

  201.     }

  202. 

  203.     memcpy(((uint8_t*)obj) + o->offset, &val, sizeof(val));

  204.     return o;

  205. }

  206. 

  207. const AVOption *av_set_string(void *obj, const char *name, const char *val){

  208.     return av_set_string2(obj, name, val, 0);

  209. }

  210. 

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

  212.     return av_set_number(obj, name, n, 1, 1);

  213. }

  214. 

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

  216.     return av_set_number(obj, name, n.num, n.den, 1);

  217. }

  218. 

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

  220.     return av_set_number(obj, name, 1, 1, n);

  221. }

  222. 

  223. /**

  224.  *

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

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

  227.  */

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

  229.     const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);

  230.     void *dst;

  231.     uint8_t *bin;

  232.     int len, i;

  233.     if(!o || o->offset<=0)

  234.         return NULL;

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

  236.         return NULL;

  237. 

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

  239.     if(o_out) *o_out= o;

  240. 

  241.     switch(o->type){

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

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

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

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

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

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

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

  249.     case FF_OPT_TYPE_BINARY:

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

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

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

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

  254.         break;

  255.     default: return NULL;

  256.     }

  257.     return buf;

  258. }

  259. 

  260. static int av_get_number(void *obj, const char *name, const AVOption **o_out, double *num, int *den, int64_t *intnum){

  261.     const AVOption *o= av_find_opt(obj, name, NULL, 0, 0);

  262.     void *dst;

  263.     if(!o || o->offset<=0)

  264.         goto error;

  265. 

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

  267. 

  268.     if(o_out) *o_out= o;

  269. 

  270.     switch(o->type){

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

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

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

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

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

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

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

  278.                                                         return 0;

  279.     }

  280. error:

  281.     *den=*intnum=0;

  282.     return -1;

  283. }

  284. 

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

  286.     int64_t intnum=1;

  287.     double num=1;

  288.     int den=1;

  289. 

  290.     av_get_number(obj, name, o_out, &num, &den, &intnum);

  291.     return num*intnum/den;

  292. }

  293. 

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

  295.     int64_t intnum=1;

  296.     double num=1;

  297.     int den=1;

  298. 

  299.     av_get_number(obj, name, o_out, &num, &den, &intnum);

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

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

  302.     else

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

  304. }

  305. 

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

  307.     int64_t intnum=1;

  308.     double num=1;

  309.     int den=1;

  310. 

  311.     av_get_number(obj, name, o_out, &num, &den, &intnum);

  312.     return num*intnum/den;

  313. }

  314. 

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

  316. {

  317.     const AVOption *opt=NULL;

  318. 

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

  320.         if(!(opt->flags & (AV_OPT_FLAG_ENCODING_PARAM|AV_OPT_FLAG_DECODING_PARAM)))

  321.             continue;

  322. 

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

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

  325.          * Only print items from the requested unit.

  326.          */

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

  328.             continue;

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

  330.             continue;

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

  332.             continue;

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

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

  335.         else

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

  337. 

  338.         switch( opt->type )

  339.         {

  340.             case FF_OPT_TYPE_FLAGS:

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

  342.                 break;

  343.             case FF_OPT_TYPE_INT:

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

  345.                 break;

  346.             case FF_OPT_TYPE_INT64:

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

  348.                 break;

  349.             case FF_OPT_TYPE_DOUBLE:

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

  351.                 break;

  352.             case FF_OPT_TYPE_FLOAT:

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

  354.                 break;

  355.             case FF_OPT_TYPE_STRING:

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

  357.                 break;

  358.             case FF_OPT_TYPE_RATIONAL:

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

  360.                 break;

  361.             case FF_OPT_TYPE_BINARY:

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

  363.                 break;

  364.             case FF_OPT_TYPE_CONST:

  365.             default:

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

  367.                 break;

  368.         }

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

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

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

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

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

  374. 

  375.         if(opt->help)

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

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

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

  379.             opt_list(obj, av_log_obj, opt->unit);

  380.         }

  381.     }

  382. }

  383. 

  384. int av_opt_show(void *obj, void *av_log_obj){

  385.     if(!obj)

  386.         return -1;

  387. 

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

  389. 

  390.     opt_list(obj, av_log_obj, NULL);

  391. 

  392.     return 0;

  393. }

  394. 

  395. /** Set the values of the AVCodecContext or AVFormatContext structure.

  396.  * They are set to the defaults specified in the according AVOption options

  397.  * array default_val field.

  398.  *

  399.  * @param s AVCodecContext or AVFormatContext for which the defaults will be set

  400.  */

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

  402. {

  403.     const AVOption *opt = NULL;

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

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

  406.             continue;

  407.         switch(opt->type) {

  408.             case FF_OPT_TYPE_CONST:

  409.                 /* Nothing to be done here */

  410.             break;

  411.             case FF_OPT_TYPE_FLAGS:

  412.             case FF_OPT_TYPE_INT: {

  413.                 int val;

  414.                 val = opt->default_val;

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

  416.             }

  417.             break;

  418.             case FF_OPT_TYPE_FLOAT: {

  419.                 double val;

  420.                 val = opt->default_val;

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

  422.             }

  423.             break;

  424.             case FF_OPT_TYPE_RATIONAL: {

  425.                 AVRational val;

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

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

  428.             }

  429.             break;

  430.             case FF_OPT_TYPE_STRING:

  431.             case FF_OPT_TYPE_BINARY:

  432.                 /* Cannot set default for string as default_val is of type * double */

  433.             break;

  434.             default:

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

  436.         }

  437.     }

  438. }

  439. 

  440. void av_opt_set_defaults(void *s){

  441.     av_opt_set_defaults2(s, 0, 0);

  442. }