TrackingParticle.h

Go to the documentation of this file.

#ifndef SimDataFormats_TrackingParticle_h
#define SimDataFormats_TrackingParticle_h

#include <vector>
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertexContainer.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"

//
// Forward declarations
//
class TrackingVertex;
class SimTrack;
class EncodedEventId;

class TrackingParticle
{
    friend std::ostream& operator<< (std::ostream& s, TrackingParticle const& tp);
public:
    typedef int Charge; 
    typedef math::XYZTLorentzVectorD LorentzVector; 
    typedef math::PtEtaPhiMLorentzVector PolarLorentzVector; 
    typedef math::XYZPointD Point; 
    typedef math::XYZVectorD Vector; 

    typedef reco::GenParticleRefVector::iterator   genp_iterator;
    typedef std::vector<SimTrack>::const_iterator  g4t_iterator;

    TrackingParticle();

    TrackingParticle( const SimTrack& simtrk, const TrackingVertexRef& parentVertex );

    // destructor
    ~TrackingParticle();

    int pdgId() const;
    EncodedEventId eventId() const;

    // Setters for G4 and reco::GenParticle
    void addGenParticle( const reco::GenParticleRef& ref);
    void addG4Track( const SimTrack& t);
    genp_iterator genParticle_begin() const;
    genp_iterator genParticle_end() const;
    g4t_iterator g4Track_begin() const;
    g4t_iterator g4Track_end() const;
    void setParentVertex(const TrackingVertexRef& ref);
    void addDecayVertex(const TrackingVertexRef& ref);
    void clearParentVertex();
    void clearDecayVertices();
    // Getters for Embd and Sim Tracks
    const reco::GenParticleRefVector& genParticles() const;
    const std::vector<SimTrack>& g4Tracks() const;
    const TrackingVertexRef& parentVertex() const;

    // Accessors for vector of decay vertices
    const TrackingVertexRefVector& decayVertices() const;
    tv_iterator decayVertices_begin() const;
    tv_iterator decayVertices_end() const;


    float charge() const { return g4Tracks_[0].charge(); }
    int threeCharge() const { return lrintf(3.f*charge()); }

    const LorentzVector& p4() const; 

    Vector momentum() const; 

    Vector boostToCM() const; 

    double p() const; 
    double energy() const; 
    double et() const; 
    double mass() const; 
    double massSqr() const; 
    double mt() const; 
    double mtSqr() const; 
    double px() const; 
    double py() const; 
    double pz() const; 
    double pt() const; 
    double phi() const; 
    double theta() const; 
    double eta() const; 
    double rapidity() const; 
    double y() const; 

    Point vertex() const {
       const TrackingVertex::LorentzVector & p = (*parentVertex_).position();
       return Point(p.x(),p.y(),p.z());
    }  

    double vx() const; 
    double vy() const; 
    double vz() const; 

    int status() const {
      return genParticles_.empty() ? -99 : (*genParticles_[0]).status();
    }

    static const unsigned int longLivedTag; 

    bool longLived() const { return status()&longLivedTag;}

   int numberOfHits() const {return numberOfHits_;}

   int numberOfTrackerHits() const {return numberOfTrackerHits_;}

   int matchedHit() const;
   int numberOfTrackerLayers() const {return numberOfTrackerLayers_;}

   void setNumberOfHits( int numberOfHits );
   void setNumberOfTrackerHits( int numberOfTrackerHits );
   void setNumberOfTrackerLayers( const int numberOfTrackerLayers );
private:
    int numberOfHits_; 
    int numberOfTrackerHits_; 
    int numberOfTrackerLayers_; 

    std::vector<SimTrack> g4Tracks_;
    reco::GenParticleRefVector genParticles_;

    // Source and decay vertices
    TrackingVertexRef parentVertex_;
    TrackingVertexRefVector decayVertices_;
};

#endif // SimDataFormats_TrackingParticle_H