setter to assume pion mass for clusters
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliJetModelBaseTask.h
1 #ifndef ALIJETMODELBASETASK_H
2 #define ALIJETMODELBASETASK_H
3
4 // $Id$
5
6 class TClonesArray;
7 class AliEMCALGeometry;
8 class AliVCluster;
9 class AliPicoTrack;
10 class AliVCaloCells;
11 class AliAODMCParticle;
12 class AliNamedArrayI;
13 class TF2;
14
15 #include <TH1F.h>
16 #include <TF1.h>
17
18 #include "AliAnalysisTaskSE.h"
19
20 class AliJetModelBaseTask : public AliAnalysisTaskSE {
21  public:
22   AliJetModelBaseTask();
23   AliJetModelBaseTask(const char *name, Bool_t drawqa=kFALSE); 
24   virtual ~AliJetModelBaseTask();
25
26   void                   SetEtaRange(Float_t min, Float_t max) { fEtaMin       = min;  fEtaMax = max; }
27   void                   SetPhiRange(Float_t min, Float_t max) { fPhiMin       = min;  fPhiMax = max; }
28   void                   SetPtRange(Float_t min, Float_t max)  { fPtMin        = min;  fPtMax  = max; }
29   void                   SetPtSpectrum(TH1 *f)                 { fPtSpectrum   = f;    }
30   void                   SetPtSpectrum(TF1 *f)                 { fPtSpectrum   = new TH1F("ptSpectrum","ptSpectrum",1000,f->GetXmin(),f->GetXmax()); 
31                                                                  fPtSpectrum->Add(f); }
32   void                   SetPtPhiEvPlDistribution(TF2 *f)      { fPtPhiEvPlDistribution   = f;    }
33   void                   SetDensitySpectrum(TH1 *f)            { fDensitySpectrum = f;    }
34   void                   SetDensitySpectrum(TF1 *f)            { fDensitySpectrum = new TH1F("densitypectrum","densitypectrum",1000,f->GetXmin(),f->GetXmax()); 
35                                                                  fDensitySpectrum->Add(f); }
36   void                   SetDifferentialV2(TF1* f)             { fDifferentialV2 = f;  }
37   void                   SetAddV2(Bool_t b)                    { fAddV2 = b;           }
38   void                   SetAddFlowFluctuations(Bool_t b)      { fFlowFluctuations = b;}
39   void                   SetMC(Bool_t a)                       { fIsMC         = a   ; }
40   void                   SetCopyArray(Bool_t copy)             { fCopyArray    = copy; }
41   void                   SetTracksName(const char *n)          { fTracksName   = n;    }
42   void                   SetClusName(const char *n)            { fCaloName     = n;    }
43   void                   SetCellsName(const char *n)           { fCellsName    = n;    }
44   void                   SetMCParticlesName(const char *n)     { fMCParticlesName = n; }
45   void                   SetSuffix(const char *s)              { fSuffix       = s;    }
46   void                   SetGeometryName(const char *n)        { fGeomName     = n;    }
47   void                   SetMarkMC(Int_t m)                    { fMarkMC       = m;    }
48   virtual void           SetNClusters(Int_t n)                 { fNClusters    = n;    }
49   virtual void           SetNCells(Int_t n)                    { fNCells       = n;    }
50   virtual void           SetNTracks(Int_t n)                   { fNTracks      = n;    }
51
52  protected:
53   void                   UserExec(Option_t* /*option*/);
54   void                   UserCreateOutputObjects();
55   Int_t                  SetNumberOfOutCells(Int_t n);                                          // set the number of cells
56   Int_t                  AddCell(Double_t e = -1, Double_t eta = -999, Double_t phi = -1);      // add a cell; if values are -1 generate random parameters
57   Int_t                  AddCell(Double_t e, Int_t absId, Double_t time = 0, Int_t label=0);    // add a cell with given energy, position and times
58   AliVCluster           *AddCluster(Double_t e = -1, Double_t eta = -999, Double_t phi = -1, Int_t label=0); // add a cluster; if values are -1 generate random parameters
59   AliVCluster           *AddCluster(Double_t e, Int_t absId, Int_t label=0);                    // add a cluster with given energy and position
60   AliVCluster           *AddCluster(AliVCluster *oc);                                           // add a cluster (copy)
61   AliPicoTrack          *AddTrack(Double_t pt = -1, Double_t eta = -999, Double_t phi = -1, Byte_t type=0, 
62                                   Double_t etaemc=0, Double_t phiemc=0, Double_t ptemc=0, Bool_t ise=kFALSE, 
63                                   Int_t label=0, Short_t charge=1, Double_t mass = 0.1396);          // add a track; if values are -1 generate random parameters
64   AliAODMCParticle      *AddMCParticle(AliAODMCParticle *part, Int_t origIndex);                // add a MC particle
65   void                   AddV2(Double_t &phi, Double_t &pt) const;
66   void                   CopyCells();
67   void                   CopyClusters();
68   void                   CopyTracks();
69   void                   CopyMCParticles();
70   void                   GetRandomCell(Double_t &eta, Double_t &phi, Int_t &absId);             // generate a random cell in the calorimeter
71   Double_t               GetRandomEta(Bool_t emcal=kFALSE);                                     // generate a random eta value in the given range
72   Double_t               GetRandomPhi(Bool_t emcal=kFALSE);                                     // generate a random phi value in the given range
73   Double_t               GetRandomPt();                                                         // generate a random pt value in the given range
74   void                   GetRandomParticle(Double_t &pt, Double_t &eta, Double_t &phi, Bool_t emcal=kFALSE);  // generate a particle with random eta,phi,pt values
75   virtual Bool_t         ExecOnce();                                                            // intialize task
76   virtual void           Run();                                                                 // do jet model action
77
78   TString                fGeomName;               // EMCal geometry name
79   TString                fTracksName;             // name of track collection
80   TString                fOutTracksName;          // name of output track collection
81   TString                fCaloName;               // name of calo cluster collection
82   TString                fOutCaloName;            // name of output cluster collection
83   TString                fCellsName;              // name of calo cells collection
84   TString                fOutCellsName;           // name of output cells collection
85   TString                fMCParticlesName;        // name of MC particle collection
86   TString                fOutMCParticlesName;     // name of output MC particle collection
87   Bool_t                 fIsMC;                   // whether the current event is MC or not
88   TString                fSuffix;                 // suffix to add in the name of new collections
89   Float_t                fEtaMin;                 // eta minimum value
90   Float_t                fEtaMax;                 // eta maximum value
91   Float_t                fPhiMin;                 // phi minimum value
92   Float_t                fPhiMax;                 // phi maximum value
93   Float_t                fPtMin;                  // pt minimum value
94   Float_t                fPtMax;                  // pt maximum value
95   Bool_t                 fCopyArray;              // whether or not the array will be copied to a new one before modelling
96   Int_t                  fNClusters;              // how many clusters are being processed
97   Int_t                  fNCells;                 // how many cells are being processed
98   Int_t                  fNTracks;                // how many tracks are being processed
99   Int_t                  fMarkMC;                 // which MC label is to be used (default=100)
100   TH1                   *fPtSpectrum;             // pt spectrum to extract random pt values
101   TF2                   *fPtPhiEvPlDistribution;  // pt vs. (phi-psi) distribution to extract random pt/phi values
102   TH1                   *fDensitySpectrum;        // particle density spectrum to extract random density values
103   TF1                   *fDifferentialV2;         // v2 as function of pt
104   Bool_t                 fAddV2;                  // add v2 sampled from a tf1
105   Bool_t                 fFlowFluctuations;       // introduce gaussian flow fluctuation 
106   Bool_t                 fQAhistos;               // draw QA histograms
107   Double_t               fPsi;                    //!simmetry plane for the elliptic flow
108   Bool_t                 fIsInit;                 //!=true if initialized
109   AliEMCALGeometry      *fGeom;                   //!pointer to EMCal geometry
110   Double_t               fVertex[3];              //!event vertex
111   TClonesArray          *fClusters;               //!cluster collection
112   TClonesArray          *fOutClusters;            //!output cluster collection
113   TClonesArray          *fTracks;                 //!track collection
114   TClonesArray          *fOutTracks;              //!output track collection
115   AliVCaloCells         *fCaloCells;              //!cells collection
116   AliVCaloCells         *fOutCaloCells;           //!output cells collection
117   Int_t                  fAddedCells;             //!number of added cells
118   TClonesArray          *fMCParticles;            //!MC particles collection
119   AliNamedArrayI        *fMCParticlesMap;         //!MC particles mapping
120   TClonesArray          *fOutMCParticles;         //!output MC particles collection
121   AliNamedArrayI        *fOutMCParticlesMap;      //!MC particles mapping
122   Int_t                  fMCLabelShift;           //!MC label shift
123   Bool_t                 fEsdMode;                //!ESD/AOD mode
124   TList                 *fOutput;                 //!output list for QA histograms
125
126  private:
127   AliJetModelBaseTask(const AliJetModelBaseTask&);            // not implemented
128   AliJetModelBaseTask &operator=(const AliJetModelBaseTask&); // not implemented
129
130   ClassDef(AliJetModelBaseTask, 11) // Jet modelling task
131 };
132 #endif