pkgbase=nvidia-utils
pkgname=('nvidia-utils' 'opencl-nvidia' 'nvidia-settings')
# Edit hardware-detection/hooks/hwdetect_7_graphics and Installer job 11 on version jump from 43x.xx
pkgver=560.35.03
pkgrel=1
epoch=2
arch=('x86_64')
url="https://www.nvidia.com/page/home.html"
license=('custom')
makedepends=('xorg-server' 'libxvmc')
options=('!strip')
#_pkg="NVIDIA-Linux-x86_64-${pkgver}-no-compat32"
_pkg="NVIDIA-Linux-x86_64-${pkgver}"
source=("http://us.download.nvidia.com/XFree86/Linux-x86_64/${pkgver}/${_pkg}.run"
#source=("http://us.download.nvidia.com/XFree86/Linux-x86_64/${pkgver}/NVIDIA-Linux-x86_64-${pkgver}.run"
        'nvidia-restore-settings.desktop'
        'nvidia-drm-outputclass.conf')
md5sums=('d4f54004bb80da17b3e2fb21ac17c018'
         '9237516792da7b6df32f8c7e8a98f475'
         'cb27b0f4a78af78aa96c5aacae23256c')

create_links() {
  while read -d '' _lib; do
        _soname="$(dirname "${_lib}")/$(LC_ALL=C readelf -d "${_lib}" | sed -nr 's/.*Library soname: \[(.*)\].*/\1/p')"
        [[ -e "${_soname}" ]] || ln -s "$(basename "${_lib}")" "${_soname}"
        [[ -e "${_soname/.[0-9]*/}" ]] || ln -s "$(basename "${_soname}")" "${_soname/.[0-9]*/}"
    done < <(find "${pkgdir}" -type f -name '*.so*' -print0)
}

build() {
  cd ${srcdir}
  sh ${_pkg}.run --extract-only
  # https://download.nvidia.com/XFree86/Linux-x86_64/396.51/README/nvidia-persistenced.html
  cd ${_pkg}
  bsdtar -xf nvidia-persistenced-init.tar.bz2

}

package_opencl-nvidia() {
  pkgdesc="OpenCL implemention for NVIDIA"
  depends=('opencl-headers' 'zlib' 'nvidia-utils')

  cd ${srcdir}/${_pkg}

  # OpenCL
  install -D -m644 nvidia.icd ${pkgdir}/etc/OpenCL/vendors/nvidia.icd
  #install -D -m755 libnvidia-compiler.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-compiler.so.${pkgver}
  install -D -m755 libnvidia-opencl.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-opencl.so.${pkgver}

  create_links
}

