STrackMatchingComparator/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 = "./new_matcher/num10evt500pt020pipm"
0015 in_pref = "sPhenixG4_testingNewTruthMatcher_"
0016 in_suff = ".num10evt500pt020pipm.d12m9y2023_g4trackmatching.root"
0017 #in_suff = ".et020n5pipm.d11m8y2023_g4trackmatching.root"
0018
0019 # output parameters
0020 out_list = "sPhenixG4_testingNewMatcher_newMatcher.pt020num10evt500pipm.d15m9y2023.list"
0021 out_file = "sPhenixG4_testingNewMatcher_newMatcher.pt020num10evt500pipm.d15m9y2023.root"
0022
0023 # create input matching pattern
0024 in_pattern = in_path + "/" + in_pref + "*" + in_suff
0025 in_pattern.gsub!("//", "/")
0026 in_pattern.gsub!("..", ".")
0027
0028 # create list of files to merge
0029 File.open(out_list, "w") { |out|
0030   Dir[in_pattern].each do |file|
0031     out.puts file
0032   end
0033 }
0034
0035 # grab number of files to merge
0036 num_files = Dir[in_pattern].size
0037
0038 # merge files
0039 exec("root -b -q \'MergeFiles.C(#{num_files}, \"#{out_list}\", \"#{out_file}\")\'")
0040
0041 # end -------------------------------------------------------------------------