Skip to content
Snippets Groups Projects
Commit 48bcd760 authored by Frederik Hennig's avatar Frederik Hennig
Browse files

atan2

parent 16c9fce5
No related branches found
No related tags found
1 merge request!394Extend symbolic language support
......@@ -51,6 +51,7 @@ class MathFunctions(Enum):
Max = ("max", 2)
Pow = ("pow", 2)
ATan2 = ("atan2", 2)
def __init__(self, func_name, num_args):
self.function_name = func_name
......
......@@ -387,6 +387,8 @@ class FreezeExpressions:
return PsCall(PsMathFunction(MathFunctions.ACos), args)
case sp.atan():
return PsCall(PsMathFunction(MathFunctions.ATan), args)
case sp.atan2():
return PsCall(PsMathFunction(MathFunctions.ATan2), args)
case integer_functions.int_div():
return PsIntDiv(*args)
case integer_functions.bit_shift_left():
......
......@@ -77,6 +77,7 @@ class GenericCpu(Platform):
| MathFunctions.ASin
| MathFunctions.ACos
| MathFunctions.ATan
| MathFunctions.ATan2
| MathFunctions.Pow
| MathFunctions.Floor
| MathFunctions.Ceil
......
......@@ -24,6 +24,7 @@ BINARY_FUNCTIONS = {
"min": (sp.Min, np.fmin),
"max": (sp.Max, np.fmax),
"pow": (sp.Pow, np.power),
"atan2": (sp.atan2, np.arctan2),
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment