build

#!/bin/bash

#Original Project: https://github.com/markus-perl/ffmpeg-build-script
#This fork of the project is to target Google Colab Platform for CUDA support

VERSION=1.16
CWD=$(pwd)
PACKAGES="$CWD/packages"
WORKSPACE="$CWD/workspace"
CC=clang
LDFLAGS="-L${WORKSPACE}/lib -lm"
CFLAGS="-I${WORKSPACE}/include"
ADDITIONAL_CONFIGURE_OPTIONS=""

# Speed up the process
# Env Var NUMJOBS overrides automatic detection
if [[ -n $NUMJOBS ]]; then
    MJOBS=$NUMJOBS
elif [[ -f /proc/cpuinfo ]]; then
    MJOBS=$(grep -c processor /proc/cpuinfo)
elif [[ "$OSTYPE" == "darwin"* ]]; then
	MJOBS=$(sysctl -n machdep.cpu.thread_count)
	ADDITIONAL_CONFIGURE_OPTIONS="--enable-videotoolbox"
else
    MJOBS=4
fi

make_dir () {
	if [ ! -d "$1" ]; then
		if ! mkdir "$1"; then
			printf "\n Failed to create dir %s" "$1";
			exit 1
		fi
	fi
}

remove_dir () {
	if [ -d "$1" ]; then
		rm -r "$1"
	fi
}

download () {

	DOWNLOAD_PATH=$PACKAGES;

	if [ -n "$3" ]; then
		mkdir -p "$PACKAGES"/"$3"
		DOWNLOAD_PATH=$PACKAGES/$3
	fi;

	if [ ! -f "$DOWNLOAD_PATH/$2" ]; then

		echo "Downloading $1"
		curl -L --silent -o "$DOWNLOAD_PATH/$2" "$1"

		EXITCODE=$?
		if [ $EXITCODE -ne 0 ]; then
			echo ""
			echo "Failed to download $1. Exitcode $EXITCODE. Retrying in 10 seconds";
			sleep 10
			curl -L --silent -o "$DOWNLOAD_PATH/$2" "$1"
		fi

		EXITCODE=$?
		if [ $EXITCODE -ne 0 ]; then
			echo ""
			echo "Failed to download $1. Exitcode $EXITCODE";
			exit 1
		fi

		echo "... Done"

		if ! tar -xvf "$DOWNLOAD_PATH/$2" -C "$DOWNLOAD_PATH" 2>/dev/null >/dev/null; then
			echo "Failed to extract $2";
			exit 1
		fi

	fi
}

execute () {
	echo "$ $*"

	OUTPUT=$("$@" 2>&1)

	# shellcheck disable=SC2181
	if [ $? -ne 0 ]; then
        echo "$OUTPUT"
        echo ""
        echo "Failed to Execute $*" >&2
        exit 1
    fi
}

build () {
	echo ""
	echo "building $1"
	echo "======================="

	if [ -f "$PACKAGES/$1.done" ]; then
		echo "$1 already built. Remove $PACKAGES/$1.done lockfile to rebuild it."
		return 1
	fi

	return 0
}

command_exists() {
    if ! [[ -x $(command -v "$1") ]]; then
        return 1
    fi

    return 0
}


build_done () {
    touch "$PACKAGES/$1.done"
}

echo "ffmpeg-build-script v$VERSION"
echo "========================="
echo ""

case "$1" in
"--cleanup")
	remove_dir "$PACKAGES"
	remove_dir "$WORKSPACE"
	echo "Cleanup done."
	echo ""
	exit 0
    ;;
"--build")

    ;;
*)
    echo "Usage: $0"
    echo "   --build: start building process"
    echo "   --cleanup: remove all working dirs"
    echo "   --help: show this help"
    echo ""
    exit 0
    ;;
esac

echo "Using $MJOBS make jobs simultaneously."

make_dir "$PACKAGES"
make_dir "$WORKSPACE"

export PATH=${WORKSPACE}/bin:$PATH

if ! command_exists "make"; then
    echo "make not installed.";
    exit 1
fi

if ! command_exists "g++"; then
    echo "g++ not installed.";
    exit 1
fi

if ! command_exists "curl"; then
    echo "curl not installed.";
    exit 1
fi

if build "gperf-3.1"; then
	download "http://ftp.gnu.org/pub/gnu/gperf/gperf-3.1.tar.gz" "gperf-3.1.tar.gz"
	cd "$PACKAGES"/gperf-3.1 || exit
	execute ./configure --prefix="${WORKSPACE}"
	execute make -j $MJOBS
	execute make install
	build_done "gperf-3.1"
