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 "Par04ParallelFullSensitiveDetector.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 Par04ParallelFullSensitiveDetector::Par04ParallelFullSensitiveDetector(G4String aName) 56 : G4VSensitiveDetector(aName) 57 { 58 collectionName.insert("physicalCellsFullSim"); 59 } 60 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 61 62 Par04ParallelFullSensitiveDetector::Par04ParallelFullSensitiveDetector(G4String aName, 63 G4int aNbOfLayers, 64 G4int aNbOfSlices, 65 G4int aNbOfRows) 66 : G4VSensitiveDetector(aName), 67 fNbOfLayers(aNbOfLayers), 68 fNbOfSlices(aNbOfSlices), 69 fNbOfRows(aNbOfRows) 70 { 71 collectionName.insert("physicalCellsFullSim"); 72 } 73 74 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 75 76 Par04ParallelFullSensitiveDetector::~Par04ParallelFullSensitiveDetector() = default; 77 78 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 79 80 void Par04ParallelFullSensitiveDetector::Initialize(G4HCofThisEvent* aHCE) 81 { 82 fHitsCollection = new Par04HitsCollection(SensitiveDetectorName, collectionName[0]); 83 if (fHitCollectionID < 0) { 84 fHitCollectionID = G4SDManager::GetSDMpointer()->GetCollectionID(fHitsCollection); 85 } 86 aHCE->AddHitsCollection(fHitCollectionID, fHitsCollection); 87 } 88 89 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 90 91 G4bool Par04ParallelFullSensitiveDetector::ProcessHits(G4Step* aStep, G4TouchableHistory*) 92 { 93 G4double edep = aStep->GetTotalEnergyDeposit(); 94 if (edep == 0.) return true; 95 96 auto touchable = (G4TouchableHistory*)(aStep->GetPreStepPoint()->GetTouchable()); 97 98 G4int layerNo = touchable->GetCopyNumber(0); 99 G4int sliceNo = touchable->GetCopyNumber(1); 100 G4int rowNo = touchable->GetCopyNumber(2); 101 102 G4int hitID = fNbOfLayers * fNbOfSlices * rowNo + fNbOfLayers * sliceNo + layerNo; 103 if (layerNo >= fNbOfLayers) { 104 G4cout << "ERROR, problem with Layer IDs: " << layerNo << " > " << fNbOfLayers << G4endl; 105 return false; 106 } 107 if (sliceNo >= fNbOfSlices) { 108 G4cout << "ERROR, problem with Slice IDs: " << sliceNo << " >= " << fNbOfSlices << G4endl; 109 return false; 110 } 111 if (rowNo >= fNbOfRows) { 112 G4cout << "ERROR, problem with Row IDs: " << rowNo << " >= " << fNbOfRows << G4endl; 113 return false; 114 } 115 auto hit = fHitsMap[hitID].get(); 116 if (hit == nullptr) { 117 fHitsMap[hitID] = std::unique_ptr<Par04Hit>(new Par04Hit()); 118 hit = fHitsMap[hitID].get(); 119 hit->SetPhiId(sliceNo); 120 hit->SetRhoId(layerNo); 121 hit->SetZid(rowNo); 122 } 123 124 // Add energy deposit from G4Step 125 hit->AddEdep(edep); 126 // Increment the counter 127 hit->AddNdep(1); 128 129 // Fill time information from G4Step 130 // If it's already filled, choose hit with earliest global time 131 if (hit->GetTime() == -1 || hit->GetTime() > aStep->GetTrack()->GetGlobalTime()) 132 hit->SetTime(aStep->GetTrack()->GetGlobalTime()); 133 134 // Set type to parallel world full hit 135 hit->SetType(2); 136 137 return true; 138 } 139 140 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 141 void Par04ParallelFullSensitiveDetector::EndOfEvent(G4HCofThisEvent*) 142 { 143 for (const auto& hits : fHitsMap) { 144 fHitsCollection->insert(new Par04Hit(*hits.second.get())); 145 } 146 fHitsMap.clear(); 147 } 148