GGSAcceptanceCheckAction.cpp

Go to the documentation of this file.

/*
 * GGSAcceptanceCheckAction.cpp
 *
 *  Created on: 08 Oct 2013
 *      Author: Nicola Mori
 */
#include "montecarlo/useractions/GGSAcceptanceCheckAction.h"
#include "montecarlo/dataobjs/GGSPrimaryParticleInfo.h"
#include "montecarlo/mccore/GGSDetectorConstruction.h"
#include "montecarlo/mccore/GGSRunManager.h"
#include "montecarlo/pluginmanagers/GGSMCPluginMacros.h"

#include "G4GenericMessenger.hh"
#include "G4Track.hh"

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

RegisterUA(GGSAcceptanceCheckAction);

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

GGSAcceptanceCheckAction::GGSAcceptanceCheckAction() : _reSimulateKilled(false) {
  _messenger = new G4GenericMessenger(this, "/GGS/userActions/acceptanceCheckAction/");
  _messenger->DeclareProperty("reSimulateKilled", _reSimulateKilled, "Simulate again events outside acceptance")
      .SetDefaultValue("true");
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

GGSAcceptanceCheckAction::~GGSAcceptanceCheckAction() { delete _messenger; }

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

G4ClassificationOfNewTrack GGSAcceptanceCheckAction::ClassifyNewTrack(const G4Track *track) {
  static const std::string routineName("GGSAcceptanceCheckAction::ClassifyNewTrack");
  if (track->GetParentID() == 0) {
    GGSDetectorConstruction *detConst =
        (GGSDetectorConstruction *)(GGSRunManager::GetRunManager()->GetUserDetectorConstruction());
    if (detConst->GetGeometry()->IsInsideAcceptance(track->GetPosition(), track->GetMomentum().unit())) {
      return fUrgent;
    } else {
      GGSPrimaryParticleInfo *userInfo =
          (GGSPrimaryParticleInfo *)(track->GetDynamicParticle()->GetPrimaryParticle()->GetUserInformation());
      if (!userInfo)
        userInfo = new GGSPrimaryParticleInfo;
      userInfo->isTrackKilled = true;
      GGSCOUT(DEBUG) << "Killed primary track " << track->GetTrackID() << GGSENDL;
      track->GetDynamicParticle()->GetPrimaryParticle()->SetUserInformation(userInfo);
      if (G4EventManager::GetEventManager()->GetStackManager()->GetNTotalTrack() == 0) {
        GGSRunManagerExtensions &rmExt = dynamic_cast<GGSRunManagerExtensions &>(*(G4RunManager::GetRunManager()));
        rmExt.KillEvent();
        if (_reSimulateKilled) {
          rmExt.SimulateAgainKilledEvent();
          GGSCOUT(DEBUG) << "Will be simulated again" << GGSENDL;
        }
      }
      return fKill;
    }
  }
  return G4UserStackingAction::ClassifyNewTrack(track);
}