Project import generated by Copybara.

GitOrigin-RevId: 5b23708185311ae39a8605b0c2eff721e7b4939f

mediapipe/framework/stream_handler/immediate_input_stream_handler.cc

@@ -56,6 +56,9 @@ class ImmediateInputStreamHandler : public InputStreamHandler {
   void FillInputSet(Timestamp input_timestamp,
                     InputStreamShardSet* input_set) override;
 
+  // Returns the number of sync-sets maintained by this input-handler.
+  int SyncSetCount() override;
+
   absl::Mutex mutex_;
   // The packet-set builder for each input stream.
   std::vector<SyncSet> sync_sets_ ABSL_GUARDED_BY(mutex_);
@@ -169,4 +172,9 @@ void ImmediateInputStreamHandler::FillInputSet(Timestamp input_timestamp,
   }
 }
 
+int ImmediateInputStreamHandler::SyncSetCount() {
+  absl::MutexLock lock(&mutex_);
+  return sync_sets_.size();
+}
+
 }  // namespace mediapipe

mediapipe/framework/stream_handler/sync_set_input_stream_handler.cc

@@ -67,6 +67,9 @@ class SyncSetInputStreamHandler : public InputStreamHandler {
                        InputStreamShardSet* input_set)
       ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_);
 
+  // Returns the number of sync-sets maintained by this input-handler.
+  int SyncSetCount() override;
+
  private:
   absl::Mutex mutex_;
   // The ids of each set of inputs.
@@ -194,4 +197,9 @@ void SyncSetInputStreamHandler::FillInputSet(Timestamp input_timestamp,
   ready_timestamp_ = Timestamp::Done();
 }
 
+int SyncSetInputStreamHandler::SyncSetCount() {
+  absl::MutexLock lock(&mutex_);
+  return sync_sets_.size();
+}
+
 }  // namespace mediapipe