TrackingParticle.h

Go to the documentation of this file.

#ifndef SimDataFormats_TrackingParticle_h
#define SimDataFormats_TrackingParticle_h

#include <map>

#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "SimDataFormats/TrackingAnalysis/interface/ParticleBase.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 ParticleBase
{

    friend std::ostream& operator<< (std::ostream& s, TrackingParticle const & 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;

    typedef std::multimap<DetId::Detector, PSimHit> DetectorToPSimHit;

    TrackingParticle() {}
    
    TrackingParticle( char q, const LorentzVector & p4, const Point & vtx,
                      double t, const int pdgId,  const int status, const EncodedEventId eventId);

    // destructor
    ~TrackingParticle();

    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;

    // 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);

    // 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_;
    }

    // PSimHits discriminated by subdetector
    std::vector<PSimHit> trackPSimHit(DetId::Detector) const;

    // 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:

    float t_;
    int pdgId_;
    EncodedEventId eventId_;

    int matchedHit_;

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

    // TrackPSimHitRefVector trackPSimHit_;
    std::vector<PSimHit> trackPSimHit_;

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

#endif // SimDataFormats_TrackingParticle_H