Merge pull request #4 from GraphBLAS/michelp/build-script

Michelp/build script

Author: jim22k

.gitignore

@@ -127,3 +127,9 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+# emacs
+*~
+
+# Vim
+*.sw?

README.md

@@ -1,2 +1,8 @@
 # python-suitesparse-graphblas
-Python CFFI Binding around SuiteSparse:GraphBLAS
+Python CFFI Binding around
+[SuiteSparse:GraphBLAS](https://github.com/DrTimothyAldenDavis/GraphBLAS)
+
+This is a base package that exposes only the low level CFFI API
+bindings and symbols.  This package is shared by the syntax bindings
+[pygraphblas](https://github.com/Graphegon/pygraphblas) and
+[grblas](https://github.com/metagraph-dev/grblas).

setup.py

@@ -0,0 +1,14 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='suitesparse-graphblas',
+    version='4.0.3',
+    description='SuiteSparse:GraphBLAS Python bindings.',
+    packages=find_packages(),
+    author='Michel Pelletier, James Kitchen, Erik Welch',
+    cffi_modules=["suitesparse/graphblas/build.py:ffibuilder"],
+    install_requires=["cffi>=1.0.0"],
+    setup_requires=["cffi>=1.0.0", "pytest-runner"],
+    tests_require=["pytest"],
+)
+

suitesparse/__init__.py

@@ -0,0 +1 @@
+

suitesparse/graphblas/__init__.py

@@ -0,0 +1 @@
+from ._graphblas import ffi, lib

suitesparse/graphblas/build.py

@@ -0,0 +1,28 @@
+import os
+import sys
+from cffi import FFI
+
+is_win = sys.platform.startswith("win")
+thisdir = os.path.dirname(__file__)
+
+ffibuilder = FFI()
+
+with open(os.path.join(thisdir, "source.c")) as f:
+    source = f.read()
+
+ffibuilder.set_source(
+    "suitesparse.graphblas._graphblas",
+    source,
+    libraries=["graphblas"],
+    include_dirs=[os.path.join(sys.prefix, "include")],
+)
+
+header = "suitesparse_graphblas.h"
+if is_win:
+    header = "suitesparse_graphblas_no_complex.h"
+gb_cdef = open(os.path.join(thisdir, header))
+
+ffibuilder.cdef(gb_cdef.read())
+
+if __name__ == "__main__":
+    ffibuilder.compile(verbose=True)

suitesparse/graphblas/clean_header.py

@@ -0,0 +1,97 @@
+import sys
+import os
+import re
+
+
+def rename_defined_constants(text):
+    # #define GB_PUBLIC extern
+    text = text.replace("GB_PUBLIC", "extern")
+
+    return text
+
+
+def remove_directives(text):
+    # There are a few cases of safe `#define` directives that we need to keep
+    #  - #define FOO 12
+    #  - #define BAR ...
+    safe_define = re.compile(r"^#define\s+\w+\s+(\d+|\.{3})(\s|$)")
+
+    out = []
+    multiline = False
+    for line in text.splitlines():
+        if not line:
+            out.append(line)
+        elif multiline:
+            if line[-1] != "\\":
+                multiline = False
+            out.append(f"/* {line} */")
+        elif line.lstrip()[0] == "#":
+            if line[-1] == "\\":
+                multiline = True
+            if not multiline and safe_define.match(line):
+                out.append(line)
+            else:
+                out.append(f"/* {line} */")
+        else:
+            out.append(line)
+    return "\n".join(out)
+
+
+def remove_complex(text):
+    out = []
+    extern_block = None
+    complex_extern = False
+    for line in text.splitlines():
+        if extern_block is not None:
+            if "FC32" in line or "FC64" in line:
+                complex_extern = True
+            if line.replace(" ", "") == ");":  # End of extern block
+                extern_block.append(line)
+                if complex_extern:
+                    for i in range(len(extern_block)):
+                        extern_block[i] = f"// {extern_block[i]}"
+                out.extend(extern_block)
+                extern_block = None
+                complex_extern = False
+            else:
+                extern_block.append(line)
+        elif not line:
+            out.append(line)
+        elif line.strip() == "extern":
+            extern_block = [line]
+        elif "FC32" in line or "FC64" in line:
+            # Check if the line is a terminating line
+            if re.search(r"FC(32|64)\s*;", line):
+                # By commenting the terminating line out, we lose the closing semicolon
+                # Walk up the lines, looking for the last non-commented out line
+                # then replace the trailing comma with a semicolon
+                for i in range(5):  # it's never more than 5 away
+                    if out[-i].startswith("//"):
+                        continue
+                    last_comma_pos = out[-i].rfind(",")
+                    if last_comma_pos < 0:
+                        continue
+                    out[-i] = f"{out[-i][:last_comma_pos]};{out[-i][last_comma_pos+1:]}"
+                    break
+            out.append(f"// {line}")
+        else:
+            out.append(line)
+    return "\n".join(out)
+
+
+def main(filename):
+    with open(filename, "r") as f:
+        text = f.read()
+    text = rename_defined_constants(text)
+    text = remove_directives(text)
+    if "_no_complex_" in filename:
+        text = remove_complex(text)
+    with open(filename, "w") as f:
+        f.write(text)
+
+
+if __name__ == "__main__":
+    filename = sys.argv[1]
+    if not os.path.exists(filename):
+        raise Exception(f'"{filename}" does not exist')
+    main(filename)