def coefficients(tree, histdict): '''Computation of the rescaling coefficient, beta, theta and phi uncertainties''' from_tree = TreeInfo() # Run on events for iev, evt in enumerate(tree): if iev % 5000 == 0: print("Processing events {} over {}".format( iev, tree.GetEntries())) # Object from_tree to access the rootfile information from_tree.set_evt(evt) if from_tree.get_dim("reco_jet") != 2: continue if from_tree.get_dim("quark") != 2: continue if from_tree.get_pdgid("reco_lepton") < -90: continue # if evt.gen_jet2_theta < -1. or evt.gen_jet2_theta > 1.: # continue ######### Leptonic part g_lepton = Particle(from_tree.get_p4("gen_lepton")) r_lepton = Particle(from_tree.get_p4("reco_lepton")) if not r_lepton: continue histdict["h_uncert_energy"].Fill(r_lepton.get_E() - g_lepton.get_E()) ######## Hadronic part partons = [Particle() for i in range(from_tree.get_dim("quark"))] for iparton, parton in enumerate(partons): nparton = str("quark" + str(iparton + 1)) parton.set_p4(from_tree.get_p4(nparton)) gen_jets = [Jet() for i in range(from_tree.get_dim("gen_jet"))] for ijet, jet in enumerate(gen_jets): njet = str("gen_jet" + str(ijet + 1)) jet.set_p4(from_tree.get_p4(njet)) rec_jets = [Jet() for i in range(from_tree.get_dim("reco_jet"))] for ijet, jet in enumerate(rec_jets): njet = str("reco_jet" + str(ijet + 1)) jet.set_p4(from_tree.get_p4(njet)) if not rec_jets: continue association(partons, rec_jets, gen_jets, histdict)
def coefficients(tree, histdict, reco_type=''): '''Computation of the rescaling coefficient, theta and phi uncertainty''' from_tree = TreeInfo() # Run on events for iev, evt in enumerate(tree): if iev < 50000: if iev%5000 == 0: print("Processing events {} over {}".format(iev, tree.GetEntries())) # Object from_tree to access the rootfile information from_tree.set_evt(evt) if from_tree.get_dim("reco_jet{}".format(reco_type)) != 4: continue if from_tree.get_dim("quark") != 4: continue partons = [Particle() for i in range(from_tree.get_dim("quark"))] for iparton, parton in enumerate(partons): nparton = str("quark" + str(iparton + 1)) parton.set_p4(from_tree.get_p4(nparton)) gen_jets = [Jet()for i in range(from_tree.get_dim("gen_jet{}".format(reco_type)))] for ijet, jet in enumerate(gen_jets): njet = str("gen_jet" + str(ijet + 1) + reco_type) if from_tree.get_p4(njet).E() < 0: continue jet.set_p4(from_tree.get_p4(njet)) rec_jets = [Jet()for i in range(from_tree.get_dim("reco_jet{}".format(reco_type)))] for ijet, jet in enumerate(rec_jets): njet = str("reco_jet" + str(ijet + 1) + reco_type) if from_tree.get_p4(njet).E() < 0: continue jet.set_p4(from_tree.get_p4(njet)) if not rec_jets: continue association(partons, rec_jets, gen_jets, histdict) else: break
def w_mass_semi_leptonic(ecm, tree, histdict, reco_level): '''Run over the events and fill the histograms''' from_tree = TreeInfo() nDiscarded = 0 for iev, evt in enumerate(tree): if iev % 5000 == 0: print("Processing event {} on {} ".format(iev, tree.GetEntries())) # Object from_tree to access the rootfile information from_tree.set_evt(evt, reco_level) if from_tree.get_dim("lepton") != 1 or from_tree.get_dim("jet") != 2: nDiscarded += 1 continue if not check_jet_compo(from_tree): nDiscarded += 1 continue #### Leptonic decay part lepton = Particle(from_tree.get_p4("lepton")) misenergy = Particle(from_tree.get_p4("misenergy")) # Compute W mass from the lepton and the missing energy lep_decay = LeptonicDecay(lepton, misenergy) lep_decay.set_true_mass(from_tree.get_w_mass("leptonic_W")) lep_decay.compute() #### hadronic decay part jets = [Jet() for i in range(from_tree.get_dim("jet"))] for jid, jet in enumerate(jets): njet = str("jet" + str(jid + 1)) jet.set_p4(from_tree.get_p4(njet)) # Compute W mass from the jet and their angle dijet = Dijet(jets[0], jets[1]) dijet.set_true_mass(from_tree.get_w_mass("hadronic_W")) dijet.compute() fill_histograms(ecm, histdict, dijet, lep_decay) print nDiscarded
def w_mass_hadronic(tree, ecm, m_w, histdict, reconstruction, reco_level, reco_type=''): '''Run over the hadronic events and save the histograms''' from_tree = TreeInfo() for iev, evt in enumerate(tree): if iev % 5000 == 0: print("Processing events {} on {} ".format(iev, tree.GetEntries())) # Object from_tree to access the rootfile information from_tree.set_evt(evt, reco_level, reco_type) # Discard event where one jet is containing only photons and/or no clustered particles if from_tree.get_dim("jet") != 4: continue if not check_content(from_tree): continue # cut y34 # if from_tree.get_y34()<0.0015: # continue # L = ['1','2','3','4'] # for jetnumber in range(0,4): energyjet1 = from_tree.get_reco_jet1_e() energyjet2 = from_tree.get_reco_jet2_e() energyjet3 = from_tree.get_reco_jet3_e() energyjet4 = from_tree.get_reco_jet4_e() # print energyjet1 # print energyjet2 # print energyjet3 # print energyjet4 # partchargees=['11','13','130','211'] energypartchargeesjet1 = from_tree.get_reco_jet1_11_e( ) + from_tree.get_reco_jet1_13_e() + from_tree.get_reco_jet1_211_e() energypartchargeesjet2 = from_tree.get_reco_jet2_11_e( ) + from_tree.get_reco_jet2_13_e() + from_tree.get_reco_jet2_211_e() energypartchargeesjet3 = from_tree.get_reco_jet3_11_e( ) + from_tree.get_reco_jet3_13_e() + from_tree.get_reco_jet3_211_e() energypartchargeesjet4 = from_tree.get_reco_jet4_11_e( ) + from_tree.get_reco_jet4_13_e() + from_tree.get_reco_jet4_211_e() #cut energy part chargees < 0.9 * energy jet # if energypartchargeesjet1 > 0.9 * energyjet1 or energypartchargeesjet2 > 0.9 * energyjet2 or energypartchargeesjet3 > 0.9 * energyjet3 or energypartchargeesjet4 > 0.9 * energyjet4: # continue # print energypartchargeesjet1 # print energypartchargeesjet2 # print energypartchargeesjet3 # print energypartchargeesjet4 #for ptcharged in partchargees #get_reco_jet + L[jetnumber]+"_e" ## partchargees=['11','13','130','211'] ## for ptcharged in partchargees: ## energy_part_chargees_jet_number+=from_tree.get_reco_jet+L[jetnumber]+_+partchargees[ptcharged]+_e ## if energy_part_chargees_jet_number < 0.9*energyjet+L[jetnumber]: #continue #print y34 value for each event #print from_tree.get_y34() #for jet in range (4): #energypartjet = fromtree.getenergyjet #if pdgif = 211 or 11 or 13 # energypartchargeesjet+=energypartchargeesjet #if from_tree_get_jet1_e()<0.9.*energypartchargeesjet #continue #cut particules chargees, #for jet_"number" in range (0,4): #energyjet"number" = fromtree.get_reco_jet_"number"_e # for part in (list particules chargees): # energy_part_chargees_jet_"number"+=from_tree?get_reco_jet_"number"_"part"_e #if energy_part_chargees_jet_"number"<0.9.*energy_part_chargees_jet_"number" # continue # Jets pvec = [] for part in range(4): njet = str("jet" + str(part + 1)) pvec.append(from_tree.get_p4(njet)) jets = [Jet() for i in range(from_tree.get_dim("jet"))] # Direct reconstruction if reconstruction == "direct": for jid, jet in enumerate(jets): jet.set_id(str("jet" + str(jid + 1))) jet.set_mref(m_w) jet.set_p4(pvec[jid]) # Energy Rescaling else: beta = get_beta_matrix(pvec) energy = energy_from_direction(ecm, beta, from_tree.get_dim("jet")) # discarded events with energy < 0 solutions (no physics) if any(energy < 0): continue # Need to rescale the p4 matrix p4_resc = rescaling_factors.rescale( pvec, rescaling_factors.factor(energy, pvec)) for jid, jet in enumerate(jets): jet.set_id(str("jet" + str(jid + 1))) jet.set_mref(m_w) jet.set_p4(p4_resc[jid]) dijet = get_dijets_pairs(jets) # If the pairing conditions are not full if not dijet: continue fill_histograms(dijet, histdict)
def w_mass_hadronic(tree, ecm, m_w, histdict, reconstruction, reco_level, reco_type=''): '''Run over the hadronic events and save the histograms''' from_tree = TreeInfo() sign = {} #eeqq bkg1 = {} #ZZqqll bkg2 = {} #ZZqqqq bkg3 = {} #WWqqll bkg4 = {} purete = [] efficacite = [] x = [] #cut y34 # for i in range(0,101): # j=i*0.001 # x.append(j) # print x # for index, value in enumerate(x): # print value #cut ejetcut # for i in range(0,2): # j=i*0.001 # x.append(j) # print x # print len(x) #boucle sur les valeurs de cut for iev, evt in enumerate(tree): if iev % 5000 == 0: print("Processing events {} on {} ".format(iev, tree.GetEntries())) # Object from_tree to access the rootfile information from_tree.set_evt(evt, reco_level, reco_type) # Discard event where one jet is containing only photons and/or no clustered particles if from_tree.get_dim("jet") != 4: continue if not check_content(from_tree): continue #cut masses "similaires" des 2 W #if abs(m_large-m_small)<13: # continue # cut y34 if from_tree.get_y34() <= 0.001: continue #cut energy part chargees < 0.96 * energy jet energyjet1 = from_tree.get_reco_jet1_e() energyjet2 = from_tree.get_reco_jet2_e() energyjet3 = from_tree.get_reco_jet3_e() energyjet4 = from_tree.get_reco_jet4_e() # partchargees=['11','13','211'] energypartchargeesjet1 = from_tree.get_reco_jet1_11_e( ) + from_tree.get_reco_jet1_13_e() + from_tree.get_reco_jet1_211_e() energypartchargeesjet2 = from_tree.get_reco_jet2_11_e( ) + from_tree.get_reco_jet2_13_e() + from_tree.get_reco_jet2_211_e() energypartchargeesjet3 = from_tree.get_reco_jet3_11_e( ) + from_tree.get_reco_jet3_13_e() + from_tree.get_reco_jet3_211_e() energypartchargeesjet4 = from_tree.get_reco_jet4_11_e( ) + from_tree.get_reco_jet4_13_e() + from_tree.get_reco_jet4_211_e() if energypartchargeesjet1 > 0.996 * energyjet1 or energypartchargeesjet2 > 0.996 * energyjet2 or energypartchargeesjet3 > 0.996 * energyjet3 or energypartchargeesjet4 > 0.996 * energyjet4: continue # Jets pvec = [] for part in range(4): njet = str("jet" + str(part + 1)) pvec.append(from_tree.get_p4(njet)) jets = [Jet() for i in range(from_tree.get_dim("jet"))] # Direct reconstruction if reconstruction == "direct": for jid, jet in enumerate(jets): jet.set_id(str("jet" + str(jid + 1))) jet.set_mref(m_w) jet.set_p4(pvec[jid]) # Energy Rescaling else: beta = get_beta_matrix(pvec) energy = energy_from_direction(ecm, beta, from_tree.get_dim("jet")) # discarded events with energy < 0 solutions (no physics) if any(energy < 0): continue # Need to rescale the p4 matrix p4_resc = rescaling_factors.rescale( pvec, rescaling_factors.factor(energy, pvec)) for jid, jet in enumerate(jets): jet.set_id(str("jet" + str(jid + 1))) jet.set_mref(m_w) jet.set_p4(p4_resc[jid]) dijet = get_dijets_pairs(jets) # If the pairing conditions are not full if not dijet: continue #cut energie dijets similaires diffedjets = return_edijets(dijet) if diffedjets >= 30: continue openingangledj1, openingangledj2 = return_opening_angle(dijet) #if openingangledj1 <= 80 or openingangledj2 <= 80: # continue m_small, m_large = return_mass(dijet) #cut sur la masse de mW2 #if m_large<=65 or m_small<40: # continue #fonction qui rempli les histos fill_histograms(dijet, histdict)
def coefficients(tree, histdict, reco_type=''): '''Computation of the rescaling coefficient, theta and phi uncertainty''' from_tree = TreeInfo() # Run on events for iev, evt in enumerate(tree): if iev % 5000 == 0: print("Processing events {} over {}".format( iev, tree.GetEntries())) # Object from_tree to access the rootfile information from_tree.set_evt(evt) #value for normalisation histweight = from_tree.get_file_type() if (histweight == 1 or histweight == 2 or histweight == 3): print histweight #value distinguish ZZllll from ZZqqll from ZZqqqq in ZZall ZZqqll = 0 ZZqqqq = 0 if (histweight == 3 & from_tree.get_dim("quark") == 2): ZZqqll = 1 if (histweight == 3 & from_tree.get_dim("quark") == 4): ZZqqqq = 1 #ecarte les evenements purement leptoniques du background ZZAll #if (histweight == 3 & from_tree.get_dim("quark") == 0): # continue #if (histweight==0 or histweight ==3): #if from_tree.get_dim("reco_jet{}".format(reco_type)) != 4: # continue #if(histweight==1 or histweight ==2): # if from_tree.get_dim("reco_jet{}".format(reco_type)) != 2: # continue #if from_tree.get_dim("quark") != 4: # continue partons = [Particle() for i in range(from_tree.get_dim("quark"))] #print "range partons "+str(from_tree.get_dim("quark")) for iparton, parton in enumerate(partons): nparton = str("quark" + str(iparton + 1)) #print "nparton "+str(nparton) parton.set_p4(from_tree.get_p4(nparton)) gen_jets = [ Jet() for i in range(from_tree.get_dim("gen_jet{}".format(reco_type))) ] #print "gen_jets"+str(from_tree.get_dim("gen_jet{}".format(reco_type))) for ijet, jet in enumerate(gen_jets): #print "range ijet "+str(gen_jets) njet = str("gen_jet" + str(ijet + 1) + reco_type) #print njet if from_tree.get_p4(njet).E() < 0: continue jet.set_p4(from_tree.get_p4(njet)) rec_jets = [ Jet() for i in range(from_tree.get_dim("reco_jet{}".format(reco_type))) ] for ijet, jet in enumerate(rec_jets): njet = str("reco_jet" + str(ijet + 1) + reco_type) if from_tree.get_p4(njet).E() < 0: continue jet.set_p4(from_tree.get_p4(njet)) if not rec_jets: print "No rec_jets" continue #print iev association(partons, rec_jets, gen_jets, histdict, histweight, ZZqqll, ZZqqqq)
def w_mass_hadronic(tree, ecm, m_w, histdict, reconstruction, reco_level, reco_type=''): '''Run over the hadronic events and save the histograms''' f1 = TFile('signal_and_bkg_cut_240GeV.root', "update") treecut = TTree(treename, "tree title") y34 = array('f', [0]) reco_jet1_e = array('f', [0]) reco_jet2_e = array('f', [0]) reco_jet3_e = array('f', [0]) reco_jet4_e = array('f', [0]) reco_jet1_px = array('f', [0]) reco_jet2_px = array('f', [0]) reco_jet3_px = array('f', [0]) reco_jet4_px = array('f', [0]) reco_jet1_py = array('f', [0]) reco_jet2_py = array('f', [0]) reco_jet3_py = array('f', [0]) reco_jet4_py = array('f', [0]) reco_jet1_pz = array('f', [0]) reco_jet2_pz = array('f', [0]) reco_jet3_pz = array('f', [0]) reco_jet4_pz = array('f', [0]) reco_jet_size = array('f', [0]) file_type = array('f', [0]) quark_size = array('f', [0]) quark1_e = array('f', [0]) quark2_e = array('f', [0]) quark3_e = array('f', [0]) quark4_e = array('f', [0]) quark1_px = array('f', [0]) quark2_px = array('f', [0]) quark3_px = array('f', [0]) quark4_px = array('f', [0]) quark1_py = array('f', [0]) quark2_py = array('f', [0]) quark3_py = array('f', [0]) quark4_py = array('f', [0]) quark1_pz = array('f', [0]) quark2_pz = array('f', [0]) quark3_pz = array('f', [0]) quark4_pz = array('f', [0]) gen_jet_size = array('f', [0]) gen_jet1_e = array('f', [0]) gen_jet2_e = array('f', [0]) gen_jet3_e = array('f', [0]) gen_jet4_e = array('f', [0]) gen_jet1_px = array('f', [0]) gen_jet2_px = array('f', [0]) gen_jet3_px = array('f', [0]) gen_jet4_px = array('f', [0]) gen_jet1_py = array('f', [0]) gen_jet2_py = array('f', [0]) gen_jet3_py = array('f', [0]) gen_jet4_py = array('f', [0]) gen_jet1_pz = array('f', [0]) gen_jet2_pz = array('f', [0]) gen_jet3_pz = array('f', [0]) gen_jet4_pz = array('f', [0]) treecut.Branch("y34", y34, 'y34/F') treecut.Branch("reco_jet1_e", reco_jet1_e, 'reco_jet1_e/F') treecut.Branch("reco_jet2_e", reco_jet2_e, 'reco_jet2_e/F') treecut.Branch("reco_jet3_e", reco_jet3_e, 'reco_jet3_e/F') treecut.Branch("reco_jet4_e", reco_jet4_e, 'reco_jet4_e/F') treecut.Branch("reco_jet1_px", reco_jet1_px, 'reco_jet1_px/F') treecut.Branch("reco_jet2_px", reco_jet2_px, 'reco_jet2_px/F') treecut.Branch("reco_jet3_px", reco_jet3_px, 'reco_jet3_px/F') treecut.Branch("reco_jet4_px", reco_jet4_px, 'reco_jet4_px/F') treecut.Branch("reco_jet1_py", reco_jet1_py, 'reco_jet1_py/F') treecut.Branch("reco_jet2_py", reco_jet2_py, 'reco_jet2_py/F') treecut.Branch("reco_jet3_py", reco_jet3_py, 'reco_jet3_py/F') treecut.Branch("reco_jet4_py", reco_jet4_py, 'reco_jet4_py/F') treecut.Branch("reco_jet1_pz", reco_jet1_pz, 'reco_jet1_pz/F') treecut.Branch("reco_jet2_pz", reco_jet2_pz, 'reco_jet2_pz/F') treecut.Branch("reco_jet3_pz", reco_jet3_pz, 'reco_jet3_pz/F') treecut.Branch("reco_jet4_pz", reco_jet4_pz, 'reco_jet4_pz/F') treecut.Branch("reco_jet_size", reco_jet_size, 'reco_jet_size/F') treecut.Branch("file_type", file_type, 'file_type/F') treecut.Branch("quark_size", quark_size, 'quark_size/F') treecut.Branch("quark1_e", quark1_e, 'quark1_e/F') treecut.Branch("quark2_e", quark2_e, 'quark2_e/F') treecut.Branch("quark3_e", quark3_e, 'quark3_e/F') treecut.Branch("quark4_e", quark4_e, 'quark4_e/F') treecut.Branch("quark1_px", quark1_px, 'quark1_px/F') treecut.Branch("quark2_px", quark2_px, 'quark2_px/F') treecut.Branch("quark3_px", quark3_px, 'quark3_px/F') treecut.Branch("quark4_px", quark4_px, 'quark4_px/F') treecut.Branch("quark1_py", quark1_py, 'quark1_py/F') treecut.Branch("quark2_py", quark2_py, 'quark2_py/F') treecut.Branch("quark3_py", quark3_py, 'quark3_py/F') treecut.Branch("quark4_py", quark4_py, 'quark4_py/F') treecut.Branch("quark1_pz", quark1_pz, 'quark1_pz/F') treecut.Branch("quark2_pz", quark2_pz, 'quark2_pz/F') treecut.Branch("quark3_pz", quark3_pz, 'quark3_pz/F') treecut.Branch("quark4_pz", quark4_pz, 'quark4_pz/F') treecut.Branch("gen_jet_size", gen_jet_size, 'gen_jet_size/F') treecut.Branch("gen_jet1_e", gen_jet1_e, 'gen_jet1_e/F') treecut.Branch("gen_jet2_e", gen_jet2_e, 'gen_jet2_e/F') treecut.Branch("gen_jet3_e", gen_jet3_e, 'gen_jet3_e/F') treecut.Branch("gen_jet4_e", gen_jet4_e, 'gen_jet4_e/F') treecut.Branch("gen_jet1_px", gen_jet1_px, 'gen_jet1_px/F') treecut.Branch("gen_jet2_px", gen_jet2_px, 'gen_jet2_px/F') treecut.Branch("gen_jet3_px", gen_jet3_px, 'gen_jet3_px/F') treecut.Branch("gen_jet4_px", gen_jet4_px, 'gen_jet4_px/F') treecut.Branch("gen_jet1_py", gen_jet1_py, 'gen_jet1_py/F') treecut.Branch("gen_jet2_py", gen_jet2_py, 'gen_jet2_py/F') treecut.Branch("gen_jet3_py", gen_jet3_py, 'gen_jet3_py/F') treecut.Branch("gen_jet4_py", gen_jet4_py, 'gen_jet4_py/F') treecut.Branch("gen_jet1_pz", gen_jet1_pz, 'gen_jet1_pz/F') treecut.Branch("gen_jet2_pz", gen_jet2_pz, 'gen_jet2_pz/F') treecut.Branch("gen_jet3_pz", gen_jet3_pz, 'gen_jet3_pz/F') treecut.Branch("gen_jet4_pz", gen_jet4_pz, 'gen_jet4_pz/F') from_tree = TreeInfo() sign = {} #eeqq bkg1 = {} #ZZqqll bkg2 = {} #ZZqqqq bkg3 = {} #WWqqll bkg4 = {} purete = [] efficacite = [] x = [] #cut y34 # for i in range(0,101): # j=i*0.001 # x.append(j) # print x # for index, value in enumerate(x): # print value #cut ejetcut # for i in range(0,2): # j=i*0.001 # x.append(j) # print x # print len(x) #boucle sur les valeurs de cut for iev, evt in enumerate(tree): if iev % 5000 == 0: print("Processing events {} on {} ".format(iev, tree.GetEntries())) # Object from_tree to access the rootfile information from_tree.set_evt(evt, reco_level, reco_type) # Discard event where one jet is containing only photons and/or no clustered particles if from_tree.get_dim("jet") != 4: continue if not check_content(from_tree): continue #cut masses "similaires" des 2 W #if abs(m_large-m_small)<13: # continue # cut y34 if from_tree.get_y34() <= 0.001: continue #cut energy part chargees < 0.96 * energy jet energyjet1 = from_tree.get_reco_jet1_e() energyjet2 = from_tree.get_reco_jet2_e() energyjet3 = from_tree.get_reco_jet3_e() energyjet4 = from_tree.get_reco_jet4_e() # partchargees=['11','13','211'] energypartchargeesjet1 = from_tree.get_reco_jet1_11_e( ) + from_tree.get_reco_jet1_13_e() + from_tree.get_reco_jet1_211_e() energypartchargeesjet2 = from_tree.get_reco_jet2_11_e( ) + from_tree.get_reco_jet2_13_e() + from_tree.get_reco_jet2_211_e() energypartchargeesjet3 = from_tree.get_reco_jet3_11_e( ) + from_tree.get_reco_jet3_13_e() + from_tree.get_reco_jet3_211_e() energypartchargeesjet4 = from_tree.get_reco_jet4_11_e( ) + from_tree.get_reco_jet4_13_e() + from_tree.get_reco_jet4_211_e() if energypartchargeesjet1 > 0.996 * energyjet1 or energypartchargeesjet2 > 0.996 * energyjet2 or energypartchargeesjet3 > 0.996 * energyjet3 or energypartchargeesjet4 > 0.996 * energyjet4: continue # Jets pvec = [] for part in range(4): njet = str("jet" + str(part + 1)) pvec.append(from_tree.get_p4(njet)) jets = [Jet() for i in range(from_tree.get_dim("jet"))] # Direct reconstruction if reconstruction == "direct": for jid, jet in enumerate(jets): jet.set_id(str("jet" + str(jid + 1))) jet.set_mref(m_w) jet.set_p4(pvec[jid]) # Energy Rescaling else: beta = get_beta_matrix(pvec) energy = energy_from_direction(ecm, beta, from_tree.get_dim("jet")) # discarded events with energy < 0 solutions (no physics) if any(energy < 0): continue # Need to rescale the p4 matrix p4_resc = rescaling_factors.rescale( pvec, rescaling_factors.factor(energy, pvec)) for jid, jet in enumerate(jets): jet.set_id(str("jet" + str(jid + 1))) jet.set_mref(m_w) jet.set_p4(p4_resc[jid]) dijet = get_dijets_pairs(jets) # If the pairing conditions are not full if not dijet: continue #cut energie dijets similaires #diffedjets = return_edijets(dijet) #if diffedjets>=30: continue openingangledj1, openingangledj2 = return_opening_angle(dijet) if openingangledj1 <= 60 or openingangledj2 <= 60 or openingangledj1 >= 145 or openingangledj2 >= 145: continue m_small, m_large = return_mass(dijet) #cut sur la masse de mW1 small mW2 large if m_large <= 70: #or m_small<40: continue #fonction qui rempli les histos fill_histograms(dijet, histdict) #y34=from_tree.get_y34() #treecuts.Fill() y34[0] = from_tree.get_y34() reco_jet1_e[0] = from_tree.get_reco_jet1_e() reco_jet2_e[0] = from_tree.get_reco_jet2_e() reco_jet3_e[0] = from_tree.get_reco_jet3_e() reco_jet4_e[0] = from_tree.get_reco_jet4_e() reco_jet1_px[0] = from_tree.get_reco_jet1_px() reco_jet2_px[0] = from_tree.get_reco_jet2_px() reco_jet3_px[0] = from_tree.get_reco_jet3_px() reco_jet4_px[0] = from_tree.get_reco_jet4_px() reco_jet1_py[0] = from_tree.get_reco_jet1_py() reco_jet2_py[0] = from_tree.get_reco_jet2_py() reco_jet3_py[0] = from_tree.get_reco_jet3_py() reco_jet4_py[0] = from_tree.get_reco_jet4_py() reco_jet1_pz[0] = from_tree.get_reco_jet1_pz() reco_jet2_pz[0] = from_tree.get_reco_jet2_pz() reco_jet3_pz[0] = from_tree.get_reco_jet3_pz() reco_jet4_pz[0] = from_tree.get_reco_jet4_pz() reco_jet_size[0] = from_tree.get_reco_jet_size() quark_size[0] = from_tree.get_quark_size() file_type[0] = float(fileoutput1) quark1_e[0] = from_tree.get_quark1_e() quark2_e[0] = from_tree.get_quark2_e() quark3_e[0] = from_tree.get_quark3_e() quark4_e[0] = from_tree.get_quark4_e() quark1_px[0] = from_tree.get_quark1_px() quark2_px[0] = from_tree.get_quark2_px() quark3_px[0] = from_tree.get_quark3_px() quark4_px[0] = from_tree.get_quark4_px() quark1_py[0] = from_tree.get_quark1_py() quark2_py[0] = from_tree.get_quark2_py() quark3_py[0] = from_tree.get_quark3_py() quark4_py[0] = from_tree.get_quark4_py() quark1_pz[0] = from_tree.get_quark1_pz() quark2_pz[0] = from_tree.get_quark2_pz() quark3_pz[0] = from_tree.get_quark3_pz() quark4_pz[0] = from_tree.get_quark4_pz() gen_jet_size[0] = from_tree.get_gen_jet_size() gen_jet1_e[0] = from_tree.get_gen_jet1_e() gen_jet2_e[0] = from_tree.get_gen_jet2_e() gen_jet3_e[0] = from_tree.get_gen_jet3_e() gen_jet4_e[0] = from_tree.get_gen_jet4_e() gen_jet1_px[0] = from_tree.get_gen_jet1_px() gen_jet2_px[0] = from_tree.get_gen_jet2_px() gen_jet3_px[0] = from_tree.get_gen_jet3_px() gen_jet4_px[0] = from_tree.get_gen_jet4_px() gen_jet1_py[0] = from_tree.get_gen_jet1_py() gen_jet2_py[0] = from_tree.get_gen_jet2_py() gen_jet3_py[0] = from_tree.get_gen_jet3_py() gen_jet4_py[0] = from_tree.get_gen_jet4_py() gen_jet1_pz[0] = from_tree.get_gen_jet1_pz() gen_jet2_pz[0] = from_tree.get_gen_jet2_pz() gen_jet3_pz[0] = from_tree.get_gen_jet3_pz() gen_jet4_pz[0] = from_tree.get_gen_jet4_pz() #print str(quark4_pz[0]) treecut.Fill() f1.cd() f1.Write() f1.Close()