trilinos.package

################################################################################
## TRILINOS                                                                   ##
################################################################################

if [ -z "${TRILINOS_MAJOR_VERSION}" ]; then
    TRILINOS_MAJOR_VERSION=AUTO
fi

if [ ${TRILINOS_MAJOR_VERSION} = "AUTO" ]; then
    TRILINOS_MAJOR_VERSION="13"
fi

if [ ${TRILINOS_MAJOR_VERSION} = "DEV" ];then
    # download from github:
    # this is the last "blessed" version we like:
    VERSION=dbf41f3b26b0326a1377f219e6e07eab861d181e
    EXTRACTSTO=trilinos-12.dbf41f

    NAME=Trilinos.git
    PACKING=git
    SOURCE=https://github.com/trilinos/
elif [ ${TRILINOS_MAJOR_VERSION} = "13" ]; then
    VERSION=13-2-0
    CHECKSUM=099680cd3660dba5ec447ddc50a8406c
    
    SOURCE=https://github.com/trilinos/Trilinos/archive/
    NAME=trilinos-release-${VERSION}
    EXTRACTSTO=Trilinos-trilinos-release-${VERSION}
    PACKING=.tar.gz

elif [ ${TRILINOS_MAJOR_VERSION} = "12" ]; then
    VERSION=12-18-1
    CHECKSUM=9c1d151169949bca6cf203831e4d6aee

    # VERSION=12-14-1
    # CHECKSUM=de912cca43c2ca3b74aa08528ac39dbd

    # VERSION=12-12-1
    # CHECKSUM=ecd4606fa332212433c98bf950a69cc7


    # note: the official trilinos tarball changed md5sum but we confirmed that the contents is
    # identical. We accept both hashes here.
    # VERSION=12-10-1
    # CHECKSUM="667333dbd7c0f031d47d7c5511fd0810 40f28628b63310f9bd17c26d9ebe32b1"
    
    # VERSION=12-4-2
    # CHECKSUM=4c25a757d86bde3531090bd900a2cea8

    SOURCE=https://github.com/trilinos/Trilinos/archive/
    NAME=trilinos-release-${VERSION}
    EXTRACTSTO=Trilinos-trilinos-release-${VERSION}
    PACKING=.tar.gz

elif [ ${TRILINOS_MAJOR_VERSION} = "11" ]; then
    VERSION=11-14-3
    CHECKSUM=dea62e57ebe51a886bee0b10a2176969
    
    #Note: For Trilinos v11 we need to force SuperLUDist=OFF,
    #      since only the deprecated SuperLUDist v3.3 together with an Amesos patch
    #      (and not the current SuperLUDist v4.x with the same Amesos patch)
    #      is working.
    #Note: This issue is resolved by setting
    #          -D HAVE_SUPERLUDIST_LUSTRUCTINIT_2ARG
    #      for Trilinos v12; this is neglegted by Trilinos v11 since it is unknown.
    cecho ${BAD} "trilinos: warning: forcing unset SUPERLU_DIR (not supported)"
    unset SUPERLU_DIR

    SOURCE=https://github.com/trilinos/Trilinos/archive/
    NAME=trilinos-release-${VERSION}
    EXTRACTSTO=Trilinos-trilinos-release-${VERSION}
    PACKING=.tar.gz
else
    cecho ${BAD} "Unknown Trilinos version ${TRILINOS_MAJOR_VERSION} forced, please use AUTO|12|11."
    exit
fi

BUILDCHAIN=cmake

INSTALL_PATH=${INSTALL_PATH}/${NAME}

################################################################################
# If you have further options for trilinos, please set:
#TRILINOS_CONFOPTS=""

################################################################################
# Please do not change the following options

# Set blas and lapack directories
if [ ! -z "${BLAS_DIR}" ]; then
    cecho ${INFO} "trilinos: configuration with BLAS_DIR=${BLAS_DIR}"
    CONFOPTS="${CONFOPTS} \
      -D BLAS_LIBRARY_DIRS:STRING=${BLAS_DIR}"
fi

if [ ! -z "${LAPACK_DIR}" ]; then
    cecho ${INFO} "trilinos: configuration with LAPACK_DIR=${LAPACK_DIR}"
    CONFOPTS="${CONFOPTS} \
      -D LAPACK_LIBRARY_DIRS:STRING=${LAPACK_DIR}"
fi

if [ ! -z "${SCALAPACK_DIR}" ]; then
    cecho ${INFO} "trilinos: configuration with SCALAPACK_DIR=${SCALAPACK_DIR}"
    CONFOPTS="\
        ${CONFOPTS} \
        -D TPL_ENABLE_SCALAPACK:BOOL=ON \
        -D SCALAPACK_LIBRARY_DIRS=${SCALAPACK_DIR}/lib"
fi

if [ ! -z "${MUMPS_DIR}" ]; then
    cecho ${INFO} "trilinos: configuration with MUMPS_DIR=${MUMPS_DIR}"
    CONFOPTS="${CONFOPTS} \
      -D TPL_ENABLE_MUMPS=ON \
      -D MUMPS_INCLUDE_DIRS:STRING=${MUMPS_DIR}/include \
      -D MUMPS_LIBRARY_DIRS:STRING=${MUMPS_DIR}/lib"
fi

# Set intel MKL options
if [ ${MKL} = ON ]; then
    cecho ${INFO} "trilinos: configuration with MKL"

    if [ -z "${BLAS_DIR}" ] && [ -z "${LAPACK_DIR}" ]; then
        cecho ${INFO} "trilinos: configuration with BLAS_DIR=${MKL_DIR}"
        cecho ${INFO} "trilinos: configuration with LAPACK_DIR=${MKL_DIR}"
        CONFOPTS="${CONFOPTS} \
          -D BLAS_LIBRARY_DIRS:STRING=${MKL_DIR} \
          -D LAPACK_LIBRARY_DIRS:STRING=${MKL_DIR}"
    fi

    # Trilinos will complain that MKL does not support HAVE_TEUCHOS_BLASFLOAT. See
    # the discussion in https://github.com/dealii/candi/pull/92 for more details.
    cecho ${INFO} "trilinos: disabling some Tpetra instantiations because you are using MKL"
    
    CONFOPTS="${CONFOPTS} \
      -D BLAS_LIBRARY_NAMES:STRING='mkl_core;mkl_sequential' \
      -D Tpetra_INST_FLOAT:BOOL=OFF \
      -D Tpetra_INST_COMPLEX_FLOAT:BOOL=OFF \
      -D LAPACK_LIBRARY_NAMES:STRING=mkl_intel_lp64"
else
    if [ ! -z "${BLAS_LIB}" ]; then
        # We need to specify the full name if using openblas.package:
        cecho ${INFO} "trilinos: configuration with BLAS_LIB=${BLAS_LIB}"
        CONFOPTS="${CONFOPTS} \
          -D TPL_BLAS_LIBRARIES:STRING=${BLAS_LIB} \
          -D TPL_LAPACK_LIBRARIES:STRING=${BLAS_LIB}"
    fi
fi

# Set compilers & compiler options
if [ ! -z "${CC}" ]; then
    CONFOPTS="${CONFOPTS} \
      -D CMAKE_C_COMPILER=${CC}"
fi

if [ ! -z "${CXX}" ]; then
    CONFOPTS="${CONFOPTS} \
      -D CMAKE_CXX_COMPILER=${CXX}"
fi

if [ ! -z "${FC}" ]; then
    CONFOPTS="${CONFOPTS} \
      -D CMAKE_Fortran_COMPILER=${FC}"
fi

if [ ${NATIVE_OPTIMIZATIONS} = OFF ]; then
CONFOPTS="${CONFOPTS} \
  -D CMAKE_CXX_FLAGS:STRING='-fPIC -g -O3' \
  -D CMAKE_C_FLAGS:STRING='-fPIC -g -O3' \
  -D CMAKE_FORTRAN_FLAGS:STRING='-g -O3'"
else
CONFOPTS="${CONFOPTS} \
  -D CMAKE_CXX_FLAGS:STRING='-fPIC -g -O3 -march=native' \
  -D CMAKE_C_FLAGS:STRING='-fPIC -g -O3 -march=native' \
  -D CMAKE_FORTRAN_FLAGS:STRING='-g -O3 -march=native'"
fi

# Add ParMETIS, if present
if [ ! -z "${PARMETIS_DIR}" ]; then
    cecho ${INFO} "trilinos: configuration with ParMETIS"

    # NOTE: if parmetis v4.0.3 is not found, but installed, add
    #  -D HAVE_PARMETIS_VERSION_4_0_3=ON"

    CONFOPTS="${CONFOPTS} \
      ${TRILINOS_PARMETIS_CONFOPTS} \
      -D TPL_ENABLE_ParMETIS:BOOL=ON \
      -D TPL_ParMETIS_LIBRARIES:FILEPATH='${PARMETIS_DIR}/lib/libparmetis.${LDSUFFIX};${PARMETIS_DIR}/lib/libmetis.${LDSUFFIX}' \
      -D TPL_ParMETIS_INCLUDE_DIRS:PATH=${PARMETIS_DIR}/include"
fi

# Add SuperLU_dist, if present
if [ ! -z "${SUPERLU_DIR}" ]; then
    cecho ${INFO} "trilinos: configuration with SuperLU_dist"

    CONFOPTS="${CONFOPTS} \
      -D TPL_ENABLE_SuperLUDist:BOOL=ON \
      -D TPL_SuperLUDist_LIBRARIES:FILEPATH=${SUPERLU_DIR}/lib/libsuperlu_dist.${LDSUFFIX} \
      -D SuperLUDist_INCLUDE_DIRS:PATH=${SUPERLU_DIR}/include \
      -D HAVE_SUPERLUDIST_LUSTRUCTINIT_2ARG:BOOL=ON \
      -D HAVE_SUPERLUDIST_ENUM_NAMESPACE:BOOL=ON"
fi

#########################################################################
# General Trilinos configuration

CONFOPTS="\
  -D TPL_ENABLE_MPI:BOOL=ON \
  -D Trilinos_ENABLE_OpenMP:BOOL=OFF \
  -D TPL_ENABLE_TBB:BOOL=OFF \
  -D Trilinos_VERBOSE_CONFIGURE:BOOL=OFF \
  -D Trilinos_ENABLE_EXPLICIT_INSTANTIATION=ON \
  -D Trilinos_ENABLE_Amesos:BOOL=ON \
  -D Trilinos_ENABLE_Epetra:BOOL=ON \
  -D Trilinos_ENABLE_EpetraExt:BOOL=ON \
  -D Trilinos_ENABLE_Ifpack:BOOL=ON \
  -D Trilinos_ENABLE_Ifpack2:BOOL=OFF \
  -D Trilinos_ENABLE_Tpetra:BOOL=ON \
  -D   Tpetra_INST_INT_LONG_LONG:BOOL=ON \
  -D Trilinos_ENABLE_AztecOO:BOOL=ON \
  -D Trilinos_ENABLE_Sacado:BOOL=ON \
  -D Trilinos_ENABLE_Teuchos:BOOL=ON \
  -D   Teuchos_ENABLE_FLOAT:BOOL=ON \
  -D Trilinos_ENABLE_MueLu:BOOL=ON \
  -D Trilinos_ENABLE_ML:BOOL=ON \
  -D Trilinos_ENABLE_ROL:BOOL=ON \
  -D Trilinos_ENABLE_Zoltan:BOOL=ON \
  -D Trilinos_ENABLE_Stratimikos:BOOL=ON \
  -D TPL_ENABLE_Boost:BOOL=OFF \
  -D Trilinos_ENABLE_Belos:BOOL=ON \
  -D Trilinos_ENABLE_Amesos2:BOOL=ON \
  -D CMAKE_BUILD_TYPE:STRING=RELEASE \
  -D CMAKE_VERBOSE_MAKEFILE:BOOL=OFF \
  -D BUILD_SHARED_LIBS:BOOL=ON \
  ${CONFOPTS}"

if [ ${TRILINOS_WITH_COMPLEX} = ON ]; then
    CONFOPTS="\
    -D Trilinos_ENABLE_COMPLEX_DOUBLE=ON \
    -D Trilinos_ENABLE_COMPLEX_FLOAT=ON \
    -D Teuchos_ENABLE_COMPLEX:BOOL=ON \
    ${CONFOPTS}"
fi

# finally append user options:
CONFOPTS="${CONFOPTS} \
  ${TRILINOS_CONFOPTS}"

package_specific_register () {
    export TRILINOS_DIR=${INSTALL_PATH}
}

package_specific_conf () {
    # Generate configuration file
    CONFIG_FILE=${CONFIGURATION_PATH}/${NAME}
    rm -f $CONFIG_FILE
    echo "
export TRILINOS_DIR=${INSTALL_PATH}
" >> $CONFIG_FILE
}