set(TARGET_EXECUTABLE "MxpiSemanticSegPostProcessor")

set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${PROJECT_SOURCE_DIR}/dist/MxpiSemanticSegPostProcessor)

file(GLOB_RECURSE SRCS MxpiSemanticSegPostProcessorTest.cpp)
add_executable(${TARGET_EXECUTABLE} ${SRCS})
target_link_libraries(${TARGET_EXECUTABLE} ${MXPLUGINS_HLT_TEST_COMMON_DEP_LIBS})

file(GLOB_RECURSE TEST_FILES *.jpg *.pipeline *.output)
file(GLOB_RECURSE PIPELINE_FILES pipeline/*)
file(GLOB_RECURSE OUTPUT_FILES output/*)

install(FILES ${OUTPUT_FILES} DESTINATION ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/output)
install(FILES ${PIPELINE_FILES} DESTINATION ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/pipeline)
install(FILES ${TEST_FILES} DESTINATION ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})
install(DIRECTORY ${PROJECT_SOURCE_DIR}/MxpiSemanticSegPostProcessor/ms_unetsimplessh DESTINATION ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})

add_test(NAME ${TARGET_EXECUTABLE}
        COMMAND ${TARGET_EXECUTABLE} --gtest_output=xml
        WORKING_DIRECTORY ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})

file(GLOB PYPOSTPROCESS_SO ${PROJECT_SOURCE_DIR}/../../../../output/${PROJECT}/python/postprocess.*.so)
install(FILES ${PYPOSTPROCESS_SO} DESTINATION ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/python)

add_test(NAME ${TARGET_EXECUTABLE}_py
        COMMAND python3 "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/../PyPostProcessTest/main.py" "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/SemanticSegPyPostProcessor.pipeline" "unet_simple" "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/people.jpg"
        WORKING_DIRECTORY ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})

add_test(NAME ${TARGET_EXECUTABLE}_deeplabv3_py
        COMMAND python3 "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/../PyPostProcessTest/main.py" "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/pipeline/Pydeeplabv3_mindspore.pipeline" "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/test_deeplabv3.jpg"
        WORKING_DIRECTORY ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})