Produce lepton time-life info table without using intermediate lepton…

… collection

DataFormats/NanoAOD/BuildFile.xml

@@ -1,4 +1,5 @@
 <use name="FWCore/Utilities"/>
+<use name="DataFormats/CLHEP"/>
 <use name="DataFormats/Common"/>
 <use name="DataFormats/Math"/>
 <use name="boost"/>

DataFormats/NanoAOD/interface/LeptonTimeLifeInfo.h

@@ -0,0 +1,73 @@
+#ifndef DataFormats_NanoAOD_LeptonTimeLifeInfo_h
+#define DataFormats_NanoAOD_LeptonTimeLifeInfo_h
+
+/**
+  \class    LeptonTimeLifeInfo
+  \brief    Structure to hold lepton life-time information
+
+  \author   Michal Bluj, NCBJ, Warsaw
+*/
+
+#include "DataFormats/GeometryCommonDetAlgo/interface/GlobalError.h"
+#include "DataFormats/GeometryCommonDetAlgo/interface/Measurement1D.h"
+#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
+#include "DataFormats/GeometryVector/interface/GlobalVector.h"
+#include "DataFormats/TrackReco/interface/Track.h"
+#include "DataFormats/VertexReco/interface/Vertex.h"
+
+class LeptonTimeLifeInfo {
+public:
+  LeptonTimeLifeInfo();
+  ~LeptonTimeLifeInfo() {}
+
+  // Secondary vertex
+  void setSV(reco::Vertex sv) {
+    sv_ = sv;
+    hasSV_ = true;
+  }
+  const reco::Vertex& sv() const { return sv_; }
+  bool hasSV() const { return hasSV_; }
+  void setFlightVector(GlobalVector flight_vec, const GlobalError& flight_cov) {
+    flight_vec_ = flight_vec;
+    flight_cov_ = flight_cov;
+  }
+  // Flight-path
+  const GlobalVector& flightVector() const { return flight_vec_; }
+  const GlobalError& flightCovariance() const { return flight_cov_; }
+  void setFlightLength(Measurement1D flightLength) { flightLength_ = flightLength; }
+  const Measurement1D& flightLength() const { return flightLength_; }
+  // Point of closest approach
+  void setPCA(GlobalPoint pca, const GlobalError& pca_cov) {
+    pca_ = pca;
+    pca_cov_ = pca_cov_;
+  }
+  const GlobalPoint& pca() const { return pca_; }
+  const GlobalError& pcaCovariance() const { return pca_cov_; }
+  // Impact parameter
+  void setIP(GlobalVector ip_vec, const GlobalError& ip_cov) {
+    ip_vec_ = ip_vec;
+    ip_cov_ = ip_cov;
+  }
+  const GlobalVector& ipVector() const { return ip_vec_; }
+  const GlobalError& ipCovariance() const { return ip_cov_; }
+  void setIPLength(Measurement1D ipLength) { ipLength_ = ipLength; }
+  const Measurement1D& ipLength() const { return ipLength_; }
+  // Track
+  void setTrack(const reco::Track* track) { track_ = track; }
+  const reco::Track* track() const { return track_; }
+  bool hasTrack() const { return track_ != nullptr; }
+  void setBFiled_z(float bField_z) { bField_z_ = bField_z; }
+  float bField_z() const { return bField_z_; }
+
+private:
+  bool hasSV_;
+  reco::Vertex sv_;
+  GlobalVector flight_vec_, ip_vec_;
+  GlobalPoint pca_;
+  GlobalError flight_cov_, pca_cov_, ip_cov_;
+  Measurement1D flightLength_, ipLength_;
+  const reco::Track* track_;
+  float bField_z_;
+};
+
+#endif

DataFormats/NanoAOD/src/LeptonTimeLifeInfo.cc

@@ -0,0 +1,15 @@
+#include "DataFormats/NanoAOD/interface/LeptonTimeLifeInfo.h"
+
+LeptonTimeLifeInfo::LeptonTimeLifeInfo()
+    : hasSV_(false),
+      sv_(reco::Vertex()),
+      flight_vec_(GlobalVector()),
+      ip_vec_(GlobalVector()),
+      pca_(GlobalPoint()),
+      flight_cov_(GlobalError()),
+      pca_cov_(GlobalError()),
+      ip_cov_(GlobalError()),
+      flightLength_(Measurement1D()),
+      ipLength_(Measurement1D()),
+      track_(nullptr),
+      bField_z_(0.){};

DataFormats/NanoAOD/src/classes.h

@@ -3,4 +3,5 @@
 #include "DataFormats/NanoAOD/interface/FlatTable.h"
 #include "DataFormats/NanoAOD/interface/MergeableCounterTable.h"
 #include "DataFormats/NanoAOD/interface/UniqueString.h"
+#include "DataFormats/NanoAOD/interface/LeptonTimeLifeInfo.h"
 #include "DataFormats/Common/interface/Wrapper.h"

DataFormats/NanoAOD/src/classes_def.xml

@@ -42,5 +42,6 @@
         <version ClassVersion="3" checksum="2793726511"/>
     </class>
     <class name="edm::Wrapper<nanoaod::UniqueString>" />
+    <class name="LeptonTimeLifeInfo" />
 </lcgdict>
 

PhysicsTools/NanoAOD/plugins/BuildFile.xml

@@ -25,6 +25,7 @@
 <use name="PhysicsTools/PatAlgos"/>
 <use name="RecoVertex/VertexPrimitives"/>
 <use name="RecoVertex/VertexTools"/>
+<use name="RecoVertex/KalmanVertexFit"/>
 <library file="*.cc" name="PhysicsToolsNanoAODPlugins">
   <flags EDM_PLUGIN="1"/>
 </library>