| import sys |
| import re |
| from numpy.distutils.fcompiler import FCompiler |
| |
| compilers = ['NAGFCompiler', 'NAGFORCompiler'] |
| |
| class BaseNAGFCompiler(FCompiler): |
| version_pattern = r'NAG.* Release (?P<version>[^(\s]*)' |
| |
| def version_match(self, version_string): |
| m = re.search(self.version_pattern, version_string) |
| if m: |
| return m.group('version') |
| else: |
| return None |
| |
| def get_flags_linker_so(self): |
| return ["-Wl,-shared"] |
| def get_flags_opt(self): |
| return ['-O4'] |
| def get_flags_arch(self): |
| return [] |
| |
| class NAGFCompiler(BaseNAGFCompiler): |
| |
| compiler_type = 'nag' |
| description = 'NAGWare Fortran 95 Compiler' |
| |
| executables = { |
| 'version_cmd' : ["<F90>", "-V"], |
| 'compiler_f77' : ["f95", "-fixed"], |
| 'compiler_fix' : ["f95", "-fixed"], |
| 'compiler_f90' : ["f95"], |
| 'linker_so' : ["<F90>"], |
| 'archiver' : ["ar", "-cr"], |
| 'ranlib' : ["ranlib"] |
| } |
| |
| def get_flags_linker_so(self): |
| if sys.platform == 'darwin': |
| return ['-unsharedf95', '-Wl,-bundle,-flat_namespace,-undefined,suppress'] |
| return BaseNAGFCompiler.get_flags_linker_so(self) |
| def get_flags_arch(self): |
| version = self.get_version() |
| if version and version < '5.1': |
| return ['-target=native'] |
| else: |
| return BaseNAGFCompiler.get_flags_arch(self) |
| def get_flags_debug(self): |
| return ['-g', '-gline', '-g90', '-nan', '-C'] |
| |
| class NAGFORCompiler(BaseNAGFCompiler): |
| |
| compiler_type = 'nagfor' |
| description = 'NAG Fortran Compiler' |
| |
| executables = { |
| 'version_cmd' : ["nagfor", "-V"], |
| 'compiler_f77' : ["nagfor", "-fixed"], |
| 'compiler_fix' : ["nagfor", "-fixed"], |
| 'compiler_f90' : ["nagfor"], |
| 'linker_so' : ["nagfor"], |
| 'archiver' : ["ar", "-cr"], |
| 'ranlib' : ["ranlib"] |
| } |
| |
| def get_flags_linker_so(self): |
| if sys.platform == 'darwin': |
| return ['-unsharedrts', |
| '-Wl,-bundle,-flat_namespace,-undefined,suppress'] |
| return BaseNAGFCompiler.get_flags_linker_so(self) |
| def get_flags_debug(self): |
| version = self.get_version() |
| if version and version > '6.1': |
| return ['-g', '-u', '-nan', '-C=all', '-thread_safe', |
| '-kind=unique', '-Warn=allocation', '-Warn=subnormal'] |
| else: |
| return ['-g', '-nan', '-C=all', '-u', '-thread_safe'] |
| |
| |
| if __name__ == '__main__': |
| from distutils import log |
| log.set_verbosity(2) |
| from numpy.distutils import customized_fcompiler |
| compiler = customized_fcompiler(compiler='nagfor') |
| print(compiler.get_version()) |
| print(compiler.get_flags_debug()) |