fi

if build "yasm"; then
	download "https://www.tortall.net/projects/yasm/releases/yasm-1.3.0.tar.gz" "yasm-1.3.0.tar.gz"
	cd "$PACKAGES"/yasm-1.3.0 || exit
	execute ./configure --prefix="${WORKSPACE}"
	execute make -j $MJOBS
	execute make install
	build_done "yasm"
fi

if build "nasm"; then
	download "https://www.nasm.us/pub/nasm/releasebuilds/2.15.05/nasm-2.15.05.tar.xz" "nasm.tar.gz"
	cd "$PACKAGES"/nasm-2.15.05 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static
	execute make -j $MJOBS
	execute make install
	build_done "nasm"
fi

if build "opencore"; then
	download "https://sonik.dl.sourceforge.net/project/opencore-amr/opencore-amr/opencore-amr-0.1.5.tar.gz" "opencore-amr-0.1.5.tar.gz"
	cd "$PACKAGES"/opencore-amr-0.1.5 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static
	execute make -j $MJOBS
	execute make install
	build_done "opencore"
fi

if build "libvpx"; then
    download "https://github.com/webmproject/libvpx/archive/refs/tags/v1.10.0.tar.gz" "libvpx-1.10.0.tar.gz"
    cd "$PACKAGES"/libvpx-1.10.0 || exit

    if [[ "$OSTYPE" == "darwin"* ]]; then
        echo "Applying Darwin patch"
        sed "s/,--version-script//g" build/make/Makefile > build/make/Makefile.patched
        sed "s/-Wl,--no-undefined -Wl,-soname/-Wl,-undefined,error -Wl,-install_name/g" build/make/Makefile.patched > build/make/Makefile
    fi

	execute ./configure --prefix="${WORKSPACE}" --disable-unit-tests --disable-shared --as=yasm --enable-vp9-highbitdepth
	execute make -j $MJOBS
	execute make install
	build_done "libvpx"
fi

if build "lame"; then
	download "https://netcologne.dl.sourceforge.net/project/lame/lame/3.100/lame-3.100.tar.gz" "lame-3.100.tar.gz"
	cd "$PACKAGES"/lame-3.100 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static
	execute make -j $MJOBS
	execute make install
	build_done "lame"
fi

if build "opus"; then
	download "https://archive.mozilla.org/pub/opus/opus-1.3.1.tar.gz" "opus-1.3.1.tar.gz"
	cd "$PACKAGES"/opus-1.3.1 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static
	execute make -j $MJOBS
	execute make install
	build_done "opus"
fi

if build "xvidcore"; then
	download "https://downloads.xvid.com/downloads/xvidcore-1.3.7.tar.gz" "xvidcore-1.3.7.tar.gz"
	cd "$PACKAGES"/xvidcore  || exit
	cd build/generic  || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static
	execute make -j $MJOBS
	execute make install

	if [[ -f ${WORKSPACE}/lib/libxvidcore.4.dylib ]]; then
	    execute rm "${WORKSPACE}/lib/libxvidcore.4.dylib"
	fi
	
    if [[ -f ${WORKSPACE}/lib/libxvidcore.so ]]; then
      execute rm "${WORKSPACE}"/lib/libxvidcore.so*
    fi

	build_done "xvidcore"
fi

if build "x264"; then
	download "https://code.videolan.org/videolan/x264/-/archive/5db6aa6cab1b146e07b60cc1736a01f21da01154/x264-5db6aa6cab1b146e07b60cc1736a01f21da01154.tar.gz" "x264-5db6aa6.tar.gz"
	cd "$PACKAGES"/x264-5db6aa6cab1b146e07b60cc1736a01f21da01154 || exit

	if [[ "$OSTYPE" == "linux-gnu" ]]; then
		execute ./configure --prefix="${WORKSPACE}" --enable-static --enable-pic CXXFLAGS="-fPIC"
    else
        execute ./configure --prefix="${WORKSPACE}" --enable-static --enable-pic
    fi

    execute make -j $MJOBS
	execute make install
	execute make install-lib-static
	build_done "x264"
fi

