| 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 // 26 //
27 /// \file eventgenerator/pythia/pythia8decayer 27 /// \file eventgenerator/pythia/pythia8decayer/src/SingleParticleGun.cc
28 /// \brief Implementation of the SingleParticl 28 /// \brief Implementation of the SingleParticleGun class
29 /// 29 ///
30 /// \author J. Yarba; FNAL 30 /// \author J. Yarba; FNAL
31 31
32 #include "SingleParticleGun.hh" 32 #include "SingleParticleGun.hh"
33 33
34 #include "G4Event.hh" 34 #include "G4Event.hh"
35 #include "G4ParticleDefinition.hh" <<
36 #include "G4ParticleGun.hh" 35 #include "G4ParticleGun.hh"
37 #include "G4ParticleTable.hh" 36 #include "G4ParticleTable.hh"
>> 37 #include "G4ParticleDefinition.hh"
38 #include "G4ThreeVector.hh" 38 #include "G4ThreeVector.hh"
39 #include "Randomize.hh" 39 #include "Randomize.hh"
40 40
41 // SPECIAL CASE - needed for treatment of pola 41 // SPECIAL CASE - needed for treatment of polarization for tau's
42 #include "G4TauMinus.hh" 42 #include "G4TauMinus.hh"
43 #include "G4TauPlus.hh" 43 #include "G4TauPlus.hh"
44 44
45 //....oooOO0OOooo........oooOO0OOooo........oo 45 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
46 46
47 SingleParticleGun::SingleParticleGun(const G4S << 47 SingleParticleGun::SingleParticleGun( const G4String& pname, const double pmom )
48 : G4VUserPrimaryGeneratorAction(), fGun(null << 48 : G4VUserPrimaryGeneratorAction(),
>> 49 fGun(nullptr),
>> 50 fMomentum(pmom)
49 { 51 {
50 int nparts = 1; <<
51 fGun = new G4ParticleGun(nparts); <<
52 52
53 G4ParticleTable* pdt = G4ParticleTable::GetP << 53 int nparts = 1;
54 G4ParticleDefinition* pd = pdt->FindParticle << 54 fGun = new G4ParticleGun( nparts );
55 fGun->SetParticleDefinition(pd); << 55
56 fGun->SetParticlePosition(G4ThreeVector(0., << 56 G4ParticleTable* pdt = G4ParticleTable::GetParticleTable();
57 double mass = pd->GetPDGMass(); << 57 G4ParticleDefinition* pd = pdt->FindParticle( pname );
58 double energy = std::sqrt(fMomentum * fMomen << 58 fGun->SetParticleDefinition( pd );
59 fGun->SetParticleEnergy(energy); << 59 fGun->SetParticlePosition( G4ThreeVector(0.,0.,0.) );
>> 60 double mass = pd->GetPDGMass();
>> 61 double energy = std::sqrt( fMomentum*fMomentum + mass*mass );
>> 62 fGun->SetParticleEnergy( energy );
>> 63
>> 64 // NOTE: do not set momentum directions
>> 65 // as it'll be generated event-by-event
60 66
61 // NOTE: do not set momentum directions <<
62 // as it'll be generated event-by-even <<
63 } 67 }
64 68
65 //....oooOO0OOooo........oooOO0OOooo........oo 69 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
66 70
67 SingleParticleGun::~SingleParticleGun() 71 SingleParticleGun::~SingleParticleGun()
68 { 72 {
69 delete fGun; << 73
>> 74 delete fGun;
>> 75
70 } 76 }
71 77
72 //....oooOO0OOooo........oooOO0OOooo........oo 78 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
73 79
74 void SingleParticleGun::GeneratePrimaries(G4Ev << 80 void SingleParticleGun::GeneratePrimaries( G4Event* evt )
75 { 81 {
76 if (!fGun) { <<
77 G4cout << " ParticleGun is NOT set up; BAI <<
78 G4cout << " Check if you are using ctor " <<
79 << "SinglePartGun(const G4string& p <<
80 return; <<
81 } <<
82 <<
83 double phi = -1. * CLHEP::pi + CLHEP::twopi <<
84 double theta = CLHEP::pi / 4. + CLHEP::halfp <<
85 <<
86 double x = std::sin(theta) * std::cos(phi); <<
87 double y = std::sin(theta) * std::sin(phi); <<
88 double z = std::cos(theta); <<
89 <<
90 fGun->SetParticleMomentumDirection(G4ThreeVe <<
91 <<
92 // SPECIAL CASE: for particle such as tau po <<
93 // while for tau_bar it should <<
94 // <<
95 // NOTE: one could probably compare by name <<
96 // <<
97 if (fGun->GetParticleDefinition() == G4TauMi <<
98 fGun->SetParticlePolarization(G4ThreeVecto <<
99 } <<
100 else if (fGun->GetParticleDefinition() == G4 <<
101 fGun->SetParticlePolarization(G4ThreeVecto <<
102 } <<
103 82
104 fGun->GeneratePrimaryVertex(evt); << 83 if ( !fGun )
>> 84 {
>> 85 G4cout << " ParticleGun is NOT set up; BAIL OUT from this event " << G4endl;
>> 86 G4cout << " Check if you are using ctor "
>> 87 << "SinglePartGun(const G4string& partname, const double pmom ) "
>> 88 << G4endl;
>> 89 return;
>> 90 }
>> 91
>> 92 double phi = -1. * CLHEP::pi + CLHEP::twopi * G4UniformRand();
>> 93 double theta = CLHEP::pi/4. + CLHEP::halfpi * G4UniformRand();
>> 94
>> 95 double x = std::sin(theta) * std::cos(phi);
>> 96 double y = std::sin(theta) * std::sin(phi);
>> 97 double z = std::cos(theta);
>> 98
>> 99 fGun->SetParticleMomentumDirection( G4ThreeVector(x,y,z) );
>> 100
>> 101 // SPECIAL CASE: for particle such as tau polarization should be -1 * p3
>> 102 // while for tau_bar it should be +1
>> 103 //
>> 104 // NOTE: one could probably compare by name but that'd mean comparing strings...
>> 105 //
>> 106 if ( fGun->GetParticleDefinition() == G4TauMinus::TauMinus() )
>> 107 {
>> 108 fGun->SetParticlePolarization( G4ThreeVector(-1.0*x,-1.0*y,-1.0*z) );
>> 109 }
>> 110 else if ( fGun->GetParticleDefinition() == G4TauPlus::TauPlus() )
>> 111 {
>> 112 fGun->SetParticlePolarization( G4ThreeVector(x,y,z) );
>> 113 }
>> 114
>> 115 fGun->GeneratePrimaryVertex(evt);
>> 116
>> 117 return;
105 118
106 return; <<
107 } 119 }
108 120
109 //....oooOO0OOooo........oooOO0OOooo........oo 121 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
110 122