#
# Copyright 2020 Google LLC.
#
# Licensed 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
#
#      https://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.
#

# Check for SIMD extensions.

function(wp2_check_compiler_flag WP2_SIMD_DEFINE ENABLE_SIMD)
  if(NOT ENABLE_SIMD)
    message(STATUS "Disabling ${WP2_SIMD_DEFINE} optimization.")
    set(WP2_HAVE_${WP2_SIMD_DEFINE}
        0
        PARENT_SCOPE)
    return()
  endif()
  unset(WP2_HAVE_FLAG_${WP2_SIMD_DEFINE} CACHE)
  check_cxx_source_compiles(
    "
      int main(void) {
        #if !defined(${WP2_SIMD_DEFINE})
        this is not valid code
        #endif
        return 0;
      }
    "
    WP2_HAVE_FLAG_${WP2_SIMD_DEFINE})
  if(WP2_HAVE_FLAG_${WP2_SIMD_DEFINE})
    set(WP2_HAVE_${WP2_SIMD_DEFINE}
        1
        PARENT_SCOPE)
  else()
    set(WP2_HAVE_${WP2_SIMD_DEFINE}
        0
        PARENT_SCOPE)
  endif()
endfunction()

include(CheckCXXSourceCompiles)

# those are included in ifdef's in c++ code.
set(WP2_SIMD_DEFINES "__SSE4_2__;__ARM_NEON__")
if(MSVC)
  # apparently, there's no specific SSE4 flag, but AVX enables it even
  # if we don't use AVX instructions.
  set(SIMD_ENABLE_FLAGS "/arch:AVX;")
  set(SIMD_DISABLE_FLAGS)
else()
  set(SIMD_ENABLE_FLAGS "-msse4.2;-mfpu=neon")
  if(NOT ${ANDROID})
    set(SIMD_DISABLE_FLAGS "-mno-sse4;")  # flag not available on Android
  endif()
endif()

if(${ANDROID})
  if(${ANDROID_ABI} STREQUAL "armeabi-v7a")
    # This is because Android studio uses the configuration "-march=armv7-a
    # -mfloat-abi=softfp -mfpu=vfpv3-d16" that does not trigger neon
    # optimizations but should (as this configuration does not exist anymore).
    set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -mfpu=neon ")
  endif()
endif()

list(LENGTH WP2_SIMD_DEFINES WP2_SIMD_DEFINES_LENGTH)
math(EXPR WP2_SIMD_DEFINES_RANGE "${WP2_SIMD_DEFINES_LENGTH} - 1")

foreach(I_SIMD RANGE ${WP2_SIMD_DEFINES_RANGE})
  list(GET WP2_SIMD_DEFINES ${I_SIMD} WP2_SIMD_DEFINE)
  unset(WP2_HAVE_${WP2_SIMD_DEFINE} CACHE)
  if(NOT simd_found)
    # First try with no extra flag added as the compiler might have
    # default flags(especially on Android).
    set(CMAKE_REQUIRED_FLAGS_INI ${CMAKE_REQUIRED_FLAGS})
    set(CMAKE_REQUIRED_FLAGS)
    wp2_check_compiler_flag(${WP2_SIMD_DEFINE} ${WP2_ENABLE_SIMD})
    if(NOT WP2_HAVE_${WP2_SIMD_DEFINE})
      list(GET SIMD_ENABLE_FLAGS ${I_SIMD} SIMD_COMPILE_FLAG)
      if(EMSCRIPTEN)
        set(SIMD_COMPILE_FLAG "-msimd128 ${SIMD_COMPILE_FLAG}")
      endif()
      set(CMAKE_REQUIRED_FLAGS ${SIMD_COMPILE_FLAG})
      wp2_check_compiler_flag(${WP2_SIMD_DEFINE} ${WP2_ENABLE_SIMD})
    else()
      set(SIMD_COMPILE_FLAG " ")
    endif()
    set(CMAKE_REQUIRED_FLAGS ${CMAKE_REQUIRED_FLAGS_INI})
  endif()
  # Check which files we should include or not.
  if(WP2_HAVE_${WP2_SIMD_DEFINE})
    # Add the flags to the compiler for all files.
    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_COMPILE_FLAG}")
    set(simd_found 1)
  else()
    # Explicitly disable SIMD.
    if(SIMD_DISABLE_FLAGS)
      list(GET SIMD_DISABLE_FLAGS ${I_SIMD} SIMD_COMPILE_FLAG)
      include(CheckCXXCompilerFlag)
      if(SIMD_COMPILE_FLAG)
        unset(HAS_COMPILE_FLAG CACHE)
        check_cxx_compiler_flag(${SIMD_COMPILE_FLAG} HAS_COMPILE_FLAG)
        if(HAS_COMPILE_FLAG)
          # Do one more check for Clang to circumvent CMake issue 13194.
          if(COMMAND check_compiler_flag_common_patterns)
            # Only in CMake 3.0 and above.
            check_compiler_flag_common_patterns(COMMON_PATTERNS)
          else()
            set(COMMON_PATTERNS)
          endif()
          set(CMAKE_REQUIRED_DEFINITIONS ${SIMD_COMPILE_FLAG})
          check_cxx_source_compiles(
            "int main(void) {return 0;}" FLAG2 FAIL_REGEX
            "warning: argument unused during compilation:" ${COMMON_PATTERNS})
          if(NOT FLAG2)
            unset(HAS_COMPILE_FLAG CACHE)
          endif()
        endif()
        if(HAS_COMPILE_FLAG)
          set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${SIMD_COMPILE_FLAG}")
        endif()
      endif()
    endif()
  endif()
endforeach()