Added shell script for building cocoapods archive
This commit is contained in:
Prianka Liz Kariat
parent
a774399630
commit
7ad2b7b32f
184
mediapipe/tasks/ios/build_ios_framework.sh
Executable file
184
mediapipe/tasks/ios/build_ios_framework.sh
Executable file
|
|
@ -0,0 +1,184 @@
|
|||
#!/usr/bin/env bash
|
||||
# Copyright 2023 The MediaPipe Authors. All Rights Reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
# Set the following variables as appropriate.
|
||||
# * BAZEL: path to bazel. defaults to the first one available in PATH
|
||||
# * FRAMEWORK_NAME: name of the iOS framework to be built. Currently the
|
||||
# * accepted values are TensorFlowLiteTaskVision, TensorFlowLiteTaskText.
|
||||
# * MPP_BUILD_VERSION: to specify the release version. defaults to 0.0.1-dev
|
||||
# * IS_RELEASE_BUILD: set as true if this build should be a release build
|
||||
# * ARCHIVE_FRAMEWORK: set as true if the framework should be archived
|
||||
# * DEST_DIR: destination directory to which the framework will be copied
|
||||
|
||||
set -ex
|
||||
|
||||
if [[ "$(uname)" != "Darwin" ]]; then
|
||||
echo "This build script only works on macOS."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
BAZEL="${BAZEL:-$(which bazel)}"
|
||||
MPP_BUILD_VERSION=${MPP_BUILD_VERSION:-0.0.1-dev}
|
||||
MPP_ROOT_DIR=$(git rev-parse --show-toplevel)
|
||||
MPP_DISABLE_GPU=true
|
||||
|
||||
if [[ ! -x "${BAZEL}" ]]; then
|
||||
echo "bazel executable is not found."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [ -z ${FRAMEWORK_NAME+x} ]; then
|
||||
echo "Name of the iOS framework, which is to be built, must be set."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
case $FRAMEWORK_NAME in
|
||||
"MediaPipeTaskText")
|
||||
;;
|
||||
*)
|
||||
echo "Wrong framework name. The following framework names are allowed: MediaPipeTaskText"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
if [[ -z "${DEST_DIR+x}" || "${DEST_DIR}" == ${MPP_ROOT_DIR}* ]]; then
|
||||
echo "DEST_DIR variable must be set and not be under the repository root."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# get_output_file_path takes one bazel target label as an argument, and prints
|
||||
# the path of the first output file of the specified target.
|
||||
function get_output_file_path {
|
||||
local STARLARK_OUTPUT_TMPDIR="$(mktemp -d)"
|
||||
|
||||
local STARLARK_FILE="${STARLARK_OUTPUT_TMPDIR}/print_output_file.starlark"
|
||||
cat > "${STARLARK_FILE}" << EOF
|
||||
def format(target):
|
||||
return target.files.to_list()[0].path
|
||||
EOF
|
||||
|
||||
local OUTPUT_PATH=$(bazel cquery $1 --output=starlark --starlark:file="${STARLARK_FILE}" 2> /dev/null)
|
||||
|
||||
rm -rf "${STARLARK_OUTPUT_TMPDIR}"
|
||||
|
||||
echo ${OUTPUT_PATH}
|
||||
}
|
||||
|
||||
# build_target builds a target using the command passed in as argument and
|
||||
# uses cquery to find the path to the output of the target.
|
||||
function build_target {
|
||||
# Build using the command passed as argument.
|
||||
"${BAZEL}" build $1
|
||||
|
||||
# Get the path to the output file of the target.
|
||||
local OUTPUT_PATH=$(get_output_file_path "$1")
|
||||
|
||||
echo ${OUTPUT_PATH}
|
||||
}
|
||||
|
||||
# build_ios_frameworks_and_libraries builds 3 targets:
|
||||
# 1. The ios task library xcframework
|
||||
# 2. Fat static library including graphs needed for the xcframework for all simulator archs (x86_64, arm64).
|
||||
# 2. Static library including graphs needed for the xcframework for all iOS device archs (arm64).
|
||||
function build_ios_frameworks_and_libraries {
|
||||
local TARGET_PREFIX="//mediapipe/tasks/ios"
|
||||
FULL_FRAMEWORK_TARGET="${TARGET_PREFIX}:${FRAMEWORK_NAME}_framework"
|
||||
FULL_GRAPH_LIBRARY_TARGET="${TARGET_PREFIX}:${FRAMEWORK_NAME}_GraphLibrary"
|
||||
|
||||
local FRAMEWORK_CQUERY_COMMAND="-c opt --define MEDIAPIPE_DISABLE_GPU=${MPP_DISABLE_GPU} --define MEDIAPIPE_AVOID_LINKING_GRAPHS=1 \
|
||||
--apple_generate_dsym=false ${FULL_FRAMEWORK_TARGET}"
|
||||
IOS_FRAMEWORK_PATH="$(build_target "${FRAMEWORK_CQUERY_COMMAND}")"
|
||||
|
||||
local IOS_SIM_FAT_LIBRARY_CQUERY_COMMAND="-c opt --config=ios_sim_fat --define MEDIAPIPE_DISABLE_GPU=${MPP_DISABLE_GPU} \
|
||||
--apple_generate_dsym=false ${FULL_GRAPH_LIBRARY_TARGET}"
|
||||
IOS_GRAPHS_SIMULATOR_LIBRARY_PATH="$(build_target "${IOS_SIM_FAT_LIBRARY_CQUERY_COMMAND}")"
|
||||
|
||||
local IOS_DEVICE_LIBRARY_CQUERY_COMMAND="-c opt --config=ios_arm64 --define MEDIAPIPE_DISABLE_GPU=${MPP_DISABLE_GPU} \
|
||||
--apple_generate_dsym=false ${FULL_GRAPH_LIBRARY_TARGET}"
|
||||
IOS_GRAPHS_DEVICE_LIBRARY_PATH="$(build_target "${IOS_DEVICE_LIBRARY_CQUERY_COMMAND}")"
|
||||
}
|
||||
|
||||
function create_framework_archive {
|
||||
# Change to the Bazel iOS output directory.
|
||||
pushd "${BAZEL_IOS_OUTDIR}"
|
||||
|
||||
# Create the temporary directory for the given framework.
|
||||
local ARCHIVE_NAME="${FRAMEWORK_NAME}-${MPP_BUILD_VERSION}"
|
||||
local MPP_TMPDIR="$(mktemp -d)"
|
||||
|
||||
# Copy the license file to MPP_TMPDIR
|
||||
cp "LICENSE" ${MPP_TMPDIR}
|
||||
|
||||
# Unzip the iOS framework zip generated by bazel to MPP_TMPDIR
|
||||
local FRAMEWORKS_DIR="${MPP_TMPDIR}/frameworks"
|
||||
|
||||
echo ${IOS_FRAMEWORK_PATH}
|
||||
unzip "${IOS_FRAMEWORK_PATH}" -d "${FRAMEWORKS_DIR}"
|
||||
|
||||
local GRAPH_LIBRARIES_DIR="graph_libraries"
|
||||
|
||||
# Create the parent folder which will hold the graph libraries of all architectures.
|
||||
mkdir -p "${FRAMEWORKS_DIR}/${GRAPH_LIBRARIES_DIR}"
|
||||
|
||||
local SIMULATOR_GRAPH_LIBRARY_PATH="${FRAMEWORKS_DIR}/${GRAPH_LIBRARIES_DIR}/lib${FRAMEWORK_NAME}_simulator_graph.a"
|
||||
|
||||
# Copy ios simulator fat library into a separate directory.
|
||||
echo ${IOS_GRAPHS_SIMULATOR_LIBRARY_PATH}
|
||||
cp "${IOS_GRAPHS_SIMULATOR_LIBRARY_PATH}" "${SIMULATOR_GRAPH_LIBRARY_PATH}"
|
||||
|
||||
|
||||
local IOS_DEVICE_GRAPH_LIBRARY_PATH="${FRAMEWORKS_DIR}/${GRAPH_LIBRARIES_DIR}/lib${FRAMEWORK_NAME}_device_graph.a"
|
||||
|
||||
# Copy ios device library into a separate directory.
|
||||
echo ${IOS_GRAPHS_DEVICE_LIBRARY_PATH}
|
||||
cp "${IOS_GRAPHS_DEVICE_LIBRARY_PATH}" "${IOS_DEVICE_GRAPH_LIBRARY_PATH}"
|
||||
|
||||
#----- (3) Move the framework to the destination -----
|
||||
if [[ "${ARCHIVE_FRAMEWORK}" == true ]]; then
|
||||
local TARGET_DIR="$(realpath "${FRAMEWORK_NAME}")"
|
||||
|
||||
# Create the framework archive directory.
|
||||
|
||||
local FRAMEWORK_ARCHIVE_DIR
|
||||
if [[ "${IS_RELEASE_BUILD}" == true ]]; then
|
||||
# Get the first 16 bytes of the sha256 checksum of the root directory.
|
||||
local SHA256_CHECKSUM=$(find "${MPP_TMPDIR}" -type f -print0 | xargs -0 shasum -a 256 | sort | shasum -a 256 | cut -c1-16)
|
||||
FRAMEWORK_ARCHIVE_DIR="${TARGET_DIR}/${MPP_BUILD_VERSION}/${SHA256_CHECKSUM}"
|
||||
else
|
||||
FRAMEWORK_ARCHIVE_DIR="${TARGET_DIR}/${MPP_BUILD_VERSION}"
|
||||
fi
|
||||
mkdir -p "${FRAMEWORK_ARCHIVE_DIR}"
|
||||
|
||||
# Zip up the framework and move to the archive directory.
|
||||
pushd "${MPP_TMPDIR}"
|
||||
local MPP_ARCHIVE_FILE="${ARCHIVE_NAME}.tar.gz"
|
||||
tar -cvzf "${MPP_ARCHIVE_FILE}" .
|
||||
mv "${MPP_ARCHIVE_FILE}" "${FRAMEWORK_ARCHIVE_DIR}"
|
||||
popd
|
||||
|
||||
# Move the target directory to the Kokoro artifacts directory.
|
||||
mv "${TARGET_DIR}" "$(realpath "${DEST_DIR}")"/
|
||||
else
|
||||
rsync -r "${MPP_TMPDIR}/" "$(realpath "${DEST_DIR}")/"
|
||||
fi
|
||||
|
||||
# Clean up the temporary directory for the framework.
|
||||
rm -rf "${MPP_TMPDIR}"
|
||||
echo ${MPP_TMPDIR}
|
||||
}
|
||||
|
||||
cd "${MPP_ROOT_DIR}"
|
||||
build_ios_frameworks_and_libraries
|
||||
create_framework_archive
|
||||
Loading…
Reference in New Issue
Block a user