package_nvidia-utils() {
  pkgdesc="NVIDIA drivers utilities and libraries."
  depends=('xorg-server' 'libxvmc')
  optdepends=('pkg-config: nvidia-xconfig'
              'opencl-nvidia: OpenCL support'
              'nvidia-settings: Gtk GUI')
  conflicts=('nvidia-utils-bumblebee')
  provides=('nvidia-utils-bumblebee')

  # see https://us.download.nvidia.com/XFree86/Linux-x86_64/${pkgver}/README/installedcomponents.html
  # for order and components
  cd ${srcdir}/${_pkg}
  # X driver
  install -D -m755 nvidia_drv.so ${pkgdir}/usr/lib/xorg/modules/drivers/nvidia_drv.so

  # GLX extension module for X
  install -D -m755 libglxserver_nvidia.so.${pkgver} ${pkgdir}/usr/lib/xorg/modules/extensions/libglxserver_nvidia.so.${pkgver}
  install -D -m755 libglxserver_nvidia.so.${pkgver} ${pkgdir}/usr/lib/nvidia/xorg/libglxserver_nvidia.so.${pkgver}
  # X doesn't find glx otherwise:
  ln -s libglxserver_nvidia.so.${pkgver} ${pkgdir}/usr/lib/xorg/modules/extensions/libglxserver_nvidia.so 
  ln -s libglxserver_nvidia.so.${pkgver} ${pkgdir}/usr/lib/nvidia/xorg/libglx.so
  
  # EGL and OpenGL ES libraries
  install -D -m755 libEGL_nvidia.so.${pkgver} ${pkgdir}/usr/lib/libEGL_nvidia.so.${pkgver}
  install -D -m755 libGLESv1_CM_nvidia.so.${pkgver} ${pkgdir}/usr/lib/libGLESv1_CM_nvidia.so.${pkgver}
  install -D -m755 libGLESv2_nvidia.so.${pkgver} ${pkgdir}/usr/lib/libGLESv2_nvidia.so.${pkgver}
  install -D -m644 10_nvidia.json ${pkgdir}/usr/share/glvnd/egl_vendor.d/10_nvidia.json

  # Wayland and GBM EGL
  install -D -m755 libnvidia-egl-gbm.so.1* -t ${pkgdir}/usr/lib/
  install -D -m644 15_nvidia_gbm.json ${pkgdir}/usr/share/egl/egl_external_platform.d/15_nvidia_gbm.json
  install -D -m755 libnvidia-egl-wayland.so.1* -t ${pkgdir}/usr/lib/
  install -D -m644 10_nvidia_wayland.json ${pkgdir}/usr/share/egl/egl_external_platform.d/10_nvidia_wayland.json
  mkdir -p ${pkgdir}/usr/lib/gbm
  ln -sr ${pkgdir}/usr/lib/libnvidia-allocator.so.${pkgver} ${pkgdir}/usr/lib/gbm/nvidia-drm_gbm.so
  
  # XWayland
  install -D -m755 libnvidia-egl-xcb.so.1 ${pkgdir}/usr/lib/libnvidia-egl-xcb.so.1
  install -D -m644 20_nvidia_xcb.json ${pkgdir}/usr/share/egl/egl_external_platform.d/20_nvidia_xcb.json

  # GLVND vendor implementation libraries for GLX
  install -D -m755 libGLX_nvidia.so.${pkgver} ${pkgdir}/usr/lib/libGLX_nvidia.so.${pkgver}

  # Vulkan
  #install -D -m755 libnvidia-vulkan-producer.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-vulkan-producer.so.${pkgver}
  #ln -s libnvidia-vulkan-producer.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-vulkan-producer.so.1
  #ln -s libnvidia-vulkan-producer.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-vulkan-producer.so

  install -D -m644 nvidia_icd.json ${pkgdir}/usr/share/vulkan/icd.d/nvidia_icd.json
  install -D -m644 nvidia_layers.json ${pkgdir}/usr/share/vulkan/implicit_layer.d/nvidia_layers.json

  # Various libraries that are used internally by other driver components
  install -D -m755 libnvidia-cfg.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-cfg.so.${pkgver}
  install -D -m755 libnvidia-eglcore.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-eglcore.so.${pkgver}
  install -D -m755 libnvidia-glcore.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-glcore.so.${pkgver}
  install -D -m755 libnvidia-glsi.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-glsi.so.${pkgver}
  install -D -m755 libnvidia-glvkspirv.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-glvkspirv.so.${pkgver}
  install -D -m755 libnvidia-gpucomp.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-gpucomp.so.${pkgver}
  install -D -m755 libnvidia-rtcore.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-rtcore.so.${pkgver}
  install -D -m755 libnvidia-allocator.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-allocator.so.${pkgver}

  # VDPAU
  install -D -m755 libvdpau_nvidia.so.${pkgver} ${pkgdir}/usr/lib/vdpau/libvdpau_nvidia.so.${pkgver}

  # CUDA
  install -D -m755 libcuda.so.${pkgver} ${pkgdir}/usr/lib/libcuda.so.${pkgver}
  install -D -m755 libnvcuvid.so.${pkgver} ${pkgdir}/usr/lib/libnvcuvid.so.${pkgver}
  install -D -m755 libcudadebugger.so.${pkgver} ${pkgdir}/usr/lib/libcudadebugger.so.${pkgver}

  # Cryptography library wrappers
  install -D -m755 libnvidia-pkcs11.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-pkcs11.so.${pkgver}
  install -D -m755 libnvidia-pkcs11-openssl3.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-pkcs11-openssl3.so.${pkgver}
  
  # PTX JIT Compiler
  install -D -m755 libnvidia-ptxjitcompiler.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-ptxjitcompiler.so.${pkgver}

  # Allow MPI processes to run concurrently on a single GPU
  install -D -m755 nvidia-cuda-mps-server ${pkgdir}/usr/bin/nvidia-cuda-mps-server
  install -D -m755 nvidia-cuda-mps-control ${pkgdir}/usr/bin/nvidia-cuda-mps-control

  # nvidia-tls & nvidia-ml libraries
  install -D -m755 libnvidia-tls.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-tls.so.${pkgver}
  install -D -m755 libnvidia-ml.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-ml.so.${pkgver}

  # Application nvidia-modprobe
  install -D -m4755 nvidia-modprobe ${pkgdir}/usr/bin/nvidia-modprobe

  # nvidia-xconfig
  install -D -m755 nvidia-xconfig ${pkgdir}/usr/bin/nvidia-xconfig
  install -D -m644 nvidia-xconfig.1.gz ${pkgdir}/usr/share/man/man1/nvidia-xconfig.1.gz

  # nvidia-smi
  install -D -m755 nvidia-smi ${pkgdir}/usr/bin/nvidia-smi
  install -D -m644 nvidia-smi.1.gz ${pkgdir}/usr/share/man/man1/nvidia-smi.1.gz

  # nvidia-bug-report
  install -D -m755 nvidia-bug-report.sh ${pkgdir}/usr/bin/nvidia-bug-report.sh

  # nvidia-persistenced
  install -D -m755 nvidia-persistenced ${pkgdir}/usr/bin/nvidia-persistenced
  install -D -m644 nvidia-persistenced.1.gz ${pkgdir}/usr/share/man/man1/nvidia-persistenced.1.gz
  install -D -m644 nvidia-persistenced-init/systemd/nvidia-persistenced.service.template ${pkgdir}/usr/lib/systemd/system/nvidia-persistenced.service
  sed -i 's/__USER__/nvidia-persistenced/' ${pkgdir}/usr/lib/systemd/system/nvidia-persistenced.service
  
  # NvEncodeAPI, NVAPI & NvFBC libraries
  install -D -m755 libnvidia-encode.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-encode.so.${pkgver}
  install -D -m755 libnvidia-api.so.1 ${pkgdir}/usr/lib/libnvidia-api.so.1
  install -D -m755 libnvidia-fbc.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-fbc.so.${pkgver}

  # OptiX
  install -D -m755 libnvoptix.so.${pkgver} ${pkgdir}/usr/lib/libnvoptix.so.${pkgver}
  install -D -m755 nvoptix.bin ${pkgdir}/usr/share/nvidia/nvoptix.bin

  # Optical Flow
  install -D -m755 libnvidia-opticalflow.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-opticalflow.so.${pkgver}

  # NGX
  install -D -m755 libnvidia-ngx.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-ngx.so.${pkgver}

  # NGX for Proton and Wine
  install -D -m644 nvngx.dll -t ${pkgdir}/usr/lib/nvidia/wine
  install -D -m644 _nvngx.dll -t ${pkgdir}/usr/lib/nvidia/wine

  # Firmware
  install -D -m644 firmware/gsp_ga10x.bin ${pkgdir}/lib/firmware/nvidia/${pkgver}/gsp_ga10x.bin
  install -D -m644 firmware/gsp_tu10x.bin ${pkgdir}/lib/firmware/nvidia/${pkgver}/gsp_tu10x.bin

  # VulkanSC ICD
  install -D -m755 libnvidia-vksc-core.so.${pkgver} ${pkgdir}/usr/lib/libnvidia-vksc-core.so.${pkgver}
  install -D -m755 nvidia-pcc ${pkgdir}/usr/bin/nvidia-pcc
  install -D -m644 nvidia_icd_vksc.json ${pkgdir}/etc/vulkansc/icd.d/nvidia_icd_vksc.json

  # license
  install -D -m644 LICENSE ${pkgdir}/usr/share/licenses/nvidia/LICENSE
  ln -s nvidia ${pkgdir}/usr/share/licenses/nvidia-utils
  
  # docs
  install -D -m644 README.txt ${pkgdir}/usr/share/doc/nvidia/README
  install -D -m644 NVIDIA_Changelog ${pkgdir}/usr/share/doc/nvidia/NVIDIA_Changelog
  ln -s nvidia ${pkgdir}/usr/share/doc/nvidia-utils

  # Systemd suspend/hibernate/resume service files
  # http://us.download.nvidia.com/XFree86/Linux-x86_64/430.40/README/powermanagement.html
  install -D -m644 systemd/system/nvidia-suspend.service ${pkgdir}/usr/lib/systemd/system/nvidia-suspend.service
  install -D -m644 systemd/system/nvidia-hibernate.service ${pkgdir}/usr/lib/systemd/system/nvidia-hibernate.service
  install -D -m644 systemd/system/nvidia-resume.service ${pkgdir}/usr/lib/systemd/system/nvidia-resume.service
  install -D -m755 systemd/system-sleep/nvidia ${pkgdir}/usr/lib/systemd/system-sleep/nvidia
  install -D -m755 systemd/nvidia-sleep.sh ${pkgdir}/usr/bin/nvidia-sleep.sh
  install -D -m644 nvidia-powerd ${pkgdir}/usr/bin/nvidia-powerd

  # conf
  install -D -m644 ${srcdir}/nvidia-drm-outputclass.conf ${pkgdir}/etc/X11/xorg.conf.d/nvidia-drm-outputclass.conf

  create_links
}

