AliGeant4/AliEventAction.cxx

   1 // $Id$
   2 // Category: event
   3 //
   4 // See the class description in the header file.
   5
   6 #include "AliEventAction.h"
   7 #include "AliEventActionMessenger.h"
   8 #include "AliRun.h"
   9 #include "AliTrackingAction.h"
  10 #include "AliSensitiveDetector.h"
  11 #include "AliGlobals.h"
  12
  13 #include "TG4GeometryManager.h"
  14
  15 #include <G4Event.hh>
  16 #include <G4TrajectoryContainer.hh>
  17 #include <G4Trajectory.hh>
  18 #include <G4VVisManager.hh>
  19 #include <G4UImanager.hh>
  20 #include <G4LogicalVolumeStore.hh>
  21 #include <G4VSensitiveDetector.hh>
  22
  23 AliEventAction::AliEventAction()
  24   : fVerboseLevel(1),
  25     fDrawFlag("CHARGED")
  26 {
  27 //
  28   fMessenger = new AliEventActionMessenger(this);
  29 }
  30
  31 AliEventAction::AliEventAction(const AliEventAction& right) {
  32 //
  33   AliGlobals::Exception("AliEventAction is protected from copying.");
  34 }
  35
  36 AliEventAction::~AliEventAction() {
  37 //
  38   delete fMessenger;
  39 }
  40
  41 // operators
  42
  43 AliEventAction& AliEventAction::operator=(const AliEventAction &right)
  44 {
  45   // check assignement to self
  46   if (this == &right) return *this;
  47
  48   AliGlobals::Exception("AliEventAction is protected from assigning.");
  49
  50   return *this;
  51 }
  52
  53 // private methods
  54
  55 void AliEventAction::DisplayEvent(const G4Event* event) const
  56 {
  57 // Draws trajectories.
  58 // ---
  59
  60
  61   // trajectories processing
  62   G4TrajectoryContainer* trajectoryContainer
  63     = event->GetTrajectoryContainer();
  64
  65   G4int nofTrajectories = 0;
  66   if (trajectoryContainer)
  67   { nofTrajectories = trajectoryContainer->entries(); }
  68
  69   if (fVerboseLevel>0) {
  70     G4cout << "    " << nofTrajectories;
  71     G4cout << " trajectories stored." << G4endl;
  72   }
  73
  74   G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
  75   if(pVVisManager && nofTrajectories>0)
  76   {
  77     G4UImanager::GetUIpointer()->ApplyCommand("/vis~/draw/current");
  78
  79     for (G4int i=0; i<nofTrajectories; i++)
  80     {
  81       G4VTrajectory* vtrajectory = (*(event->GetTrajectoryContainer()))[i];
  82       G4Trajectory* trajectory = dynamic_cast<G4Trajectory*>(vtrajectory);
  83       if (!trajectory) {
  84         AliGlobals::Exception(
  85           "AliEventAction::DisplayEvent: Unknown trajectory type.");
  86       }
  87       if ( (fDrawFlag == "ALL") ||
  88           ((fDrawFlag == "CHARGED") && (trajectory->GetCharge() != 0.))){
  89          trajectory->DrawTrajectory(50);
  90             // the argument number defines the size of the step points
  91             // use 2000 to make step points well visible
  92       }
  93     }
  94     G4UImanager::GetUIpointer()->ApplyCommand("/vis~/show/view");
  95   }
  96 }
  97
  98 // public methods
  99
 100 void AliEventAction::BeginOfEventAction(const G4Event* event)
 101 {
 102 // Called by G4 kernel at the beginning of event.
 103 // ---
 104
 105   G4int eventID = event->GetEventID();
 106
 107   // reset the counters (primary tracks, saved tracks)
 108   AliTrackingAction* trackingAction
 109     = AliTrackingAction::Instance();
 110   trackingAction->PrepareNewEvent();
 111
 112   if (fVerboseLevel>0)
 113     G4cout << ">>> Event " << event->GetEventID() << G4endl;
 114 }
 115
 116 void AliEventAction::EndOfEventAction(const G4Event* event)
 117 {
 118 // Called by G4 kernel at the end of event.
 119 // ---
 120
 121   // save the last primary track store of
 122   // the current event
 123   AliTrackingAction* trackingAction
 124     = AliTrackingAction::Instance();
 125   trackingAction->SaveAndDestroyTrack();
 126
 127   if (fVerboseLevel>0) {
 128     G4int nofPrimaryTracks = trackingAction->GetNofPrimaryTracks();
 129     G4cout  << "    " << nofPrimaryTracks <<
 130                " primary tracks processed." << G4endl;
 131   }
 132
 133   // display event
 134   DisplayEvent(event);
 135
 136   // aliroot
 137   // store event header data
 138   gAlice->GetHeader()->SetEvent(event->GetEventID());
 139   gAlice->GetHeader()->SetNvertex(event->GetNumberOfPrimaryVertex());
 140   gAlice->GetHeader()->SetNprimary(trackingAction->GetNofPrimaryTracks());
 141   gAlice->GetHeader()->SetNtrack(trackingAction->GetNofSavedTracks());
 142
 143   gAlice->FinishEvent();
 144 }
 145
 146
 147