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 // (adapted from B1PrimaryGeneratorAction) 27 // Author: A.Knaian (ara@nklabs.com), N.MacFadden (natemacfadden@gmail.com) 28 29 #include "FAPrimaryGeneratorAction.hh" 30 #include "G4LogicalVolumeStore.hh" 31 #include "G4LogicalVolume.hh" 32 #include "G4Box.hh" 33 #include "G4RunManager.hh" 34 #include "G4ParticleGun.hh" 35 #include "G4ParticleTable.hh" 36 #include "G4ParticleDefinition.hh" 37 #include "G4SystemOfUnits.hh" 38 #include "Randomize.hh" 39 40 PrimaryGeneratorAction::PrimaryGeneratorAction() 41 : G4VUserPrimaryGeneratorAction(), fParticleGun(nullptr), fWorldBox(nullptr) 42 { 43 G4int n_particle = 1; 44 fParticleGun = new G4ParticleGun(n_particle); 45 46 // default particle kinematic 47 G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable(); 48 G4String particleName; 49 G4ParticleDefinition* particle 50 = particleTable->FindParticle(particleName="proton"); 51 fParticleGun->SetParticleDefinition(particle); 52 fParticleGun->SetParticleMomentumDirection(G4ThreeVector(0.,0.,1.)); 53 fParticleGun->SetParticleEnergy(50.*MeV); 54 } 55 56 PrimaryGeneratorAction::~PrimaryGeneratorAction() 57 { 58 delete fParticleGun; 59 } 60 61 void PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent) 62 { 63 G4double worldSizeXY = 0; 64 G4double worldSizeZ = 0; 65 66 if (!fWorldBox) 67 { 68 G4LogicalVolume* worldLV 69 = G4LogicalVolumeStore::GetInstance()->GetVolume("World"); 70 if ( worldLV ) fWorldBox = dynamic_cast<G4Box*>(worldLV->GetSolid()); 71 } 72 if ( fWorldBox ) 73 { 74 worldSizeXY = fWorldBox->GetXHalfLength()*2.; 75 worldSizeZ = fWorldBox->GetZHalfLength()*2.; 76 } 77 else 78 { 79 G4ExceptionDescription msg; 80 msg << "World volume of box shape not found.\n"; 81 msg << "Perhaps you have changed geometry.\n"; 82 msg << "The gun will be place at the center."; 83 G4Exception("PrimaryGeneratorAction::GeneratePrimaries()", 84 "MyCode0002",JustWarning,msg); 85 } 86 87 // shoot on XY disk centered on Z-axis behind the cloud 88 G4double sigma = worldSizeXY/10.0; // spread in x and y 89 G4double x0 = G4RandGauss::shoot(0,sigma); 90 G4double y0 = G4RandGauss::shoot(0,sigma); 91 G4double z0 = 0.95 * (-0.5) * worldSizeZ; 92 93 fParticleGun->SetParticlePosition(G4ThreeVector(x0,y0,z0)); 94 fParticleGun->GeneratePrimaryVertex(anEvent); 95 } 96 97