Project import generated by Copybara.

PiperOrigin-RevId: 256060078

mediapipe/framework/calculator_runner.cc

@@ -216,6 +216,10 @@ mediapipe::Counter* CalculatorRunner::GetCounter(const std::string& name) {
   return graph_->GetCounterFactory()->GetCounter(name);
 }
 
+std::map<std::string, int64> CalculatorRunner::GetCountersValues() {
+  return graph_->GetCounterFactory()->GetCounterSet()->GetCountersValues();
+}
+
 ::mediapipe::Status CalculatorRunner::BuildGraph() {
   if (graph_ != nullptr) {
     // The graph was already built.

mediapipe/framework/calculator_runner.h

@@ -127,6 +127,9 @@ class CalculatorRunner {
   // Returns a graph counter.
   mediapipe::Counter* GetCounter(const std::string& name);
 
+  // Returns all graph counters values.
+  std::map<std::string, int64> GetCountersValues();
+
  private:
   static const char kSourcePrefix[];
   static const char kSinkPrefix[];

mediapipe/framework/counter_factory.cc

@@ -73,6 +73,16 @@ Counter* CounterSet::Get(const std::string& name) LOCKS_EXCLUDED(mu_) {
   return counters_[name].get();
 }
 
+std::map<std::string, int64> CounterSet::GetCountersValues()
+    LOCKS_EXCLUDED(mu_) {
+  absl::ReaderMutexLock lock(&mu_);
+  std::map<std::string, int64> result;
+  for (const auto& it : counters_) {
+    result[it.first] = it.second->Get();
+  }
+  return result;
+}
+
 Counter* BasicCounterFactory::GetCounter(const std::string& name) {
   return counter_set_.Emplace<BasicCounter>(name, name);
 }

mediapipe/framework/counter_factory.h

@@ -65,6 +65,9 @@ class CounterSet {
   // exist.
   Counter* Get(const std::string& name);
 
+  // Retrieves all counters names and current values from the internal map.
+  std::map<std::string, int64> GetCountersValues() LOCKS_EXCLUDED(mu_);
+
  private:
   absl::Mutex mu_;
   std::map<std::string, std::unique_ptr<Counter>> counters_ GUARDED_BY(mu_);