summaryrefslogtreecommitdiff
path: root/libavutil/eval.c
diff options
context:
space:
mode:
authorStefano Sabatini <stefano.sabatini-lala@poste.it>2011-04-09 13:49:54 +0200
committerAnton Khirnov <anton@khirnov.net>2011-04-10 09:24:21 +0200
commit5ae4251c3b603cc7f44dd0e0839c5b6f6bbac61e (patch)
tree22c45f3f3115c2880fb5c6af3b0502d2e911c193 /libavutil/eval.c
parent2fd8756b9f5f5e809bec8f2635216f236106dff4 (diff)
eval: add support for trunc, ceil, and floor functions
Signed-off-by: Stefano Sabatini <stefano.sabatini-lala@poste.it> Signed-off-by: Anton Khirnov <anton@khirnov.net>
Diffstat (limited to 'libavutil/eval.c')
-rw-r--r--libavutil/eval.c21
1 files changed, 19 insertions, 2 deletions
diff --git a/libavutil/eval.c b/libavutil/eval.c
index 8994393cdb..7488a310a7 100644
--- a/libavutil/eval.c
+++ b/libavutil/eval.c
@@ -121,7 +121,7 @@ struct AVExpr {
e_squish, e_gauss, e_ld, e_isnan,
e_mod, e_max, e_min, e_eq, e_gt, e_gte,
e_pow, e_mul, e_div, e_add,
- e_last, e_st, e_while,
+ e_last, e_st, e_while, e_floor, e_ceil, e_trunc,
} type;
double value; // is sign in other types
union {
@@ -145,6 +145,9 @@ static double eval_expr(Parser *p, AVExpr *e)
case e_gauss: { double d = eval_expr(p, e->param[0]); return exp(-d*d/2)/sqrt(2*M_PI); }
case e_ld: return e->value * p->var[av_clip(eval_expr(p, e->param[0]), 0, VARS-1)];
case e_isnan: return e->value * !!isnan(eval_expr(p, e->param[0]));
+ case e_floor: return e->value * floor(eval_expr(p, e->param[0]));
+ case e_ceil : return e->value * ceil (eval_expr(p, e->param[0]));
+ case e_trunc: return e->value * trunc(eval_expr(p, e->param[0]));
case e_while: {
double d = NAN;
while (eval_expr(p, e->param[0]))
@@ -276,6 +279,9 @@ static int parse_primary(AVExpr **e, Parser *p)
else if (strmatch(next, "isnan" )) d->type = e_isnan;
else if (strmatch(next, "st" )) d->type = e_st;
else if (strmatch(next, "while" )) d->type = e_while;
+ else if (strmatch(next, "floor" )) d->type = e_floor;
+ else if (strmatch(next, "ceil" )) d->type = e_ceil;
+ else if (strmatch(next, "trunc" )) d->type = e_trunc;
else {
for (i=0; p->func1_names && p->func1_names[i]; i++) {
if (strmatch(next, p->func1_names[i])) {
@@ -439,7 +445,11 @@ static int verify_expr(AVExpr *e)
case e_squish:
case e_ld:
case e_gauss:
- case e_isnan: return verify_expr(e->param[0]);
+ case e_isnan:
+ case e_floor:
+ case e_ceil:
+ case e_trunc:
+ return verify_expr(e->param[0]);
default: return verify_expr(e->param[0]) && verify_expr(e->param[1]);
}
}
@@ -611,6 +621,13 @@ int main(void)
"st(0, 1); while(lte(ld(0),100), st(1, ld(1)+ld(0)); st(0, ld(0)+1))",
"isnan(1)",
"isnan(NAN)",
+ "floor(NAN)",
+ "floor(123.123)",
+ "floor(-123.123)",
+ "trunc(123.123)",
+ "trunc(-123.123)",
+ "ceil(123.123)",
+ "ceil(-123.123)",
NULL
};