| 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 // G4SingleParticleSource source implementatio <<
27 // <<
28 // Author: Fan Lei, QinetiQ ltd. - 05/02/2004 <<
29 // Customer: ESA/ESTEC <<
30 // Revision: Andrea Dotti, SLAC <<
31 // ------------------------------------------- <<
32 26
>> 27 ///////////////////////////////////////////////////////////////////////////////
>> 28 //
>> 29 // MODULE: G4SingleParticleSource.hh
>> 30 //
>> 31 // Version: 1.0
>> 32 // Date: 5/02/04
>> 33 // Author: Fan Lei
>> 34 // Organisation: QinetiQ ltd.
>> 35 // Customer: ESA/ESTEC
>> 36 //
>> 37 ///////////////////////////////////////////////////////////////////////////////
>> 38 //
>> 39 // CHANGE HISTORY
>> 40 // --------------
>> 41 //
>> 42 // Version 1.0, 05/02/2004, Fan Lei, Created.
>> 43 // Based on the G4GeneralParticleSource class in Geant4 v6.0
>> 44 //
>> 45 ///////////////////////////////////////////////////////////////////////////////
>> 46 //
33 #include <cmath> 47 #include <cmath>
34 48
35 #include "G4SingleParticleSource.hh" 49 #include "G4SingleParticleSource.hh"
36 50
37 #include "G4SystemOfUnits.hh" 51 #include "G4SystemOfUnits.hh"
38 #include "G4PrimaryParticle.hh" 52 #include "G4PrimaryParticle.hh"
39 #include "G4Event.hh" 53 #include "G4Event.hh"
40 #include "Randomize.hh" 54 #include "Randomize.hh"
41 #include "G4ParticleTable.hh" 55 #include "G4ParticleTable.hh"
42 #include "G4Geantino.hh" 56 #include "G4Geantino.hh"
43 #include "G4ParticleDefinition.hh" 57 #include "G4ParticleDefinition.hh"
44 #include "G4IonTable.hh" 58 #include "G4IonTable.hh"
45 #include "G4Ions.hh" 59 #include "G4Ions.hh"
46 #include "G4TrackingManager.hh" 60 #include "G4TrackingManager.hh"
47 #include "G4Track.hh" 61 #include "G4Track.hh"
48 #include "G4AutoLock.hh" 62 #include "G4AutoLock.hh"
49 63
50 G4SingleParticleSource::part_prop_t::part_prop << 64 G4SingleParticleSource::part_prop_t::part_prop_t() {
51 { << 65 //definition = G4Geantino::GeantinoDefinition();
52 momentum_direction = G4ParticleMomentum(1,0, 66 momentum_direction = G4ParticleMomentum(1,0,0);
53 energy = 1.*MeV; 67 energy = 1.*MeV;
54 position = G4ThreeVector(); 68 position = G4ThreeVector();
55 } 69 }
56 70
57 G4SingleParticleSource::G4SingleParticleSource << 71 G4SingleParticleSource::G4SingleParticleSource() {
58 { << 72 // // Initialise all variables
59 // Initialise all variables << 73 // // Position distribution Variables
60 // Position distribution Variables << 74 //
61 << 75 NumberOfParticlesToBeGenerated = 1;
62 NumberOfParticlesToBeGenerated = 1; << 76 definition = G4Geantino::GeantinoDefinition();
63 definition = G4Geantino::GeantinoDefinition( << 77 // G4ThreeVector zero;
64 << 78 // particle_momentum_direction = G4ParticleMomentum(1, 0, 0);
65 charge = 0.0; << 79 // particle_energy = 1.0 * MeV;
66 time = 0; << 80 // particle_position = zero;
67 polarization = G4ThreeVector(); << 81 // particle_time = 0.0;
68 << 82 // particle_polarization = zero;
69 biasRndm = new G4SPSRandomGenerator(); << 83 charge = 0.0;
70 posGenerator = new G4SPSPosDistribution(); << 84 time = 0;
71 posGenerator->SetBiasRndm(biasRndm); << 85 polarization = G4ThreeVector();
72 angGenerator = new G4SPSAngDistribution(); << 86
73 angGenerator->SetPosDistribution(posGenerato << 87 biasRndm = new G4SPSRandomGenerator();
74 angGenerator->SetBiasRndm(biasRndm); << 88 posGenerator = new G4SPSPosDistribution();
75 eneGenerator = new G4SPSEneDistribution(); << 89 posGenerator->SetBiasRndm(biasRndm);
76 eneGenerator->SetBiasRndm(biasRndm); << 90 angGenerator = new G4SPSAngDistribution();
>> 91 angGenerator->SetPosDistribution(posGenerator);
>> 92 angGenerator->SetBiasRndm(biasRndm);
>> 93 eneGenerator = new G4SPSEneDistribution();
>> 94 eneGenerator->SetBiasRndm(biasRndm);
77 95
78 verbosityLevel = 0; << 96 // verbosity
>> 97 verbosityLevel = 0;
79 98
80 G4MUTEXINIT(mutex); << 99 G4MUTEXINIT(mutex);
>> 100
81 } 101 }
82 102
83 G4SingleParticleSource::~G4SingleParticleSourc << 103 G4SingleParticleSource::~G4SingleParticleSource() {
84 { << 104 delete biasRndm;
85 delete biasRndm; << 105 delete posGenerator;
86 delete posGenerator; << 106 delete angGenerator;
87 delete angGenerator; << 107 delete eneGenerator;
88 delete eneGenerator; << 108 G4MUTEXDESTROY(mutex);
89 << 109 }
90 G4MUTEXDESTROY(mutex); << 110
91 } << 111 void G4SingleParticleSource::SetVerbosity(int vL) {
92 << 112 G4AutoLock l(&mutex);
93 void G4SingleParticleSource::SetVerbosity(G4in << 113 verbosityLevel = vL;
94 { << 114 posGenerator->SetVerbosity(vL);
95 G4AutoLock l(&mutex); << 115 angGenerator->SetVerbosity(vL);
96 verbosityLevel = vL; << 116 eneGenerator->SetVerbosity(vL);
97 posGenerator->SetVerbosity(vL); << 117 //G4cout << "Verbosity Set to: " << verbosityLevel << G4endl;
98 angGenerator->SetVerbosity(vL); << 118 }
99 eneGenerator->SetVerbosity(vL); << 119
100 } << 120 void G4SingleParticleSource::SetParticleDefinition(
101 << 121 G4ParticleDefinition* aParticleDefinition) {
102 void G4SingleParticleSource:: << 122 definition = aParticleDefinition;
103 SetParticleDefinition(G4ParticleDefinition* aP << 123 charge = aParticleDefinition->GetPDGCharge();
104 { << 124 }
105 definition = aParticleDefinition; << 125
106 charge = aParticleDefinition->GetPDGCharge() << 126 void G4SingleParticleSource::GeneratePrimaryVertex(G4Event *evt) {
107 } << 127
108 << 128 //G4AutoLock l(&mutex);
109 void G4SingleParticleSource::GeneratePrimaryVe << 129 //part_prop_t& pp = ParticleProperties.Get();
110 { << 130 if (definition == NULL) {
111 if (definition == nullptr) << 131 //TODO: Should this rise an exception???
112 { << 132 return ;
113 // TODO: Should this rise an exception??? << 133 }
114 return; << 134 //return;
115 } << 135
116 << 136 if (verbosityLevel > 1)
117 if (verbosityLevel > 1) << 137 G4cout << " NumberOfParticlesToBeGenerated: "
118 { << 138 <<NumberOfParticlesToBeGenerated << G4endl;
119 G4cout << " NumberOfParticlesToBeGenerated << 139
120 << NumberOfParticlesToBeGenerated < << 140 part_prop_t& pp = ParticleProperties.Get();
121 } << 141 // Position stuff
122 << 142 pp.position = posGenerator->GenerateOne();
123 part_prop_t& pp = ParticleProperties.Get(); << 143
124 << 144 // create a new vertex
125 // Position stuff << 145 G4PrimaryVertex* vertex = new G4PrimaryVertex(pp.position,time);
126 pp.position = posGenerator->GenerateOne(); << 146
127 << 147 for (G4int i = 0; i < NumberOfParticlesToBeGenerated; i++) {
128 // Create a new vertex << 148 // Angular stuff
129 auto* vertex = new G4PrimaryVertex(pp.positi << 149 pp.momentum_direction = angGenerator->GenerateOne();
130 << 150 // Energy stuff
131 for (G4int i=0; i<NumberOfParticlesToBeGener << 151 pp.energy = eneGenerator->GenerateOne(definition);
132 { << 152
133 // Angular stuff << 153 if (verbosityLevel >= 2)
134 pp.momentum_direction = angGenerator->Gene << 154 G4cout << "Creating primaries and assigning to vertex" << G4endl;
135 << 155 // create new primaries and set them to the vertex
136 // Energy stuff << 156 G4double mass = definition->GetPDGMass();
137 pp.energy = eneGenerator->GenerateOne(defi << 157 G4PrimaryParticle* particle =
138 << 158 new G4PrimaryParticle(definition);
139 if (verbosityLevel >= 2) << 159 particle->SetKineticEnergy(pp.energy );
140 { << 160 particle->SetMass( mass );
141 G4cout << "Creating primaries and assign << 161 particle->SetMomentumDirection( pp.momentum_direction );
142 } << 162 particle->SetCharge( charge );
143 << 163 particle->SetPolarization(polarization.x(),
144 // Create new primaries and set them to th << 164 polarization.y(),
145 // << 165 polarization.z());
146 G4double mass = definition->GetPDGMass(); << 166 if (verbosityLevel > 1) {
147 auto* particle = new G4PrimaryParticle(def << 167 G4cout << "Particle name: "
148 particle->SetKineticEnergy(pp.energy ); << 168 << definition->GetParticleName() << G4endl;
149 particle->SetMass( mass ); << 169 G4cout << " Energy: " << pp.energy << G4endl;
150 particle->SetMomentumDirection( pp.momentu << 170 G4cout << " Position: " << pp.position << G4endl;
151 particle->SetCharge( charge ); << 171 G4cout << " Direction: " << pp.momentum_direction
152 particle->SetPolarization(polarization.x() << 172 << G4endl;
153 polarization.y() << 173 }
154 polarization.z() << 174 // Set bweight equal to the multiple of all non-zero weights
155 if (verbosityLevel > 1) << 175 G4double weight = eneGenerator->GetWeight()*biasRndm->GetBiasWeight();
156 { << 176 // pass it to primary particle
157 G4cout << "Particle name: " << definitio << 177 particle->SetWeight(weight);
158 G4cout << " Energy: " << pp.energy << 178
159 G4cout << " Position: " << pp.positi << 179 vertex->SetPrimary(particle);
160 G4cout << " Direction: " << pp.moment << 180
161 } << 181 }
162 << 182 // now pass the weight to the primary vertex. CANNOT be used here!
163 // Set bweight equal to the multiple of al << 183 // vertex->SetWeight(particle_weight);
164 // << 184 evt->AddPrimaryVertex(vertex);
165 G4double weight = eneGenerator->GetWeight( << 185 if (verbosityLevel > 1)
166 << 186 G4cout << " Primary Vetex generated !" << G4endl;
167 if(eneGenerator->IfApplyEnergyWeight()) <<
168 { <<
169 weight *= eneGenerator->GetArbEneWeight( <<
170 } <<
171 <<
172 // Pass it to primary particle <<
173 // <<
174 particle->SetWeight(weight); <<
175 vertex->SetPrimary(particle); <<
176 } <<
177 <<
178 // Now pass the weight to the primary vertex <<
179 // vertex->SetWeight(particle_weight); <<
180 evt->AddPrimaryVertex(vertex); <<
181 <<
182 if (verbosityLevel > 1) <<
183 { <<
184 G4cout << " Primary Vetex generated !" << <<
185 } <<
186 } 187 }
>> 188
187 189