TrackKinematics.cc

Go to the documentation of this file.

#include <cmath>
#include <vector>

#include <Math/GenVector/PxPyPzE4D.h>
#include <Math/GenVector/PxPyPzM4D.h>

#include "DataFormats/BTauReco/interface/ParticleMasses.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

#include "RecoBTag/SecondaryVertex/interface/TrackKinematics.h"

using namespace reco;

TrackKinematics::TrackKinematics() :
    n(0), sumWeights(0)
{
}

TrackKinematics::TrackKinematics(const std::vector<Track> &tracks) :
    n(0), sumWeights(0)
{
    for(std::vector<Track>::const_iterator iter = tracks.begin();
        iter != tracks.end(); iter++)
        add(*iter);
}

TrackKinematics::TrackKinematics(const TrackRefVector &tracks) :
    n(0), sumWeights(0)
{
    for(TrackRefVector::const_iterator iter = tracks.begin();
        iter != tracks.end(); iter++)
        add(**iter);
}

TrackKinematics::TrackKinematics(const std::vector<CandidatePtr> &tracks) :
    n(0), sumWeights(0)
{
    for(std::vector<CandidatePtr>::const_iterator iter = tracks.begin();
        iter != tracks.end(); iter++)
        add(*iter);
}

TrackKinematics::TrackKinematics(const CandidatePtrVector &tracks) :
    n(0), sumWeights(0)
{
    for(CandidatePtrVector::const_iterator iter = tracks.begin();
        iter != tracks.end(); iter++)
        add(*iter);
}

TrackKinematics::TrackKinematics(const Vertex &vertex) :
    n(0), sumWeights(0)
{
    bool hasRefittedTracks = vertex.hasRefittedTracks();
    for(Vertex::trackRef_iterator iter = vertex.tracks_begin();
        iter != vertex.tracks_end(); ++iter) {
        if (hasRefittedTracks)
            add(vertex.refittedTrack(*iter),
                vertex.trackWeight(*iter));
        else
            add(**iter, vertex.trackWeight(*iter));
    }
}

TrackKinematics &TrackKinematics::operator += (const TrackKinematics &other)
{
    n += other.n;
    sumWeights += other.sumWeights;
    sum += other.sum;
    weightedSum += other.weightedSum;

    return *this;
}

void TrackKinematics::add(const Track &track, double weight)
{
    ROOT::Math::LorentzVector<ROOT::Math::PxPyPzM4D<double> > vec;

    vec.SetPx(track.px());
    vec.SetPy(track.py());
    vec.SetPz(track.pz());
    vec.SetM(ParticleMasses::piPlus);

    n++;
    sumWeights += weight;
    sum += vec;
    weightedSum += weight * vec;
}

void TrackKinematics::add(const CandidatePtr &track)
{
    double weight = 1.0;

    n++;
    sumWeights += weight;
    sum += track->p4();
    weightedSum += weight * track->p4();
}