project( Proto )

#	Google Protocol Buffers
find_package( Protobuf REQUIRED )

# As of Ubuntu 14.04 protoc is no longer a part of libprotobuf-dev package and should be installed
# separately as in: sudo apt-get install protobuf-compiler
if(PROTOBUF_PROTOC_EXECUTABLE)
	message(STATUS "Found PROTOBUF Compiler: ${PROTOBUF_PROTOC_EXECUTABLE}")
else()
	message(FATAL_ERROR "Could not find PROTOBUF Compiler")
endif()

include_directories(${PROTOBUF_INCLUDE_DIR})
file(GLOB ProtoFiles "${CMAKE_CURRENT_SOURCE_DIR}/*.proto")
PROTOBUF_GENERATE_CPP(ProtoSources ProtoHeaders ${ProtoFiles})
PROTOBUF_GENERATE_PYTHON(ProtoSourcesPy ${ProtoFiles})

add_custom_target(protoPy DEPENDS ${ProtoSourcesPy})

add_library(proto
        ${ProtoSources}
        ${ProtoHeaders}
        )
   

target_link_libraries(proto ${PROTOBUF_LIBRARIES})

# Create proto include directory
file(MAKE_DIRECTORY ${CMAKE_SOURCE_DIR}/include/caffe/proto)

# Copy proto headers to include/caffe/proto/
foreach(header ${ProtoHeaders})

    ADD_CUSTOM_COMMAND(TARGET proto
        COMMAND cmake -E copy ${header}
        ${Caffe_INCLUDE_DIRS}/caffe/proto/
        DEPENDS ${header}
)

endforeach(header)

file(WRITE __init__.py)
install(PROGRAMS __init__.py       DESTINATION python/caffe/proto)
install(PROGRAMS ${ProtoSourcesPy} DESTINATION python/caffe/proto)