From 8f877f45258976839fad76defb4c66ac50677a57 Mon Sep 17 00:00:00 2001
From: Christian Chapman-Bird <c.chapmanbird@gmail.com>
Date: Mon, 18 Nov 2024 00:01:40 +0000
Subject: [PATCH] re-add removed install.sh script

---
 install.sh | 158 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 158 insertions(+)
 create mode 100644 install.sh

diff --git a/install.sh b/install.sh
new file mode 100644
index 00000000..47e68d61
--- /dev/null
+++ b/install.sh
@@ -0,0 +1,158 @@
+#!/bin/bash
+
+usage="$(basename "$0") [-h] -- program to install FastEMRIWaveforms
+
+where:
+    -h  show this help text
+
+keyword argument options (given as key=value):
+    env_name:  Name of generated conda environment. Default is 'few_env'.
+    install_type:  Type of install. 'basic', 'development', or 'sampling'. 
+        'development' adds packages needed for development and documentation.
+        'sampling' adds packages for sampling like eryn, lisatools, corner, chainconsumer.
+        Default is 'basic'. 
+    run_tests: Either true or false. Whether to run tests after install. Default is true.
+    
+"
+
+while getopts 'h' option; do
+  case "$option" in
+    h) echo "$usage"
+       exit 3
+       ;;
+  esac
+done
+
+for ARGUMENT in "$@"
+do
+   KEY=$(echo $ARGUMENT | cut -f1 -d=)
+
+   KEY_LENGTH=${#KEY}
+   VALUE="${ARGUMENT:$KEY_LENGTH+1}"
+
+   export "$KEY"="$VALUE"
+done
+
+if [ -z ${env_name+x} ]; then env_name="few_env"; fi
+if [ -z ${install_type+x} ]; then install_type="basic"; fi
+if [ -z ${run_tests+x} ]; then run_tests="true"; fi
+
+if [[ "$run_tests" != "true" ]] && [[ "$run_tests" != "false" ]]; 
+    then echo "run_tests variable must be 'true' or 'false'.";
+    exit 2;
+fi
+
+nvcc=$(which nvcc)
+if [[ "$nvcc" == "" ]];
+        then nvcc="$CUDAHOME";
+        if [[ "$nvcc" != "" ]];
+                then nvcc="$nvcc/bin/nvcc";
+        fi
+fi
+
+if [[ "$nvcc" == "" ]];
+        then nvcc="$CUDA_HOME";
+        if [[ "$nvcc" != "" ]];
+                then nvcc=""$nvcc"bin/nvcc";
+        fi
+fi
+
+if [[ "$nvcc" == "" ]];
+        then use_gpu=false;
+else
+        echo "found nvcc: $nvcc";
+        use_gpu=true;
+fi
+echo "use_gpu: "$use_gpu"";
+
+if [[ "$use_gpu" == true ]];
+    then export CUDA_HOME="${nvcc:0:-8}";
+    tmp=$("$nvcc" --version | grep -i "Build cuda_");
+    cuda_version="${tmp:11:4}";
+    echo "CUDA_HOME: $CUDA_HOME";
+    echo "cuda version: $cuda_version";
+fi
+
+echo "Now installing into conda environment named $env_name."
+
+if [[ "$install_type" == "basic" ]]; 
+    then echo "Installing basic setup."; 
+elif [[ "$install_type" == "sampling" ]];
+    then echo "Installing sampling setup."; 
+elif [[ "$install_type" == "development" ]];
+    then echo "Installing development setup."; 
+else
+    echo "If providing install_type variable, must be 'basic', 'sampling', or 'development'";
+    exit 1;
+fi
+
+if [[ "$OSTYPE" == "darwin"* ]]; then
+    echo "This system is macOS."
+
+    machine=$(uname -m)
+    if [[ "$machine" == "arm64" ]]; then
+        echo "This is an M1 Mac."
+        conda create -n "$env_name" -c conda-forge -y wget gsl hdf5 numpy Cython scipy tqdm jupyter ipython h5py requests matplotlib python=3.12 openblas lapack liblapacke rich numba
+    else
+        echo "This is not an M1 Mac."
+        conda create -n "$env_name" -c conda-forge -y clangxx_osx-64 clang_osx-64 wget gsl lapack=3.6.1 hdf5 numpy Cython scipy tqdm jupyter ipython h5py requests matplotlib rich numba python=3.12
+    fi
+elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+    echo "This system is Unix/Linux."
+    conda create -n "$env_name" -c conda-forge -y gcc_linux-64 gxx_linux-64 wget gsl lapack=3.6.1 hdf5 rich numba numpy Cython scipy tqdm jupyter ipython h5py requests matplotlib python=3.12
+else
+    echo "Unsupported operating system."
+fi
+
+# get conda base information
+IN="$(conda info | grep -i 'base environment')"
+
+tmp=$(echo $IN | tr " " "\n")
+i=0
+out=()
+for tmp_i in $tmp
+do
+    out[i]=$tmp_i
+    let "i = $i + 1"
+done
+
+conda_base="${out[3]}"
+conda_init_sh="$conda_base/etc/profile.d/conda.sh"
+echo "$conda_init_sh"
+source $conda_init_sh
+
+conda activate "$env_name"
+echo "should be done"
+conda_check="$(conda info | grep -i 'active environment')"
+echo "conda env: $conda_check"
+echo "Environment created. Installing additional packages before FEW install."
+
+python_here=""$conda_base"/envs/"$env_name"/bin/python"
+pip_here=""$conda_base"/envs/"$env_name"/bin/pip"
+
+echo "python: $python_here" 
+echo "pip: $pip_here"
+if [[ "$install_type" == "sampling" ]]; then
+    "$pip_here" install corner eryn chainconsumer;
+    "$pip_here" install git+https://github.com/mikekatz04/LISAanalysistools.git@dev;
+elif [[ "$install_type" == "development" ]]; then
+    conda install sphinx sphinx_rtd_theme pypandoc --yes;
+    "$pip_here" install nbsphinx;
+fi
+
+if [[ "$use_gpu" == true ]];
+    then pip install "cupy-cuda"${cuda_version:0:2}""${cuda_version:3:1}"";
+fi
+
+machine=$(uname -m)
+
+if [[ "$machine" == "arm64" ]]; then
+    "$pip_here" install . --ccbin /usr/bin/
+else
+    "$pip_here" install .
+fi
+
+if [[ "$run_tests" == "true" ]]; 
+ then echo "Running tests...";
+ "$python_here" -m unittest discover;
+fi
\ No newline at end of file