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