/
setup.py
119 lines (107 loc) · 4.43 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from numpy.distutils.core import Extension
import numpy.distutils.command.build_ext
import distutils
from distutils.command import clean
from distutils.core import Command
from distutils import log
from distutils.dir_util import mkpath, remove_tree
import os
import sys
# original Guptri website went offline
# UPSTREAM_TAR_URL = ("https://www8.cs.umu.se/research/nla/singular_pairs/guptri/archive/guptri.tar.gz",
# "https://www8.cs.umu.se/research/nla/singular_pairs/guptri/archive/fguptri.tar.gz")
UPSTREAM_TAR_URL = ("https://web.archive.org/web/20210310205556/https://www8.cs.umu.se/research/nla/singular_pairs/guptri/archive/guptri.tar.gz",
"https://web.archive.org/web/20210310204056/https://www8.cs.umu.se/research/nla/singular_pairs/guptri/archive/fguptri.tar.gz")
CHECKSUMS = ("42ba92fb3b58334d99b43e0e4338caeeba91d8d8", "3416bed23a0bf4175bec3e1579821c217c29a7bb")
UPSTREAM_TAR_NAMES = ("guptri.tar.gz", "fguptri.tar.gz")
cwd = os.path.abspath(os.getcwd())
UPSTREAM_DIR = os.path.join(cwd, "upstream")
SRC = "guptri_py"
VERSION = open("VERSION").read().strip()
TMPDIR = "local/var/tmp"
def sha1sum(filename):
import hashlib
BLOCKSIZE = 65536
hasher = hashlib.sha1()
with open(filename, 'rb') as f:
buf = f.read(BLOCKSIZE)
while len(buf) > 0:
hasher.update(buf)
buf = f.read(BLOCKSIZE)
return hasher.hexdigest()
class DownloadCommand(Command):
description = "download, extract and patch guptri files"
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
log.info("Checking if upstream tar files exist.")
if not os.path.exists(UPSTREAM_DIR):
mkpath(UPSTREAM_DIR)
for tarname, checksum, url in zip(UPSTREAM_TAR_NAMES, CHECKSUMS, UPSTREAM_TAR_URL):
t_file = os.path.join(UPSTREAM_DIR, tarname)
if not os.path.exists(t_file):
log.warn("File %s does not exist. Attempting to download from %s." % (t_file, url))
os.system("wget -O %s %s" % (t_file, url))
if not os.path.exists(t_file):
log.error("""Download failed. You may wish to download the file "%s" manually from "%s" and place it in the "upstream/" directory.""" %
(tarname, url))
sys.exit(1)
if sha1sum(t_file) != checksum:
log.error("Checksum for file %s is different." % t_file)
sys.exit(1)
log.info("Creating directories.")
if os.path.exists("local"):
remove_tree("local")
mkpath(TMPDIR)
log.info("Extracting tar files.")
for tarname in UPSTREAM_TAR_NAMES:
t_file = os.path.join(UPSTREAM_DIR, tarname)
import tarfile
tar = tarfile.open(t_file)
tar.extractall(TMPDIR)
tar.close()
log.info("Applying patches.")
if os.system("patch -d %s < patches/0001-fix-compile-errors-fguptri.patch" % TMPDIR):
log.error("Failed to apply patches.")
sys.exit(1)
class BuildExtCommand(numpy.distutils.command.build_ext.build_ext):
def run(self):
self.run_command('download')
super().run()
class CleanCommand(clean.clean):
def run(self):
if os.path.exists("local"):
remove_tree("local")
super().run()
class TestCommand(Command):
description = "run Sage tests (after installation)"
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
errno = os.system("sage -t --force-lib %s" % SRC)
if errno != 0:
sys.exit(1)
ext1 = Extension(name='guptri_py._fguptri_py',
# the following silences type mismatch errors in gfortran 10
# (-fallow-argument-mismatch is not recognized by older fortran versions)
extra_f77_compile_args=['-std=legacy', '-fPIC'],
sources=['guptri_py/_fguptri_py.pyf'] +
[os.path.join(TMPDIR, s) for s in ('fguptri.f', 'guptribase.f', 'zguptri.f')])
from numpy.distutils.core import setup
setup(
cmdclass={
'download': DownloadCommand,
'build_ext': BuildExtCommand,
'test': TestCommand,
'clean': CleanCommand,
},
name='guptri_py',
version=VERSION,
packages=["guptri_py"],
ext_modules=[ext1])