| Geant4 Cross Reference |
1 // 1
2 // *******************************************
3 // * License and Disclaimer
4 // *
5 // * The Geant4 software is copyright of th
6 // * the Geant4 Collaboration. It is provided
7 // * conditions of the Geant4 Software License
8 // * LICENSE and available at http://cern.ch/
9 // * include a list of copyright holders.
10 // *
11 // * Neither the authors of this software syst
12 // * institutes,nor the agencies providing fin
13 // * work make any representation or warran
14 // * regarding this software system or assum
15 // * use. Please see the license in the file
16 // * for the full disclaimer and the limitatio
17 // *
18 // * This code implementation is the result
19 // * technical work of the GEANT4 collaboratio
20 // * By using, copying, modifying or distri
21 // * any work based on the software) you ag
22 // * use in resulting scientific publicati
23 // * acceptance of all terms of the Geant4 Sof
24 // *******************************************
25 //
26 /// \file runAndEvent/RE01/src/RE01StackingAct
27 /// \brief Implementation of the RE01StackingA
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........oo
48 RE01StackingAction::RE01StackingAction()
49 : G4UserStackingAction(), fStage(0), fCalori
50 {
51 ;
52 }
53
54 //....oooOO0OOooo........oooOO0OOooo........oo
55 RE01StackingAction::~RE01StackingAction()
56 {
57 ;
58 }
59
60 //....oooOO0OOooo........oooOO0OOooo........oo
61 G4ClassificationOfNewTrack RE01StackingAction
62 {
63 G4ClassificationOfNewTrack classification =
64
65 if (fStage == 0) {
66 RE01TrackInformation* trackInfo;
67 if (aTrack->GetTrackStatus() == fSuspend)
68 {
69 trackInfo = (RE01TrackInformation*)(aTra
70 trackInfo->SetTrackingStatus(0);
71 trackInfo->SetSourceTrackInformation(aTr
72 // G4cout << "Track " << aTrack->Ge
73 // << " (parentID " << aTrac
74 // << ") has reached to calo
75 // << aTrack->GetPosition()
76 classification = fWaiting;
77 }
78 else if (aTrack->GetParentID() == 0) // P
79 {
80 trackInfo = new RE01TrackInformation(aTr
81 trackInfo->SetTrackingStatus(1);
82 G4Track* theTrack = (G4Track*)aTrack;
83 theTrack->SetUserInformation(trackInfo);
84 }
85 }
86 return classification;
87 }
88
89 //....oooOO0OOooo........oooOO0OOooo........oo
90 G4VHitsCollection* RE01StackingAction::GetCalC
91 {
92 G4SDManager* SDMan = G4SDManager::GetSDMpoin
93 G4RunManager* runMan = G4RunManager::GetRunM
94 if (fCalorimeterHitsColID < 0) {
95 fCalorimeterHitsColID = SDMan->GetCollecti
96 }
97 if (fCalorimeterHitsColID >= 0) {
98 const G4Event* currentEvent = runMan->GetC
99 G4HCofThisEvent* HCE = currentEvent->GetHC
100 return HCE->GetHC(fCalorimeterHitsColID);
101 }
102 return 0;
103 }
104
105 //....oooOO0OOooo........oooOO0OOooo........oo
106 void RE01StackingAction::NewStage()
107 {
108 // G4cout << "+++++++++++ Stage " << fStage
109 if (fStage == 0) {
110 // display trajetory information in the tr
111 G4cout << G4endl;
112 G4cout << "Tracks in tracking region have
113 G4cout << G4endl;
114 }
115 else {
116 // display calorimeter information caused
117 // in the tracker region
118 // G4cout << G4endl;
119 // G4cout << "Processing one shower ori
120 // << "in tracker region is over
121 // G4cout << G4endl;
122 RE01CalorimeterHitsCollection* CHC = (RE01
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]->Get
129 if (edepByATrack > 0.) {
130 totE += edepByATrack;
131 if (n_hitByATrack == 0) {
132 (*CHC)[i]->GetTrackInformation()->
133 }
134 n_hitByATrack++;
135 G4cout << "Cell[" << (*CHC)[i]->GetZ
136 << edepByATrack / GeV << " [G
137 (*CHC)[i]->ClearEdepByATrack();
138 }
139 }
140 if (n_hitByATrack > 0) {
141 G4cout << "### Total energy depositio
142 << n_hitByATrack << " cells : "
143 G4cout << G4endl;
144 }
145 }
146 }
147
148 if (stackManager->GetNUrgentTrack()) {
149 // Transfer all tracks in Urgent stack to
150 // in Waiting stack have already been tran
151 // invokation of this method.
152 stackManager->TransferStackedTracks(fUrgen
153
154 // Then, transfer only one track to Urgent
155 stackManager->TransferOneStackedTrack(fWai
156
157 fStage++;
158 }
159 }
160
161 //....oooOO0OOooo........oooOO0OOooo........oo
162 void RE01StackingAction::PrepareNewEvent()
163 {
164 fStage = 0;
165 }
166