2
0
mirror of https://gitlab.com/ita1024/waf.git synced 2024-11-16 23:19:50 +01:00
waf/waflib/Tools/compiler_fc.py
2011-09-10 11:13:51 +02:00

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)