]>
Commit | Line | Data |
---|---|---|
676fb573 | 1 | // $Id$ |
2 | // Category: run | |
3 | // | |
4 | // Class that defines primary generator action. | |
5 | // Available primary generators (AliPrimaryGenerator): | |
6 | // kGun, // gun (can be set interactively) | |
7 | // kGeantino, // geantino with random direction | |
8 | // kChargedGeantino, // chargedgeantino with random direction | |
9 | // kAliGenerator // AliGenerator from AliRoot | |
10 | ||
11 | #ifndef ALI_PRIMARY_GENERATOR_ACTION_H | |
12 | #define ALI_PRIMARY_GENERATOR_ACTION_H | |
13 | ||
14 | #include "AliPrimaryGenerator.h" | |
15 | ||
16 | #include <G4VUserPrimaryGeneratorAction.hh> | |
17 | #include <globals.hh> | |
18 | ||
19 | #include <TClonesArray.h> | |
20 | ||
21 | class AliParticleGun; | |
22 | class AliPrimaryGeneratorMessenger; | |
23 | class G4ParticleGun; | |
24 | class G4Event; | |
25 | ||
26 | class AliPrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction | |
27 | { | |
28 | public: | |
29 | AliPrimaryGeneratorAction(); | |
30 | // --> protected | |
31 | // AliPrimaryGeneratorAction(const AliPrimaryGeneratorAction& right); | |
32 | virtual ~AliPrimaryGeneratorAction(); | |
33 | ||
34 | // methods | |
35 | virtual void GeneratePrimaries(G4Event* event); | |
36 | ||
37 | // set methods | |
38 | void SetGenerator(AliPrimaryGenerator generator); | |
39 | void SetNofGunParticles(G4int nofParticles); | |
40 | void SetVerboseLevel(G4int level); | |
41 | ||
42 | // get methods | |
43 | AliPrimaryGenerator GetGenerator() const; | |
44 | G4int GetNofGunParticles() const; | |
45 | G4int GetVerboseLevel() const; | |
46 | ||
47 | protected: | |
48 | AliPrimaryGeneratorAction(const AliPrimaryGeneratorAction& right); | |
49 | ||
50 | // operators | |
51 | AliPrimaryGeneratorAction& operator=( | |
52 | const AliPrimaryGeneratorAction& right); | |
53 | ||
54 | private: | |
55 | // methods | |
56 | void ConstructGenerator(); | |
57 | void ConstructGeantinoGenerator(G4bool isCharged); | |
58 | void ConstructAliGenerator(); | |
59 | void GenerateAliGeneratorPrimaries(G4Event* event); | |
60 | ||
61 | // data members | |
62 | AliPrimaryGenerator fGenerator; //selected AliPrimaryGenerator | |
63 | G4int fNofGunParticles; //number of gun particles | |
64 | G4int fVerboseLevel; //verbose level | |
65 | AliParticleGun* fParticleGun; //AliParticleGun | |
66 | TClonesArray* fParticleArray; //AliRun::fParticles | |
67 | AliPrimaryGeneratorMessenger* fMessenger; //messenger | |
68 | }; | |
69 | ||
70 | // inline methods | |
71 | ||
72 | inline void AliPrimaryGeneratorAction::SetVerboseLevel(G4int level) | |
73 | { fVerboseLevel = level; } | |
74 | ||
75 | inline AliPrimaryGenerator AliPrimaryGeneratorAction::GetGenerator() const | |
76 | { return fGenerator; } | |
77 | ||
78 | inline G4int AliPrimaryGeneratorAction::GetNofGunParticles() const | |
79 | { return fNofGunParticles; } | |
80 | ||
81 | inline G4int AliPrimaryGeneratorAction::GetVerboseLevel() const | |
82 | { return fVerboseLevel; } | |
83 | ||
84 | #endif //ALI_PRIMARY_GENERATOR_ACTION_H | |
85 | ||
86 |