Added scalar bitwise ops

src/compute/bitwise.rs

@@ -5,7 +5,7 @@ use crate::array::PrimitiveArray;
 use crate::compute::arity::{binary, unary};
 use crate::types::NativeType;
 
-/// Performs `OR` operation on two arrays.
+/// Performs `OR` operation on two [`PrimitiveArray`]s.
 /// # Panic
 /// This function errors when the arrays have different lengths.
 pub fn or<T>(lhs: &PrimitiveArray<T>, rhs: &PrimitiveArray<T>) -> PrimitiveArray<T>
@@ -15,7 +15,7 @@ where
     binary(lhs, rhs, lhs.data_type().clone(), |a, b| a | b)
 }
 
-/// Performs `XOR` operation between two arrays.
+/// Performs `XOR` operation between two [`PrimitiveArray`]s.
 /// # Panic
 /// This function errors when the arrays have different lengths.
 pub fn xor<T>(lhs: &PrimitiveArray<T>, rhs: &PrimitiveArray<T>) -> PrimitiveArray<T>
@@ -25,7 +25,7 @@ where
     binary(lhs, rhs, lhs.data_type().clone(), |a, b| a ^ b)
 }
 
-/// Performs `AND` operation on two arrays.
+/// Performs `AND` operation on two [`PrimitiveArray`]s.
 /// # Panic
 /// This function panics when the arrays have different lengths.
 pub fn and<T>(lhs: &PrimitiveArray<T>, rhs: &PrimitiveArray<T>) -> PrimitiveArray<T>
@@ -43,3 +43,33 @@ where
     let op = move |a: T| !a;
     unary(array, op, array.data_type().clone())
 }
+
+/// Performs `OR` operation between a [`PrimitiveArray`] and scalar.
+/// # Panic
+/// This function errors when the arrays have different lengths.
+pub fn or_scalar<T>(lhs: &PrimitiveArray<T>, rhs: &T) -> PrimitiveArray<T>
+where
+    T: NativeType + BitOr<Output = T>,
+{
+    unary(lhs, |a| a | *rhs, lhs.data_type().clone())
+}
+
+/// Performs `XOR` operation between a [`PrimitiveArray`] and scalar.
+/// # Panic
+/// This function errors when the arrays have different lengths.
+pub fn xor_scalar<T>(lhs: &PrimitiveArray<T>, rhs: &T) -> PrimitiveArray<T>
+where
+    T: NativeType + BitXor<Output = T>,
+{
+    unary(lhs, |a| a ^ *rhs, lhs.data_type().clone())
+}
+
+/// Performs `AND` operation between a [`PrimitiveArray`] and scalar.
+/// # Panic
+/// This function panics when the arrays have different lengths.
+pub fn and_scalar<T>(lhs: &PrimitiveArray<T>, rhs: &T) -> PrimitiveArray<T>
+where
+    T: NativeType + BitAnd<Output = T>,
+{
+    unary(lhs, |a| a & *rhs, lhs.data_type().clone())
+}