TG4TrackingAction.h

Go to the documentation of this file.

#ifndef TG4_TRACKING_ACTION_H
#define TG4_TRACKING_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 "TG4TrackSaveControl.h"
#include "TG4TrackingActionMessenger.h"
#include "TG4Verbose.h"
 
#include <G4UserTrackingAction.hh>
 
class TG4TrackInformation;
class TG4TrackManager;
class TG4StepManager;
class TG4StackPopper;
class TG4SpecialControlsV2;
 
class TVirtualMCApplication;
class TVirtualMCStack;
 
class G4Track;
 
 
class TG4TrackingAction : public G4UserTrackingAction, public TG4Verbose
{
 public:
  TG4TrackingAction();
  virtual ~TG4TrackingAction();
 
  // static access method
  static TG4TrackingAction* Instance();
 
  // methods
  void LateInitialize();
  virtual void PrepareNewEvent();
  virtual void PreTrackingAction(const G4Track* aTrack);
  virtual void PostTrackingAction(const G4Track* aTrack);
  // the following methods should not
  // be overwritten in a derived class
  virtual void PreUserTrackingAction(const G4Track* aTrack);
  virtual void PostUserTrackingAction(const G4Track* aTrack);
  void FinishPrimaryTrack();
 
  // set methods
  void SetMCStack(TVirtualMCStack* mcStack);
  void SetNewVerboseLevel(G4int level);
  void SetNewVerboseTrackID(G4int trackID);
  void SetSpecialControls(TG4SpecialControlsV2* specialControls);
 
  // get methods
  G4int GetNewVerboseLevel() const;
  G4int GetNewVerboseTrackID() const;
  TG4TrackManager* GetTrackManager() const;
 
 private:
  TG4TrackingAction(const TG4TrackingAction& right);
  TG4TrackingAction& operator=(const TG4TrackingAction& right);
 
  // methods
  void UserProcessHits(const G4Track* aTrack);
  void Verbose() const;
 
  // static data members
  static G4ThreadLocal TG4TrackingAction* fgInstance; 
 
  //
  // data members
 
  TG4TrackingActionMessenger fMessenger;
 
  TG4SpecialControlsV2* fSpecialControls;
 
  TG4TrackManager* fTrackManager;
 
  TVirtualMCApplication* fMCApplication;
 
  TVirtualMCStack* fMCStack;
 
  TG4StepManager* fStepManager;
 
  TG4StackPopper* fStackPopper;
 
  G4int fPrimaryTrackID;
 
  G4int fCurrentTrackID;
 
  TG4TrackSaveControl fTrackSaveControl;
 
  G4bool fOverwriteLastTrack;
 
  G4int fNewVerboseLevel;
 
  G4int fNewVerboseTrackID;
 
  G4bool fDoFinishPrimary;
};
 
// inline methods
 
inline TG4TrackingAction* TG4TrackingAction::Instance()
{
  return fgInstance;
}
 
inline void TG4TrackingAction::PreTrackingAction(const G4Track* /*aTrack*/)
{
}
 
inline void TG4TrackingAction::PostTrackingAction(const G4Track* /*aTrack*/)
{
}
 
inline void TG4TrackingAction::SetMCStack(TVirtualMCStack* mcStack)
{
  fMCStack = mcStack;
}
 
inline void TG4TrackingAction::SetSpecialControls(
  TG4SpecialControlsV2* specialControls)
{
  fSpecialControls = specialControls;
}
 
#endif // TG4_TRACKING_ACTION_H