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 // $Id: G4MultiFunctionalDetector.cc,v 1.4 2006/06/29 18:05:47 gunter Exp $ >> 28 // GEANT4 tag $Name: geant4-08-03-patch-02 $ 27 // 29 // 28 // G4MultiFunctionalDetector 30 // G4MultiFunctionalDetector 29 // << 30 // 2010-07-23 T.Aso Call PS if the step length << 31 // << 32 // << 33 #include "G4MultiFunctionalDetector.hh" 31 #include "G4MultiFunctionalDetector.hh" 34 << 35 #include "G4SDManager.hh" 32 #include "G4SDManager.hh" 36 #include "G4VPrimitiveScorer.hh" 33 #include "G4VPrimitiveScorer.hh" 37 34 38 G4MultiFunctionalDetector::G4MultiFunctionalDe << 35 G4MultiFunctionalDetector::G4MultiFunctionalDetector(G4String name) 39 : G4VSensitiveDetector(name) << 36 :G4VSensitiveDetector(name) 40 { << 37 {;} 41 ; << 42 } << 43 38 44 G4MultiFunctionalDetector::~G4MultiFunctionalD 39 G4MultiFunctionalDetector::~G4MultiFunctionalDetector() 45 { << 40 {;} 46 for (auto pr : primitives) << 47 delete pr; << 48 primitives.clear(); << 49 } << 50 41 51 G4bool G4MultiFunctionalDetector::ProcessHits( << 42 G4bool G4MultiFunctionalDetector::ProcessHits(G4Step* aStep,G4TouchableHistory* aTH) 52 { << 43 { 53 if (aStep->GetStepLength() > 0. || aStep->Ge << 44 G4int nPrim = primitives.size(); 54 for (auto pr : primitives) { << 45 for(G4int iPrim=0;iPrim<nPrim;iPrim++) 55 pr->HitPrimitive(aStep, aTH); << 46 { primitives[iPrim]->HitPrimitive(aStep,aTH); } 56 } << 47 return true; 57 } << 58 return true; << 59 } 48 } 60 49 61 G4bool G4MultiFunctionalDetector::RegisterPrim 50 G4bool G4MultiFunctionalDetector::RegisterPrimitive(G4VPrimitiveScorer* aPS) 62 { 51 { 63 for (auto pr : primitives) { << 52 G4int nPrim = primitives.size(); 64 if (pr == aPS) { << 53 for(G4int iPrim=0;iPrim<nPrim;iPrim++) 65 G4ExceptionDescription ED; << 54 { 66 ED << "Primitive <" << aPS->GetName() << << 55 if(primitives[iPrim]==aPS) 67 << ">." << G4endl << "Method Register << 56 { 68 G4Exception("G4MultiFunctionalDetector:: << 57 G4cerr << "Primitive <" << aPS->GetName() << "> is already defined in <" << SensitiveDetectorName 69 return false; << 58 << ">." << G4endl << "Method RegisterPrimitive() is ignored." << G4endl; 70 } << 59 return false; 71 } << 60 } 72 primitives.push_back(aPS); << 61 } 73 aPS->SetMultiFunctionalDetector(this); << 62 primitives.push_back(aPS); 74 collectionName.insert(aPS->GetName()); << 63 aPS->SetMultiFunctionalDetector(this); 75 if (G4SDManager::GetSDMpointer()->FindSensit << 64 collectionName.insert(aPS->GetName()); 76 { << 65 if(G4SDManager::GetSDMpointer()->FindSensitiveDetector(SensitiveDetectorName,false)) 77 // This G4MultiFunctionalDetector has alre << 66 { 78 // G4SDManager. Make sure this new primiti << 67 // This G4MultiFunctionalDetector has already been registered to G4SDManager. 79 G4SDManager::GetSDMpointer()->AddNewCollec << 68 // Make sure this new primitive is registered as well. 80 } << 69 G4SDManager::GetSDMpointer()->AddNewCollection(SensitiveDetectorName,aPS->GetName()); 81 return true; << 70 } >> 71 return true; 82 } 72 } 83 73 84 G4bool G4MultiFunctionalDetector::RemovePrimit 74 G4bool G4MultiFunctionalDetector::RemovePrimitive(G4VPrimitiveScorer* aPS) 85 { 75 { 86 auto pr = std::find(primitives.begin(), prim << 76 std::vector<G4VPrimitiveScorer*>::iterator iterPS; 87 if (pr != primitives.end()) { << 77 std::vector<G4String>::iterator iterName = collectionName.begin(); 88 primitives.erase(pr); << 78 for(iterPS=primitives.begin();iterPS!=primitives.end();iterPS++) 89 aPS->SetMultiFunctionalDetector(nullptr); << 79 { 90 return true; << 80 if(*iterPS==aPS) 91 } << 81 { 92 G4cerr << "Primitive <" << aPS->GetName() << << 82 primitives.erase(iterPS); 93 << ">." << G4endl << "Method RemovePr << 83 aPS->SetMultiFunctionalDetector(0); 94 return false; << 84 return true; 95 } << 85 } >> 86 iterName++; >> 87 } >> 88 G4cerr << "Primitive <" << aPS->GetName() << "> is not defined in <" << SensitiveDetectorName >> 89 << ">." << G4endl << "Method RemovePrimitive() is ignored." << G4endl; >> 90 return false; >> 91 } 96 92 97 void G4MultiFunctionalDetector::Initialize(G4H 93 void G4MultiFunctionalDetector::Initialize(G4HCofThisEvent* HC) 98 { 94 { 99 for (auto pr : primitives) { << 95 G4int nPrim = primitives.size(); 100 pr->Initialize(HC); << 96 for(G4int iPrim=0;iPrim<nPrim;iPrim++) 101 } << 97 { primitives[iPrim]->Initialize(HC); } 102 } 98 } 103 99 104 void G4MultiFunctionalDetector::EndOfEvent(G4H 100 void G4MultiFunctionalDetector::EndOfEvent(G4HCofThisEvent* HC) 105 { 101 { 106 for (auto pr : primitives) { << 102 G4int nPrim = primitives.size(); 107 pr->EndOfEvent(HC); << 103 for(G4int iPrim=0;iPrim<nPrim;iPrim++) 108 } << 104 { primitives[iPrim]->EndOfEvent(HC); } 109 } 105 } 110 106 111 void G4MultiFunctionalDetector::clear() 107 void G4MultiFunctionalDetector::clear() 112 { 108 { 113 for (auto pr : primitives) { << 109 G4int nPrim = primitives.size(); 114 pr->clear(); << 110 for(G4int iPrim=0;iPrim<nPrim;iPrim++) 115 } << 111 { primitives[iPrim]->clear(); } 116 } 112 } 117 113 118 void G4MultiFunctionalDetector::DrawAll() 114 void G4MultiFunctionalDetector::DrawAll() 119 { 115 { 120 for (auto pr : primitives) { << 116 G4int nPrim = primitives.size(); 121 pr->DrawAll(); << 117 for(G4int iPrim=0;iPrim<nPrim;iPrim++) 122 } << 118 { primitives[iPrim]->DrawAll(); } 123 } 119 } 124 120 125 void G4MultiFunctionalDetector::PrintAll() 121 void G4MultiFunctionalDetector::PrintAll() 126 { 122 { 127 for (auto pr : primitives) { << 123 G4int nPrim = primitives.size(); 128 pr->PrintAll(); << 124 for(G4int iPrim=0;iPrim<nPrim;iPrim++) 129 } << 125 { primitives[iPrim]->PrintAll(); } 130 } 126 } >> 127 131 128