| 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 #include "Par04ParallelFastSensitiveDetector.hh"
27
28 #include "Par04EventInformation.hh" // for Par04EventInformation
29 #include "Par04Hit.hh" // for Par04Hit, Par04HitsCollection
30
31 #include "G4Event.hh" // for G4Event
32 #include "G4EventManager.hh" // for G4EventManager
33 #include "G4HCofThisEvent.hh" // for G4HCofThisEvent
34 #include "G4SDManager.hh" // for G4SDManager
35 #include "G4Step.hh" // for G4Step
36 #include "G4Track.hh" // for G4Track
37
38 #include <CLHEP/Vector/Rotation.h> // for HepRotation
39 #include <CLHEP/Vector/ThreeVector.h> // for Hep3Vector
40 #include <G4CollectionNameVector.hh> // for G4CollectionNameVector
41 #include <G4FastHit.hh> // for G4FastHit
42 #include <G4FastTrack.hh> // for G4FastTrack
43 #include <G4RotationMatrix.hh> // for G4RotationMatrix
44 #include <G4StepPoint.hh> // for G4StepPoint
45 #include <G4THitsCollection.hh> // for G4THitsCollection
46 #include <G4ThreeVector.hh> // for G4ThreeVector
47 #include <G4VSensitiveDetector.hh> // for G4VSensitiveDetector
48 #include <G4VUserEventInformation.hh> // for G4VUserEventInformation
49 #include <cmath> // for floor
50 #include <cstddef> // for size_t
51 #include <vector> // for vector
52
53 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
54
55 Par04ParallelFastSensitiveDetector::Par04ParallelFastSensitiveDetector(G4String aName)
56 : G4VSensitiveDetector(aName)
57 {
58 collectionName.insert("physicalCellsFastSim");
59 }
60 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
61
62 Par04ParallelFastSensitiveDetector::Par04ParallelFastSensitiveDetector(G4String aName,
63 G4int aNbOfLayers,
64 G4int aNbOfSlices)
65 : G4VSensitiveDetector(aName), fNbOfLayers(aNbOfLayers), fNbOfSlices(aNbOfSlices)
66 {
67 collectionName.insert("physicalCellsFastSim");
68 }
69
70 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
71
72 Par04ParallelFastSensitiveDetector::~Par04ParallelFastSensitiveDetector() = default;
73
74 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
75
76 void Par04ParallelFastSensitiveDetector::Initialize(G4HCofThisEvent* aHCE)
77 {
78 fHitsCollection = new Par04HitsCollection(SensitiveDetectorName, collectionName[0]);
79 if (fHitCollectionID < 0) {
80 fHitCollectionID = G4SDManager::GetSDMpointer()->GetCollectionID(fHitsCollection);
81 }
82 aHCE->AddHitsCollection(fHitCollectionID, fHitsCollection);
83 }
84
85 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
86 G4bool Par04ParallelFastSensitiveDetector::ProcessHits(G4Step*, G4TouchableHistory*)
87 {
88 return true;
89 }
90
91 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
92 G4bool Par04ParallelFastSensitiveDetector::ProcessHits(const G4FastHit* aHit, const G4FastTrack*,
93 G4TouchableHistory* aTouchable)
94 {
95 G4double edep = aHit->GetEnergy();
96 if (edep == 0.) return true;
97
98 G4int layerNo = aTouchable->GetCopyNumber(0);
99 G4int sliceNo = aTouchable->GetCopyNumber(1);
100 G4int rowNo = aTouchable->GetCopyNumber(2);
101
102 G4int hitID = fNbOfLayers * fNbOfSlices * rowNo + fNbOfLayers * sliceNo + layerNo;
103 auto hit = fHitsMap[hitID].get();
104 if (hit == nullptr) {
105 fHitsMap[hitID] = std::unique_ptr<Par04Hit>(new Par04Hit());
106 hit = fHitsMap[hitID].get();
107 hit->SetPhiId(sliceNo);
108 hit->SetRhoId(layerNo);
109 hit->SetZid(rowNo);
110 }
111 if (layerNo != hit->GetRhoId()) {
112 G4cout << "ERROR, problem with Layer IDs: " << layerNo << " != " << hit->GetRhoId() << G4endl;
113 return false;
114 }
115 if (sliceNo != hit->GetPhiId()) {
116 G4cout << "ERROR, problem with Slice IDs: " << sliceNo << " != " << hit->GetPhiId() << G4endl;
117 return false;
118 }
119 if (rowNo != hit->GetZid()) {
120 G4cout << "ERROR, problem with Row IDs: " << rowNo << " != " << hit->GetZid() << G4endl;
121 return false;
122 }
123
124 // Add energy deposit from G4Step
125 hit->AddEdep(edep);
126 // Increment the counter
127 hit->AddNdep(1);
128 // Set hit type to parallel world fast simulation (even if hit was already marked as full
129 // sim, overwrite it)
130 hit->SetType(3);
131
132 return true;
133 }
134
135 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
136 void Par04ParallelFastSensitiveDetector::EndOfEvent(G4HCofThisEvent*)
137 {
138 for (const auto& hits : fHitsMap) {
139 fHitsCollection->insert(new Par04Hit(*hits.second.get()));
140 }
141 fHitsMap.clear();
142 }
143