91 lines
3.1 KiB
Bash
91 lines
3.1 KiB
Bash
# sets up an environment for DAMASK on zsh
|
|
# usage: source DAMASK.zsh
|
|
|
|
function canonicalPath {
|
|
python -c "import os,sys; print(os.path.normpath(os.path.realpath(os.path.expanduser(sys.argv[1]))))" $1
|
|
}
|
|
|
|
function blink {
|
|
echo -e "\033[2;5m$1\033[0m"
|
|
}
|
|
|
|
ENV_ROOT=$(canonicalPath "${0:a:h}")
|
|
DAMASK_ROOT=$(canonicalPath "${0:a:h}'/..")
|
|
|
|
# shorthand command to change to DAMASK_ROOT directory
|
|
eval "function DAMASK_root() { cd $DAMASK_ROOT; }"
|
|
|
|
# defining set() allows to source the same file for tcsh and zsh, with and without space around =
|
|
set() {
|
|
export $1$2$3
|
|
}
|
|
source $ENV_ROOT/CONFIG
|
|
unset -f set
|
|
|
|
# add BRANCH if DAMASK_ROOT is a git repository
|
|
cd $DAMASK_ROOT >/dev/null; BRANCH=$(git branch 2>/dev/null| grep -E '^\* '); cd - >/dev/null
|
|
|
|
PATH=${DAMASK_ROOT}/bin:$PATH
|
|
|
|
SOLVER=$(which DAMASK_spectral || true 2>/dev/null)
|
|
[[ "x$SOLVER" == "x" ]] && SOLVER=$(blink 'Not found!')
|
|
|
|
[[ "x$DAMASK_NUM_THREADS" == "x" ]] && DAMASK_NUM_THREADS=1
|
|
|
|
# currently, there is no information that unlimited stack size causes problems
|
|
# still, http://software.intel.com/en-us/forums/topic/501500 suggest to fix it
|
|
# more info https://jblevins.org/log/segfault
|
|
# https://stackoverflow.com/questions/79923/what-and-where-are-the-stack-and-heap
|
|
# http://superuser.com/questions/220059/what-parameters-has-ulimit
|
|
ulimit -s unlimited 2>/dev/null # maximum stack size (kB)
|
|
|
|
# disable output in case of scp
|
|
if [ ! -z "$PS1" ]; then
|
|
echo
|
|
echo Düsseldorf Advanced Materials Simulation Kit --- DAMASK
|
|
echo Max-Planck-Institut für Eisenforschung GmbH, Düsseldorf
|
|
echo https://damask.mpie.de
|
|
echo
|
|
echo "Using environment with ..."
|
|
echo "DAMASK $DAMASK_ROOT $BRANCH"
|
|
echo "Grid Solver $SOLVER"
|
|
if [ "x$PETSC_DIR" != "x" ]; then
|
|
echo -n "PETSc location "
|
|
[ -d $PETSC_DIR ] && echo $PETSC_DIR || blink $PETSC_DIR
|
|
[[ $(canonicalPath "$PETSC_DIR") == $PETSC_DIR ]] \
|
|
|| echo " ~~> "$(canonicalPath "$PETSC_DIR")
|
|
fi
|
|
[[ "x$PETSC_ARCH" == "x" ]] \
|
|
|| echo "PETSc architecture $PETSC_ARCH"
|
|
echo -n "MSC.Marc/Mentat "
|
|
[ -d $MSC_ROOT ] && echo $MSC_ROOT || blink $MSC_ROOT
|
|
echo
|
|
echo "Multithreading DAMASK_NUM_THREADS=$DAMASK_NUM_THREADS"
|
|
echo -n "heap size "
|
|
[[ "$(ulimit -d)" == "unlimited" ]] \
|
|
&& echo "unlimited" \
|
|
|| echo $(python -c \
|
|
"import math; \
|
|
size=$(( 1024*$(ulimit -d) )); \
|
|
print('{:.4g} {}'.format(size / (1 << ((int(math.log(size,2) / 10) if size else 0) * 10)), \
|
|
['bytes','KiB','MiB','GiB','TiB','EiB','ZiB'][int(math.log(size,2) / 10) if size else 0]))")
|
|
echo -n "stack size "
|
|
[[ "$(ulimit -s)" == "unlimited" ]] \
|
|
&& echo "unlimited" \
|
|
|| echo $(python -c \
|
|
"import math; \
|
|
size=$(( 1024*$(ulimit -s) )); \
|
|
print('{:.4g} {}'.format(size / (1 << ((int(math.log(size,2) / 10) if size else 0) * 10)), \
|
|
['bytes','KiB','MiB','GiB','TiB','EiB','ZiB'][int(math.log(size,2) / 10) if size else 0]))")
|
|
echo
|
|
fi
|
|
|
|
export DAMASK_NUM_THREADS
|
|
export PYTHONPATH=$DAMASK_ROOT/python:$PYTHONPATH
|
|
|
|
for var in SOLVER BRANCH; do
|
|
unset "${var}"
|
|
done
|
|
unset "ENV_ROOT"
|
|
unset "DAMASK_ROOT"
|