Enable/disable benchmarking through config

Signed-off-by: Victor Chang <vicchang@nvidia.com>

Author: mocsharp

applications/distributed/ucx/ucx_endoscopy_tool_tracking/cpp/endoscopy_tool_tracking.yaml

@@ -21,10 +21,17 @@ extensions:
   - libgxf_serialization.so
   - lib/gxf_extensions/libgxf_lstm_tensor_rt_inference.so
 
+application:
+  title: Endoscopy Tool Tracking - UCX
+  version: 1.0
+  inputFormats: []
+  outputFormats: ["screen"]
+  benchmarking: true # default: false, true to enable Data Flow Benchmarking, false otherwise
+
 replayer:
   basename: "surgical_video"
   frame_rate: 0   # as specified in timestamps
-  repeat: true    # default: false
+  repeat: false    # default: false
   realtime: true  # default: true
   count: 0        # default: 0 (no frame count restriction)
 

applications/distributed/ucx/ucx_endoscopy_tool_tracking/cpp/main.cpp

@@ -178,6 +178,23 @@ bool parse_arguments(int argc, char** argv, std::string& data_path, std::string&
   return true;
 }
 
+/** Helper function to parse fragment mode and benchmarking settings from the configuration file */
+void parse_config(const std::string& config_path, bool& benchmarking) {
+  auto config = holoscan::Config(config_path);
+  auto& yaml_nodes = config.yaml_nodes();
+  for (const auto& yaml_node : yaml_nodes) {
+    try {
+      auto application = yaml_node["application"];
+      if (application.IsMap()) {
+        benchmarking = application["benchmarking"].as<bool>();
+      }
+    } catch (std::exception& e) {
+      HOLOSCAN_LOG_ERROR("Error parsing configuration file: {}", e.what());
+      benchmarking = false;
+    }
+  }
+}
+
 /** Main function */
 int main(int argc, char** argv) {
   std::string config_path = "";
@@ -207,15 +224,25 @@ int main(int argc, char** argv) {
     }
   }
 
+  bool benchmarking = false;
+  parse_config(config_path, benchmarking);
+
   auto app = holoscan::make_application<App>();
   app->config(config_path);
   app->set_datapath(data_directory);
 
-  auto trackers = app->track_distributed();
+  std::unordered_map<std::string, DataFlowTracker*> trackers;
+  if (benchmarking) {
+    trackers = app->track_distributed();
+  }
+
   app->run();
-  for (const auto& [name, tracker] : trackers) {
-    std::cout << "Fragment: " << name << std::endl;
-    tracker->print();
+
+  if (benchmarking) {
+    for (const auto& [name, tracker] : trackers) {
+      std::cout << "Fragment: " << name << std::endl;
+      tracker->print();
+    }
   }
   return 0;
 }