Homec4science
Diffusion Gear (master) Browse

Gear/PyGear/setup.py
b87f34a0ce7emaster

Last Change
Raw File

setup.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
'''
In order to compile with mpi,
export CC=mpicc
python setup.py build
'''
incdir = os.path.join(get_python_inc(plat_specific=1), 'numpy')
libdir = os.path.join(get_python_lib(plat_specific=1), 'numpy')
gadget_files=glob.glob("src/*.c")
PYTHON_INC = "/usr/include/python2.6/"
MPI_DIR = "/usr/lib64/openmpi/lib/"
MPI_LIB = "mpi"
MPI_INC = "/usr/include/openmpi-x86_64/"
H5_LIB = "hdf5"
# defines flags
opts=[('UNEQUALSOFTENINGS', '1'),
('PEANOHILBERT', '1'),
('WALLCLOCK', '1'),
('SYNCHRONIZATION', '1'),
('NOSTOP_WHEN_BELOW_MINTIMESTEP', '1'),
#('SYNCHRONIZE_NGB_TIMESTEP', '1'),
#('TIMESTEP_UPDATE_FOR_FEEDBACK', '1'),
#('IMPROVED_TIMESTEP_CRITERION_FORGAS', '1'),
('ADVANCEDSTATISTICS', '1'),
('ADVANCEDCPUSTATISTICS', '1'),
('SYSTEMSTATISTICS', '1'),
('BLOCK_SKIPPING', '1'),
#('OUTPUTSTELLAR_PROP', '1'),
('COMPUTE_POTENTIAL_ENERGY', '1'),
('NO_NEGATIVE_PRESSURE', '1'),
('WRITE_ALL_MASSES', '1'),
('COUNT_ACTIVE_PARTICLES', '1'),
('RANDOMSEED_AS_PARAMETER', '1'),
('COSMICTIME', '1'),
('ONLY_MASTER_READ_EWALD', '1'),
#('COOLING', '1'),
('SFR', '1'),
('COMPUTE_SFR_ENERGY', '1'),
#('SFR_NEG_DIV_ONLY', '1'),
#('STELLAR_PROP', '1'),
#('CHIMIE', '1'),
#('CHIMIE_THERMAL_FEEDBACK', '1'),
#('CHIMIE_COMPUTE_THERMAL_FEEDBACK_ENERGY', '1'),
#('CHIMIE_EXTRAHEADER', '1'),
#('CHIMIE_INPUT_ALL', '1'),
#('CHECK_BLOCK_ORDER', '1'),
('CHECK_ENTROPY_SIGN', '1'),
('CHECK_TYPE_DURING_IO', '1'),
#('CHECK_ID_CORRESPONDENCE', '1'),
('ENTROPYPRED', '1'),
('DENSITY_INDEPENDENT_SPH', '1'),
('PY_INTERFACE', '1')
]
setup(name='PyGear',
version='0.0',
description='Python Gadget Wrapping',
author='Yves Revaz',
author_email='yves.revaz@epfl.ch',
url='http://obswww.unige.ch/~revaz/pNbody',
packages=['PyGear'],
# 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('PyGear.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 tessel
# ext_modules=[
## Extension('PyGear.gadget', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC],define_macros=[('PY_INTERFACE', '1'),('UNEQUALSOFTENINGS', '1') ,('PEANOHILBERT', '1'),('PERIODIC', '1'),('TESSEL', '1'),('NOGRAVITY', '1'),('LONG_X', '1'),('LONG_Y', '1'),('LONG_Z', '1'),('TWODIMS', '1')] , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB] )
# Extension('PyGear.gadget', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC],define_macros=[('PY_INTERFACE', '1'),('UNEQUALSOFTENINGS', '1') ,('PERIODIC', '1'),('TESSEL', '1'),('NOGRAVITY', '1'),('LONG_X', '1'),('LONG_Y', '1'),('LONG_Z', '1'),('TWODIMS', '1'),('CHECK_IN_TESSEL', '1')] , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB] )
# ]
# This is used for cosmic time experiments
# 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'),('COSMICTIME', '1')] , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB] ),
# Extension('PyGear.cosmic', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC],define_macros=[('PY_INTERFACE', '1'),('UNEQUALSOFTENINGS', '1') ,('PERIODIC', '1'),('PEANOHILBERT', '1'),('NOGRAVITY', '1'),('COSMICTIME', '1')] , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB] )
# ]
# This is used for star formation tests
ext_modules=[
Extension('PyGear.gadget', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC],define_macros=opts , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB,H5_LIB] ),
Extension('PyGear.cosmic', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC],define_macros=opts , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB,H5_LIB] ),
Extension('PyGear.sfr', gadget_files,include_dirs=[MPI_INC,"src/",PYTHON_INC] ,define_macros=opts , library_dirs=[MPI_DIR],libraries=['gsl','gslcblas','m',MPI_LIB,H5_LIB] )
]
)
c4science · Help