350fbb2100
GitOrigin-RevId: d073f8e21be2fcc0e503cb97c6695078b6b75310
270 lines
6.6 KiB
Python
270 lines
6.6 KiB
Python
load("//mediapipe/framework/tool:mediapipe_graph.bzl", "mediapipe_binary_graph")
|
|
|
|
package(default_visibility = ["//visibility:private"])
|
|
|
|
licenses(["notice"])
|
|
|
|
cc_library(
|
|
name = "CFHolder",
|
|
# Header is excluded on non-ios so you can still build :all.
|
|
hdrs = select({
|
|
"//mediapipe:apple": ["CFHolder.h"],
|
|
"//conditions:default": [],
|
|
}),
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "util",
|
|
srcs = ["util.cc"],
|
|
hdrs = ["util.h"],
|
|
features = ["-parse_headers"],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
deps = [
|
|
":CFHolder",
|
|
"//mediapipe/framework:packet",
|
|
"//mediapipe/framework/formats:image_frame",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
objc_library(
|
|
name = "Weakify",
|
|
hdrs = ["Weakify.h"],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
)
|
|
|
|
MEDIAPIPE_IOS_SRCS = [
|
|
"MPPGraph.mm",
|
|
"MPPTimestampConverter.mm",
|
|
"NSError+util_status.mm",
|
|
]
|
|
|
|
MEDIAPIPE_IOS_HDRS = [
|
|
"MPPGraph.h",
|
|
"MPPTimestampConverter.h",
|
|
"NSError+util_status.h",
|
|
]
|
|
|
|
objc_library(
|
|
name = "mediapipe_framework_ios",
|
|
srcs = MEDIAPIPE_IOS_SRCS,
|
|
hdrs = MEDIAPIPE_IOS_HDRS,
|
|
copts = [
|
|
"-Wno-shorten-64-to-32",
|
|
"-std=c++17",
|
|
],
|
|
sdk_frameworks = [
|
|
# Needed for OpenCV.
|
|
"Accelerate",
|
|
],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
deps = [
|
|
":CFHolder",
|
|
":util",
|
|
"//mediapipe/framework:calculator_framework",
|
|
"//mediapipe/framework:mediapipe_profiling",
|
|
"//mediapipe/framework/port:map_util",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/port:statusor",
|
|
"//mediapipe/framework/port:threadpool",
|
|
"//mediapipe/gpu:MPPGraphGPUData",
|
|
"//mediapipe/gpu:gl_base",
|
|
"//mediapipe/gpu:gpu_buffer",
|
|
"//mediapipe/gpu:gpu_shared_data_internal",
|
|
"//mediapipe/gpu:graph_support",
|
|
"//mediapipe/gpu:pixel_buffer_pool_util",
|
|
"//mediapipe/util:cpu_util",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/synchronization",
|
|
"@google_toolbox_for_mac//:GTM_Defines",
|
|
],
|
|
)
|
|
|
|
objc_library(
|
|
name = "mediapipe_input_sources_ios",
|
|
srcs = [
|
|
"MPPInputSource.m",
|
|
"MPPPlayerInputSource.m",
|
|
] + select({
|
|
"//mediapipe:ios": [
|
|
"MPPCameraInputSource.m",
|
|
"MPPDisplayLinkWeakTarget.m",
|
|
],
|
|
"//conditions:default": [],
|
|
}),
|
|
hdrs = [
|
|
"MPPInputSource.h",
|
|
"MPPPlayerInputSource.h",
|
|
] + select({
|
|
"//mediapipe:ios": [
|
|
"MPPCameraInputSource.h",
|
|
"MPPDisplayLinkWeakTarget.h",
|
|
],
|
|
"//conditions:default": [],
|
|
}),
|
|
sdk_frameworks = [
|
|
"CoreVideo",
|
|
],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
)
|
|
|
|
objc_library(
|
|
name = "mediapipe_gl_view_renderer",
|
|
srcs = [
|
|
"MPPGLViewRenderer.mm",
|
|
],
|
|
hdrs = [
|
|
"MPPGLViewRenderer.h",
|
|
],
|
|
copts = [
|
|
"-Wno-shorten-64-to-32",
|
|
"-std=c++17",
|
|
],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
deps = [
|
|
":mediapipe_framework_ios",
|
|
"//mediapipe/gpu:gl_calculator_helper_ios",
|
|
"//mediapipe/gpu:gl_quad_renderer",
|
|
"//mediapipe/gpu:gl_simple_shaders",
|
|
],
|
|
)
|
|
|
|
objc_library(
|
|
name = "mediapipe_layer_renderer",
|
|
srcs = [
|
|
"MPPLayerRenderer.m",
|
|
],
|
|
hdrs = [
|
|
"MPPLayerRenderer.h",
|
|
],
|
|
copts = [
|
|
"-Wno-shorten-64-to-32",
|
|
],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
deps = [
|
|
":mediapipe_framework_ios",
|
|
":mediapipe_gl_view_renderer",
|
|
"//mediapipe/gpu:gl_calculator_helper_ios",
|
|
],
|
|
)
|
|
|
|
objc_library(
|
|
name = "CGImageRefUtils",
|
|
srcs = [
|
|
"CGImageRefUtils.mm",
|
|
],
|
|
hdrs = [
|
|
"CGImageRefUtils.h",
|
|
],
|
|
copts = [
|
|
"-Wno-shorten-64-to-32",
|
|
"-std=c++17",
|
|
],
|
|
sdk_frameworks = [
|
|
"CoreVideo",
|
|
],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
deps = [
|
|
":mediapipe_framework_ios",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
objc_library(
|
|
name = "MPPGraphTestBase",
|
|
testonly = 1,
|
|
srcs = [
|
|
"MPPGraphTestBase.mm",
|
|
],
|
|
hdrs = [
|
|
"MPPGraphTestBase.h",
|
|
],
|
|
copts = [
|
|
"-Wno-shorten-64-to-32",
|
|
"-std=c++17",
|
|
],
|
|
sdk_frameworks = [
|
|
"Accelerate",
|
|
"AVFoundation",
|
|
"CoreVideo",
|
|
"CoreGraphics",
|
|
"CoreMedia",
|
|
"GLKit",
|
|
"OpenGLES",
|
|
"QuartzCore",
|
|
"UIKit",
|
|
],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
deps = [
|
|
":CGImageRefUtils",
|
|
":Weakify",
|
|
":mediapipe_framework_ios",
|
|
"//mediapipe/framework:calculator_framework",
|
|
],
|
|
)
|
|
|
|
objc_library(
|
|
name = "mediapipe_framework_ios_testLib",
|
|
testonly = 1,
|
|
srcs = [
|
|
"CFHolderTests.mm",
|
|
"MPPGraphTests.mm",
|
|
],
|
|
copts = [
|
|
"-Wno-shorten-64-to-32",
|
|
"-std=c++17",
|
|
],
|
|
data = [
|
|
"testdata/googlelogo_color_272x92dp.png",
|
|
],
|
|
sdk_frameworks = [
|
|
"Accelerate",
|
|
"AVFoundation",
|
|
"CoreVideo",
|
|
"CoreGraphics",
|
|
"CoreMedia",
|
|
"GLKit",
|
|
"QuartzCore",
|
|
"UIKit",
|
|
],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
deps = [
|
|
":CGImageRefUtils",
|
|
":MPPGraphTestBase",
|
|
":Weakify",
|
|
":mediapipe_framework_ios",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
],
|
|
)
|
|
|
|
[
|
|
mediapipe_binary_graph(
|
|
name = graph.split("/")[-1].rsplit(".", 1)[0] + "_graph",
|
|
graph = graph,
|
|
output_name = "%s.binarypb" % graph.split("/")[-1].rsplit(".", 1)[0],
|
|
visibility = ["//mediapipe/framework:mediapipe_internal"],
|
|
)
|
|
for graph in glob(["testdata/*.pbtxt"])
|
|
]
|
|
|
|
exports_files(
|
|
[
|
|
"testdata/googlelogo_color_272x92dp.png",
|
|
"testdata/googlelogo_color_272x92dp_luminance.png",
|
|
"testdata/sergey.png",
|
|
],
|
|
visibility = [
|
|
"//mediapipe:__subpackages__",
|
|
],
|
|
)
|