TG4PrimaryGeneratorAction.h

Go to the documentation of this file.

#ifndef TG4_PRIMARY_GENERATOR_ACTION_H
#define TG4_PRIMARY_GENERATOR_ACTION_H
 
//------------------------------------------------
// The Geant4 Virtual Monte Carlo package
// Copyright (C) 2007 - 2014 Ivana Hrivnacova
// All rights reserved.
//
// For the licensing terms see geant4_vmc/LICENSE.
// Contact: root-vmc@cern.ch
//-------------------------------------------------
 
 
#include "TG4Verbose.h"
 
#include <G4VUserPrimaryGeneratorAction.hh>
#include <globals.hh>
 
class TVirtualMCStack;
class TMCManagerStack;
class TParticle;
 
class TG4PrimaryGeneratorMessenger;
class TG4ParticlesManager;
class TG4TrackManager;
 
class G4Event;
class G4ParticleDefinition;
class G4PrimaryVertex;
 
 
class TG4PrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction,
                                  public TG4Verbose
{
 public:
  TG4PrimaryGeneratorAction();
  virtual ~TG4PrimaryGeneratorAction();
 
  // methods
  virtual void GeneratePrimaries(G4Event* event);
 
  // set methods
  void SetSkipUnknownParticles(G4bool value);
 
  // get methods
  G4bool GetSkipUnknownParticles() const;
 
 private:
  // methods
 
  G4bool CheckVMCStack(TVirtualMCStack* stack) const;
  G4bool CheckParticleDefinition(const G4ParticleDefinition* particleDefinition,
    const TParticle* particle) const;
  G4double GetProperCharge(const G4ParticleDefinition* particleDefinition,
    const TParticle* particle) const;
  G4PrimaryVertex* AddParticleToVertex(G4Event* event, G4PrimaryVertex* vertex,
    const G4ParticleDefinition* particleDefinition,
    const G4ThreeVector& position, G4double time, const G4ThreeVector& momentum,
    G4double energy, const G4ThreeVector& polarization, G4double charge,
    G4double weight) const;
  void TransformPrimaries(G4Event* event);
  void TransformTracks(G4Event* event);
 
  // data members
  TG4PrimaryGeneratorMessenger* fMessenger;
  TG4ParticlesManager* fParticlesManager;
  TG4TrackManager* fTrackManager;
  TVirtualMCStack* fMCStack;
  TMCManagerStack* fMCManagerStack;
  G4bool fCached;
  G4bool fSkipUnknownParticles;
};
 
// inline functions
 
inline void TG4PrimaryGeneratorAction::SetSkipUnknownParticles(G4bool value)
{
  fSkipUnknownParticles = value;
}
 
inline G4bool TG4PrimaryGeneratorAction::GetSkipUnknownParticles() const
{
  return fSkipUnknownParticles;
}
 
#endif // TG4_PRIMARY_GENERATOR_ACTION_H