Added files for iOS task text cocoapods

mediapipe/tasks/ios/BUILD

@@ -0,0 +1,63 @@
+# 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.
+
+load(
+    "//mediapipe/tasks/ios:ios.bzl",
+    "MPP_TASK_MINIMUM_OS_VERSION",
+    "strip_api_include_path_prefix",
+)
+load(
+    "@build_bazel_rules_apple//apple:ios.bzl",
+    "ios_static_framework",
+)
+
+package(default_visibility = ["//mediapipe/tasks:internal"])
+
+licenses(["notice"])
+
+strip_api_include_path_prefix(
+    name = "strip_api_include_path",
+    hdr_labels = [
+        "//mediapipe/tasks/ios/common:sources/MPPCommon.h",
+        "//mediapipe/tasks/ios/components/containers:sources/MPPCategory.h",
+        "//mediapipe/tasks/ios/components/containers:sources/MPPClassificationResult.h",
+        "//mediapipe/tasks/ios/core:sources/MPPBaseOptions.h",
+        "//mediapipe/tasks/ios/core:sources/MPPTaskOptions.h",
+        "//mediapipe/tasks/ios/core:sources/MPPTaskResult.h",
+        "//mediapipe/tasks/ios/text/text_classifier:sources/MPPTextClassifier.h",
+        "//mediapipe/tasks/ios/text/text_classifier:sources/MPPTextClassifierOptions.h",
+        "//mediapipe/tasks/ios/text/text_classifier:sources/MPPTextClassifierResult.h",
+    ],
+)
+
+ios_static_framework(
+    name = "MediaPipeTaskText_framework",
+    hdrs = [
+        ":MPPBaseOptions.h",
+        ":MPPCategory.h",
+        ":MPPClassificationResult.h",
+        ":MPPCommon.h",
+        ":MPPTaskOptions.h",
+        ":MPPTaskResult.h",
+        ":MPPTextClassifier.h",
+        ":MPPTextClassifierOptions.h",
+        ":MPPTextClassifierResult.h",
+    ],
+    bundle_name = "MediaPipeTaskText",
+    minimum_os_version = MPP_TASK_MINIMUM_OS_VERSION,
+    deps = [
+        # "//mediapipe/tasks/cc/text/text_classifier:text_classifier_graph",
+        "//mediapipe/tasks/ios/text/text_classifier:MPPTextClassifier",
+    ],
+)

mediapipe/tasks/ios/MediaPipeTaskText.podspec.template

@@ -0,0 +1,20 @@
+Pod::Spec.new do |s|
+  s.name             = 'MediaPipeTaskText'
+  s.version          = '${MPP_BUILD_VERSION}'
+  s.authors          = 'Google Inc.'
+  s.license          = { :type => 'Apache',:file => "LICENSE" }
+  s.homepage         = 'https://github.com/google/mediapipe'
+  s.source           = { :http => '${MPP_DOWNLOAD_URL}' }
+  s.summary          = 'MediaPipe Task Library - Text'
+  s.description      = 'The Natural Language APIs of the MediaPipe Task Library'
+
+  s.ios.deployment_target = '10.0'
+
+  s.module_name = 'MediaPipeTaskText'
+  s.static_framework = true
+  s.pod_target_xcconfig = {
+    'VALID_ARCHS' => 'x86_64, arm64, armv7',
+  }
+  s.library = 'c++'
+  s.vendored_frameworks = 'Frameworks/MediaPipeTaskText.framework'
+end

mediapipe/tasks/ios/build_ios_framework.sh

@@ -0,0 +1,139 @@
+#!/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)
+
+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 FRAMEWORK_PATH=$(bazel cquery -c opt --config=ios_fat --define=MEDIAPIPE_DISABLE_GPU=1  $1 \
+    --output=starlark --starlark:file="${STARLARK_FILE}" 2> /dev/null)
+
+  # Clean up the temporary directory for bazel cquery.
+  rm -rf "${STARLARK_OUTPUT_TMPDIR}"
+
+  echo ${FRAMEWORK_PATH}
+}
+
+function build_ios_api_framework {
+  local TARGET_PREFIX="//mediapipe/tasks/ios"
+  FULL_TARGET="${TARGET_PREFIX}:${FRAMEWORK_NAME}_framework"
+
+  "${BAZEL}" build -c opt --config=ios_fat --define=MEDIAPIPE_DISABLE_GPU=1 ${FULL_TARGET}
+
+  # Find the path of the iOS framework generated by the above bazel build command
+  IOS_FRAMEWORK_PATH="$(get_output_file_path "${FULL_TARGET}")"
+  echo ${IOS_FRAMEWORK_PATH}
+}
+
+function create_framework_archive {
+  # Change to the Bazel iOS output directory.
+  pushd "${BAZEL_IOS_OUTDIR}"
+
+  # Create the temporary directory for the given framework.
+  ARCHIVE_NAME="${FRAMEWORK_NAME}-${MPP_BUILD_VERSION}"
+  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
+  echo ${IOS_FRAMEWORK_PATH}
+  unzip "${IOS_FRAMEWORK_PATH}" -d "${MPP_TMPDIR}"/Frameworks
+
+  #----- (3) Move the framework to the destination -----
+  if [[ "${ARCHIVE_FRAMEWORK}" == true ]]; then
+    TARGET_DIR="$(realpath "${FRAMEWORK_NAME}")"
+
+    # Create the framework archive directory.
+    if [[ "${IS_RELEASE_BUILD}" == true ]]; then
+      # Get the first 16 bytes of the sha256 checksum of the root directory.
+      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}"
+    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_api_framework
+create_framework_archive

mediapipe/tasks/ios/ios.bzl

@@ -0,0 +1,39 @@
+"""MediaPipe Task Library Helper Rules for iOS"""
+
+MPP_TASK_MINIMUM_OS_VERSION = "11.0"
+
+# When the static framework is built with bazel, the all header files are moved
+# to the "Headers" directory with no header path prefixes. This auxiliary rule
+# is used for stripping the path prefix to the C/iOS API header files included by
+# other C/iOS API header files.
+# In case of C header files includes start with a keyword of "#include'.
+# Imports in iOS header files start with a keyword of '#import'.
+def strip_api_include_path_prefix(name, hdr_labels, prefix = ""):
+    """Create modified header files with the import path stripped out.
+
+    Args:
+      name: The name to be used as a prefix to the generated genrules.
+      hdr_labels: List of header labels to strip out the include path. Each
+          label must end with a colon followed by the header file name.
+      prefix: Optional prefix path to prepend to the header inclusion path.
+    """
+    for hdr_label in hdr_labels:
+        hdr_filename = hdr_label.split(":")[-1]
+
+        # The last path component of iOS header files is sources/some_file.h
+        # Hence it wiill contain a '/'. So the string can be split at '/' to get 
+        # the header file name.
+        if "/" in hdr_filename:
+            hdr_filename = hdr_filename.split("/")[-1]
+
+        hdr_basename = hdr_filename.split(".")[0]
+        native.genrule(
+            name = "{}_{}".format(name, hdr_basename),
+            srcs = [hdr_label],
+            outs = [hdr_filename],
+            cmd = """
+            sed 's|#\\([a-z]*\\) ".*/\\([^/]\\{{1,\\}}\\.h\\)"|#\\1 "{}\\2"|'\
+            "$(location {})"\
+            > "$@"
+            """.format(prefix, hdr_label),
+        )