| Geant4 Cross Reference |
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 //
27 /// \file B4/B4d/src/EventAction.cc
28 /// \brief Implementation of the B4d::EventAction class
29
30 #include "EventAction.hh"
31
32 #include "G4AnalysisManager.hh"
33 #include "G4Event.hh"
34 #include "G4HCofThisEvent.hh"
35 #include "G4RunManager.hh"
36 #include "G4SDManager.hh"
37 #include "G4THitsMap.hh"
38 #include "G4UnitsTable.hh"
39
40 #include <iomanip>
41
42 namespace B4d
43 {
44
45 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
46
47 G4THitsMap<G4double>* EventAction::GetHitsCollection(G4int hcID, const G4Event* event) const
48 {
49 auto hitsCollection = static_cast<G4THitsMap<G4double>*>(event->GetHCofThisEvent()->GetHC(hcID));
50
51 if (!hitsCollection) {
52 G4ExceptionDescription msg;
53 msg << "Cannot access hitsCollection ID " << hcID;
54 G4Exception("EventAction::GetHitsCollection()", "MyCode0003", FatalException, msg);
55 }
56
57 return hitsCollection;
58 }
59
60 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
61
62 G4double EventAction::GetSum(G4THitsMap<G4double>* hitsMap) const
63 {
64 G4double sumValue = 0.;
65 for (auto it : *hitsMap->GetMap()) {
66 // hitsMap->GetMap() returns the map of std::map<G4int, G4double*>
67 sumValue += *(it.second);
68 }
69 return sumValue;
70 }
71
72 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
73
74 void EventAction::PrintEventStatistics(G4double absoEdep, G4double absoTrackLength,
75 G4double gapEdep, G4double gapTrackLength) const
76 {
77 // Print event statistics
78 //
79 G4cout << " Absorber: total energy: " << std::setw(7) << G4BestUnit(absoEdep, "Energy")
80 << " total track length: " << std::setw(7) << G4BestUnit(absoTrackLength, "Length")
81 << G4endl << " Gap: total energy: " << std::setw(7) << G4BestUnit(gapEdep, "Energy")
82 << " total track length: " << std::setw(7) << G4BestUnit(gapTrackLength, "Length")
83 << G4endl;
84 }
85
86 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
87
88 void EventAction::BeginOfEventAction(const G4Event* /*event*/) {}
89
90 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
91
92 void EventAction::EndOfEventAction(const G4Event* event)
93 {
94 // Get hist collections IDs
95 if (fAbsoEdepHCID == -1) {
96 fAbsoEdepHCID = G4SDManager::GetSDMpointer()->GetCollectionID("Absorber/Edep");
97 fGapEdepHCID = G4SDManager::GetSDMpointer()->GetCollectionID("Gap/Edep");
98 fAbsoTrackLengthHCID = G4SDManager::GetSDMpointer()->GetCollectionID("Absorber/TrackLength");
99 fGapTrackLengthHCID = G4SDManager::GetSDMpointer()->GetCollectionID("Gap/TrackLength");
100 }
101
102 // Get sum values from hits collections
103 //
104 auto absoEdep = GetSum(GetHitsCollection(fAbsoEdepHCID, event));
105 auto gapEdep = GetSum(GetHitsCollection(fGapEdepHCID, event));
106
107 auto absoTrackLength = GetSum(GetHitsCollection(fAbsoTrackLengthHCID, event));
108 auto gapTrackLength = GetSum(GetHitsCollection(fGapTrackLengthHCID, event));
109
110 // get analysis manager
111 auto analysisManager = G4AnalysisManager::Instance();
112
113 // fill histograms
114 //
115 analysisManager->FillH1(0, absoEdep);
116 analysisManager->FillH1(1, gapEdep);
117 analysisManager->FillH1(2, absoTrackLength);
118 analysisManager->FillH1(3, gapTrackLength);
119
120 // fill ntuple
121 //
122 analysisManager->FillNtupleDColumn(0, absoEdep);
123 analysisManager->FillNtupleDColumn(1, gapEdep);
124 analysisManager->FillNtupleDColumn(2, absoTrackLength);
125 analysisManager->FillNtupleDColumn(3, gapTrackLength);
126 analysisManager->AddNtupleRow();
127
128 // print per event (modulo n)
129 //
130 auto eventID = event->GetEventID();
131 auto printModulo = G4RunManager::GetRunManager()->GetPrintProgress();
132 if ((printModulo > 0) && (eventID % printModulo == 0)) {
133 PrintEventStatistics(absoEdep, absoTrackLength, gapEdep, gapTrackLength);
134 G4cout << "--> End of event: " << eventID << "\n" << G4endl;
135 }
136 }
137
138 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
139
140 } // namespace B4d
141