2011-09-10 11:13:51 +02:00
|
|
|
#!/usr/bin/env python
|
|
|
|
# encoding: utf-8
|
|
|
|
|
2014-01-27 16:20:01 +01:00
|
|
|
import os, sys, imp, types, re
|
2011-09-10 11:13:51 +02:00
|
|
|
from waflib import Utils, Configure, Options, Logs, Errors
|
|
|
|
from waflib.Tools import fc
|
|
|
|
|
|
|
|
fc_compiler = {
|
|
|
|
'win32' : ['gfortran','ifort'],
|
|
|
|
'darwin' : ['gfortran', 'g95', 'ifort'],
|
|
|
|
'linux' : ['gfortran', 'g95', 'ifort'],
|
|
|
|
'java' : ['gfortran', 'g95', 'ifort'],
|
|
|
|
'default': ['gfortran'],
|
|
|
|
'aix' : ['gfortran']
|
|
|
|
}
|
2014-01-25 18:47:15 +01:00
|
|
|
"""
|
|
|
|
Dict mapping the platform names to lists of names of Fortran compilers to try, in order of preference::
|
|
|
|
|
|
|
|
from waflib.Tools.compiler_c import c_compiler
|
|
|
|
c_compiler['linux'] = ['gfortran', 'g95', 'ifort']
|
|
|
|
"""
|
2011-09-10 11:13:51 +02:00
|
|
|
|
2014-02-10 09:27:11 +01:00
|
|
|
def default_compilers():
|
|
|
|
build_platform = Utils.unversioned_sys_platform()
|
|
|
|
possible_compiler_list = fc_compiler.get(build_platform, fc_compiler['default'])
|
|
|
|
return ' '.join(possible_compiler_list)
|
|
|
|
|
2011-09-10 11:13:51 +02:00
|
|
|
def configure(conf):
|
|
|
|
"""
|
|
|
|
Try to find a suitable Fortran compiler or raise a :py:class:`waflib.Errors.ConfigurationError`.
|
|
|
|
"""
|
2014-02-10 09:27:11 +01:00
|
|
|
try: test_for_compiler = conf.options.check_fortran_compiler or default_compilers()
|
2011-09-10 11:13:51 +02:00
|
|
|
except AttributeError: conf.fatal("Add options(opt): opt.load('compiler_fc')")
|
2014-01-25 18:47:15 +01:00
|
|
|
for compiler in re.split('[ ,]+', test_for_compiler):
|
2012-01-14 13:32:35 +01:00
|
|
|
conf.env.stash()
|
2014-01-25 18:47:15 +01:00
|
|
|
conf.start_msg('Checking for %r (Fortran compiler)' % compiler)
|
2011-09-10 11:13:51 +02:00
|
|
|
try:
|
|
|
|
conf.load(compiler)
|
|
|
|
except conf.errors.ConfigurationError as e:
|
2012-01-14 13:32:35 +01:00
|
|
|
conf.env.revert()
|
2011-09-10 11:13:51 +02:00
|
|
|
conf.end_msg(False)
|
|
|
|
Logs.debug('compiler_fortran: %r' % e)
|
|
|
|
else:
|
|
|
|
if conf.env['FC']:
|
|
|
|
conf.end_msg(conf.env.get_flat('FC'))
|
|
|
|
conf.env.COMPILER_FORTRAN = compiler
|
|
|
|
break
|
|
|
|
conf.end_msg(False)
|
|
|
|
else:
|
2014-01-25 19:28:55 +01:00
|
|
|
conf.fatal('could not configure a Fortran compiler!')
|
2011-09-10 11:13:51 +02:00
|
|
|
|
|
|
|
def options(opt):
|
|
|
|
"""
|
|
|
|
Restrict the compiler detection from the command-line::
|
|
|
|
|
|
|
|
$ waf configure --check-fortran-compiler=ifort
|
|
|
|
"""
|
2014-02-10 09:27:11 +01:00
|
|
|
test_for_compiler = default_compilers()
|
2011-09-10 11:13:51 +02:00
|
|
|
opt.load_special_tools('fc_*.py')
|
2014-01-25 18:47:15 +01:00
|
|
|
fortran_compiler_opts = opt.add_option_group('Configuration options')
|
2014-02-10 09:27:11 +01:00
|
|
|
fortran_compiler_opts.add_option('--check-fortran-compiler', default=None,
|
2014-01-25 18:47:15 +01:00
|
|
|
help='list of Fortran compiler to try [%s]' % test_for_compiler,
|
2014-02-10 09:27:11 +01:00
|
|
|
dest="check_fortran_compiler")
|
2011-09-10 11:13:51 +02:00
|
|
|
|
2014-02-10 09:27:11 +01:00
|
|
|
for x in test_for_compiler.split():
|
|
|
|
opt.load('%s' % x)
|
2011-09-10 11:13:51 +02:00
|
|
|
|