ParticleInformation.h

Go to the documentation of this file.

#ifndef SIMG4COMMON_PARTICLEINFORMATION_H
#define SIMG4COMMON_PARTICLEINFORMATION_H
 
// datamodel
#include "edm4hep/MCParticle.h"
 
// Geant4
#include "G4VUserPrimaryParticleInformation.hh"
 
// CLHEP
#include "CLHEP/Vector/ThreeVector.h"
 
namespace sim {
class ParticleInformation : public G4VUserPrimaryParticleInformation {
public:
  explicit ParticleInformation(const edm4hep::MCParticle& aMCpart);
  virtual ~ParticleInformation();
  virtual void Print() const final;
  edm4hep::MCParticle& mcParticle();
  void setEndMomentum(const CLHEP::Hep3Vector& aMom);
  const CLHEP::Hep3Vector& endMomentum() const;
  void setVertexPosition(const CLHEP::Hep3Vector& aPos);
  const CLHEP::Hep3Vector& vertexPosition() const;
  void setEndStatus(int aStat);
  int endStatus() const;
  void setSmeared(bool aFlag);
  bool smeared() const;
 
private:
  edm4hep::MCParticle m_mcParticle;
  CLHEP::Hep3Vector m_endMomentum;
  CLHEP::Hep3Vector m_vertexPosition;
  int m_endStatus;
  bool m_smeared;
};
}
 
#endif /* SIMG4COMMON_PARTICLEINFORMATION_H */