You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
1.3 KiB
30 lines
1.3 KiB
--- setup.py
|
|
+++ setup.py
|
|
@@ -2,7 +2,7 @@
|
|
|
|
from distutils.core import setup, Extension
|
|
from distutils.command.install_headers import install_headers
|
|
-import os, sys, platform
|
|
+import os, sys, platform, subprocess
|
|
from glob import glob
|
|
|
|
class Dummy:
|
|
@@ -103,13 +103,16 @@
|
|
netcdf_include = os.path.join(netcdf_prefix, 'include')
|
|
netcdf_h_file = os.path.join(netcdf_prefix, 'include', 'netcdf.h')
|
|
netcdf_lib = os.path.join(netcdf_prefix, 'lib')
|
|
+ mpi_cflags = subprocess.Popen(["mpicc", "-showme:compile"], stdout=subprocess.PIPE).communicate()[0].rstrip().split()
|
|
+ mpi_ldflags = subprocess.Popen(["mpicc", "-showme:link"], stdout=subprocess.PIPE).communicate()[0].rstrip().split()
|
|
ext_modules = [Extension('Scientific_netcdf',
|
|
['Src/Scientific_netcdf.c'],
|
|
include_dirs=['Include', netcdf_include]
|
|
+ arrayobject_h_include,
|
|
library_dirs=[netcdf_lib],
|
|
libraries = ['netcdf'],
|
|
- extra_compile_args=extra_compile_args)]
|
|
+ extra_compile_args=extra_compile_args + mpi_cflags,
|
|
+ extra_link_args=mpi_ldflags)]
|
|
|
|
try:
|
|
# Add code for including documentation in Mac packages
|