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 // G4PSFlatSurfaceFlux 28 // G4PSFlatSurfaceFlux 29 #include "G4PSFlatSurfaceFlux.hh" 29 #include "G4PSFlatSurfaceFlux.hh" 30 30 31 #include "G4SystemOfUnits.hh" 31 #include "G4SystemOfUnits.hh" 32 #include "G4StepStatus.hh" 32 #include "G4StepStatus.hh" 33 #include "G4Track.hh" 33 #include "G4Track.hh" 34 #include "G4VSolid.hh" 34 #include "G4VSolid.hh" 35 #include "G4VPhysicalVolume.hh" 35 #include "G4VPhysicalVolume.hh" 36 #include "G4VPVParameterisation.hh" 36 #include "G4VPVParameterisation.hh" 37 #include "G4UnitsTable.hh" 37 #include "G4UnitsTable.hh" 38 #include "G4GeometryTolerance.hh" 38 #include "G4GeometryTolerance.hh" 39 #include "G4VScoreHistFiller.hh" << 40 << 41 ////////////////////////////////////////////// 39 //////////////////////////////////////////////////////////////////////////////// 42 // (Description) 40 // (Description) 43 // This is a primitive scorer class for scor 41 // This is a primitive scorer class for scoring Surface Flux. 44 // Current version assumes only for G4Box sha 42 // Current version assumes only for G4Box shape, and the surface 45 // is fixed on -Z plane. 43 // is fixed on -Z plane. 46 // 44 // 47 // Surface is defined at the -Z surface. 45 // Surface is defined at the -Z surface. 48 // Direction -Z +Z 46 // Direction -Z +Z 49 // 0 IN || OUT ->|<- | 47 // 0 IN || OUT ->|<- | 50 // 1 IN ->| | 48 // 1 IN ->| | 51 // 2 OUT |<- | 49 // 2 OUT |<- | 52 // 50 // 53 // Created: 2005-11-14 Tsukasa ASO, Akinori K 51 // Created: 2005-11-14 Tsukasa ASO, Akinori Kimura. 54 // << 52 // 55 // 18-Nov-2005 T.Aso, To use always positive 53 // 18-Nov-2005 T.Aso, To use always positive value for anglefactor. 56 // 29-Mar-2007 T.Aso, Bug fix for momentum d 54 // 29-Mar-2007 T.Aso, Bug fix for momentum direction at outgoing flux. 57 // 2010-07-22 Introduce Unit specification. 55 // 2010-07-22 Introduce Unit specification. 58 // 2010-07-22 Add weighted and divideByAre o 56 // 2010-07-22 Add weighted and divideByAre options 59 // 2020-10-06 Use G4VPrimitivePlotter and fi << 60 // vs. cell flux * track weight << 61 ////////////////////////////////////////////// 57 /////////////////////////////////////////////////////////////////////////////// 62 58 63 G4PSFlatSurfaceFlux::G4PSFlatSurfaceFlux(const << 59 G4PSFlatSurfaceFlux::G4PSFlatSurfaceFlux(G4String name, 64 G4int << 60 G4int direction, G4int depth) 65 : G4PSFlatSurfaceFlux(name, direction, "perc << 61 : G4VPrimitiveScorer(name,depth),HCID(-1),fDirection(direction),EvtMap(0), 66 {} << 62 weighted(true),divideByArea(true) 67 << 63 { 68 G4PSFlatSurfaceFlux::G4PSFlatSurfaceFlux(const << 64 DefineUnitAndCategory(); 69 const << 65 SetUnit("percm2"); 70 : G4VPrimitivePlotter(name, depth) << 66 } 71 , HCID(-1) << 67 72 , fDirection(direction) << 68 G4PSFlatSurfaceFlux::G4PSFlatSurfaceFlux(G4String name, 73 , EvtMap(nullptr) << 69 G4int direction, 74 , weighted(true) << 70 const G4String& unit, 75 , divideByArea(true) << 71 G4int depth) >> 72 : G4VPrimitiveScorer(name,depth),HCID(-1),fDirection(direction),EvtMap(0), >> 73 weighted(true),divideByArea(true) 76 { 74 { 77 DefineUnitAndCategory(); << 75 DefineUnitAndCategory(); 78 SetUnit(unit); << 76 SetUnit(unit); 79 } 77 } 80 78 81 G4bool G4PSFlatSurfaceFlux::ProcessHits(G4Step << 79 G4PSFlatSurfaceFlux::~G4PSFlatSurfaceFlux() >> 80 {;} >> 81 >> 82 G4bool G4PSFlatSurfaceFlux::ProcessHits(G4Step* aStep,G4TouchableHistory*) 82 { 83 { 83 G4StepPoint* preStep = aStep->Ge << 84 G4StepPoint* preStep = aStep->GetPreStepPoint(); 84 G4VPhysicalVolume* physVol = preStep-> << 85 G4VPhysicalVolume* physVol = preStep->GetPhysicalVolume(); 85 G4VPVParameterisation* physParam = physVol-> 86 G4VPVParameterisation* physParam = physVol->GetParameterisation(); 86 G4VSolid* solid = nullptr; << 87 G4VSolid * solid = 0; 87 if(physParam != nullptr) << 88 if(physParam) 88 { // for parameterized volume << 89 { // for parameterized volume 89 G4int idx = << 90 G4int idx = ((G4TouchableHistory*)(aStep->GetPreStepPoint()->GetTouchable())) 90 ((G4TouchableHistory*) (aStep->GetPreSte << 91 ->GetReplicaNumber(indexDepth); 91 ->GetReplicaNumber(indexDepth); << 92 solid = physParam->ComputeSolid(idx, physV 92 solid = physParam->ComputeSolid(idx, physVol); 93 solid->ComputeDimensions(physParam, idx, p << 93 solid->ComputeDimensions(physParam,idx,physVol); 94 } 94 } 95 else 95 else 96 { // for ordinary volume << 96 { // for ordinary volume 97 solid = physVol->GetLogicalVolume()->GetSo 97 solid = physVol->GetLogicalVolume()->GetSolid(); 98 } 98 } 99 99 100 auto boxSolid = (G4Box*) (solid); << 100 G4Box* boxSolid = (G4Box*)(solid); 101 101 102 G4int dirFlag = IsSelectedSurface(aStep, box << 102 G4int dirFlag =IsSelectedSurface(aStep,boxSolid); 103 if(dirFlag > 0) << 103 if ( dirFlag > 0 ) { 104 { << 104 if ( fDirection == fFlux_InOut || fDirection == dirFlag ){ 105 if(fDirection == fFlux_InOut || fDirection << 105 106 { << 106 G4StepPoint* thisStep=0; 107 G4StepPoint* thisStep = nullptr; << 107 if ( dirFlag == fFlux_In ){ 108 if(dirFlag == fFlux_In) << 108 thisStep = preStep; 109 { << 109 }else if ( dirFlag == fFlux_Out ){ 110 thisStep = preStep; << 110 thisStep = aStep->GetPostStepPoint(); 111 } << 111 }else{ 112 else if(dirFlag == fFlux_Out) << 112 return FALSE; 113 { << 114 thisStep = aStep->GetPostStepPoint(); << 115 } << 116 else << 117 { << 118 return false; << 119 } 113 } 120 114 121 G4TouchableHandle theTouchable = thisSte 115 G4TouchableHandle theTouchable = thisStep->GetTouchableHandle(); 122 G4ThreeVector pdirection = thisSte << 116 G4ThreeVector pdirection = thisStep->GetMomentumDirection(); 123 G4ThreeVector localdir = << 117 G4ThreeVector localdir = 124 theTouchable->GetHistory()->GetTopTran << 118 theTouchable->GetHistory()->GetTopTransform().TransformAxis(pdirection); 125 // 119 // 126 G4double angleFactor = localdir.z(); 120 G4double angleFactor = localdir.z(); 127 if(angleFactor < 0) << 121 if ( angleFactor < 0 ) angleFactor *= -1.; 128 angleFactor *= -1.; << 129 G4double flux = 1.0; 122 G4double flux = 1.0; 130 if(weighted) << 123 if ( weighted ) flux *=preStep->GetWeight(); // Current (Particle Weight) 131 flux *= preStep->GetWeight(); // Curr << 132 // 124 // 133 G4double square = << 125 G4double square = 4.*boxSolid->GetXHalfLength()*boxSolid->GetYHalfLength(); 134 4. * boxSolid->GetXHalfLength() * boxS << 135 // 126 // 136 flux = flux / angleFactor; // Flux with << 127 flux = flux/angleFactor; // Flux with angle. 137 if(divideByArea) << 128 if ( divideByArea ) flux /= square; 138 flux /= square; << 139 // 129 // 140 G4int index = GetIndex(aStep); 130 G4int index = GetIndex(aStep); 141 EvtMap->add(index, flux); << 131 EvtMap->add(index,flux); 142 << 143 if(!hitIDMap.empty() && hitIDMap.find(in << 144 { << 145 auto filler = G4VScoreHistFiller::Inst << 146 if(filler == nullptr) << 147 { << 148 G4Exception("G4PSFlatSurfaceFlux::Pr << 149 JustWarning, << 150 "G4TScoreHistFiller is n << 151 "not filled."); << 152 } << 153 else << 154 { << 155 filler->FillH1(hitIDMap[index], preS << 156 } << 157 } << 158 } 132 } 159 } 133 } 160 #ifdef debug 134 #ifdef debug 161 G4cout << " PASSED vol " << index << " trk " << 135 G4cout << " PASSED vol " 162 << fFlatSurfaceFlux << G4endl; << 136 << index << " trk "<<trkid<<" len " << fFlatSurfaceFlux<<G4endl; 163 #endif 137 #endif 164 138 165 return true; << 139 return TRUE; 166 } 140 } 167 141 168 G4int G4PSFlatSurfaceFlux::IsSelectedSurface(G << 142 G4int G4PSFlatSurfaceFlux::IsSelectedSurface(G4Step* aStep, G4Box* boxSolid){ 169 { << 143 170 G4TouchableHandle theTouchable = << 144 G4TouchableHandle theTouchable = 171 aStep->GetPreStepPoint()->GetTouchableHand 145 aStep->GetPreStepPoint()->GetTouchableHandle(); 172 G4double kCarTolerance = << 146 G4double kCarTolerance=G4GeometryTolerance::GetInstance()->GetSurfaceTolerance(); 173 G4GeometryTolerance::GetInstance()->GetSur << 174 147 175 if(aStep->GetPreStepPoint()->GetStepStatus() << 148 if (aStep->GetPreStepPoint()->GetStepStatus() == fGeomBoundary ){ 176 { << 177 // Entering Geometry 149 // Entering Geometry 178 G4ThreeVector stppos1 = aStep->GetPreStepP << 150 G4ThreeVector stppos1= aStep->GetPreStepPoint()->GetPosition(); 179 G4ThreeVector localpos1 = << 151 G4ThreeVector localpos1 = 180 theTouchable->GetHistory()->GetTopTransf 152 theTouchable->GetHistory()->GetTopTransform().TransformPoint(stppos1); 181 if(std::fabs(localpos1.z() + boxSolid->Get << 153 if(std::fabs( localpos1.z() + boxSolid->GetZHalfLength())<kCarTolerance ){ 182 { << 183 return fFlux_In; 154 return fFlux_In; 184 } 155 } 185 } 156 } 186 157 187 if(aStep->GetPostStepPoint()->GetStepStatus( << 158 if (aStep->GetPostStepPoint()->GetStepStatus() == fGeomBoundary ){ 188 { << 189 // Exiting Geometry 159 // Exiting Geometry 190 G4ThreeVector stppos2 = aStep->GetPostStep << 160 G4ThreeVector stppos2= aStep->GetPostStepPoint()->GetPosition(); 191 G4ThreeVector localpos2 = << 161 G4ThreeVector localpos2 = 192 theTouchable->GetHistory()->GetTopTransf 162 theTouchable->GetHistory()->GetTopTransform().TransformPoint(stppos2); 193 if(std::fabs(localpos2.z() + boxSolid->Get << 163 if(std::fabs( localpos2.z() + boxSolid->GetZHalfLength())<kCarTolerance ){ 194 { << 195 return fFlux_Out; 164 return fFlux_Out; 196 } 165 } 197 } 166 } 198 167 199 return -1; 168 return -1; 200 } 169 } 201 170 202 void G4PSFlatSurfaceFlux::Initialize(G4HCofThi 171 void G4PSFlatSurfaceFlux::Initialize(G4HCofThisEvent* HCE) 203 { 172 { 204 EvtMap = new G4THitsMap<G4double>(GetMultiFu 173 EvtMap = new G4THitsMap<G4double>(GetMultiFunctionalDetector()->GetName(), 205 GetName()) << 174 GetName()); 206 if(HCID < 0) << 175 if ( HCID < 0 ) HCID = GetCollectionID(0); 207 HCID = GetCollectionID(0); << 176 HCE->AddHitsCollection(HCID, (G4VHitsCollection*)EvtMap); 208 HCE->AddHitsCollection(HCID, (G4VHitsCollect << 177 } >> 178 >> 179 void G4PSFlatSurfaceFlux::EndOfEvent(G4HCofThisEvent*) >> 180 {;} >> 181 >> 182 void G4PSFlatSurfaceFlux::clear(){ >> 183 EvtMap->clear(); 209 } 184 } 210 185 211 void G4PSFlatSurfaceFlux::clear() { EvtMap->cl << 186 void G4PSFlatSurfaceFlux::DrawAll() >> 187 {;} 212 188 213 void G4PSFlatSurfaceFlux::PrintAll() 189 void G4PSFlatSurfaceFlux::PrintAll() 214 { 190 { 215 G4cout << " MultiFunctionalDet " << detecto 191 G4cout << " MultiFunctionalDet " << detector->GetName() << G4endl; 216 G4cout << " PrimitiveScorer" << GetName() << << 192 G4cout << " PrimitiveScorer" << GetName() <<G4endl; 217 G4cout << " Number of entries " << EvtMap->e 193 G4cout << " Number of entries " << EvtMap->entries() << G4endl; 218 for(const auto& [copy, flux] : *(EvtMap->Get << 194 std::map<G4int,G4double*>::iterator itr = EvtMap->GetMap()->begin(); 219 { << 195 for(; itr != EvtMap->GetMap()->end(); itr++) { 220 G4cout << " copy no.: " << copy << 196 G4cout << " copy no.: " << itr->first 221 << " flux : " << *(flux) / GetUni << 197 << " flux : " << *(itr->second)/GetUnitValue() 222 << GetUnit() << "]" << G4endl; << 198 << " [" << GetUnit() <<"]" >> 199 << G4endl; 223 } 200 } 224 } 201 } 225 202 226 void G4PSFlatSurfaceFlux::SetUnit(const G4Stri 203 void G4PSFlatSurfaceFlux::SetUnit(const G4String& unit) 227 { 204 { 228 if(divideByArea) << 205 if ( divideByArea ) { 229 { << 206 CheckAndSetUnit(unit,"Per Unit Surface"); 230 CheckAndSetUnit(unit, "Per Unit Surface"); << 207 } else { 231 } << 208 if (unit == "" ){ 232 else << 209 unitName = unit; 233 { << 210 unitValue = 1.0; 234 if(unit.empty()) << 211 }else{ 235 { << 212 G4String msg = "Invalid unit ["+unit+"] (Current unit is [" +GetUnit()+"] ) for " + GetName(); 236 unitName = unit; << 213 G4Exception("G4PSFlatSurfaceFlux::SetUnit","DetPS0008",JustWarning,msg); 237 unitValue = 1.0; << 214 } 238 } 215 } 239 else << 240 { << 241 G4String msg = "Invalid unit [" + unit + << 242 GetUnit() + "] ) for " + << 243 G4Exception("G4PSFlatSurfaceFlux::SetUni << 244 msg); << 245 } << 246 } << 247 } 216 } 248 217 249 void G4PSFlatSurfaceFlux::DefineUnitAndCategor << 218 void G4PSFlatSurfaceFlux::DefineUnitAndCategory(){ 250 { << 219 // Per Unit Surface 251 // Per Unit Surface << 220 new G4UnitDefinition("percentimeter2","percm2","Per Unit Surface",(1./cm2)); 252 new G4UnitDefinition("percentimeter2", "perc << 221 new G4UnitDefinition("permillimeter2","permm2","Per Unit Surface",(1./mm2)); 253 (1. / cm2)); << 222 new G4UnitDefinition("permeter2","perm2","Per Unit Surface",(1./m2)); 254 new G4UnitDefinition("permillimeter2", "perm << 255 (1. / mm2)); << 256 new G4UnitDefinition("permeter2", "perm2", " << 257 } 223 } >> 224 >> 225 258 226