def build_roofunctor_dict(basename, wsname = 'fit_efficiency', functor_name = 'efficiency'):
ret = {}
for i in optimizer.lep_id:
ret[i] = build_roofunctor(
basename % i,
wsname, # workspace name
functor_name
)
return ret
def build_roofunctor_dict(basename, wsname = 'fit_efficiency', functor_name = 'efficiency', mapper = None):
ret = {}
for i in optimizer.lep_id:
lepid = i
if mapper:
lepid = mapper(lepid)
ret[i] = build_roofunctor(
basename % lepid,
wsname, # workspace name
functor_name
)
return ret
def build_2Droofunctor_dict(basename,
xvar, yvar,
wsname = 'fit_efficiency', functor_name = 'efficiency', mapper = None):
ret = {}
for i in optimizer.lep_id:
lepid = i
if mapper:
lepid = mapper(lepid)
ret[i] = build_uncorr_2Droofunctor(
build_roofunctor(
basename % (lepid, xvar),
wsname, # workspace name
functor_name
),
build_roofunctor(
basename % (lepid, yvar),
wsname, # workspace name
functor_name
),
(basename % (lepid, yvar)).replace('.root','.corrected_inputs.root'),
)
return ret
def build_roofunctor_dict(basename,
wsname='fit_efficiency',
functor_name='efficiency',
mapper=None):
ret = {}
for i in optimizer.lep_id:
lepid = i
if mapper:
lepid = mapper(lepid)
ret[i] = build_roofunctor(
basename % lepid,
wsname, # workspace name
functor_name)
return ret
def build_2Droofunctor_dict(basename,
xvar,
yvar,
wsname='fit_efficiency',
functor_name='efficiency',
mapper=None):
ret = {}
for i in optimizer.lep_id:
lepid = i
if mapper:
lepid = mapper(lepid)
ret[i] = build_uncorr_2Droofunctor(
build_roofunctor(
basename % (lepid, xvar),
wsname, # workspace name
functor_name),
build_roofunctor(
basename % (lepid, yvar),
wsname, # workspace name
functor_name),
(basename % (lepid, yvar)).replace('.root',
'.corrected_inputs.root'),
)
return ret
#EET
highpt_ee_fr = make_mva_functor_dict(frfit_dir + '/ee_wjetsNoZmass_pt20_%s_electronInfo.kNN.weights.xml', variables)
lowpt_ee_fr = make_mva_functor_dict(frfit_dir + '/ee_wjetsNoZmass_pt10_%s_electronInfo.kNN.weights.xml', variables)
highpt_ee_qcd_fr = make_mva_functor_dict(frfit_dir + '/ee_qcd_pt20_%s_electronInfo.kNN.weights.xml', variables)
lowpt_ee_qcd_fr = make_mva_functor_dict(frfit_dir + '/ee_qcd_pt10_%s_electronInfo.kNN.weights.xml', variables)
##################
## 1D Taus Func ##
##################
tau_fr = build_roofunctor(
frfit_dir + '/t_ztt_pt20_mvaloose_tauPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
tau_qcd_fr = tau_fr
e_charge_flip = make_corrector_dict(frfit_dir+"/charge_flip_prob_map_%s.root", "efficiency_map")
e_charge_flip_up = make_corrector_dict(frfit_dir+"/charge_flip_prob_map_%s.root", "efficiency_map_statUp")
e_charge_flip_down = make_corrector_dict(frfit_dir+"/charge_flip_prob_map_%s.root", "efficiency_map_statDown")
mass_scaler = make_scaler_dict(frfit_dir+"/charge_flip_prob_map_%s.root", 'mass_scale')
default_scaler = mass_scaler[mass_scaler.keys()[0]]
e1_charge_flip = make_corrector_dict(frfit_dir+"/charge_flip_prob_map_e1_%s.root", "efficiency_map")
e1_charge_flip_up = make_corrector_dict(frfit_dir+"/charge_flip_prob_map_e1_%s.root", "efficiency_map_statUp")
e1_charge_flip_down = make_corrector_dict(frfit_dir+"/charge_flip_prob_map_e1_%s.root", "efficiency_map_statDown")
import glob
from MuMuTauTree import MuMuTauTree
import os
import FinalStateAnalysis.TagAndProbe.MuonPOGCorrections as MuonPOGCorrections
import FinalStateAnalysis.TagAndProbe.PileupWeight as PileupWeight
import WHAnalyzerBase
import ROOT
################################################################################
#### Fitted fake rate functions ################################################
################################################################################
# Get fitted fake rate functions
frfit_dir = os.path.join('results', os.environ['jobid'], 'fakerate_fits')
highpt_mu_fr = build_roofunctor(
frfit_dir + '/m_wjets_pt20_pfidiso02_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency')
lowpt_mu_fr = build_roofunctor(
frfit_dir + '/m_wjets_pt10_pfidiso02_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency')
tau_fr = build_roofunctor(
frfit_dir + '/t_ztt_pt20_mvaloose_tauPt.root',
'fit_efficiency', # workspace name
'efficiency')
################################################################################
#### MC-DATA and PU corrections ################################################
################################################################################
import glob
from MuMuTauTree import MuMuTauTree
import os
import FinalStateAnalysis.TagAndProbe.MuonPOGCorrections as MuonPOGCorrections
import FinalStateAnalysis.TagAndProbe.PileupWeight as PileupWeight
import WHAnalyzerBase
import ROOT
################################################################################
#### Fitted fake rate functions ################################################
################################################################################
# Get fitted fake rate functions
frfit_dir = os.path.join("results", os.environ["jobid"], "fakerate_fits")
highpt_mu_fr = build_roofunctor(
frfit_dir + "/m_wjets_pt20_pfidiso02_muonJetPt.root", "fit_efficiency", "efficiency" # workspace name
)
lowpt_mu_fr = build_roofunctor(
frfit_dir + "/m_wjets_pt10_pfidiso02_muonJetPt.root", "fit_efficiency", "efficiency" # workspace name
)
tau_fr = build_roofunctor(
frfit_dir + "/t_ztt_pt20_mvaloose_tauPt.root", "fit_efficiency", "efficiency" # workspace name
)
################################################################################
#### MC-DATA and PU corrections ################################################
################################################################################
# Determine MC-DATA corrections
import os
import re
import glob
import ROOT
from math import sqrt
from FinalStateAnalysis.StatTools.RooFunctorFromWS import build_roofunctor, make_corrector_from_histo
frfit_dir = os.path.join('results', os.environ['jobid'], 'fakerate_fits') + '/'
tau_fake_rate = build_roofunctor(frfit_dir + 't_os_tLoose_tTigh_tAbsEta.root',
'fit_efficiency', 'efficiency')
def inflate_systematics(functor, inflation):
'''Christian recipe: INFLATE ALL THE SYSTEMATICS!'''
def fcn(*args):
return functor(*args) * inflation
return fcn
#tau_fake_rate_up = inflate_systematics(tau_fake_rate, 1.3)
#tau_fake_rate_dw = inflate_systematics(tau_fake_rate, 0.7)
###Tau fakes as it should be
tau_fake_rate_up = make_corrector_from_histo(
frfit_dir + 't_os_tLoose_tTigh_tAbsEta_2.root', 'efficiency_up', '1D')
tau_fake_rate_dw = make_corrector_from_histo(
frfit_dir + 't_os_tLoose_tTigh_tAbsEta_2.root', 'efficiency_dw', '1D')
'''
from EETauTree import EETauTree
from FinalStateAnalysis.StatTools.RooFunctorFromWS import build_roofunctor
import os
#from PUWeighting import pu_weight
pu_weight=1.0
import ROOT
import WHAnalyzerBase
# Get fitted fake rate functions
frfit_dir = os.path.join('results', os.environ['jobid'], 'fakerate_fits')
highpt_e_fr = build_roofunctor(
frfit_dir + '/e_wjets_pt10_mvaidiso02_eJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
lowpt_e_fr = build_roofunctor(
frfit_dir + '/e_wjets_pt10_mvaidiso03_eJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
tau_fr = build_roofunctor(
frfit_dir + '/t_ztt_pt20_mvaloose_tauPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
# Get correction functions
ROOT.gSystem.Load("Corrector_C")
# mu_loose_fr = build_roofunctor( #lambda x: 0.1
# frfit_dir + '/m_zlt_pt10_looseId_muonPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
# )
# e_loose_fr =build_roofunctor( #lambda x: 0.2
# frfit_dir + '/e_zlt_pt10_looseId_electronPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
# )
mu_tight_jetpt_fr = build_roofunctor( # lambda x: 0.1
frfit_dir + "/m_zlt_pt10_tightId_muonJetPt.root", "fit_efficiency", "efficiency" # workspace name
)
e_tight_jetpt_fr = build_roofunctor( # lambda x: 0.2
frfit_dir + "/e_zlt_pt10_tightId_electronJetPt.root", "fit_efficiency", "efficiency" # workspace name
)
mu_loose_jetpt_fr = build_roofunctor( # lambda x: 0.1
frfit_dir + "/m_zlt_pt10_looseId_muonJetPt.root", "fit_efficiency", "efficiency" # workspace name
)
e_loose_jetpt_fr = build_roofunctor( # lambda x: 0.2
frfit_dir + "/e_zlt_pt10_looseId_electronJetPt.root", "fit_efficiency", "efficiency" # workspace name
)
## /e_zlt_pt10_looseId_electronJetPt.root
Analyze EET events for the WH analysis
'''
from EETauTree import EETauTree
from FinalStateAnalysis.StatTools.RooFunctorFromWS import build_roofunctor
import os
from PUWeighting import pu_weight
import ROOT
import WHAnalyzerBase
# Get fitted fake rate functions
frfit_dir = os.path.join('results', os.environ['jobid'], 'fakerate_fits')
highpt_ee_fr = build_roofunctor(
frfit_dir + '/ee_wjets_pt20_mvaidiso01_e2JetPt-data_ee.root',
'fit_efficiency', # workspace name
'efficiency')
lowpt_ee_fr = build_roofunctor(
frfit_dir + '/ee_wjets_pt10_mvaidiso01_e2JetPt-data_ee.root',
'fit_efficiency', # workspace name
'efficiency')
tau_fr = build_roofunctor(
frfit_dir + '/t_ztt_pt20_mvaloose_tauPt-data_mm.root',
'fit_efficiency', # workspace name
'efficiency')
# Get correction functions
ROOT.gSystem.Load("Corrector_C")
class WHAnalyzeEET(WHAnalyzerBase.WHAnalyzerBase):
import os
from FinalStateAnalysis.StatTools.RooFunctorFromWS import build_roofunctor
################################################################################
#### Fitted fake rate functions ################################################
################################################################################
# Get fitted fake rate functions
frfit_dir = os.path.join('results', os.environ['jobid'], 'fakerate_fits')
mu_tight_fr = build_roofunctor( #lambda x: 0.1
frfit_dir + '/m_zlt_pt10_tightId_muonPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
e_tight_fr =build_roofunctor( #lambda x: 0.2
frfit_dir + '/e_zlt_pt10_tightId_electronPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
mu_loose_fr = build_roofunctor( #lambda x: 0.1
frfit_dir + '/m_zlt_pt10_looseId_muonPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
e_loose_fr =build_roofunctor( #lambda x: 0.2
frfit_dir + '/e_zlt_pt10_looseId_electronPt.root',
'fit_efficiency', # workspace name
'efficiency'
data_views.log.setLevel(logging.INFO)
#import sys
#logging.basicConfig(stream=sys.stderr, level=logging.INFO)
import WHAnalyzerBase
import ROOT
from TwoDimFakeRate import TwoDimFakeRate
################################################################################
#### Fitted fake rate functions ################################################
################################################################################
# Get fitted fake rate functions
frfit_dir = os.path.join('results', os.environ['jobid'], 'fakerate_fits')
highpt_mu_fr = build_roofunctor(
#frfit_dir + '/m_wjets_pt20_pfidiso02_muonJetPt.root',
frfit_dir + '/m_wjets_pt20_h2taucuts_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
lowpt_mu_fr = build_roofunctor(
#frfit_dir + '/m_wjets_pt10_pfidiso02_muonJetPt.root',
frfit_dir + '/m_wjets_pt10_h2taucuts_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
tau_fr = build_roofunctor(
frfit_dir + '/t_ztt_pt20_mvaloose_tauPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
highpt_mu_qcd_fr = build_roofunctor(
Analyze EET events for the WH analysis
'''
from EETauTree import EETauTree
from FinalStateAnalysis.StatTools.RooFunctorFromWS import build_roofunctor
import os
from PUWeighting import pu_weight
import ROOT
import WHAnalyzerBase
# Get fitted fake rate functions
frfit_dir = os.path.join('results', os.environ['jobid'], 'fakerate_fits')
highpt_ee_fr = build_roofunctor(
frfit_dir + '/ee_wjets_pt20_mvaidiso01_e2JetPt-data_ee.root',
'fit_efficiency', # workspace name
'efficiency'
)
lowpt_ee_fr = build_roofunctor(
frfit_dir + '/ee_wjets_pt10_mvaidiso01_e2JetPt-data_ee.root',
'fit_efficiency', # workspace name
'efficiency'
)
tau_fr = build_roofunctor(
frfit_dir + '/t_ztt_pt20_mvaloose_tauPt-data_mm.root',
'fit_efficiency', # workspace name
'efficiency'
)
# Get correction functions
ROOT.gSystem.Load("Corrector_C")
#mu_loose_fr = build_roofunctor( #lambda x: 0.1
# frfit_dir + '/m_zlt_pt10_looseId_muonPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
#)
#e_loose_fr =build_roofunctor( #lambda x: 0.2
# frfit_dir + '/e_zlt_pt10_looseId_electronPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
#)
mu_tight_jetpt_fr = build_roofunctor( #lambda x: 0.1
frfit_dir + '/m_zlt_pt10_tightId_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
#e_tight_jetpt_fr =build_roofunctor( #lambda x: 0.2
# frfit_dir + '/e_zlt_pt10_tightId_electronJetPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
#)
mu_loose_jetpt_fr = build_roofunctor( #lambda x: 0.1
frfit_dir + '/m_zlt_pt10_looseId_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
variables)
lowpt_ee_fr = make_mva_functor_dict(
frfit_dir + '/ee_wjetsNoZmass_pt10_%s_electronInfo.kNN.weights.xml',
variables)
highpt_ee_qcd_fr = make_mva_functor_dict(
frfit_dir + '/ee_qcd_pt20_%s_electronInfo.kNN.weights.xml', variables)
lowpt_ee_qcd_fr = make_mva_functor_dict(
frfit_dir + '/ee_qcd_pt10_%s_electronInfo.kNN.weights.xml', variables)
##################
## 1D Taus Func ##
##################
tau_fr = build_roofunctor(
frfit_dir + '/t_ztt_pt20_mvaloose_tauPt.root',
'fit_efficiency', # workspace name
'efficiency')
tau_qcd_fr = tau_fr
e_charge_flip = make_corrector_dict(
frfit_dir + "/charge_flip_prob_map_%s.root", "efficiency_map")
e_charge_flip_up = make_corrector_dict(
frfit_dir + "/charge_flip_prob_map_%s.root", "efficiency_map_statUp")
e_charge_flip_down = make_corrector_dict(
frfit_dir + "/charge_flip_prob_map_%s.root", "efficiency_map_statDown")
mass_scaler = make_scaler_dict(frfit_dir + "/charge_flip_prob_map_%s.root",
'mass_scale')
default_scaler = mass_scaler[mass_scaler.keys()[0]]
e1_charge_flip = make_corrector_dict(
#mu_loose_fr = build_roofunctor( #lambda x: 0.1
# frfit_dir + '/m_zlt_pt10_looseId_muonPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
#)
#e_loose_fr =build_roofunctor( #lambda x: 0.2
# frfit_dir + '/e_zlt_pt10_looseId_electronPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
#)
mu_tight_jetpt_fr = build_roofunctor( #lambda x: 0.1
frfit_dir + '/m_zlt_pt10_tightId_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
e_tight_jetpt_fr =build_roofunctor( #lambda x: 0.2
frfit_dir + '/e_zlt_pt10_tightId_electronJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
mu_loose_jetpt_fr = build_roofunctor( #lambda x: 0.1
frfit_dir + '/m_zlt_pt10_looseId_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
#mu_loose_fr = build_roofunctor( #lambda x: 0.1
# frfit_dir + '/m_zlt_pt10_looseId_muonPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
#)
#e_loose_fr =build_roofunctor( #lambda x: 0.2
# frfit_dir + '/e_zlt_pt10_looseId_electronPt.root',
# 'fit_efficiency', # workspace name
# 'efficiency'
#)
mu_tight_jetpt_fr = build_roofunctor( #lambda x: 0.1
frfit_dir + '/m_zlt_pt10_tightId_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
e_tight_jetpt_fr =build_roofunctor( #lambda x: 0.2
frfit_dir + '/e_zlt_pt10_tightId_electronJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
mu_loose_jetpt_fr = build_roofunctor( #lambda x: 0.1
frfit_dir + '/m_zlt_pt10_looseId_muonJetPt.root',
'fit_efficiency', # workspace name
'efficiency'
)
