setter to assume pion mass for clusters
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliJetEmbeddingFromGenTask.h
1 #ifndef ALIJETEMBEDDINGFROMGENTASK_H
2 #define ALIJETEMBEDDINGFROMGENTASK_H
3
4 // $Id$
5
6 class TClonesArray;
7 class TProfile;
8 class AliEMCALGeometry;
9
10 #include "AliJetModelBaseTask.h"
11 class AliGenerator;
12
13 class AliJetEmbeddingFromGenTask : public AliJetModelBaseTask {
14  public:
15   AliJetEmbeddingFromGenTask();
16   AliJetEmbeddingFromGenTask(const char *name, Bool_t drawqa);
17   virtual ~AliJetEmbeddingFromGenTask();
18
19   void           UserCreateOutputObjects();
20   void           FillPythiaHistograms();
21
22   void           SetGen(AliGenerator *gen)      { fGen = gen       ; }
23   void           SetMasslessParticles(Bool_t b) { fMassless = b    ; }
24   void           SetChargedOnly(Bool_t b)       { fChargedOnly = b ; }
25
26  protected:
27   Bool_t         ExecOnce();
28   void           Run();
29
30   AliGenerator  *fGen;                    //generator
31   Bool_t         fMassless;               //make particles massless
32   Bool_t         fChargedOnly;            //accept only charged particles
33
34   TH1F          *fHistPt;                 //!pT spectrum of generated particles
35   TH2F          *fHistEtaPhi;             //!eta-phi of generated particles
36   TH1F          *fHistTrials;             //!trials from generator
37   TProfile      *fHistXsection;           //!x-section from generator
38   TH1           *fHistPtHard;             //!pt hard distribution
39
40  private:
41   AliJetEmbeddingFromGenTask(const AliJetEmbeddingFromGenTask&);            // not implemented
42   AliJetEmbeddingFromGenTask &operator=(const AliJetEmbeddingFromGenTask&); // not implemented
43
44   ClassDef(AliJetEmbeddingFromGenTask, 4) // Jet embedding task
45 };
46 #endif