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