CMakeLists.txt

SET(CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR})

CMAKE_MINIMUM_REQUIRED(VERSION 2.6 FATAL_ERROR)
CMAKE_POLICY(VERSION 2.6)

FIND_PACKAGE(Torch REQUIRED)

SET(BUILD_STATIC YES) # makes sure static targets are enabled in ADD_TORCH_PACKAGE

SET(CMAKE_C_FLAGS "--std=c99 -pedantic -Werror -Wall -Wextra -Wno-unused-function -D_GNU_SOURCE ${CMAKE_C_FLAGS}")
SET(src
   src/dataset.c
)
SET(luasrc
   lua/Batch.lua
   lua/Cache.lua
   lua/Dataset.lua
   lua/Getters.lua
   lua/Index.lua
   lua/IndexCSV.lua
   lua/IndexDirectory.lua
   lua/IndexSlowFS.lua
   lua/IndexTensor.lua
   lua/IndexUtils.lua
   lua/Reader.lua
   lua/Sampler.lua
   lua/SlowFS.lua
   lua/SlowFSStream.lua
   lua/StreamedDataset.lua
)

ADD_TORCH_PACKAGE(dataset "${src}" "${luasrc}" "A dataset library, for Torch")

TARGET_LINK_LIBRARIES(dataset luaT TH)

SET_TARGET_PROPERTIES(dataset_static PROPERTIES COMPILE_FLAGS "-fPIC -DSTATIC_TH")

INSTALL(FILES ${luasrc} DESTINATION "${Torch_INSTALL_LUA_PATH_SUBDIR}/dataset")