install: arm64 compilation on MacOS

Author: william-silversmith

build-macos-arm.sh

@@ -1,6 +1,13 @@
 #!/bin/zsh
 
+source /usr/local/bin/virtualenvwrapper.sh
+
+export ARCHFLAGS="-arch arm64"
+
 function build {
+	pip install numpy cython -U
+	rm src/DracoPy.cpp
+	cython -3 src/DracoPy.pyx
 	python setup-macos-arm.py develop
 	python -m pytest -v -x tests.py
 	python setup-macos-arm.py bdist_wheel	

setup-macos-arm.py

@@ -35,7 +35,7 @@ def read(fname):
                 np.get_include(),
             ],
             extra_compile_args=[
-              '-std=c++11','-O3', '-stdlib=libc++'
+              '-std=c++17','-O3', '-stdlib=libc++'
             ],
             extra_link_args=[ "-ldraco" ],
         )

src/DracoPy.pxd

@@ -3,9 +3,11 @@ from libcpp.vector cimport vector
 from libc.stdint cimport uint8_t, uint32_t
 from libcpp cimport bool
 
-cimport numpy
+cimport numpy as cnp
 import numpy as np
 
+cnp.import_array()
+
 cdef extern from "DracoPy.h" namespace "DracoFunctions":
 
     cdef enum decoding_status:

src/DracoPy.pyx

@@ -11,10 +11,10 @@ from libc.stdint cimport (
   uint8_t, uint16_t, uint32_t, uint64_t,
 )
 
-cimport numpy as np
-import numpy as np
-np.import_array()
+cimport numpy as cnp
+cnp.import_array()
 
+import numpy as np
 
 class DracoPointCloud:
     def __init__(self, data_struct):
@@ -170,7 +170,7 @@ def encode(
     elif np.issubdtype(points.dtype, np.unsignedinteger):
         integer_mark = 2
 
-    cdef np.ndarray[float, ndim=1] qorigin = np.zeros((3,), dtype=np.float32)
+    cdef cnp.ndarray[float, ndim=1] qorigin = np.zeros((3,), dtype=np.float32)
     cdef float[:] quant_origin = qorigin
 
     if quantization_origin is not None: