#!/bin/bash


source /shared/setup_env.sh


ARCH=$(uname -m)
#Build packages
if [ $ARCH = "x86_64" ]
then
    time spack install wps %intel ^intel-oneapi-mpi #This is going to install WPS and WRF
    time spack install ncview%intel
    time spack install wgrib2%intel 
    ARCH_CONFIG=spack load wgrib2%intel
elif [ $ARCH = "aarch64" ]
then
    #Remove mirrors.yaml because of errors during build
    mv ${SHARED_DIR}/spack/etc/spack/mirrors.yaml ${SHARED_DIR}/spack/etc/spack/mirrors.yaml.old
    touch ${SHARED_DIR}/spack/etc/spack/mirrors.yaml

    #Update packages.yaml to avoid rebuildling already installed componentis
    sudo pip3 install pyyaml
    mv ${SHARED_DIR}/spack/etc/spack/packages.yaml ${SHARED_DIR}/spack/etc/spack/packages.yaml.old
    python3 ${SHARED_DIR}/hpc-workshop-wrf/wrf_setup_scripts/update_packages_yaml.py  ${SHARED_DIR}/spack/etc/spack/packages.yaml.old >  ${SHARED_DIR}/spack/etc/spack/packages.yaml

    spack install wps   #This is going to install WPS and WRF
    spack install ncview
    
    #spack install    wgrib2
    #Manual install wgrib2 because of spack open issue on Graviton2
    mkdir -p ${BUILDDIR}
    cd ${SHARED_DIR}/download
     wget ftp://ftp.cpc.ncep.noaa.gov/wd51we/wgrib2/wgrib2.tgz.v3.1.0 -O wgrib2.tgz
    ################## COMPILE wgrib2  ################################################
    cd $BUILDDIR
    tar -zxvf /shared/download/wgrib2.tgz
    cd grib2
    bash <<@EOF
    export FC=gfortran
    export COMP_SYS=gnu_linux
    export NETCDF=\$(spack location -i netcdf-c)
    export JASPERDIR=\$(spack location -i jasper)    
    export CPPFLAGS="-I\$NETCDF/include/ -I\$JASPERDIR/include  -L\$NETCDF/lib/ -L\$JASPERDIR/lib64/"
    echo "makefile < ${SETUP_DIR}/wgrib_makefile.patch"
    patch --fuzz 3 makefile < ${SETUP_DIR}/wrf_setup_scripts/wgrib_makefile.patch
    make
    cp wgrib2/wgrib2  $DIR/bin
    ARCH_CONFIG=""
@EOF
else
    echo "Unsupported Architecture $ARCH"
    exit 1
fi


cat <<@EOF >> /shared/setup_env.sh
spack load wrf
spack load wps
spack load ncview

$ARCH_CONFIG

export WRF_DIR=\$(spack location -i wrf)
export WPS_DIR=\$(spack location -i wps)
export NETCDF=\$(spack location -i netcdf-c)
export JASPER=\$(spack location -i jasper)

export NETCDFRLIB=\$NETCDF/lib
export NETCDFINC=\$NETCDF/include
export JASPERLIB=\$JASPER/lib64
export JASPERINC=\$JASPER/include
export LD_LIBRARY_PATH=\${NETCDFRLIB}:\${JASPERLIB}:\$LD_LIBRARY_PATH

export WRF_ENV=true
@EOF