expanded dimuon content

Author: drkovalskyi

PhysicsTools/BPHNano/plugins/DiLeptonBuilder.cc

@@ -8,6 +8,7 @@
 
 #include "CommonTools/Statistics/interface/ChiSquaredProbability.h"
 #include "CommonTools/Utils/interface/StringCutObjectSelector.h"
+#include "DataFormats/BeamSpot/interface/BeamSpot.h"
 #include "DataFormats/Math/interface/deltaR.h"
 #include "DataFormats/PatCandidates/interface/CompositeCandidate.h"
 #include "FWCore/Framework/interface/Event.h"
@@ -33,6 +34,7 @@ class DiLeptonBuilder : public edm::global::EDProducer<> {
         pre_vtx_selection_{cfg.getParameter<std::string>("preVtxSelection")},
         post_vtx_selection_{cfg.getParameter<std::string>("postVtxSelection")},
         src_{consumes<LeptonCollection>(cfg.getParameter<edm::InputTag>("src"))},
+        beamspot_{consumes<reco::BeamSpot>(cfg.getParameter<edm::InputTag>("beamSpot"))},
         ttracks_src_{consumes<TransientTrackCollection>(cfg.getParameter<edm::InputTag>("transientTracksSrc"))} {
     produces<pat::CompositeCandidateCollection>("SelectedDiLeptons");
   }
@@ -47,6 +49,7 @@ class DiLeptonBuilder : public edm::global::EDProducer<> {
   const StringCutObjectSelector<pat::CompositeCandidate> pre_vtx_selection_;   // cut on the di-lepton before the SV fit
   const StringCutObjectSelector<pat::CompositeCandidate> post_vtx_selection_;  // cut on the di-lepton after the SV fit
   const edm::EDGetTokenT<LeptonCollection> src_;
+  const edm::EDGetTokenT<reco::BeamSpot> beamspot_;
   const edm::EDGetTokenT<TransientTrackCollection> ttracks_src_;
 };
 
@@ -56,6 +59,9 @@ void DiLeptonBuilder<Lepton>::produce(edm::StreamID, edm::Event &evt, edm::Event
   edm::Handle<LeptonCollection> leptons;
   evt.getByToken(src_, leptons);
 
+  edm::Handle<reco::BeamSpot> beamspot;
+  evt.getByToken(beamspot_, beamspot);
+
   edm::Handle<TransientTrackCollection> ttracks;
   evt.getByToken(ttracks_src_, ttracks);
 
@@ -103,11 +109,16 @@ void DiLeptonBuilder<Lepton>::produce(edm::StreamID, edm::Event &evt, edm::Event
       lepton_pair.addUserFloat(
           "fitted_massErr",
           fitter.success() ? sqrt(fitter.fitted_candidate().kinematicParametersError().matrix()(6, 6)) : -1);
+      auto fit_p4 = fitter.fitted_p4();
       lepton_pair.addUserFloat("vtx_x", lepton_pair.vx());
       lepton_pair.addUserFloat("vtx_y", lepton_pair.vy());
       lepton_pair.addUserFloat("vtx_z", lepton_pair.vz());
+      lepton_pair.addUserFloat("cos_theta_2D", bph::cos_theta_2D(fitter, *beamspot, lepton_pair.p4()));
+      lepton_pair.addUserFloat("fitted_cos_theta_2D", bph::cos_theta_2D(fitter, *beamspot, fit_p4));
 
-      // if needed, add here more stuff
+      auto lxy = bph::l_xy(fitter, *beamspot);
+      lepton_pair.addUserFloat("l_xy", lxy.value());
+      lepton_pair.addUserFloat("l_xy_unc", lxy.error());
 
       // cut on the SV info
       if (!post_vtx_selection_(lepton_pair))

PhysicsTools/BPHNano/python/MuMu_cff.py

@@ -9,6 +9,7 @@
     transientTracksSrc = cms.InputTag('muonBPH', 'SelectedTransientMuons'),
     lep1Selection = cms.string('pt > 4.0 && abs(eta) < 2.4 && isLooseMuon && isGlobalMuon'),
     lep2Selection = cms.string('pt > 3.0 && abs(eta) < 2.4 && isLooseMuon && isGlobalMuon'),
+    beamSpot = cms.InputTag("offlineBeamSpot"),
     preVtxSelection  = cms.string('abs(userCand("l1").vz - userCand("l2").vz) <= 1.'
                                   '&& 0 < mass() && mass() < 15.0 '
                                   '&& charge() == 0'
@@ -33,11 +34,18 @@
     singleton = cms.bool(False), # the number of entries is variable
     extension = cms.bool(False), # this is the main table for the muons
     variables = cms.PSet(CandVars,
-          fitted_mass = Var("userFloat('fitted_mass')", float, doc="Fitted dilepton mass", precision=10),
-          svprob = Var("userFloat('sv_prob')", float, doc="Vtx fit probability", precision=10),
-          vtx_x =Var("userFloat('vtx_x')", float, doc="Vtx position in x", precision=10),
-          vtx_y = Var("userFloat('vtx_y')", float, doc="Vtx position in y", precision=10),
-          vtx_z = Var("userFloat('vtx_z')", float, doc="Vtx position in y", precision=10),
+        l1_idx = Var("userInt('l1_idx')", int, doc = "leading muon index to the BPH muon collection"),
+        l2_idx = Var("userInt('l2_idx')", int, doc = "subleading muon index to the BPH muon collection"),
+        fit_mass = Var("userFloat('fitted_mass')", float, doc="Fitted dilepton mass", precision=10),
+        fit_massErr = Var("userFloat('fitted_massErr')", float, doc = "post-fit uncertainty of the mass of the B candidate", precision=10),
+        svprob = Var("userFloat('sv_prob')", float, doc="Vtx fit probability", precision=10),
+        l_xy     = Var("userFloat('l_xy')", float, doc = "post-fit vertex displacement on transverse plane wrt beamspot", precision=10),
+        l_xy_unc = Var("userFloat('l_xy_unc')", float, doc = "post-fit vertex uncertainty of displacement on transverse plane wrt beamspot", precision=10),
+        vtx_x =Var("userFloat('vtx_x')", float, doc="Vtx position in x", precision=10),
+        vtx_y = Var("userFloat('vtx_y')", float, doc="Vtx position in y", precision=10),
+        vtx_z = Var("userFloat('vtx_z')", float, doc="Vtx position in y", precision=10),
+        cos2D     = Var("userFloat('cos_theta_2D')", float, doc = "cos 2D of pre-fit candidate wrt beamspot", precision=10),
+        fit_cos2D = Var("userFloat('fitted_cos_theta_2D')", float, doc = "cos 2D of fitted vertex wrt beamspot", precision=10),
 
     )
 )