prepare_env.sh

#!/bin/bash

EDSR_DIR=$PWD

# when script is run from the enhanced-edsr project directory (as expected), let's create the python virtual environment in its parent directory, above the enhanced-edsr folder
ENV_PARENT_DIR_BASENAME=`basename $EDSR_DIR`
if [[ "$ENV_PARENT_DIR_BASENAME" != *"enhanced-edsr"* ]]; then
    echo "Please run this script after changing directory into this enhanced-edsr folder"
    exit -1
fi

ENV_PARENT_DIR=`dirname $EDSR_DIR`
ENV_DIR=$ENV_PARENT_DIR/venv_edsr

if [[ ! -d  $ENV_DIR ]]; then
    echo "Creating python virtual environment: ${ENV_DIR}"
    if [[ -z `which python3.12` ]]; then
        echo "ERROR: python3 is not found. please install python then rerun this script"
        exit -1
    fi
    if [[ -z `which pip3.12` ]]; then
	echo "ERROR: pip3 is not found. please install pip3 then rerun this script"
	exit -1
    fi
    python3.12 -m pip install --upgrade pip
    python3.12 -m venv $ENV_DIR
    if [ $? -ne 0 ]; then
        echo "ERROR: python virtualenv module did not succeed. please install virtualenv then rerun this script"
        exit -1
    fi
fi

if [[ ! -d $ENV_DIR ]]; then
    echo "ERROR: Creating python virtual environment was unsuccessful"
    echo "exiting"
    exit -1
fi

echo "Activating virtual environment: $ENV_DIR"
source $ENV_DIR/bin/activate

echo "Installing BasicSR requirements"
pushd $EDSR_DIR/BasicSR
pip3.12 install -r requirements.txt
if [ $? -ne 0 ]; then
    echo "ERROR: installing BasicSR dependencies was unsuccessful. please fix and then rerun this script"
    exit -1
fi
popd

FFMPEG_BIN=`which ffmpeg`
if [ $? -ne 0 ]; then
    echo "Could not find ffmpeg binary. Pulling ffmpeg 8.0 from github source, and attempting to build it"
    git clone https://github.com/FFmpeg/FFmpeg.git
    pushd FFmpeg
    git checkout -b n.8.0 tags/n8.0
    ./configure
    make -j
    sudo make install
    popd
fi

echo ""
echo ""
echo "Creation of virtual environment for EDSR succeeded."
echo "Run \"source $ENV_DIR/bin/activate\" in your environment before working with EDSR"
FFMPEG_BIN=`which ffmpeg`
if [ $? -ne 0 ]; then
    echo "Could not find ffmpeg binary. Please consider installing ffmpeg if using the make_dataset.sh script for dataset preparation."
fi

echo "Exiting with SUCCESS"