File indexing completed on 2024-11-25 02:29:53
0001 import ROOT
0002 from DataFormats.FWLite import Handle, Events
0003 import numpy as np
0004 import pickle
0005
0006 events_c = Events('output_test_DDX.root')
0007
0008 handleJ = Handle("std::vector<pat::Jet>")
0009 labelJ = ("selectedUpdatedPatJets","","PATtest")
0010
0011 h_probQ_ddb = ROOT.TH1F('h_probQ_ddb', ';prob Q;', 40, 0., 1.)
0012 h_probH_ddb = ROOT.TH1F('h_probH_ddb', ';prob H;', 40, 0., 1.)
0013
0014 h_probQ_ddc = ROOT.TH1F('h_probQ_ddc', ';prob Q;', 40, 0., 1.)
0015 h_probH_ddc = ROOT.TH1F('h_probH_ddc', ';prob H;', 40, 0., 1.)
0016
0017 info = {}
0018 info['pt'] = []
0019 info['eta'] = []
0020 info['mass'] = []
0021 info['BvL'] = []
0022 info['CvL'] = []
0023 info['CvB'] = []
0024
0025
0026 for iev, event in enumerate(events_c):
0027 event.getByLabel (labelJ, handleJ)
0028 jets = handleJ.product()
0029 print(iev)
0030 for jet in jets:
0031
0032
0033
0034 print(jet.pt(), jet.mass(), jet.eta())
0035 print("DDB", jet.bDiscriminator("pfDeepDoubleBvLJetTags:probQCD"), jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb"))
0036
0037 print("DDCvL", jet.bDiscriminator("pfDeepDoubleCvLJetTags:probQCD"), jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHcc"))
0038
0039 print("DDCvB", jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHbb"), jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHcc") )
0040
0041 h_probQ_ddb.Fill(jet.bDiscriminator("pfDeepDoubleBvLJetTags:probQCD"))
0042 h_probH_ddb.Fill(jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb"))
0043 h_probQ_ddc.Fill(jet.bDiscriminator("pfDeepDoubleCvLJetTags:probQCD"))
0044 h_probH_ddc.Fill(jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHcc"))
0045
0046 info['mass'].append(jet.mass())
0047 info['pt'].append(jet.pt())
0048 info['eta'].append(jet.eta())
0049 info['BvL'].append(jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb"))
0050 info['CvL'].append(jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHcc"))
0051 info['CvB'].append(jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHcc"))
0052
0053 with open('outputs.pkl', 'wb') as handle:
0054 pickle.dump(info, handle)
0055
0056 c1a = ROOT.TCanvas()
0057 h_probH_ddb.Draw("HISTO")
0058 h_probH_ddb.SetLineColor(632)
0059 h_probH_ddb.SetLineStyle(10)
0060 h_probQ_ddb.Draw("SAME")
0061 c1a.Draw()
0062 c1a.SaveAs("ProbQ_vc_vb.png")
0063
0064 c1b = ROOT.TCanvas()
0065 h_probH_ddc.Draw("HISTO")
0066 h_probH_ddc.SetLineColor(632)
0067 h_probH_ddc.SetLineStyle(10)
0068 h_probQ_ddc.Draw("SAME")
0069 c1b.Draw()
0070 c1b.SaveAs("ProbH_vc_vb.png")