util.h

Go to the documentation of this file.

 
#include <filesystem>
#include <regex>
#include <iostream>
#include "TChain.h"
 
namespace Util {
  inline int prepare_icefinal_chain(std::filesystem::path PUEO_MC_DATA,
                                    std::unique_ptr<TChain> &passTree_chain_ptr,
                                    std::unique_ptr<TChain> &allTree_chain_ptr)
  {
    passTree_chain_ptr = std::make_unique<TChain>("passTree");
    allTree_chain_ptr = std::make_unique<TChain>("allTree");
 
    std::regex re(R"(.*run(\d+))");
    std::smatch match;
    std::string run_name;   // eg. run420/
    std::string run_number; // eg. 420
    std::filesystem::path icefinal_path; // eg. .../run35/IceFinal_35_allTree.root
    
    // loop through the directory that contains all the runs
    for (auto& p: std::filesystem::directory_iterator(PUEO_MC_DATA)){
      run_name = p.path()     // eg. /work/e513ad03_2025Feb28_f0ce4e36_2025Feb28_nnt_100_energy_20/run35
                  .filename() // eg. run35
                  .string();
      // extract the number
      if(std::regex_match(run_name, match, re)) {
         run_number = match[1].str();
      }
      
      // check if the IceFinal_allTree.root file exists.
      icefinal_path = p.path() / ("IceFinal_" + run_number + "_allTree.root");
      if (!std::filesystem::exists(icefinal_path)){
        std::cerr << "file not found: " << icefinal_path << "\n";
        return 1;
      }
 
      passTree_chain_ptr -> Add(icefinal_path.string().c_str());
      allTree_chain_ptr -> Add(icefinal_path.string().c_str());
    }
 
    return 0;
  };
 
}