numpy distutils - 如果你失败了,尝试编译一些东西并设置标志

时间:2013-04-26 14:41:34

标签: python numpy distutils

在* NIX平台上构建源代码的一种常用方法是使用configure脚本。在引擎盖下,configure尝试构建一堆测试程序,以确定您可以访问哪些库。然后它生成一个头文件,它包含在项目中,有条件地定义了一堆宏,以便程序员可以提供替代或构建库/程序的精简版本,如果特定的"依赖"不见了。使用numpy.distutils是否有任何功能相同的东西?

例如,这是我的setup.py

from numpy.distutils.misc_util import Configuration

def configuration(parent_package='',top_path=None):
    config = Configuration('pyggcm',parent_package,top_path)

    #TODO: Currently, I have some macros to conditionally build the seek-code
    #Unfortunately, that's not the best solution (by far).  Perhaps if we
    #changed to using stream access it would work better, without the need
    #for these silly macros.
    config.add_extension('_fortfile',sources=['_fortfile/_fortfile.F90'],
                         define_macros=[
                             ('FSEEKABLE',1),  #compiler provides fseek and ftell
                             ('HAVE_STREAM',1) #compiler provides access='stream' for opening files. (f2003 standard)
                             ])  

    config.add_extension('jrrle',sources=['jrrle/jrrle.f90'])
    config.add_scripts(['scripts/ggcm_timehist',
                        'scripts/ggcm_plasmasheet',
                        'scripts/ggcm_plot'])
    return config


from numpy.distutils.core import setup    
setup(configuration=configuration)

这是无条件构建FSEEKABLE代码,如果用户Fortran编译器不支持,则需要手动编辑(宏包裹fseekftell GNU内在功能)。有没有办法确定Fortran编译器是否提供这些内部函数?

2 个答案:

答案 0 :(得分:1)

试试这个:

import os
import shutil
import tempfile
from distutils.ccompiler import new_compiler

def hasfunction(cc, funcname, include=None, extra_postargs=None):
    tmpdir = tempfile.mkdtemp(prefix='hasfunction-')
    devnull = oldstderr = None
    try:
        try:
            fname = os.path.join(tmpdir, 'funcname.c')
            f = open(fname, 'w')
            if include is not None:
                f.write('#include %s\n' % include)
            f.write('int main(void) {\n')
            f.write('    %s;\n' % funcname)
            f.write('}\n')
            f.close()
            devnull = open(os.devnull, 'w')
            oldstderr = os.dup(sys.stderr.fileno())
            os.dup2(devnull.fileno(), sys.stderr.fileno())
            objects = cc.compile([fname], output_dir=tmpdir,
                                 extra_postargs=extra_postargs)
            cc.link_executable(objects, os.path.join(tmpdir, 'a.out'))
        except Exception as e:
            return False
        return True
    finally:
        if oldstderr is not None:
            os.dup2(oldstderr, sys.stderr.fileno())
        if devnull is not None:
            devnull.close()
        shutil.rmtree(tmpdir)

示例:

def detect_sse3():
    "Does this compiler support SSE3 intrinsics?"
    compiler = new_compiler()
    return hasfunction(compiler, '__m128 v; _mm_hadd_ps(v,v)',
                       include='<pmmintrin.h>',
                       extra_postargs=['-msse3'])

答案 1 :(得分:0)

将代码片段生成为文件&amp;尝试编译它检查错误代码是正常的方法,应该很好地工作 - 但是AFAIK没有人已经为你做了这个作为distutils的一部分。我怀疑其中一个,可能是基于python的构建工具,可能会支持这个。我认为你最好的选择是GNU autoconfigure。