def test_pythia(self):
components = load_components(mode='pythia')
def test_heppy(self):
components = load_components(mode='heppy')
from cpyroot import *
# from tdrstyle import tdrstyle
from fcc_ee_higgs.components.all import load_components
components = load_components(mode='heppy')
from fitter import TemplateFitter, BaseFitter, BallFitter
from plotter import Plotter
from marginal_efficiency import marginal_efficiency
plot = None
if __name__ == '__main__':
from ROOT import RooRealVar, RooDataHist, RooHistPdf, RooArgList, RooArgSet, TH1
ZZ = components['ee_to_ZZ_Sep12_ZHnunubb_Sep21_A_13']
ZZ.name = 'ZZ'
ZH = components['ee_to_ZH_Z_to_nunu_Sep12_ZHnunubb_Sep21_A_12']
ZH.name = 'ZH'
ffbar = components['ee_to_ffbar_Sep12_ZHnunubb_Sep21_B_15']
ffbar.name = 'ffbar'
comps = [ZZ, ZH]
lumi = 5e5
# lumi = 5e6 # 5ab-1
plotter = Plotter(comps, lumi)
cut_missmass = 'misenergy_m>65 && misenergy_m<125'
# cut_missmass= 'misenergy_m>75 && misenergy_m<150'
# cut_h_bb = 'jet1_e>0 && jet2_e>0 && (jet1_bmatch==1 && jet2_bmatch==1)'
from fcc_ee_higgs.components.all import load_components
all_components = load_components()
from varmap_janik import varmap
from cpyroot.tools.treecomparator import TreeComparator
from cpyroot.tools.splitcanvas import SplitCanvas
from ROOT import TPad, TCanvas
import pprint
def print_vars(tree):
names = []
for v in tree.GetListOfBranches():
names.append(v.GetName())
pprint.pprint(names)
comps = dict(janik=all_components['ee_to_ZZ_Janik_Sep18_ZHnunubb_A_10'],
colin=all_components['ee_to_ZZ_Sep19_ZHnunubb_A_11'])
for vcolin, vjanik in varmap.iteritems():
comps['janik'].tree.SetAlias(vcolin, vjanik)
tc = comps['colin'].tree
tj = comps['janik'].tree
vars = [
'higgs_acol',
'higgs_cross',
'higgs_rescaled_m',
]
# definition of the collider
# help(Collider) for more information
from heppy.configuration import Collider
Collider.BEAMS = 'ee'
Collider.SQRTS = 240.
mode = 'ee_to_ZZ_Sep12_A_2'
nfiles = 20
#mode = 'test'
# definition of input samples
# from components.ZH_Znunu import components as cps
from fcc_ee_higgs.components.all import load_components
cps = load_components(mode='pythia')
selectedComponents = cps.values()
for comp in selectedComponents:
comp.splitFactor = len(comp.files)
test_filename = 'samples/test/ee_ZZ_nunu.root'
if mode == 'test':
comp = cps['ee_to_ZZ_Sep12_A_2']
comp.files = [test_filename]
comp.splitFactor = 1
selectedComponents = [comp]
elif mode == 'debug':
comp = cfg.Component('Debug', files=['ee_ffbar.root'])
selectedComponents = [comp]
else: