Add helper to create Connection array

PiperOrigin-RevId: 565706612
google-ai-edge · Sep 15, 2023 · 30590fe · 30590fe
1 parent 4a8a811
commit 30590fe
Show file tree

Hide file tree

Showing 5 changed files with 470 additions and 948 deletions.
diff --git a/mediapipe/tasks/web/vision/core/BUILD b/mediapipe/tasks/web/vision/core/BUILD
@@ -1,7 +1,7 @@
 # This package contains options shared by all MediaPipe Vision Tasks for Web.
 
-load("@npm//@bazel/jasmine:index.bzl", "jasmine_node_test")
 load("//mediapipe/framework/port:build_config.bzl", "mediapipe_ts_declaration", "mediapipe_ts_library")
+load("@npm//@bazel/jasmine:index.bzl", "jasmine_node_test")
 
 package(default_visibility = ["//mediapipe/tasks:internal"])
 
@@ -21,9 +21,9 @@ mediapipe_ts_declaration(
     ],
 )
 
-mediapipe_ts_declaration(
+mediapipe_ts_library(
     name = "types",
-    srcs = ["types.d.ts"],
+    srcs = ["types.ts"],
     deps = [
         "//mediapipe/tasks/web/components/containers:keypoint",
     ],

diff --git a/mediapipe/tasks/web/vision/core/types.d.ts → mediapipe/tasks/web/vision/core/types.ts b/mediapipe/tasks/web/vision/core/types.d.ts → mediapipe/tasks/web/vision/core/types.ts
@@ -30,3 +30,9 @@ export declare interface Connection {
   start: number;
   end: number;
 }
+
+/** Converts a list of connection in array notation to a list of Connections. */
+export function convertToConnections(...connections: Array<[number, number]>):
+    Connection[] {
+  return connections.map(([start, end]) => ({start, end}));
+}