Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/examples/extended/runAndEvent/RE01/src/RE01StackingAction.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

  1 //
  2 // ********************************************************************
  3 // * License and Disclaimer                                           *
  4 // *                                                                  *
  5 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
  6 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
  7 // * conditions of the Geant4 Software License,  included in the file *
  8 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
  9 // * include a list of copyright holders.                             *
 10 // *                                                                  *
 11 // * Neither the authors of this software system, nor their employing *
 12 // * institutes,nor the agencies providing financial support for this *
 13 // * work  make  any representation or  warranty, express or implied, *
 14 // * regarding  this  software system or assume any liability for its *
 15 // * use.  Please see the license in the file  LICENSE  and URL above *
 16 // * for the full disclaimer and the limitation of liability.         *
 17 // *                                                                  *
 18 // * This  code  implementation is the result of  the  scientific and *
 19 // * technical work of the GEANT4 collaboration.                      *
 20 // * By using,  copying,  modifying or  distributing the software (or *
 21 // * any work based  on the software)  you  agree  to acknowledge its *
 22 // * use  in  resulting  scientific  publications,  and indicate your *
 23 // * acceptance of all terms of the Geant4 Software license.          *
 24 // ********************************************************************
 25 //
 26 /// \file runAndEvent/RE01/src/RE01StackingAction.cc
 27 /// \brief Implementation of the RE01StackingAction class
 28 //
 29 //
 30 
 31 #include "RE01StackingAction.hh"
 32 
 33 #include "RE01CalorimeterHit.hh"
 34 #include "RE01TrackInformation.hh"
 35 
 36 #include "G4Event.hh"
 37 #include "G4HCofThisEvent.hh"
 38 #include "G4ParticleDefinition.hh"
 39 #include "G4ParticleTypes.hh"
 40 #include "G4RunManager.hh"
 41 #include "G4SDManager.hh"
 42 #include "G4SystemOfUnits.hh"
 43 #include "G4Track.hh"
 44 #include "G4TrackStatus.hh"
 45 #include "G4ios.hh"
 46 
 47 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 48 RE01StackingAction::RE01StackingAction()
 49   : G4UserStackingAction(), fStage(0), fCalorimeterHitsColID(-1)
 50 {
 51   ;
 52 }
 53 
 54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 55 RE01StackingAction::~RE01StackingAction()
 56 {
 57   ;
 58 }
 59 
 60 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 61 G4ClassificationOfNewTrack RE01StackingAction ::ClassifyNewTrack(const G4Track* aTrack)
 62 {
 63   G4ClassificationOfNewTrack classification = fUrgent;
 64 
 65   if (fStage == 0) {
 66     RE01TrackInformation* trackInfo;
 67     if (aTrack->GetTrackStatus() == fSuspend)  // Track reached to calorimeter
 68     {
 69       trackInfo = (RE01TrackInformation*)(aTrack->GetUserInformation());
 70       trackInfo->SetTrackingStatus(0);
 71       trackInfo->SetSourceTrackInformation(aTrack);
 72       //      G4cout << "Track " << aTrack->GetTrackID()
 73       //             << " (parentID " << aTrack->GetParentID()
 74       //             << ") has reached to calorimeter and has been suspended at "
 75       //             << aTrack->GetPosition() << G4endl;
 76       classification = fWaiting;
 77     }
 78     else if (aTrack->GetParentID() == 0)  // Primary particle
 79     {
 80       trackInfo = new RE01TrackInformation(aTrack);
 81       trackInfo->SetTrackingStatus(1);
 82       G4Track* theTrack = (G4Track*)aTrack;
 83       theTrack->SetUserInformation(trackInfo);
 84     }
 85   }
 86   return classification;
 87 }
 88 
 89 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 90 G4VHitsCollection* RE01StackingAction::GetCalCollection()
 91 {
 92   G4SDManager* SDMan = G4SDManager::GetSDMpointer();
 93   G4RunManager* runMan = G4RunManager::GetRunManager();
 94   if (fCalorimeterHitsColID < 0) {
 95     fCalorimeterHitsColID = SDMan->GetCollectionID("calCollection");
 96   }
 97   if (fCalorimeterHitsColID >= 0) {
 98     const G4Event* currentEvent = runMan->GetCurrentEvent();
 99     G4HCofThisEvent* HCE = currentEvent->GetHCofThisEvent();
100     return HCE->GetHC(fCalorimeterHitsColID);
101   }
102   return 0;
103 }
104 
105 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
106 void RE01StackingAction::NewStage()
107 {
108   // G4cout << "+++++++++++ Stage " << fStage << G4endl;
109   if (fStage == 0) {
110     // display trajetory information in the tracking region
111     G4cout << G4endl;
112     G4cout << "Tracks in tracking region have been processed. -- Stage 0 over." << G4endl;
113     G4cout << G4endl;
114   }
115   else {
116     // display calorimeter information caused by a "source" track
117     // in the tracker region
118     //    G4cout << G4endl;
119     //    G4cout << "Processing one shower originated by a source track "
120     //           << "in tracker region is over." << G4endl;
121     //    G4cout << G4endl;
122     RE01CalorimeterHitsCollection* CHC = (RE01CalorimeterHitsCollection*)GetCalCollection();
123     if (CHC) {
124       int n_hit = CHC->entries();
125       G4double totE = 0;
126       G4int n_hitByATrack = 0;
127       for (int i = 0; i < n_hit; i++) {
128         G4double edepByATrack = (*CHC)[i]->GetEdepByATrack();
129         if (edepByATrack > 0.) {
130           totE += edepByATrack;
131           if (n_hitByATrack == 0) {
132             (*CHC)[i]->GetTrackInformation()->Print();
133           }
134           n_hitByATrack++;
135           G4cout << "Cell[" << (*CHC)[i]->GetZ() << "," << (*CHC)[i]->GetPhi() << "]    "
136                  << edepByATrack / GeV << " [GeV]" << G4endl;
137           (*CHC)[i]->ClearEdepByATrack();
138         }
139       }
140       if (n_hitByATrack > 0) {
141         G4cout << "###  Total energy deposition in calorimeter by a source track in "
142                << n_hitByATrack << " cells : " << totE / GeV << " (GeV)" << G4endl;
143         G4cout << G4endl;
144       }
145     }
146   }
147 
148   if (stackManager->GetNUrgentTrack()) {
149     // Transfer all tracks in Urgent stack to Waiting stack, since all tracks
150     // in Waiting stack have already been transfered to Urgent stack before
151     // invokation of this method.
152     stackManager->TransferStackedTracks(fUrgent, fWaiting);
153 
154     // Then, transfer only one track to Urgent stack.
155     stackManager->TransferOneStackedTrack(fWaiting, fUrgent);
156 
157     fStage++;
158   }
159 }
160 
161 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
162 void RE01StackingAction::PrepareNewEvent()
163 {
164   fStage = 0;
165 }
166