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