| 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 //
27 /// \file B5/src/PrimaryGeneratorAction.cc
28 /// \brief Implementation of the B5::PrimaryGe
29
30 #include "PrimaryGeneratorAction.hh"
31
32 #include "G4GenericMessenger.hh"
33 #include "G4ParticleDefinition.hh"
34 #include "G4ParticleGun.hh"
35 #include "G4ParticleTable.hh"
36 #include "G4SystemOfUnits.hh"
37 #include "G4ThreeVector.hh"
38 #include "Randomize.hh"
39
40 namespace B5
41 {
42
43 //....oooOO0OOooo........oooOO0OOooo........oo
44
45 PrimaryGeneratorAction::PrimaryGeneratorAction
46 {
47 G4int nofParticles = 1;
48 fParticleGun = new G4ParticleGun(nofParticle
49
50 auto particleTable = G4ParticleTable::GetPar
51 fPositron = particleTable->FindParticle("e+"
52 fMuon = particleTable->FindParticle("mu+");
53 fPion = particleTable->FindParticle("pi+");
54 fKaon = particleTable->FindParticle("kaon+")
55 fProton = particleTable->FindParticle("proto
56
57 // default particle kinematics
58 fParticleGun->SetParticlePosition(G4ThreeVec
59 fParticleGun->SetParticleDefinition(fPositro
60
61 // define commands for this class
62 DefineCommands();
63 }
64
65 //....oooOO0OOooo........oooOO0OOooo........oo
66
67 PrimaryGeneratorAction::~PrimaryGeneratorActio
68 {
69 delete fParticleGun;
70 delete fMessenger;
71 }
72
73 //....oooOO0OOooo........oooOO0OOooo........oo
74
75 void PrimaryGeneratorAction::GeneratePrimaries
76 {
77 G4ParticleDefinition* particle;
78 if (fRandomizePrimary) {
79 auto i = (int)(5. * G4UniformRand());
80 switch (i) {
81 case 0:
82 particle = fPositron;
83 break;
84 case 1:
85 particle = fMuon;
86 break;
87 case 2:
88 particle = fPion;
89 break;
90 case 3:
91 particle = fKaon;
92 break;
93 default:
94 particle = fProton;
95 break;
96 }
97 fParticleGun->SetParticleDefinition(partic
98 }
99 else {
100 particle = fParticleGun->GetParticleDefini
101 }
102
103 auto pp = fMomentum + (G4UniformRand() - 0.5
104 auto mass = particle->GetPDGMass();
105 auto ekin = std::sqrt(pp * pp + mass * mass)
106 fParticleGun->SetParticleEnergy(ekin);
107
108 auto angle = (G4UniformRand() - 0.5) * fSigm
109 fParticleGun->SetParticleMomentumDirection(G
110
111 fParticleGun->GeneratePrimaryVertex(event);
112 }
113
114 //....oooOO0OOooo........oooOO0OOooo........oo
115
116 void PrimaryGeneratorAction::DefineCommands()
117 {
118 // Define /B5/generator command directory us
119 fMessenger = new G4GenericMessenger(this, "/
120
121 // momentum command
122 auto& momentumCmd = fMessenger->DeclarePrope
123
124 momentumCmd.SetParameterName("p", true);
125 momentumCmd.SetRange("p>=0.");
126 momentumCmd.SetDefaultValue("1.");
127 // ok
128 // momentumCmd.SetParameterName("p", true);
129 // momentumCmd.SetRange("p>=0.");
130
131 // sigmaMomentum command
132 auto& sigmaMomentumCmd = fMessenger->Declare
133 "sigmaMomentum", "MeV", fSigmaMomentum, "S
134 sigmaMomentumCmd.SetParameterName("sp", true
135 sigmaMomentumCmd.SetRange("sp>=0.");
136 sigmaMomentumCmd.SetDefaultValue("50.");
137
138 // sigmaAngle command
139 auto& sigmaAngleCmd = fMessenger->DeclarePro
140
141 sigmaAngleCmd.SetParameterName("t", true);
142 sigmaAngleCmd.SetRange("t>=0.");
143 sigmaAngleCmd.SetDefaultValue("2.");
144
145 // randomizePrimary command
146 auto& randomCmd = fMessenger->DeclarePropert
147 G4String guidance = "Boolean flag for random
148 guidance += "In case this flag is false, you
149 guidance += " with /gun/particle command.";
150 randomCmd.SetGuidance(guidance);
151 randomCmd.SetParameterName("flg", true);
152 randomCmd.SetDefaultValue("true");
153 }
154
155 //..oooOO0OOooo........oooOO0OOooo........oooO
156
157 } // namespace B5
158