| 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 /// \file PrimaryGeneratorAction.cc 26 /// \file PrimaryGeneratorAction.cc
27 /// \brief Implementation of the PrimaryGenera 27 /// \brief Implementation of the PrimaryGeneratorAction class
28 // 28 //
29 // 29 //
30 //....oooOO0OOooo........oooOO0OOooo........oo 30 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
31 //....oooOO0OOooo........oooOO0OOooo........oo 31 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
32 32
33 #include "PrimaryGeneratorAction.hh" 33 #include "PrimaryGeneratorAction.hh"
34 34
>> 35 #include "G4GenericMessenger.hh"
35 #include "DetectorConstruction.hh" 36 #include "DetectorConstruction.hh"
36 #include "HistoManager.hh" 37 #include "HistoManager.hh"
37 38
38 #include "G4Event.hh" 39 #include "G4Event.hh"
39 #include "G4GenericMessenger.hh" <<
40 #include "G4ParticleDefinition.hh" <<
41 #include "G4ParticleGun.hh" 40 #include "G4ParticleGun.hh"
42 #include "G4ParticleTable.hh" 41 #include "G4ParticleTable.hh"
>> 42 #include "G4ParticleDefinition.hh"
43 #include "G4SystemOfUnits.hh" 43 #include "G4SystemOfUnits.hh"
44 #include "Randomize.hh" 44 #include "Randomize.hh"
45 45
46 //....oooOO0OOooo........oooOO0OOooo........oo 46 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
47 47
48 PrimaryGeneratorAction::PrimaryGeneratorAction << 48 PrimaryGeneratorAction::PrimaryGeneratorAction(DetectorConstruction* det)
>> 49 :G4VUserPrimaryGeneratorAction(),
>> 50 fParticleGun(nullptr),
>> 51 fDetector(det),
>> 52 fMessenger(nullptr),
>> 53 fRndmBeam(0.)
>> 54
49 { 55 {
50 G4int n_particle = 1; 56 G4int n_particle = 1;
51 fParticleGun = new G4ParticleGun(n_particle) << 57 fParticleGun = new G4ParticleGun(n_particle);
52 SetDefaultKinematic(); 58 SetDefaultKinematic();
53 << 59
54 // define commands for this class 60 // define commands for this class
55 DefineCommands(); << 61 DefineCommands();
56 } 62 }
57 63
58 //....oooOO0OOooo........oooOO0OOooo........oo 64 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
59 65
60 PrimaryGeneratorAction::~PrimaryGeneratorActio 66 PrimaryGeneratorAction::~PrimaryGeneratorAction()
61 { 67 {
62 delete fParticleGun; 68 delete fParticleGun;
63 delete fMessenger; 69 delete fMessenger;
64 } 70 }
65 71
66 //....oooOO0OOooo........oooOO0OOooo........oo << 72 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
67 73
68 void PrimaryGeneratorAction::SetDefaultKinemat 74 void PrimaryGeneratorAction::SetDefaultKinematic()
69 { 75 {
70 G4ParticleTable* particleTable = G4ParticleT 76 G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable();
71 G4String particleName; 77 G4String particleName;
72 G4ParticleDefinition* particle = particleTab << 78 G4ParticleDefinition* particle
>> 79 = particleTable->FindParticle(particleName="proton");
73 fParticleGun->SetParticleDefinition(particle 80 fParticleGun->SetParticleDefinition(particle);
74 fParticleGun->SetParticleMomentumDirection(G << 81 fParticleGun->SetParticleMomentumDirection(G4ThreeVector(1.,0.,0.));
75 fParticleGun->SetParticleEnergy(5. * GeV); << 82 fParticleGun->SetParticleEnergy(5.*GeV);
76 G4double position = -0.5 * (fDetector->GetWo << 83 G4double position = -0.5*(fDetector->GetWorldSizeX());
77 fParticleGun->SetParticlePosition(G4ThreeVec << 84 fParticleGun->SetParticlePosition(G4ThreeVector(position,0.*cm,0.*cm));
78 } 85 }
79 86
80 //....oooOO0OOooo........oooOO0OOooo........oo 87 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
81 88
82 void PrimaryGeneratorAction::GeneratePrimaries 89 void PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent)
83 { 90 {
84 // this function is called at the begining o << 91 //this function is called at the begining of event
85 // 92 //
86 // randomize the beam, if requested. << 93 //randomize the beam, if requested.
87 if (fRndmBeam > 0.) { << 94 if (fRndmBeam > 0.)
88 G4ThreeVector oldPosition = fParticleGun-> << 95 {
89 G4double rbeam = 0.5 * (fDetector->GetCalo << 96 G4ThreeVector oldPosition = fParticleGun->GetParticlePosition();
90 G4double x0 = oldPosition.x(); << 97 G4double rbeam = 0.5*(fDetector->GetCalorSizeYZ())*fRndmBeam;
91 G4double y0 = oldPosition.y() + (2 * G4Uni << 98 G4double x0 = oldPosition.x();
92 G4double z0 = oldPosition.z() + (2 * G4Uni << 99 G4double y0 = oldPosition.y() + (2*G4UniformRand()-1.)*rbeam;
93 fParticleGun->SetParticlePosition(G4ThreeV << 100 G4double z0 = oldPosition.z() + (2*G4UniformRand()-1.)*rbeam;
94 fParticleGun->GeneratePrimaryVertex(anEven << 101 fParticleGun->SetParticlePosition(G4ThreeVector(x0,y0,z0));
95 fParticleGun->SetParticlePosition(oldPosit << 102 fParticleGun->GeneratePrimaryVertex(anEvent);
96 } << 103 fParticleGun->SetParticlePosition(oldPosition);
97 else << 104 }
98 fParticleGun->GeneratePrimaryVertex(anEven << 105 else fParticleGun->GeneratePrimaryVertex(anEvent);
99 } 106 }
100 107
101 //....oooOO0OOooo........oooOO0OOooo........oo 108 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
102 109
103 void PrimaryGeneratorAction::DefineCommands() 110 void PrimaryGeneratorAction::DefineCommands()
104 { 111 {
105 // Define /testhadr/gun command directory us 112 // Define /testhadr/gun command directory using generic messenger class
106 fMessenger = new G4GenericMessenger(this, "/ << 113 fMessenger = new G4GenericMessenger(this,
107 << 114 "/testhadr/gun/",
>> 115 "gun control");
>> 116
108 // default kinematic command 117 // default kinematic command
109 auto& defaultCmd = << 118 auto& defaultCmd
110 fMessenger->DeclareMethod("setDefault", &P << 119 = fMessenger->DeclareMethod("setDefault",
111 "set/reset kinem << 120 &PrimaryGeneratorAction::SetDefaultKinematic,
112 << 121 "set/reset kinematic in PrimaryGenerator");
113 defaultCmd.SetStates(G4State_PreInit, G4Stat << 122
>> 123 defaultCmd.SetStates(G4State_PreInit, G4State_Idle);
114 124
115 // randomize beam extension command 125 // randomize beam extension command
116 auto& rndmCmd = fMessenger->DeclareProperty( << 126 auto& rndmCmd
>> 127 = fMessenger->DeclareProperty("rndm", fRndmBeam);
117 128
118 rndmCmd.SetGuidance("lateral size of the bea 129 rndmCmd.SetGuidance("lateral size of the beam, in fraction of sizeYZ ");
119 rndmCmd.SetParameterName("rBeam", false); 130 rndmCmd.SetParameterName("rBeam", false);
120 rndmCmd.SetRange("rBeam>=0.&&rBeam<=1."); 131 rndmCmd.SetRange("rBeam>=0.&&rBeam<=1.");
121 rndmCmd.SetDefaultValue("0."); 132 rndmCmd.SetDefaultValue("0.");
122 rndmCmd.SetStates(G4State_Idle); << 133 rndmCmd.SetStates(G4State_Idle);
123 } 134 }
124 135
125 //..oooOO0OOooo........oooOO0OOooo........oooO 136 //..oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
>> 137
126 138