Add missing functions (arrayfire#13)

Author: roaffix

arrayfire_wrapper/lib/__init__.py

@@ -202,6 +202,7 @@
     "lock_device_ptr",
     "print_mem_info",
     "set_device",
+    "sync",
     "set_kernel_cache_directory",
     "set_mem_step_size",
     "unlock_array",
@@ -235,6 +236,7 @@
     set_device,
     set_kernel_cache_directory,
     set_mem_step_size,
+    sync,
     unlock_array,
     unlock_device_ptr,
 )
@@ -745,6 +747,7 @@
 # Features
 
 __all__ += [
+    "AFFeatures",
     "create_features",
     "get_features_num",
     "get_features_orientation",
@@ -756,6 +759,7 @@
     "retain_features",
 ]
 from .features import (
+    AFFeatures,
     create_features,
     get_features_num,
     get_features_orientation,
@@ -848,6 +852,7 @@
     "ifft3",
     "ifft3_inplace",
     "ifft_inplace",
+    "set_fft_plan_cache_size",
 ]
 
 from .signal_processing.fast_fourier_transforms import (
@@ -869,6 +874,7 @@
     ifft3,
     ifft3_inplace,
     ifft_inplace,
+    set_fft_plan_cache_size,
 )
 
 __all__ += ["fir", "iir"]
@@ -925,6 +931,30 @@
 
 # Constants
 
-__all__ += ["Match", "Moment", "Pad", "PointerSource", "TopK", "VarianceBias", "MatProp", "ImageFormat"]
+__all__ += [
+    "Match",
+    "Moment",
+    "Pad",
+    "PointerSource",
+    "TopK",
+    "VarianceBias",
+    "MatProp",
+    "ImageFormat",
+    "ConvMode",
+    "ConvDomain",
+    "ConvGradient",
+]
 
-from ._constants import ImageFormat, Match, MatProp, Moment, Pad, PointerSource, TopK, VarianceBias
+from ._constants import (
+    ConvDomain,
+    ConvGradient,
+    ConvMode,
+    ImageFormat,
+    Match,
+    MatProp,
+    Moment,
+    Pad,
+    PointerSource,
+    TopK,
+    VarianceBias,
+)

arrayfire_wrapper/lib/create_and_modify_array/manage_array.py

@@ -89,11 +89,11 @@ def eval(arr: AFArray, /) -> None:
     return None
 
 
-def eval_multiple(arr: AFArray, data: int, /) -> None:
+def eval_multiple(num: int, /, *arrs: AFArray) -> None:
     """
     source: https://arrayfire.org/docs/group__c__api__mat.htm#ga9e08f4cda2471a477d2fa91c2356f72c
     """
-    call_from_clib(eval_multiple.__name__, ctypes.c_int(data), arr)
+    call_from_clib(eval_multiple.__name__, ctypes.c_int(num), ctypes.pointer(arrs))
     return None
 
 

arrayfire_wrapper/lib/create_and_modify_array/manage_device.py

@@ -420,8 +420,20 @@ def set_device(device_id: int, /) -> None:
     Change the active device to the specified id.
 
     Parameters
