SCorrelatorJetTreeMaker/scripts/MergeFiles.rb

0001 #!/usr/bin/env ruby
0002 # -----------------------------------------------------------------------------
0003 # 'MergeFiles.rb'
0004 # Derek Anderson
0005 # 04.22.2023
0006 #
0007 # For merging files using 'hadd_files.C'
0008 # -----------------------------------------------------------------------------
0009
0010 # modules to use
0011 require 'fileutils'
0012
0013 # input parameters
0014 in_path = "./condor/intermediate_merge/pp200py8jet10run6_trksAndChargPars_2023may7"
0015 in_pref = "correlatorJetTree.pp200py8jet10run6_trksAndChrgPars_"
0016 in_suff = ".d7m5y2023.root"
0017
0018 # output parameters
0019 out_list = "correlatorJetTree.pp200py8jet10run6_trksAndChrgPars.d7m5y2023.list"
0020 out_file = "correlatorJetTree.pp200py8jet10run6_trksAndChrgPars.d7m5y2023.root"
0021
0022 # create input matching pattern
0023 in_pattern = in_path + "/" + in_pref + "*" + in_suff
0024 in_pattern.gsub!("//", "/")
0025 in_pattern.gsub!("..", ".")
0026
0027 # create list of files to merge
0028 File.open(out_list, "w") { |out|
0029   Dir[in_pattern].each do |file|
0030     out.puts file
0031   end
0032 }
0033
0034 # grab number of files to merge
0035 num_files = Dir[in_pattern].size
0036
0037 # merge files
0038 exec("root -b -q \'MergeFiles.C(#{num_files}, \"#{out_list}\", \"#{out_file}\")\'")
0039
0040 # end -------------------------------------------------------------------------