git.sesse.net Git - ffmpeg/blob - libavutil/eval.c

   1 /*
   2  * Copyright (c) 2002-2006 Michael Niedermayer <michaelni@gmx.at>
   3  * Copyright (c) 2006 Oded Shimon <ods15@ods15.dyndns.org>
   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
  24  * simple arithmetic expression evaluator.
  25  *
  26  * see http://joe.hotchkiss.com/programming/eval/eval.html
  27  */
  28
  29 #include "avutil.h"
  30 #include "eval.h"
  31 #include "log.h"
  32
  33 typedef struct Parser {
  34     const AVClass *class;
  35     int stack_index;
  36     char *s;
  37     const double *const_values;
  38     const char * const *const_names;          // NULL terminated
  39     double (* const *funcs1)(void *, double a);           // NULL terminated
  40     const char * const *func1_names;          // NULL terminated
  41     double (* const *funcs2)(void *, double a, double b); // NULL terminated
  42     const char * const *func2_names;          // NULL terminated
  43     void *opaque;
  44     int log_offset;
  45     void *log_ctx;
  46 #define VARS 10
  47     double *var;
  48 } Parser;
  49
  50 static const AVClass class = { "Eval", av_default_item_name, NULL, LIBAVUTIL_VERSION_INT, offsetof(Parser,log_offset), offsetof(Parser,log_ctx) };
  51
  52 static const int8_t si_prefixes['z' - 'E' + 1] = {
  53     ['y'-'E']= -24,
  54     ['z'-'E']= -21,
  55     ['a'-'E']= -18,
  56     ['f'-'E']= -15,
  57     ['p'-'E']= -12,
  58     ['n'-'E']= - 9,
  59     ['u'-'E']= - 6,
  60     ['m'-'E']= - 3,
  61     ['c'-'E']= - 2,
  62     ['d'-'E']= - 1,
  63     ['h'-'E']=   2,
  64     ['k'-'E']=   3,
  65     ['K'-'E']=   3,
  66     ['M'-'E']=   6,
  67     ['G'-'E']=   9,
  68     ['T'-'E']=  12,
  69     ['P'-'E']=  15,
  70     ['E'-'E']=  18,
  71     ['Z'-'E']=  21,
  72     ['Y'-'E']=  24,
  73 };
  74
  75 static const struct {
  76     const char *name;
  77     double value;
  78 } constants[] = {
  79     { "E",   M_E   },
  80     { "PI",  M_PI  },
  81     { "PHI", M_PHI },
  82 };
  83
  84 double av_strtod(const char *numstr, char **tail)
  85 {
  86     double d;
  87     char *next;
  88     if(numstr[0]=='0' && (numstr[1]|0x20)=='x') {
  89         d = strtoul(numstr, &next, 16);
  90     } else
  91         d = strtod(numstr, &next);
  92     /* if parsing succeeded, check for and interpret postfixes */
  93     if (next!=numstr) {
  94         if (*next >= 'E' && *next <= 'z') {
  95             int e= si_prefixes[*next - 'E'];
  96             if (e) {
  97                 if (next[1] == 'i') {
  98                     d*= pow( 2, e/0.3);
  99                     next+=2;
 100                 } else {
 101                     d*= pow(10, e);
 102                     next++;
 103                 }
 104             }
 105         }
 106
 107         if (*next=='B') {
 108             d*=8;
 109             next++;
 110         }
 111     }
 112     /* if requested, fill in tail with the position after the last parsed
 113        character */
 114     if (tail)
 115         *tail = next;
 116     return d;
 117 }
 118
 119 #define IS_IDENTIFIER_CHAR(c) ((c) - '0' <= 9U || (c) - 'a' <= 25U || (c) - 'A' <= 25U || (c) == '_')
 120
 121 static int strmatch(const char *s, const char *prefix)
 122 {
 123     int i;
 124     for (i=0; prefix[i]; i++) {
 125         if (prefix[i] != s[i]) return 0;
 126     }
 127     /* return 1 only if the s identifier is terminated */
 128     return !IS_IDENTIFIER_CHAR(s[i]);
 129 }
 130
 131 struct AVExpr {
 132     enum {
 133         e_value, e_const, e_func0, e_func1, e_func2,
 134         e_squish, e_gauss, e_ld, e_isnan,
 135         e_mod, e_max, e_min, e_eq, e_gt, e_gte,
 136         e_pow, e_mul, e_div, e_add,
 137         e_last, e_st, e_while, e_floor, e_ceil, e_trunc,
 138         e_sqrt, e_not, e_random, e_hypot, e_gcd,
 139         e_if, e_ifnot,
 140     } type;
 141     double value; // is sign in other types
 142     union {
 143         int const_index;
 144         double (*func0)(double);
 145         double (*func1)(void *, double);
 146         double (*func2)(void *, double, double);
 147     } a;
 148     struct AVExpr *param[2];
 149     double *var;
 150 };
 151
 152 static double eval_expr(Parser *p, AVExpr *e)
 153 {
 154     switch (e->type) {
 155         case e_value:  return e->value;
 156         case e_const:  return e->value * p->const_values[e->a.const_index];
 157         case e_func0:  return e->value * e->a.func0(eval_expr(p, e->param[0]));
 158         case e_func1:  return e->value * e->a.func1(p->opaque, eval_expr(p, e->param[0]));
 159         case e_func2:  return e->value * e->a.func2(p->opaque, eval_expr(p, e->param[0]), eval_expr(p, e->param[1]));
 160         case e_squish: return 1/(1+exp(4*eval_expr(p, e->param[0])));
 161         case e_gauss: { double d = eval_expr(p, e->param[0]); return exp(-d*d/2)/sqrt(2*M_PI); }
 162         case e_ld:     return e->value * p->var[av_clip(eval_expr(p, e->param[0]), 0, VARS-1)];
 163         case e_isnan:  return e->value * !!isnan(eval_expr(p, e->param[0]));
 164         case e_floor:  return e->value * floor(eval_expr(p, e->param[0]));
 165         case e_ceil :  return e->value * ceil (eval_expr(p, e->param[0]));
 166         case e_trunc:  return e->value * trunc(eval_expr(p, e->param[0]));
 167         case e_sqrt:   return e->value * sqrt (eval_expr(p, e->param[0]));
 168         case e_not:    return e->value * (eval_expr(p, e->param[0]) == 0);
 169         case e_if:     return e->value * ( eval_expr(p, e->param[0]) ? eval_expr(p, e->param[1]) : 0);
 170         case e_ifnot:  return e->value * (!eval_expr(p, e->param[0]) ? eval_expr(p, e->param[1]) : 0);
 171         case e_random:{
 172             int idx= av_clip(eval_expr(p, e->param[0]), 0, VARS-1);
 173             uint64_t r= isnan(p->var[idx]) ? 0 : p->var[idx];
 174             r= r*1664525+1013904223;
 175             p->var[idx]= r;
 176             return e->value * (r * (1.0/UINT64_MAX));
 177         }
 178         case e_while: {
 179             double d = NAN;
 180             while (eval_expr(p, e->param[0]))
 181                 d=eval_expr(p, e->param[1]);
 182             return d;
 183         }
 184         default: {
 185             double d = eval_expr(p, e->param[0]);
 186             double d2 = eval_expr(p, e->param[1]);
 187             switch (e->type) {
 188                 case e_mod: return e->value * (d - floor(d/d2)*d2);
 189                 case e_gcd: return e->value * av_gcd(d,d2);
 190                 case e_max: return e->value * (d >  d2 ?   d : d2);
 191                 case e_min: return e->value * (d <  d2 ?   d : d2);
 192                 case e_eq:  return e->value * (d == d2 ? 1.0 : 0.0);
 193                 case e_gt:  return e->value * (d >  d2 ? 1.0 : 0.0);
 194                 case e_gte: return e->value * (d >= d2 ? 1.0 : 0.0);
 195                 case e_pow: return e->value * pow(d, d2);
 196                 case e_mul: return e->value * (d * d2);
 197                 case e_div: return e->value * (d / d2);
 198                 case e_add: return e->value * (d + d2);
 199                 case e_last:return e->value * d2;
 200                 case e_st : return e->value * (p->var[av_clip(d, 0, VARS-1)]= d2);
 201                 case e_hypot:return e->value * (sqrt(d*d + d2*d2));
 202             }
 203         }
 204     }
 205     return NAN;
 206 }
 207
 208 static int parse_expr(AVExpr **e, Parser *p);
 209
 210 void av_expr_free(AVExpr *e)
 211 {
 212     if (!e) return;
 213     av_expr_free(e->param[0]);
 214     av_expr_free(e->param[1]);
 215     av_freep(&e->var);
 216     av_freep(&e);
 217 }
 218
 219 static int parse_primary(AVExpr **e, Parser *p)
 220 {
 221     AVExpr *d = av_mallocz(sizeof(AVExpr));
 222     char *next = p->s, *s0 = p->s;
 223     int ret, i;
 224
 225     if (!d)
 226         return AVERROR(ENOMEM);
 227
 228     /* number */
 229     d->value = av_strtod(p->s, &next);
 230     if (next != p->s) {
 231         d->type = e_value;
 232         p->s= next;
 233         *e = d;
 234         return 0;
 235     }
 236     d->value = 1;
 237
 238     /* named constants */
 239     for (i=0; p->const_names && p->const_names[i]; i++) {
 240         if (strmatch(p->s, p->const_names[i])) {
 241             p->s+= strlen(p->const_names[i]);
 242             d->type = e_const;
 243             d->a.const_index = i;
 244             *e = d;
 245             return 0;
 246         }
 247     }
 248     for (i = 0; i < FF_ARRAY_ELEMS(constants); i++) {
 249         if (strmatch(p->s, constants[i].name)) {
 250             p->s += strlen(constants[i].name);
 251             d->type = e_value;
 252             d->value = constants[i].value;
 253             *e = d;
 254             return 0;
 255         }
 256     }
 257
 258     p->s= strchr(p->s, '(');
 259     if (p->s==NULL) {
 260         av_log(p, AV_LOG_ERROR, "Undefined constant or missing '(' in '%s'\n", s0);
 261         p->s= next;
 262         av_expr_free(d);
 263         return AVERROR(EINVAL);
 264     }
 265     p->s++; // "("
 266     if (*next == '(') { // special case do-nothing
 267         av_freep(&d);
 268         if ((ret = parse_expr(&d, p)) < 0)
 269             return ret;
 270         if (p->s[0] != ')') {
 271             av_log(p, AV_LOG_ERROR, "Missing ')' in '%s'\n", s0);
 272             av_expr_free(d);
 273             return AVERROR(EINVAL);
 274         }
 275         p->s++; // ")"
 276         *e = d;
 277         return 0;
 278     }
 279     if ((ret = parse_expr(&(d->param[0]), p)) < 0) {
 280         av_expr_free(d);
 281         return ret;
 282     }
 283     if (p->s[0]== ',') {
 284         p->s++; // ","
 285         parse_expr(&d->param[1], p);
 286     }
 287     if (p->s[0] != ')') {
 288         av_log(p, AV_LOG_ERROR, "Missing ')' or too many args in '%s'\n", s0);
 289         av_expr_free(d);
 290         return AVERROR(EINVAL);
 291     }
 292     p->s++; // ")"
 293
 294     d->type = e_func0;
 295          if (strmatch(next, "sinh"  )) d->a.func0 = sinh;
 296     else if (strmatch(next, "cosh"  )) d->a.func0 = cosh;
 297     else if (strmatch(next, "tanh"  )) d->a.func0 = tanh;
 298     else if (strmatch(next, "sin"   )) d->a.func0 = sin;
 299     else if (strmatch(next, "cos"   )) d->a.func0 = cos;
 300     else if (strmatch(next, "tan"   )) d->a.func0 = tan;
 301     else if (strmatch(next, "atan"  )) d->a.func0 = atan;
 302     else if (strmatch(next, "asin"  )) d->a.func0 = asin;
 303     else if (strmatch(next, "acos"  )) d->a.func0 = acos;
 304     else if (strmatch(next, "exp"   )) d->a.func0 = exp;
 305     else if (strmatch(next, "log"   )) d->a.func0 = log;
 306     else if (strmatch(next, "abs"   )) d->a.func0 = fabs;
 307     else if (strmatch(next, "squish")) d->type = e_squish;
 308     else if (strmatch(next, "gauss" )) d->type = e_gauss;
 309     else if (strmatch(next, "mod"   )) d->type = e_mod;
 310     else if (strmatch(next, "max"   )) d->type = e_max;
 311     else if (strmatch(next, "min"   )) d->type = e_min;
 312     else if (strmatch(next, "eq"    )) d->type = e_eq;
 313     else if (strmatch(next, "gte"   )) d->type = e_gte;
 314     else if (strmatch(next, "gt"    )) d->type = e_gt;
 315     else if (strmatch(next, "lte"   )) { AVExpr *tmp = d->param[1]; d->param[1] = d->param[0]; d->param[0] = tmp; d->type = e_gt; }
 316     else if (strmatch(next, "lt"    )) { AVExpr *tmp = d->param[1]; d->param[1] = d->param[0]; d->param[0] = tmp; d->type = e_gte; }
 317     else if (strmatch(next, "ld"    )) d->type = e_ld;
 318     else if (strmatch(next, "isnan" )) d->type = e_isnan;
 319     else if (strmatch(next, "st"    )) d->type = e_st;
 320     else if (strmatch(next, "while" )) d->type = e_while;
 321     else if (strmatch(next, "floor" )) d->type = e_floor;
 322     else if (strmatch(next, "ceil"  )) d->type = e_ceil;
 323     else if (strmatch(next, "trunc" )) d->type = e_trunc;
 324     else if (strmatch(next, "sqrt"  )) d->type = e_sqrt;
 325     else if (strmatch(next, "not"   )) d->type = e_not;
 326     else if (strmatch(next, "pow"   )) d->type = e_pow;
 327     else if (strmatch(next, "random")) d->type = e_random;
 328     else if (strmatch(next, "hypot" )) d->type = e_hypot;
 329     else if (strmatch(next, "gcd"   )) d->type = e_gcd;
 330     else if (strmatch(next, "if"    )) d->type = e_if;
 331     else if (strmatch(next, "ifnot" )) d->type = e_ifnot;
 332     else {
 333         for (i=0; p->func1_names && p->func1_names[i]; i++) {
 334             if (strmatch(next, p->func1_names[i])) {
 335                 d->a.func1 = p->funcs1[i];
 336                 d->type = e_func1;
 337                 *e = d;
 338                 return 0;
 339             }
 340         }
 341
 342         for (i=0; p->func2_names && p->func2_names[i]; i++) {
 343             if (strmatch(next, p->func2_names[i])) {
 344                 d->a.func2 = p->funcs2[i];
 345                 d->type = e_func2;
 346                 *e = d;
 347                 return 0;
 348             }
 349         }
 350
 351         av_log(p, AV_LOG_ERROR, "Unknown function in '%s'\n", s0);
 352         av_expr_free(d);
 353         return AVERROR(EINVAL);
 354     }
 355
 356     *e = d;
 357     return 0;
 358 }
 359
 360 static AVExpr *new_eval_expr(int type, int value, AVExpr *p0, AVExpr *p1)
 361 {
 362     AVExpr *e = av_mallocz(sizeof(AVExpr));
 363     if (!e)
 364         return NULL;
 365     e->type     =type   ;
 366     e->value    =value  ;
 367     e->param[0] =p0     ;
 368     e->param[1] =p1     ;
 369     return e;
 370 }
 371
 372 static int parse_pow(AVExpr **e, Parser *p, int *sign)
 373 {
 374     *sign= (*p->s == '+') - (*p->s == '-');
 375     p->s += *sign&1;
 376     return parse_primary(e, p);
 377 }
 378
 379 static int parse_factor(AVExpr **e, Parser *p)
 380 {
 381     int sign, sign2, ret;
 382     AVExpr *e0, *e1, *e2;
 383     if ((ret = parse_pow(&e0, p, &sign)) < 0)
 384         return ret;
 385     while(p->s[0]=='^'){
 386         e1 = e0;
 387         p->s++;
 388         if ((ret = parse_pow(&e2, p, &sign2)) < 0) {
 389             av_expr_free(e1);
 390             return ret;
 391         }
 392         e0 = new_eval_expr(e_pow, 1, e1, e2);
 393         if (!e0) {
 394             av_expr_free(e1);
 395             av_expr_free(e2);
 396             return AVERROR(ENOMEM);
 397         }
 398         if (e0->param[1]) e0->param[1]->value *= (sign2|1);
 399     }
 400     if (e0) e0->value *= (sign|1);
 401
 402     *e = e0;
 403     return 0;
 404 }
 405
 406 static int parse_term(AVExpr **e, Parser *p)
 407 {
 408     int ret;
 409     AVExpr *e0, *e1, *e2;
 410     if ((ret = parse_factor(&e0, p)) < 0)
 411         return ret;
 412     while (p->s[0]=='*' || p->s[0]=='/') {
 413         int c= *p->s++;
 414         e1 = e0;
 415         if ((ret = parse_factor(&e2, p)) < 0) {
 416             av_expr_free(e1);
 417             return ret;
 418         }
 419         e0 = new_eval_expr(c == '*' ? e_mul : e_div, 1, e1, e2);
 420         if (!e0) {
 421             av_expr_free(e1);
 422             av_expr_free(e2);
 423             return AVERROR(ENOMEM);
 424         }
 425     }
 426     *e = e0;
 427     return 0;
 428 }
 429
 430 static int parse_subexpr(AVExpr **e, Parser *p)
 431 {
 432     int ret;
 433     AVExpr *e0, *e1, *e2;
 434     if ((ret = parse_term(&e0, p)) < 0)
 435         return ret;
 436     while (*p->s == '+' || *p->s == '-') {
 437         e1 = e0;
 438         if ((ret = parse_term(&e2, p)) < 0) {
 439             av_expr_free(e1);
 440             return ret;
 441         }
 442         e0 = new_eval_expr(e_add, 1, e1, e2);
 443         if (!e0) {
 444             av_expr_free(e1);
 445             av_expr_free(e2);
 446             return AVERROR(ENOMEM);
 447         }
 448     };
 449
 450     *e = e0;
 451     return 0;
 452 }
 453
 454 static int parse_expr(AVExpr **e, Parser *p)
 455 {
 456     int ret;
 457     AVExpr *e0, *e1, *e2;
 458     if (p->stack_index <= 0) //protect against stack overflows
 459         return AVERROR(EINVAL);
 460     p->stack_index--;
 461
 462     if ((ret = parse_subexpr(&e0, p)) < 0)
 463         return ret;
 464     while (*p->s == ';') {
 465         p->s++;
 466         e1 = e0;
 467         if ((ret = parse_subexpr(&e2, p)) < 0) {
 468             av_expr_free(e1);
 469             return ret;
 470         }
 471         e0 = new_eval_expr(e_last, 1, e1, e2);
 472         if (!e0) {
 473             av_expr_free(e1);
 474             av_expr_free(e2);
 475             return AVERROR(ENOMEM);
 476         }
 477     };
 478
 479     p->stack_index++;
 480     *e = e0;
 481     return 0;
 482 }
 483
 484 static int verify_expr(AVExpr *e)
 485 {
 486     if (!e) return 0;
 487     switch (e->type) {
 488         case e_value:
 489         case e_const: return 1;
 490         case e_func0:
 491         case e_func1:
 492         case e_squish:
 493         case e_ld:
 494         case e_gauss:
 495         case e_isnan:
 496         case e_floor:
 497         case e_ceil:
 498         case e_trunc:
 499         case e_sqrt:
 500         case e_not:
 501         case e_random:
 502             return verify_expr(e->param[0]);
 503         default: return verify_expr(e->param[0]) && verify_expr(e->param[1]);
 504     }
 505 }
 506
 507 int av_expr_parse(AVExpr **expr, const char *s,
 508                   const char * const *const_names,
 509                   const char * const *func1_names, double (* const *funcs1)(void *, double),
 510                   const char * const *func2_names, double (* const *funcs2)(void *, double, double),
 511                   int log_offset, void *log_ctx)
 512 {
 513     Parser p = { 0 };
 514     AVExpr *e = NULL;
 515     char *w = av_malloc(strlen(s) + 1);
 516     char *wp = w;
 517     const char *s0 = s;
 518     int ret = 0;
 519
 520     if (!w)
 521         return AVERROR(ENOMEM);
 522
 523     while (*s)
 524         if (!isspace(*s++)) *wp++ = s[-1];
 525     *wp++ = 0;
 526
 527     p.class      = &class;
 528     p.stack_index=100;
 529     p.s= w;
 530     p.const_names = const_names;
 531     p.funcs1      = funcs1;
 532     p.func1_names = func1_names;
 533     p.funcs2      = funcs2;
 534     p.func2_names = func2_names;
 535     p.log_offset = log_offset;
 536     p.log_ctx    = log_ctx;
 537
 538     if ((ret = parse_expr(&e, &p)) < 0)
 539         goto end;
 540     if (*p.s) {
 541         av_expr_free(e);
 542         av_log(&p, AV_LOG_ERROR, "Invalid chars '%s' at the end of expression '%s'\n", p.s, s0);
 543         ret = AVERROR(EINVAL);
 544         goto end;
 545     }
 546     if (!verify_expr(e)) {
 547         av_expr_free(e);
 548         ret = AVERROR(EINVAL);
 549         goto end;
 550     }
 551     e->var= av_mallocz(sizeof(double) *VARS);
 552     *expr = e;
 553 end:
 554     av_free(w);
 555     return ret;
 556 }
 557
 558 double av_expr_eval(AVExpr *e, const double *const_values, void *opaque)
 559 {
 560     Parser p = { 0 };
 561     p.var= e->var;
 562
 563     p.const_values = const_values;
 564     p.opaque     = opaque;
 565     return eval_expr(&p, e);
 566 }
 567
 568 int av_expr_parse_and_eval(double *d, const char *s,
 569                            const char * const *const_names, const double *const_values,
 570                            const char * const *func1_names, double (* const *funcs1)(void *, double),
 571                            const char * const *func2_names, double (* const *funcs2)(void *, double, double),
 572                            void *opaque, int log_offset, void *log_ctx)
 573 {
 574     AVExpr *e = NULL;
 575     int ret = av_expr_parse(&e, s, const_names, func1_names, funcs1, func2_names, funcs2, log_offset, log_ctx);
 576
 577     if (ret < 0) {
 578         *d = NAN;
 579         return ret;
 580     }
 581     *d = av_expr_eval(e, const_values, opaque);
 582     av_expr_free(e);
 583     return isnan(*d) ? AVERROR(EINVAL) : 0;
 584 }
 585
 586 #if FF_API_OLD_EVAL_NAMES
 587 // LCOV_EXCL_START
 588 int av_parse_expr(AVExpr **expr, const char *s,
 589                   const char * const *const_names,
 590                   const char * const *func1_names, double (* const *funcs1)(void *, double),
 591                   const char * const *func2_names, double (* const *funcs2)(void *, double, double),
 592                   int log_offset, void *log_ctx)
 593 {
 594     return av_expr_parse(expr, s, const_names, func1_names, funcs1, func2_names, funcs2,
 595                       log_offset, log_ctx);
 596 }
 597
 598 double av_eval_expr(AVExpr *e, const double *const_values, void *opaque)
 599 {
 600     return av_expr_eval(e, const_values, opaque);
 601 }
 602
 603 int av_parse_and_eval_expr(double *res, const char *s,
 604                            const char * const *const_names, const double *const_values,
 605                            const char * const *func1_names, double (* const *funcs1)(void *, double),
 606                            const char * const *func2_names, double (* const *funcs2)(void *, double, double),
 607                            void *opaque, int log_offset, void *log_ctx)
 608 {
 609     return av_expr_parse_and_eval(res, s, const_names, const_values, func1_names, funcs1, func2_names, funcs2,
 610                                   opaque, log_offset, log_ctx);
 611 }
 612
 613 void av_free_expr(AVExpr *e)
 614 {
 615     av_expr_free(e);
 616 }
 617 // LCOV_EXCL_STOP
 618 #endif /* FF_API_OLD_EVAL_NAMES */
 619
 620 #ifdef TEST
 621 // LCOV_EXCL_START
 622 #undef printf
 623 #include <string.h>
 624
 625 static double const_values[] = {
 626     M_PI,
 627     M_E,
 628     0
 629 };
 630
 631 static const char *const_names[] = {
 632     "PI",
 633     "E",
 634     0
 635 };
 636
 637 int main(int argc, char **argv)
 638 {
 639     int i;
 640     double d;
 641     const char **expr, *exprs[] = {
 642         "",
 643         "1;2",
 644         "-20",
 645         "-PI",
 646         "+PI",
 647         "1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)",
 648         "80G/80Gi",
 649         "1k",
 650         "1Gi",
 651         "1gi",
 652         "1GiFoo",
 653         "1k+1k",
 654         "1Gi*3foo",
 655         "foo",
 656         "foo(",
 657         "foo()",
 658         "foo)",
 659         "sin",
 660         "sin(",
 661         "sin()",
 662         "sin)",
 663         "sin 10",
 664         "sin(1,2,3)",
 665         "sin(1 )",
 666         "1",
 667         "1foo",
 668         "bar + PI + E + 100f*2 + foo",
 669         "13k + 12f - foo(1, 2)",
 670         "1gi",
 671         "1Gi",
 672         "st(0, 123)",
 673         "st(1, 123); ld(1)",
 674         /* compute 1+2+...+N */
 675         "st(0, 1); while(lte(ld(0), 100), st(1, ld(1)+ld(0));st(0, ld(0)+1)); ld(1)",
 676         /* compute Fib(N) */
 677         "st(1, 1); st(2, 2); st(0, 1); while(lte(ld(0),10), st(3, ld(1)+ld(2)); st(1, ld(2)); st(2, ld(3)); st(0, ld(0)+1)); ld(3)",
 678         "while(0, 10)",
 679         "st(0, 1); while(lte(ld(0),100), st(1, ld(1)+ld(0)); st(0, ld(0)+1))",
 680         "isnan(1)",
 681         "isnan(NAN)",
 682         "floor(NAN)",
 683         "floor(123.123)",
 684         "floor(-123.123)",
 685         "trunc(123.123)",
 686         "trunc(-123.123)",
 687         "ceil(123.123)",
 688         "ceil(-123.123)",
 689         "sqrt(1764)",
 690         "isnan(sqrt(-1))",
 691         "not(1)",
 692         "not(NAN)",
 693         "not(0)",
 694         "pow(0,1.23)",
 695         "pow(PI,1.23)",
 696         "PI^1.23",
 697         "pow(-1,1.23)",
 698         "if(1, 2)",
 699         "ifnot(0, 23)",
 700         "ifnot(1, NaN) + if(0, 1)",
 701         NULL
 702     };
 703
 704     for (expr = exprs; *expr; expr++) {
 705         printf("Evaluating '%s'\n", *expr);
 706         av_expr_parse_and_eval(&d, *expr,
 707                                const_names, const_values,
 708                                NULL, NULL, NULL, NULL, NULL, 0, NULL);
 709         if(isnan(d)){
 710             printf("'%s' -> nan\n\n", *expr);
 711         }else{
 712             printf("'%s' -> %f\n\n", *expr, d);
 713         }
 714     }
 715
 716     av_expr_parse_and_eval(&d, "1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)",
 717                            const_names, const_values,
 718                            NULL, NULL, NULL, NULL, NULL, 0, NULL);
 719     printf("%f == 12.7\n", d);
 720     av_expr_parse_and_eval(&d, "80G/80Gi",
 721                            const_names, const_values,
 722                            NULL, NULL, NULL, NULL, NULL, 0, NULL);
 723     printf("%f == 0.931322575\n", d);
 724
 725     if (argc > 1 && !strcmp(argv[1], "-t")) {
 726         for (i = 0; i < 1050; i++) {
 727             START_TIMER;
 728             av_expr_parse_and_eval(&d, "1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)",
 729                                    const_names, const_values,
 730                                    NULL, NULL, NULL, NULL, NULL, 0, NULL);
 731             STOP_TIMER("av_expr_parse_and_eval");
 732         }
 733     }
 734
 735     return 0;
 736 }
 737 // LCOV_EXCL_STOP
 738 #endif