| 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 eventgenerator/HepMC/HepMCEx02/include/H02PrimaryGeneratorAction.hh
27 /// \brief Definition of the H02PrimaryGeneratorAction class
28 //
29 // ====================================================================
30 //
31 // H02PrimaryGeneratorAction.hh
32 //
33 // ====================================================================
34 #ifndef H02_PRIMARY_GENERATOR_ACTION_H
35 #define H02_PRIMARY_GENERATOR_ACTION_H
36
37 #include "G4VUserPrimaryGeneratorAction.hh"
38 #include "globals.hh"
39
40 #include <map>
41
42 class G4Event;
43 class G4VPrimaryGenerator;
44 class H02PrimaryGeneratorMessenger;
45
46 class H02PrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction
47 {
48 public:
49 H02PrimaryGeneratorAction();
50 ~H02PrimaryGeneratorAction();
51
52 virtual void GeneratePrimaries(G4Event* anEvent);
53
54 void SetGenerator(G4VPrimaryGenerator* gen);
55 void SetGenerator(G4String genname);
56
57 G4VPrimaryGenerator* GetGenerator() const;
58 G4String GetGeneratorName() const;
59
60 private:
61 G4VPrimaryGenerator* fParticleGun;
62 G4VPrimaryGenerator* fHepmcAscii;
63 G4VPrimaryGenerator* fPythiaGen;
64
65 G4VPrimaryGenerator* fCurrentGenerator;
66 G4String fCurrentGeneratorName;
67 std::map<G4String, G4VPrimaryGenerator*> fGentypeMap;
68
69 H02PrimaryGeneratorMessenger* fMessenger;
70 };
71
72 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
73 inline void H02PrimaryGeneratorAction::SetGenerator(G4VPrimaryGenerator* gen)
74 {
75 fCurrentGenerator = gen;
76 }
77
78 inline void H02PrimaryGeneratorAction::SetGenerator(G4String genname)
79 {
80 std::map<G4String, G4VPrimaryGenerator*>::iterator pos = fGentypeMap.find(genname);
81 if (pos != fGentypeMap.end()) {
82 fCurrentGenerator = pos->second;
83 fCurrentGeneratorName = genname;
84 }
85 }
86
87 inline G4VPrimaryGenerator* H02PrimaryGeneratorAction::GetGenerator() const
88 {
89 return fCurrentGenerator;
90 }
91
92 inline G4String H02PrimaryGeneratorAction::GetGeneratorName() const
93 {
94 return fCurrentGeneratorName;
95 }
96
97 #endif
98