First implementation of phi, pt and eta weights for the Q vector

[u/mrichter/AliRoot.git] / PWG2 / FLOW / AliFlowEventSimpleMaker.h

diff --git a/PWG2/FLOW/AliFlowEventSimpleMaker.h b/PWG2/FLOW/AliFlowEventSimpleMaker.h
index 26c6da73c08711a0ca33edfc7e3596c524903b3c..1d1af7b852d2d985dc13be298d8e48aef1e3e589 100644 (file)
--- a/PWG2/FLOW/AliFlowEventSimpleMaker.h
+++ b/PWG2/FLOW/AliFlowEventSimpleMaker.h
@@ -18,7 +18,6 @@ class AliMCEvent;
 class AliESDEvent;
 class AliAODEvent;
 
-
 // AliFlowEventSimpleMaker:
 // Class to fill the AliFlowEventSimple with AliFlowTrackSimple objects
 // author: N. van der Kolk (kolk@nikhef.nl)
@@ -28,7 +27,19 @@ class AliFlowEventSimpleMaker {
  public:
 
   AliFlowEventSimpleMaker();             //constructor
-  virtual  ~AliFlowEventSimpleMaker();   //destructor
+  virtual ~AliFlowEventSimpleMaker();    //destructor
+  
+  virtual void Init(TFile *file);
+  
+  //setters and getters
+  void SetUseWeightsPhi(Bool_t const uwPhi) { this->fUseWeightsPhi = uwPhi;};
+  Bool_t GetUseWeightsPhi() const { return this->fUseWeightsPhi;};
+  
+  void SetUseWeightsPt(Bool_t const uwPt) { this->fUseWeightsPt = uwPt;};
+  Bool_t GetUseWeightsPt() const { return this->fUseWeightsPt;};
+  
+  void SetUseWeightsEta(Bool_t const uwEta) { this->fUseWeightsEta = uwEta;};
+  Bool_t GetUseWeightsEta() const { return this->fUseWeightsEta;};
   
   //TTree
   AliFlowEventSimple* FillTracks(TTree* anInput, AliFlowTrackSimpleCuts* intCuts, AliFlowTrackSimpleCuts* diffCuts);   //use own cut class
@@ -50,6 +61,13 @@ class AliFlowEventSimpleMaker {
  private:
   AliFlowEventSimpleMaker(const AliFlowEventSimpleMaker& anAnalysis);            //copy constructor
   AliFlowEventSimpleMaker& operator=(const AliFlowEventSimpleMaker& anAnalysis); //assignment operator
+ 
+  Bool_t fUseWeightsPhi;       // phi weights
+  Bool_t fUseWeightsPt;        // v_2(pt) weights
+  Bool_t fUseWeightsEta;       // v_2(eta) weights   
+  TH1F *fPhiWeights;           // histogram with phi weights
+  TH1D *fPtWeights;            // histogram with v_2(pt) weights   
+  TH1D *fEtaWeights;           // histogram with v_2(eta) weights
           
   ClassDef(AliFlowEventSimpleMaker,0)    // macro for rootcint
 };