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