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 // 28 // CaTS (Calorimetry and Tracking Simulation) 29 // 30 // Authors : Hans Wenzel 31 // Soon Yung Jun 32 // (Fermi National Accelerator Laboratory) 33 // 34 // History 35 // October 18th, 2021 : first implementation 36 // 37 // ******************************************************************** 38 // 39 /// \file CalorimeterSD.cc 40 /// \brief Implementation of the CaTS::CalorimeterSD class 41 42 // Geant4 headers 43 #include "G4HCofThisEvent.hh" 44 #include "G4Step.hh" 45 #include "G4SDManager.hh" 46 #include "G4ios.hh" 47 // project headers 48 #include "CalorimeterSD.hh" 49 #include "ConfigurationManager.hh" 50 51 CalorimeterSD::CalorimeterSD(G4String name) 52 : G4VSensitiveDetector(name) 53 { 54 G4String HCname = name + "_HC"; 55 collectionName.insert(HCname); 56 G4cout << collectionName.size() << " CalorimeterSD name: " << name 57 << " collection Name: " << HCname << G4endl; 58 fHCID = -1; 59 verbose = ConfigurationManager::getInstance()->isEnable_verbose(); 60 } 61 62 void CalorimeterSD::Initialize(G4HCofThisEvent* hce) 63 { 64 fCalorimeterHitsCollection = 65 new CalorimeterHitsCollection(SensitiveDetectorName, collectionName[0]); 66 if(fHCID < 0) 67 { 68 if(verbose) 69 G4cout << "CalorimeterSD::Initialize: " << SensitiveDetectorName << " " 70 << collectionName[0] << G4endl; 71 fHCID = G4SDManager::GetSDMpointer()->GetCollectionID(collectionName[0]); 72 } 73 hce->AddHitsCollection(fHCID, fCalorimeterHitsCollection); 74 } 75 76 G4bool CalorimeterSD::ProcessHits(G4Step* aStep, G4TouchableHistory*) 77 { 78 // auto can be used for all results from retrieving functions 79 auto edep = aStep->GetTotalEnergyDeposit() / CLHEP::MeV; 80 if(edep == 0.) 81 return false; 82 auto time = aStep->GetPreStepPoint()->GetGlobalTime() / CLHEP::ns; 83 auto touch = aStep->GetPreStepPoint()->GetTouchable(); 84 auto cellpos = touch->GetTranslation(); 85 unsigned int ID = aStep->GetPreStepPoint()->GetPhysicalVolume()->GetCopyNo(); 86 auto theTrack = aStep->GetTrack(); 87 auto particleType = theTrack->GetDefinition()->GetParticleName(); 88 // 89 // check if this cell has been hit before 90 // fCalorimeterHitsCollection 91 for(unsigned int j = 0; j < fCalorimeterHitsCollection->entries(); j++) 92 { 93 CalorimeterHit* aPreviousHit = (*fCalorimeterHitsCollection)[j]; 94 if(ID == aPreviousHit->GetId()) 95 { 96 aPreviousHit->SetEdep(aStep->GetTotalEnergyDeposit() + 97 aPreviousHit->GetEdep()); 98 if((particleType == "e+") || (particleType == "gamma") || 99 (particleType == "e-")) 100 { 101 aPreviousHit->Setem_Edep(edep + aPreviousHit->Getem_Edep()); 102 } 103 return true; 104 } 105 } 106 // 107 // otherwise create a new hit: 108 // 109 CalorimeterHit* newHit; 110 if((particleType == "e+") || (particleType == "gamma") || 111 (particleType == "e-")) 112 { 113 newHit = new CalorimeterHit(ID, edep, edep, time, cellpos); 114 } 115 else 116 { 117 newHit = new CalorimeterHit(ID, edep, 0.0, time, cellpos); 118 } 119 fCalorimeterHitsCollection->insert(newHit); 120 return true; 121 } 122 123 void CalorimeterSD::EndOfEvent(G4HCofThisEvent*) 124 { 125 G4int NbHits = fCalorimeterHitsCollection->entries(); 126 if(verbose) 127 G4cout << " Number of CalorimeterHits: " << NbHits << G4endl; 128 } 129