#!/bin/bash
# Copyright # Copyright (C) 2022 by Amazon.com, Inc. or its affiliates.  All Rights Reserved.
# SPDX-License-Identifier: MIT

OPENFOAM_VERSION=${OPENFOAM_VERSION:-2212}

case ${HPC_COMPILER} in
    "amdclang")
	WM_COMPILER=Amd
	;;
    "armclang")
	WM_COMPILER=Arm
	;;
    "armgcc"|"gcc")
	WM_COMPILER=Gcc
	;;
    "clang")
	WM_COMPILER=Clang
	;;
    "nvc")
	WM_COMPILER=Nvidia
	;;
    "icc")
	WM_COMPILER=Icc
	;;
    "icx")
	WM_COMPILER=Icx
	;;
    *)
	exit 1
	;;
esac


case ${HPC_MPI} in
    "intelmpi")
	export MPI_ROOT=${I_MPI_ROOT}
        export MPI_ARCH_FLAGS="-DMPICH_SKIP_MPICXX -DOMPI_SKIP_MPICXX"
	WM_MPLIB="INTELMPI"
	;;
    "mpich"|"mvapich"|"openmpi")
	export MPI_ROOT=${HPC_PREFIX}/${HPC_COMPILER}/${HPC_MPI}
	export MPI_ARCH_FLAGS="-DMPICH_SKIP_MPICXX -DOMPI_SKIP_MPICXX"
	export MPI_ARCH_INC="-isystem ${MPI_ROOT}/include"
	WM_MPLIB="SYSTEMMPI"
	if [ "${HPC_MPI}" == "openmpi" ]
	then 
	    export MPI_ARCH_LIBS="-L${MPI_ROOT}/lib -lmpi"
	else
	    export MPI_ARCH_LIBS="-L${MPI_ROOT}/lib -lmpi -lrt"
	fi
	;;
    *)
	echo "unknown or unsupported MPI"
	exit 1
	;;
esac

. "${HPC_PREFIX}/${HPC_COMPILER}/${HPC_MPI}/OpenFOAM/OpenFOAM-${OPENFOAM_VERSION}/etc/bashrc" WM_COMPILER=${WM_COMPILER} WM_MPLIB=${WM_MPLIB}