if build "libogg"; then
	download "https://ftp.osuosl.org/pub/xiph/releases/ogg/libogg-1.3.3.tar.gz" "libogg-1.3.3.tar.gz"
	cd "$PACKAGES"/libogg-1.3.3 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static
	execute make -j $MJOBS
	execute make install
	build_done "libogg"
fi

if build "libvorbis"; then
	download "https://ftp.osuosl.org/pub/xiph/releases/vorbis/libvorbis-1.3.6.tar.gz" "libvorbis-1.3.6.tar.gz"
	cd "$PACKAGES"/libvorbis-1.3.6 || exit
	execute ./configure --prefix="${WORKSPACE}" --with-ogg-libraries="${WORKSPACE}"/lib --with-ogg-includes="${WORKSPACE}"/include/ --enable-static --disable-shared --disable-oggtest
	execute make -j $MJOBS
	execute make install
	build_done "libvorbis"
fi

if build "libtheora"; then
	download "https://ftp.osuosl.org/pub/xiph/releases/theora/libtheora-1.1.1.tar.gz" "libtheora-1.1.1.tar.bz"
	cd "$PACKAGES"/libtheora-1.1.1 || exit
	sed "s/-fforce-addr//g" configure > configure.patched
	chmod +x configure.patched
	mv configure.patched configure
	execute ./configure --prefix="${WORKSPACE}" --with-ogg-libraries="${WORKSPACE}"/lib --with-ogg-includes="${WORKSPACE}"/include/ --with-vorbis-libraries="${WORKSPACE}"/lib --with-vorbis-includes="${WORKSPACE}"/include/ --enable-static --disable-shared --disable-oggtest --disable-vorbistest --disable-examples --disable-asm --disable-spec
	execute make -j $MJOBS
	execute make install
	build_done "libtheora"
fi

if build "pkg-config"; then
	download "https://pkgconfig.freedesktop.org/releases/pkg-config-0.29.2.tar.gz" "pkg-config-0.29.2.tar.gz"
	cd "$PACKAGES"/pkg-config-0.29.2 || exit
	execute ./configure --silent --prefix="${WORKSPACE}" --with-pc-path="${WORKSPACE}"/lib/pkgconfig --with-internal-glib
	execute make -j $MJOBS
	execute make install
	build_done "pkg-config"
fi

if build "cmake"; then
	download "https://cmake.org/files/v3.15/cmake-3.15.4.tar.gz" "cmake-3.15.4.tar.gz"
	cd "$PACKAGES"/cmake-3.15.4  || exit
	rm Modules/FindJava.cmake
	perl -p -i -e "s/get_filename_component.JNIPATH/#get_filename_component(JNIPATH/g" Tests/CMakeLists.txt
	perl -p -i -e "s/get_filename_component.JNIPATH/#get_filename_component(JNIPATH/g" Tests/CMakeLists.txt
	execute ./configure --prefix="${WORKSPACE}"
	execute make -j $MJOBS
	execute make install
	build_done "cmake"
fi

if build "vid_stab"; then
	download "https://github.com/georgmartius/vid.stab/archive/v1.1.0.tar.gz" "georgmartius-vid.stab-v1.1.0-0-g60d65da.tar.tgz"
	cd "$PACKAGES"/vid.stab-1.1.0 || exit
	execute cmake -DBUILD_SHARED_LIBS=OFF -DCMAKE_INSTALL_PREFIX:PATH="${WORKSPACE}" -DUSE_OMP=OFF -DENABLE_SHARED:bool=off .
	execute make
	execute make install
	build_done "vid_stab"
fi

if build "expat-2.4.8"; then
	download "https://sonik.dl.sourceforge.net/project/expat/expat/2.4.8/expat-2.4.8.tar.xz" "expat-2.4.8.tar.xz"
	cd "$PACKAGES"/expat-2.4.8 || exit
	execute ./configure --prefix="${WORKSPACE}" --enable-shared=no --without-docbook
	execute make -j $MJOBS
	execute make install
	build_done "expat-2.4.8"
fi

if build "freetype2"; then
	download "https://sonik.dl.sourceforge.net/project/freetype/freetype2/2.12.1/freetype-2.12.1.tar.xz" "freetype-2.12.1.tar.gz"
	cd "$PACKAGES"/freetype-2.12.1 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static --with-png=no --with-harfbuzz=no
	execute make -j $MJOBS
	execute make install
	build_done "freetype2"
fi

