mirror of https://gitlab.com/ita1024/waf.git
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
|
|
import os, sys, imp, types
|
|
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']
|
|
}
|
|
|
|
def __list_possible_compiler(platform):
|
|
try:
|
|
return fc_compiler[platform]
|
|
except KeyError:
|
|
return fc_compiler["default"]
|
|
|
|
def configure(conf):
|
|
"""
|
|
Try to find a suitable Fortran compiler or raise a :py:class:`waflib.Errors.ConfigurationError`.
|
|
"""
|
|
try: test_for_compiler = conf.options.check_fc
|
|
except AttributeError: conf.fatal("Add options(opt): opt.load('compiler_fc')")
|
|
orig = conf.env
|
|
for compiler in test_for_compiler.split():
|
|
try:
|
|
conf.start_msg('Checking for %r (fortran compiler)' % compiler)
|
|
conf.env = orig.derive()
|
|
conf.load(compiler)
|
|
except conf.errors.ConfigurationError as e:
|
|
conf.end_msg(False)
|
|
Logs.debug('compiler_fortran: %r' % e)
|
|
else:
|
|
if conf.env['FC']:
|
|
orig.table = conf.env.get_merged_dict()
|
|
conf.env = orig
|
|
conf.end_msg(conf.env.get_flat('FC'))
|
|
conf.env.COMPILER_FORTRAN = compiler
|
|
break
|
|
conf.end_msg(False)
|
|
else:
|
|
conf.fatal('could not configure a fortran compiler!')
|
|
|
|
def options(opt):
|
|
"""
|
|
Restrict the compiler detection from the command-line::
|
|
|
|
$ waf configure --check-fortran-compiler=ifort
|
|
"""
|
|
opt.load_special_tools('fc_*.py')
|
|
build_platform = Utils.unversioned_sys_platform()
|
|
detected_platform = Options.platform
|
|
possible_compiler_list = __list_possible_compiler(detected_platform)
|
|
test_for_compiler = ' '.join(possible_compiler_list)
|
|
fortran_compiler_opts = opt.add_option_group("Fortran Compiler Options")
|
|
fortran_compiler_opts.add_option('--check-fortran-compiler',
|
|
default="%s" % test_for_compiler,
|
|
help='On this platform (%s) the following Fortran Compiler will be checked by default: "%s"' % (detected_platform, test_for_compiler),
|
|
dest="check_fc")
|
|
|
|
for compiler in test_for_compiler.split():
|
|
opt.load('%s' % compiler)
|
|
|