package_nvidia-settings() {
  pkgdesc="Gtk Graphical User Interface for nVidia"
  depends=('nvidia-utils' 'gtk3')

  cd ${srcdir}/${_pkg}


  install -D -m755 nvidia-settings ${pkgdir}/usr/bin/nvidia-settings
  install -D -m644 nvidia-settings.1.gz ${pkgdir}/usr/share/man/man1/nvidia-settings.1.gz
  install -D -m644 nvidia-settings.desktop ${pkgdir}/usr/share/applications/nvidia-settings.desktop
  install -D -m644 nvidia-settings.png ${pkgdir}/usr/share/pixmaps/nvidia-settings.png
  install -D -m755 libnvidia-gtk3.so.${pkgver} $pkgdir/usr/lib/libnvidia-gtk3.so.${pkgver}
  sed -e 's:__UTILS_PATH__:/usr/bin:' -e 's:__PIXMAP_PATH__:/usr/share/pixmaps:' -i ${pkgdir}/usr/share/applications/nvidia-settings.desktop

  # Install the autostart .desktop file, so that parameters saved through nvidia-settings
  # are automatically reloaded at each login
  install -D -m644 ${srcdir}/nvidia-restore-settings.desktop ${pkgdir}/etc/xdg/autostart/nvidia-restore-settings.desktop
  # key documentation and application profile
  install -D -m644 nvidia-application-profiles-${pkgver}-key-documentation ${pkgdir}/usr/share/nvidia/nvidia-application-profiles-${pkgver}-key-documentation
  install -D -m644 nvidia-application-profiles-${pkgver}-rc ${pkgdir}/usr/share/nvidia/nvidia-application-profiles-${pkgver}-rc
}