args = parser.parse_args() file_list = args.infile in_dir = './' if (args.directory): in_dir = args.directory in_dir = os.path.expanduser(in_dir) suffix = '_calib.root' if (args.suffix): suffix = args.suffix fgeoFile = 'geo_standard.root' if (args.geometry): fgeoFile = args.geometry fgeoFile = os.path.expanduser(fgeoFile) # list_cluster = batch.get_list(file_list, suffix, in_dir) list_elist = batch.get_list(file_list, '_EntryList.root', in_dir) for fcluster, felist in zip(list_cluster, list_elist): command = [ exec_bin, macro, fcluster, '2.6', felist, 'fwd', 'elist_fwdhit', '33', '5', '0.3', '0.9', '3', '2', fgeoFile, 'true' ] print(command) process = subprocess.Popen(command) process.wait()
import argparse import os import subprocess import batch vmc_dir = os.environ['VMCWORKDIR'] macro = os.path.join(vmc_dir, 'macro/koala/ana_tasks/export_koala.C') exec_bin = os.path.join(vmc_dir, 'build/bin/koa_execute') # arguments definitions parser = argparse.ArgumentParser() parser.add_argument("infile", help="the file list to be processed") parser.add_argument("-d", "--directory", default="./", help="directory where files are located") args = parser.parse_args() in_dir = os.path.expanduser(args.directory) # add rec each file in the list list_input = batch.get_list(args.infile, '_TriggerSumCount.root', in_dir) # invoking the command for fin in list_input: command = [exec_bin, macro, fin] print(command) process = subprocess.Popen(command) process.wait()
parser.add_argument("--time_high", default='700') parser.add_argument("--elist_dir", help="directory where elist file is located") parser.add_argument("--elist_suffix", help="suffix of the elist filename") parser.add_argument("--elist_dirname", help="directory name of ROOT folder containing elist") parser.add_argument("--elist_name", help="name of the elist") args = parser.parse_args() file_list = args.infile in_dir = os.path.expanduser(args.directory) fwd_dir = os.path.expanduser(args.fwddir) # add rec each file in the list list_files = batch.get_list(file_list, args.suffix, in_dir) list_fwd = batch.get_list(file_list, "_FwdHit.root", fwd_dir) if (args.elist_dir): elist_dir = os.path.expanduser(args.elist_dir) list_elist = batch.get_list(file_list, args.elist_suffix, elist_dir) for fin, ffwd, felist in zip(list_files, list_fwd, list_elist): command = [ exec_bin, macro, fin, 'koalasim', args.branch, ffwd, args.amp_nbin, args.amp_low, args.amp_high, args.time_nbin, args.time_low, args.time_high, felist, args.elist_dirname, args.elist_name ] print(command) process = subprocess.Popen(command) process.wait() else:
outfile = args.outfile in_dir = './' if (args.input): in_dir = args.input in_dir = os.path.expanduser(in_dir) out_dir = in_dir if (args.output): out_dir = args.output out_dir = os.path.expanduser(out_dir) ################## real job ######################################## # 1. fill the adc and tdc histograms list_decoded = batch.get_list(file_list, '.root', in_dir) for fin in list_decoded: command = [exec_bin, macro_fillHistogram, fin, out_dir] print(command) process = subprocess.Popen(command) process.wait() for fin in list_decoded: command = [exec_bin, macro_fillAmpVsTime, fin, out_dir] print(command) process = subprocess.Popen(command) process.wait() # 2. fit the adc and tdc spectrum list_result = batch.get_list(file_list, '_result.root', out_dir) for fin in list_result:
default="930", help="higher limit of the time spectrum of fwd1") parser.add_argument("--fwd2_low", default="915", help="lower limit of the time spectrum of fwd2") parser.add_argument("--fwd2_high", default="930", help="higher limit of the time spectrum of fwd2") parser.add_argument("--window_low", default="-10", help="lower limit of the time windown") parser.add_argument("--window_high", default="10", help="higher limit of the time window") args = parser.parse_args() in_dir = os.path.expanduser(args.directory) # list_file = batch.get_list(args.infile, args.suffix, in_dir) for fin in list_file: command = [ exec_bin, macro, fin, 'koalasim', 'KoaFwdDigi', args.fwd1_low, args.fwd1_high, args.fwd2_low, args.fwd2_high, args.window_low, args.window_high ] print(command) process = subprocess.Popen(command) process.wait()
"--suffix", default=".root", help="suffix of the digi file") parser.add_argument("--fwd1_amp_trigger", default="780", help="fwd1 amplitude trigger") parser.add_argument("--fwd2_amp_trigger", default="790", help="fwd2 amplitude trigger") args = parser.parse_args() in_dir = os.path.expanduser(args.directory) # add rec each file in the list list_input = batch.get_list(args.infile, args.suffix, in_dir) list_elist = batch.get_list(args.infile, "_EntryList.root", in_dir) # open the output file out_filename = os.path.join(in_dir, args.output) fout = open(out_filename, 'w') # invoking the command for fin, felist in zip(list_input, list_elist): command = [ exec_bin, macro, fin, felist, args.fwd1_amp_trigger, args.fwd2_amp_trigger ] print(command) # process = subprocess.Popen(command) # process.wait()
help="the output file containing selection efficiency") parser.add_argument("-d", "--directory", default="./", help="directory where files are located") parser.add_argument("-s", "--suffix", default=".root", help="suffix of the input file") args = parser.parse_args() in_dir = os.path.expanduser(args.directory) # add rec each file in the list list_input = batch.get_list(args.infile, args.suffix, in_dir) list_rate = batch.get_list(args.infile, "_Rate_Selected.root", in_dir) list_elist = batch.get_list(args.infile, "_EntryList.root", in_dir) # open the output file out_filename = os.path.join(in_dir, args.output) fout = open(out_filename, 'w') # invoking the command for fin, frate, felist in zip(list_input, list_rate, list_elist): command = [exec_bin, macro, fin, frate, felist] print(command) # process = subprocess.Popen(command) # process.wait() with subprocess.Popen(command, stdout=subprocess.PIPE,
help="suffix of the input file") parser.add_argument("--fwd2_thresh_mip", default="1040", help="suffix of the input file") parser.add_argument("--fwd1_thresh_daq", default="780", help="suffix of the input file") parser.add_argument("--fwd2_thresh_daq", default="790", help="suffix of the input file") args = parser.parse_args() # file_list = args.infile list_fcluster = batch.get_list(file_list, args.suffix, args.directory) list_fdigi = batch.get_list(file_list, ".root", args.digi_dir) # open the output file out_filename = os.path.join(args.directory, args.output) fout = open(out_filename, 'w') for fcluster, fdigi in zip(list_fcluster, list_fdigi): command = [ exec_bin, macro, fcluster, args.branch, fdigi, args.p0, args.p1, args.p2, args.window, args.fwd1_thresh_mip, args.fwd2_thresh_mip, args.fwd1_thresh_daq, args.fwd2_thresh_daq ] print(command) with subprocess.Popen(command, stdout=subprocess.PIPE,
# arguments definitions parser = argparse.ArgumentParser() parser.add_argument("infile", help="the file list to be processed") parser.add_argument("-d", "--directory", default="./", help="directory where files are located") parser.add_argument("-s", "--suffix", default="_EmsRawEvent.root", help="suffix of the input file") parser.add_argument("-w", "--window", default="2", help="Half size of the event window") args = parser.parse_args() in_dir = os.path.expanduser(args.directory) # add rec each file in the list list_input = batch.get_list(args.infile, args.suffix, in_dir) list_output = batch.get_list(args.infile, "_Rate.root", in_dir) # invoking the command for fin, fout in zip(list_input, list_output): command = [exec_bin, macro, fin, fout, args.window] print(command) process = subprocess.Popen(command) process.wait()
import os import subprocess import batch vmc_dir = os.environ['VMCWORKDIR'] macro = os.path.join(vmc_dir, 'macro/koala/daq_tasks/unpack.C') exec_bin = os.path.join(vmc_dir, 'build/bin/koa_execute') # arguments definitions parser = argparse.ArgumentParser() parser.add_argument("infile", help="the file list to be processed") parser.add_argument("-d", "--directory", help="directory where files are located") args = parser.parse_args() file_list = args.infile in_dir = './' if (args.directory): in_dir = args.directory in_dir = os.path.expanduser(in_dir) # add rec each file in the list list_decoded = batch.get_list(file_list, '.cl', in_dir) for fin in list_decoded: command = [exec_bin, macro, fin] print(command) process = subprocess.Popen(command) process.wait()
exec_bin = os.path.join(vmc_dir, 'build/bin/koa_execute') # arguments definitions parser = argparse.ArgumentParser() parser.add_argument("infile", help="the file list to be processed") parser.add_argument("-d", "--directory", default="./", help="directory where files are located") parser.add_argument( "--trigger_threshfile", default="cluster_seed_threshold.txt", help="File containing the trigger thresholds for each ADC channel") args = parser.parse_args() in_dir = os.path.expanduser(args.directory) # add rec each file in the list list_input = batch.get_list(args.infile, '.root', in_dir) list_param = batch.get_list(args.infile, '_param.root', in_dir) # invoking the command for fin, fparam in zip(list_input, list_param): command = [ exec_bin, macro, fin, fparam, '_Trigger.root', args.trigger_threshfile ] print(command) process = subprocess.Popen(command) process.wait()
parser.add_argument( "--elist_suffix", help="suffix of the file containing the event list, empty if not used") parser.add_argument("--elist_dir", default="/", help="directory containing the event list") parser.add_argument("--elist_name", default="ems_rate_elist", help="name of the event list") args = parser.parse_args() in_dir = os.path.expanduser(args.directory) # add rec each file in the list list_input = batch.get_list(args.infile, args.suffix, in_dir) if args.elist_suffix: list_elist = batch.get_list(args.infile, args.elist_suffix, in_dir) for fin, felist in zip(list_input, list_elist): command = [ exec_bin, macro, fin, felist, args.elist_dir, args.elist_name ] print(command) process = subprocess.Popen(command) process.wait() else: for fin in list_input: command = [exec_bin, macro, fin] print(command) process = subprocess.Popen(command)
exec_bin = os.path.join(vmc_dir, 'build/bin/koa_execute') # arguments definitions parser = argparse.ArgumentParser() parser.add_argument("infile", help="the file list to be processed") parser.add_argument("-d", "--directory", help="directory where files are located") parser.add_argument("-s", "--suffix", help="suffix of the input file") args = parser.parse_args() file_list = args.infile in_dir = './' if (args.directory): in_dir = args.directory in_dir = os.path.expanduser(in_dir) suffix = '.root' if (args.suffix): suffix = args.suffix # list_file = batch.get_list(file_list, suffix, in_dir) for fin in list_file: command = [exec_bin, macro, fin, 'koalasim'] print(command) process = subprocess.Popen(command) process.wait()
parser.add_argument("-d", "--directory", default="digi", help="directory where files are located") parser.add_argument("-s", "--suffix", default=".root", help="suffix of the input file") parser.add_argument("--cluster_dir", default="calib", help="directory containing the cluster file") parser.add_argument("--cluster_suffix", default="_calib.root", help="suffix of the cluster file") parser.add_argument("--cluster_branch", default="KoaRecCluster_ThresholdFilter", help="name of the cluster branch") args = parser.parse_args() file_list = args.infile # list_file = batch.get_list(file_list, args.suffix, args.directory) list_cluster = batch.get_list(file_list, args.cluster_suffix, args.cluster_dir) for fin, fcluster in zip(list_file, list_cluster): command = [exec_bin, macro, fin, fcluster, args.cluster_branch] print(command) process = subprocess.Popen(command) process.wait()
args = parser.parse_args() in_dir = os.path.expanduser(args.directory) para_dir = os.path.expanduser(args.para_dir) thresh_file = "" if args.thresh_file != "": thresh_file = os.path.expanduser(args.thresh_file) if args.mode == 'absolute': macro = os.path.join( vmc_dir, 'macro/koala/rec_tasks/run_cluster_purification_AbsoluteMode.C') elif args.mode == 'percentage': macro = os.path.join( vmc_dir, 'macro/koala/rec_tasks/run_cluster_purification_PercentageMode.C') # add rec each file in the list list_input = batch.get_list(args.infile, args.suffix, in_dir) list_para = batch.get_list(args.infile, "_param.root", para_dir) # invoking the command for fin, fpara in zip(list_input, list_para): command = [ exec_bin, macro, fin, fpara, args.branch, args.threshold, thresh_file ] print(command) process = subprocess.Popen(command) process.wait()
default="_calib_ThresholdFilter_smear.root") parser.add_argument("--cluster_br", default="KoaRecCluster_Smear") parser.add_argument("--cluster_nr", default="-1", help="Threshold of number of clusters") args = parser.parse_args() file_list = args.infile in_dir = './' if (args.directory): in_dir = args.directory in_dir = os.path.expanduser(in_dir) suffix = '.root' if (args.suffix): suffix = args.suffix # list_file = batch.get_list(file_list, suffix, in_dir) list_fcluster = batch.get_list(file_list, args.cluster_suffix, args.cluster_dir) for fin, fcluster in zip(list_file, list_fcluster): command = [ exec_bin, macro, fin, fcluster, args.cluster_br, args.cluster_nr ] print(command) process = subprocess.Popen(command) process.wait()
'--suffix', default='.root', help='suffix of the input file') parser.add_argument('--hdir', default='rec_adc', help='directory for histograms to be fitted') parser.add_argument('--hsuffix', default='adc', help='suffix for histogram name') parser.add_argument('--seed', default='adc_pedestal_20190902_003449.txt', help='pedestal seed file') parser.add_argument('-o', '--outdir', default='./', help='output directory') # arguments parsing args = parser.parse_args() file_list = args.infile in_dir = os.path.expanduser(args.directory) out_dir = os.path.expanduser(args.outdir) # list_infile = batch.get_list(file_list, args.suffix, in_dir) list_outfile = batch.get_list(file_list, '_pedestal_fit.txt', out_dir) for fin, fout in zip(list_infile, list_outfile): command = [exec_bin, macro, fin, args.hdir, args.hsuffix, fout, args.seed] print(command) process = subprocess.Popen(command) process.wait()
macro = os.path.join(vmc_dir, 'macro/koala/ana_tasks/accumulate_koala_cluster.C') exec_bin = os.path.join(vmc_dir,'build/bin/koa_execute') # arguments definitions parser = argparse.ArgumentParser() parser.add_argument("infile",help="the file list to be processed") parser.add_argument("--cluster_dir", help="directory where files are located") parser.add_argument("--cluster_suffix", help="directory where files are located") parser.add_argument("--digi_dir", help="directory where files are located") parser.add_argument("--digi_suffix", help="directory where files are located") args = parser.parse_args() cluster_dir = os.path.expanduser(args.cluster_dir) digi_dir = os.path.expanduser(args.digi_dir) # add rec each file in the list list_cluster = batch.get_list(args.infile, args.cluster_suffix, cluster_dir) list_digi = batch.get_list(args.infile, args.digi_suffix, digi_dir) # invoking the command for fcluster,fdigi in zip(list_cluster, list_digi): command = [exec_bin, macro, fcluster, fdigi] print(command) process = subprocess.Popen(command) process.wait()
help="directory where files are located") parser.add_argument("-o", "--output", default="./", help="directory where the output files are saved") parser.add_argument("-s", "--suffix", default="_calib.root", help="suffix of the output") parser.add_argument("--pedestal", default="adc_pedestal_20190902_003449.txt") parser.add_argument("--adc", default="adc_calib_energy_huagen.txt") parser.add_argument("--tdc", default="tdc_calib_shift.txt") parser.add_argument("--seed", default="cluster_seed_threshold.txt") args = parser.parse_args() in_dir = os.path.expanduser(args.directory) out_dir = os.path.expanduser(args.output) # add rec each file in the list list_decoded = batch.get_list(args.infile, '.root', in_dir) for fin in list_decoded: fpara = fin.replace(".root", "_param.root") command = [ exec_bin, macro, fin, fpara, out_dir, args.suffix, args.pedestal, args.adc, args.tdc, args.seed ] print(command) process = subprocess.Popen(command) process.wait()
parser.add_argument("infile", help="the file list to be processed") parser.add_argument("outfile", help="the output summed root file name") parser.add_argument("-d", "--directory", help="directory where files are located") parser.add_argument("-s", "--suffix", help="common suffix of the file") args = parser.parse_args() file_list = args.infile outfile = args.outfile in_dir = './' if (args.directory): in_dir = args.directory in_dir = os.path.expanduser(in_dir) suffix = '_result.root' if (args.suffix): suffix = args.suffix list_decoded = batch.get_list(file_list, suffix, in_dir) command.append(os.path.join(in_dir, outfile)) #command = [haddbin, os.path.join(in_dir, outfile)] for fin in list_decoded: command.append(fin) print(command) process = subprocess.Popen(command) # wait until process returns, so that mv in next step be meaningful process.wait()