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