TG4TrackManager.h

Go to the documentation of this file.

#ifndef TG4_TRACK_MANAGER_H
#define TG4_TRACK_MANAGER_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 "TG4Verbose.h"
 
#ifdef USE_G4ROOT
#include <TG4RootNavMgr.h>
#endif
 
#include <G4TrackVector.hh>
#include <G4UserTrackingAction.hh>
 
class TG4TrackInformation;
class TG4StackPopper;
 
class TVirtualMCStack;
class TMCManagerStack;
 
class G4Track;
class G4PrimaryVertex;
class G4PrimaryParticle;
struct TMCParticleStatus;
#ifdef USE_G4ROOT
class TMCManager;
#endif
 
 
class TG4TrackManager : public TG4Verbose
{
 public:
  TG4TrackManager();
  virtual ~TG4TrackManager();
 
  // static access method
  static TG4TrackManager* Instance();
 
  // methods
  void LateInitialize();
  void AddPrimaryParticleId(G4int id);
  void AddParticleStatus(const TMCParticleStatus* particleStatus);
  TG4TrackInformation* SetTrackInformation(
    const G4Track* aTrack, G4bool overWrite = false);
  void SetParentToTrackInformation(const G4Track* aTrack);
  void SetBackPDGLifetime(const G4Track* aTrack);
 
  void TrackToStack(const G4Track* track, G4bool overWrite = false);
  void PrimaryToStack(
    const G4PrimaryVertex* vertex, const G4PrimaryParticle* particle);
 
  void SaveSecondaries(const G4Track* track, const G4TrackVector* secondaries);
 
  // set methods
  void SetMCStack(TVirtualMCStack* mcStack);
  void SetMCManagerStack(TMCManagerStack* mcManagerStack);
  void SetTrackSaveControl(TG4TrackSaveControl control);
  void SetSaveDynamicCharge(G4bool saveDynamicCharge);
  void SetNofTracks(G4int nofTracks);
  void SetG4TrackingManager(G4TrackingManager* trackingManager);
  void ResetPrimaryParticleIds();
  void ResetParticlesStatus();
 
  // get methods
  TG4TrackInformation* GetTrackInformation(const G4Track* track) const;
  TG4TrackSaveControl GetTrackSaveControl() const;
  G4bool GetSaveDynamicCharge() const;
  G4int GetNofTracks() const;
  G4bool IsUserTrack(const G4Track* track) const;
 
 private:
  TG4TrackManager(const TG4TrackManager& right);
  TG4TrackManager& operator=(const TG4TrackManager& right);
 
  // static data members
  static G4ThreadLocal TG4TrackManager* fgInstance; 
 
  // data members
  G4TrackingManager* fG4TrackingManager;  
  std::vector<G4int> fPrimaryParticleIds; 
  std::vector<TMCParticleStatus*> fParticlesStatus; 
  TG4TrackSaveControl fTrackSaveControl; 
 
  TVirtualMCStack* fMCStack;
 
  TMCManagerStack* fMCManagerStack;
 
  TG4StackPopper* fStackPopper;
 
  G4bool fSaveDynamicCharge;  
  G4int fTrackCounter;        
  G4int fCurrentTrackID;      
  G4int fNofSavedSecondaries; 
#ifdef USE_G4ROOT
  TG4RootNavMgr* fRootNavMgr; 
  TMCManager* fMCManager;     
#endif
};
 
// inline methods
 
inline TG4TrackManager* TG4TrackManager::Instance()
{
  return fgInstance;
}
 
inline void TG4TrackManager::SetMCStack(TVirtualMCStack* mcStack)
{
  fMCStack = mcStack;
}
 
inline void TG4TrackManager::SetMCManagerStack(TMCManagerStack* mcManagerStack)
{
  fMCManagerStack = mcManagerStack;
}
 
inline void TG4TrackManager::SetTrackSaveControl(TG4TrackSaveControl control)
{
  fTrackSaveControl = control;
}
 
inline void TG4TrackManager::SetSaveDynamicCharge(G4bool saveDynamicCharge)
{
  fSaveDynamicCharge = saveDynamicCharge;
}
 
inline void TG4TrackManager::SetNofTracks(G4int nofTracks)
{
  fTrackCounter = nofTracks;
}
 
inline void TG4TrackManager::SetG4TrackingManager(
  G4TrackingManager* trackingManager)
{
  fG4TrackingManager = trackingManager;
#ifdef USE_G4ROOT
  if (fRootNavMgr) {
    fRootNavMgr->SetG4TrackingManager(fG4TrackingManager);
  }
#endif
}
 
inline TG4TrackSaveControl TG4TrackManager::GetTrackSaveControl() const
{
  return fTrackSaveControl;
}
 
inline G4bool TG4TrackManager::GetSaveDynamicCharge() const
{
  return fSaveDynamicCharge;
}
 
inline G4int TG4TrackManager::GetNofTracks() const
{
  return fTrackCounter;
}
 
#endif // TG4_TRACK_MANAGER_H