From 05f54bdd5cd36565b41da7753f43b436e2efb2b4 Mon Sep 17 00:00:00 2001
From: Hrishikesh Gohain <hrishikeshgohain123@gmail.com>
Date: Sun, 29 Mar 2026 01:46:19 +0530
Subject: [PATCH] feat: support input sam files for conversion time benchmarks

currently synthetic SAM files are generated for conversion benchmark
Now we will be able to give our own input sam files ( real data ) as args and get the benchmark results

Signed-off-by: Hrishikesh Gohain <hrishikeshgohain123@gmail.com>
---
 benchmark/conversion_time_benchmark.cxx | 60 +++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/benchmark/conversion_time_benchmark.cxx b/benchmark/conversion_time_benchmark.cxx
index 80f77b0..2b3e19b 100644
--- a/benchmark/conversion_time_benchmark.cxx
+++ b/benchmark/conversion_time_benchmark.cxx
@@ -5,6 +5,8 @@
 #include <filesystem>
 #include <iostream>
 #include <cstdio>
+#include <vector>
+#include <string>
 
 #ifdef _WIN32
 #define NULL_DEVICE "NUL"
@@ -12,6 +14,9 @@
 #define NULL_DEVICE "/dev/null"
 #endif
 
+// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
+static std::vector<std::string> input_files;
+
 static void BM_SamToTTree(benchmark::State &state)
 {
    int num_reads = state.range(0);
@@ -72,8 +77,63 @@ static void BM_SamToRNTuple(benchmark::State &state)
    state.counters["reads_per_second"] = benchmark::Counter(num_reads, benchmark::Counter::kIsRate);
 }
 
+static void BM_SamToRNTupleRealData(benchmark::State &state)
+{
+   auto file_index = static_cast<std::size_t>(state.range(0));
+   std::string sam_file = input_files[file_index];
+   std::string rntuple_file = sam_file + ".rntuple.root";
+
+   // NOLINTNEXTLINE(clang-analyzer-deadcode.DeadStores)
+   for (auto _ : state) {
+
+      FILE *original_stdout = stdout;
+      stdout = fopen(NULL_DEVICE, "w");
+
+      samtoramntuple(sam_file.c_str(), rntuple_file.c_str(),
+                     /*index=*/true,
+                     /*split=*/true,
+                     /*cache=*/true,
+                     /*compression_algorithm=*/505,
+                     /*quality_policy=*/0);
+
+      // NOLINTNEXTLINE(cppcoreguidelines-owning-memory)
+      fclose(stdout);
+      stdout = original_stdout;
+
+      if (std::filesystem::exists(rntuple_file)) {
+         auto file_size = std::filesystem::file_size(rntuple_file);
+         double file_size_mb = static_cast<double>(file_size) / (1024.0 * 1024.0);
+
+         state.counters["file_size_mb"] = file_size_mb;
+         state.counters["MB_per_sec"] = benchmark::Counter(file_size_mb, benchmark::Counter::kIsRate);
+      }
+
+      std::remove(rntuple_file.c_str());
+   }
+}
+
 int main(int argc, char **argv)
 {
+   if (argc > 1) {
+      std::cout << "Input Files Conversion Benchmark\n";
+
+      for (int i = 1; i < argc; i++) {
+         // NOLINTNEXTLINE(cppcoreguidelines-pro-bounds-pointer-arithmetic)
+         input_files.emplace_back(argv[i]);
+      }
+
+      for (int i = 0; i < input_files.size(); i++) {
+         ::benchmark::RegisterBenchmark((input_files[i]), BM_SamToRNTupleRealData)
+            ->Args({i})
+            ->Unit(benchmark::kMillisecond);
+      }
+
+      ::benchmark::Initialize(&argc, argv);
+      ::benchmark::RunSpecifiedBenchmarks();
+
+      return 0;
+   }
+
    std::cout << "Individual Conversion Time Benchmark" << std::endl;
    std::cout << "====================================" << std::endl;
    std::cout << "Measuring TTree and RNTuple conversion times separately" << std::endl;