Convert selection.c.src to selection.cpp

numpy · Jan 17, 2022 · 584ddc9 · 584ddc9
1 parent b83286e
commit 584ddc9
Show file tree

Hide file tree

Showing 4 changed files with 654 additions and 420 deletions.
diff --git a/numpy/core/setup.py b/numpy/core/setup.py
@@ -952,7 +952,7 @@ def gl_if_msvc(build_cmd):
             join('src', 'npysort', 'heapsort.cpp'),
             join('src', 'npysort', 'radixsort.cpp'),
             join('src', 'common', 'npy_partition.h.src'),
-            join('src', 'npysort', 'selection.c.src'),
+            join('src', 'npysort', 'selection.cpp'),
             join('src', 'common', 'npy_binsearch.h'),
             join('src', 'npysort', 'binsearch.cpp'),
             ]

diff --git a/numpy/core/src/common/npy_partition.h.src b/numpy/core/src/common/npy_partition.h.src
@@ -28,6 +28,10 @@
 
 #define NPY_MAX_PIVOT_STACK 50
 
+#ifdef __cplusplus
+extern "C" {
+#endif
+
 /**begin repeat
  *
  * #TYPE = BOOL, BYTE, UBYTE, SHORT, USHORT, INT, UINT, LONG, ULONG,
@@ -56,6 +60,10 @@ NPY_NO_EXPORT int aintroselect_@suff@(@type@ *v, npy_intp* tosort, npy_intp num,
 
 /**end repeat**/
 
+#ifdef __cplusplus
+}
+#endif
+
 typedef struct {
     int typenum;
     PyArray_PartitionFunc * part[NPY_NSELECTS];