| Geant4 Cross Reference |
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 /// \file PrimaryGeneratorAction.cc
27 /// \brief Implementation of the PrimaryGeneratorAction class
28 //
29 //
30 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
31 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
32
33 #include "PrimaryGeneratorAction.hh"
34
35 #include "DetectorConstruction.hh"
36 #include "HistoManager.hh"
37
38 #include "G4Event.hh"
39 #include "G4GenericMessenger.hh"
40 #include "G4ParticleDefinition.hh"
41 #include "G4ParticleGun.hh"
42 #include "G4ParticleTable.hh"
43 #include "G4SystemOfUnits.hh"
44 #include "Randomize.hh"
45
46 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
47
48 PrimaryGeneratorAction::PrimaryGeneratorAction(DetectorConstruction* det) : fDetector(det)
49 {
50 G4int n_particle = 1;
51 fParticleGun = new G4ParticleGun(n_particle);
52 SetDefaultKinematic();
53
54 // define commands for this class
55 DefineCommands();
56 }
57
58 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
59
60 PrimaryGeneratorAction::~PrimaryGeneratorAction()
61 {
62 delete fParticleGun;
63 delete fMessenger;
64 }
65
66 //....oooOO0OOooo........oooOO0OOooo........oooOO0OO ooo........oooOO0OOooo......
67
68 void PrimaryGeneratorAction::SetDefaultKinematic()
69 {
70 G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable();
71 G4String particleName;
72 G4ParticleDefinition* particle = particleTable->FindParticle(particleName = "proton");
73 fParticleGun->SetParticleDefinition(particle);
74 fParticleGun->SetParticleMomentumDirection(G4ThreeVector(1., 0., 0.));
75 fParticleGun->SetParticleEnergy(5. * GeV);
76 G4double position = -0.5 * (fDetector->GetWorldSizeX()) + 1 * nm;
77 fParticleGun->SetParticlePosition(G4ThreeVector(position, 0. * cm, 0. * cm));
78 }
79
80 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
81
82 void PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent)
83 {
84 // this function is called at the begining of event
85 //
86 // randomize the beam, if requested.
87 if (fRndmBeam > 0.) {
88 G4ThreeVector oldPosition = fParticleGun->GetParticlePosition();
89 G4double rbeam = 0.5 * (fDetector->GetCalorSizeYZ()) * fRndmBeam;
90 G4double x0 = oldPosition.x();
91 G4double y0 = oldPosition.y() + (2 * G4UniformRand() - 1.) * rbeam;
92 G4double z0 = oldPosition.z() + (2 * G4UniformRand() - 1.) * rbeam;
93 fParticleGun->SetParticlePosition(G4ThreeVector(x0, y0, z0));
94 fParticleGun->GeneratePrimaryVertex(anEvent);
95 fParticleGun->SetParticlePosition(oldPosition);
96 }
97 else
98 fParticleGun->GeneratePrimaryVertex(anEvent);
99 }
100
101 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
102
103 void PrimaryGeneratorAction::DefineCommands()
104 {
105 // Define /testhadr/gun command directory using generic messenger class
106 fMessenger = new G4GenericMessenger(this, "/testhadr/gun/", "gun control");
107
108 // default kinematic command
109 auto& defaultCmd =
110 fMessenger->DeclareMethod("setDefault", &PrimaryGeneratorAction::SetDefaultKinematic,
111 "set/reset kinematic in PrimaryGenerator");
112
113 defaultCmd.SetStates(G4State_PreInit, G4State_Idle);
114
115 // randomize beam extension command
116 auto& rndmCmd = fMessenger->DeclareProperty("rndm", fRndmBeam);
117
118 rndmCmd.SetGuidance("lateral size of the beam, in fraction of sizeYZ ");
119 rndmCmd.SetParameterName("rBeam", false);
120 rndmCmd.SetRange("rBeam>=0.&&rBeam<=1.");
121 rndmCmd.SetDefaultValue("0.");
122 rndmCmd.SetStates(G4State_Idle);
123 }
124
125 //..oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
126