# Maintainer: Dmitry Kharitonov <darksab0r at gmail com>
# Contributor: Daniel Bermond < yahoo-com: danielbermond >
# Contributor: rcpoison <rc dot poison at gmail dot com>
# Contributor: Gerad Munsch <gmunsch@unforgivendevelopment.com>
# Contributor: Rudolf Polzer <divVerent[at]xonotic[dot]org>
# Contributor: Ionut Biru <ibiru@archlinux.org>
# Contributor: Tom Newsom <Jeepster@gmx.co.uk>
# Contributor: Paul Mattal <paul@archlinux.org>

# If you don't want CUDA/CUVID support, you can remove
# depends_x86_64=('cuda') line and 
# $_cuda \, $_cuvid \, $_libnpp \ lines from PKGBUILD
#
# Replace "--disable-libsrt" with "--enable-libsrt"
# and add 'srt' to depends array if you need the support
# for Haivision Open SRT (Secure Reliable Transport).
# Enabling it is known to cause segfaults with OBS, 
# that's why it's disabled by default.
# 
# Add "--enable-decklink \" to configure flags
# if you have decklink-sdk installed
# Add "--enable-libndi_newtek \" to configure flags
# if you have ndi-sdk installed

pkgname=ffmpeg-full-nvenc
_pkgbasename=ffmpeg
pkgver=4.2.2
pkgrel=1
epoch=1
pkgdesc="Record, convert, and stream audio and video (all codecs including Nvidia NVENC)"
arch=('i686' 'x86_64')
url="http://ffmpeg.org/"
license=('custom: nonfree and unredistributable')
depends=('alsa-lib' 'aom' 'aribb24' 'bzip2' 'celt' 'chromaprint-fftw' 'codec2'  
	       'dav1d' 'davs2' 'fontconfig' 'ffnvcodec-headers' 'frei0r-plugins' 
         'fribidi' 'gsm' 'jack' 'ladspa' 'lame' 'lensfun' 'libass' 'libavc1394'
         'libbluray' 'libbs2b' 'libcaca' 'libcdio-paranoia' 'libdc1394'
         'libfdk-aac' 'libgme' 'libiec61883' 'libilbc' 'libmodplug' 'libomxil-bellagio'
         'libmysofa' 'libpulse' 'librsvg' 'libsoxr' 'libssh' 'libtheora'
         'libva' 'libvdpau' 'libxcb' 'libxext' 'libwebp' 'libxml2' 'libxv' 'lilv'
         'libgl' 'lv2' 'openal' 'opencore-amr' 'opencl-driver' 'opencl-icd-loader'
         'openh264' 'openjpeg2' 'libklvanc-git' 'libopenmpt-svn' 'opus' 'pocketsphinx'
         'rockchip-mpp' 'rubberband' 'rtmpdump' 'sdl2' 'speex' 'shine' 'tensorflow' 
         'tesseract' 'twolame' 'v4l-utils' 'vapoursynth' 'vid.stab' 'vmaf'
         'vo-amrwbenc' 'xavs2' 'xvidcore' 'xz' 'wavpack' 'zeromq' 'zimg' 'zlib'
         'zvbi' 'libvorbisenc.so' 'libvorbis.so' 'libvpx.so' 'x264' 'x265'
         'snappy' 'sndio' 'xavs')
depends_x86_64=('cuda')
makedepends=('flite' 'ffnvcodec-headers' 'libmfx' 'libvdpau' 'nasm' 'opencl-headers')
optdepends=('avxsynth-git: for Avisynth support'
            'decklink-sdk: for Blackmagic DeckLink support; need to add --enable-decklink option in this PKGBUILD')
optdepends_x86_64=('intel-media-sdk: for Intel QSV support (Experimental! See PKGBUILD of that package for additional info)')
conflicts=('ffmpeg')
provides=('libavcodec.so' 'libavdevice.so' 'libavfilter.so' 'libavformat.so'
          'libavutil.so' 'libpostproc.so' 'libswresample.so'
          'libswscale.so' 'ffmpeg')
source=(https://ffmpeg.org/releases/$_pkgbasename-$pkgver.tar.xz{,.asc}
        'UNREDISTRIBUTABLE.txt')
validpgpkeys=('FCF986EA15E6E293A5644F10B4322F04D67658D8')
sha256sums=('cb754255ab0ee2ea5f66f8850e1bd6ad5cac1cd855d0a2f4990fb8c668b0d29c'
            'SKIP'
            'e0c1b126862072a71e18b9580a6b01afc76a54aa6e642d2c413ba0ac9d3010c4')

