| 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 optical/OpNovice2/src/PrimaryGeneratorAction.cc
27 /// \brief Implementation of the PrimaryGeneratorAction class
28 //
29 //
30 //
31 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
32 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
33
34 #include "PrimaryGeneratorAction.hh"
35
36 #include "PrimaryGeneratorMessenger.hh"
37
38 #include "G4Event.hh"
39 #include "G4OpticalPhoton.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()
49 : G4VUserPrimaryGeneratorAction(), fParticleGun(nullptr)
50 {
51 G4int n_particle = 1;
52 fParticleGun = new G4ParticleGun(n_particle);
53
54 // create a messenger for this class
55 fGunMessenger = new PrimaryGeneratorMessenger(this);
56
57 G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable();
58 G4ParticleDefinition* particle = particleTable->FindParticle("e+");
59
60 fParticleGun->SetParticleDefinition(particle);
61 fParticleGun->SetParticleTime(0.0 * ns);
62 fParticleGun->SetParticlePosition(G4ThreeVector(0.0 * cm, 0.0 * cm, 0.0 * cm));
63 fParticleGun->SetParticleMomentumDirection(G4ThreeVector(1., 0., 0.));
64 fParticleGun->SetParticleEnergy(500.0 * keV);
65 }
66
67 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
68
69 PrimaryGeneratorAction::~PrimaryGeneratorAction()
70 {
71 delete fParticleGun;
72 delete fGunMessenger;
73 }
74
75 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
76
77 void PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent)
78 {
79 if (fRandomDirection) {
80 G4double theta = CLHEP::halfpi * G4UniformRand();
81 G4double phi = CLHEP::twopi * G4UniformRand();
82 G4double x = std::cos(theta);
83 G4double y = std::sin(theta) * std::sin(phi);
84 G4double z = std::sin(theta) * std::cos(phi);
85 G4ThreeVector dir(x, y, z);
86 fParticleGun->SetParticleMomentumDirection(dir);
87 }
88 if (fParticleGun->GetParticleDefinition() == G4OpticalPhoton::OpticalPhotonDefinition()) {
89 if (fPolarized)
90 SetOptPhotonPolar(fPolarization);
91 else
92 SetOptPhotonPolar();
93 }
94 fParticleGun->GeneratePrimaryVertex(anEvent);
95 }
96
97 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
98
99 void PrimaryGeneratorAction::SetOptPhotonPolar()
100 {
101 G4double angle = G4UniformRand() * 360.0 * deg;
102 SetOptPhotonPolar(angle);
103 }
104
105 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
106
107 void PrimaryGeneratorAction::SetOptPhotonPolar(G4double angle)
108 {
109 if (fParticleGun->GetParticleDefinition() != G4OpticalPhoton::OpticalPhotonDefinition()) {
110 G4ExceptionDescription ed;
111 ed << "The particleGun is not an opticalphoton.";
112 G4Exception("PrimaryGeneratorAction::SetOptPhotonPolar", "OpNovice2_004", JustWarning, ed);
113 return;
114 }
115
116 fPolarized = true;
117 fPolarization = angle;
118
119 G4ThreeVector normal(1., 0., 0.);
120 G4ThreeVector kphoton = fParticleGun->GetParticleMomentumDirection();
121 G4ThreeVector product = normal.cross(kphoton);
122 G4double modul2 = product * product;
123
124 G4ThreeVector e_perpend(0., 0., 1.);
125 if (modul2 > 0.) e_perpend = (1. / std::sqrt(modul2)) * product;
126 G4ThreeVector e_paralle = e_perpend.cross(kphoton);
127
128 G4ThreeVector polar = std::cos(angle) * e_paralle + std::sin(angle) * e_perpend;
129 fParticleGun->SetParticlePolarization(polar);
130 }
131
132 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
133 void PrimaryGeneratorAction::SetRandomDirection(G4bool val)
134 {
135 fRandomDirection = val;
136 }
137