Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer << 3 // * DISCLAIMER * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th << 5 // * The following disclaimer summarizes all the specific disclaimers * 6 // * the Geant4 Collaboration. It is provided << 6 // * of contributors to this software. The specific disclaimers,which * 7 // * conditions of the Geant4 Software License << 7 // * govern, are listed with their locations in: * 8 // * LICENSE and available at http://cern.ch/ << 8 // * http://cern.ch/geant4/license * 9 // * include a list of copyright holders. << 10 // * 9 // * * 11 // * Neither the authors of this software syst 10 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 11 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 12 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 13 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file << 14 // * use. * 16 // * for the full disclaimer and the limitatio << 17 // * 15 // * * 18 // * This code implementation is the result << 16 // * This code implementation is the intellectual property of the * 19 // * technical work of the GEANT4 collaboratio << 17 // * GEANT4 collaboration. * 20 // * By using, copying, modifying or distri << 18 // * By copying, distributing or modifying the Program (or any work * 21 // * any work based on the software) you ag << 19 // * based on the Program) you indicate your acceptance of this * 22 // * use in resulting scientific publicati << 20 // * statement, and all its terms. * 23 // * acceptance of all terms of the Geant4 Sof << 24 // ******************************************* 21 // ******************************************************************** 25 // 22 // 26 /// \file eventgenerator/HepMC/HepMCEx01/src/E << 27 /// \brief Implementation of the ExN04Stacking << 28 // << 29 // << 30 23 31 #include "ExN04StackingAction.hh" 24 #include "ExN04StackingAction.hh" 32 << 25 #include "G4SDManager.hh" 33 #include "ExN04StackingActionMessenger.hh" << 26 #include "G4RunManager.hh" 34 << 35 #include "G4Event.hh" 27 #include "G4Event.hh" 36 #include "G4HCofThisEvent.hh" 28 #include "G4HCofThisEvent.hh" 37 #include "G4ParticleDefinition.hh" << 38 #include "G4ParticleTypes.hh" << 39 #include "G4RunManager.hh" << 40 #include "G4SDManager.hh" << 41 #include "G4SystemOfUnits.hh" << 42 #include "G4Track.hh" 29 #include "G4Track.hh" 43 #include "G4TrackStatus.hh" 30 #include "G4TrackStatus.hh" >> 31 #include "G4ParticleDefinition.hh" >> 32 #include "G4ParticleTypes.hh" >> 33 #include "ExN04StackingActionMessenger.hh" >> 34 #include "G4ios.hh" 44 35 45 //....oooOO0OOooo........oooOO0OOooo........oo << 46 ExN04StackingAction::ExN04StackingAction() 36 ExN04StackingAction::ExN04StackingAction() 47 : G4UserStackingAction(), fTrkHits(0), fMuon << 37 : trkHits(0), muonHits(0), stage(0) 48 { << 38 { 49 fAngRoI = 30.0 * deg; << 39 angRoI = 30.0*deg; 50 fReqMuon = 2; << 40 reqMuon = 2; 51 fReqIso = 10; << 41 reqIso = 10; 52 fMessenger = new ExN04StackingActionMessenge << 42 theMessenger = new ExN04StackingActionMessenger(this); 53 } 43 } 54 44 55 //....oooOO0OOooo........oooOO0OOooo........oo << 56 ExN04StackingAction::~ExN04StackingAction() 45 ExN04StackingAction::~ExN04StackingAction() 57 { << 46 { delete theMessenger; } 58 delete fMessenger; << 59 } << 60 47 61 //....oooOO0OOooo........oooOO0OOooo........oo << 48 G4ClassificationOfNewTrack 62 G4ClassificationOfNewTrack ExN04StackingAction << 49 ExN04StackingAction::ClassifyNewTrack(const G4Track * aTrack) 63 { 50 { 64 G4ClassificationOfNewTrack classification = 51 G4ClassificationOfNewTrack classification = fWaiting; 65 switch (fStage) { << 52 switch(stage) 66 case 0: // Stage 0 : Primary muons only << 53 { 67 if (aTrack->GetParentID() == 0) { << 54 case 0: // Stage 0 : Primary muons only 68 G4ParticleDefinition* particleType = a << 55 if(aTrack->GetParentID()==0) 69 if ((particleType == G4MuonPlus::MuonP << 56 { 70 || (particleType == G4MuonMinus::M << 57 G4ParticleDefinition * particleType = aTrack->GetDefinition(); 71 { << 58 if((particleType==G4MuonPlus::MuonPlusDefinition()) 72 classification = fUrgent; << 59 ||(particleType==G4MuonMinus::MuonMinusDefinition())) 73 } << 60 { classification = fUrgent; } 74 } << 61 } >> 62 break; >> 63 >> 64 case 1: // Stage 1 : Charged primaries only >> 65 // Suspended tracks will be sent to the waiting stack >> 66 if(aTrack->GetParentID()!=0) { break; } >> 67 if(aTrack->GetTrackStatus()==fSuspend) { break; } >> 68 if(aTrack->GetDefinition()->GetPDGCharge()==0.) { break; } >> 69 classification = fUrgent; >> 70 break; >> 71 >> 72 default: // Stage 2 : Accept all primaries >> 73 // Accept all secondaries in RoI >> 74 // Kill secondaries outside RoI >> 75 if(aTrack->GetParentID()==0) >> 76 { >> 77 classification = fUrgent; 75 break; 78 break; 76 << 79 } 77 case 1: // Stage 1 : Charged primaries on << 80 if((angRoI<0.)||InsideRoI(aTrack,angRoI)) 78 // Suspended tracks wil << 81 { 79 if (aTrack->GetParentID() != 0) { << 80 break; << 81 } << 82 if (aTrack->GetTrackStatus() == fSuspend << 83 break; << 84 } << 85 if (aTrack->GetDefinition()->GetPDGCharg << 86 break; << 87 } << 88 classification = fUrgent; 82 classification = fUrgent; 89 break; 83 break; 90 << 84 } 91 default: // Stage 2 : Accept all primarie << 85 classification = fKill; 92 // Accept all secondar << 93 // Kill secondaries ou << 94 if (aTrack->GetParentID() == 0) { << 95 classification = fUrgent; << 96 break; << 97 } << 98 if ((fAngRoI < 0.) || InsideRoI(aTrack, << 99 classification = fUrgent; << 100 break; << 101 } << 102 classification = fKill; << 103 } 86 } 104 return classification; 87 return classification; 105 } 88 } 106 89 107 //....oooOO0OOooo........oooOO0OOooo........oo << 90 G4bool ExN04StackingAction::InsideRoI(const G4Track * aTrack,G4double ang) 108 G4bool ExN04StackingAction::InsideRoI(const G4 << 109 { 91 { 110 if (!fMuonHits) { << 92 if(!muonHits) 111 fMuonHits = (ExN04MuonHitsCollection*)GetC << 93 { muonHits = (ExN04MuonHitsCollection*)GetCollection("muonCollection"); } 112 } << 94 if(!muonHits) 113 if (!fMuonHits) { << 95 { G4cerr << "muonCollection NOT FOUND" << G4endl; 114 G4cerr << "muonCollection NOT FOUND" << G4 << 96 return true; } 115 return true; << 116 } << 117 97 118 G4int nhits = fMuonHits->entries(); << 98 G4int nhits = muonHits->entries(); 119 99 120 const G4ThreeVector trPos = aTrack->GetPosit 100 const G4ThreeVector trPos = aTrack->GetPosition(); 121 for (G4int i = 0; i < nhits; i++) { << 101 for(G4int i=0;i<nhits;i++) 122 G4ThreeVector muHitPos = (*fMuonHits)[i]-> << 102 { >> 103 G4ThreeVector muHitPos = (*muonHits)[i]->GetPos(); 123 G4double angl = muHitPos.angle(trPos); 104 G4double angl = muHitPos.angle(trPos); 124 if (angl < ang) { << 105 if(angl<ang) { return true; } 125 return true; << 126 } << 127 } 106 } 128 107 129 return false; 108 return false; 130 } 109 } 131 110 132 //....oooOO0OOooo........oooOO0OOooo........oo << 133 G4VHitsCollection* ExN04StackingAction::GetCol 111 G4VHitsCollection* ExN04StackingAction::GetCollection(G4String colName) 134 { 112 { 135 G4SDManager* SDMan = G4SDManager::GetSDMpoin 113 G4SDManager* SDMan = G4SDManager::GetSDMpointer(); 136 G4RunManager* runMan = G4RunManager::GetRunM 114 G4RunManager* runMan = G4RunManager::GetRunManager(); 137 int colID = SDMan->GetCollectionID(colName); 115 int colID = SDMan->GetCollectionID(colName); 138 if (colID >= 0) { << 116 if(colID>=0) >> 117 { 139 const G4Event* currentEvent = runMan->GetC 118 const G4Event* currentEvent = runMan->GetCurrentEvent(); 140 G4HCofThisEvent* HCE = currentEvent->GetHC 119 G4HCofThisEvent* HCE = currentEvent->GetHCofThisEvent(); 141 return HCE->GetHC(colID); 120 return HCE->GetHC(colID); 142 } 121 } 143 return 0; 122 return 0; 144 } 123 } 145 124 146 //....oooOO0OOooo........oooOO0OOooo........oo << 147 void ExN04StackingAction::NewStage() 125 void ExN04StackingAction::NewStage() 148 { 126 { 149 fStage++; << 127 stage++; 150 G4int nhits; 128 G4int nhits; 151 if (fStage == 1) { << 129 if(stage==1) 152 // Stage 0->1 : check if at least "fReqMuo << 130 { 153 // otherwise abort current ev << 131 // Stage 0->1 : check if at least "reqMuon" hits on muon chamber 154 if (!fMuonHits) { << 132 // otherwise abort current event 155 fMuonHits = (ExN04MuonHitsCollection*)Ge << 133 if(!muonHits) 156 } << 134 { muonHits = (ExN04MuonHitsCollection*)GetCollection("muonCollection"); } 157 if (!fMuonHits) { << 135 if(!muonHits) 158 G4cerr << "muonCollection NOT FOUND" << << 136 { G4cerr << "muonCollection NOT FOUND" << G4endl; 159 return; << 137 return; } 160 } << 138 nhits = muonHits->entries(); 161 nhits = fMuonHits->entries(); << 139 G4cout << "Stage 0->1 : " << nhits << " hits found in the muon chamber." 162 G4cout << "Stage 0->1 : " << nhits << " hi << 140 << G4endl; 163 if (nhits < fReqMuon) { << 141 if(nhits<reqMuon) >> 142 { 164 stackManager->clear(); 143 stackManager->clear(); 165 G4cout << "++++++++ event aborted" << G4 144 G4cout << "++++++++ event aborted" << G4endl; 166 return; 145 return; 167 } 146 } 168 stackManager->ReClassify(); 147 stackManager->ReClassify(); 169 return; 148 return; 170 } 149 } 171 150 172 else if (fStage == 2) { << 151 else if(stage==2) 173 // Stage 1->2 : check the isolation of muo << 152 { 174 // at least "fReqIsoMuon" iso << 153 // Stage 1->2 : check the isolation of muon tracks 175 // otherwise abort current ev << 154 // at least "reqIsoMuon" isolated muons 176 // Isolation requires "fReqIs << 155 // otherwise abort current event. 177 // (including own hits) in th << 156 // Isolation requires "reqIso" or less hits 178 // in the tracker layers. << 157 // (including own hits) in the RoI region 179 nhits = fMuonHits->entries(); << 158 // in the tracker layers. 180 if (!fTrkHits) { << 159 nhits = muonHits->entries(); 181 fTrkHits = (ExN04TrackerHitsCollection*) << 160 if(!trkHits) 182 } << 161 { trkHits = (ExN04TrackerHitsCollection*)GetCollection("trackerCollection"); } 183 if (!fTrkHits) { << 162 if(!trkHits) 184 G4cerr << "trackerCollection NOT FOUND" << 163 { G4cerr << "trackerCollection NOT FOUND" << G4endl; 185 return; << 164 return; } 186 } << 165 G4int nTrkhits = trkHits->entries(); 187 G4int nTrkhits = fTrkHits->entries(); << 188 G4int isoMuon = 0; 166 G4int isoMuon = 0; 189 for (G4int j = 0; j < nhits; j++) { << 167 for(G4int j=0;j<nhits;j++) 190 G4ThreeVector hitPos = (*fMuonHits)[j]-> << 168 { >> 169 G4ThreeVector hitPos = (*muonHits)[j]->GetPos(); 191 G4int nhitIn = 0; 170 G4int nhitIn = 0; 192 for (G4int jj = 0; (jj < nTrkhits) && (n << 171 for(G4int jj=0;(jj<nTrkhits)&&(nhitIn<=reqIso);jj++) 193 G4ThreeVector trkhitPos = (*fTrkHits)[ << 172 { 194 if (trkhitPos.angle(hitPos) < fAngRoI) << 173 G4ThreeVector trkhitPos = (*trkHits)[jj]->GetPos(); >> 174 if(trkhitPos.angle(hitPos)<angRoI) nhitIn++; 195 } 175 } 196 if (nhitIn <= fReqIso) isoMuon++; << 176 if(nhitIn<=reqIso) isoMuon++; 197 } 177 } 198 G4cout << "Stage 1->2 : " << isoMuon << " 178 G4cout << "Stage 1->2 : " << isoMuon << " isolated muon found." << G4endl; 199 if (isoMuon < fReqIsoMuon) { << 179 if(isoMuon<reqIsoMuon) >> 180 { 200 stackManager->clear(); 181 stackManager->clear(); 201 G4cout << "++++++++ event aborted" << G4 182 G4cout << "++++++++ event aborted" << G4endl; 202 return; 183 return; 203 } 184 } 204 stackManager->ReClassify(); 185 stackManager->ReClassify(); 205 return; 186 return; 206 } 187 } 207 188 208 else { << 189 else 209 // Other fStage change : just re-classify << 190 { >> 191 // Other stage change : just re-classify 210 stackManager->ReClassify(); 192 stackManager->ReClassify(); 211 } 193 } 212 } 194 } 213 << 195 214 //....oooOO0OOooo........oooOO0OOooo........oo << 215 void ExN04StackingAction::PrepareNewEvent() 196 void ExN04StackingAction::PrepareNewEvent() 216 { << 197 { 217 fStage = 0; << 198 stage = 0; 218 fTrkHits = 0; << 199 trkHits = 0; 219 fMuonHits = 0; << 200 muonHits = 0; 220 } 201 } >> 202 >> 203 221 204