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);
    void exoticRelativesChecker(const HepMC::GenVertex* originVertex, int& hasExoticAssociation, int ngendepth);


    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

inline bool isExotic(int pdgid_) {
  unsigned int pdgid = std::abs(pdgid_);
  return ((pdgid >= 1000000 && pdgid < 4000000 && pdgid != 3000022) ||  // SUSY, R-hadron, and technicolor particles
          pdgid == 17 ||                                                // 4th generation lepton
          pdgid == 34 ||                                                // W-prime
          pdgid == 37 ||                                                // charged Higgs
          pdgid == 39);                                                 // bulk graviton
}

#endif