Add conditional standard flags

setup.py

@@ -75,6 +75,7 @@
     metadata["install_requires"] = install_requires
 else:
     from setuptools.extension import Extension
+    from setuptools.command.build_ext import build_ext
     from Cython.Build import cythonize
     import numpy as np
 
@@ -93,7 +94,7 @@
             "discretize._extensions.tree_ext",
             ["discretize/_extensions/tree_ext.pyx", "discretize/_extensions/tree.cpp"],
             include_dirs=[np.get_include()],
-            extra_compile_args=["-std:c++17", "-std=c++17"],
+            #extra_compile_args=["-std:c++17", "-std=c++17"],
             **ext_kwargs
         ),
         Extension(
@@ -104,6 +105,23 @@
         ),
     ]
 
+    class build_ext_cpp_standard(build_ext):
+        # add compiler specific standard argument specifier
+        def build_extension(self, ext):
+            # This module requires c++17 standard
+            if ext.name == "discretize._extensions.tree_ext":
+                comp_type = self.compiler.compiler_type
+                if comp_type == 'msvc':
+                    std_arg = "\std:c++17"
+                elif comp_type == 'bcpp':
+                    raise Exception('Must use cpp compiler that support C++17 standard.')
+                else:
+                    std_arg = "-std=c++17"
+                ext.extra_compile_args = [std_arg,]
+            super().build_extension(ext)
+
+
     metadata["ext_modules"] = cythonize(extensions)
+    metadata["cmdclass"] = {"build_ext":build_ext_cpp_standard}
 
 setup(**metadata)