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 // 27 /// \file field/field04/src/F04PrimaryGeneratorAction.cc 28 /// \brief Implementation of the F04PrimaryGeneratorAction class 29 // 30 31 #include "F04PrimaryGeneratorAction.hh" 32 33 #include "F04DetectorConstruction.hh" 34 #include "F04PrimaryGeneratorMessenger.hh" 35 36 #include "G4Event.hh" 37 #include "G4GeometryManager.hh" 38 #include "G4ParticleDefinition.hh" 39 #include "G4ParticleGun.hh" 40 #include "G4ParticleTable.hh" 41 #include "G4PhysicalConstants.hh" 42 #include "G4SystemOfUnits.hh" 43 #include "G4TouchableHandle.hh" 44 #include "G4ios.hh" 45 #include "Randomize.hh" 46 47 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 48 49 F04PrimaryGeneratorAction::F04PrimaryGeneratorAction(F04DetectorConstruction* detectorConstruction) 50 : fDetector(detectorConstruction) 51 { 52 G4int n_particle = 1; 53 fParticleGun = new G4ParticleGun(n_particle); 54 55 fGunMessenger = new F04PrimaryGeneratorMessenger(this); 56 57 G4String particleName; 58 G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable(); 59 60 fParticleGun->SetParticleDefinition(particleTable->FindParticle(particleName = "proton")); 61 fParticleGun->SetParticleEnergy(500. * MeV); 62 fParticleGun->SetParticleMomentumDirection(G4ThreeVector(0., 0., 1.)); 63 64 fZvertex = -0.5 * (fDetector->GetTargetThickness()); 65 fParticleGun->SetParticlePosition(G4ThreeVector(fXvertex, fYvertex, fZvertex)); 66 } 67 68 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 69 70 F04PrimaryGeneratorAction::~F04PrimaryGeneratorAction() 71 { 72 delete fParticleGun; 73 delete fGunMessenger; 74 } 75 76 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 77 78 void F04PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent) 79 { 80 if (!fFirst) { 81 fFirst = true; 82 G4ThreeVector direction(0.0, 0.0, 1.0); 83 84 G4Navigator* theNavigator = 85 G4TransportationManager::GetTransportationManager()->GetNavigatorForTracking(); 86 if (theNavigator->GetWorldVolume()) { 87 auto aNavigator = new G4Navigator(); 88 aNavigator->SetWorldVolume(theNavigator->GetWorldVolume()); 89 90 G4ThreeVector center(0., 0., 0.); 91 aNavigator->LocateGlobalPointAndSetup(center, nullptr, false); 92 93 G4TouchableHandle touchable = aNavigator->CreateTouchableHistoryHandle(); 94 95 // set Global2local transform 96 fGlobal2local = touchable->GetHistory()->GetTopTransform(); 97 98 direction = fGlobal2local.Inverse().TransformAxis(direction); 99 delete aNavigator; 100 } 101 102 fParticleGun->SetParticleMomentumDirection(direction); 103 } 104 105 G4double x0, y0, z0; 106 107 if (fVertexDefined) { 108 x0 = fXvertex; 109 y0 = fYvertex; 110 z0 = fZvertex; 111 } 112 else { 113 x0 = 0.; 114 y0 = 0.; 115 z0 = -0.5 * (fDetector->GetTargetThickness()); 116 } 117 118 G4double r0, phi0; 119 120 if (fRndmFlag == "on") { 121 r0 = (fDetector->GetTargetRadius()) * std::sqrt(G4UniformRand()); 122 phi0 = twopi * G4UniformRand(); 123 x0 = r0 * std::cos(phi0); 124 y0 = r0 * std::sin(phi0); 125 } 126 127 G4ThreeVector localPosition(x0, y0, z0); 128 G4ThreeVector globalPosition = fGlobal2local.Inverse().TransformPoint(localPosition); 129 130 fParticleGun->SetParticlePosition(globalPosition); 131 fParticleGun->GeneratePrimaryVertex(anEvent); 132 } 133 134 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 135 136 void F04PrimaryGeneratorAction::SetXvertex(G4double x) 137 { 138 fVertexDefined = true; 139 fXvertex = x; 140 G4cout << " X coordinate of the primary vertex = " << fXvertex / mm << " mm." << G4endl; 141 } 142 143 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 144 145 void F04PrimaryGeneratorAction::SetYvertex(G4double y) 146 { 147 fVertexDefined = true; 148 fYvertex = y; 149 G4cout << " Y coordinate of the primary vertex = " << fYvertex / mm << " mm." << G4endl; 150 } 151 152 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 153 154 void F04PrimaryGeneratorAction::SetZvertex(G4double z) 155 { 156 fVertexDefined = true; 157 fZvertex = z; 158 G4cout << " Z coordinate of the primary vertex = " << fZvertex / mm << " mm." << G4endl; 159 } 160