setup.py

from distutils.core import setup
from distutils.extension import Extension
import os
#from Cython.Distutils import build_ext

#Check the dependencies
try:
  import scipy
except ImportError:
  print("scipy not found. Please install scipy: https://www.scipy.org/install.html")

try:
  import numpy
except ImportError:
  print("Numpy not found. Please install Numpy: https://docs.scipy.org/doc/numpy/user/install.html")

try:
  import mpi4py
except ImportError:
  print("mpi4py not found. Please install mpi4py: https://mpi4py.scipy.org/docs/usrman/install.html")

try:
  import matplotlib
except ImportError:
  print("matplotlib not found. Please install matplotlib: http://matplotlib.org/users/installing.html")


# If Cython is installed, cythonize the .pyx files. 
# Else, the .c files are used.
try:
    from Cython.Distutils import build_ext
except ImportError:
    use_cython = False
else:
    use_cython = True

use_cython = False

cmdclass = {}
if use_cython:
  ext_modules=[ Extension("matvec2D",
              ["PoissonSolver/MV_2D_cy/matvec2D.pyx"],
              extra_compile_args = ["-ffast-math"]),
              Extension("matvec1D",
              ["PoissonSolver/MV_1D_cy/matvec1D.pyx"],
              extra_compile_args = ["-ffast-math"]),
              Extension("ps3d",
              ["PoissonSolver/PS_3D_cy/ps3d.pyx"],
              extra_compile_args = ["-ffast-math"])
            ]
  cmdclass.update({ 'build_ext': build_ext })
else:
  ext_modules=[ Extension("matvec2D",
              ["PoissonSolver/MV_2D_cy/matvec2D.c"],
              extra_compile_args = ["-ffast-math"]),
              Extension("matvec1D",
              ["PoissonSolver/MV_1D_cy/matvec1D.c"],
              extra_compile_args = ["-ffast-math"]),
              Extension("ps3d",
              ["PoissonSolver/PS_3D_cy/ps3d.c"],
              extra_compile_args = ["-ffast-math"])
            ]

setup(
  name = "CoFFEE",
  author="Mit H Naik",
  author_email = "mitwise@gmail.com",
  cmdclass = cmdclass,
  ext_modules = ext_modules,
  include_dirs = [numpy.get_include()] 
)

os.system("mv PoissonSolver/matvec1D*.so PoissonSolver/matvec1D.so")
os.system("mv PoissonSolver/matvec2D*.so PoissonSolver/matvec2D.so")
os.system("mv PoissonSolver/ps3d*.so PoissonSolver/ps3d.so")