#! /usr/bin/env python
# encoding: utf-8
# KWS 2010
# Thomas Nagy 2016-2018 (ita)

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

@conf
def find_g95(conf):
	fc = conf.find_program('g95', var='FC')
	conf.get_g95_version(fc)
	conf.env.FC_NAME = 'G95'

@conf
def g95_flags(conf):
	v = conf.env
	v.FCFLAGS_fcshlib   = ['-fPIC']
	v.FORTRANMODFLAG  = ['-fmod=', ''] # template for module path
	v.FCFLAGS_DEBUG = ['-Werror'] # why not

@conf
def g95_modifier_win32(conf):
	fc_config.fortran_modifier_win32(conf)

@conf
def g95_modifier_cygwin(conf):
	fc_config.fortran_modifier_cygwin(conf)

@conf
def g95_modifier_darwin(conf):
	fc_config.fortran_modifier_darwin(conf)

@conf
def g95_modifier_platform(conf):
	dest_os = conf.env.DEST_OS or Utils.unversioned_sys_platform()
	g95_modifier_func = getattr(conf, 'g95_modifier_' + dest_os, None)
	if g95_modifier_func:
		g95_modifier_func()

@conf
def get_g95_version(conf, fc):
	"""get the compiler version"""

	version_re = re.compile(r"g95\s*(?P<major>\d*)\.(?P<minor>\d*)").search
	cmd = fc + ['--version']
	out, err = fc_config.getoutput(conf, cmd, stdin=False)
	if out:
		match = version_re(out)
	else:
		match = version_re(err)
	if not match:
		conf.fatal('cannot determine g95 version')
	k = match.groupdict()
	conf.env.FC_VERSION = (k['major'], k['minor'])

def configure(conf):
	conf.find_g95()
	conf.find_ar()
	conf.fc_flags()
	conf.fc_add_flags()
	conf.g95_flags()
	conf.g95_modifier_platform()