Skip to content

Commit 2efb205

Browse files
committed
Add fmmlib to setup.py
1 parent 48abe09 commit 2efb205

File tree

1 file changed

+12
-7
lines changed

1 file changed

+12
-7
lines changed

setup.py

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,12 @@
88
import re
99
import sys
1010

11-
#if sys.platform == 'darwin':
11+
# if sys.platform == 'darwin':
1212
# from distutils import sysconfig
1313
# vars = sysconfig.get_config_vars()
1414
# vars['LDSHARED'] = vars['LDSHARED'].replace('-bundle', '-dynamiclib')
1515

1616

17-
1817
if 'CC' in os.environ:
1918
print("Using CC={}".format(os.environ['CC']))
2019
else:
@@ -32,7 +31,7 @@
3231
MICRO_DIR = os.path.join(SRC_DIR, "micro", "lib")
3332
BARYAKHTAR_DIR = os.path.join(MICRO_DIR, "baryakhtar")
3433
DEMAG_DIR = os.path.join(SRC_DIR, "common", "dipolar")
35-
FMM_DIR = os.path.join(SRC_DIR, "atomistic", "lib", "demag_multipoles")
34+
FMM_DIR = os.path.join(SRC_DIR, "atomistic", "lib", "fmmlib")
3635
LOCAL_DIR = os.path.join(MODULE_DIR, "local")
3736
INCLUDE_DIR = os.path.join(LOCAL_DIR, "include")
3837
LIB_DIR = os.path.join(LOCAL_DIR, "lib")
@@ -51,6 +50,7 @@ def get_version():
5150
return m.group(2)
5251
raise Exception("Couldn't find __version__ in %s" % pkg_init_path)
5352

53+
5454
version = get_version()
5555

5656

@@ -62,6 +62,7 @@ def glob_cfiles(path, excludes, extension="*.c"):
6262
cfiles.append(cfile)
6363
return cfiles
6464

65+
6566
sources = []
6667
sources.append(os.path.join(ATOM_DIR, 'clib.pyx'))
6768
sources += glob_cfiles(ATOM_DIR, excludes=["clib.c"])
@@ -119,11 +120,12 @@ def glob_cfiles(path, excludes, extension="*.c"):
119120
fmm_sources.append(os.path.join(FMM_DIR, 'fmmlib.pyx'))
120121
fmm_sources += glob_cfiles(FMM_DIR, excludes=[], extension="*.cpp")
121122

123+
print(fmm_sources)
122124

123125
com_libs = ['m', 'fftw3_omp', 'fftw3', 'sundials_cvodes',
124126
'sundials_nvecserial', 'sundials_nvecopenmp', 'blas', 'lapack']
125127

126-
com_args = ['-std=c99', '-O3']
128+
com_args = ['-O3']
127129
# rpath is the path relative to the compiled shared object files (e.g. clib.so, etc)
128130
# which the dynamic linker looks for the linked libraries (e.g. libsundials_*.so) in.
129131
# We need to set it relatively in order for it to be preserved if the parent directory is moved
@@ -153,7 +155,9 @@ def glob_cfiles(path, excludes, extension="*.c"):
153155
com_inc.append(os.environ['FFTW_INC'])
154156

155157
com_args_cpp = com_args.copy()
156-
com_args_cpp.append('std=c++14')
158+
com_args_cpp.append('-std=c++14')
159+
160+
com_args.append('-std=c99')
157161
com_inc_cpp = com_inc.copy()
158162
com_link_cpp = com_link.copy()
159163
com_libs_cpp = com_libs.copy()
@@ -164,15 +168,15 @@ def glob_cfiles(path, excludes, extension="*.c"):
164168
sources=sources,
165169
include_dirs=com_inc,
166170
libraries=com_libs,
167-
library_dirs=lib_paths, runtime_library_dirs=lib_paths,
171+
library_dirs=lib_paths, runtime_library_dirs=lib_paths,
168172
extra_compile_args=com_args,
169173
extra_link_args=com_link,
170174
),
171175
Extension("fidimag.extensions.common_clib",
172176
sources=common_sources,
173177
include_dirs=com_inc,
174178
libraries=com_libs,
175-
library_dirs=lib_paths, runtime_library_dirs=lib_paths,
179+
library_dirs=lib_paths, runtime_library_dirs=lib_paths,
176180
extra_compile_args=com_args,
177181
extra_link_args=com_link,
178182
),
@@ -271,6 +275,7 @@ def glob_cfiles(path, excludes, extension="*.c"):
271275
library_dirs=lib_paths_cpp, runtime_library_dirs=lib_paths_cpp,
272276
extra_compile_args=com_args_cpp,
273277
extra_link_args=com_link_cpp,
278+
language="c++",
274279
)
275280
]
276281

0 commit comments

Comments
 (0)