"""distutils.extension

Provides the Extension class, used to describe C/C++ extension

modules in setup scripts.

Overridden to support f2py.

"""

from __future__ import division, absolute_import, print_function

import sys

import re

from distutils.extension import Extension as old_Extension

if sys.version_info[0] >= 3:

    basestring = str

cxx_ext_re = re.compile(r'.*[.](cpp|cxx|cc)\Z', re.I).match

fortran_pyf_ext_re = re.compile(r'.*[.](f90|f95|f77|for|ftn|f|pyf)\Z', re.I).match

class Extension(old_Extension):

    def __init__ (

            self, name, sources,

            include_dirs=None,

            define_macros=None,

            undef_macros=None,

            library_dirs=None,

            libraries=None,

            runtime_library_dirs=None,

            extra_objects=None,

            extra_compile_args=None,

            extra_link_args=None,

            export_symbols=None,

            swig_opts=None,

            depends=None,

            language=None,

            f2py_options=None,

            module_dirs=None,

            extra_f77_compile_args=None,

            extra_f90_compile_args=None,):

        old_Extension.__init__(

                self, name, [],

                include_dirs=include_dirs,

                define_macros=define_macros,

                undef_macros=undef_macros,

                library_dirs=library_dirs,

                libraries=libraries,

                runtime_library_dirs=runtime_library_dirs,

                extra_objects=extra_objects,

                extra_compile_args=extra_compile_args,

                extra_link_args=extra_link_args,

                export_symbols=export_symbols)

        # Avoid assert statements checking that sources contains strings:

        self.sources = sources

        # Python 2.4 distutils new features

        self.swig_opts = swig_opts or []

        # swig_opts is assumed to be a list. Here we handle the case where it

        # is specified as a string instead.

        if isinstance(self.swig_opts, basestring):

            import warnings

            msg = "swig_opts is specified as a string instead of a list"

            warnings.warn(msg, SyntaxWarning, stacklevel=2)

            self.swig_opts = self.swig_opts.split()

        # Python 2.3 distutils new features

        self.depends = depends or []

        self.language = language

        # numpy_distutils features

        self.f2py_options = f2py_options or []

        self.module_dirs = module_dirs or []

        self.extra_f77_compile_args = extra_f77_compile_args or []

        self.extra_f90_compile_args = extra_f90_compile_args or []

        return

    def has_cxx_sources(self):

        for source in self.sources:

            if cxx_ext_re(str(source)):

                return True

        return False

    def has_f2py_sources(self):

        for source in self.sources:

            if fortran_pyf_ext_re(source):

                return True

        return False

# class Extension