1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

use num_traits::Float;

use crate::array::PrimitiveArray;
use crate::compute::arity::binary;
use crate::types::NativeType;

pub fn atan2<T: NativeType>(
    arr_1: &PrimitiveArray<T>,
    arr_2: &PrimitiveArray<T>,
) -> PrimitiveArray<T>
where
    T: Float,
{
    binary(arr_1, arr_2, arr_1.data_type().clone(), |a, b| a.atan2(b))
}