| 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 // This example is provided by the Geant4-DNA 26 // This example is provided by the Geant4-DNA collaboration
27 // Any report or published results obtained us << 27 // Any report or published results obtained using the Geant4-DNA software
28 // shall cite the following Geant4-DNA collabo 28 // shall cite the following Geant4-DNA collaboration publication:
29 // Med. Phys. 37 (2010) 4692-4708 29 // Med. Phys. 37 (2010) 4692-4708
30 // The Geant4-DNA web site is available at htt 30 // The Geant4-DNA web site is available at http://geant4-dna.org
31 // 31 //
32 /// \file SplitProcess.cc 32 /// \file SplitProcess.cc
33 /// \brief Implementation of the variance redu 33 /// \brief Implementation of the variance reduction particle split class
34 34
35 //....oooOO0OOooo........oooOO0OOooo........oo 35 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
36 36
37 #include "SplitProcess.hh" 37 #include "SplitProcess.hh"
38 <<
39 #include "UserTrackInformation.hh" 38 #include "UserTrackInformation.hh"
40 39
>> 40 #include "G4Track.hh"
>> 41 #include "G4VParticleChange.hh"
41 #include "G4LogicalVolume.hh" 42 #include "G4LogicalVolume.hh"
>> 43 #include "G4TouchableHandle.hh"
42 #include "G4Region.hh" 44 #include "G4Region.hh"
43 #include "G4RegionStore.hh" 45 #include "G4RegionStore.hh"
44 #include "G4TouchableHandle.hh" <<
45 #include "G4Track.hh" <<
46 #include "G4VParticleChange.hh" <<
47 46
48 #include <vector> 47 #include <vector>
49 48
50 //....oooOO0OOooo........oooOO0OOooo........oo 49 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
51 50
52 SplitProcess::SplitProcess(G4String regName, G << 51 SplitProcess::SplitProcess(G4String regName, G4int nsplit)
>> 52 :fRegionName(regName), fNSplit(nsplit)
53 { 53 {
54 fRegion = G4RegionStore::GetInstance()->Find << 54 fRegion = G4RegionStore::GetInstance()->FindOrCreateRegion(fRegionName);
55 } 55 }
56 56
57 //....oooOO0OOooo........oooOO0OOooo........oo 57 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
58 58
59 SplitProcess::~SplitProcess() {} << 59 SplitProcess::~SplitProcess() {
>> 60 }
60 61
61 //....oooOO0OOooo........oooOO0OOooo........oo 62 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
62 63
63 G4VParticleChange* SplitProcess::PostStepDoIt( << 64 G4VParticleChange* SplitProcess::PostStepDoIt(const G4Track& track, const G4Step& step) {
64 { << 65 G4VParticleChange* particleChange(0);
65 G4VParticleChange* particleChange(0); << 66
66 << 67 if ( step.GetPreStepPoint()->GetTouchableHandle()->GetVolume()->GetLogicalVolume()
67 if (step.GetPreStepPoint()->GetTouchableHand << 68 ->GetRegion() != fRegion ) {
68 != fRegion) << 69 particleChange = pRegProcess->PostStepDoIt(track, step);
69 { << 70 assert ( 0 != particleChange);
70 particleChange = pRegProcess->PostStepDoIt << 71 return particleChange;
71 assert(0 != particleChange); << 72 }
72 return particleChange; << 73
73 } << 74 if ( fNSplit == 1 ) {
74 << 75 particleChange = pRegProcess->PostStepDoIt(track, step);
75 if (fNSplit == 1) { << 76 assert ( 0 != particleChange);
76 particleChange = pRegProcess->PostStepDoIt << 77 return particleChange;
77 assert(0 != particleChange); << 78 }
78 return particleChange; << 79
79 } << 80 UserTrackInformation* parentInformation =
80 << 81 (UserTrackInformation*)(step.GetTrack()->GetUserInformation());
81 UserTrackInformation* parentInformation = << 82 G4int initialSplitTrackID = parentInformation->GetSplitTrackID();
82 (UserTrackInformation*)(step.GetTrack()->G << 83
83 G4int initialSplitTrackID = parentInformatio << 84 if ( initialSplitTrackID > 1 ) {
84 << 85 particleChange = pRegProcess->PostStepDoIt(track, step);
85 if (initialSplitTrackID > 1) { << 86 assert ( 0 != particleChange);
86 particleChange = pRegProcess->PostStepDoIt << 87 return particleChange;
87 assert(0 != particleChange); << 88 }
>> 89
>> 90 G4double weight = track.GetWeight()/fNSplit;
>> 91 G4int splitTrackID = 3;
>> 92
>> 93 std::vector<G4Track*> secondaries;
>> 94 std::vector<G4int> vSplitTrack;
>> 95
>> 96 for ( int i = 0; i < fNSplit; i++ ) {
>> 97 particleChange = pRegProcess->PostStepDoIt(track, step);
>> 98 assert( 0 != particleChange);
>> 99 particleChange->SetVerboseLevel(0);
>> 100 G4Track* newTrack = new G4Track(*(particleChange->GetSecondary(0)));
>> 101
>> 102 secondaries.push_back( newTrack );
>> 103 vSplitTrack.push_back( splitTrackID );
>> 104
>> 105 splitTrackID++;
>> 106 }
>> 107
>> 108 parentInformation->SetSplitTrackID(2);
>> 109
>> 110 particleChange->SetNumberOfSecondaries(secondaries.size());
>> 111 particleChange->SetSecondaryWeightByProcess(true);
>> 112
>> 113 std::vector<G4Track*>::iterator iter = secondaries.begin();
>> 114 G4int i = 0;
>> 115 while( iter != secondaries.end() ) {
>> 116 G4Track* newTrack = *iter;
>> 117 newTrack->SetWeight(weight);
>> 118
>> 119 UserTrackInformation* secondaryInformation = new UserTrackInformation();
>> 120 secondaryInformation->SetSplitTrackID(vSplitTrack[i]);
>> 121 newTrack->SetUserInformation(secondaryInformation);
>> 122
>> 123 particleChange->AddSecondary(newTrack);
>> 124
>> 125 iter++;
>> 126 i++;
>> 127 }
>> 128
88 return particleChange; 129 return particleChange;
89 } <<
90 <<
91 G4double weight = track.GetWeight() / fNSpli <<
92 G4int splitTrackID = 3; <<
93 <<
94 std::vector<G4Track*> secondaries; <<
95 std::vector<G4int> vSplitTrack; <<
96 <<
97 for (int i = 0; i < fNSplit; i++) { <<
98 particleChange = pRegProcess->PostStepDoIt <<
99 assert(0 != particleChange); <<
100 particleChange->SetVerboseLevel(0); <<
101 G4Track* newTrack = new G4Track(*(particle <<
102 <<
103 secondaries.push_back(newTrack); <<
104 vSplitTrack.push_back(splitTrackID); <<
105 <<
106 splitTrackID++; <<
107 } <<
108 <<
109 parentInformation->SetSplitTrackID(2); <<
110 <<
111 particleChange->SetNumberOfSecondaries(secon <<
112 particleChange->SetSecondaryWeightByProcess( <<
113 <<
114 std::vector<G4Track*>::iterator iter = secon <<
115 G4int i = 0; <<
116 while (iter != secondaries.end()) { <<
117 G4Track* newTrack = *iter; <<
118 newTrack->SetWeight(weight); <<
119 <<
120 UserTrackInformation* secondaryInformation <<
121 secondaryInformation->SetSplitTrackID(vSpl <<
122 newTrack->SetUserInformation(secondaryInfo <<
123 <<
124 particleChange->AddSecondary(newTrack); <<
125 <<
126 iter++; <<
127 i++; <<
128 } <<
129 <<
130 return particleChange; <<
131 } 130 }
132 131
133 //....oooOO0OOooo........oooOO0OOooo........oo 132 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
134 133