FBaseSimEvent.h

Go to the documentation of this file.

#ifndef FastSimulation_Event_FBaseSimEvent_H
#define FastSimulation_Event_FBaseSimEvent_H
 
// Data Formats
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
#include "DataFormats/Math/interface/Point3D.h"
 
// HepPDT Headers
#include "SimGeneral/HepPDTRecord/interface/ParticleDataTable.h"
 
// Famos Headers
#include "CommonTools/BaseParticlePropagator/interface/RawParticle.h"
#include "FastSimDataFormats/NuclearInteractions/interface/FSimVertexType.h"
#include "FastSimDataFormats/NuclearInteractions/interface/FSimVertexTypeFwd.h"
 
#include <vector>
 
//class FSimEvent;
class FSimTrack;
class FSimVertex;
class KineParticleFilter;
 
class SimTrack;
class SimVertex;
 
namespace edm {
  class ParameterSet;
}
 
namespace HepMC {
  class GenEvent;
  class GenParticle;
  class GenVertex;
}  // namespace HepMC
 
class FBaseSimEvent {
public:
  FBaseSimEvent(const edm::ParameterSet& kine);
 
  ~FBaseSimEvent();
 
  void initializePdt(const HepPDT::ParticleDataTable* aPdt);
 
  inline const HepPDT::ParticleDataTable* theTable() const { return pdt; }
 
  void fill(const HepMC::GenEvent& hev);
 
  void fill(const std::vector<SimTrack>&, const std::vector<SimVertex>&);
 
  void printMCTruth(const HepMC::GenEvent& hev);
 
  void addParticles(const HepMC::GenEvent& hev);
 
  void print() const;
 
  void clear();
 
  void addChargedTrack(int id);
 
  inline unsigned int nTracks() const { return nSimTracks; }
 
  inline unsigned int nVertices() const { return nSimVertices; }
 
  inline unsigned int nGenParts() const { return nGenParticles; }
 
  inline unsigned int nChargedTracks() const { return nChargedParticleTracks; }
 
  inline FSimTrack& track(int id) const;
 
  inline FSimVertex& vertex(int id) const;
 
  inline FSimVertexType& vertexType(int id) const;
 
  int chargedTrack(int id) const;
 
  inline const SimTrack& embdTrack(int i) const;
 
  inline const SimVertex& embdVertex(int i) const;
 
  inline const FSimVertexType& embdVertexType(int i) const;
 
  const HepMC::GenParticle* embdGenpart(int i) const;
 
  int addSimTrack(const RawParticle* p, int iv, int ig = -1, const HepMC::GenVertex* ev = nullptr);
 
  int addSimVertex(const XYZTLorentzVector& decayVertex,
                   int im = -1,
                   FSimVertexType::VertexType type = FSimVertexType::ANY);
 
  const KineParticleFilter& filter() const { return *myFilter; }
 
protected:
  inline std::vector<FSimTrack>* tracks() const { return theSimTracks; }
 
  inline std::vector<FSimVertex>* vertices() const { return theSimVertices; }
 
  inline std::vector<HepMC::GenParticle*>* genparts() const { return theGenParticles; }
 
private:
  std::vector<FSimTrack>* theSimTracks;
  std::vector<FSimVertex>* theSimVertices;
  FSimVertexTypeCollection* theFSimVerticesType;
  std::vector<HepMC::GenParticle*>* theGenParticles;
 
  std::vector<unsigned>* theChargedTracks;
 
  unsigned int nSimTracks;
  unsigned int nSimVertices;
  unsigned int nGenParticles;
  unsigned int nChargedParticleTracks;
 
  unsigned int theTrackSize;
  unsigned int theVertexSize;
  unsigned int theGenSize;
  unsigned int theChargedSize;
  unsigned int initialSize;
 
  KineParticleFilter* myFilter;
 
  double sigmaVerteX;
  double sigmaVerteY;
  double sigmaVerteZ;
 
  const ParticleDataTable* pdt;
 
  double lateVertexPosition;
 
  //  Histos* myHistos;
};
 
#include "FastSimulation/Event/interface/FBaseSimEvent.icc"
 
#endif  // FBaseSimEvent_H