def merge_files_by_process(root_files):
    global input_folder, output_folder
    electron_qcd_samples = [ 'QCD_Pt-20to30_BCtoE',
                 'QCD_Pt-30to80_BCtoE',
                 'QCD_Pt-80to170_BCtoE',
                 'QCD_Pt-20to30_EMEnriched',
                 'QCD_Pt-30to80_EMEnriched',
                 'QCD_Pt-80to170_EMEnriched',
                 'GJets_HT-40To100',
                 'GJets_HT-100To200',
                 'GJets_HT-200']
    singleTop_samples = [ 'T_tW-channel',
                 'T_t-channel',
                 'T_s-channel',
                 'Tbar_tW-channel',
                 'Tbar_t-channel',
                 'Tbar_s-channel']
    wplusjets_samples = [ 'W1Jet', 'W2Jets', 'W3Jets', 'W4Jets']
    vplusjets_samples = wplusjets_samples
    vplusjets_samples.append('DYJetsToLL')
    diboson_samples = [ 'WWtoAnything', 'WZtoAnything', 'ZZtoAnything']
    signal_samples = [ 'TTJet', 'SingleTop']
    
    summations = {
                  'QCD_Electron':electron_qcd_samples,
                  'SingleTop' : singleTop_samples,
                  'WPlusJets' : wplusjets_samples,
                  'VPlusJets' : vplusjets_samples,
                  'DiBoson': diboson_samples,
                  'Signal': signal_samples
                  }
    
    summation_files = {}
    file_template = ''
    template_token = '<temp>'
    for summation, samples in summations.iteritems():
        summation_files[summation] = []
        for file_in_path in root_files:
            process_name = get_process_from_file(file_in_path)
            if not file_template:
                file_template = file_in_path.replace(process_name, template_token)
                file_template = file_template.replace(input_folder, output_folder)
            if process_name in samples:
                summation_files[summation].append(file_in_path)
    
    for summation, files in summation_files.iteritems():
        output_file = file_template.replace(template_token, summation)
        merge_ROOT_files(files, output_file)        
def merge_files_by_process(root_files):
    global input_folder, output_folder
    electron_qcd_samples = [
        'QCD_Pt-20to30_BCtoE', 'QCD_Pt-30to80_BCtoE', 'QCD_Pt-80to170_BCtoE',
        'QCD_Pt-20to30_EMEnriched', 'QCD_Pt-30to80_EMEnriched',
        'QCD_Pt-80to170_EMEnriched', 'GJets_HT-40To100', 'GJets_HT-100To200',
        'GJets_HT-200'
    ]
    singleTop_samples = [
        'T_tW-channel', 'T_t-channel', 'T_s-channel', 'Tbar_tW-channel',
        'Tbar_t-channel', 'Tbar_s-channel'
    ]
    wplusjets_samples = ['W1Jet', 'W2Jets', 'W3Jets', 'W4Jets']
    vplusjets_samples = wplusjets_samples
    vplusjets_samples.append('DYJetsToLL')
    diboson_samples = ['WWtoAnything', 'WZtoAnything', 'ZZtoAnything']
    signal_samples = ['TTJet', 'SingleTop']

    summations = {
        'QCD_Electron': electron_qcd_samples,
        'SingleTop': singleTop_samples,
        'WPlusJets': wplusjets_samples,
        'VPlusJets': vplusjets_samples,
        'DiBoson': diboson_samples,
        'Signal': signal_samples
    }

    summation_files = {}
    file_template = ''
    template_token = '<temp>'
    for summation, samples in summations.iteritems():
        summation_files[summation] = []
        for file_in_path in root_files:
            process_name = get_process_from_file(file_in_path)
            if not file_template:
                file_template = file_in_path.replace(process_name,
                                                     template_token)
                file_template = file_template.replace(input_folder,
                                                      output_folder)
            if process_name in samples:
                summation_files[summation].append(file_in_path)

    for summation, files in summation_files.iteritems():
        output_file = file_template.replace(template_token, summation)
        merge_ROOT_files(files, output_file)
示例#3
0
def scale_file(file_in_path):
    process_name = get_process_from_file(file_in_path)
示例#4
0
def scale_file(file_in_path):
    process_name = get_process_from_file(file_in_path)