Chris@87: from __future__ import division, absolute_import, print_function
Chris@87: 
Chris@87: import os
Chris@87: import re
Chris@87: import sys
Chris@87: 
Chris@87: from numpy.distutils.fcompiler import FCompiler
Chris@87: from numpy.distutils.exec_command import exec_command, find_executable
Chris@87: from numpy.distutils.misc_util import make_temp_file
Chris@87: from distutils import log
Chris@87: 
Chris@87: compilers = ['IBMFCompiler']
Chris@87: 
Chris@87: class IBMFCompiler(FCompiler):
Chris@87:     compiler_type = 'ibm'
Chris@87:     description = 'IBM XL Fortran Compiler'
Chris@87:     version_pattern =  r'(xlf\(1\)\s*|)IBM XL Fortran ((Advanced Edition |)Version |Enterprise Edition V|for AIX, V)(?P<version>[^\s*]*)'
Chris@87:     #IBM XL Fortran Enterprise Edition V10.1 for AIX \nVersion: 10.01.0000.0004
Chris@87: 
Chris@87:     executables = {
Chris@87:         'version_cmd'  : ["<F77>", "-qversion"],
Chris@87:         'compiler_f77' : ["xlf"],
Chris@87:         'compiler_fix' : ["xlf90", "-qfixed"],
Chris@87:         'compiler_f90' : ["xlf90"],
Chris@87:         'linker_so'    : ["xlf95"],
Chris@87:         'archiver'     : ["ar", "-cr"],
Chris@87:         'ranlib'       : ["ranlib"]
Chris@87:         }
Chris@87: 
Chris@87:     def get_version(self,*args,**kwds):
Chris@87:         version = FCompiler.get_version(self,*args,**kwds)
Chris@87: 
Chris@87:         if version is None and sys.platform.startswith('aix'):
Chris@87:             # use lslpp to find out xlf version
Chris@87:             lslpp = find_executable('lslpp')
Chris@87:             xlf = find_executable('xlf')
Chris@87:             if os.path.exists(xlf) and os.path.exists(lslpp):
Chris@87:                 s, o = exec_command(lslpp + ' -Lc xlfcmp')
Chris@87:                 m = re.search('xlfcmp:(?P<version>\d+([.]\d+)+)', o)
Chris@87:                 if m: version = m.group('version')
Chris@87: 
Chris@87:         xlf_dir = '/etc/opt/ibmcmp/xlf'
Chris@87:         if version is None and os.path.isdir(xlf_dir):
Chris@87:             # linux:
Chris@87:             # If the output of xlf does not contain version info
Chris@87:             # (that's the case with xlf 8.1, for instance) then
Chris@87:             # let's try another method:
Chris@87:             l = sorted(os.listdir(xlf_dir))
Chris@87:             l.reverse()
Chris@87:             l = [d for d in l if os.path.isfile(os.path.join(xlf_dir, d, 'xlf.cfg'))]
Chris@87:             if l:
Chris@87:                 from distutils.version import LooseVersion
Chris@87:                 self.version = version = LooseVersion(l[0])
Chris@87:         return version
Chris@87: 
Chris@87:     def get_flags(self):
Chris@87:         return ['-qextname']
Chris@87: 
Chris@87:     def get_flags_debug(self):
Chris@87:         return ['-g']
Chris@87: 
Chris@87:     def get_flags_linker_so(self):
Chris@87:         opt = []
Chris@87:         if sys.platform=='darwin':
Chris@87:             opt.append('-Wl,-bundle,-flat_namespace,-undefined,suppress')
Chris@87:         else:
Chris@87:             opt.append('-bshared')
Chris@87:         version = self.get_version(ok_status=[0, 40])
Chris@87:         if version is not None:
Chris@87:             if sys.platform.startswith('aix'):
Chris@87:                 xlf_cfg = '/etc/xlf.cfg'
Chris@87:             else:
Chris@87:                 xlf_cfg = '/etc/opt/ibmcmp/xlf/%s/xlf.cfg' % version
Chris@87:             fo, new_cfg = make_temp_file(suffix='_xlf.cfg')
Chris@87:             log.info('Creating '+new_cfg)
Chris@87:             fi = open(xlf_cfg, 'r')
Chris@87:             crt1_match = re.compile(r'\s*crt\s*[=]\s*(?P<path>.*)/crt1.o').match
Chris@87:             for line in fi:
Chris@87:                 m = crt1_match(line)
Chris@87:                 if m:
Chris@87:                     fo.write('crt = %s/bundle1.o\n' % (m.group('path')))
Chris@87:                 else:
Chris@87:                     fo.write(line)
Chris@87:             fi.close()
Chris@87:             fo.close()
Chris@87:             opt.append('-F'+new_cfg)
Chris@87:         return opt
Chris@87: 
Chris@87:     def get_flags_opt(self):
Chris@87:         return ['-O3']
Chris@87: 
Chris@87: if __name__ == '__main__':
Chris@87:     log.set_verbosity(2)
Chris@87:     compiler = IBMFCompiler()
Chris@87:     compiler.customize()
Chris@87:     print(compiler.get_version())