#!/bin/bash
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$DIR"

#Simple dependency checker that will apt-get stuff if something is missing
# sudo apt-get install python3-venv python3-pip
SYSTEM_DEPENDENCIES="build-essential python3-venv python3-pip freeglut3-dev libglew-dev protobuf-compiler libprotobuf-dev libavdevice-dev"

for REQUIRED_PKG in $SYSTEM_DEPENDENCIES
do
PKG_OK=$(dpkg-query -W --showformat='${Status}\n' $REQUIRED_PKG|grep "install ok installed")
echo "Checking for $REQUIRED_PKG: $PKG_OK"
if [ "" = "$PKG_OK" ]; then

  echo "No $REQUIRED_PKG. Setting up $REQUIRED_PKG."

  #If this is uncommented then only packages that are missing will get prompted..
  #sudo apt-get --yes install $REQUIRED_PKG

  #if this is uncommented then if one package is missing then all missing packages are immediately installed..
  sudo apt-get --yes install $SYSTEM_DEPENDENCIES  
  break
fi
done
#------------------------------------------------------------------------------

HEADLESS=" --headless"
OPENCV=" "

if [[ $* == *--collab* ]]
then 
 echo "Collab mode!"
 echo "Performing python package installation without a virtual environment"
HEADLESS=" --headless"
OPENCV="opencv-python-headless"
rm -rf /content/sample_data/ #remove sample data
ln -s $DIR /content/mnet4
sudo apt-get --yes install build-essential freeglut3-dev libglew-dev #For some reason collab has some errors on previous check so just double-make sure we have these libraries

else 
 echo "Using a python pythonVirtualEnvironment environment"
 echo "Activate it using : source pythonVirtualEnvironment/bin/activate"
 python3 -m venv pythonVirtualEnvironment
 source pythonVirtualEnvironment/bin/activate
OPENCV="opencv-python"
HEADLESS=" "
fi



cd "$DIR"
if [ -f RGBDAcquisition/README.md ]; then
echo "RGBDAcquisition appears to already exist .."
else
 git clone https://github.com/AmmarkoV/RGBDAcquisition
 cd RGBDAcquisition
 #----------------------------------------------------------------------------------------------------------------------
 #mkdir build
 #cd build
 #cmake ..
 #cd ..
 #----------------------------------------------------------------------------------------------------------------------
 cd opengl_acquisition_shared_library/opengl_depth_and_color_renderer/
 mkdir build 
 cd build
 cmake ..
 make BVHTester
 #----------------------------------------------------------------------------------------------------------------------
 cd "$DIR"
 ln -s RGBDAcquisition/opengl_acquisition_shared_library/opengl_depth_and_color_renderer/src/Applications/BVHTester BVH
 ln -s RGBDAcquisition/opengl_acquisition_shared_library/opengl_depth_and_color_renderer/BVHTester GroundTruthDumper
fi


 
 python3 -m pip install --upgrade pip #2.8.0 nvidia-tensorrt
 python3 -m pip install tensorflow numpy numba tensorboard_plugin_profile tensorflow-model-optimization keras pillow tf2onnx onnxruntime onnx matplotlib pydot mediapipe $OPENCV scikit-learn

 
 #Two ways to get Models!
 #One through the database and compile them yourself (commented out to save Google Collab CPU time)..
 #python3 -m getModelFromDatabase --get 318
 #To generate the archive in the next lines do :
 #   zip mnet4_318.zip step1_*/*

 #Another just take the precompiled ensembles ready to run
 wget http://ammar.gr/mocapnet/mnet4/mnet4_-314.zip -O mnet4_-314.zip
 unzip -o mnet4_-314.zip #<- upgrade to 324 mnet4_318.zip

 #Unzip models for PoseNET.py operation..
 wget http://ammar.gr/mocapnet/mnet4/mediapipe.zip -O mediapipe.zip
 unzip -o mediapipe.zip

 #To get :
 # python3 -m evaluateMocapNET --config dataset/body_configuration.json --all body --engine onnx 
 #  to work

 #wget http://cvrlcode.ics.forth.gr/web_share/mocapnet/MocapNETBodyValidationSet.zip
 #wget http://ammar.gr/datasets/MocapNETBodyValidationSet.zip
 #unzip MocapNETBodyValidationSet.zip



 #Doing a test run!

# echo "To enable Tensorflow profiling for non-root users please add : "
# echo "options nvidia \"NVreg_RestrictProfilingToAdminUsers=0\""
# echo "to /etc/modprobe.d/nvidia-graphics-drivers.conf"
# echo " "
 echo "Ready for use"
 echo "From now on to work with MocapNET you can use :  "
 echo "source pythonVirtualEnvironment/bin/activate"
 echo "!python3 -m mediapipeHolisticWebcamMocapNET --from webcam  --save --plot"
 echo " OR "
 echo "!python3 -m mediapipeHolisticWebcamMocapNET --from shuffle.webm --save --plot"
 echo " OR "
 echo "!python3 -m csvNET --from ../../../dataset/sample.csv  --save --plot"
 echo "  "
 echo " if you are running on collab don't forget to append --headless to your command"
 echo " after the run is complete out.bvh will contain your results  "

#Force making of library during last-stage of setup..
BVH/makeLibrary.sh