| Geant4 Cross Reference |
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 // Class G4PSVolumeFlux
27 //
28 // Class description:
29 // Scorer that scores number of tracks coming into the associated volume.
30 // Optionally number of tracks can be divided by the surface area to
31 // score the volume current and divided by cos(theta) for volume flux
32 // where theta is the incident angle
33 //
34 // - Created M. Asai, Sept. 2020
35 //
36 //
37 #include "G4PSVolumeFlux.hh"
38
39 #include "G4SystemOfUnits.hh"
40 #include "G4StepStatus.hh"
41 #include "G4Track.hh"
42 #include "G4VSolid.hh"
43 #include "G4VPhysicalVolume.hh"
44 #include "G4VPVParameterisation.hh"
45 #include "G4UnitsTable.hh"
46 #include "G4GeometryTolerance.hh"
47 #include "G4VScoreHistFiller.hh"
48
49 G4PSVolumeFlux::G4PSVolumeFlux(const G4String& name, G4int direction, G4int depth)
50 : G4VPrimitivePlotter(name, depth)
51 , fDirection(direction)
52 {}
53
54 G4bool G4PSVolumeFlux::ProcessHits(G4Step* aStep, G4TouchableHistory*)
55 {
56 G4StepPoint* preStepPoint = aStep->GetPreStepPoint();
57 G4StepPoint* postStepPoint = aStep->GetPostStepPoint();
58 G4StepPoint* thisStepPoint = nullptr;
59 if(fDirection == 1) // Score only the inward particle
60 {
61 if(preStepPoint->GetStepStatus() == fGeomBoundary)
62 {
63 thisStepPoint = preStepPoint;
64 }
65 else
66 {
67 return false;
68 }
69 }
70 else if(fDirection == 2) // Score only the outward particle
71 {
72 if(postStepPoint->GetStepStatus() == fGeomBoundary)
73 {
74 thisStepPoint = postStepPoint;
75 }
76 else
77 {
78 return false;
79 }
80 }
81
82 G4double flux = preStepPoint->GetWeight();
83
84 if(divare || divcos)
85 {
86 G4VPhysicalVolume* physVol = preStepPoint->GetPhysicalVolume();
87 G4VPVParameterisation* physParam = physVol->GetParameterisation();
88 G4VSolid* solid = nullptr;
89 if(physParam != nullptr)
90 { // for parameterized volume
91 auto idx = ((G4TouchableHistory*) (preStepPoint->GetTouchable()))
92 ->GetReplicaNumber(indexDepth);
93 solid = physParam->ComputeSolid(idx, physVol);
94 solid->ComputeDimensions(physParam, idx, physVol);
95 }
96 else
97 { // for ordinary volume
98 solid = physVol->GetLogicalVolume()->GetSolid();
99 }
100
101 if(divare)
102 {
103 flux /= solid->GetSurfaceArea();
104 }
105
106 if(divcos)
107 {
108 G4TouchableHandle theTouchable = thisStepPoint->GetTouchableHandle();
109 G4ThreeVector pdirection = thisStepPoint->GetMomentumDirection();
110 G4ThreeVector localdir =
111 theTouchable->GetHistory()->GetTopTransform().TransformAxis(pdirection);
112 G4ThreeVector globalPos = thisStepPoint->GetPosition();
113 G4ThreeVector localPos =
114 theTouchable->GetHistory()->GetTopTransform().TransformPoint(globalPos);
115 G4ThreeVector surfNormal = solid->SurfaceNormal(localPos);
116 G4double cosT = surfNormal.cosTheta(localdir);
117 if(cosT != 0.)
118 flux /= std::abs(cosT);
119 }
120 }
121
122 G4int index = GetIndex(aStep);
123 EvtMap->add(index, flux);
124
125 if(!hitIDMap.empty() && hitIDMap.find(index) != hitIDMap.cend())
126 {
127 auto filler = G4VScoreHistFiller::Instance();
128 if(filler == nullptr)
129 {
130 G4Exception(
131 "G4PSVolumeFlux::ProcessHits", "SCORER0123", JustWarning,
132 "G4TScoreHistFiller is not instantiated!! Histogram is not filled.");
133 }
134 else
135 {
136 filler->FillH1(hitIDMap[index], thisStepPoint->GetKineticEnergy(), flux);
137 }
138 }
139
140 return true;
141 }
142
143 void G4PSVolumeFlux::Initialize(G4HCofThisEvent* HCE)
144 {
145 if(HCID < 0)
146 HCID = GetCollectionID(0);
147 EvtMap = new G4THitsMap<G4double>(GetMultiFunctionalDetector()->GetName(),
148 GetName());
149 HCE->AddHitsCollection(HCID, (G4VHitsCollection*) EvtMap);
150 }
151
152 void G4PSVolumeFlux::clear() { EvtMap->clear(); }
153
154 void G4PSVolumeFlux::PrintAll()
155 {
156 G4cout << " MultiFunctionalDet " << detector->GetName() << G4endl;
157 G4cout << " PrimitiveScorer" << GetName() << G4endl;
158 G4cout << " Number of entries " << EvtMap->entries() << G4endl;
159 for(const auto& [copy, flux] : *(EvtMap->GetMap()))
160 {
161 G4cout << " copy no.: " << copy << " flux : " << *(flux)
162 << G4endl;
163 }
164 }
165