Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 // 26 // 27 // ------------------------------------------- 27 // ------------------------------------------------------------ 28 // GEANT 4 class implementation file 28 // GEANT 4 class implementation file 29 // CERN Geneva Switzerland 29 // CERN Geneva Switzerland 30 // 30 // 31 // 31 // 32 // ------------ GammaRayTelDigitizer --- 32 // ------------ GammaRayTelDigitizer ------ 33 // by F.Longo, R.Giannitrapani & G 33 // by F.Longo, R.Giannitrapani & G.Santin (13 nov 2000) 34 // 34 // 35 // ******************************************* 35 // ************************************************************ 36 36 37 #include <vector> 37 #include <vector> 38 38 39 #include "GammaRayTelAnticoincidenceHit.hh" 39 #include "GammaRayTelAnticoincidenceHit.hh" 40 #include "GammaRayTelCalorimeterHit.hh" 40 #include "GammaRayTelCalorimeterHit.hh" 41 #include "GammaRayTelDigi.hh" 41 #include "GammaRayTelDigi.hh" 42 #include "GammaRayTelDigitizer.hh" 42 #include "GammaRayTelDigitizer.hh" 43 #include "GammaRayTelDigitizerMessenger.hh" 43 #include "GammaRayTelDigitizerMessenger.hh" 44 #include "GammaRayTelTrackerHit.hh" 44 #include "GammaRayTelTrackerHit.hh" 45 45 46 #include "G4DigiManager.hh" 46 #include "G4DigiManager.hh" 47 #include "G4Event.hh" 47 #include "G4Event.hh" 48 #include "G4EventManager.hh" 48 #include "G4EventManager.hh" 49 #include "G4ios.hh" 49 #include "G4ios.hh" 50 #include "G4SDManager.hh" 50 #include "G4SDManager.hh" 51 #include "G4SystemOfUnits.hh" 51 #include "G4SystemOfUnits.hh" 52 52 53 //....oooOO0OOooo........oooOO0OOooo........oo 53 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 54 54 55 GammaRayTelDigitizer::GammaRayTelDigitizer(G4S 55 GammaRayTelDigitizer::GammaRayTelDigitizer(G4String name) : G4VDigitizerModule(name) { 56 constexpr auto DIGIT_COLLECTION_NAME{"Digi 56 constexpr auto DIGIT_COLLECTION_NAME{"DigitsCollection"}; 57 collectionName.push_back(DIGIT_COLLECTION_ 57 collectionName.push_back(DIGIT_COLLECTION_NAME); 58 58 59 // create a messenger for this class 59 // create a messenger for this class 60 digitizerMessenger = new GammaRayTelDigiti 60 digitizerMessenger = new GammaRayTelDigitizerMessenger(this); 61 } 61 } 62 62 63 //....oooOO0OOooo........oooOO0OOooo........oo 63 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 64 64 65 GammaRayTelDigitizer::~GammaRayTelDigitizer() 65 GammaRayTelDigitizer::~GammaRayTelDigitizer() { 66 delete digitizerMessenger; 66 delete digitizerMessenger; 67 } 67 } 68 68 69 //....oooOO0OOooo........oooOO0OOooo........oo 69 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 70 70 71 void GammaRayTelDigitizer::Digitize() { 71 void GammaRayTelDigitizer::Digitize() { 72 digitsCollection = new GammaRayTelDigitsCo 72 digitsCollection = new GammaRayTelDigitsCollection("GammaRayTelDigitizer", "DigitsCollection"); // to create the Digi Collection 73 73 74 auto *digitManager = G4DigiManager::GetDMp 74 auto *digitManager = G4DigiManager::GetDMpointer(); 75 75 76 G4int trackerCollectionIdentifier{0}; // T 76 G4int trackerCollectionIdentifier{0}; // TrackerCollection 77 G4int calorimeterCollectionIdentifier{0}; 77 G4int calorimeterCollectionIdentifier{0}; // CalorimeterCollection 78 G4int anticoincidenceCollectionIdentifier{ 78 G4int anticoincidenceCollectionIdentifier{0}; // AnticoincidenceCollection 79 79 80 totalEnergy = 0.; // at each event 80 totalEnergy = 0.; // at each event 81 81 82 // TKR hits collection 82 // TKR hits collection 83 83 84 trackerCollectionIdentifier = digitManager 84 trackerCollectionIdentifier = digitManager->GetHitsCollectionID("TrackerCollection"); 85 GammaRayTelTrackerHitsCollection *trackerC 85 GammaRayTelTrackerHitsCollection *trackerCollection{nullptr}; 86 trackerCollection = (GammaRayTelTrackerHit 86 trackerCollection = (GammaRayTelTrackerHitsCollection*) (digitManager->GetHitsCollection(trackerCollectionIdentifier)); 87 87 88 // CAL hits collection 88 // CAL hits collection 89 89 90 calorimeterCollectionIdentifier = digitMan 90 calorimeterCollectionIdentifier = digitManager->GetHitsCollectionID("CalorimeterCollection"); 91 GammaRayTelCalorimeterHitsCollection *calo 91 GammaRayTelCalorimeterHitsCollection *calorimeterCollection{nullptr}; 92 calorimeterCollection = (GammaRayTelCalori 92 calorimeterCollection = (GammaRayTelCalorimeterHitsCollection*) (digitManager->GetHitsCollection(calorimeterCollectionIdentifier)); 93 93 94 // ACD hits collection 94 // ACD hits collection 95 95 96 anticoincidenceCollectionIdentifier = digi 96 anticoincidenceCollectionIdentifier = digitManager->GetHitsCollectionID("AnticoincidenceCollection"); 97 GammaRayTelAnticoincidenceHitsCollection * 97 GammaRayTelAnticoincidenceHitsCollection *anticoincidenceCollection{nullptr}; 98 anticoincidenceCollection = (GammaRayTelAn 98 anticoincidenceCollection = (GammaRayTelAnticoincidenceHitsCollection*) (digitManager->GetHitsCollection(anticoincidenceCollectionIdentifier)); 99 99 100 if (trackerCollection != nullptr) { 100 if (trackerCollection != nullptr) { 101 G4int numberOfHits = trackerCollection 101 G4int numberOfHits = trackerCollection->entries(); 102 102 103 for (auto i = 0; i < numberOfHits; i++ 103 for (auto i = 0; i < numberOfHits; i++) { 104 G4double depositedEnergy = (*track 104 G4double depositedEnergy = (*trackerCollection)[i]->GetDepositedEnergy(); 105 G4int stripNumber = (*trackerColle 105 G4int stripNumber = (*trackerCollection)[i]->GetStripNumber(); 106 G4int planeNumber = (*trackerColle 106 G4int planeNumber = (*trackerCollection)[i]->GetSiliconPlaneNumber(); 107 G4int IsX = (*trackerCollection)[i 107 G4int IsX = (*trackerCollection)[i]->GetPlaneType(); 108 108 109 // digi generation only if energy 109 // digi generation only if energy deposit is greater than threshold 110 110 111 if (depositedEnergy > energyThresh 111 if (depositedEnergy > energyThreshold) { 112 auto *digit = new GammaRayTelD 112 auto *digit = new GammaRayTelDigi(); 113 digit->SetPlaneNumber(planeNum 113 digit->SetPlaneNumber(planeNumber); 114 digit->SetPlaneType(IsX); 114 digit->SetPlaneType(IsX); 115 digit->SetStripNumber(stripNum 115 digit->SetStripNumber(stripNumber); 116 digit->SetDigitType(0); 116 digit->SetDigitType(0); 117 digit->SetEnergy(0.); 117 digit->SetEnergy(0.); 118 digitsCollection->insert(digit 118 digitsCollection->insert(digit); 119 } 119 } 120 } 120 } 121 } 121 } 122 122 123 if (calorimeterCollection != nullptr) { 123 if (calorimeterCollection != nullptr) { 124 G4int numberOfHits = calorimeterCollec 124 G4int numberOfHits = calorimeterCollection->entries(); 125 125 126 for (auto i = 0; i < numberOfHits; i++ 126 for (auto i = 0; i < numberOfHits; i++) { 127 totalEnergy += (*calorimeterCollec 127 totalEnergy += (*calorimeterCollection)[i]->GetCALDepositedEnergy(); 128 } 128 } 129 129 130 // digit generation only if energy dep 130 // digit generation only if energy deposit is greater than 0. 131 131 132 if (totalEnergy > 0.) { 132 if (totalEnergy > 0.) { 133 auto *digit = new GammaRayTelDigi( 133 auto *digit = new GammaRayTelDigi(); 134 digit->SetPlaneNumber(0); 134 digit->SetPlaneNumber(0); 135 digit->SetPlaneType(0); 135 digit->SetPlaneType(0); 136 digit->SetStripNumber(0); 136 digit->SetStripNumber(0); 137 digit->SetDigitType(1); 137 digit->SetDigitType(1); 138 digit->SetEnergy(totalEnergy); 138 digit->SetEnergy(totalEnergy); 139 digitsCollection->insert(digit); 139 digitsCollection->insert(digit); 140 } 140 } 141 } 141 } 142 142 143 if (anticoincidenceCollection != nullptr) 143 if (anticoincidenceCollection != nullptr) { 144 G4int numberOfHits = anticoincidenceCo 144 G4int numberOfHits = anticoincidenceCollection->entries(); 145 145 146 for (G4int i = 0; i < numberOfHits; i+ 146 for (G4int i = 0; i < numberOfHits; i++) { 147 auto energy = (*anticoincidenceCol 147 auto energy = (*anticoincidenceCollection)[i]->GetEdepACD(); 148 auto type = (*anticoincidenceColle 148 auto type = (*anticoincidenceCollection)[i]->GetACDTileNumber(); 149 149 150 // digit generation only if energy 150 // digit generation only if energy deposit is greater than 0. 151 151 152 if (energy > acdThreshold) { 152 if (energy > acdThreshold) { 153 auto *digit = new GammaRayTelD 153 auto *digit = new GammaRayTelDigi(); 154 digit->SetPlaneNumber(0); 154 digit->SetPlaneNumber(0); 155 digit->SetPlaneType(0); 155 digit->SetPlaneType(0); 156 digit->SetStripNumber(type); 156 digit->SetStripNumber(type); 157 digit->SetDigitType(2); 157 digit->SetDigitType(2); 158 digit->SetEnergy(energy); 158 digit->SetEnergy(energy); 159 digitsCollection->insert(digit 159 digitsCollection->insert(digit); 160 } 160 } 161 } 161 } 162 } 162 } 163 163 164 if (trackerCollection != nullptr || antico 164 if (trackerCollection != nullptr || anticoincidenceCollection != nullptr || calorimeterCollection != nullptr) { 165 G4cout << "Number of digits in this ev 165 G4cout << "Number of digits in this event: " << digitsCollection->entries() 166 // << " " << digitsCollection->GetName 166 // << " " << digitsCollection->GetName() 167 // << " " << digitsCollection->GetDMna 167 // << " " << digitsCollection->GetDMname() 168 << G4endl; 168 << G4endl; 169 } 169 } 170 170 171 StoreDigiCollection(digitsCollection); 171 StoreDigiCollection(digitsCollection); 172 172 173 G4int digitsCollectionIdentifier{-1}; 173 G4int digitsCollectionIdentifier{-1}; 174 if (digitsCollectionIdentifier < 0) { 174 if (digitsCollectionIdentifier < 0) { 175 // digiManager->List(); 175 // digiManager->List(); 176 digitsCollectionIdentifier = digitMana 176 digitsCollectionIdentifier = digitManager->GetDigiCollectionID("GammaRayTelDigitizer/DigitsCollection"); 177 } 177 } 178 } 178 } 179 179