from ROOT import TFile, TCanvas, TH1F, TLorentzVector, TH2F
%jsroot on
file = TFile("data/Dielectrons.root","READ");
Let's calculate the invariant mass $M$ of two electrons
Canvas = TCanvas()
histogram = TH1F("Invariant Mass","ee mass; ee mass [GeV];Events", 900, 2, 110)
for diele in file.Dielectrons:
electron1 = TLorentzVector(diele.Electron1_Px, diele.Electron1_Py, diele.Electron1_Pz, diele.Electron1_Energy)
electron2 = TLorentzVector(diele.Electron2_Px, diele.Electron2_Py, diele.Electron2_Pz, diele.Electron2_Energy)
InvariantMass = (electron1 + electron2).M()
histogram.Fill(InvariantMass)
histogram.Draw()
Canvas.SetLogy()
Canvas.Draw()
dielectroncharge = TH2F("dielectroncharge", "charge comparison; electron 1 charge; electron 2 charge", 10, -2, 2, 10, -2, 2)