if build "fontconfig-2.13.96"; then
	download "https://www.freedesktop.org/software/fontconfig/release/fontconfig-2.14.0.tar.gz" "fontconfig-2.14.0.tar.gz"
	cd "$PACKAGES"/fontconfig-2.14.0 || exit
	execute ./configure --prefix="${WORKSPACE}" --enable-shared=no
	execute make -j $MJOBS
	execute make install
	build_done "fontconfig-2.14.0"
fi

if build "harfbuzz"; then
	download "https://github.com/harfbuzz/harfbuzz/releases/download/5.1.0/harfbuzz-5.1.0.tar.xz" "harfbuzz-5.1.0.tar.xz"
	cd "$PACKAGES"/harfbuzz-5.1.0 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static --with-freetype=yes
	execute make -j $MJOBS
	execute make install
	build_done "harfbuzz"
fi

if build "fribidi"; then
	download "https://github.com/fribidi/fribidi/releases/download/v1.0.12/fribidi-1.0.12.tar.xz" "fribidi-1.0.12.tar.xz"
	cd "$PACKAGES"/fribidi-1.0.12 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static
	execute make -j $MJOBS
	execute make install
	build_done "fribidi"
fi

if build "x265"; then
	download "https://github.com/videolan/x265/archive/3.4.tar.gz" "x265-3.4.tar.gz"
	cd "$PACKAGES"/x265-*/ || exit
	cd source || exit
	execute cmake -DCMAKE_INSTALL_PREFIX:PATH="${WORKSPACE}" -DENABLE_SHARED:bool=off .
	execute make -j $MJOBS
	execute make install
	sed "s/-lx265/-lx265 -lstdc++/g" "$WORKSPACE/lib/pkgconfig/x265.pc" > "$WORKSPACE/lib/pkgconfig/x265.pc.tmp"
	mv "$WORKSPACE/lib/pkgconfig/x265.pc.tmp" "$WORKSPACE/lib/pkgconfig/x265.pc"
	build_done "x265"
fi

if build "fdk_aac"; then
	download "https://sourceforge.net/projects/opencore-amr/files/fdk-aac/fdk-aac-2.0.2.tar.gz/download?use_mirror=gigenet" "fdk-aac-2.0.2.tar.gz"
	cd "$PACKAGES"/fdk-aac-2.0.2 || exit
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --enable-static --enable-pic
	execute make -j $MJOBS
	execute make install
	build_done "fdk_aac"
fi

if build "av1"; then
	download "https://aomedia.googlesource.com/aom/+archive/ae2be8030200925895fa6e98bd274ffdb595cbf6.tar.gz" "av1.tar.gz" "av1"
	cd "$PACKAGES"/av1 || exit
	mkdir -p "$PACKAGES"/aom_build
	cd "$PACKAGES"/aom_build || exit
	execute cmake -DENABLE_TESTS=0 -DCMAKE_INSTALL_PREFIX:PATH="${WORKSPACE}" "$PACKAGES"/av1
	execute make -j $MJOBS
	execute make install
	build_done "av1"
fi

if build "zlib"; then
	download "https://www.zlib.net/zlib-1.2.12.tar.gz" "zlib-1.2.12.tar.gz"
	cd "$PACKAGES"/zlib-1.2.12 || exit
	execute ./configure --static --prefix="${WORKSPACE}"
	execute make -j $MJOBS
	execute make install
	build_done "zlib"
fi

if build "openssl"; then
	download "https://www.openssl.org/source/openssl-1.1.1q.tar.gz" "openssl-1.1.1q.tar.gz"
	cd "$PACKAGES"/openssl-1.1.1q || exit
	execute ./config --prefix="${WORKSPACE}" --openssldir="${WORKSPACE}" --with-zlib-include="${WORKSPACE}"/include/ --with-zlib-lib="${WORKSPACE}"/lib no-shared zlib
	execute make -j $MJOBS
	execute make install
	build_done "openssl"
fi

if build "srt"; then
	download "https://github.com/Haivision/srt/archive/refs/tags/v1.5.1.tar.gz" "srt-1.5.1.tar.gz"
	cd "$PACKAGES"/srt-1.5.1 || exit
	export OPENSSL_ROOT_DIR="${WORKSPACE}"
	export OPENSSL_LIB_DIR="${WORKSPACE}"/lib
	export OPENSSL_INCLUDE_DIR="${WORKSPACE}"/include/
	execute cmake "$PACKAGES"/srt-1.5.1 -DCMAKE_INSTALL_PREFIX="${WORKSPACE}" -DCMAKE_INSTALL_LIBDIR=lib -DCMAKE_INSTALL_BINDIR=bin -DCMAKE_INSTALL_INCLUDEDIR=include -DENABLE_SHARED=OFF -DENABLE_STATIC=ON -DENABLE_APPS=OFF -DUSE_STATIC_LIBSTDCXX=ON
	execute make install
	build_done "srt"
