ParticleManager.h

Go to the documentation of this file.

#ifndef FASTSIM_PARTICLEMANAGER_H
#define FASTSIM_PARTICLEMANAGER_H
 
#include "DataFormats/Math/interface/LorentzVector.h"
#include "HepMC/GenEvent.h"
#include <vector>
#include <memory>
 
#include "SimDataFormats/Track/interface/SimTrack.h"
#include "SimDataFormats/Vertex/interface/SimVertex.h"
 
// Author: L. Vanelderen, S. Kurz
// Date: 29 May 2017
 
namespace HepPDT {
  class ParticleDataTable;
}
 
class RandomEngineAndDistribution;
 
namespace fastsim {
  class Particle;
  class ParticleFilter;
  class SimplifiedGeometry;
 
 
  class ParticleManager {
  public:
 
    ParticleManager(const HepMC::GenEvent& genEvent,
                    const HepPDT::ParticleDataTable& particleDataTable,
                    double beamPipeRadius,
                    double deltaRchargedMother,
                    const ParticleFilter& particleFilter,
                    std::vector<SimTrack>& simTracks,
                    std::vector<SimVertex>& simVertices);
 
    ~ParticleManager();
 
 
    std::unique_ptr<Particle> nextParticle(const RandomEngineAndDistribution& random);
 
 
    void addSecondaries(const math::XYZTLorentzVector& vertexPosition,
                        int motherSimTrackId,
                        std::vector<std::unique_ptr<Particle> >& secondaries,
                        const SimplifiedGeometry* layer = nullptr);
 
    const SimVertex getSimVertex(unsigned i) { return simVertices_->at(i); }
 
    const SimTrack getSimTrack(unsigned i) { return simTracks_->at(i); }
 
 
    unsigned addEndVertex(const Particle* particle);
 
  private:
 
    unsigned addSimVertex(const math::XYZTLorentzVector& position, int motherIndex);
 
 
    unsigned addSimTrack(const Particle* particle);
 
 
    std::unique_ptr<Particle> nextGenParticle();
 
    // data members
    const HepMC::GenEvent* const genEvent_;  
    HepMC::GenEvent::particle_const_iterator
        genParticleIterator_;  
    const HepMC::GenEvent::particle_const_iterator genParticleEnd_;  
    int genParticleIndex_;  
    const HepPDT::ParticleDataTable* const
        particleDataTable_;         
    const double beamPipeRadius2_;  
    const double
        deltaRchargedMother_;  
    const ParticleFilter* const particleFilter_;  
    std::vector<SimTrack>* simTracks_;            
    std::vector<SimVertex>* simVertices_;         
    double momentumUnitConversionFactor_;         
    double lengthUnitConversionFactor_;           
    double lengthUnitConversionFactor2_;          
    double timeUnitConversionFactor_;             
    std::vector<std::unique_ptr<Particle> >
        particleBuffer_;  
  };
}  // namespace fastsim
 
#endif