CMakeLists.txt

########################################################################
# Project setup
########################################################################
cmake_minimum_required(VERSION 3.3)
project(PothosGPU CXX)

find_package(Pothos 0.7 CONFIG REQUIRED)

list(INSERT CMAKE_MODULE_PATH 0 ${CMAKE_SOURCE_DIR}/cmake)

########################################################################
# Find ArrayFire
########################################################################
find_package(ArrayFire 3.6.0 REQUIRED)

########################################################################
# Find Python modules used to implement blocks
########################################################################
find_package(PythonInterp REQUIRED)
include(FindPythonModule)
find_python_module(mako REQUIRED)
find_python_module(yaml REQUIRED)

########################################################################
# json.hpp header
########################################################################
find_path(JSON_HPP_INCLUDE_DIR NAMES json.hpp PATH_SUFFIXES nlohmann)

if (NOT JSON_HPP_INCLUDE_DIR)
    message(STATUS "Pothos GPU toolkit requires json.hpp. Exiting...")
    return()
endif ()

########################################################################
# Auto-generate the majority of blocks
########################################################################

file(MAKE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/BlockGen)
set(autogenDeps
    ${CMAKE_CURRENT_SOURCE_DIR}/BlockGen/Blocks.yaml
    ${CMAKE_CURRENT_SOURCE_DIR}/BlockGen/Factory.mako.cpp
    ${CMAKE_CURRENT_SOURCE_DIR}/BlockGen/BlockExecutionTestAuto.mako.cpp
    ${CMAKE_CURRENT_SOURCE_DIR}/BlockGen/GenBlocks.py)
set(autogenOutputs
    ${CMAKE_CURRENT_BINARY_DIR}/BlockGen/Factory.cpp
    ${CMAKE_CURRENT_BINARY_DIR}/BlockGen/BlockExecutionTestAuto.cpp)
add_custom_command(
    OUTPUT ${autogenOutputs}
    DEPENDS ${autogenDeps}
    COMMAND ${PYTHON_EXECUTABLE} "${CMAKE_CURRENT_SOURCE_DIR}/BlockGen/GenBlocks.py" "${CMAKE_CURRENT_BINARY_DIR}/BlockGen" "${ArrayFire_VERSION}"
    COMMENT "Generating block factories and block execution tests")
add_custom_target(
    autogen_files ALL
    DEPENDS ${autogenOutputs})

# We need all sources to be relative paths to be able to use ENABLE_DOCS instead
# of manually specifying files to be scanned.
foreach(autogenOutput ${autogenOutputs})
    file(RELATIVE_PATH relativeAutogenOutput ${CMAKE_CURRENT_SOURCE_DIR} ${autogenOutput})
    list(APPEND relativeAutogenOutputs ${relativeAutogenOutput})
endforeach()

include_directories(
    ${CMAKE_CURRENT_SOURCE_DIR}
    ${CMAKE_CURRENT_SOURCE_DIR}/Source
    ${CMAKE_CURRENT_SOURCE_DIR}/Testing
    ${JSON_HPP_INCLUDE_DIR}
    ${ArrayFire_INCLUDE_DIRS})

set(sources
    ${relativeAutogenOutputs}

    Source/ArrayFireBlock.cpp
    Source/ArrayOpBlock.cpp
    Source/BitShift.cpp
    Source/BitwiseNot.cpp
    Source/BufferConversions.cpp
    Source/Cast.cpp
    Source/Clamp.cpp
    Source/Complex.cpp
    Source/Constant.cpp
    Source/Convolve.cpp
    Source/CorrCoef.cpp
    Source/Covariance.cpp
    Source/DeviceCache.cpp
    Source/EnumConversions.cpp
    Source/FactoryOnly.cpp
    Source/Fallback.cpp
    Source/FFT.cpp
    Source/FileSink.cpp
    Source/FileSource.cpp
    Source/Filter.cpp
    Source/IsX.cpp
    Source/LogN.cpp
    Source/MinMax.cpp
    Source/ModF.cpp
    Source/ModuleInfo.cpp
    Source/NToOneBlock.cpp
    Source/NumericConversions.cpp
    Source/ObjectFunctions.cpp
    Source/OneToOneBlock.cpp
    Source/Pow.cpp
    Source/PowersOfN.cpp
    Source/Random.cpp
    Source/ReducedBlock.cpp
    Source/Replace.cpp
    Source/Root.cpp
    Source/ScalarOpBlock.cpp
    Source/SharedBufferAllocator.cpp
    Source/Sort.cpp
    Source/Statistics.cpp
    Source/TopK.cpp
    Source/TwoToOneBlock.cpp
    Source/Utility.cpp

    # TODO: test constant
    Testing/BlockValueComparisonTests.cpp
    Testing/CastBlockTest.cpp
    Testing/ClampBlockTest.cpp
    Testing/ComparatorBlockTest.cpp
    Testing/ComplexBlockTest.cpp
    Testing/IsXBlockTest.cpp
    Testing/ReplaceBlockTest.cpp
    Testing/NToOneBlockExecutionTest.cpp
    Testing/TestObjectFunctions.cpp
    Testing/OneToOneBlockExecutionTest.cpp
    Testing/TwoToOneBlockExecutionTest.cpp
    Testing/TestArithmeticBlocks.cpp
    Testing/TestBitwise.cpp
    Testing/TestBufferCombos.cpp
    Testing/TestBufferConversions.cpp
    Testing/TestConjugate.cpp
    Testing/TestEnumConversions.cpp
    Testing/TestFFT.cpp
    Testing/TestFileSink.cpp
    Testing/TestFileSource.cpp
    Testing/TestGamma.cpp
    Testing/TestGPUConfig.cpp
    Testing/TestLog.cpp
    Testing/TestLogical.cpp
    Testing/TestManagedDeviceCache.cpp
    Testing/TestMinMax.cpp
    Testing/TestModF.cpp
    Testing/TestNumericConversions.cpp
    Testing/TestPowRoot.cpp
    Testing/TestRoundBlocks.cpp
    Testing/TestRSqrt.cpp
    Testing/TestSetUnion.cpp
    Testing/TestSetUnique.cpp
    Testing/TestSinc.cpp
    Testing/TestStatistics.cpp
    Testing/TestTrigonometric.cpp
    Testing/TestUtility.cpp)

if(POTHOS_ABI_VERSION STRLESS "0.7-2")
    list(APPEND sources
        Source/PinnedBufferManager.cpp)

    add_definitions(-DPOTHOSGPU_LEGACY_BUFFER_MANAGER)
endif()

include(PothosUtil)
POTHOS_MODULE_UTIL(
    TARGET GPUBlocks
    SOURCES
        ${sources}
    LIBRARIES
        ArrayFire::af
    DESTINATION gpu
    ENABLE_DOCS ON
)