new examples`

zingale · zingale · commit 3e908d32ba9c · 2017-04-11T15:15:31.000-04:00
diff --git a/examples/extensions/ctypes/simple/numpy-example.py b/examples/extensions/ctypes/simple/numpy-example.py
@@ -0,0 +1,23 @@
+# see https://docs.scipy.org/doc/numpy/reference/routines.ctypeslib.html
+
+import numpy as np
+import numpy.ctypeslib as npc
+import ctypes as C
+
+# load our extension -- this returns a ctypes library object
+cfunc = npc.load_library('cfunc', '.')
+
+# set the return type
+cfunc.my_subroutine.restype = C.c_int
+
+# set the argument types
+cfunc.my_subroutine.argtypes = [npc.ndpointer(C.c_double, flags="C_CONTIGUOUS"),
+                                C.c_int]
+
+def my_subroutine(A):
+    return cfunc.my_subroutine(A, len(A))
+
+A = np.arange(10, dtype=np.float64)
+n = my_subroutine(A)
+
+
diff --git a/examples/extensions/ctypes/simple/pointer-example.py b/examples/extensions/ctypes/simple/pointer-example.py
@@ -0,0 +1,24 @@
+# this version of the driver uses the normal ctypes POINTER(), so you
+# need to have your numpy array make itself appear as a C pointer
+# explicitly.
+
+import numpy as np
+import ctypes as C
+
+# load our extension -- this returns a ctypes library object
+cfunc = np.ctypeslib.load_library('cfunc', '.')
+
+# set the return type
+cfunc.my_subroutine.restype = C.c_int
+
+# set the argument types
+cfunc.my_subroutine.argtypes = [C.POINTER(C.c_double),
+                                C.c_int]
+
+def my_subroutine(A):
+    return cfunc.my_subroutine(A.ctypes.data_as(C.POINTER(C.c_double)), len(A))
+
+A = np.arange(10, dtype=np.float64)
+n = my_subroutine(A)
+
+
diff --git a/examples/extensions/ctypes/simple/test.py b/examples/extensions/ctypes/simple/test.py
