install.ffmpeg.sh

#!/bin/bash
#
# Copyright (c) 2015 - 2020 imm studios, z.s.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
##############################################################################
## COMMON UTILS

base_dir=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
temp_dir=/tmp/$(basename "${BASH_SOURCE[0]}")

function error_exit {
    printf "\n\033[0;31mInstallation failed\033[0m\n"
    cd ${base_dir}
    exit 1
}

function finished {
    printf "\n\033[0;92mInstallation completed\033[0m\n"
    cd ${base_dir}
    exit 0
}

if [ "$(id -u)" != "0" ]; then
   echo "This script must be run as root" 1>&2
   error_exit
fi

if [ ! -d ${temp_dir} ]; then
    mkdir ${temp_dir} || error_exit
fi

## COMMON UTILS
##############################################################################

make_threads=8

NASM_VERSION="2.14.02"

REPOS=(
    "https://github.com/mstorsjo/fdk-aac"
    "https://github.com/mirror/x264"
    "https://github.com/Haivision/srt"
    "https://github.com/stoth68000/libklvanc"
    "https://github.com/martastain/bmd-sdk"
    "https://github.com/immstudios/ffmpeg"
    "https://github.com/immstudios/ndi-headers"
)

extra_flags=""

if [ -z "$PREFIX" ]; then
    PREFIX="/usr/local"
fi

function install_prerequisites {
    # bulid tools
    apt-get -y install\
        build-essential \
        unzip \
        cmake \
        yasm \
        git \
        libtool \
        autoconf \
        automake \
        pkg-config \
        libxml2-dev \
        tclsh \
        wget \
        || exit 1

    # network and security
    apt-get -y install \
        avahi-daemon \
        avahi-discover \
        avahi-utils \
        libssl-dev \
        || exit 1

    # text rendering
    apt-get -y install \
        fontconfig \
        libfontconfig1 \
        libfontconfig1-dev \
        libfribidi-dev \
        libfribidi0 \
        libfreetype6-dev \
        libass-dev \
        || exit 1

    # 3rd party codecs and filters
    apt-get -y install \
        libx265-dev \
        libmp3lame-dev \
        libtwolame-dev \
        libopus-dev \
        libv4l-dev \
        libvpx-dev \
        libwebp-dev \
        libzvbi-dev \
        librubberband-dev \
        libfftw3-dev \
        ocl-icd-opencl-dev \
        opencl-headers \
        || exit 1
}

function download_repos {
    cd ${temp_dir}
    for i in ${REPOS[@]}; do
        MNAME=`basename $i`
        if [ -d $MNAME ]; then
            cd $MNAME
            git checkout master || return 1
            git pull || return 1
            cd ..
        else
            git clone $i || return 1
        fi
    done
    return 0
}

function install_nasm {
    cd $temp_dir
    nasm_name=nasm-${NASM_VERSION}
    if [ -f ${nasm_name}.tar.gz ]; then
        rm ${nasm_name}.tar.gz
    fi
    wget https://www.nasm.us/pub/nasm/releasebuilds/${NASM_VERSION}/${nasm_name}.tar.gz || return 1
    tar -xf ${nasm_name}.tar.gz
    cd ${nasm_name}
    ./configure || return 1
    make -j$make_threads || return 1
    make install || return 1
    return 0
}

#
# 3rd party libraries
#

function install_fdk_aac {
    cd ${temp_dir}/fdk-aac
    autoreconf -fiv || return 1
    ./configure --prefix=$PREFIX || return 1
    make -j$make_threads || return 1
    make install || return 1
    extra_flags="$extra_flags --enable-libfdk-aac"
    return 0
}


function install_nvcodec {
    HAS_NVIDIA=`hash nvidia-smi 2> /dev/null && echo 1 || echo ""`

    if [ $HAS_NVIDIA ] ; then
        cd $temp_dir
        if [ -d nv-codec-headers ]; then
            rm -rf nv-codec-headers
        fi
        git clone "https://git.videolan.org/git/ffmpeg/nv-codec-headers"
        cd nv-codec-headers
        make || return 1
        make install || return 1
        extra_flags="$extra_flags --enable-nvenc --enable-cuda --enable-cuvid --enable-libnpp --enable-cuda-nvcc"
        extra_flags="$extra_flags --extra-cflags=-I/usr/local/cuda/include --extra-ldflags=-L/usr/local/cuda/lib64"
    fi
    return 0
}

