agsantos | 66b9adb | 2020-06-23 10:40:48 -0400 | [diff] [blame^] | 1 | #! /bin/bash |
| 2 | # Build the plugin for the project |
| 3 | if [ -z $DAEMON ]; then |
| 4 | DAEMON="./../../daemon" |
| 5 | echo "DAEMON not provided, building for ${DAEMON}" |
| 6 | fi |
| 7 | |
| 8 | PLUGIN_NAME="foregroundsegmentation" |
| 9 | JPL_FILE_NAME=${PLUGIN_NAME}".jpl" |
| 10 | SO_FILE_NAME="lib"${PLUGIN_NAME}".so" |
| 11 | DAEMON_SRC="${DAEMON}/src" |
| 12 | CONTRIB_PATH="${DAEMON}/contrib" |
| 13 | DESTINATION_PATH="./../build/" |
| 14 | PLUGINS_LIB="../lib" |
| 15 | LIBS_DIR="/home/${USER}/Libs" |
| 16 | |
| 17 | |
| 18 | CONTRIB_PLATFORM_CURT=x86_64 |
| 19 | CONTRIB_PLATFORM=${CONTRIB_PLATFORM_CURT}-linux-gnu |
| 20 | |
| 21 | mkdir -p lib/${CONTRIB_PLATFORM_CURT} |
| 22 | mkdir -p ${DESTINATION_PATH}/${CONTRIB_PLATFORM}/jpl |
| 23 | |
| 24 | # Compile |
| 25 | clang++ -std=c++14 -shared -fPIC \ |
| 26 | -Wl,-Bsymbolic,-rpath,"\${ORIGIN}" \ |
| 27 | -Wall -Wextra \ |
| 28 | -Wno-unused-variable \ |
| 29 | -Wno-unused-function \ |
| 30 | -Wno-unused-parameter \ |
| 31 | -I"." \ |
| 32 | -I${DAEMON_SRC} \ |
| 33 | -I"${CONTRIB_PATH}/${CONTRIB_PLATFORM}/include" \ |
| 34 | -I"${CONTRIB_PATH}/${CONTRIB_PLATFORM}/include/opencv4" \ |
| 35 | -I${LIBS_DIR}/_tensorflow_cc/include \ |
| 36 | -I${LIBS_DIR}/_tensorflow_cc/include/third_party/eigen3 \ |
| 37 | -I${PLUGINS_LIB} \ |
| 38 | main.cpp \ |
| 39 | videoSubscriber.cpp \ |
| 40 | pluginProcessor.cpp \ |
| 41 | pluginMediaHandler.cpp \ |
| 42 | TFInference.cpp \ |
| 43 | pluginInference.cpp \ |
| 44 | pluginParameters.cpp \ |
| 45 | -L${CONTRIB_PATH}/${CONTRIB_PLATFORM}/lib/ \ |
| 46 | -L${CONTRIB_PATH}/${CONTRIB_PLATFORM}/lib/opencv4/3rdparty/ \ |
| 47 | -L${LIBS_DIR}/_tensorflow_cc/lib/${CONTRIB_PLATFORM}-gpu61/ \ |
| 48 | -lswscale \ |
| 49 | -lavutil \ |
| 50 | -lopencv_imgcodecs \ |
| 51 | -lopencv_imgproc \ |
| 52 | -lopencv_core \ |
| 53 | -ltensorflow_cc \ |
| 54 | -lpng \ |
| 55 | -o lib/${CONTRIB_PLATFORM_CURT}/${SO_FILE_NAME} |
| 56 | # (above) Always put opencv_core after all other opencv libs |
| 57 | # (above) Always put avutil after all other ffmpeg libs |
| 58 | # (above) Always put png after all other libs |
| 59 | |
| 60 | cp ${LIBS_DIR}/_tensorflow_cc/lib/${CONTRIB_PLATFORM}-gpu61/libtensorflow_cc.so lib/$CONTRIB_PLATFORM_CURT/libtensorflow_cc.so.2 |
| 61 | cp /usr/lib/${CONTRIB_PLATFORM}/libswscale.so.4 lib/$CONTRIB_PLATFORM_CURT |
| 62 | cp /usr/lib/${CONTRIB_PLATFORM}/libavutil.so.55 lib/$CONTRIB_PLATFORM_CURT |
| 63 | cp /usr/lib/${CONTRIB_PLATFORM}/libpng16.so.16 lib/$CONTRIB_PLATFORM_CURT |
| 64 | |
| 65 | zip -r ${JPL_FILE_NAME} data manifest.json lib |
| 66 | mv ${JPL_FILE_NAME} ${DESTINATION_PATH}/${CONTRIB_PLATFORM}/jpl/ |
| 67 | |
| 68 | # Cleanup |
| 69 | # Remove lib after compilation |
| 70 | rm -rf lib |