CMakeLists.txt

cmake_minimum_required(VERSION 3.16)

project(vectorindex VERSION 0.1.0 LANGUAGES CXX)

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)

option(VECTORINDEX_BUILD_TESTS "Build vectorindex tests" ON)
option(VECTORINDEX_BUILD_BENCH "Build vectorindex benchmark runner" ON)

add_library(vectorindex
    src/brute_force_index.cpp
    src/distance.cpp
    src/hnsw_index.cpp
    src/ivf_index.cpp
    src/kmeans.cpp
    src/mmap_vector_store.cpp
    src/wal_index.cpp
)

target_include_directories(vectorindex
    PUBLIC
        $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
        $<INSTALL_INTERFACE:include>
)

target_compile_options(vectorindex PRIVATE
    $<$<CXX_COMPILER_ID:GNU,Clang>:-Wall -Wextra -Wpedantic>
)

if(VECTORINDEX_BUILD_TESTS)
    enable_testing()

    add_executable(vectorindex_tests
        tests/test_brute_force.cpp
        tests/test_hnsw.cpp
        tests/test_infra.cpp
        tests/test_ivf.cpp
        tests/test_main.cpp
        tests/test_recall.cpp
    )
    target_link_libraries(vectorindex_tests PRIVATE vectorindex)
    add_test(NAME vectorindex_tests COMMAND vectorindex_tests)
endif()

if(VECTORINDEX_BUILD_BENCH)
    add_executable(vectorindex_bench
        bench/ann_dataset_loader.cpp
        bench/benchmark_runner.cpp
        bench/dataset_generator.cpp
        bench/ground_truth.cpp
    )
    target_link_libraries(vectorindex_bench PRIVATE vectorindex)

    add_executable(hnsw_scale_bench
        bench/hnsw_scale_bench.cpp
        bench/dataset_generator.cpp
        bench/ground_truth.cpp
    )
    target_link_libraries(hnsw_scale_bench PRIVATE vectorindex)
endif()