SOLARIS_Analysis/SetupNewExp

#!/bin/bash -l


###################################################
##
##    This is a script for Setup/switch Experiment
##
##   
##
####################################################


RED='\033[1;31m'
YELLOW='\033[1;33m'
ORANGE='\033[0;33m'
GREEN='\033[1;32m'
BLUE='\033[0;34m'
Cyan='\033[0;36m'
NC='\033[0m'  #no color


##############   need to distingish mac and daq
Arch="$(uname -s)"
PCName="$(hostname)"
PCID=-1      #if PCID == 1 (DAQ), 2 (MAC), -1(OTHER)

#------ Set up data folder, check disk space
echo -e "${YELLOW} 0) ################## Check computer name and arch. ${NC}"
echo "PC name  : ${PCName}"
echo "Archetech: ${Arch}"

if [ ${Arch} == "Linux" ] && [ ${PCName} == "solaris-daq" ]; then 

  PCID=1

  pathsSetting=${HOME}/SOLARIS_QT6_DAQ/programSettings.txt
  if [ -e ${pathsSetting} ]; then
    echo "Found DAQ programSettings.txt for paths settings"

    analysisPath=$(cat ${pathsSetting} | head -n 2 | tail -n 1)

    if [ "${analysisPath}" = "$(pwd)" ]; then
      echo "The analysisPath from ${analysisPath} is same as the present folder $(pwd)"
    else
      echo "The analysisPath from ${analysisPath} is different from present folder $(pwd). Abort."
      exit
    fi

    rawDataPathParent=$(cat ${pathsSetting} | head -n 3 | tail -n 1)
    rootDataPathParent=$(cat ${pathsSetting} | head -n 4 | tail -n 1)

    databaseIP=$(cat ${pathsSetting} | head -n 6 | tail -n 1)
    databaseName=$(cat ${pathsSetting} | head -n 7 | tail -n 1)

    #echo ${rawDataPathParent}
    #echo ${rootDataPathParent}
    #echo ${databaseIP}
    #echo ${databaseName}

  else 

    echo "${RED} Cannot found DAQ programSettings.txt for path settings ${NC}"
    echo "Seek Ryan for help"
    exit

  fi

fi


if [ ${Arch} == "Darwin" ] && [ ${PCName} == "SOLARISs-Mac-Studio.local" ]; then 
  PCID=2
  rawDataPathParent=${HOME}/experimentalData/
  rootDataPathParent=${HOME}/experimentalData/
fi

############### Check is git exist
isGitExist=0
if [ -d $(pwd)/".git" ]; then
  echo -e "Git repository exists."
else
  echo -e "Git repository does not exist. Abort."
  if [ ${PCID} -eq 1 ]; then
    echo -e "Use SOLARIS DAQ to setup new experiment"
  fi
  exit
fi

if [ ${PCID} -eq 1 ]; then
  echo -e "================================================ ${YELLOW}SOLARIS DAQ ${NC}"
elif [ ${PCID} -eq 2 ]; then
  echo -e "================================================ ${YELLOW}SOLARIS MAC ${NC}"
else
  echo -e "================================================ ${YELLOW}Other PC ${NC}"
fi
echo " The bach script do following things "
echo " 1) git-fetch  "
echo " 2) check repository is clean"
echo " 3) check is git branch exist"
echo "    branch exist --> git checkout "
echo "    branch not exist --> create expName.sh"
echo " 4) making directories for data, root_data"
echo " 5) create symbolic links to data, root_data"
echo " 6) Tell Database SetupNewExp change exp. (only from the DAQ)"
echo " 7) git push (if new exp.)"
echo "============================================================="


expName=$1

###############  this will define global enviroment varibales, like the analysis path, export armory
#source SOLARIS.sh

###############  Check git
echo -e "${YELLOW} 1) ################## Git Fetch ${NC}"
echo "when password is needed, please edit .git/config"
git fetch

echo -e "${YELLOW} 2) ################## Checking git repository is clean or not.... ${NC}"
gitCheck=`git status --porcelain --untracked-files=no | wc -l`
if [ ${gitCheck} -eq 0 ]; then
  echo "---- clean."
