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 :fDetector(det) 49 { 50 { 50 G4int n_particle = 1; 51 G4int n_particle = 1; 51 fParticleGun = new G4ParticleGun(n_particle) << 52 fParticleGun = new G4ParticleGun(n_particle); 52 SetDefaultKinematic(); 53 SetDefaultKinematic(); 53 << 54 54 // define commands for this class 55 // define commands for this class 55 DefineCommands(); << 56 DefineCommands(); 56 } 57 } 57 58 58 //....oooOO0OOooo........oooOO0OOooo........oo 59 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 59 60 60 PrimaryGeneratorAction::~PrimaryGeneratorActio 61 PrimaryGeneratorAction::~PrimaryGeneratorAction() 61 { 62 { 62 delete fParticleGun; 63 delete fParticleGun; 63 delete fMessenger; 64 delete fMessenger; 64 } 65 } 65 66 66 //....oooOO0OOooo........oooOO0OOooo........oo << 67 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 67 68 68 void PrimaryGeneratorAction::SetDefaultKinemat 69 void PrimaryGeneratorAction::SetDefaultKinematic() 69 { 70 { 70 G4ParticleTable* particleTable = G4ParticleT 71 G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable(); 71 G4String particleName; 72 G4String particleName; 72 G4ParticleDefinition* particle = particleTab << 73 G4ParticleDefinition* particle >> 74 = particleTable->FindParticle(particleName="proton"); 73 fParticleGun->SetParticleDefinition(particle 75 fParticleGun->SetParticleDefinition(particle); 74 fParticleGun->SetParticleMomentumDirection(G << 76 fParticleGun->SetParticleMomentumDirection(G4ThreeVector(1.,0.,0.)); 75 fParticleGun->SetParticleEnergy(5. * GeV); << 77 fParticleGun->SetParticleEnergy(5.*GeV); 76 G4double position = -0.5 * (fDetector->GetWo << 78 G4double position = -0.5*(fDetector->GetWorldSizeX()); 77 fParticleGun->SetParticlePosition(G4ThreeVec << 79 fParticleGun->SetParticlePosition(G4ThreeVector(position,0.*cm,0.*cm)); 78 } 80 } 79 81 80 //....oooOO0OOooo........oooOO0OOooo........oo 82 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 81 83 82 void PrimaryGeneratorAction::GeneratePrimaries 84 void PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent) 83 { 85 { 84 // this function is called at the begining o << 86 //this function is called at the begining of event 85 // 87 // 86 // randomize the beam, if requested. << 88 //randomize the beam, if requested. 87 if (fRndmBeam > 0.) { << 89 if (fRndmBeam > 0.) 88 G4ThreeVector oldPosition = fParticleGun-> << 90 { 89 G4double rbeam = 0.5 * (fDetector->GetCalo << 91 G4ThreeVector oldPosition = fParticleGun->GetParticlePosition(); 90 G4double x0 = oldPosition.x(); << 92 G4double rbeam = 0.5*(fDetector->GetCalorSizeYZ())*fRndmBeam; 91 G4double y0 = oldPosition.y() + (2 * G4Uni << 93 G4double x0 = oldPosition.x(); 92 G4double z0 = oldPosition.z() + (2 * G4Uni << 94 G4double y0 = oldPosition.y() + (2*G4UniformRand()-1.)*rbeam; 93 fParticleGun->SetParticlePosition(G4ThreeV << 95 G4double z0 = oldPosition.z() + (2*G4UniformRand()-1.)*rbeam; 94 fParticleGun->GeneratePrimaryVertex(anEven << 96 fParticleGun->SetParticlePosition(G4ThreeVector(x0,y0,z0)); 95 fParticleGun->SetParticlePosition(oldPosit << 97 fParticleGun->GeneratePrimaryVertex(anEvent); 96 } << 98 fParticleGun->SetParticlePosition(oldPosition); 97 else << 99 } 98 fParticleGun->GeneratePrimaryVertex(anEven << 100 else fParticleGun->GeneratePrimaryVertex(anEvent); 99 } 101 } 100 102 101 //....oooOO0OOooo........oooOO0OOooo........oo 103 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 102 104 103 void PrimaryGeneratorAction::DefineCommands() 105 void PrimaryGeneratorAction::DefineCommands() 104 { 106 { 105 // Define /testhadr/gun command directory us 107 // Define /testhadr/gun command directory using generic messenger class 106 fMessenger = new G4GenericMessenger(this, "/ << 108 fMessenger = new G4GenericMessenger(this, 107 << 109 "/testhadr/gun/", >> 110 "gun control"); >> 111 108 // default kinematic command 112 // default kinematic command 109 auto& defaultCmd = << 113 auto& defaultCmd 110 fMessenger->DeclareMethod("setDefault", &P << 114 = fMessenger->DeclareMethod("setDefault", 111 "set/reset kinem << 115 &PrimaryGeneratorAction::SetDefaultKinematic, 112 << 116 "set/reset kinematic in PrimaryGenerator"); 113 defaultCmd.SetStates(G4State_PreInit, G4Stat << 117 >> 118 defaultCmd.SetStates(G4State_PreInit, G4State_Idle); 114 119 115 // randomize beam extension command 120 // randomize beam extension command 116 auto& rndmCmd = fMessenger->DeclareProperty( << 121 auto& rndmCmd >> 122 = fMessenger->DeclareProperty("rndm", fRndmBeam); 117 123 118 rndmCmd.SetGuidance("lateral size of the bea 124 rndmCmd.SetGuidance("lateral size of the beam, in fraction of sizeYZ "); 119 rndmCmd.SetParameterName("rBeam", false); 125 rndmCmd.SetParameterName("rBeam", false); 120 rndmCmd.SetRange("rBeam>=0.&&rBeam<=1."); 126 rndmCmd.SetRange("rBeam>=0.&&rBeam<=1."); 121 rndmCmd.SetDefaultValue("0."); 127 rndmCmd.SetDefaultValue("0."); 122 rndmCmd.SetStates(G4State_Idle); << 128 rndmCmd.SetStates(G4State_Idle); 123 } 129 } 124 130 125 //..oooOO0OOooo........oooOO0OOooo........oooO 131 //..oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 132 126 133