Changes to create MediaPipe Tasks Audio AAR.
PiperOrigin-RevId: 487661296
This commit is contained in:
parent
d6ba5e4bbe
commit
b3ac1c1387
|
@ -25,7 +25,7 @@ android_library(
|
||||||
deps = [
|
deps = [
|
||||||
":libmediapipe_tasks_audio_jni_lib",
|
":libmediapipe_tasks_audio_jni_lib",
|
||||||
"//mediapipe/java/com/google/mediapipe/framework:android_framework_no_mff",
|
"//mediapipe/java/com/google/mediapipe/framework:android_framework_no_mff",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:audio_data",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:audiodata",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/core",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/core",
|
||||||
"@maven//:com_google_guava_guava",
|
"@maven//:com_google_guava_guava",
|
||||||
],
|
],
|
||||||
|
@ -66,7 +66,7 @@ android_library(
|
||||||
"//mediapipe/tasks/cc/audio/audio_classifier/proto:audio_classifier_graph_options_java_proto_lite",
|
"//mediapipe/tasks/cc/audio/audio_classifier/proto:audio_classifier_graph_options_java_proto_lite",
|
||||||
"//mediapipe/tasks/cc/components/containers/proto:classifications_java_proto_lite",
|
"//mediapipe/tasks/cc/components/containers/proto:classifications_java_proto_lite",
|
||||||
"//mediapipe/tasks/cc/core/proto:base_options_java_proto_lite",
|
"//mediapipe/tasks/cc/core/proto:base_options_java_proto_lite",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:audio_data",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:audiodata",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:classificationresult",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:classificationresult",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/processors:classifieroptions",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/processors:classifieroptions",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/core",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/core",
|
||||||
|
@ -74,3 +74,11 @@ android_library(
|
||||||
"@maven//:com_google_guava_guava",
|
"@maven//:com_google_guava_guava",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
load("//mediapipe/tasks/java/com/google/mediapipe/tasks:mediapipe_tasks_aar.bzl", "mediapipe_tasks_audio_aar")
|
||||||
|
|
||||||
|
mediapipe_tasks_audio_aar(
|
||||||
|
name = "tasks_audio",
|
||||||
|
srcs = glob(["**/*.java"]),
|
||||||
|
native_library = ":libmediapipe_tasks_audio_jni_lib",
|
||||||
|
)
|
||||||
|
|
|
@ -188,7 +188,7 @@ public final class AudioClassifier extends BaseAudioTaskApi {
|
||||||
});
|
});
|
||||||
if (options.resultListener().isPresent()) {
|
if (options.resultListener().isPresent()) {
|
||||||
ResultListener<AudioClassifierResult, Void> resultListener =
|
ResultListener<AudioClassifierResult, Void> resultListener =
|
||||||
new ResultListener<>() {
|
new ResultListener<AudioClassifierResult, Void>() {
|
||||||
@Override
|
@Override
|
||||||
public void run(AudioClassifierResult audioClassifierResult, Void input) {
|
public void run(AudioClassifierResult audioClassifierResult, Void input) {
|
||||||
options.resultListener().get().run(audioClassifierResult);
|
options.resultListener().get().run(audioClassifierResult);
|
||||||
|
|
|
@ -17,7 +17,7 @@ package(default_visibility = ["//mediapipe/tasks:internal"])
|
||||||
licenses(["notice"])
|
licenses(["notice"])
|
||||||
|
|
||||||
android_library(
|
android_library(
|
||||||
name = "audio_data",
|
name = "audiodata",
|
||||||
srcs = ["AudioData.java"],
|
srcs = ["AudioData.java"],
|
||||||
deps = [
|
deps = [
|
||||||
"//third_party:autovalue",
|
"//third_party:autovalue",
|
||||||
|
|
|
@ -24,3 +24,10 @@ android_library(
|
||||||
"@maven//:com_google_guava_guava",
|
"@maven//:com_google_guava_guava",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Expose the java source files for building mediapipe tasks core AAR.
|
||||||
|
filegroup(
|
||||||
|
name = "java_src",
|
||||||
|
srcs = glob(["*.java"]),
|
||||||
|
visibility = ["//mediapipe/tasks/java/com/google/mediapipe/tasks/core:__subpackages__"],
|
||||||
|
)
|
||||||
|
|
|
@ -44,6 +44,7 @@ mediapipe_tasks_core_aar(
|
||||||
srcs = glob(["*.java"]) + [
|
srcs = glob(["*.java"]) + [
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:java_src",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:java_src",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/processors:java_src",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/processors:java_src",
|
||||||
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/utils:java_src",
|
||||||
"//mediapipe/java/com/google/mediapipe/framework/image:java_src",
|
"//mediapipe/java/com/google/mediapipe/framework/image:java_src",
|
||||||
],
|
],
|
||||||
manifest = "AndroidManifest.xml",
|
manifest = "AndroidManifest.xml",
|
||||||
|
|
|
@ -30,6 +30,10 @@ _CORE_TASKS_JAVA_PROTO_LITE_TARGETS = [
|
||||||
"//mediapipe/tasks/cc/core/proto:external_file_java_proto_lite",
|
"//mediapipe/tasks/cc/core/proto:external_file_java_proto_lite",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
_AUDIO_TASKS_JAVA_PROTO_LITE_TARGETS = [
|
||||||
|
"//mediapipe/tasks/cc/audio/audio_classifier/proto:audio_classifier_graph_options_java_proto_lite",
|
||||||
|
]
|
||||||
|
|
||||||
_VISION_TASKS_JAVA_PROTO_LITE_TARGETS = [
|
_VISION_TASKS_JAVA_PROTO_LITE_TARGETS = [
|
||||||
"//mediapipe/tasks/cc/vision/object_detector/proto:object_detector_options_java_proto_lite",
|
"//mediapipe/tasks/cc/vision/object_detector/proto:object_detector_options_java_proto_lite",
|
||||||
"//mediapipe/tasks/cc/vision/image_classifier/proto:image_classifier_graph_options_java_proto_lite",
|
"//mediapipe/tasks/cc/vision/image_classifier/proto:image_classifier_graph_options_java_proto_lite",
|
||||||
|
@ -62,6 +66,11 @@ def mediapipe_tasks_core_aar(name, srcs, manifest):
|
||||||
_mediapipe_tasks_java_proto_src_extractor(target = target),
|
_mediapipe_tasks_java_proto_src_extractor(target = target),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
for target in _AUDIO_TASKS_JAVA_PROTO_LITE_TARGETS:
|
||||||
|
mediapipe_tasks_java_proto_srcs.append(
|
||||||
|
_mediapipe_tasks_java_proto_src_extractor(target = target),
|
||||||
|
)
|
||||||
|
|
||||||
for target in _VISION_TASKS_JAVA_PROTO_LITE_TARGETS:
|
for target in _VISION_TASKS_JAVA_PROTO_LITE_TARGETS:
|
||||||
mediapipe_tasks_java_proto_srcs.append(
|
mediapipe_tasks_java_proto_srcs.append(
|
||||||
_mediapipe_tasks_java_proto_src_extractor(target = target),
|
_mediapipe_tasks_java_proto_src_extractor(target = target),
|
||||||
|
@ -119,9 +128,43 @@ def mediapipe_tasks_core_aar(name, srcs, manifest):
|
||||||
"@maven//:com_google_flogger_flogger_system_backend",
|
"@maven//:com_google_flogger_flogger_system_backend",
|
||||||
"@maven//:com_google_code_findbugs_jsr305",
|
"@maven//:com_google_code_findbugs_jsr305",
|
||||||
] +
|
] +
|
||||||
|
_AUDIO_TASKS_JAVA_PROTO_LITE_TARGETS +
|
||||||
_CORE_TASKS_JAVA_PROTO_LITE_TARGETS +
|
_CORE_TASKS_JAVA_PROTO_LITE_TARGETS +
|
||||||
_VISION_TASKS_JAVA_PROTO_LITE_TARGETS +
|
_TEXT_TASKS_JAVA_PROTO_LITE_TARGETS +
|
||||||
_TEXT_TASKS_JAVA_PROTO_LITE_TARGETS,
|
_VISION_TASKS_JAVA_PROTO_LITE_TARGETS,
|
||||||
|
)
|
||||||
|
|
||||||
|
def mediapipe_tasks_audio_aar(name, srcs, native_library):
|
||||||
|
"""Builds medaipipe tasks audio AAR.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
name: The bazel target name.
|
||||||
|
srcs: MediaPipe Audio Tasks' source files.
|
||||||
|
native_library: The native library that contains audio tasks' graph and calculators.
|
||||||
|
"""
|
||||||
|
|
||||||
|
native.genrule(
|
||||||
|
name = name + "tasks_manifest_generator",
|
||||||
|
outs = ["AndroidManifest.xml"],
|
||||||
|
cmd = """
|
||||||
|
cat > $(OUTS) <<EOF
|
||||||
|
<?xml version="1.0" encoding="utf-8"?>
|
||||||
|
<manifest xmlns:android="http://schemas.android.com/apk/res/android"
|
||||||
|
package="com.google.mediapipe.tasks.audio">
|
||||||
|
<uses-sdk
|
||||||
|
android:minSdkVersion="24"
|
||||||
|
android:targetSdkVersion="30" />
|
||||||
|
</manifest>
|
||||||
|
EOF
|
||||||
|
""",
|
||||||
|
)
|
||||||
|
|
||||||
|
_mediapipe_tasks_aar(
|
||||||
|
name = name,
|
||||||
|
srcs = srcs,
|
||||||
|
manifest = "AndroidManifest.xml",
|
||||||
|
java_proto_lite_targets = _CORE_TASKS_JAVA_PROTO_LITE_TARGETS + _AUDIO_TASKS_JAVA_PROTO_LITE_TARGETS,
|
||||||
|
native_library = native_library,
|
||||||
)
|
)
|
||||||
|
|
||||||
def mediapipe_tasks_vision_aar(name, srcs, native_library):
|
def mediapipe_tasks_vision_aar(name, srcs, native_library):
|
||||||
|
@ -232,6 +275,7 @@ def _mediapipe_tasks_aar(name, srcs, manifest, java_proto_lite_targets, native_l
|
||||||
"//mediapipe/framework/formats:landmark_java_proto_lite",
|
"//mediapipe/framework/formats:landmark_java_proto_lite",
|
||||||
"//mediapipe/framework/formats:location_data_java_proto_lite",
|
"//mediapipe/framework/formats:location_data_java_proto_lite",
|
||||||
"//mediapipe/framework/formats:rect_java_proto_lite",
|
"//mediapipe/framework/formats:rect_java_proto_lite",
|
||||||
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:audiodata",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:detection",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:detection",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:category",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:category",
|
||||||
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:classificationresult",
|
"//mediapipe/tasks/java/com/google/mediapipe/tasks/components/containers:classificationresult",
|
||||||
|
|
Loading…
Reference in New Issue
Block a user