| Geant4 Cross Reference |
1 // 1
2 // *******************************************
3 // * License and Disclaimer
4 // *
5 // * The Geant4 software is copyright of th
6 // * the Geant4 Collaboration. It is provided
7 // * conditions of the Geant4 Software License
8 // * LICENSE and available at http://cern.ch/
9 // * include a list of copyright holders.
10 // *
11 // * Neither the authors of this software syst
12 // * institutes,nor the agencies providing fin
13 // * work make any representation or warran
14 // * regarding this software system or assum
15 // * use. Please see the license in the file
16 // * for the full disclaimer and the limitatio
17 // *
18 // * This code implementation is the result
19 // * technical work of the GEANT4 collaboratio
20 // * By using, copying, modifying or distri
21 // * any work based on the software) you ag
22 // * use in resulting scientific publicati
23 // * acceptance of all terms of the Geant4 Sof
24 // *******************************************
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
33 #include <cmath>
34
35 #include "G4SingleParticleSource.hh"
36
37 #include "G4SystemOfUnits.hh"
38 #include "G4PrimaryParticle.hh"
39 #include "G4Event.hh"
40 #include "Randomize.hh"
41 #include "G4ParticleTable.hh"
42 #include "G4Geantino.hh"
43 #include "G4ParticleDefinition.hh"
44 #include "G4IonTable.hh"
45 #include "G4Ions.hh"
46 #include "G4TrackingManager.hh"
47 #include "G4Track.hh"
48 #include "G4AutoLock.hh"
49
50 G4SingleParticleSource::part_prop_t::part_prop
51 {
52 momentum_direction = G4ParticleMomentum(1,0,
53 energy = 1.*MeV;
54 position = G4ThreeVector();
55 }
56
57 G4SingleParticleSource::G4SingleParticleSource
58 {
59 // Initialise all variables
60 // Position distribution Variables
61
62 NumberOfParticlesToBeGenerated = 1;
63 definition = G4Geantino::GeantinoDefinition(
64
65 charge = 0.0;
66 time = 0;
67 polarization = G4ThreeVector();
68
69 biasRndm = new G4SPSRandomGenerator();
70 posGenerator = new G4SPSPosDistribution();
71 posGenerator->SetBiasRndm(biasRndm);
72 angGenerator = new G4SPSAngDistribution();
73 angGenerator->SetPosDistribution(posGenerato
74 angGenerator->SetBiasRndm(biasRndm);
75 eneGenerator = new G4SPSEneDistribution();
76 eneGenerator->SetBiasRndm(biasRndm);
77
78 verbosityLevel = 0;
79
80 G4MUTEXINIT(mutex);
81 }
82
83 G4SingleParticleSource::~G4SingleParticleSourc
84 {
85 delete biasRndm;
86 delete posGenerator;
87 delete angGenerator;
88 delete eneGenerator;
89
90 G4MUTEXDESTROY(mutex);
91 }
92
93 void G4SingleParticleSource::SetVerbosity(G4in
94 {
95 G4AutoLock l(&mutex);
96 verbosityLevel = vL;
97 posGenerator->SetVerbosity(vL);
98 angGenerator->SetVerbosity(vL);
99 eneGenerator->SetVerbosity(vL);
100 }
101
102 void G4SingleParticleSource::
103 SetParticleDefinition(G4ParticleDefinition* aP
104 {
105 definition = aParticleDefinition;
106 charge = aParticleDefinition->GetPDGCharge()
107 }
108
109 void G4SingleParticleSource::GeneratePrimaryVe
110 {
111 if (definition == nullptr)
112 {
113 // TODO: Should this rise an exception???
114 return;
115 }
116
117 if (verbosityLevel > 1)
118 {
119 G4cout << " NumberOfParticlesToBeGenerated
120 << NumberOfParticlesToBeGenerated <
121 }
122
123 part_prop_t& pp = ParticleProperties.Get();
124
125 // Position stuff
126 pp.position = posGenerator->GenerateOne();
127
128 // Create a new vertex
129 auto* vertex = new G4PrimaryVertex(pp.positi
130
131 for (G4int i=0; i<NumberOfParticlesToBeGener
132 {
133 // Angular stuff
134 pp.momentum_direction = angGenerator->Gene
135
136 // Energy stuff
137 pp.energy = eneGenerator->GenerateOne(defi
138
139 if (verbosityLevel >= 2)
140 {
141 G4cout << "Creating primaries and assign
142 }
143
144 // Create new primaries and set them to th
145 //
146 G4double mass = definition->GetPDGMass();
147 auto* particle = new G4PrimaryParticle(def
148 particle->SetKineticEnergy(pp.energy );
149 particle->SetMass( mass );
150 particle->SetMomentumDirection( pp.momentu
151 particle->SetCharge( charge );
152 particle->SetPolarization(polarization.x()
153 polarization.y()
154 polarization.z()
155 if (verbosityLevel > 1)
156 {
157 G4cout << "Particle name: " << definitio
158 G4cout << " Energy: " << pp.energy
159 G4cout << " Position: " << pp.positi
160 G4cout << " Direction: " << pp.moment
161 }
162
163 // Set bweight equal to the multiple of al
164 //
165 G4double weight = eneGenerator->GetWeight(
166
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