| 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 PrimaryGeneratorAction.cc
28 /// \brief Primary Generator for Molecular DNA simulation
29
30 #include "PrimaryGeneratorAction.hh"
31
32 #include "PrimaryGeneratorMessenger.hh"
33 #include "PrimaryGeneratorSourceGRASCSV.hh"
34
35 #include "G4GeneralParticleSource.hh"
36 #include "G4ParticleGun.hh"
37
38 // define a Mutex to avoid concurrent reading in multi-thread
39 #include "G4AutoLock.hh"
40 namespace
41 {
42 G4Mutex PrimaryGeneratorMutex = G4MUTEX_INITIALIZER;
43 }
44
45 // instance of PrimaryGeneratorSource
46 PrimaryGeneratorSourceGRASCSV* PrimaryGeneratorAction::fPrimarySource = nullptr;
47
48 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
49
50 PrimaryGeneratorAction::PrimaryGeneratorAction()
51 {
52 G4AutoLock lock(&PrimaryGeneratorMutex);
53 fParticleGun = new G4GeneralParticleSource();
54
55 fFirstEvent = true;
56 fGunMessenger = new PrimaryGeneratorMessenger(this);
57 }
58
59 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
60
61 PrimaryGeneratorAction::~PrimaryGeneratorAction()
62 {
63 G4AutoLock lock(&PrimaryGeneratorMutex);
64 delete fParticleGun;
65 if (fPrimarySource != nullptr) {
66 delete fPrimarySource;
67 fPrimarySource = nullptr;
68 }
69 delete fGunMessenger;
70 }
71
72 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
73
74 void PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent)
75 {
76 if (fMyInputFileName != "") {
77 G4AutoLock lock(&PrimaryGeneratorMutex);
78 // Read primaries from file
79 // Before first event, instantiate file reader if fInputFileName is not empty
80 if (fFirstEvent) {
81 fPrimarySource = new PrimaryGeneratorSourceGRASCSV(fMyInputFileName);
82 fFirstEvent = false;
83 }
84
85 auto* fpParticleGun = new G4ParticleGun();
86
87 // Get a new primary particle.
88 Primary* primary = fPrimarySource->GetPrimary();
89 if (primary != nullptr) {
90 G4String particleName = primary->GetName();
91 G4ThreeVector pos = primary->GetPosition();
92 G4ThreeVector momdir = primary->GetMomentumDirection();
93 G4double energy = primary->GetEnergy();
94 G4ParticleDefinition* particle = primary->GetParticleDefinition();
95 // primary->Print(); // print of the data of the primary particle
96
97 fpParticleGun->SetParticleDefinition(particle);
98 fpParticleGun->SetParticlePosition(pos);
99 fpParticleGun->SetParticleEnergy(energy * CLHEP::MeV);
100 fpParticleGun->SetParticleMomentumDirection(momdir);
101 fpParticleGun->GeneratePrimaryVertex(anEvent);
102 }
103 else {
104 // If primary is NULL, the end of file has been reached or the file format is not consistent
105 // A Geantino placed at kInfinity will be fired in this case
106 G4cout << "WARNING: The phase space source is ended. Maybe you reach the end of file, or "
107 "your file is broken. A Geantino will be generated."
108 << G4endl;
109 G4ParticleTable* pTable = G4ParticleTable::GetParticleTable();
110 G4ParticleDefinition* particle = pTable->FindParticle("geantino");
111 fpParticleGun->SetParticleDefinition(particle);
112 G4ThreeVector pos = G4ThreeVector(kInfinity, kInfinity, kInfinity);
113 fpParticleGun->SetParticlePosition(pos);
114 fpParticleGun->SetParticleEnergy(0);
115 fpParticleGun->SetParticleMomentumDirection(pos);
116 fpParticleGun->GeneratePrimaryVertex(anEvent);
117 }
118 }
119 else {
120 fParticleGun->GeneratePrimaryVertex(anEvent);
121 }
122 }
123
124 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
125