"""This module implements additional tests ala autoconf which can be useful.

"""

from __future__ import division, absolute_import, print_function

# We put them here since they could be easily reused outside numpy.distutils

def check_inline(cmd):

    """Return the inline identifier (may be empty)."""

    cmd._check_compiler()

    body = """

#ifndef __cplusplus

static %(inline)s int static_func (void)

{

    return 0;

}

%(inline)s int nostatic_func (void)

{

    return 0;

}

#endif"""

    for kw in ['inline', '__inline__', '__inline']:

        st = cmd.try_compile(body % {'inline': kw}, None, None)

        if st:

            return kw

    return ''

def check_restrict(cmd):

    """Return the restrict identifier (may be empty)."""

    cmd._check_compiler()

    body = """

static int static_func (char * %(restrict)s a)

{

    return 0;

}

"""

    for kw in ['restrict', '__restrict__', '__restrict']:

        st = cmd.try_compile(body % {'restrict': kw}, None, None)

        if st:

            return kw

    return ''

def check_compiler_gcc4(cmd):

    """Return True if the C compiler is GCC 4.x."""

    cmd._check_compiler()

    body = """

int

main()

{

#if (! defined __GNUC__) || (__GNUC__ < 4)

#error gcc >= 4 required

#endif

    return 0;

}

"""

    return cmd.try_compile(body, None, None)

def check_gcc_function_attribute(cmd, attribute, name):

    """Return True if the given function attribute is supported."""

    cmd._check_compiler()

    body = """

#pragma GCC diagnostic error "-Wattributes"

#pragma clang diagnostic error "-Wattributes"

int %s %s(void*);

int

main()

{

    return 0;

}

""" % (attribute, name)

    return cmd.try_compile(body, None, None) != 0

def check_gcc_variable_attribute(cmd, attribute):

    """Return True if the given variable attribute is supported."""

    cmd._check_compiler()

    body = """

#pragma GCC diagnostic error "-Wattributes"

#pragma clang diagnostic error "-Wattributes"

int %s foo;

int

main()

{

    return 0;

}

""" % (attribute, )

    return cmd.try_compile(body, None, None) != 0