TG4SpecialStackingAction.cxx

Go to the documentation of this file.

//------------------------------------------------
// 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 "TG4SpecialStackingAction.h"
#include "TG4Globals.h"
 
#include <G4StackManager.hh>
#include <G4StackedTrack.hh>
#include <G4Track.hh>
#include <G4TrackStack.hh>
 
#include <TPDGCode.h>
 
//_____________________________________________________________________________
TG4SpecialStackingAction::TG4SpecialStackingAction()
  : G4UserStackingAction(),
    TG4Verbose("stackingAction", 1),
    fMessenger(this),
    fStage(0),
    fSkipNeutrino(false),
    fWaitPrimary(true)
{
 
  G4cout << "### TG4SpecialStackingAction activated" << G4endl;
}
 
//_____________________________________________________________________________
TG4SpecialStackingAction::~TG4SpecialStackingAction()
{
}
 
//
// public methods
//
 
//_____________________________________________________________________________
G4ClassificationOfNewTrack TG4SpecialStackingAction::ClassifyNewTrack(
  const G4Track* track)
{
 
  if (fWaitPrimary && fStage == 0) {
    // move all primaries to PrimaryStack
    return fPostpone;
  }
 
  if (fSkipNeutrino) {
    G4int pdgCode = track->GetDefinition()->GetPDGEncoding();
    if (pdgCode == kNuE || pdgCode == kNuEBar || pdgCode == kNuMu ||
        pdgCode == kNuMuBar || pdgCode == kNuTau || pdgCode == kNuTauBar) {
 
      return fKill;
    }
  }
 
  return fUrgent;
}
 
//_____________________________________________________________________________
void TG4SpecialStackingAction::NewStage()
{
 
  fStage++;
 
  if (VerboseLevel() > 1) {
    G4cout << "TG4SpecialStackingAction::NewStage " << fStage
           << " has been started." << G4endl;
  }
 
  if (fWaitPrimary && stackManager->GetNUrgentTrack() == 0 &&
      stackManager->GetNPostponedTrack() != 0) {
 
    stackManager->TransferOneStackedTrack(fPostpone, fUrgent);
  }
}
 
//_____________________________________________________________________________
void TG4SpecialStackingAction::PrepareNewEvent()
{
 
  fStage = 0;
}