]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWG2/FLOW/AliFlowTasks/AliAnalysisTaskQCumulants.h
1.) QC code runs now even faster
[u/mrichter/AliRoot.git] / PWG2 / FLOW / AliFlowTasks / AliAnalysisTaskQCumulants.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 Q-cumulants      * 
9  *                                    * 
10  * authors: Naomi van der Kolk        *
11  *           (kolk@nikhef.nl)         *  
12  *          Raimond Snellings         *
13  *           (snelling@nikhef.nl)     * 
14  *          Ante Bilandzic            *
15  *           (anteb@nikhef.nl)        * 
16  * ***********************************/
17
18 #ifndef ALIANALYSISTASKQCUMULANTS_H
19 #define ALIANALYSISTASKQCUMULANTS_H
20
21 #include "TString.h"
22 #include "AliAnalysisTaskSE.h"
23
24 class TList;
25 class AliFlowEventSimple;
26 class AliFlowAnalysisWithQCumulants;
27
28 //================================================================================================================
29
30 class AliAnalysisTaskQCumulants : public AliAnalysisTaskSE{
31  public:
32   AliAnalysisTaskQCumulants();
33   AliAnalysisTaskQCumulants(const char *name, Bool_t useParticleWeights=kFALSE);
34   virtual ~AliAnalysisTaskQCumulants(){}; 
35   
36   virtual void UserCreateOutputObjects();
37   virtual void UserExec(Option_t *option);
38   virtual void Terminate(Option_t *);
39   
40   // Common:
41   void SetFillMultipleControlHistograms(Bool_t const fmch) {this->fFillMultipleControlHistograms = fmch;};
42   Bool_t GetFillMultipleControlHistograms() const {return this->fFillMultipleControlHistograms;}; 
43   void SetHarmonic(Int_t const harmonic) {this->fHarmonic = harmonic;};
44   Int_t GetHarmonic() const {return this->fHarmonic;};
45   void SetApplyCorrectionForNUA(Bool_t const applyCorrectionForNUA) {this->fApplyCorrectionForNUA = applyCorrectionForNUA;};
46   Bool_t GetApplyCorrectionForNUA() const {return this->fApplyCorrectionForNUA;};
47   void SetApplyCorrectionForNUAVsM(Bool_t const applyCorrectionForNUAVsM) {this->fApplyCorrectionForNUAVsM = applyCorrectionForNUAVsM;};
48   Bool_t GetApplyCorrectionForNUAVsM() const {return this->fApplyCorrectionForNUAVsM;};      
49   void SetPropagateErrorAlsoFromNIT(Bool_t const peafNIT) {this->fPropagateErrorAlsoFromNIT = peafNIT;};
50   Bool_t GetPropagateErrorAlsoFromNIT() const {return this->fPropagateErrorAlsoFromNIT;};  
51   void SetCalculateDiffFlow(Bool_t const calculateDiffFlow) {this->fCalculateDiffFlow = calculateDiffFlow;};
52   Bool_t GetCalculateDiffFlow() const {return this->fCalculateDiffFlow;};
53   void SetCalculate2DDiffFlow(Bool_t const calculate2DDiffFlow) {this->fCalculate2DDiffFlow = calculate2DDiffFlow;};
54   Bool_t GetCalculate2DDiffFlow() const {return this->fCalculate2DDiffFlow;};
55   void SetStoreDistributions(Bool_t const storeDistributions) {this->fStoreDistributions = storeDistributions;};
56   Bool_t GetStoreDistributions() const {return this->fStoreDistributions;};
57   void SetCalculateCumulantsVsM(Bool_t const ccvm) {this->fCalculateCumulantsVsM = ccvm;};
58   Bool_t GetCalculateCumulantsVsM() const {return this->fCalculateCumulantsVsM;};
59   void SetMinimumBiasReferenceFlow(Bool_t const mmrf) {this->fMinimumBiasReferenceFlow = mmrf;};
60   Bool_t GetMinimumBiasReferenceFlow() const {return this->fMinimumBiasReferenceFlow;};     
61   void SetForgetAboutCovariances(Bool_t const fac) {this->fForgetAboutCovariances = fac;};
62   Bool_t GetForgetAboutCovariances() const {return this->fForgetAboutCovariances;};   
63   void SetStorePhiDistributionForOneEvent(Bool_t const spdfoe) {this->fStorePhiDistributionForOneEvent = spdfoe;};
64   Bool_t GetStorePhiDistributionForOneEvent() const {return this->fStorePhiDistributionForOneEvent;};
65   void SetPhiDistributionForOneEventSettings(Double_t const pdfoes, Int_t const i) {this->fPhiDistributionForOneEventSettings[i] = pdfoes;};
66   Double_t GetPhiDistributionForOneEventSettings(Int_t const i) const {return this->fPhiDistributionForOneEventSettings[i];};  
67  
68   // Multiparticle correlations vs multiplicity:
69   void SetnBinsMult(Int_t const nbm) {this->fnBinsMult = nbm;};
70   Int_t GetnBinsMult() const {return this->fnBinsMult;};  
71   void SetMinMult(Double_t const minm) {this->fMinMult = minm;};
72   Double_t GetMinMult() const {return this->fMinMult;};
73   void SetMaxMult(Double_t const maxm) {this->fMaxMult = maxm;};
74   Double_t GetMaxMult() const {return this->fMaxMult;};
75   // Particle weights:
76   void SetUsePhiWeights(Bool_t const uPhiW) {this->fUsePhiWeights = uPhiW;};
77   Bool_t GetUsePhiWeights() const {return this->fUsePhiWeights;};
78   void SetUsePtWeights(Bool_t const uPtW) {this->fUsePtWeights = uPtW;};
79   Bool_t GetUsePtWeights() const {return this->fUsePtWeights;};
80   void SetUseEtaWeights(Bool_t const uEtaW) {this->fUseEtaWeights = uEtaW;};
81   Bool_t GetUseEtaWeights() const {return this->fUseEtaWeights;};
82   // Event weights:
83   void SetMultiplicityWeight(const char *multiplicityWeight) {*this->fMultiplicityWeight = multiplicityWeight;};
84  
85  private:
86   AliAnalysisTaskQCumulants(const AliAnalysisTaskQCumulants& aatqc);
87   AliAnalysisTaskQCumulants& operator=(const AliAnalysisTaskQCumulants& aatqc);
88   
89   AliFlowEventSimple *fEvent;         // the input event
90   AliFlowAnalysisWithQCumulants *fQC; // Q-cumulant object
91   TList *fListHistos;                 // collection of output 
92   // Common:
93   Bool_t fFillMultipleControlHistograms; // fill separately control histos for events with >= 2, 4, 6 and 8 particles
94   Int_t fHarmonic;                       // harmonic  
95   Bool_t fApplyCorrectionForNUA;         // apply correction for non-uniform acceptance 
96   Bool_t fApplyCorrectionForNUAVsM;      // apply correction for non-uniform acceptance versus M    
97   Bool_t fPropagateErrorAlsoFromNIT;     // propagate error by taking into account also non-isotrpic terms  
98   Bool_t fCalculateDiffFlow;             // calculate differential flow in pt or eta
99   Bool_t fCalculate2DDiffFlow;           // calculate differential flow in (pt,eta) (Remark: this is very expensive in terms of CPU time)
100   Bool_t fStoreDistributions;            // store or not distributions of correlations
101   Bool_t fCalculateCumulantsVsM;         // calculate cumulants versus multiplicity  
102   Bool_t fMinimumBiasReferenceFlow;      // store as reference flow in AliFlowCommonHistResults the minimum bias result (kFALSE by default)     
103   Bool_t fForgetAboutCovariances;        // when propagating error forget about the covariances  
104   Bool_t fStorePhiDistributionForOneEvent; // store phi distribution for one event to illustrate flow
105   Double_t fPhiDistributionForOneEventSettings[4]; // [v_min,v_max,refMult_min,refMult_max]        
106   // Multiparticle correlations vs multiplicity:
107   Int_t fnBinsMult;                   // number of multiplicity bins for flow analysis versus multiplicity  
108   Double_t fMinMult;                  // minimal multiplicity for flow analysis versus multiplicity  
109   Double_t fMaxMult;                  // maximal multiplicity for flow analysis versus multiplicity    
110   // Particle weights:
111   Bool_t fUseParticleWeights;         // use any particle weights
112   Bool_t fUsePhiWeights;              // use phi weights
113   Bool_t fUsePtWeights;               // use pt weights
114   Bool_t fUseEtaWeights;              // use eta weights  
115   TList *fWeightsList;                // list with weights
116   // Event weights:
117   TString *fMultiplicityWeight;       // event-by-event weights for multiparticle correlations ("combinations","unit" or "multiplicity")  
118   
119   ClassDef(AliAnalysisTaskQCumulants, 1); 
120 };
121
122 //================================================================================================================
123
124 #endif
125
126
127
128
129
130
131
132
133
134
135