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 runAndEvent/RE01/src/RE01PrimaryGeneratorAction.cc 27 /// \brief Implementation of the RE01PrimaryGeneratorAction class 28 // 29 // 30 31 #include "RE01PrimaryGeneratorAction.hh" 32 33 #include "RE01PrimaryGeneratorMessenger.hh" 34 35 #include "G4Event.hh" 36 #include "G4HEPEvtInterface.hh" 37 #include "G4ParticleDefinition.hh" 38 #include "G4ParticleGun.hh" 39 #include "G4ParticleTable.hh" 40 #include "G4SystemOfUnits.hh" 41 42 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 43 RE01PrimaryGeneratorAction::RE01PrimaryGeneratorAction() 44 : G4VUserPrimaryGeneratorAction(), fHEPEvt(0), fParticleGun(0), fMessenger(0) 45 { 46 const char* filename = "pythia_event.data"; 47 fHEPEvt = new G4HEPEvtInterface(filename); 48 49 G4int n_particle = 1; 50 G4ParticleGun* particleGun = new G4ParticleGun(n_particle); 51 G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable(); 52 G4String particleName; 53 G4ParticleDefinition* particle = particleTable->FindParticle(particleName = "mu+"); 54 particleGun->SetParticleDefinition(particle); 55 particleGun->SetParticleMomentumDirection(G4ThreeVector(0., 1., 0.)); 56 particleGun->SetParticleEnergy(100. * GeV); 57 particleGun->SetParticlePosition(G4ThreeVector(0. * cm, 0. * cm, 0. * cm)); 58 fParticleGun = particleGun; 59 60 fMessenger = new RE01PrimaryGeneratorMessenger(this); 61 fUseHEPEvt = true; 62 } 63 64 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 65 #include "G4AutoLock.hh" 66 namespace 67 { 68 G4Mutex RE01PrimGenDestrMutex = G4MUTEX_INITIALIZER; 69 G4Mutex RE01PrimGenMutex = G4MUTEX_INITIALIZER; 70 } // namespace 71 72 RE01PrimaryGeneratorAction::~RE01PrimaryGeneratorAction() 73 { 74 G4AutoLock lock(&RE01PrimGenDestrMutex); 75 if (fHEPEvt) { 76 delete fHEPEvt; 77 fHEPEvt = 0; 78 } 79 delete fParticleGun; 80 delete fMessenger; 81 } 82 83 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 84 void RE01PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent) 85 { 86 if (fUseHEPEvt) { 87 G4AutoLock lock(&RE01PrimGenMutex); 88 fHEPEvt->GeneratePrimaryVertex(anEvent); 89 if (anEvent->GetNumberOfPrimaryVertex() == 0) { // End-of-file 90 G4Exception("RE01PrimaryGeneratorAction::GeneratePrimaries", "RE01_0001", RunMustBeAborted, 91 "End of file detected."); 92 } 93 } 94 else { 95 fParticleGun->GeneratePrimaryVertex(anEvent); 96 } 97 } 98