#!/bin/bash
#
# Creates NA62 MC software tarballs for installation to the grid sites 
# protopop@cern.ch - May 9, 2012
# 
# Please check thoroughly before use!

VERSION=5 

SWPK1=NA62MC_v$VERSION
SWPK2=na62_geant4lib
SWPK3=na62_root
SWPK4=na62_X11
SWPK5=na62_ext

NA62SW_URL="http://ppewww.physics.gla.ac.uk/~protopop/na62/mc/gridsw"
NA62OFFLINE="/afs/cern.ch/na62/offline/"
ENVSCRIPT=~/NA62MC/scripts/env.sh

echo "Setting environment with $ENVSCRIPT ..."
source $ENVSCRIPT > /dev/null

echo " G4INSTALL = $G4INSTALL"
echo " ROOTSYS   = $ROOTSYS"

function pack {
   src="$1"
   if [[ "$src" == *"$NA62OFFLINE"* ]]; then
      dir=${src##$NA62OFFLINE}
      echo -n "Packing $dir ... "
      tar -C $NA62OFFLINE -zcf $2.tgz $dir
   else
      echo -n "Packing $src ... "
      tar -zcf $2.tgz $src
   fi
   ls -lah $2.tgz | awk  '{print " -> " $9 "  " $5 "B"}'
}

# na62mc
echo pack NA62MC $SWPK1

# geant4
echo pack $G4LIB/$G4SYSTEM $SWPK2

# root
# pack $ROOTSYS $SWPK3

# X11
# pack  ... $SWPK4

# sw
# pack  ... $SWPK5

NA62SW_DIR=na62sw_v$VERSION
NA62SW_ENV=na62env_v$VERSION".sh"

echo "Creating job environment file $NA62SW_ENV to be shipped with the tarballs ... "

cat > $NA62SW_ENV <<EOF
#!/bin/bash
# Script automatically produced by CreateTarballs.sh - protopop@cern.ch

echo Setting up job environment ...

LD_LIBRARY_PATH_ORIGINAL=\$LD_LIBRARY_PATH
LSD=\$VO_NA62_VO_GRIDPP_AC_UK_SW_DIR/$NA62SW_DIR
PATH=\$LSD/NA62MC/bin/$G4SYSTEM:\$PATH

# NA62 internal libraries
NA62_LIBS=\$LSD/NA62MC/tmp/$G4SYSTEM/NA62MC:\$LSD/NA62MC/Persistency/lib:\$LSD/NA62MC/Beam/src
for DETLIB in RICH GigaTracker CHOD Spectrometer MUV1 MUV2 MUV3 SAC LAV IRC LKr Cedar
do
  NA62_LIBS=\$NA62_LIBS:\$LSD/NA62MC/\$DETLIB/Persistency/lib
done

LD_LIBRARY_PATH=\$NA62_LIBS
EXTLIBS="$G4INSTALL $XERCESCROOT $CLHEP_BASE_DIR $ROOTSYS sw/lcg/contrib/gcc/4.3.2/x86_64-slc5-gcc34-opt/lib64 root"
for EXTLIB in \$EXTLIBS
do
  LOCLIB=\${EXTLIB##\$NA62OFFLINE}
  LD_LIBRARY_PATH=\$LD_LIBRARY_PATH:\$LSD/\$LOCLIB/lib
done 
LD_LIBRARY_PATH=\$LD_LIBRARY_PATH:\$LSD/usr/lib64/root:\$LSD/sw/usr/lib64

echo " PATH=\$PATH"
echo " LD_LIBRARY_PATH=\$LD_LIBRARY_PATH"

export PATH LD_LIBRARY_PATH

EOF

echo "Creating the s/w installation script ..."

cat > na62swinstall.sh <<EOF 
#!/bin/bash
# Script automatically produced by CreateTarballs.sh - protopop@cern.ch

NA62SW_URL=$NA62SW_URL
NA62SW_DIR=$NA62SW_DIR
NA62SW_ENV=$NA62SW_ENV

function unpack {
   wget -r \$NA62SW_URL/\$1.tgz
   tar zxvf \$1.tgz
}

mkdir -p \$NA62SW_DIR
cd \$NA62SW_DIR

echo "Downloading software tarballs from \$NA62SW_URL ..."

unpack $SWPK1
unpack $SWPK2
unpack $SWPK3
unpack $SWPK4
unpack $SWPK5

wget -r \$NA62SW_URL/\$NA62SW_ENV
source \$NA62SW_ENV

EOF

chmod u+x na62swinstall.sh