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