コード例 #1
0
def CopyHists(infile, outfile, prefix = ''):
    
    from load_data import TprimeData
    _hists = TprimeData(1)

    _hists.load_all_hists(infile, prefix)
    _hists.SaveAllHists(outfile)

    return
コード例 #2
0
def ProjectOnHt(infile, outfile, prefix = ''):
    from load_data import TprimeData
    _hists = TprimeData(1)
    _hists.load_all_hists(infile, projection_y = True,
                          rebin_ngroup = 4,
                          rebin_low    = 200,
                          rebin_high   = 2000,
                          prefix=prefix)
    _hists.SaveAllHists(outfile)

    return
コード例 #3
0
def MakeSignalBoxDiagonal(infile, outfile, prefix = '',
                          mass = None,
                          signal_only = False):

    from load_data import TprimeData
    _hists = TprimeData(1)
    
    _hists.load_all_hists(infile,
                          prefix=prefix)
    
    _hists.ConvertToSignalDiagonal(mass,signal_only=signal_only)
    
    _hists.SaveAllHists(outfile)
    
    return
コード例 #4
0
def MakeSignalBoxTemplates(infile, outfile, prefix = '',
                           xmin = None, xmax = None,
                           ymin = None, ymax = None,
                           signal_only = False):
    
    from load_data import TprimeData
    _hists = TprimeData(1)

    _hists.load_all_hists(infile,
                          prefix=prefix)

    _hists.ConvertToSignalBox(xmin, xmax, ymin, ymax, signal_only=signal_only)
    
    _hists.SaveAllHists(outfile)

    return
コード例 #5
0
def MakeSignalMassDiagPerp(infile, outfile, prefix = '',
                           mass = None,
                           rebin_ngroup = None,
                           rebin_low = None,
                           rebin_high= None
                           ):
    
        from load_data import TprimeData
        _hists = TprimeData(1)
        
        _hists.load_all_hists(infile,
                              prefix=prefix)
        
        _hists.ConvertToMassDiagPerp(mass,rebin_ngroup,rebin_low,rebin_high)

        _hists.SaveAllHists(outfile)
        
        return
コード例 #6
0
ファイル: tprime_powheg.py プロジェクト: TENorbert/SUSY-TOOLS
def CreateTemplates(options):
    #
    # create two "standard" files with "plus" and "minus"
    # shape templates for POWHEG-Nominal difference
    # Since there is only one POWHEG template, the idea is
    # to interpolate in some way between it and the nominal one
    #

    legend = '[tprime_powheg.CreateTemplates]:'

    channels = ['e', 'mu']

    input_nom = {}
    input_pow = {}

    input_nom[
        'mu'] = 'data/mujets_4600/from_gueorgi/27jan2011v1/csv/v1/mujets_4601ipb_2D_v1_csv_jer_nom.root'
    input_pow[
        'mu'] = 'data/mujets_4600/from_gueorgi/27jan2011v1/csv/v1/mujets_4601ipb_2D_v1_csv_POWHEG.root'

    input_nom[
        'e'] = 'data/ejets_4700/ricardo_4683ipb_2d_csv_22jan2012v2/ht4jetsAfterFit_vs_fitMass.root'
    input_pow[
        'e'] = 'data/ejets_4700/ricardo_4683ipb_2d_csv_22jan2012v2/ht4jetsAfterFit_vs_fitMass_powheg.root'

    hist_name = {}
    hist_name['e'] = 'Top_ht4jetsAfterFit:fitMass'
    hist_name['mu'] = 'TTjets_HtvsMfit'

    for chan in channels:

        data = TprimeData()
        data_powheg = TprimeData()

        _suffix = ''

        data.load_all_hists(input_nom[chan],
                            merge_ewk=False,
                            merge_top=False,
                            strip_suffix=False,
                            suffix=_suffix)

        data_powheg.load_all_hists(input_pow[chan],
                                   merge_ewk=False,
                                   merge_top=False,
                                   strip_suffix=False,
                                   suffix=_suffix)

        hPlus = copy.deepcopy(data.hists[hist_name[chan]])
        hMinus = copy.deepcopy(data_powheg.hists[hist_name[chan]])

        hPlus.SetMarkerColor(ROOT.kRed)

        #hMinus.Draw()
        #hPlus.Draw('same')

        #raw_input('press enter to continue')

        hNom = copy.deepcopy(hMinus)
        hNom.Add(hPlus)
        hNom.Scale(0.5)

        # save nominal

        data.hists.pop(hist_name[chan])
        iTop = 0
        for name in data.hist_names:
            if name == hist_name[chan]:
                break
            iTop += 1
        data.hist_names.pop(iTop)

        data.AddHist(hNom)

        data.SaveAllHists(chan + '_powheg_nom.root')

        # save minus

        data.hists.pop(hist_name[chan])
        iTop = 0
        for name in data.hist_names:
            if name == hist_name[chan]:
                break
            iTop += 1
        data.hist_names.pop(iTop)

        data.AddHist(hMinus)

        data.SaveAllHists(chan + '_powheg_minus.root')

        # save plus

        data.hists.pop(hist_name[chan])
        iTop = 0
        for name in data.hist_names:
            if name == hist_name[chan]:
                break
            iTop += 1
        data.hist_names.pop(iTop)

        data.AddHist(hPlus)

        data.SaveAllHists(chan + '_powheg_plus.root')
コード例 #7
0
ファイル: tprime.py プロジェクト: TENorbert/SUSY-TOOLS
                              projection_x=options.projection_x,
                              projection_y=options.projection_y,
                              rebin_ngroup=options.rebin_ngroup,
                              rebin_low=options.rebin_low,
                              rebin_high=options.rebin_high,
                              scale_bin_error=options.scale_bin_error)
            i_suff += 1

        # add pseudo data if requested
        if options.add_pseudodata:
            _map = {}
            _map['ewk'] = 380
            _map['top'] = 1766
            ds.CreatePseudoData('DATA', _map)

        ds.SaveAllHists(options.out_file)

    if name == 'mujets_merge':
        ds.load_all_hists('data/mujets/5jets/v2/mujets_1889ipb_2D_v2.root',
                          merge_ewk=merge_ewk)
        ds.load_all_hists(
            'data/mujets/5jets/v2/mujets_1889ipb_2D_v2_JES_minus.root',
            "",
            "_JESdown",
            merge_ewk=merge_ewk)
        ds.load_all_hists(
            'data/mujets/5jets/v2/mujets_1889ipb_2D_v2_JES_plus.root',
            "",
            "_JESup",
            merge_ewk=merge_ewk)
        ds.SaveAllHists('data/mujets/tprime_mujets_2D_1889ipb_5j_v2.root')