#!/usr/bin/perl

#

# 1. Not specified

#   1.1 Automatically detect, then check compiler

#   1.2 If no fortran compiler is detected, g77 is default with NOFORTRAN definition

# 2. Specified

#   2.1 If path is correct, check compiler

#   2.2 If path is not correct, but still valid compiler name, force setting

#     2.2.2 Path is not correct, invalid compiler name, then g77 is default with NOFORTRAN definition

#

$makefile = shift(@ARGV);

$config   = shift(@ARGV);

$nofortran = 0;

$compiler = join(" ", @ARGV);

# f77 is too ambiguous

$compiler = "" if $compiler eq "f77";

@path = split(/:/, $ENV{"PATH"});

if ($compiler eq "") {

    @lists = ("f77", "g77", "g95", "gfortran", "frt", "fort", "openf90", "openf95",

	      "sunf77", "sunf90", "sunf95",

              "xlf95", "xlf90", "xlf", 

              "ppuf77", "ppuf95", "ppuf90", "ppuxlf", 

	      "pathf90", "pathf95", 

	      "pgf95", "pgf90", "pgf77",

              "ifort");

    foreach $lists (@lists) {

        foreach $path (@path) {

            if (-f $path . "/" . $lists) {

                $compiler = $lists;

                break;

            }

        }

    }

}

if ($compiler eq "") {

    $nofortran = 1;

    $compiler = "g77";

    $vendor = G77;

    $bu       = "_";

} else {

    $data = `which $compiler > /dev/null 2> /dev/null`;

    $vendor = "";

    if (!$?) {

	$data = `$compiler -O2 -S ftest.f > /dev/null 2>&1 && cat ftest.s && rm -f ftest.s`;

	if ($data =~ /zhoge_/) {

	    $bu       = "_";

	}

	if ($data =~ /GNU/) {

	    $data =~ /(\d)\.(\d).(\d)/;

	    $major = $1;

	    $minor = $2;

	    if ($major >= 4) {

		$vendor = GFORTRAN;

		$openmp = "-fopenmp";

	    } else {

		$vendor = G77;

		$openmp = "";

	    }

	}

	if ($data =~ /g95/) {

	    $vendor = G95;

	    $openmp = "";

	}

	if ($data =~ /Intel/) {

	    $vendor = INTEL;

	    $openmp = "-openmp";

	}

        if ($data =~ /Sun Fortran/) {

            $vendor = SUN;

	    $openmp = "-xopenmp=parallel";

        }

	if ($data =~ /PathScale/) {

	    $vendor = PATHSCALE;

	    $openmp = "-openmp";

	}

	if ($data =~ /Open64/) {

	    $vendor = OPEN64;

	    $openmp = "-mp";

	}

	if ($data =~ /PGF/) {

	    $vendor = PGI;

	    $openmp = "-mp";

	}

	if ($data =~ /IBM/) {

	    $vendor = IBM;

	    $openmp = "-openmp";

	}

    }

    if ($vendor eq "") {

	if ($compiler =~ /g77/) {

	    $vendor = G77;

	    $bu       = "_";

	    $openmp = "";

	}

	if ($compiler =~ /g95/) {

	    $vendor = G95;

	    $bu       = "_";

	    $openmp = "";

	}

	if ($compiler =~ /gfortran/) {

	    $vendor = GFORTRAN;

	    $bu       = "_";

	    $openmp = "-fopenmp";

	}

	if ($compiler =~ /ifort/) {

	    $vendor = INTEL;

	    $bu       = "_";

	    $openmp = "-openmp";

	}

	if ($compiler =~ /pathf/) {

	    $vendor = PATHSCALE;

	    $bu       = "_";

	    $openmp = "-mp";

	}

	if ($compiler =~ /pgf/) {

	    $vendor = PGI;

	    $bu       = "_";

	    $openmp = "-mp";

	}

	if ($compiler =~ /ftn/) {

	    $vendor = PGI;

	    $bu       = "_";

	    $openmp = "-openmp";

	}

	if ($compiler =~ /frt/) {

	    $vendor = FUJITSU;

	    $bu       = "_";

	    $openmp = "-openmp";

	}

	if ($compiler =~ /sunf77|sunf90|sunf95/) {

	    $vendor = SUN;

	    $bu       = "_";

	    $openmp = "-xopenmp=parallel";

	}

	if ($compiler =~ /ppuf/) {

	    $vendor = IBM;

	    $openmp = "-openmp";

	}

	if ($compiler =~ /xlf/) {

	    $vendor = IBM;

	    $openmp = "-openmp";

	}

	if ($compiler =~ /open64/) {

	    $vendor = OPEN64;

	    $openmp = "-mp";

	}

	if ($vendor eq "") {

	    $nofortran = 1;

	    $compiler = "g77";

	    $vendor = G77;

	    $bu       = "_";

	    $openmp = "";

	}

    }

}