else
  git status
  echo -e "\033[0;93m=============== Please fix the git status \033[0m"
  exit
fi

if [ $# -eq 0 ]; then
  git branch -a
  read -p 'Enter the new experiment name: ' expName    
fi

############### check is there any GIt branch in repository, 
#               if not create Git Branch, if yes, checkout
echo -e "${YELLOW} 3) ################## Checking/Create Git Branch ${NC}"
pullFlag="K"
isBranchExist=`git branch -a | grep ${expName} | wc -l`
if [ ${expName} == "Master" ]; then
  echo "this is the Master experiment name. no branch create. checkout Master"
  git checkout Master
  isBranchExist=1
else

  if [ ${isBranchExist} -eq 0 ]; then

    if [ ${PCID} -eq 1 ]; then

      echo -e "You are going to create a${GREEN} NEW ${NC}branch ${Cyan}${expName}${NC} from branch : $(git branch --show-current)"
      read -p "Are you sure? (Y/N) " createNewBranch
      if [ ${createNewBranch} == "Y" ]; then
        git checkout -b ${expName}
      else
        exit
      fi
    
    else

      echo -e "${Cyan} ONLY the DAQ machines can create branch ${NC}"

    fi

  else
    echo -e "Experimental (branch) Name ${expName} ${GEEN} already exist ${NC}."
    echo "Please take another name or git pull origin ${expName}"
    read -p "Do you want to checkout origin/${expName} (Y/N):"  pullFlag
    if [ ${pullFlag} == "N" ]; then
      exit
    else
      git checkout ${expName}
    fi
  fi
fi




############### set up expName.sh, so that all experimental Name is refered to this name
if [ ${isBranchExist} -eq 0 ]; then

  workingDir=$(pwd)"/working/"
  expNamePath=$(pwd)"/working/expName.sh"
  echo -e "${YELLOW} 3a) ################## Setting up ${expNamePath} ${NC}"

  if [ ! -d ${workingDir} ]; then
      mkdir -p ${workingDir}  # create parent directory
  fi

  touch ${expNamePath}
  echo "#!/bin/bash -l" > ${expNamePath}
  echo "expName=${expName}" >> ${expNamePath}
  echo "rawDataPath=${rawDataPathParent}"/${expName} >> ${expNamePath}
  echo "rootDataPath=${rootDataPathParent}"/${expName} >> ${expNamePath}
  echo "runID=0" >> ${expNamePath}
  echo "elogID=0" >> ${expNamePath}
fi

echo -e "${YELLOW} 4) ##################  making new folders in ${rawDataPathParent} ${NC}"
rawData=${rawDataPathParent}/${expName}
rootData=${rootDataPathParent}/${expName}

mkdir -v ${rawData}
mkdir -v ${rootData}

############### create symbolic links
echo -e "${YELLOW} 5) ################## creating symbolic links ${NC}"
rm -f data_raw
rm -f root_data

ln -sfv ${rawData} data_raw
ln -sfv ${rootData} root_data


if [ ${PCID} -eq 1 ]; then 

  ############### tell Database SetupNewExp change exp.
  
  echo -e "=================== Tell the database"

  haha1=${databaseIP}"/write?db="${databaseName}
  haha2="RunID,start=0 value=0,expName=\"${expName}\",comment=\"New_experiment_[${expName}]_was_created\""

  #echo $haha1
  #echo $haha2

  curl -s -v -XPOST "$haha1" --data-binary "$haha2"  --max-time 1 --connect-timeout 1

  #===== clean up working if it is new
  if [ ${isBranchExist} -eq 0 ]; then

    echo "======== Clean up working directory "
    #rm -fv working/correction_*.dat
    rm -fv working/reaction.dat
    rm -fv working/run_Summary.dat
    rm -fv working/example.*
    rm -fv working/RunTimeStamp.dat
    rm -fv working/*.root
    rm -fv working/*.d
    rm -fv working/*.so
    rm -fv working/*.pcm

    echo "======== git commit "
    git add -A
    git commit -m "New experiment ${expName}"

    echo -e "${YELLOW} 5) ################## git push ${NC}"
    git push origin ${expName}

  fi 

fi