Detect and use PARI/GP installation prefix

Author: jdemeyer

autogen/__init__.py

@@ -5,7 +5,7 @@
 from os.path import join, getmtime, exists
 
 from .generator import PariFunctionGenerator
-from .parser import pari_share
+from .paths import pari_share
 
 
 def rebuild(force=False):

autogen/parser.py

@@ -23,33 +23,7 @@
 
 from .args import pari_arg_types
 from .ret import pari_ret_types
-
-
-def pari_share():
-    r"""
-    Return the directory where the PARI data files are stored.
-
-    EXAMPLES::
-
-        sage: from sage_setup.autogen.pari.parser import pari_share
-        sage: pari_share()
-        '.../share/pari'
-    """
-    from subprocess import Popen, PIPE
-    gp = Popen(["gp", "-f", "-q"], stdin=PIPE, stdout=PIPE)
-    out = gp.communicate(b"print(default(datadir))")[0]
-    datadir = out.strip()
-    if not os.path.isdir(datadir):
-        # As a fallback, try a path relative to the gp executable.
-        # This is useful for broken Conda versions, see
-        # https://github.com/conda-forge/pari-feedstock/issues/4
-        from distutils.spawn import find_executable
-        gppath = find_executable("gp")
-        if gppath is not None:
-            datadir = os.path.join(os.path.dirname(gppath), "..", "share", "pari")
-        if not os.path.isdir(datadir):
-            raise EnvironmentError("PARI data directory {!r} does not exist".format(datadir))
-    return datadir
+from .paths import pari_share
 
 
 paren_re = re.compile(r"[(](.*)[)]")

autogen/paths.py

@@ -0,0 +1,69 @@
+"""
+Find out installation paths of PARI/GP
+"""
+
+#*****************************************************************************
+#       Copyright (C) 2017 Jeroen Demeyer <[email protected]>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 2 of the License, or
+# (at your option) any later version.
+#                  http://www.gnu.org/licenses/
+#*****************************************************************************
+
+from __future__ import absolute_import
+
+import os
+from glob import glob
+from distutils.spawn import find_executable
+
+
+# find_executable() returns None if nothing was found
+gppath = find_executable("gp") or ""
+if gppath is None:
+    # This almost certainly won't work, but we need to put something here
+    prefix = "."
+else:
+    # Assume gppath is ${prefix}/bin/gp
+    prefix = os.path.dirname(os.path.dirname(gppath))
+
+
+def pari_share():
+    r"""
+    Return the directory where the PARI data files are stored.
+
+    EXAMPLES::
+
+        sage: from sage_setup.autogen.pari.parser import pari_share
+        sage: pari_share()
+        '.../share/pari'
+    """
+    from subprocess import Popen, PIPE
+    if not gppath:
+        raise EnvironmentError("cannot find an installation of PARI/GP: make sure that the 'gp' program is in your $PATH")
+    gp = Popen([gppath, "-f", "-q"], stdin=PIPE, stdout=PIPE)
+    out = gp.communicate(b"print(default(datadir))")[0]
+    datadir = out.strip()
+    if not os.path.isdir(datadir):
+        # As a fallback, try a path relative to the prefix
+        datadir = os.path.join(prefix, "share", "pari")
+        if not os.path.isdir(datadir):
+            raise EnvironmentError("PARI data directory {!r} does not exist".format(datadir))
+    return datadir
+
+
+def include_dirs():
+    """
+    Return the directory containing PARI include files.
+    """
+    dirs = [os.path.join(prefix, "include")]
+    return [d for d in dirs if os.path.isdir(os.path.join(d, "pari"))]
+
+
+def library_dirs():
+    """
+    Return the directory containing PARI library files.
+    """
+    dirs = [os.path.join(prefix, s) for s in ("lib", "lib32", "lib64")]
+    return [d for d in dirs if glob(os.path.join(d, "libpari*"))]

setup.py

@@ -23,6 +23,9 @@
 from setuptools.command.bdist_egg import bdist_egg as _bdist_egg
 from setuptools.extension import Extension
 
+from autogen import rebuild
+from autogen.paths import include_dirs, library_dirs
+
 
 # Adapted from Cython's new_build_ext
 class build_ext(_build_ext):
@@ -44,7 +47,6 @@ def finalize_options(self):
             sys.exit(1)
 
         # Generate auto-generated sources from pari.desc
-        from autogen import rebuild
         rebuild()
 
         self.directives = dict(binding=True)
@@ -70,7 +72,8 @@ def run(self):
     author='Many people',
     author_email="[email protected]",
     license='GNU General Public License, version 2 or later',
-    ext_modules=[Extension("*", ["cypari2/*.pyx"])],
+    ext_modules=[Extension("*", ["cypari2/*.pyx"],
+            include_dirs=include_dirs(), library_dirs=library_dirs())],
     keywords='PARI/GP number theory',
     packages=['cypari2'],
     package_dir={'cypari2': 'cypari2'},