diff mbox

[FFmpeg-devel] avutil/eval: add atan2 function

Message ID 20170129125322.23442-1-onemda@gmail.com
State Accepted
Commit 13564fc24d8248a1f9a987ce1bbd483c72609438
Headers show

Commit Message

Paul B Mahol Jan. 29, 2017, 12:53 p.m. UTC 
Signed-off-by: Paul B Mahol <onemda@gmail.com>
---
 doc/utils.texi   | 3 +++
 libavutil/eval.c | 4 +++-
 2 files changed, 6 insertions(+), 1 deletion(-)

Comments

Michael Niedermayer Jan. 30, 2017, 3:20 a.m. UTC | #1 
On Sun, Jan 29, 2017 at 01:53:22PM +0100, Paul B Mahol wrote:
> Signed-off-by: Paul B Mahol <onemda@gmail.com>
> ---
>  doc/utils.texi   | 3 +++
>  libavutil/eval.c | 4 +++-
>  2 files changed, 6 insertions(+), 1 deletion(-)

LGTM

thx

[...]
diff mbox

Patch

diff --git a/doc/utils.texi b/doc/utils.texi
index 30a962a..1734439 100644
--- a/doc/utils.texi
+++ b/doc/utils.texi
@@ -776,6 +776,9 @@  Compute arcsine of @var{x}.
 @item atan(x)
 Compute arctangent of @var{x}.
 
+@item atan2(x, y)
+Compute principal value of the arc tangent of @var{y}/@var{x}.
+
 @item between(x, min, max)
 Return 1 if @var{x} is greater than or equal to @var{min} and lesser than or
 equal to @var{max}, 0 otherwise.
diff --git a/libavutil/eval.c b/libavutil/eval.c
index 2d49154..7e86615 100644
--- a/libavutil/eval.c
+++ b/libavutil/eval.c
@@ -155,7 +155,7 @@  struct AVExpr {
         e_pow, e_mul, e_div, e_add,
         e_last, e_st, e_while, e_taylor, e_root, e_floor, e_ceil, e_trunc,
         e_sqrt, e_not, e_random, e_hypot, e_gcd,
-        e_if, e_ifnot, e_print, e_bitand, e_bitor, e_between, e_clip
+        e_if, e_ifnot, e_print, e_bitand, e_bitor, e_between, e_clip, e_atan2
     } type;
     double value; // is sign in other types
     union {
@@ -306,6 +306,7 @@  static double eval_expr(Parser *p, AVExpr *e)
                 case e_last:return e->value * d2;
                 case e_st : return e->value * (p->var[av_clip(d, 0, VARS-1)]= d2);
                 case e_hypot:return e->value * hypot(d, d2);
+                case e_atan2:return e->value * atan2(d, d2);
                 case e_bitand: return isnan(d) || isnan(d2) ? NAN : e->value * ((long int)d & (long int)d2);
                 case e_bitor:  return isnan(d) || isnan(d2) ? NAN : e->value * ((long int)d | (long int)d2);
             }
@@ -452,6 +453,7 @@  static int parse_primary(AVExpr **e, Parser *p)
     else if (strmatch(next, "bitor" )) d->type = e_bitor;
     else if (strmatch(next, "between"))d->type = e_between;
     else if (strmatch(next, "clip"  )) d->type = e_clip;
+    else if (strmatch(next, "atan2" )) d->type = e_atan2;
     else {
         for (i=0; p->func1_names && p->func1_names[i]; i++) {
             if (strmatch(next, p->func1_names[i])) {