$data = `which $compiler > /dev/null 2> /dev/null`;

if (!$?) {

    $binary = $ENV{"BINARY"};

    $openmp = "" if $ENV{USE_OPENMP} != 1;

    if ($binary == 32) {

	$link = `$compiler $openmp -m32 -v ftest2.f 2>&1 && rm -f a.out a.exe`;

	if ($?) {

	    $link = `$compiler $openmp -q32 -v ftest2.f 2>&1 && rm -f a.out a.exe`;

	}

	$binary = "" if ($?);

    }

    if ($binary == 64) {

	$link = `$compiler $openmp -m64 -v ftest2.f 2>&1 && rm -f a.out a.exe`;

	if ($?) {

	    $link = `$compiler $openmp -q64 -v ftest2.f 2>&1 && rm -f a.out a.exe`;

	}

	$binary = "" if ($?);

    }

    if ($binary eq "") {

	$link = `$compiler $openmp -v ftest2.f 2>&1 && rm -f a.out a.exe`;

    }

}

$linker_L = "";

$linker_l = "";

$linker_a = "";

if ($link ne "") {

    $link =~ s/\-Y\sP\,/\-Y/g;

    $link =~ s/\-rpath\s+/\-rpath\@/g;

    @flags = split(/[\s\,\n]/, $link);

    # remove leading and trailing quotes from each flag.

    @flags = map {s/^['"]|['"]$//g; $_} @flags;

    foreach $flags (@flags) {

	if (

	    ($flags =~ /^\-L/)

	    && ($flags !~ /^-LIST:/)

	    && ($flags !~ /^-LANG:/)

	    ) {

	    if ($vendor eq "PGI") {

		$flags =~ s/lib$/libso/;

	    }

	    $linker_L .= $flags . " ";

	}

	if ($flags =~ /^\-Y/) {

	    $linker_L .= "-Wl,". $flags . " ";

	    }

	if ($flags =~ /^\-rpath/) {

	    $flags =~ s/\@/\,/g;

	    if ($vendor eq "PGI") {

		$flags =~ s/lib$/libso/;

	    }

	    $linker_L .= "-Wl,". $flags . " " ;

	}

	if (

	    ($flags =~ /^\-l/) 

	    && ($flags !~ /gfortranbegin/)

	    && ($flags !~ /frtbegin/)

	    && ($flags !~ /pathfstart/)

	    && ($flags !~ /numa/)

	    && ($flags !~ /crt[0-9]/)

	    && ($flags !~ /gcc/)

	    && ($flags !~ /user32/)

	    && ($flags !~ /kernel32/)

	    && ($flags !~ /advapi32/)

	    && ($flags !~ /shell32/)

		&& ($flags !~ /^\-l$/)

	    ) {

	    $linker_l .= $flags . " ";

	}

	$linker_a .= $flags . " " if $flags =~ /\.a$/;

    }

}

if ($vendor eq "INTEL"){

    $linker_a .= "-lgfortran"

}

open(MAKEFILE, ">> $makefile") || die "Can't append $makefile";

open(CONFFILE, ">> $config"  ) || die "Can't append $config";

print MAKEFILE "F_COMPILER=$vendor\n";

print MAKEFILE "FC=$compiler\n";

print MAKEFILE "BU=$bu\n" if $bu ne "";

print MAKEFILE "NOFORTRAN=1\n" if $nofortran == 1;

print CONFFILE "#define BUNDERSCORE\t$bu\n" if $bu ne "";

print CONFFILE "#define NEEDBUNDERSCORE\t1\n" if $bu ne "";

if (($linker_l ne "") || ($linker_a ne "")) {

    print MAKEFILE "FEXTRALIB=$linker_L $linker_l $linker_a\n";

}

close(MAKEFILE);

close(CONFFILE);