Geant4 Cross Reference |
1 // 1 2 // ******************************************* 3 // * License and Disclaimer 4 // * 5 // * The Geant4 software is copyright of th 6 // * the Geant4 Collaboration. It is provided 7 // * conditions of the Geant4 Software License 8 // * LICENSE and available at http://cern.ch/ 9 // * include a list of copyright holders. 10 // * 11 // * Neither the authors of this software syst 12 // * institutes,nor the agencies providing fin 13 // * work make any representation or warran 14 // * regarding this software system or assum 15 // * use. Please see the license in the file 16 // * for the full disclaimer and the limitatio 17 // * 18 // * This code implementation is the result 19 // * technical work of the GEANT4 collaboratio 20 // * By using, copying, modifying or distri 21 // * any work based on the software) you ag 22 // * use in resulting scientific publicati 23 // * acceptance of all terms of the Geant4 Sof 24 // ******************************************* 25 // 26 // 27 // 28 // G4PSTrackCounter 29 #include "G4PSTrackCounter.hh" 30 #include "G4UnitsTable.hh" 31 #include "G4VScoreHistFiller.hh" 32 33 ////////////////////////////////////////////// 34 // (Description) 35 // This is a primitive scorer class for scor 36 // 37 // Created: 2007-02-02 Tsukasa ASO, Akinori K 38 // 2010-07-22 Introduce Unit specification. 39 // 40 ////////////////////////////////////////////// 41 42 G4PSTrackCounter::G4PSTrackCounter(const G4Str 43 : G4VPrimitivePlotter(name, depth) 44 , fDirection(direction) 45 { 46 SetUnit(""); 47 } 48 49 G4bool G4PSTrackCounter::ProcessHits(G4Step* a 50 { 51 G4StepPoint* preStep = aStep->GetPreStepPoin 52 G4StepPoint* posStep = aStep->GetPostStepPoi 53 G4bool IsEnter = preStep->GetStepStatu 54 G4bool IsExit = posStep->GetStepStatu 55 56 // Regular : count in prestep volume. 57 G4int index = GetIndex(aStep); 58 59 // G4cout << " trk " << aStep->GetTrack()-> 60 // << " index " << index << " In " << IsEn 61 // << " PreStatus " << preStep->GetStepSta 62 // << " PostStatus " << posStep->GetStepSt 63 // << " w " << aStep->GetPreStepPoint()->G 64 // << G4endl; 65 66 G4bool flag = false; 67 68 if(IsEnter && fDirection == fCurrent_In) 69 flag = true; 70 else if(IsExit && fDirection == fCurrent_Out 71 flag = true; 72 else if((IsExit || IsEnter) && fDirection == 73 flag = true; 74 75 if(flag) 76 { 77 // G4cout << " Count " << G4endl; 78 G4double val = 1.0; 79 if(weighted) 80 val *= aStep->GetPreStepPoint()->GetWeig 81 EvtMap->add(index, val); 82 83 if(!hitIDMap.empty() && hitIDMap.find(inde 84 { 85 auto filler = G4VScoreHistFiller::Instan 86 if(filler == nullptr) 87 { 88 G4Exception( 89 "G4PSTrackCounter::ProcessHits", "SC 90 "G4TScoreHistFiller is not instantia 91 } 92 else 93 { 94 filler->FillH1(hitIDMap[index], 95 aStep->GetPreStepPoint( 96 } 97 } 98 } 99 100 return true; 101 } 102 103 void G4PSTrackCounter::Initialize(G4HCofThisEv 104 { 105 EvtMap = new G4THitsMap<G4double>(detector-> 106 if(HCID < 0) 107 { 108 HCID = GetCollectionID(0); 109 } 110 HCE->AddHitsCollection(HCID, (G4VHitsCollect 111 // G4cout << "----- PS Initialize " << G4e 112 } 113 114 void G4PSTrackCounter::clear() { EvtMap->clear 115 116 void G4PSTrackCounter::PrintAll() 117 { 118 G4cout << " MultiFunctionalDet " << detecto 119 G4cout << " PrimitiveScorer " << GetName() < 120 G4cout << " Number of entries " << EvtMap->e 121 for(const auto& [copy, count] : *(EvtMap->Ge 122 { 123 G4cout << " copy no.: " << copy 124 << " track count: " << *(count) << 125 } 126 } 127 128 void G4PSTrackCounter::SetUnit(const G4String& 129 { 130 if(unit.empty()) 131 { 132 unitName = unit; 133 unitValue = 1.0; 134 } 135 else 136 { 137 G4String msg = "Invalid unit [" + unit + " 138 GetUnit() + "] ) for " + Ge 139 G4Exception("G4PSTrackCounter::SetUnit", " 140 } 141 } 142