let setup script automatically download Eigen

Author: yixuan

.gitignore

@@ -1,8 +1,4 @@
-# R/RStudio config files
-.Rproj.user
-.Rhistory
-.RData
-.Ruserdata
+# Config files
 
 # Binary files
 *.o
@@ -12,12 +8,12 @@
 *.csv
 
 # Python cache and build files
-python/benchmark/benchmark_FairSVM/__cache__/
-python/benchmark/benchmark_SVM/__cache__/
-python/benchmark/benchmark_QR/__cache__/
-python/benchmark/benchmark_Huber/__cache__/
+**/__pycache__/
+build/
+rehline.egg-info/
+*.log
+
+# Files during package building
+eigen-3.4.0/
+eigen3.zip
 
-__pycache__/
-python/build/
-python/ReHLine.egg-info/
-*.log

setup.py

@@ -1,8 +1,7 @@
-
-import sys
-
-from pybind11 import get_cmake_dir
-# Available at setup time due to pyproject.toml
+import os
+from pathlib import Path
+import zipfile
+import requests
 from pybind11.setup_helpers import Pybind11Extension, build_ext
 from setuptools import setup
 
@@ -17,22 +16,57 @@
 #   Sort input source files if you glob sources to ensure bit-for-bit
 #   reproducible builds (https://github.com/pybind/python_example/pull/53)
 
+# The directory that contains setup.py
+SETUP_DIRECTORY = Path(__file__).resolve().parent
+
+# Download Eigen source files
+# Modified from https://github.com/tohtsky/irspack/blob/main/setup.py
+class get_eigen_include(object):
+    EIGEN3_URL = "https://gitlab.com/libeigen/eigen/-/archive/3.4.0/eigen-3.4.0.zip"
+    EIGEN3_DIRNAME = "eigen-3.4.0"
+
+    def __str__(self) -> str:
+        # Test whether the environment variable EIGEN3_INCLUDE_DIR is set
+        # If yes, directly return this directory
+        eigen_include_dir = os.environ.get("EIGEN3_INCLUDE_DIR", None)
+        if eigen_include_dir is not None:
+            return eigen_include_dir
+
+        # If the directory already exists (e.g. from previous setup),
+        # directly return it
+        target_dir = SETUP_DIRECTORY / self.EIGEN3_DIRNAME
+        if target_dir.exists():
+            return target_dir.name
+
+        # Filename for the downloaded Eigen source package
+        download_target_dir = SETUP_DIRECTORY / "eigen3.zip"
+        response = requests.get(self.EIGEN3_URL, stream=True)
+        with download_target_dir.open("wb") as ofs:
+            for chunk in response.iter_content(chunk_size=1024):
+                ofs.write(chunk)
+        # Unzip package
+        with zipfile.ZipFile(download_target_dir) as ifs:
+            ifs.extractall()
+
+        return target_dir.name
+
 ext_modules = [
-    Pybind11Extension("rehline",
+    Pybind11Extension("rehline._internal",
         ["src/rehline.cpp"],
+        include_dirs=[get_eigen_include()],
         # Example: passing in the version to the compiled code
-        define_macros = [('VERSION_INFO', __version__)],
+        define_macros=[('VERSION_INFO', __version__)],
         ),
 ]
 
 setup(
-    name="ReHLine",
+    name="rehline",
     version=__version__,
     author=["Ben Dai", "Yixuan Qiu"],
     author_email="[email protected]",
-    url="https://github.com/softmin/ReHLine",
-    description=" Minimizing Regularized Composite ReHU/ReLU Losses with Linear Computational Complexity ",
-    long_description="https://github.com/softmin/ReHLine",
+    url="https://github.com/softmin/ReHLine-python",
+    description="Regularized Composite ReLU-ReHU Loss Minimization with Linear Computation and Linear Convergence",
+    packages=["rehline"],
     ext_modules=ext_modules,
     # extras_require={"test": "pytest"},
     # Currently, build_ext only provides an optional "highest supported C++