Homec4science
Diffusion Gear (master) Browse

Gear/PyChem/setup-r2.py
b87f34a0ce7emaster

Last Change
Raw File

setup-r2.py
View Options

#!/usr/bin/env python
import os,sys
from distutils.core import setup, Extension
from distutils.sysconfig import get_python_inc
from distutils.sysconfig import get_python_lib
import glob
import numpy
'''
In order to compile with mpi,
export CC=mpicc
python setup.py build
'''
gadget_files=glob.glob("src/*.c")
INCLUDES=["src/"]
INCLUDES.append(numpy.get_include())
INCLUDES.append(numpy.get_numarray_include())
INCLUDES.append(os.path.join(numpy.get_numarray_include(),'numpy'))
MPI_DIR = "/dios/shared/apps/openmpi/gcc/1.6/lib/"
MPI_LIB = "mpi"
INCLUDES.append("/dios/shared/apps/openmpi/gcc/1.6/include/")
setup(name='PyChem',
version='0.0',
description='Python Chemical evolution code base wrapped around GEAR',
author='Yves Revaz',
author_email='yves.revaz@epfl.ch',
url='http://obswww.unige.ch/~revaz/pNbody',
packages=['PyChem'],
# ext_modules=[
# Extension('PyGear.gadget', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC],define_macros=[('PY_INTERFACE', '1'),('UNEQUALSOFTENINGS', '1') ,('PERIODIC', '1'),('PEANOHILBERT', '1'),('NOGRAVITY', '1'),('LONG_X', '1'),('LONG_Y', '1'),('LONG_Z', '0.125')] , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB] )
# ]
# ext_modules=[
# Extension('PyGear.gadget', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC],define_macros=[('PY_INTERFACE', '1'),('UNEQUALSOFTENINGS', '1') ,('PERIODIC', '1'),('PEANOHILBERT', '1')] , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB] )
# ]
# This is used for turb
# ext_modules=[
# Extension('PyChem.gadget', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC],define_macros=[('PY_INTERFACE', '1'),('UNEQUALSOFTENINGS', '1') ,('PERIODIC', '1'),('PEANOHILBERT', '1'),('NOGRAVITY', '1'),('ISOTHERM_EQS', '1'),('ONLY_MASTER_READ_EWALD', '1')] , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB] )
# ]
# This is used for pychem
ext_modules=[
Extension('PyChem.chemistry', gadget_files,include_dirs=INCLUDES,define_macros=[('PY_INTERFACE', '1'),('UNEQUALSOFTENINGS', '1') ,('PERIODIC', '1'),('PEANOHILBERT', '1'),('CHIMIE', '1'),('STELLAR_PROP', '1'),('ENTROPYPRED', '1'), ('PYCHEM', '1'),('ONLY_MASTER_READ_EWALD', '1')] , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB] ),
]
)
c4science · Help