---

TrackingParticle.h

Go to the documentation of this file.

#ifndef SimDataFormats_TrackingParticle_h
#define SimDataFormats_TrackingParticle_h

#include "DataFormats/Candidate/interface/Particle.h"

#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
#include "SimDataFormats/HepMCProduct/interface/HepMCProduct.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertex.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "DataFormats/DetId/interface/DetId.h"

namespace HepMC {
  class GenParticle;
}
class TrackingVertex;

class TrackingParticle : public reco::Particle {
  
  friend std::ostream& operator<< (std::ostream& s, const TrackingParticle & tp);
  
public:
  typedef edm::RefVector<edm::HepMCProduct, HepMC::GenParticle > GenParticleRefVector;
  typedef edm::Ref<edm::HepMCProduct, HepMC::GenParticle >       GenParticleRef;
  typedef GenParticleRefVector::iterator                         genp_iterator;
  typedef std::vector<SimTrack>::const_iterator                  g4t_iterator;
  
  typedef std::vector<TrackingVertex>                TrackingVertexCollection;
  typedef edm::Ref<TrackingVertexCollection>         TrackingVertexRef;
  typedef edm::RefVector<TrackingVertexCollection>   TrackingVertexRefVector;
  typedef TrackingVertexRefVector::iterator   tv_iterator;
  
  TrackingParticle() { subdetVectorFill_ = false; }
  // destructor
  ~TrackingParticle();
  TrackingParticle( char q, const LorentzVector & p4, const Point & vtx,
                    double t, const int pdgId,  const int status, const EncodedEventId eventId);
  
  int pdgId() const { return pdgId_; }
  EncodedEventId eventId() const { return eventId_; }
  
  genp_iterator genParticle_begin() const;
  genp_iterator genParticle_end()   const;
  g4t_iterator  g4Track_begin()     const;
  g4t_iterator  g4Track_end()       const;
  
  const std::vector<PSimHit>::const_iterator  pSimHit_begin() const;
  const std::vector<PSimHit>::const_iterator  pSimHit_end()   const;
  
  const std::vector<PSimHit>::const_iterator  trackerPSimHit_begin();
  const std::vector<PSimHit>::const_iterator  trackerPSimHit_end();
  
  const std::vector<PSimHit>::const_iterator  muonPSimHit_begin();
  const std::vector<PSimHit>::const_iterator  muonPSimHit_end();
  
  // Setters for G4 and HepMC
  void addG4Track(    const SimTrack&);
  void addGenParticle(const GenParticleRef&);
  
  void addPSimHit(const PSimHit&);
  void setParentVertex(const TrackingVertexRef&);
  void addDecayVertex(const TrackingVertexRef&);
  void clearParentVertex();
  void clearDecayVertices();
  void setMatchedHit(const int&);
  void setVertex(const Point & vtx, double t);
  void fillSubDetHitVectors();

  // Getters for Embd and Sim Tracks
  const GenParticleRefVector&     genParticle() const { return genParticles_; }
  const std::vector<SimTrack>&       g4Tracks() const { return g4Tracks_ ;    }
  const TrackingVertexRef&       parentVertex() const { return parentVertex_; }
  const std::vector<PSimHit>&    trackPSimHit() const { return trackPSimHit_; }

  // Sub-detectors details
  const std::vector<PSimHit>& trackerPSimHit();
  const std::vector<PSimHit>&    muonPSimHit();
  
  // 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(); }
  int matchedHit() const {return matchedHit_;}
  
private:
  double t_;
  int pdgId_;
  EncodedEventId eventId_;
  
  int matchedHit_;
  
  std::vector<SimTrack> g4Tracks_;
  GenParticleRefVector  genParticles_;
  
  //  TrackPSimHitRefVector trackPSimHit_;
  std::vector<PSimHit> trackPSimHit_;
  // TrackPSimHitRefVector for Tracker and Muon (not persistent)
  std::vector<PSimHit> trackerPSimHit_;
  std::vector<PSimHit> muonPSimHit_;
  bool subdetVectorFill_;
  
  // Source and decay vertices
  TrackingVertexRef       parentVertex_;
  TrackingVertexRefVector decayVertices_;
};

#endif // SimDataFormats_TrackingParticle_H

---