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