-    -----------
+    ----------
     num: int.
          id of the desired device.
     """
     call_from_clib(set_device.__name__, device_id)
+
+
+def sync(device_id: int, /) -> None:
+    """
+    Blocks until all operations on device are finished.
+
+    Parameters
+    ----------
+    device_id: int
+        Id of the target device.
+    """
+    call_from_clib(sync.__name__, device_id)

arrayfire_wrapper/lib/create_and_modify_array/move_and_reorder.py

@@ -55,7 +55,7 @@ def moddims(arr: AFArray, shape: tuple[int, ...], /) -> AFArray:
     return out
 
 
-def reorder(arr: AFArray, d0: int = 1, d1: int = 0, d2: int = 2, d3: int = 3, /) -> AFArray:
+def reorder(arr: AFArray, /, d0: int = 1, d1: int = 0, d2: int = 2, d3: int = 3) -> AFArray:
     """
     source: https://arrayfire.org/docs/group__manip__func__reorder.htm#ga57383f4d00a3a86eab08dddd52c3ad3d
     """
@@ -99,7 +99,7 @@ def select_scalar_r(lhs: AFArray, cond_arr: AFArray, rhs: int | float, /) -> Non
     call_from_clib(select_scalar_l.__name__, cond_arr, lhs, ctypes.c_double(rhs))
 
 
-def shift(arr: AFArray, d0: int, d1: int = 0, d2: int = 0, d3: int = 0, /) -> AFArray:
+def shift(arr: AFArray, /, d0: int, d1: int = 0, d2: int = 0, d3: int = 0) -> AFArray:
     """
     source: https://arrayfire.org/docs/group__manip__func__shift.htm#ga64a0cd7680b71e87f3ab372876153b66
     """
@@ -108,7 +108,7 @@ def shift(arr: AFArray, d0: int, d1: int = 0, d2: int = 0, d3: int = 0, /) -> AF
     return out
 
 
-def tile(arr: AFArray, d0: int, d1: int = 1, d2: int = 1, d3: int = 1, /) -> AFArray:
+def tile(arr: AFArray, /, d0: int, d1: int = 1, d2: int = 1, d3: int = 1) -> AFArray:
     """
     source: https://arrayfire.org/docs/group__manip__func__tile.htm#ga3540329723c9876839e0c790075ab076
     """

arrayfire_wrapper/lib/image_processing/filters.py

@@ -15,6 +15,7 @@ def anisotropic_diffusion(
     call_from_clib(
         anisotropic_diffusion.__name__,
         ctypes.pointer(out),
+        image,
         ctypes.c_float(timestep),
         ctypes.c_float(conductance),
         ctypes.c_uint(iterations),
@@ -133,7 +134,7 @@ def medfilt1(image: AFArray, wind_length: int, edge_pad: Pad, /) -> AFArray:
     source: https://arrayfire.org/docs/group__image__func__medfilt.htm#gad108ea62cbbb5371bd14a17d06384359
     """
     out = AFArray.create_null_pointer()
-    call_from_clib(medfilt1.__name__, ctypes.pointer(out), CDimT(wind_length), edge_pad.value)
+    call_from_clib(medfilt1.__name__, ctypes.pointer(out), image, CDimT(wind_length), edge_pad.value)
     return out
 
 

arrayfire_wrapper/lib/signal_processing/fast_fourier_transforms.py

@@ -18,7 +18,7 @@ def fft_inplace(arr: AFArray, norm_factor: float, /) -> AFArray:
     source: https://arrayfire.org/docs/group__signal__func__fft.htm#gaa2f03c9ee1cb80dc184c0b0a13176da1
     """
     out = AFArray.create_null_pointer()
-    call_from_clib(fft_inplace.__name__, ctypes.pointer(out), arr, ctypes.c_double(norm_factor))
+    call_from_clib(fft_inplace.__name__, arr, ctypes.c_double(norm_factor))
     return out
 
 
@@ -44,7 +44,7 @@ def fft2_inplace(arr: AFArray, norm_factor: float, /) -> AFArray:
     source: https://arrayfire.org/docs/group__signal__func__fft2.htm#gacdeebb3f221ae698833dc4900a172b8c
     """
     out = AFArray.create_null_pointer()
-    call_from_clib(fft2_inplace.__name__, ctypes.pointer(out), arr, ctypes.c_double(norm_factor))
+    call_from_clib(fft2_inplace.__name__, arr, ctypes.c_double(norm_factor))
     return out
 
 

arrayfire_wrapper/version.py

@@ -1,7 +1,7 @@
 import os
 
 _MAJOR = "0"
-_MINOR = "3"
+_MINOR = "4"
 # On main and in a nightly release the patch should be one ahead of the last
 # released build.
 _PATCH = "0"