added the possibility to use weights of the Q vector to FittingQDistributions

[u/mrichter/AliRoot.git] / PWG2 / FLOW / AliFlowCommon / AliFittingQDistribution.h

diff --git a/PWG2/FLOW/AliFlowCommon/AliFittingQDistribution.h b/PWG2/FLOW/AliFlowCommon/AliFittingQDistribution.h
index 6b6069c3e6ccccc9cbecb98a5803461699874724..61f5a2bca6c1b4a6d91a2c3d00d2d70ee7de13eb 100644 (file)
--- a/PWG2/FLOW/AliFlowCommon/AliFittingQDistribution.h
+++ b/PWG2/FLOW/AliFlowCommon/AliFittingQDistribution.h
@@ -37,7 +37,7 @@ class AliFittingQDistribution{
   AliFittingQDistribution();
   virtual ~AliFittingQDistribution(); 
   
-  virtual void CreateOutputObjects();
+  virtual void Init();
   virtual void Make(AliFlowEventSimple* anEvent);
   virtual void Finish();
   virtual void WriteHistograms(TString* outputFileName);
@@ -46,7 +46,10 @@ class AliFittingQDistribution{
 //----------------------------------------------------------------------------------------------------------------
 //                                            setters and getters                                                 
 //----------------------------------------------------------------------------------------------------------------      
-  TList* GetHistList() const {return this->fHistList;}      //output histogram list
+  TList* GetHistList() const {return this->fHistList;}
+  
+  void SetWeightsList(TList* wlist) {this->fWeightsList = wlist;}
+  TList* GetWeightsList() const {return this->fWeightsList;}   
   
   void SetIntFlowResults(TH1D* ifr)  {this->fIntFlowResultsFQD = ifr;};
   TH1D* GetIntFlowResults() const    {return this->fIntFlowResultsFQD;};
@@ -62,6 +65,15 @@ class AliFittingQDistribution{
   
   void SetSigma2(TH1D* s2)  {this->fSigma2 = s2;};
   TH1D* GetSigma2() const   {return this->fSigma2;};
+  
+  void SetUsePhiWeights(Bool_t const uPhiW) {this->fUsePhiWeights = uPhiW;};
+  Bool_t GetUsePhiWeights() const {return this->fUsePhiWeights;};
+  
+  void SetUsePtWeights(Bool_t const uPtW) {this->fUsePtWeights = uPtW;};
+  Bool_t GetUsePtWeights() const {return this->fUsePtWeights;};
+  
+  void SetUseEtaWeights(Bool_t const uEtaW) {this->fUseEtaWeights = uEtaW;};
+  Bool_t GetUseEtaWeights() const {return this->fUseEtaWeights;};
 //----------------------------------------------------------------------------------------------------------------
  
  private:
@@ -71,12 +83,18 @@ class AliFittingQDistribution{
   AliFlowTrackSimple*        fTrack;                   //track
    
   TList*                     fHistList;                //list to hold all output histograms
+  TList*                     fWeightsList;             //list to hold all histograms with weights
+  
   TProfile*                  fAvMultIntFlowFQD;        //avarage selected multiplicity
   TH1D*                      fIntFlowResultsFQD;       //integrated flow final results
   TH1D*                      fSigma2;                  //sigma^2
   AliFlowCommonHist*         fCommonHists;             //common control histograms
   AliFlowCommonHistResults*  fCommonHistsResults;      //final results for integrated flow stored in the common histograms 
   TH1D*                      fQDistributionFQD;        //q-distribution 
+  
+  Bool_t                     fUsePhiWeights;           //phi weights
+  Bool_t                     fUsePtWeights;            //v_2(pt) weights
+  Bool_t                     fUseEtaWeights;           //v_2(eta) weights
          
   ClassDef(AliFittingQDistribution, 0);
 };