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