feat: add std.atan2

Fixes: google#1118
CertainLach · Nov 30, 2023 · a22be2f · a22be2f
1 parent 0679858
commit a22be2f
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/doc/_stdlib_gen/stdlib-content.jsonnet b/doc/_stdlib_gen/stdlib-content.jsonnet
@@ -109,6 +109,7 @@ local html = import 'html.libsonnet';
               <ul><code>std.asin(x)</code></ul>
               <ul><code>std.acos(x)</code></ul>
               <ul><code>std.atan(x)</code></ul>
+              <ul><code>std.atan2(y, x)</code></ul>
               <ul><code>std.round(x)</code></ul>
               <ul><code>std.isEven(x)</code></ul>
               <ul><code>std.isOdd(x)</code></ul>

diff --git a/stdlib/std.jsonnet b/stdlib/std.jsonnet
@@ -1785,4 +1785,20 @@ limitations under the License.
   sha3(str):: go_only_function,
 
   trim(str):: std.stripChars(str, ' \t\n\f\r\u0085\u00A0'),
+
+  atan2(y, x)::
+    local pi = std.acos(-1);
+    if x > 0 then
+      std.atan(y / x)
+    else if x < 0 && y >= 0 then
+      std.atan(y / x) + pi
+    else if x < 0 && y < 0 then
+      std.atan(y / x) - pi
+    else if x == 0 && y > 0 then
+      pi / 2
+    else if x == 0 && y < 0 then
+      -pi / 2
+    else
+      0
+  ,
 }