From 036ddef0df37cd2a95b38680366dd10fd2cf9e6f Mon Sep 17 00:00:00 2001
From: Cristian C Lalescu <Cristian.Lalescu@ds.mpg.de>
Date: Fri, 31 Jul 2015 15:03:19 +0200
Subject: [PATCH] move local settings to outside file

---
 machine_settings.py | 33 +++++++++++++++++++++++++++++++++
 setup.py            | 16 +++-------------
 2 files changed, 36 insertions(+), 13 deletions(-)
 create mode 100644 machine_settings.py

diff --git a/machine_settings.py b/machine_settings.py
new file mode 100644
index 00000000..a4f7d29b
--- /dev/null
+++ b/machine_settings.py
@@ -0,0 +1,33 @@
+import os
+
+########################################################################
+#### these you're supposed to adapt to your environment
+########################################################################
+
+hostname = os.getenv('HOSTNAME')
+
+extra_compile_args = ['-mtune=native', '-ffast-math', '-std=c++11']
+
+include_dirs = []
+library_dirs = []
+
+if hostname in ['frontend01', 'frontend02']:
+    include_dirs = ['/usr/lib64/mpi/gcc/openmpi/include',
+                    '/usr/include/mpich',
+                    '/usr/nld/mvapich2-1.9a2-gcc/include',
+                    '/usr/nld/fftw-3.3.3-mvapich2-1.9a2-gcc/include',
+                    '/usr/nld/fftw-3.3.3-float-mvapich2-1.9a2-gcc/include']
+
+    library_dirs = ['/usr/nld/fftw-3.3.3-mvapich2-1.9a2-gcc/lib',
+                    '/usr/nld/fftw-3.3.3-float-mvapich2-1.9a2-gcc/lib']
+
+if hostname == 'tolima':
+    local_install_dir = '/scratch.local/chichi/installs'
+
+    include_dirs = ['/usr/lib64/mpi/gcc/openmpi/include',
+                    '/usr/include/mpich',
+                    os.path.join(local_install_dir, 'include')]
+
+    library_dirs = [os.path.join(local_install_dir, 'lib'),
+                    os.path.join(local_install_dir, 'lib64')]
+
diff --git a/setup.py b/setup.py
index fbffcdd4..e5aa603f 100644
--- a/setup.py
+++ b/setup.py
@@ -22,16 +22,7 @@
 
 import os
 
-########################################################################
-#### these you're supposed to adapt to your environment
-
-# first off, this is the install folder for fftw and/or openmpi etc
-local_install_dir = '/scratch.local/chichi/installs'
-
-include_dirs = ['/usr/lib64/mpi/gcc/openmpi/include',
-                '/usr/include/mpich',
-                os.path.join(local_install_dir, 'include')]
-########################################################################
+from machine_settings import include_dirs, library_dirs, extra_compile_args
 
 
 AUTHOR = 'Cristian C Lalescu'
@@ -72,9 +63,8 @@ libbfps = Extension(
         sources = ['bfps/cpp/' + fname + '.cpp' for fname in src_file_list],
         include_dirs = include_dirs,
         libraries = libraries,
-        extra_compile_args = ['-mtune=native', '-ffast-math', '-std=c++11'],
-        library_dirs = [os.path.join(local_install_dir, 'lib'),
-                        os.path.join(local_install_dir, 'lib64')])
+        extra_compile_args = extra_compile_args,
+        library_dirs = library_dirs)
 
 setup(
         name = 'bfps',
-- 
GitLab