1700 lines
49 KiB
Python
1700 lines
49 KiB
Python
# Copyright 2019 The MediaPipe Authors.
|
|
#
|
|
# 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/framework/port:build_config.bzl", "mediapipe_proto_library")
|
|
load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
|
|
|
|
licenses(["notice"])
|
|
|
|
package(default_visibility = ["//visibility:private"])
|
|
|
|
# The MediaPipe internal package group. No mediapipe users should be added to this group.
|
|
package_group(
|
|
name = "mediapipe_internal",
|
|
packages = [
|
|
"//mediapipe/...",
|
|
],
|
|
)
|
|
|
|
bzl_library(
|
|
name = "transitive_protos_bzl",
|
|
srcs = [
|
|
"transitive_protos.bzl",
|
|
],
|
|
visibility = [
|
|
"//mediapipe/framework:__subpackages__",
|
|
],
|
|
)
|
|
|
|
bzl_library(
|
|
name = "encode_binary_proto_bzl",
|
|
srcs = [
|
|
"encode_binary_proto.bzl",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
alias(
|
|
name = "encode_binary_proto",
|
|
actual = ":encode_binary_proto_bzl",
|
|
deprecation = "Use encode_binary_proto_bzl",
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "calculator_proto",
|
|
srcs = ["calculator.proto"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_options_proto",
|
|
":mediapipe_options_proto",
|
|
":packet_factory_proto",
|
|
":packet_generator_proto",
|
|
":status_handler_proto",
|
|
":stream_handler_proto",
|
|
"@com_google_protobuf//:any_proto",
|
|
],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "calculator_options_proto",
|
|
srcs = ["calculator_options.proto"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "calculator_contract_test_proto",
|
|
testonly = 1,
|
|
srcs = ["calculator_contract_test.proto"],
|
|
visibility = ["//mediapipe/framework:__subpackages__"],
|
|
deps = [
|
|
":calculator_options_proto",
|
|
":calculator_proto",
|
|
],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "calculator_profile_proto",
|
|
srcs = ["calculator_profile.proto"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_options_proto",
|
|
":calculator_proto",
|
|
],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "mediapipe_options_proto",
|
|
srcs = ["mediapipe_options.proto"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "packet_factory_proto",
|
|
srcs = ["packet_factory.proto"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "packet_generator_proto",
|
|
srcs = ["packet_generator.proto"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "packet_test_proto",
|
|
testonly = 1,
|
|
srcs = ["packet_test.proto"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
"//mediapipe/framework:__subpackages__",
|
|
],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "status_handler_proto",
|
|
srcs = ["status_handler.proto"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [":mediapipe_options_proto"],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "stream_handler_proto",
|
|
srcs = ["stream_handler.proto"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [":mediapipe_options_proto"],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "test_calculators_proto",
|
|
testonly = 1,
|
|
srcs = ["test_calculators.proto"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":calculator_options_proto",
|
|
":calculator_proto",
|
|
],
|
|
)
|
|
|
|
mediapipe_proto_library(
|
|
name = "thread_pool_executor_proto",
|
|
srcs = ["thread_pool_executor.proto"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [":mediapipe_options_proto"],
|
|
)
|
|
|
|
# It is for pure-native Android builds where the library can't have any dependency on libandroid.so
|
|
config_setting(
|
|
name = "android_no_jni",
|
|
define_values = {"MEDIAPIPE_NO_JNI": "1"},
|
|
values = {"crosstool_top": "//external:android/crosstool"},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_base",
|
|
srcs = ["calculator_base.cc"],
|
|
hdrs = ["calculator_base.h"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
deps = [
|
|
":calculator_context",
|
|
":calculator_contract",
|
|
":port",
|
|
":timestamp",
|
|
"//mediapipe/framework/deps:registration",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_context",
|
|
srcs = ["calculator_context.cc"],
|
|
hdrs = ["calculator_context.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":calculator_state",
|
|
":counter",
|
|
":graph_service",
|
|
":input_stream_shard",
|
|
":output_stream_shard",
|
|
":packet",
|
|
":packet_set",
|
|
":port",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:any_proto",
|
|
"//mediapipe/framework/port:status",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_context_manager",
|
|
srcs = ["calculator_context_manager.cc"],
|
|
hdrs = ["calculator_context_manager.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":calculator_context",
|
|
":calculator_state",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:tag_map",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_contract",
|
|
srcs = ["calculator_contract.cc"],
|
|
hdrs = ["calculator_contract.h"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
deps = [
|
|
":calculator_cc_proto",
|
|
":graph_service",
|
|
":mediapipe_options_cc_proto",
|
|
":packet_generator_cc_proto",
|
|
":packet_type",
|
|
":port",
|
|
":status_handler_cc_proto",
|
|
"//mediapipe/framework/port:any_proto",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:options_map",
|
|
"//mediapipe/framework/tool:packet_generator_wrapper_calculator_cc_proto",
|
|
"//mediapipe/framework/tool:tag_map",
|
|
"@com_google_absl//absl/container:flat_hash_map",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "legacy_calculator_support",
|
|
srcs = ["legacy_calculator_support.cc"],
|
|
hdrs = ["legacy_calculator_support.h"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
deps = [
|
|
":calculator_context",
|
|
":calculator_contract",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_framework",
|
|
hdrs = ["calculator_framework.h"],
|
|
visibility = [
|
|
"//visibility:public",
|
|
],
|
|
deps = [
|
|
":calculator_base",
|
|
":calculator_graph",
|
|
":calculator_registry",
|
|
":counter_factory",
|
|
":input_stream",
|
|
":output_side_packet",
|
|
":output_stream",
|
|
":packet",
|
|
":packet_generator",
|
|
":packet_generator_graph",
|
|
":packet_set",
|
|
":packet_type",
|
|
":port",
|
|
":status_handler",
|
|
":subgraph",
|
|
":timestamp",
|
|
":validated_graph_config",
|
|
"//mediapipe/framework/tool:sink",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/framework/tool:validate",
|
|
"//mediapipe/framework/tool:validate_name",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_graph",
|
|
srcs = [
|
|
"calculator_graph.cc",
|
|
"scheduler.cc",
|
|
],
|
|
hdrs = [
|
|
"calculator_graph.h",
|
|
"scheduler.h",
|
|
],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
deps = [
|
|
":calculator_base",
|
|
":calculator_cc_proto",
|
|
":calculator_node",
|
|
":counter_factory",
|
|
":delegating_executor",
|
|
":executor",
|
|
":graph_output_stream",
|
|
":graph_service",
|
|
":graph_service_manager",
|
|
":input_stream_manager",
|
|
":mediapipe_profiling",
|
|
":output_side_packet_impl",
|
|
":output_stream",
|
|
":output_stream_manager",
|
|
":output_stream_poller",
|
|
":output_stream_shard",
|
|
":packet",
|
|
":packet_generator",
|
|
":packet_generator_cc_proto",
|
|
":packet_generator_graph",
|
|
":packet_set",
|
|
":packet_type",
|
|
":port",
|
|
":scheduler_queue",
|
|
":status_handler",
|
|
":status_handler_cc_proto",
|
|
":thread_pool_executor",
|
|
":thread_pool_executor_cc_proto",
|
|
":timestamp",
|
|
":validated_graph_config",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:fill_packet_set",
|
|
"//mediapipe/framework/tool:packet_generator_wrapper_calculator",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/framework/tool:tag_map",
|
|
"//mediapipe/framework/tool:validate",
|
|
"//mediapipe/framework/tool:validate_name",
|
|
"//mediapipe/gpu:gpu_service",
|
|
"//mediapipe/gpu:graph_support",
|
|
"//mediapipe/util:cpu_util",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/container:fixed_array",
|
|
"@com_google_absl//absl/container:flat_hash_map",
|
|
"@com_google_absl//absl/container:flat_hash_set",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/strings:str_format",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "graph_service_manager",
|
|
srcs = ["graph_service_manager.cc"],
|
|
hdrs = ["graph_service_manager.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":graph_service",
|
|
":packet",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "graph_service_manager_test",
|
|
srcs = ["graph_service_manager_test.cc"],
|
|
deps = [
|
|
":graph_service_manager",
|
|
":packet",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_node",
|
|
srcs = ["calculator_node.cc"],
|
|
hdrs = ["calculator_node.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":calculator_base",
|
|
":calculator_cc_proto",
|
|
":calculator_context",
|
|
":calculator_context_manager",
|
|
":calculator_state",
|
|
":counter_factory",
|
|
":input_side_packet_handler",
|
|
":input_stream_handler",
|
|
":input_stream_manager",
|
|
":legacy_calculator_support",
|
|
":mediapipe_profiling",
|
|
":output_side_packet_impl",
|
|
":output_stream_handler",
|
|
":output_stream_manager",
|
|
":packet",
|
|
":packet_set",
|
|
":packet_type",
|
|
":port",
|
|
":stream_handler_cc_proto",
|
|
":timestamp",
|
|
":validated_graph_config",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/stream_handler:default_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:in_order_output_stream_handler",
|
|
"//mediapipe/framework/tool:name_util",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/framework/tool:tag_map",
|
|
"//mediapipe/framework/tool:validate_name",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_registry",
|
|
hdrs = ["calculator_registry.h"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
deps = [
|
|
":calculator_base",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_runner",
|
|
testonly = 1,
|
|
srcs = ["calculator_runner.cc"],
|
|
hdrs = ["calculator_runner.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_framework",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:sink",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "calculator_state",
|
|
srcs = ["calculator_state.cc"],
|
|
hdrs = ["calculator_state.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":calculator_cc_proto",
|
|
":counter",
|
|
":counter_factory",
|
|
":graph_service",
|
|
":graph_service_manager",
|
|
":input_stream",
|
|
":output_stream",
|
|
":packet",
|
|
":packet_set",
|
|
":port",
|
|
"//mediapipe/framework/port:any_proto",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/tool:options_map",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "camera_intrinsics",
|
|
hdrs = ["camera_intrinsics.h"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "collection",
|
|
hdrs = ["collection.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":collection_item_id",
|
|
":type_map",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/tool:tag_map",
|
|
"//mediapipe/framework/tool:tag_map_helper",
|
|
"//mediapipe/framework/tool:validate_name",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "collection_item_id",
|
|
srcs = ["collection_item_id.cc"],
|
|
hdrs = ["collection_item_id.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"//mediapipe/framework/deps:intops",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "counter",
|
|
hdrs = ["counter.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = ["//mediapipe/framework/port:integral_types"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "counter_factory",
|
|
srcs = ["counter_factory.cc"],
|
|
hdrs = ["counter_factory.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":counter",
|
|
":port",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:map_util",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/synchronization",
|
|
"@com_google_absl//absl/time",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "delegating_executor",
|
|
srcs = ["delegating_executor.cc"],
|
|
hdrs = ["delegating_executor.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":executor",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "demangle",
|
|
hdrs = ["demangle.h"],
|
|
defines = select({
|
|
"//mediapipe/framework/profiler:android_release": [
|
|
"MEDIAPIPE_HAS_CXA_DEMANGLE=0",
|
|
],
|
|
"//conditions:default": [],
|
|
}),
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "mediapipe_profiling",
|
|
hdrs = [
|
|
"mediapipe_profiling.h",
|
|
"platform_specific_profiling.h",
|
|
],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
deps = [
|
|
"//mediapipe/framework/profiler:graph_profiler",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "executor",
|
|
srcs = ["executor.cc"],
|
|
hdrs = ["executor.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":mediapipe_options_cc_proto",
|
|
"//mediapipe/framework/deps:registration",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/port:statusor",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "graph_output_stream",
|
|
srcs = ["graph_output_stream.cc"],
|
|
hdrs = ["graph_output_stream.h"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
deps = [
|
|
":input_stream_handler",
|
|
":input_stream_manager",
|
|
":output_stream_manager",
|
|
":packet",
|
|
":packet_set",
|
|
":packet_type",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "graph_service",
|
|
hdrs = ["graph_service.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":packet",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/port:statusor",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "input_side_packet_handler",
|
|
srcs = ["input_side_packet_handler.cc"],
|
|
hdrs = ["input_side_packet_handler.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":collection_item_id",
|
|
":packet",
|
|
":packet_set",
|
|
":packet_type",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:fill_packet_set",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "input_stream",
|
|
hdrs = ["input_stream.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":packet",
|
|
":port",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "input_stream_handler",
|
|
srcs = ["input_stream_handler.cc"],
|
|
hdrs = ["input_stream_handler.h"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
"//research/interaction/mediapipe/calculators:__pkg__",
|
|
],
|
|
deps = [
|
|
":calculator_context",
|
|
":calculator_context_manager",
|
|
":collection",
|
|
":collection_item_id",
|
|
":input_stream_manager",
|
|
":input_stream_shard",
|
|
":mediapipe_options_cc_proto",
|
|
":mediapipe_profiling",
|
|
":packet",
|
|
":packet_set",
|
|
":packet_type",
|
|
"//mediapipe/framework/deps:registration",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:tag_map",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "input_stream_manager",
|
|
srcs = ["input_stream_manager.cc"],
|
|
hdrs = ["input_stream_manager.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":packet",
|
|
":packet_type",
|
|
":port",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "input_stream_shard",
|
|
srcs = ["input_stream_shard.cc"],
|
|
hdrs = ["input_stream_shard.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":input_stream",
|
|
":packet",
|
|
":packet_type",
|
|
":port",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "lifetime_tracker",
|
|
testonly = 1,
|
|
hdrs = ["lifetime_tracker.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "output_side_packet",
|
|
hdrs = ["output_side_packet.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":packet",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "output_side_packet_impl",
|
|
srcs = ["output_side_packet_impl.cc"],
|
|
hdrs = ["output_side_packet_impl.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":collection_item_id",
|
|
":input_side_packet_handler",
|
|
":output_side_packet",
|
|
":packet",
|
|
":packet_type",
|
|
":port",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "output_stream",
|
|
hdrs = ["output_stream.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":packet",
|
|
":port",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:logging",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "output_stream_handler",
|
|
srcs = ["output_stream_handler.cc"],
|
|
hdrs = ["output_stream_handler.h"],
|
|
visibility = [
|
|
":mediapipe_internal",
|
|
],
|
|
deps = [
|
|
":calculator_context_manager",
|
|
":collection",
|
|
":collection_item_id",
|
|
":mediapipe_options_cc_proto",
|
|
":output_stream_manager",
|
|
":output_stream_shard",
|
|
":packet_set",
|
|
":packet_type",
|
|
":timestamp",
|
|
"//mediapipe/framework/deps:registration",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:tag_map",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "output_stream_manager",
|
|
srcs = ["output_stream_manager.cc"],
|
|
hdrs = ["output_stream_manager.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":input_stream_handler",
|
|
":output_stream_shard",
|
|
":packet",
|
|
":packet_type",
|
|
":port",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "output_stream_poller",
|
|
hdrs = ["output_stream_poller.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":graph_output_stream",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "output_stream_shard",
|
|
srcs = ["output_stream_shard.cc"],
|
|
hdrs = ["output_stream_shard.h"],
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":output_stream",
|
|
":packet",
|
|
":packet_type",
|
|
":port",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
# Defines Packet, a data carrier used throughout the framework.
|
|
cc_library(
|
|
name = "packet",
|
|
srcs = ["packet.cc"],
|
|
hdrs = ["packet.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":port",
|
|
":timestamp",
|
|
":type_map",
|
|
"//mediapipe/framework/deps:no_destructor",
|
|
"//mediapipe/framework/deps:registration",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/port:statusor",
|
|
"//mediapipe/framework/tool:type_util",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "packet_generator",
|
|
hdrs = ["packet_generator.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":packet",
|
|
":packet_generator_cc_proto",
|
|
":packet_set",
|
|
":packet_type",
|
|
":port",
|
|
"//mediapipe/framework/deps:registration",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "packet_generator_graph",
|
|
srcs = ["packet_generator_graph.cc"],
|
|
hdrs = ["packet_generator_graph.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":delegating_executor",
|
|
":executor",
|
|
":packet",
|
|
":packet_factory_cc_proto",
|
|
":packet_generator",
|
|
":packet_generator_cc_proto",
|
|
":packet_type",
|
|
":port",
|
|
":thread_pool_executor",
|
|
":validated_graph_config",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "packet_set",
|
|
hdrs = ["packet_set.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":collection",
|
|
":packet",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "packet_type",
|
|
srcs = ["packet_type.cc"],
|
|
hdrs = ["packet_type.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":collection",
|
|
":packet",
|
|
":packet_set",
|
|
":type_map",
|
|
"//mediapipe/framework/deps:no_destructor",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:map_util",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/framework/tool:type_util",
|
|
"//mediapipe/framework/tool:validate_name",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/types:span",
|
|
"@com_google_absl//absl/types:variant",
|
|
],
|
|
)
|
|
|
|
# When --copt=-fno-rtti is set, MEDIAPIPE_HAS_RTTI is cleared in port.h.
|
|
# To explicitly clear MEDIAPIPE_HAS_RTTI, compile with:
|
|
# bazel build --define=disable_rtti_and_exceptions=true
|
|
config_setting(
|
|
name = "disable_rtti_and_exceptions",
|
|
define_values = {"disable_rtti_and_exceptions": "true"},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "port",
|
|
hdrs = ["port.h"],
|
|
defines = select({
|
|
"//conditions:default": [],
|
|
}) + select({
|
|
"//conditions:default": [],
|
|
"//mediapipe/gpu:disable_gpu": ["MEDIAPIPE_DISABLE_GPU=1"],
|
|
}) + select({
|
|
"//conditions:default": [],
|
|
"//mediapipe/framework/port:disable_opencv": ["MEDIAPIPE_DISABLE_OPENCV=1"],
|
|
}) + select({
|
|
"//conditions:default": [],
|
|
# TODO: Improve this. This only sets MEDIAPIPE_DISABLE_OPENCV as a "defines" Make
|
|
# value, not as a bazel "--define" variable, which has effects in C++ code but not in
|
|
# select() statements.
|
|
"//mediapipe/framework/port:enable_halide": [
|
|
"MEDIAPIPE_ENABLE_HALIDE=1",
|
|
"MEDIAPIPE_DISABLE_OPENCV=1",
|
|
],
|
|
}) + select({
|
|
"//conditions:default": [],
|
|
"//mediapipe/framework:disable_rtti_and_exceptions": [
|
|
"MEDIAPIPE_HAS_RTTI=0",
|
|
],
|
|
}),
|
|
visibility = [
|
|
"//fitbit/research/sensing/mobisense:__subpackages__",
|
|
"//mediapipe/calculators:__subpackages__",
|
|
"//mediapipe/framework:__subpackages__",
|
|
"//mediapipe/framework/port:__pkg__",
|
|
"//mediapipe/gpu:__pkg__",
|
|
"//mediapipe/util:__subpackages__",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "scheduler_queue",
|
|
srcs = ["scheduler_queue.cc"],
|
|
hdrs = [
|
|
"scheduler_queue.h",
|
|
"scheduler_shared.h",
|
|
],
|
|
copts = select({
|
|
"//conditions:default": [],
|
|
"//mediapipe:apple": [
|
|
"-ObjC++",
|
|
],
|
|
}),
|
|
visibility = [":mediapipe_internal"],
|
|
deps = [
|
|
":calculator_context",
|
|
":calculator_node",
|
|
":executor",
|
|
"//mediapipe/framework/deps:clock",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "status_handler",
|
|
hdrs = ["status_handler.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":mediapipe_options_cc_proto",
|
|
":packet_set",
|
|
":packet_type",
|
|
":port",
|
|
"//mediapipe/framework/deps:registration",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "subgraph",
|
|
srcs = ["subgraph.cc"],
|
|
hdrs = ["subgraph.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_cc_proto",
|
|
":graph_service",
|
|
":graph_service_manager",
|
|
":port",
|
|
"//mediapipe/framework/deps:registration",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/port:statusor",
|
|
"//mediapipe/framework/tool:calculator_graph_template_cc_proto",
|
|
"//mediapipe/framework/tool:options_util",
|
|
"//mediapipe/framework/tool:template_expander",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/types:optional",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "test_calculators",
|
|
testonly = 1,
|
|
srcs = ["test_calculators.cc"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_framework",
|
|
":test_calculators_cc_proto",
|
|
"//mediapipe/framework/deps:mathutil",
|
|
"//mediapipe/framework/formats:matrix",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
"@eigen_archive//:eigen3",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
cc_library(
|
|
name = "test_service",
|
|
testonly = 1,
|
|
srcs = ["test_service.cc"],
|
|
hdrs = ["test_service.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_contract",
|
|
":calculator_framework",
|
|
":graph_service",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
],
|
|
alwayslink = True, # Defines TestServiceCalculator
|
|
)
|
|
|
|
cc_library(
|
|
name = "thread_pool_executor",
|
|
srcs = ["thread_pool_executor.cc"],
|
|
hdrs = ["thread_pool_executor.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":executor",
|
|
":thread_pool_executor_cc_proto",
|
|
"//mediapipe/framework/deps:thread_options",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/port:statusor",
|
|
"//mediapipe/framework/port:threadpool",
|
|
"//mediapipe/util:cpu_util",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "timestamp",
|
|
srcs = ["timestamp.cc"],
|
|
hdrs = ["timestamp.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"//mediapipe/framework/deps:intops",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "throttler",
|
|
hdrs = ["throttler.h"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "type_map",
|
|
hdrs = ["type_map.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":demangle",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/framework/tool:type_util",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
cc_library(
|
|
name = "basic_types_registration",
|
|
srcs = ["basic_types_registration.cc"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":type_map",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
cc_library(
|
|
name = "validated_graph_config",
|
|
srcs = ["validated_graph_config.cc"],
|
|
hdrs = ["validated_graph_config.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_base",
|
|
":calculator_cc_proto",
|
|
":calculator_contract",
|
|
":graph_service_manager",
|
|
":legacy_calculator_support",
|
|
":packet",
|
|
":packet_generator",
|
|
":packet_generator_cc_proto",
|
|
":packet_set",
|
|
":packet_type",
|
|
":port",
|
|
":status_handler",
|
|
":status_handler_cc_proto",
|
|
":stream_handler_cc_proto",
|
|
":subgraph",
|
|
":thread_pool_executor_cc_proto",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:map_util",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:source_location",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/port:topologicalsorter",
|
|
"//mediapipe/framework/tool:name_util",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/framework/tool:subgraph_expansion",
|
|
"//mediapipe/framework/tool:validate",
|
|
"//mediapipe/framework/tool:validate_name",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/container:flat_hash_set",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "validated_graph_config_test",
|
|
srcs = ["validated_graph_config_test.cc"],
|
|
deps = [
|
|
":calculator_cc_proto",
|
|
":calculator_framework",
|
|
":graph_service",
|
|
":graph_service_manager",
|
|
":validated_graph_config",
|
|
"//mediapipe/framework/api2:node",
|
|
"//mediapipe/framework/api2:port",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "graph_validation",
|
|
hdrs = ["graph_validation.h"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_framework",
|
|
"//mediapipe/framework/port:status",
|
|
],
|
|
)
|
|
|
|
# cc tests
|
|
cc_test(
|
|
name = "calculator_base_test",
|
|
size = "medium",
|
|
srcs = ["calculator_base_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":calculator_base",
|
|
":calculator_cc_proto",
|
|
":calculator_context",
|
|
":calculator_context_manager",
|
|
":calculator_registry",
|
|
":calculator_state",
|
|
":output_stream",
|
|
":output_stream_manager",
|
|
":output_stream_shard",
|
|
":packet_set",
|
|
":packet_type",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/framework/tool:tag_map_helper",
|
|
"@com_google_absl//absl/container:flat_hash_set",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_contract_test",
|
|
srcs = ["calculator_contract_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":calculator_cc_proto",
|
|
":calculator_contract",
|
|
":calculator_contract_test_cc_proto",
|
|
":packet_generator_cc_proto",
|
|
":status_handler_cc_proto",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_node_test",
|
|
size = "small",
|
|
srcs = ["calculator_node_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":calculator_framework",
|
|
":calculator_node",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:source",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_graph_event_loop_test",
|
|
size = "small",
|
|
srcs = ["calculator_graph_event_loop_test.cc"],
|
|
deps = [
|
|
":calculator_framework",
|
|
":calculator_graph",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:sink",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/synchronization",
|
|
"@com_google_absl//absl/time",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_graph_stopping_test",
|
|
size = "small",
|
|
srcs = ["calculator_graph_stopping_test.cc"],
|
|
deps = [
|
|
":calculator_framework",
|
|
":calculator_graph",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:sink",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/time",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_parallel_execution_test",
|
|
srcs = ["calculator_parallel_execution_test.cc"],
|
|
deps = [
|
|
":calculator_framework",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:integral_types",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:sink",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/synchronization",
|
|
"@com_google_absl//absl/time",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_runner_test",
|
|
size = "medium",
|
|
srcs = ["calculator_runner_test.cc"],
|
|
deps = [
|
|
":calculator_base",
|
|
":calculator_registry",
|
|
":calculator_runner",
|
|
":input_stream",
|
|
":output_stream",
|
|
":packet_type",
|
|
":timestamp",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:status",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_context_test",
|
|
size = "medium",
|
|
srcs = ["calculator_context_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":calculator_cc_proto",
|
|
":calculator_context",
|
|
":calculator_context_manager",
|
|
":calculator_state",
|
|
":output_stream",
|
|
":output_stream_manager",
|
|
":output_stream_shard",
|
|
":packet_set",
|
|
":packet_type",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/testdata:night_light_calculator_cc_proto",
|
|
"//mediapipe/framework/testdata:sky_light_calculator_cc_proto",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/framework/tool:tag_map_helper",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_graph_test",
|
|
size = "small",
|
|
srcs = [
|
|
"calculator_graph_test.cc",
|
|
],
|
|
linkstatic = 1,
|
|
tags = ["not_run:arm"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_framework",
|
|
":calculator_graph",
|
|
":collection_item_id",
|
|
":counter_factory",
|
|
":executor",
|
|
":input_stream_handler",
|
|
":lifetime_tracker",
|
|
":mediapipe_options_cc_proto",
|
|
":output_stream_poller",
|
|
":packet_set",
|
|
":packet_type",
|
|
":status_handler",
|
|
":subgraph",
|
|
":test_calculators",
|
|
":thread_pool_executor",
|
|
":thread_pool_executor_cc_proto",
|
|
":timestamp",
|
|
":type_map",
|
|
"//mediapipe/calculators/core:counting_source_calculator",
|
|
"//mediapipe/calculators/core:mux_calculator",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/stream_handler:barrier_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:early_close_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:fixed_size_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:immediate_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:mux_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:sync_set_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:timestamp_align_input_stream_handler",
|
|
"//mediapipe/framework/tool:sink",
|
|
"//mediapipe/framework/tool:status_util",
|
|
"//mediapipe/gpu:gpu_service",
|
|
"@com_google_absl//absl/container:fixed_array",
|
|
"@com_google_absl//absl/memory",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/strings:str_format",
|
|
"@com_google_absl//absl/time",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_graph_bounds_test",
|
|
size = "small",
|
|
srcs = [
|
|
"calculator_graph_bounds_test.cc",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_context",
|
|
":calculator_framework",
|
|
":test_calculators",
|
|
":thread_pool_executor",
|
|
":timestamp",
|
|
":type_map",
|
|
"//mediapipe/calculators/core:counting_source_calculator",
|
|
"//mediapipe/calculators/core:mux_calculator",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/stream_handler:barrier_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:early_close_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:fixed_size_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:immediate_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:mux_input_stream_handler",
|
|
"//mediapipe/framework/stream_handler:sync_set_input_stream_handler",
|
|
"//mediapipe/framework/tool:sink",
|
|
"//mediapipe/util:packet_test_util",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "calculator_graph_side_packet_test",
|
|
size = "small",
|
|
srcs = [
|
|
"calculator_graph_side_packet_test.cc",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_cc_proto",
|
|
":calculator_framework",
|
|
":test_calculators",
|
|
"//mediapipe/calculators/core:counting_source_calculator",
|
|
"//mediapipe/calculators/core:mux_calculator",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:sink",
|
|
"@com_google_absl//absl/time",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "collection_test",
|
|
size = "small",
|
|
srcs = ["collection_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":collection",
|
|
":packet_set",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/tool:tag_map_helper",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "graph_service_test",
|
|
size = "small",
|
|
srcs = [
|
|
"graph_service_test.cc",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":calculator_contract",
|
|
":calculator_framework",
|
|
":graph_service",
|
|
":test_service",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:logging",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/port:ret_check",
|
|
"//mediapipe/framework/port:status",
|
|
"//mediapipe/framework/port:status_matchers",
|
|
"//mediapipe/framework/tool:sink",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "input_stream_manager_test",
|
|
size = "small",
|
|
srcs = ["input_stream_manager_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":input_stream_manager",
|
|
":input_stream_shard",
|
|
":lifetime_tracker",
|
|
":packet",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "output_stream_manager_test",
|
|
size = "small",
|
|
srcs = ["output_stream_manager_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":input_stream_handler",
|
|
":input_stream_manager",
|
|
":output_stream_manager",
|
|
":output_stream_shard",
|
|
":packet",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/stream_handler:default_input_stream_handler",
|
|
"//mediapipe/framework/tool:tag_map_helper",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "packet_delete_test",
|
|
size = "small",
|
|
srcs = ["packet_delete_test.cc"],
|
|
copts = [
|
|
"-Werror",
|
|
],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":packet",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "executor_external_build_test",
|
|
size = "small",
|
|
srcs = ["executor_external_build_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":executor",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:threadpool",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "packet_test",
|
|
size = "medium",
|
|
srcs = ["packet_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":packet",
|
|
":packet_test_cc_proto",
|
|
":type_map",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "packet_registration_test",
|
|
size = "small",
|
|
srcs = ["packet_registration_test.cc"],
|
|
deps = [
|
|
":calculator_framework",
|
|
":packet",
|
|
":packet_test_cc_proto",
|
|
":type_map",
|
|
"//mediapipe/framework/port:core_proto",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "packet_generator_test",
|
|
size = "small",
|
|
srcs = ["packet_generator_test.cc"],
|
|
deps = [
|
|
":packet_generator",
|
|
":packet_generator_cc_proto",
|
|
":packet_type",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/tool:validate_type",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "timestamp_test",
|
|
size = "small",
|
|
srcs = ["timestamp_test.cc"],
|
|
linkstatic = 1,
|
|
deps = [
|
|
":timestamp",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "graph_validation_test",
|
|
srcs = ["graph_validation_test.cc"],
|
|
deps = [
|
|
":calculator_contract_test_cc_proto",
|
|
":calculator_framework",
|
|
":graph_validation",
|
|
"//mediapipe/calculators/core:constant_side_packet_calculator",
|
|
"//mediapipe/calculators/core:default_side_packet_calculator",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/tool:template_parser",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "subgraph_test",
|
|
srcs = ["subgraph_test.cc"],
|
|
deps = [
|
|
":calculator_framework",
|
|
":graph_service_manager",
|
|
":subgraph",
|
|
":test_calculators",
|
|
"//mediapipe/calculators/core:constant_side_packet_calculator",
|
|
"//mediapipe/calculators/core:constant_side_packet_calculator_cc_proto",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/framework/port:gtest_main",
|
|
"//mediapipe/framework/port:parse_text_proto",
|
|
"//mediapipe/framework/tool:sink",
|
|
"//mediapipe/framework/tool/testdata:dub_quad_test_subgraph",
|
|
"@com_google_absl//absl/strings:str_format",
|
|
],
|
|
)
|