feat: add std.atan2

Fixes: google#1118
CertainLach · Nov 30, 2023 · a8d47a8 · a8d47a8
1 parent 0679858
commit a8d47a8
Showing 1 changed file with 16 additions and 0 deletions.
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
+ ,
 }