From c3907229fed517abee19d6642f8cea68deda608d Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Thu, 9 Feb 2023 09:05:00 -0800
Subject: [PATCH] Use case: add metadata to headerless graph input audio
 streams.

PiperOrigin-RevId: 508389799
---
 mediapipe/calculators/core/BUILD                            | 2 ++
 .../calculators/core/constant_side_packet_calculator.cc     | 6 ++++++
 .../calculators/core/constant_side_packet_calculator.proto  | 2 ++
 3 files changed, 10 insertions(+)

diff --git a/mediapipe/calculators/core/BUILD b/mediapipe/calculators/core/BUILD
index ecfdd5d0b..fd9c4049b 100644
--- a/mediapipe/calculators/core/BUILD
+++ b/mediapipe/calculators/core/BUILD
@@ -117,6 +117,7 @@ mediapipe_proto_library(
         "//mediapipe/framework:calculator_proto",
         "//mediapipe/framework/formats:classification_proto",
         "//mediapipe/framework/formats:landmark_proto",
+        "//mediapipe/framework/formats:time_series_header_proto",
     ],
 )
 
@@ -1158,6 +1159,7 @@ cc_library(
         "//mediapipe/framework:collection_item_id",
         "//mediapipe/framework/formats:classification_cc_proto",
         "//mediapipe/framework/formats:landmark_cc_proto",
+        "//mediapipe/framework/formats:time_series_header_cc_proto",
         "//mediapipe/framework/port:integral_types",
         "//mediapipe/framework/port:ret_check",
         "//mediapipe/framework/port:status",
diff --git a/mediapipe/calculators/core/constant_side_packet_calculator.cc b/mediapipe/calculators/core/constant_side_packet_calculator.cc
index 54488fbfb..45ff07110 100644
--- a/mediapipe/calculators/core/constant_side_packet_calculator.cc
+++ b/mediapipe/calculators/core/constant_side_packet_calculator.cc
@@ -19,6 +19,7 @@
 #include "mediapipe/framework/collection_item_id.h"
 #include "mediapipe/framework/formats/classification.pb.h"
 #include "mediapipe/framework/formats/landmark.pb.h"
+#include "mediapipe/framework/formats/time_series_header.pb.h"
 #include "mediapipe/framework/port/canonical_errors.h"
 #include "mediapipe/framework/port/integral_types.h"
 #include "mediapipe/framework/port/ret_check.h"
@@ -84,6 +85,8 @@ class ConstantSidePacketCalculator : public CalculatorBase {
         packet.Set<LandmarkList>();
       } else if (packet_options.has_double_value()) {
         packet.Set<double>();
+      } else if (packet_options.has_time_series_header_value()) {
+        packet.Set<TimeSeriesHeader>();
       } else {
         return absl::InvalidArgumentError(
             "None of supported values were specified in options.");
@@ -118,6 +121,9 @@ class ConstantSidePacketCalculator : public CalculatorBase {
             MakePacket<LandmarkList>(packet_options.landmark_list_value()));
       } else if (packet_options.has_double_value()) {
         packet.Set(MakePacket<double>(packet_options.double_value()));
+      } else if (packet_options.has_time_series_header_value()) {
+        packet.Set(MakePacket<TimeSeriesHeader>(
+            packet_options.time_series_header_value()));
       } else {
         return absl::InvalidArgumentError(
             "None of supported values were specified in options.");
diff --git a/mediapipe/calculators/core/constant_side_packet_calculator.proto b/mediapipe/calculators/core/constant_side_packet_calculator.proto
index ec12d0115..78a773a6c 100644
--- a/mediapipe/calculators/core/constant_side_packet_calculator.proto
+++ b/mediapipe/calculators/core/constant_side_packet_calculator.proto
@@ -19,6 +19,7 @@ package mediapipe;
 import "mediapipe/framework/calculator.proto";
 import "mediapipe/framework/formats/classification.proto";
 import "mediapipe/framework/formats/landmark.proto";
+import "mediapipe/framework/formats/time_series_header.proto";
 
 message ConstantSidePacketCalculatorOptions {
   extend CalculatorOptions {
@@ -35,6 +36,7 @@ message ConstantSidePacketCalculatorOptions {
       ClassificationList classification_list_value = 6;
       LandmarkList landmark_list_value = 7;
       double double_value = 9;
+      TimeSeriesHeader time_series_header_value = 10;
     }
   }