marinone94's picture
Training in progress, epoch 0
1ce325b
# Explicitly list the source files for this subdirectory
#
# If you add any source files to this subdirectory
# that should be included in the kenlm library,
# (this excludes any unit test files)
# you should add them to the following list:
#
# Because we do not set PARENT_SCOPE in the following definition,
# CMake files in the parent directory won't be able to access this variable.
#
set(KENLM_UTIL_SOURCE
bit_packing.cc
ersatz_progress.cc
exception.cc
file.cc
file_piece.cc
float_to_string.cc
integer_to_string.cc
mmap.cc
murmur_hash.cc
parallel_read.cc
pool.cc
read_compressed.cc
scoped.cc
spaces.cc
string_piece.cc
usage.cc
)
if (WIN32)
set(KENLM_UTIL_SOURCE ${KENLM_UTIL_SOURCE} getopt.c)
endif()
# This directory has children that need to be processed
add_subdirectory(double-conversion)
add_subdirectory(stream)
add_library(kenlm_util ${KENLM_UTIL_DOUBLECONVERSION_SOURCE} ${KENLM_UTIL_STREAM_SOURCE} ${KENLM_UTIL_SOURCE})
# Since headers are relative to `include/kenlm` at install time, not just `include`
target_include_directories(kenlm_util PUBLIC $<INSTALL_INTERFACE:include/kenlm>)
set(READ_COMPRESSED_FLAGS)
find_package(ZLIB)
if (ZLIB_FOUND)
set(READ_COMPRESSED_FLAGS "${READ_COMPRESSED_FLAGS} -DHAVE_ZLIB")
target_link_libraries(kenlm_util PRIVATE ${ZLIB_LIBRARIES})
include_directories(${ZLIB_INCLUDE_DIR})
endif()
find_package(BZip2)
if (BZIP2_FOUND)
set(READ_COMPRESSED_FLAGS "${READ_COMPRESSED_FLAGS} -DHAVE_BZLIB")
target_link_libraries(kenlm_util PRIVATE ${BZIP2_LIBRARIES})
include_directories(${BZIP2_INCLUDE_DIR})
endif()
find_package(LibLZMA)
if (LIBLZMA_FOUND)
set(READ_COMPRESSED_FLAGS "${READ_COMPRESSED_FLAGS} -DHAVE_XZLIB")
target_link_libraries(kenlm_util PRIVATE ${LIBLZMA_LIBRARIES})
include_directories(${LIBLZMA_INCLUDE_DIRS})
endif()
if (NOT "${READ_COMPRESSED_FLAGS}" STREQUAL "")
set_source_files_properties(read_compressed.cc PROPERTIES COMPILE_FLAGS ${READ_COMPRESSED_FLAGS})
set_source_files_properties(read_compressed_test.cc PROPERTIES COMPILE_FLAGS ${READ_COMPRESSED_FLAGS})
set_source_files_properties(file_piece_test.cc PROPERTIES COMPILE_FLAGS ${READ_COMPRESSED_FLAGS})
endif()
if(UNIX)
include(CheckLibraryExists)
check_library_exists(rt clock_gettime "clock_gettime from librt" HAVE_CLOCKGETTIME_RT)
if (HAVE_CLOCKGETTIME_RT)
set(RT rt)
else()
check_library_exists(c clock_gettime "clock_gettime from the libc" HAVE_CLOCKGETTIME)
endif()
if (HAVE_CLOCKGETTIME_RT OR HAVE_CLOCKGETTIME)
add_definitions(-DHAVE_CLOCKGETTIME)
endif()
endif()
# Group these objects together for later use.
set_target_properties(kenlm_util PROPERTIES POSITION_INDEPENDENT_CODE ON)
target_link_libraries(kenlm_util
PUBLIC
# Boost is required for building binaries and tests
"$<BUILD_INTERFACE:${Boost_LIBRARIES}>"
PRIVATE
Threads::Threads
${RT})
install(
TARGETS kenlm_util
EXPORT kenlmTargets
RUNTIME DESTINATION bin
LIBRARY DESTINATION lib
ARCHIVE DESTINATION lib
INCLUDES DESTINATION include
)
if (NOT WIN32)
AddExes(EXES probing_hash_table_benchmark
LIBRARIES kenlm_util Threads::Threads)
endif()
# Only compile and run unit tests if tests should be run
if(BUILD_TESTING)
set(KENLM_BOOST_TESTS_LIST
bit_packing_test
integer_to_string_test
joint_sort_test
multi_intersection_test
pcqueue_test
probing_hash_table_test
read_compressed_test
sized_iterator_test
sorted_uniform_test
string_stream_test
tokenize_piece_test
)
AddTests(TESTS ${KENLM_BOOST_TESTS_LIST}
LIBRARIES kenlm_util Threads::Threads)
# file_piece_test requires an extra command line parameter
KenLMAddTest(TEST file_piece_test
LIBRARIES kenlm_util Threads::Threads
TEST_ARGS ${CMAKE_CURRENT_SOURCE_DIR}/file_piece.cc)
endif()