Merge pull request #116 from nkint/patch-1

Add atan signature with two input
thi-ng · Aug 7, 2019 · be94f59 · be94f59
2 parents f8decf8 + d97dd08
commit be94f59
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/packages/shader-ast/src/builtins.ts b/packages/shader-ast/src/builtins.ts
@@ -117,7 +117,17 @@ export const sin = primOp1("sin");
 export const tan = primOp1("tan");
 export const acos = primOp1("acos");
 export const asin = primOp1("asin");
-export const atan = primOp1("atan");
+
+export function atan<T extends Prim>(a: Term<T>): FnCall<T>;
+export function atan<A extends Prim, B extends A>(
+    a: Term<A>,
+    b: Term<B>
+): FnCall<A>;
+export function atan(a:Term<any>, b?: Term<any>) {
+    return b
+        ? builtinCall("atan", a.type, a, b)
+        : builtinCall("atan", a.type, a);
+}
 
 export const pow = primOp2("pow");
 export const exp = primOp1("exp");