Get menu for period

Author: mpuccio

EventFiltering/macros/getMenu.C

@@ -12,9 +12,17 @@
 #include "CCDB/BasicCCDBManager.h"
 
 #include <TAxis.h>
+#include <TFile.h>
+#include <TGrid.h>
 #include <TH1.h>
+#include <TSystem.h>
 
+#include <iostream>
+#include <regex>
+#include <set>
+#include <sstream>
 #include <string>
+#include <vector>
 
 void getMenu(int runNumber, std::string baseCCDBPath = "Users/m/mpuccio/EventFiltering/OTS/Chunked/")
 {
@@ -29,3 +37,87 @@ void getMenu(int runNumber, std::string baseCCDBPath = "Users/m/mpuccio/EventFil
     std::cout << "Id " << i - 2 << ": " << axis->GetBinLabel(i) << "\n";
   }
 }
+
+std::vector<std::string> getMenuForPeriod(std::string period)
+{
+  std::regex pattern(R"(LHC(\d{2})[A-Za-z]{1,2})");
+  std::smatch match;
+
+  int year{2000};
+  if (!std::regex_match(period, match, pattern)) {
+    std::cout << "Invalid format for period: " << period << std::endl;
+    return {};
+  }
+
+  year += std::stoi(match[1]);
+  gSystem->Exec(Form("alien_find /alice/data/%i/%s/ ctf_skim_full/AnalysisResults_fullrun.root > list_tmp_%s.txt", year, period.data(), period.data()));
+
+  std::ifstream file(Form("list_tmp_%s.txt", period.data()));
+  if (!file) {
+    std::cerr << "Error: could not open file for period " << period << "\n";
+    return {};
+  }
+
+  std::string firstLine;
+  if (!std::getline(file, firstLine)) {
+    std::cerr << "Error: file is empty or read failed for period " << period << "\n";
+    return {};
+  }
+
+  TGrid::Connect("alien://");
+  TFile* scalersFile = TFile::Open((std::string("alien://") + firstLine).data(), "READ");
+  TH1D* counters = (TH1D*)scalersFile->Get("central-event-filter-task/scalers/mFiltered");
+  TAxis* axis = counters->GetXaxis();
+
+  std::vector<std::string> binLabels(axis->GetNbins() - 2);
+  for (int i = 2; i < axis->GetNbins(); ++i) {
+    binLabels[i - 2] = axis->GetBinLabel(i);
+  }
+
+  scalersFile->Close();
+  delete scalersFile;
+  gSystem->Exec(Form("rm list_tmp_%s.txt", period.data()));
+
+  return binLabels;
+}
+
+void getMenu(std::string periods)
+{
+  std::stringstream ss(periods);
+  std::string period;
+  std::vector<std::string> periodList;
+
+  // Parse comma-separated periods
+  while (std::getline(ss, period, ',')) {
+    // Trim whitespace
+    period.erase(0, period.find_first_not_of(" \t"));
+    period.erase(period.find_last_not_of(" \t") + 1);
+    periodList.push_back(period);
+  }
+
+  std::map<std::vector<std::string>, std::vector<std::string>> menuGroups;
+
+  // Get menus for each period
+  for (const auto& p : periodList) {
+    auto menu = getMenuForPeriod(p);
+    if (!menu.empty()) {
+      menuGroups[menu].push_back(p);
+    }
+  }
+
+  // Report different menus
+  int menuId = 1;
+  for (const auto& [menu, periods] : menuGroups) {
+    std::cout << "\n=== Menu " << menuId++ << " (periods: ";
+    for (size_t i = 0; i < periods.size(); ++i) {
+      std::cout << periods[i];
+      if (i < periods.size() - 1)
+        std::cout << ", ";
+    }
+    std::cout << ") ===\n";
+
+    for (size_t i = 0; i < menu.size(); ++i) {
+      std::cout << "Id " << i << ": " << menu[i] << "\n";
+    }
+  }
+}