fi

if build "libass"; then
	download "https://github.com/libass/libass/releases/download/0.17.1/libass-0.17.1.tar.gz" "libass-0.17.1.tar.gz"
	cd "$PACKAGES"/libass-0.17.1 || exit 
	execute ./configure --prefix="${WORKSPACE}" --disable-shared --disable-require-system-font-provider
	execute make -j $MJOBS
	execute make install
	build_done "libass"
fi

CFLAGS="-I$WORKSPACE/include"
LDFLAGS="-L$WORKSPACE/lib"
if command -v nvcc > /dev/null ; then
       if build "nv-codec"; then
               download "https://github.com/FFmpeg/nv-codec-headers/releases/download/n9.1.23.3/nv-codec-headers-9.1.23.3.tar.gz" "nv-codec-headers-9.1.23.3.tar.gz"
               cd "$PACKAGES"/nv-codec-headers-9.1.23.3 || exit
               sed -i  "s#PREFIX = /usr/local#PREFIX = ${WORKSPACE}#g" "$PACKAGES"/nv-codec-headers-9.1.23.3/Makefile
               execute make install
               build_done "nv-codec"
       fi
       CFLAGS="$CFLAGS -I/usr/local/cuda/include"
       LDFLAGS="$LDFLAGS -L/usr/local/cuda/lib64"
       ADDITIONAL_CONFIGURE_OPTIONS="$ADDITIONAL_CONFIGURE_OPTIONS --enable-cuda-nvcc --enable-cuvid --enable-nvenc --enable-libnpp  --enable-cuda-llvm"
fi

build "ffmpeg"
download "http://ffmpeg.org/releases/ffmpeg-6.0.tar.bz2" "ffmpeg-6.0.tar.bz2"
cd "$PACKAGES"/ffmpeg-6.0/ || exit
# shellcheck disable=SC2086
./configure $ADDITIONAL_CONFIGURE_OPTIONS \
    --pkgconfigdir="$WORKSPACE/lib/pkgconfig" \
    --prefix="${WORKSPACE}" \
    --pkg-config-flags="--static" \
    --extra-cflags="$CFLAGS" \
    --extra-ldflags="$LDFLAGS" \
    --extra-libs="-lpthread -lm" \
	--enable-static \
	--disable-debug \
	--disable-shared \
	--disable-ffplay \
	--disable-doc \
	--enable-openssl \
	--enable-gpl \
	--enable-version3 \
	--enable-nonfree \
	--enable-pthreads \
	--enable-libvpx \
	--enable-libmp3lame \
    --enable-libopus \
	--enable-libtheora \
	--enable-libvorbis \
	--enable-libx264 \
	--enable-libx265 \
	--enable-runtime-cpudetect \
	--enable-libfdk-aac \
	--enable-avfilter \
	--enable-libopencore_amrwb \
	--enable-libopencore_amrnb \
	--enable-filters \
	--enable-libvidstab \
	--enable-libaom \
	--enable-libsrt \
	--enable-libass \
	--enable-libfreetype \
	--enable-libfribidi \
	--enable-libfontconfig \
    --nvccflags="-gencode arch=compute_75,code=sm_75 -O2"

execute make -j $MJOBS
execute make install

INSTALL_FOLDER="/usr/bin"
if [[ "$OSTYPE" == "darwin"* ]]; then
INSTALL_FOLDER="/usr/local/bin"
fi

echo ""
echo "Building done. The binary can be found here: $WORKSPACE/bin/ffmpeg"
echo ""

if command_exists "sudo"; then
	sudo cp "$WORKSPACE/bin/ffmpeg" "$INSTALL_FOLDER/ffmpeg"
	sudo cp "$WORKSPACE/bin/ffprobe" "$INSTALL_FOLDER/ffprobe"
else
	cp "$WORKSPACE/bin/ffmpeg" "$INSTALL_FOLDER/ffmpeg"
	cp "$WORKSPACE/bin/ffprobe" "$INSTALL_FOLDER/ffprobe"
fi

echo "ffmpeg should be installed to this instance"

exit 0