TrackingAction.cc

Go to the documentation of this file.

#include "SimG4Core/Application/interface/TrackingAction.h"
#include "SimG4Core/Application/interface/EventAction.h"
#include "SimG4Core/Notification/interface/NewTrackAction.h"
#include "SimG4Core/Notification/interface/CurrentG4Track.h"
#include "SimG4Core/Notification/interface/BeginOfTrack.h"
#include "SimG4Core/Notification/interface/EndOfTrack.h"
#include "SimG4Core/Notification/interface/TrackInformation.h"
#include "SimG4Core/Notification/interface/TrackWithHistory.h"
#include "SimG4Core/Notification/interface/TrackInformationExtractor.h"
#include "SimG4Core/Notification/interface/SimG4Exception.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "G4UImanager.hh" 
#include "G4TrackingManager.hh"
#include "G4PhysicalVolumeStore.hh"
#include "G4TransportationManager.hh"

//#define DebugLog

TrackingAction::TrackingAction(EventAction * e, const edm::ParameterSet & p) 
  : eventAction_(e),currentTrack_(0),
  detailedTiming(p.getUntrackedParameter<bool>("DetailedTiming",false)),
  trackMgrVerbose(p.getUntrackedParameter<int>("G4TrackManagerVerbosity",0)) {

  worldSolid = G4TransportationManager::GetTransportationManager()->GetNavigatorForTracking()->GetWorldVolume()->GetLogicalVolume()->GetSolid();
}

TrackingAction::~TrackingAction() {}

void TrackingAction::PreUserTrackingAction(const G4Track * aTrack)
{

    CurrentG4Track::setTrack(aTrack);

    if (currentTrack_ != 0) 
    throw SimG4Exception("TrackingAction: currentTrack is a mess...");
    currentTrack_ = new TrackWithHistory(aTrack);

    /*
      Trick suggested by Vladimir I. in order to debug with high 
      level verbosity only a single problematic tracks
    */      

    /*
    if ( aTrack->GetTrackID() == palce_here_the_trackid_of_problematic_tracks  ) {
      G4UImanager::GetUIpointer()->ApplyCommand("/tracking/verbose 6");
    } else if ( aTrack->GetTrackID() == place_here_the_trackid_of_following_track_to_donwgrade_the_severity ) {
      G4UImanager::GetUIpointer()->ApplyCommand("/tracking/verbose 0");
    }
    */
    BeginOfTrack bt(aTrack);
    m_beginOfTrackSignal(&bt);

    if (isNewPrimary(aTrack)) {
      eventAction_->prepareForNewPrimary();
    }
    //    G4cout << "Track " << aTrack->GetTrackID() << " R " << (aTrack->GetVertexPosition()).r() << " Z " << std::abs((aTrack->GetVertexPosition()).z()) << G4endl << "Top Solid " << worldSolid->GetName() << " is it inside " << worldSolid->Inside(aTrack->GetVertexPosition()) << " compared to " << kOutside << G4endl;
    if (worldSolid->Inside(aTrack->GetVertexPosition()) == kOutside) {
      //      G4cout << "Kill Track " << aTrack->GetTrackID() << G4endl;
      G4Track* theTrack = (G4Track *)(aTrack);
      theTrack->SetTrackStatus(fStopAndKill);
    }      

}

void TrackingAction::PostUserTrackingAction(const G4Track * aTrack)
{
    CurrentG4Track::postTracking(aTrack);
    if (eventAction_->trackContainer() != 0)
    {

      TrackInformationExtractor extractor;
      if (extractor(aTrack).storeTrack())
    {
      currentTrack_->save();
      
      math::XYZVectorD pos((aTrack->GetStep()->GetPostStepPoint()->GetPosition()).x(),
                   (aTrack->GetStep()->GetPostStepPoint()->GetPosition()).y(),
                   (aTrack->GetStep()->GetPostStepPoint()->GetPosition()).z());
      math::XYZTLorentzVectorD mom;
      
      uint32_t id = aTrack->GetTrackID();
      
      std::pair<math::XYZVectorD,math::XYZTLorentzVectorD> p(pos,mom);
      eventAction_->addTkCaloStateInfo(id,p);
#ifdef DebugLog
      LogDebug("SimTrackManager") << "TrackingAction addTkCaloStateInfo " << id << " of momentum " << mom << " at " << pos;
#endif
    }

      bool withAncestor = ((extractor(aTrack).getIDonCaloSurface() == aTrack->GetTrackID()) || (extractor(aTrack).isAncestor()));
      if (extractor(aTrack).isInHistory())
        {
          currentTrack_->checkAtEnd(aTrack);  // check with end-of-track information
          eventAction_->addTrack(currentTrack_, true, withAncestor);
#ifdef DebugLog
      math::XYZVectorD pos((aTrack->GetStep()->GetPostStepPoint()->GetPosition()).x(),
                   (aTrack->GetStep()->GetPostStepPoint()->GetPosition()).y(),
                   (aTrack->GetStep()->GetPostStepPoint()->GetPosition()).z());
      LogDebug("SimTrackManager") << "TrackingAction addTrack "  << currentTrack_->trackID() << " added with " << true << " and " << withAncestor << " at " << pos;
#endif
        }
      else
        {
          eventAction_->addTrack(currentTrack_, false, false);
#ifdef DebugLog
      LogDebug("SimTrackManager") << "TrackingAction addTrack " << currentTrack_->trackID() << " added with " << false << " and " << false;
#endif
          delete currentTrack_;
        }
    }
    EndOfTrack et(aTrack);
    m_endOfTrackSignal(&et);
    currentTrack_ = 0; // reset for next track
}

G4TrackingManager * TrackingAction::getTrackManager()
{
    G4TrackingManager * theTrackingManager = 0;
    theTrackingManager = fpTrackingManager;
    theTrackingManager->SetVerboseLevel(trackMgrVerbose);
    return theTrackingManager;
}
 
bool TrackingAction::isNewPrimary(const G4Track * aTrack){

  TrackInformation * trkInfo = (TrackInformation *)aTrack->GetUserInformation();
  return trkInfo->isPrimary();

}