TrackKinematics.h

Go to the documentation of this file.

#ifndef RecoBTag_SecondaryVertex_TrackKinematics_h
#define RecoBTag_SecondaryVertex_TrackKinematics_h

#include <vector>

#include "DataFormats/Math/interface/LorentzVector.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/Candidate/interface/VertexCompositePtrCandidate.h"

namespace reco {

  class TrackKinematics {
  public:
    TrackKinematics();
    TrackKinematics(const std::vector<reco::Track> &tracks);
    TrackKinematics(const reco::TrackRefVector &tracks);
    TrackKinematics(const std::vector<reco::CandidatePtr> &tracks);
    TrackKinematics(const reco::CandidatePtrVector &tracks);
    TrackKinematics(const reco::Vertex &vertex);
    TrackKinematics(const reco::VertexCompositePtrCandidate &vertex)
        : n(vertex.numberOfSourceCandidatePtrs()),
          sumWeights(vertex.numberOfSourceCandidatePtrs()),
          sum(vertex.p4()),
          weightedSum(vertex.p4()) {}

    ~TrackKinematics() {}

    void add(const reco::Track &track, double weight = 1.0);
    void add(const reco::CandidatePtr &track);

    inline void add(const reco::TrackRef &track, double weight = 1.0) { return add(*track, weight); }

    TrackKinematics &operator+=(const TrackKinematics &other);
    inline TrackKinematics operator+(const TrackKinematics &other) {
      TrackKinematics copy = *this;
      copy += other;
      return copy;
    }

    inline unsigned int numberOfTracks() const { return n; }
    inline double sumOfWeights() const { return sumWeights; }

    inline const math::XYZTLorentzVector &vectorSum() const { return sum; }
    inline const math::XYZTLorentzVector &weightedVectorSum() const { return weightedSum; }

  private:
    unsigned int n;
    double sumWeights;
    math::XYZTLorentzVector sum;
    math::XYZTLorentzVector weightedSum;
  };

}  // namespace reco

#endif  // RecoBTag_SecondaryVertex_TrackKinematics_h