Adding more functions

Author: lshaw8317

src/blosc2/__init__.py

@@ -330,7 +330,6 @@ def _raise(exc):
     are_partitions_behaved,
     arange,
     broadcast_to,
-    bitwise_xor,
     linspace,
     eye,
     asarray,
@@ -440,6 +439,9 @@ def _raise(exc):
     atan,
     atan2,
     atanh,
+    bitwise_invert,
+    bitwise_or,
+    bitwise_xor,
     conj,
     contains,
     cos,
@@ -459,12 +461,15 @@ def _raise(exc):
     mean,
     min,
     multiply,
+    pow,
     prod,
     real,
     sin,
     sinh,
     sqrt,
+    square,
     std,
+    subtract,
     sum,
     tan,
     tanh,
@@ -543,6 +548,8 @@ def _raise(exc):
     "atan",
     "atan2",
     "atanh",
+    "bitwise_invert",
+    "bitwise_or",
     "bitwise_xor",
     "broadcast_to",
     "clib_info",
@@ -611,6 +618,7 @@ def _raise(exc):
     "pack_tensor",
     "permute_dims",
     "postfilter_funcs",
+    "pow",
     "prefilter_funcs",
     "print_versions",
     "prod",
@@ -628,9 +636,11 @@ def _raise(exc):
     "set_nthreads",
     "set_releasegil",
     "sort",
+    "square",
     "stack",
     "storage_dflts",
     "sum",
+    "subtract",
     "tan",
     "tanh",
     "tensordot",

src/blosc2/ndarray.py

@@ -887,7 +887,7 @@ def __array_ufunc__(self, ufunc, method, *inputs, **kwargs):
             np.conj: "conj",
             np.real: "real",
             np.imag: "imag",
-            np.bitwise_not: "~",
+            np.bitwise_invert: "~",
             np.isnan: "isnan",
             np.isfinite: "isfinite",
             np.isinf: "isinf",
@@ -2429,6 +2429,12 @@ def __matmul__(self, other):
     def __xor__(self, other):
         return bitwise_xor(self, other)
 
+    def __or__(self, other):
+        return bitwise_or(self, other)
+
+    def __invert__(self):
+        return bitwise_invert(self)
+
 
 def array_from_ffi_ptr(array_ptr) -> NDArray:
     """
@@ -3547,6 +3553,9 @@ def pow(
     x1: NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr
         First input array. May have any data type.
 
+    x2:NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr
+        Second input array. Must be compatible with x1. May have any data type.
+
     Returns
     -------
     out LazyExpr
@@ -3572,6 +3581,9 @@ def bitwise_xor(
     x1: NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr
         First input array. May have any data type.
 
+    x2:NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr
+        Second input array. Must be compatible with x1. May have any data type.
+
     Returns
     -------
     out LazyExpr
@@ -3584,6 +3596,57 @@ def bitwise_xor(
     return x1 ^ x2
 
 
+def bitwise_or(
+    x1: NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr | int | float | complex,
+    x2: NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr | int | float | complex,
+) -> blosc2.LazyExpr:
+    """
+    Computes the value of x1_i | x2_i for each element x1_i of the input array x1 and x2_i
+    of x2.
+
+    Parameters
+    -----------
+    x1: NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr
+        First input array. May have any data type.
+
+    x2: NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr
+        Second input array. Must be compatible with x1. May have any data type.
+
+    Returns
+    -------
+    out LazyExpr
+        A LazyArray containing the element-wise results.
+
+    References
+    ----------
+    `np.bitwise_or <https://numpy.org/doc/stable/reference/generated/numpy.bitwise_or.html#numpy.bitwise_or>`_
+    """
+    return x1 | x2
+
+
+def bitwise_invert(
+    x1: NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr | int | float | complex,
+) -> blosc2.LazyExpr:
+    """
+    Computes the value of ~x1_i for each element x1_i of the input array x1.
+
+    Parameters
+    -----------
+    x1: NDArray | NDField | blosc2.C2Array | blosc2.LazyExpr
+        First input array. May have any data type.
+
+    Returns
+    -------
+    out LazyExpr
+        A LazyArray containing the element-wise results.
+
+    References
+    ----------
+    `np.bitwise_invert <https://numpy.org/doc/stable/reference/generated/numpy.bitwise_invert.html#numpy.bitwise_invert>`_
+    """
+    return ~x1
+
+
 def where(
     condition: blosc2.LazyExpr | NDArray,
     x: NDArray | NDField | np.ndarray | int | float | complex | bool | str | bytes | None = None,