build() {
  cd "$_pkgbasename-$pkgver"

  # Add x86_64 (opt)depends to the build
  if [ "$CARCH" = "x86_64" ]
  then
      local _libvmaf='--enable-libvmaf'
      local _cudasdk='--enable-cuda-nvcc'
      local _libmfx='--enable-libmfx'
      local _libnpp='--enable-libnpp'
      local _cflags='-I/opt/cuda/include'      
      # '-L/usr/lib/nvidia' (for cuda_sdk) needs to be enabled only on
      # systems with nvidia-340xx-utils or nvidia-304xx-utils
      if pacman -Qqs '^nvidia-340xx-utils$' | grep -q '^nvidia-340xx-utils$' ||
         pacman -Qqs '^nvidia-304xx-utils$' | grep -q '^nvidia-304xx-utils$'
      then
          local _nvidia_340xx_ldflags='-L/usr/lib/nvidia'
      fi
      
      # fix tensorflow include dir
      _cflags+=' -I/usr/include/tensorflow'

      local _ldflags="-L/opt/cuda/lib64 ${_nvidia_340xx_ldflags}"
      local _ldflags="${_ldflags} -Wl,-rpath -Wl,/opt/intel/mediasdk/lib64:/opt/intel/mediasdk/plugins"

      # fix tensorflow include dir
      ## upstream ffmpeg excects : /usr/include/tensorflow/
      ## tensorflow package ships: /usr/include/tensorflow/tensorflow/
      _cflags+=' -I/usr/include/tensorflow'
      
      # strictly specifying nvcc path is needed if package is installing
      # cuda for the first time (nvcc path will be in $PATH only after relogin)
      sed -i "s@^nvcc_default=.*@nvcc_default='/opt/cuda/bin/nvcc'@" configure
  fi

  msg "Starting configure..."

  ## Add "--enable-decklink \" 
  ## if you have decklink-sdk installed
  ./configure \
    --prefix=/usr \
    --extra-cflags="${_cflags}" \
    --extra-ldflags="${_ldflags}" \
    \
    --toolchain=hardened \
    \
    --disable-rpath \
    --enable-gpl \
    --enable-version3 \
    --enable-nonfree \
    --disable-static \
    --disable-stripping \
    --enable-shared \
    --enable-swresample \
    \
    \
    $_cuda \
    $_cudasdk \
    $_cuvid \
    $_libnpp \
    $_libmfx \
    \
    --enable-ffnvcodec \
    --enable-libdrm \
    --enable-libmfx \
    --enable-nvenc \
    --enable-nvdec \
    --enable-omx \
    --enable-omx-rpi \
    \
    \
    --enable-alsa \
    --enable-avisynth \
    --enable-chromaprint \
    --enable-decoder=atrac3 \
    --enable-decoder=atrac3p \
    --enable-bzlib \
    --enable-fontconfig \
    --enable-frei0r \
    --enable-gpl \
    --enable-gray \
    --enable-iconv \
    --enable-ladspa \
    --enable-libaom \
    --enable-libaribb24 \
    --enable-libass \
    --enable-libcodec2 \
    --enable-libbluray \
    --enable-libbs2b \
    --enable-libcaca \
    --enable-libcdio \
    --enable-libcelt \
    --enable-libcodec2 \
    --enable-libdavs2 \
    --enable-libdc1394 \
    --enable-libfdk-aac \
    --enable-libfreetype \
    --enable-libfribidi \
    --enable-libgme \
    --enable-libgsm \
    --enable-libiec61883 \
    --enable-libilbc \
    --enable-libjack \
    --enable-libklvanc \
    --disable-libkvazaar \
    --enable-liblensfun \
    --enable-libmodplug \
    --enable-libmp3lame \
    --enable-libmysofa \
    --enable-libopencore-amrnb \
    --enable-libopencore-amrwb \
    --disable-libopencv \
    --enable-libopenh264 \
    --enable-libopenjpeg \
    --enable-libopenmpt \
    --enable-libopus \
    --enable-libpulse \
    --enable-librsvg \
    --enable-librubberband \
    --enable-librtmp  \
    --enable-libshine \
    --enable-libsnappy \
    --enable-libsoxr \
    --enable-libspeex \
    --disable-libsrt \
    --enable-libssh \
    --enable-libtensorflow \
    --enable-libtesseract \
    --enable-libtheora \
    --enable-libtwolame \
    --enable-libv4l2 \
    --enable-libvidstab \
    --disable-libvmaf \
    --enable-libvo-amrwbenc \
    --enable-libvorbis \
    --enable-libvpx \
    --enable-libwavpack \
    --enable-libwebp \
    --enable-libx264 \
    --enable-libx265 \
    --enable-libxavs \
    --enable-libxavs2 \
    --enable-libxcb \
    --enable-libxcb-shm \
    --enable-libxcb-xfixes \
    --enable-libxcb-shape \
    --enable-libxvid \
    --enable-libxml2 \
    --enable-libzimg \
    --enable-libzmq \
    --enable-libzvbi \
    --enable-lv2 \
    --enable-lzma \
    --enable-openal \
    --enable-opencl \
    --enable-opengl \
    --enable-openssl \
    --enable-pocketsphinx \
    --enable-rkmpp \
    --enable-sndio \
    --enable-sdl2 \
    --enable-vapoursynth \
    --enable-v4l2-m2m \
    --enable-vaapi \
    --enable-vdpau \
    --enable-xlib \
    --enable-zlib

    
  msg "Starting make"
  make
  make tools/qt-faststart
  make doc/ff{mpeg,play}.1
}

package() {
  cd "$_pkgbasename-$pkgver"
  make DESTDIR="$pkgdir" install install-man
  install -Dm 755 tools/qt-faststart "${pkgdir}/usr/bin"
  install -Dm 644 "$srcdir"/UNREDISTRIBUTABLE.txt "$pkgdir/usr/share/licenses/${pkgname}"
}