function install_bmd {
    cd ${temp_dir}
    cp bmd-sdk/* /usr/include/ || return 1
    extra_flags="$extra_flags --enable-decklink"
    return 0
}

function install_ndi {
    cd ${temp_dir}
    cp ndi-headers/*.h /usr/include
    ndi_file="libndi4_4.5.1-1_amd64.deb"
    if [ ! -f $ndi_file ]; then
        wget https://github.com/Palakis/obs-ndi/releases/download/4.9.1/$ndi_file || return 1
    fi
    dpkg -i $ndi_file || return 1
    extra_flags="$extra_flags --enable-libndi_newtek"
    return 0
}

function install_x264 {
    cd $temp_dir/x264
    ./configure --enable-shared --bit-depth=all --chroma-format=all || return 1
    make -j$make_threads || return 1
    make install || return 1
    extra_flags="$extra_flags --enable-libx264"
    return 0
}

function install_libsrt {
    cd $temp_dir/srt
    ./configure || return 1
    make -j$make_threads || return 1
    make install || return 1
    extra_flags="$extra_flags --enable-libsrt"
    return 0
}

function install_libklvanc {
    cd $temp_dir/libklvanc
    ./autogen.sh --build || return 1
    ./configure || return 1
    make -j$make_threads || return 1
    make install || return 1
    extra_flags="$extra_flags --enable-libklvanc"
    return 0
}

function install_openjpeg {
    cd $temp_dir
    openjpeg_src_file="v2.4.0.tar.gz"
    openjpeg_file="openjpeg-2.4.0"
    if [ ! -f $openjpeg_file.tar.gz ]; then
        wget https://github.com/uclouvain/openjpeg/archive/refs/tags/v2.4.0.tar.gz -O $openjpeg_file.tar.gz || exit 1
    fi
    if [ ! -d $openjpeg_file ]; then
        tar -xf $openjpeg_file.tar.gz
    fi
    cd $openjpeg_file
    mkdir build
    cd build
    cmake .. -DCMAKE_BUILD_TYPE=Release
    make
    make install
    extra_flags="$extra_flags --enable-libopenjpeg"
}

#
# Install FFmpeg
#

function install_ffmpeg {
    cd ${temp_dir}/ffmpeg

    PATH=/usr/local/cuda/bin${PATH:+:${PATH}}

    echo ""
    echo $extra_flags
    echo ""


    ./configure --prefix=$PREFIX \
      --enable-nonfree \
      --enable-gpl \
      --enable-version3 \
      --enable-shared \
      --enable-pic \
    \
    --enable-fontconfig      ` # enable fontconfig, useful for drawtext filter` \
    --enable-libass          ` # enable libass subtitles rendering` \
    --enable-libfreetype     ` # enable libfreetype, needed for drawtext filter` \
    --enable-libfribidi      ` # enable libfribidi, improves drawtext filter` \
    --enable-libmp3lame      ` # enable MP3 encoding via libmp3lame` \
    --enable-libtwolame      ` # enable MP2 encoding via libtwolame` \
    --enable-libwebp         ` # enable WebP encoding via libwebp` \
    --enable-libx265         ` # enable HEVC encoding via x265` \
    --enable-libopus         ` # enable Opus de/encoding via libopus` \
    --enable-libzvbi         ` # enable teletext support via libzvbi` \
    --enable-libv4l2         ` # enable libv4l2/v4l-utils` \
    --enable-openssl         ` # needed for https support if gnutls is not used` \
    --enable-libxml2         ` # enable XML parsing needed for dash demuxing support` \
    --enable-libvpx          ` # enable VP8 and VP9 de/encoding via libvpx` \
    --enable-librubberband \
    --enable-opencl \
    $extra_flags \
    || return 1

    echo "Making ffmpeg"
    make -j$make_threads || return 1
    make install || return 1
    make clean
    ldconfig
    return 0
}


################################################

install_prerequisites || error_exit
download_repos || error_exit
install_nasm || error_exit
install_fdk_aac || error_exit
install_x264 || error_exit
install_nvcodec || error_exit
install_bmd || error_exit
install_ndi || error_exit
install_libsrt || error_exit
install_libklvanc || error_exit
install_openjpeg || error_exit
install_ffmpeg || error_exit

finished