# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

# - Find Arrow CUDA (arrow/gpu/cuda_api.h, libarrow_cuda.a, libarrow_cuda.so)
#
# This module requires Arrow from which it uses
#  arrow_find_package()
#
# This module defines
#  ARROW_CUDA_FOUND, whether Arrow CUDA has been found
#  ARROW_CUDA_IMPORT_LIB, path to libarrow_cuda's import library (Windows only)
#  ARROW_CUDA_INCLUDE_DIR, directory containing headers
#  ARROW_CUDA_LIBS, deprecated. Use ARROW_CUDA_LIB_DIR instead
#  ARROW_CUDA_LIB_DIR, directory containing Arrow CUDA libraries
#  ARROW_CUDA_SHARED_IMP_LIB, deprecated. Use ARROW_CUDA_IMPORT_LIB instead
#  ARROW_CUDA_SHARED_LIB, path to libarrow_cuda's shared library
#  ARROW_CUDA_STATIC_LIB, path to libarrow_cuda.a

if(DEFINED ARROW_CUDA_FOUND)
  return()
endif()

set(find_package_arguments)
if(${CMAKE_FIND_PACKAGE_NAME}_FIND_VERSION)
  list(APPEND find_package_arguments "${${CMAKE_FIND_PACKAGE_NAME}_FIND_VERSION}")
endif()
if(${CMAKE_FIND_PACKAGE_NAME}_FIND_REQUIRED)
  list(APPEND find_package_arguments REQUIRED)
endif()
if(${CMAKE_FIND_PACKAGE_NAME}_FIND_QUIETLY)
  list(APPEND find_package_arguments QUIET)
endif()
find_package(Arrow ${find_package_arguments})

if(ARROW_FOUND)
  arrow_find_package(ARROW_CUDA
                     "${ARROW_HOME}"
                     arrow_cuda
                     arrow/gpu/cuda_api.h
                     ArrowCUDA
                     arrow-cuda)
  if(NOT ARROW_CUDA_VERSION)
    set(ARROW_CUDA_VERSION "${ARROW_VERSION}")
  endif()
endif()

if("${ARROW_CUDA_VERSION}" VERSION_EQUAL "${ARROW_VERSION}")
  set(ARROW_CUDA_VERSION_MATCH TRUE)
else()
  set(ARROW_CUDA_VERSION_MATCH FALSE)
endif()

mark_as_advanced(ARROW_CUDA_IMPORT_LIB
                 ARROW_CUDA_INCLUDE_DIR
                 ARROW_CUDA_LIBS
                 ARROW_CUDA_LIB_DIR
                 ARROW_CUDA_SHARED_IMP_LIB
                 ARROW_CUDA_SHARED_LIB
                 ARROW_CUDA_STATIC_LIB
                 ARROW_CUDA_VERSION
                 ARROW_CUDA_VERSION_MATCH)

find_package_handle_standard_args(
  ArrowCUDA
  REQUIRED_VARS ARROW_CUDA_INCLUDE_DIR ARROW_CUDA_LIB_DIR ARROW_CUDA_VERSION_MATCH
  VERSION_VAR ARROW_CUDA_VERSION)
set(ARROW_CUDA_FOUND ${ArrowCUDA_FOUND})

if(ArrowCUDA_FOUND AND NOT ArrowCUDA_FIND_QUIETLY)
  message(STATUS "Found the Arrow CUDA by ${ARROW_CUDA_FIND_APPROACH}")
  message(STATUS "Found the Arrow CUDA shared library: ${ARROW_CUDA_SHARED_LIB}")
  message(STATUS "Found the Arrow CUDA import library: ${ARROW_CUDA_IMPORT_LIB}")
  message(STATUS "Found the Arrow CUDA static library: ${ARROW_CUDA_STATIC_LIB}")
endif()