]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWG2/FLOW/AliFlowTasks/AliAnalysisTaskFittingQDistribution.h
added the possibility to use weights of the Q vector to FittingQDistributions
[u/mrichter/AliRoot.git] / PWG2 / FLOW / AliFlowTasks / AliAnalysisTaskFittingQDistribution.h
1 /* 
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. 
3  * See cxx source for full Copyright notice 
4  * $Id$ 
5  */
6
7 /**************************************
8  *    analysis task for fitting       * 
9  *         q-distribution             *
10  *                                    * 
11  * authors: Naomi van der Kolk        *
12  *           (kolk@nikhef.nl)         *  
13  *          Raimond Snellings         *
14  *           (snelling@nikhef.nl)     * 
15  *          Ante Bilandzic            *
16  *           (anteb@nikhef.nl)        * 
17  * ***********************************/
18
19 #ifndef AliAnalysisTaskFittingQDistribution_H
20 #define AliAnalysisTaskFittingQDistribution_H
21
22 #include "TString.h"
23 #include "AliAnalysisTask.h"
24
25 class AliESDEvent;
26 class AliAODEvent;
27 class AliCFManager;
28 class AliFittingQDistribution;
29 class AliFlowEventSimpleMaker;
30 class TFile;
31
32 //================================================================================================================
33
34 class AliAnalysisTaskFittingQDistribution : public AliAnalysisTask{
35  public:
36   AliAnalysisTaskFittingQDistribution();
37   AliAnalysisTaskFittingQDistribution(const char *name, Bool_t QAon = kFALSE, Bool_t useWeights=kFALSE);
38   virtual ~AliAnalysisTaskFittingQDistribution(){}; 
39   
40   virtual void   ConnectInputData(Option_t *);
41   virtual void   CreateOutputObjects();
42   virtual void   Exec(Option_t *option);
43   virtual void   Terminate(Option_t *);
44   
45   void           SetAnalysisType(TString type) {this->fAnalysisType = type;}
46   TString GetAnalysisType() const {return this->fAnalysisType;}
47   
48   void SetCFManager1(AliCFManager* cfmgr) {this->fCFManager1 = cfmgr;} 
49   AliCFManager* GetCFManager1()           {return this->fCFManager1;}
50   void SetCFManager2(AliCFManager* cfmgr) {this->fCFManager2 = cfmgr;} 
51   AliCFManager* GetCFManager2()           {return this->fCFManager2;} 
52   void          SetQAList1(TList* list)   {this->fQAInt = list; }
53   TList*        GetQAList1()              {return this->fQAInt; }
54   void          SetQAList2(TList* list)   {this->fQADiff = list; }
55   TList*        GetQAList2()              {return this->fQADiff; }
56   void          SetQAOn(Bool_t kt)        {this->fQA = kt; }
57   Bool_t        GetQAOn()                 {return this->fQA; }
58   
59   void SetUsePhiWeights(Bool_t const uPhiW) {this->fUsePhiWeights = uPhiW;};
60   Bool_t GetUsePhiWeights() const {return this->fUsePhiWeights;};
61   void SetUsePtWeights(Bool_t const uPtW) {this->fUsePtWeights = uPtW;};
62   Bool_t GetUsePtWeights() const {return this->fUsePtWeights;};
63   void SetUseEtaWeights(Bool_t const uEtaW) {this->fUseEtaWeights = uEtaW;};
64   Bool_t GetUseEtaWeights() const {return this->fUseEtaWeights;};
65  
66  private:
67   AliAnalysisTaskFittingQDistribution(const AliAnalysisTaskFittingQDistribution& aatfqd);
68   AliAnalysisTaskFittingQDistribution& operator=(const AliAnalysisTaskFittingQDistribution& aatfqd);
69
70   AliESDEvent *fESD;                      // ESD object
71   AliAODEvent* fAOD;                      // AOD object
72   AliFittingQDistribution* fFQDA;         // Fitting q-distribution Analysis (FQDA) object
73   AliFlowEventSimpleMaker* fEventMaker;   // FlowEventSimple maker object
74   TString fAnalysisType;                  // string to select which kind of input to analyse (ESD, AOD or MC)
75   AliCFManager* fCFManager1;              // correction framework manager
76   AliCFManager* fCFManager2;              // correction framework manager
77   TList  *fListHistos;                    // collection of output 
78      
79   TList*        fQAInt;                   // QA histogram list
80   TList*        fQADiff;                  // QA histogram list
81
82   Bool_t fQA;                             // flag to set the filling of the QA histograms  
83   
84   Bool_t       fUseWeights;               // use any weights
85   Bool_t       fUsePhiWeights;            // phi weights
86   Bool_t       fUsePtWeights;             // v_2(pt) weights
87   Bool_t       fUseEtaWeights;            // v_2(eta) weights  
88   TList*       fListWeights;              // list with weights                                               
89                                                            
90   ClassDef(AliAnalysisTaskFittingQDistribution, 1); 
91 };
92
93 //================================================================================================================
94
95 #endif
96
97
98
99
100
101
102
103
104
105
106