| 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
28 // ********************************************************************
29 //
30 // CaTS (Calorimetry and Tracking Simulation)
31 //
32 // Authors : Hans Wenzel
33 // Soon Yung Jun
34 // (Fermi National Accelerator Laboratory)
35 //
36 // History
37 // October 18th, 2021 : first implementation
38 //
39 // ********************************************************************
40 //
41 /// \file PrimaryGeneratorAction.hh
42 /// \brief Definition of the CaTS::PrimaryGeneratorAction class
43
44 #pragma once
45
46 #include "G4VUserPrimaryGeneratorAction.hh"
47 #include <G4String.hh>
48 #include <G4ios.hh>
49 #include <map>
50 #include <utility>
51 class G4GenericMessenger;
52 class G4VPrimaryGenerator;
53 class G4Event;
54
55 class PrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction
56 {
57 public:
58 PrimaryGeneratorAction();
59 virtual ~PrimaryGeneratorAction();
60 void GeneratePrimaries(G4Event*) final;
61 inline void SetGenerator(G4String genname)
62 {
63 std::map<G4String, G4VPrimaryGenerator*>::iterator pos =
64 gentypeMap.find(genname);
65 if(pos != gentypeMap.end())
66 {
67 fcurrentGenerator = pos->second;
68 fcurrentGeneratorName = genname;
69 }
70 else
71 {
72 G4cout << "Primary Generator: " << genname << " is not a valid Option"
73 << G4endl;
74 G4cout << "Valid Options are: ";
75 #if(G4VERSION_NUMBER > 1072)
76 for(auto [genType, generator] : gentypeMap)
77 {
78 G4cout << genType << " ";
79 }
80 G4cout << G4endl;
81 #else
82 for(std::map<G4String, G4VPrimaryGenerator*>::iterator ii =
83 gentypeMap.begin();
84 ii != gentypeMap.end(); ++ii)
85 {
86 G4cout << (*ii).first << " ";
87 }
88 G4cout << G4endl;
89 #endif
90 }
91 }
92 inline G4VPrimaryGenerator* GetGenerator() const { return fcurrentGenerator; }
93 inline G4String GetGeneratorName() const { return fcurrentGeneratorName; }
94 inline void Print()
95 {
96 G4cout << "===================================================" << G4endl;
97 G4cout << " Primary Generator: " << fcurrentGeneratorName << G4endl;
98 G4cout << "===================================================" << G4endl;
99 }
100
101 private:
102 /// Define UI commands: choice of primary generators
103 void DefineCommands();
104 G4VPrimaryGenerator* fcurrentGenerator = nullptr;
105 G4String fcurrentGeneratorName{ "particleGun" };
106 PrimaryGeneratorAction& operator=(const PrimaryGeneratorAction& right);
107 PrimaryGeneratorAction(const PrimaryGeneratorAction&);
108 std::map<G4String, G4VPrimaryGenerator*> gentypeMap;
109 /// Pointer to the messenger for UI commands
110 G4GenericMessenger* fMessenger = nullptr;
111 };
112