#! /usr/bin/env python
# encoding: utf-8
# harald at klimachs.de

import re
from waflib import Utils,Errors
from waflib.Tools import fc,fc_config,fc_scan
from waflib.Configure import conf

from waflib.Tools.compiler_fc import fc_compiler
fc_compiler['aix'].insert(0, 'fc_xlf')

@conf
def find_xlf(conf):
	"""Find the xlf program (will look in the environment variable 'FC')"""

	fc = conf.find_program(['xlf2003_r', 'xlf2003', 'xlf95_r', 'xlf95', 'xlf90_r', 'xlf90', 'xlf_r', 'xlf'], var='FC')
	fc = conf.cmd_to_list(fc)
	conf.get_xlf_version(fc)
	conf.env.FC_NAME='XLF'

@conf
def xlf_flags(conf):
	v = conf.env
	v['FCDEFINES_ST'] = '-WF,-D%s'
	v['FCFLAGS_fcshlib'] = ['-qpic=small']
	v['FCFLAGS_DEBUG'] = ['-qhalt=w']
	v['LINKFLAGS_fcshlib'] = ['-Wl,-shared']

@conf
def xlf_modifier_platform(conf):
	dest_os = conf.env['DEST_OS'] or Utils.unversioned_sys_platform()
	xlf_modifier_func = getattr(conf, 'xlf_modifier_' + dest_os, None)
	if xlf_modifier_func:
		xlf_modifier_func()

@conf
def get_xlf_version(conf, fc):
	"""Get the compiler version"""

	cmd = fc + ['-qversion']
	try:
		out, err = conf.cmd_and_log(cmd, output=0)
	except Errors.WafError:
		conf.fatal('Could not find xlf %r' % cmd)

	for v in (r"IBM XL Fortran.* V(?P<major>\d*)\.(?P<minor>\d*)",):
		version_re = re.compile(v, re.I).search
		match = version_re(out or err)
		if match:
			k = match.groupdict()
			conf.env['FC_VERSION'] = (k['major'], k['minor'])
			break
	else:
		conf.fatal('Could not determine the XLF version.')

def configure(conf):
	conf.find_xlf()
	conf.find_ar()
	conf.fc_flags()
	conf.fc_add_flags()
	conf.xlf_flags()
	conf.xlf_modifier_platform()


[((17, 6, 17, 115), 'waflib.Configure.conf.find_program', 'conf.find_program', (), '', False, 'from waflib.Configure import conf\n'), ((18, 6, 18, 26), 'waflib.Configure.conf.cmd_to_list', 'conf.cmd_to_list', ({(18, 23, 18, 25): 'fc'}, {}), '(fc)', False, 'from waflib.Configure import conf\n'), ((19, 1, 19, 25), 'waflib.Configure.conf.get_xlf_version', 'conf.get_xlf_version', ({(19, 22, 19, 24): 'fc'}, {}), '(fc)', False, 'from waflib.Configure import conf\n'), ((58, 1, 58, 16), 'waflib.Configure.conf.find_xlf', 'conf.find_xlf', ({}, {}), '()', False, 'from waflib.Configure import conf\n'), ((59, 1, 59, 15), 'waflib.Configure.conf.find_ar', 'conf.find_ar', ({}, {}), '()', False, 'from waflib.Configure import conf\n'), ((60, 1, 60, 16), 'waflib.Configure.conf.fc_flags', 'conf.fc_flags', ({}, {}), '()', False, 'from waflib.Configure import conf\n'), ((61, 1, 61, 20), 'waflib.Configure.conf.fc_add_flags', 'conf.fc_add_flags', ({}, {}), '()', False, 'from waflib.Configure import conf\n'), ((62, 1, 62, 17), 'waflib.Configure.conf.xlf_flags', 'conf.xlf_flags', ({}, {}), '()', False, 'from waflib.Configure import conf\n'), ((63, 1, 63, 29), 'waflib.Configure.conf.xlf_modifier_platform', 'conf.xlf_modifier_platform', ({}, {}), '()', False, 'from waflib.Configure import conf\n'), ((32, 34, 32, 66), 'waflib.Utils.unversioned_sys_platform', 'Utils.unversioned_sys_platform', ({}, {}), '()', False, 'from waflib import Utils, Errors\n'), ((43, 13, 43, 44), 'waflib.Configure.conf.cmd_and_log', 'conf.cmd_and_log', (), '', False, 'from waflib.Configure import conf\n'), ((55, 2, 55, 52), 'waflib.Configure.conf.fatal', 'conf.fatal', ({(55, 13, 55, 51): '"""Could not determine the XLF version."""'}, {}), "('Could not determine the XLF version.')", False, 'from waflib.Configure import conf\n'), ((45, 2, 45, 43), 'waflib.Configure.conf.fatal', 'conf.fatal', ({(45, 13, 45, 42): "('Could not find xlf %r' % cmd)"}, {}), "('Could not find xlf %r' % cmd)", False, 'from waflib.Configure import conf\n'), ((48, 15, 48, 34), 're.compile', 're.compile', ({(48, 26, 48, 27): 'v', (48, 29, 48, 33): 're.I'}, {}), '(v, re.I)', False, 'import re\n')]