PrimaryVertexInformation.hh

#ifndef __RAT__PrimaryVertexInformation__
#define __RAT__PrimaryVertexInformation__
 
#include <G4PrimaryParticle.hh>
#include <G4VUserPrimaryVertexInformation.hh>
#include <vector>
 
namespace RAT {
 
class PrimaryVertexInformation : public G4VUserPrimaryVertexInformation {
 public:
  PrimaryVertexInformation(){};
  ~PrimaryVertexInformation(){};
 
  void Print() const {};  // required for G4VUserPrimaryVertexInformation
 
  G4PrimaryParticle *GetParentParticle(int i) { return parents[i]; }
  int GetParentParticleCount() const { return parents.size(); }
  void AddNewParentParticle(G4PrimaryParticle *part) { parents.push_back(part); }
 
 private:
  std::vector<G4PrimaryParticle *> parents;
};
}  // namespace RAT
 
#endif