From c926fac941603891c7bd07ca6e6ddff96a2f3ecd Mon Sep 17 00:00:00 2001
From: snelling <snelling@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Tue, 20 Mar 2012 08:06:10 +0000
Subject: [PATCH] patch to use pt differential flow at the task level

---
 PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.cxx | 8 +++++++-
 PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.h   | 4 ++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.cxx b/PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.cxx
index 26931200cb0..7710c7b026c 100644
--- a/PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.cxx
+++ b/PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.cxx
@@ -30,6 +30,7 @@
 #include "TTree.h"
 #include "TFile.h" //needed as include
 #include "TList.h"
+#include "TF1.h"
 #include "TH2F.h"
 #include "TRandom3.h"
 #include "TTimeStamp.h"
@@ -127,6 +128,7 @@ AliAnalysisTaskFlowEvent::AliAnalysisTaskFlowEvent() :
   fV3(0.),
   fV4(0.),
   fV5(0.),
+  fDifferentialV2(0),
   fFlowEvent(NULL),
   fMyTRandom3(NULL)
 {
@@ -186,6 +188,7 @@ AliAnalysisTaskFlowEvent::AliAnalysisTaskFlowEvent(const char *name, TString RPt
   fV3(0.),
   fV4(0.),
   fV5(0.),
+  fDifferentialV2(0),
   fFlowEvent(NULL),
   fMyTRandom3(NULL)
 {
@@ -532,7 +535,10 @@ void AliAnalysisTaskFlowEvent::UserExec(Option_t *)
     if (!fFlowEvent->IsSetMCReactionPlaneAngle())
       fFlowEvent->SetMCReactionPlaneAngle(gRandom->Uniform(0.0,TMath::TwoPi()));
 
-    fFlowEvent->AddFlow(fV1,fV2,fV3,fV4,fV5);     //add flow
+    if(fDifferentialV2)
+      fFlowEvent->AddV2(fDifferentialV2);
+    else 
+      fFlowEvent->AddFlow(fV1,fV2,fV3,fV4,fV5);     //add flow
     fFlowEvent->CloneTracks(fNonFlowNumberOfTrackClones); //add nonflow by cloning tracks
   }
   //////////////////////////////////////////////////////////////////////////////
diff --git a/PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.h b/PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.h
index d8526ef29b3..9c78a7740c3 100644
--- a/PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.h
+++ b/PWG/FLOW/Tasks/AliAnalysisTaskFlowEvent.h
@@ -19,6 +19,7 @@ class AliFlowTrackCuts;
 class AliFlowEventSimpleMaker;
 class AliFlowEvent;
 class TList;
+class TF1;
 class TRandom3;
 class AliAnalysisTaskSE;
 class TString;
@@ -99,6 +100,8 @@ class AliAnalysisTaskFlowEvent : public AliAnalysisTaskSE {
   // setters for adding by hand flow values (afterburner)
   void SetAfterburnerOn(Bool_t b=kTRUE) {fAfterburnerOn=b;}
   void SetNonFlowNumberOfTrackClones(Int_t n) {fNonFlowNumberOfTrackClones=n;}
+  void SetPtDifferentialV2( TF1 *gPtV2) {
+    fDifferentialV2 = gPtV2;}
   void SetFlow( Double_t v1, Double_t v2, Double_t v3=0.0, Double_t v4=0.0, Double_t v5=0.0)
                {fV1=v1;fV2=v2;fV3=v3;fV4=v4;fV5=v5;}
   // end setters afterburner
@@ -171,6 +174,7 @@ class AliAnalysisTaskFlowEvent : public AliAnalysisTaskSE {
   Double_t  fV3;        // Add Flow. Must be in range [0,0.5].
   Double_t  fV4;        // Add Flow. Must be in range [0,0.5].
   Double_t  fV5;        // Add Flow. Must be in range [0,0.5].
+  TF1 *fDifferentialV2; // pt-differential v2
 
   AliFlowEvent* fFlowEvent; //flowevent
     
-- 
2.43.5