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