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 {
      if( genParticles_.empty() ) return g4Tracks_[0].type();
      else return (*genParticles_.begin())->pdgId();
    }

    EncodedEventId eventId() const {
      return g4Tracks_[0].eventId();
    }

    // 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 { return genParticles_; }
    const std::vector<SimTrack>& g4Tracks() const { return g4Tracks_; }
    const TrackingVertexRef& parentVertex() const { return parentVertex_; }

    // Accessors for vector of decay vertices
    const TrackingVertexRefVector& decayVertices() const { return decayVertices_; }
    tv_iterator decayVertices_begin() const { return decayVertices_.begin(); }
    tv_iterator decayVertices_end() const { return decayVertices_.end(); }


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

    const LorentzVector& p4() const {
      return g4Tracks_[0].momentum();
    }

    Vector momentum() const {
      return p4().Vect();
    }

    Vector boostToCM() const {
      return p4().BoostToCM();
    }

    double p() const {
      return p4().P();
    }

    double energy() const {
      return p4().E();
    }

    double et() const {
      return p4().Et();
    }

    double mass() const {
      return p4().M();
    }

    double massSqr() const {
      return pow( mass(), 2 );
    }

    double mt() const {
      return p4().Mt();
    }

    double mtSqr() const {
      return p4().Mt2();
    }

    double px() const {
      return p4().Px();
    }

    double py() const {
      return p4().Py();
    }

    double pz() const {
      return p4().Pz();
    }

    double pt() const {
      return p4().Pt();
    }

    double phi() const {
      return p4().Phi();
    }

    double theta() const {
      return p4().Theta();
    }

    double eta() const {
      return p4().Eta();
    }

    double rapidity() const {
      return p4().Rapidity();
    }

    double y() const {
      return rapidity();
    }

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

    double vx() const {
      const TrackingVertex& r=( *parentVertex_);
      return r.position().X();
    }

    double vy() const {
      const TrackingVertex& r=( *parentVertex_);
      return r.position().Y();
    }
    // @brief z coordinate of parent vertex position
    double vz() const {
      const TrackingVertex& r=( *parentVertex_);
      return r.position().Z();
    }

    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