MAINT: Improve Windows build support

Author: zoj613

.github/workflows/build-and-test.yml

@@ -15,7 +15,7 @@ jobs:
 
     strategy:
       matrix:
-        os: [ ubuntu-20.04, macOS-10.15, windows-2019 ]
+        os: [ ubuntu-20.04, macOS-10.15, windows-2016 ]
         python-version: [ '3.6', '3.7', '3.8', '3.9' ]
 
     name: Python version - ${{ matrix.python-version }} - ${{ matrix.os }}

build.py

@@ -1,4 +1,5 @@
 from distutils.core import Extension
+import platform
 import os
 from os.path import join
 
@@ -20,10 +21,12 @@
 if os.getenv("BUILD_WITH_COVERAGE", None):
     macros.append(('CYTHON_TRACE_NOGIL', 1))
 
-
-compile_args = ['-O2', '-std=c99', '-march=native']
-if os.getenv("BUILD_WHEEL", None):
-    compile_args.pop()
+if platform.system() == 'Windows':
+    compile_args = ['/O2']
+else:
+    compile_args = ['-O2', '-std=c99', '-march=native']
+    if os.getenv("BUILD_WHEEL", None):
+        compile_args.pop()
 
 # https://numpy.org/devdocs/reference/random/examples/cython/setup.py.html
 include_path = np.get_include()

include/pgm_random.h

@@ -3,6 +3,12 @@
 
 #include <numpy/random/bitgen.h>
 
+#if !defined(_MSC_VER)
+    #define PGM_RESTRICT restrict
+#else
+    #define PGM_RESTRICT
+#endif
+
 
 typedef enum {GAMMA, DEVROYE, ALTERNATE, SADDLE, HYBRID} sampler_t;
 
@@ -65,6 +71,6 @@ pgm_random_polyagamma_fill(bitgen_t* bitgen_state, double h, double z,
  */
 void
 pgm_random_polyagamma_fill2(bitgen_t* bitgen_state, const double* h, const double* z,
-                            sampler_t method, size_t n, double* restrict out);
+                            sampler_t method, size_t n, double* PGM_RESTRICT out);
 
 #endif

src/pgm_random.c

@@ -157,7 +157,7 @@ pgm_random_polyagamma_fill(bitgen_t* bitgen_state, double h, double z,
 
 void
 pgm_random_polyagamma_fill2(bitgen_t* bitgen_state, const double* h, const double* z,
-                            sampler_t method, size_t n, double* restrict out)
+                            sampler_t method, size_t n, double* PGM_RESTRICT out)
 {
     pgm_func_t f = sampling_method_table[method];