build-onnxruntime-linux.sh

#!/bin/bash
# build onnxruntime by benjaminwan
# CMakeFiles/onnxruntime.dir/link.txt/link/lib*.a

function is_cmd_exist() {
    retval=""
    if ! command -v $1 >/dev/null 2>&1; then
        retval="false"
    else
        retval="true"
    fi
    echo "$retval"
}

function collect_shared_lib() {
    if [ -d "install/bin" ]; then
        rm -r -f install/bin
    fi

    if [ -d "install/include/onnxruntime" ]; then
        mv install/include/onnxruntime/* install/include
        rm -rf install/include/onnxruntime
    fi

    echo "set(OnnxRuntime_INCLUDE_DIRS \"\${CMAKE_CURRENT_LIST_DIR}/include\")" >install/OnnxRuntimeConfig.cmake
    echo "include_directories(\${OnnxRuntime_INCLUDE_DIRS})" >>install/OnnxRuntimeConfig.cmake
    echo "link_directories(\${CMAKE_CURRENT_LIST_DIR}/lib)" >>install/OnnxRuntimeConfig.cmake
    echo "set(OnnxRuntime_LIBS onnxruntime)" >>install/OnnxRuntimeConfig.cmake
}

function copy_libs() {
    all_link=$(cat CMakeFiles/onnxruntime.dir/link.txt)
    link=${all_link#*onnxruntime.dir}
    regex="lib.*.a$"
    libs=""
    for var in $link; do
        if [[ ${var} =~ ${regex} ]]; then
            #echo cp ${var} install-static/lib
            cp ${var} install-static/lib
            name=$(echo $var | grep -E ${regex} -o)
            name=${name#lib}
            name=${name%.a}
            libs="${libs} ${name}"
        fi
    done
    echo "$libs"
}

function combine_libs_linux() {
    all_link=$(cat CMakeFiles/onnxruntime.dir/link.txt)
    link=${all_link#*onnxruntime.dir}
    regex="lib.*.a$"
    root_path="${PWD}"
    static_path="${PWD}/install-static"
    lib_path="${PWD}/install-static/lib"
    mkdir -p $lib_path
    echo "create ${lib_path}/libonnxruntime.a" >${static_path}/libonnxruntime.mri
    for var in $link; do
        if [[ ${var} =~ ${regex} ]]; then
            echo "addlib ${root_path}/${var}" >>${static_path}/libonnxruntime.mri
        fi
    done
    echo "save" >>${static_path}/libonnxruntime.mri
    echo "end" >>${static_path}/libonnxruntime.mri
    ar -M <${static_path}/libonnxruntime.mri
    #    ranlib ${lib_path}/libonnxruntime.a
}

function collect_static_libs() {
    if [ -d "install-static" ]; then
        rm -r -f install-static
    fi
    mkdir -p install-static/lib

    if [ -d "install/include" ]; then
        cp -r install/include install-static
    fi

    if [ ! -f "CMakeFiles/onnxruntime.dir/link.txt" ]; then
        echo "link.txt is not exist, collect static libs error."
        exit 0
    fi

    ar_exist=$(is_cmd_exist ar)
    ranlib_exist=$(is_cmd_exist ranlib)
    if [ "$ar_exist" == "true" ] && [ "$ranlib_exist" == "true" ]; then
        echo "combine_libs_linux"
        combine_libs_linux
        libs="onnxruntime"
    else
        echo "copy_libs"
        libs=$(copy_libs)
    fi

    echo "set(OnnxRuntime_INCLUDE_DIRS \"\${CMAKE_CURRENT_LIST_DIR}/include\")" >install-static/OnnxRuntimeConfig.cmake
    echo "include_directories(\${OnnxRuntime_INCLUDE_DIRS})" >>install-static/OnnxRuntimeConfig.cmake
    echo "link_directories(\${CMAKE_CURRENT_LIST_DIR}/lib)" >>install-static/OnnxRuntimeConfig.cmake
    echo "set(OnnxRuntime_LIBS $libs)" >>install-static/OnnxRuntimeConfig.cmake

    cp CMakeFiles/onnxruntime.dir/link.txt install-static/link.log
}

HOST_OS=$(uname -s)
NUM_THREADS=1
BUILD_TYPE=Release

if [ $HOST_OS == "Linux" ]; then
    NUM_THREADS=$(nproc)
else
    echo "Unsupport OS: $HOST_OS"
    exit 0
fi

JAVA_FLAG=""

while getopts "j" arg; do
    case $arg in
    j)
        echo "j's arg:$OPTARG"
        JAVA_FLAG="--build_java"
        ;;
    ?)
        echo -e "unkonw argument. \nuseage1: ./build-onnxruntim-mac.bat -a x86_64 \nuseage2: ./build-onnxruntim-mac.bat -a arm64"
        exit 1
        ;;
    esac
done

DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
python3 $DIR/tools/ci_build/build.py --build_dir $DIR/build-$HOST_OS \
--allow_running_as_root \
--config $BUILD_TYPE \
--parallel "$NUM_THREADS" \
--skip_tests \
--build_shared_lib \
--compile_no_warning_as_error \
$JAVA_FLAG \
--cmake_extra_defines CMAKE_INSTALL_PREFIX=./install \
onnxruntime_BUILD_UNIT_TESTS=OFF

if [ ! -d "build-$HOST_OS/$BUILD_TYPE" ]; then
    echo "Build error!"
    exit 0
fi

pushd build-$HOST_OS/$BUILD_TYPE
cmake --install .
if [ ! -d "install" ]; then
    echo "Cmake install  error!"
    exit 0
fi
collect_shared_lib
collect_static_libs
popd