# Maintainer: Howetuft <howetuft@gmail.com>
# Contributor:

# You may define CUDA_ARCH to target specific GPU architecture
# for instance: CUDA_ARCH=sm_50
[[ -v CUDA_ARCH ]] && _cuda_capability=(${CUDA_ARCH})

# You may define CUDA_HOST_COMPILER to target specific compiler
# for instance: CUDA_HOST_COMPILER=gcc-9
[[ -v CUDA_HOST_COMPILER ]] && _cuda_host_compiler=(${CUDA_HOST_COMPILER})

pkgname=cycles-standalone
pkgver=4.3.0
pkgrel=1
pkgdesc="Blender Cycles rendering engine, standalone version"
arch=(x86_64)
url="https://github.com/blender/cycles.git"
license=(Apache)
depends=(pugixml opencolorio boost-libs python gflags openimageio intel-oneapi-tbb openvdb embree openimagedenoise opensubdiv openshadinglanguage alembic sdl2 libepoxy usd)
makedepends=(cmake git boost llvm python)
optdepends=(google-glog cuda optix)
provides=(cycles)
source=("git+https://github.com/blender/cycles.git#tag=v${pkgver}"
        usd24_05.patch
        cycles_wrap.sh)
sha256sums=('165dbd7d8a1904aa556d97b57bd485b5ead3f7a4495a6fedfd3f43b0bfe2eb29'
            '92fef506493aa90f20378ad0ecd0bb6637f3e738116e49b050057d5f7bd216de'
            '00afc4aab5541d147b013c31ab91d78e272654a75cae60b39cf70c23a2612c96')

prepare() {
    _src_root_dir="$srcdir/cycles"

    # Remove Find*.cmake, to use local equivalent
    rm $_src_root_dir/src/cmake/Modules/FindClang.cmake

    ls "${srcdir}"
    for patch in "${srcdir}/"*.patch; do
      msg2  "apply $patch..."
      patch -Np1 -d "${srcdir}"/cycles -i "$patch"
    done
}

build() {
    _pyver=$(python -c "from sys import version_info; print(\"%d.%d\" % (version_info.major,version_info.minor))")

    # Determine whether we can precompile CUDA kernels
    # (credits to blender-2.90-git package)
    _CUDA_PKG=$(pacman -Qq cuda-sdk 2>/dev/null) || true
    if [ "$_CUDA_PKG" != "" ] && ! ((DISABLE_CUDA)) ; then
      export CUDA_PATH=/opt/cuda
      export CUDA_INC_PATH=/opt/cuda/include
      _CMAKE_FLAGS+=( -DWITH_CYCLES_CUDA_BINARIES=ON
                      -DCUDA_TOOLKIT_ROOT_DIR=/opt/cuda)
      ((DISABLE_OPTIX)) || _CMAKE_FLAGS+=( -DOPTIX_ROOT_DIR=/opt/optix )
      if [[ -v _cuda_capability ]]; then
        _CMAKE_FLAGS+=( -DCYCLES_CUDA_BINARIES_ARCH="$(IFS=';'; echo "${_cuda_capability[*]}";)" )
      fi

      [ -f "/usr/lib/ccache/bin/nvcc-ccache" ] && _CMAKE_FLAGS+=( -DCUDA_NVCC_EXECUTABLE=/usr/lib/ccache/bin/nvcc-ccache )

      if [[ -v _cuda_host_compiler ]]; then
        _CMAKE_FLAGS+=( -DCUDA_HOST_COMPILER="$_cuda_host_compiler" )
      elif _cuda_gcc=$(basename "$(readlink /opt/cuda/bin/gcc)") ; then
        [ -L "/usr/lib/ccache/bin/$_cuda_gcc" ] && _CMAKE_FLAGS+=( -DCUDA_HOST_COMPILER=/usr/lib/ccache/bin/"$_cuda_gcc" )
      fi
    fi

    echo "${_CMAKE_FLAGS[@]}"
    # Make app
    #
    # INFO (2022-10-08):
    # we don't support NanoVDB at the moment (no nanovdb package in AUR...)
    cmake -B build -S "cycles" \
        -DCMAKE_SYSTEM_NAME=Linux \
        -DPYTHON_VERSION=$_pyver \
        -DCMAKE_BUILD_TYPE='None' \
        -DCMAKE_INSTALL_PREFIX=/usr \
        -DWITH_CYCLES_STANDALONE_GUI=TRUE \
        -DWITH_CYCLES_OSL=TRUE \
        -DCMAKE_SKIP_INSTALL_RPATH=YES \
        -DWITH_CYCLES_NANOVDB=FALSE \
        -DWITH_CYCLES_HYDRA_RENDER_DELEGATE=FALSE \
        -DWITH_CYCLES_USD=TRUE \
        "${_CMAKE_FLAGS[@]}" \
        -Wno-dev

    make -C build
}

package() {
    # Application
    install -D -m755 build/bin/cycles -t "${pkgdir}/usr/lib/cycles"

    # Wrapper script
    install -D -m755 cycles_wrap.sh -T "${pkgdir}/usr/bin/cycles"

    # CUDA binaries (if any)
    if ls build/src/kernel/*.cubin 1> /dev/null 2>&1 ; then
        install -D -m444 build/src/kernel/*.cubin -t "${pkgdir}/usr/lib/cycles/lib"
    fi

    # OSL shaders
    install -D -m444 build/src/kernel/osl/shaders/*.oso -t "${pkgdir}/usr/lib/cycles/shader"

    # License
    install -D -m444 cycles/LICENSE -t "${pkgdir}/usr/share/doc/cycles/license"
}