adding some drawing macros for balance functions

[u/mrichter/AliRoot.git] / PWGCF / EBYE / macros / runBalanceFunctionPsi.C

diff --git a/PWGCF/EBYE/macros/runBalanceFunctionPsi.C b/PWGCF/EBYE/macros/runBalanceFunctionPsi.C
old mode 100755 (executable)
new mode 100644 (file)
index 4d927ff..748e99e
--- a/PWGCF/EBYE/macros/runBalanceFunctionPsi.C
+++ b/PWGCF/EBYE/macros/runBalanceFunctionPsi.C
@@ -21,11 +21,13 @@ Float_t vZ[numberOfSyst]     = {10.,12.,6.,8.,10.,10.,10.,10.,10.,10.,10.,10.,10
 Float_t DCAxy[numberOfSyst]  = {-1.,2.4,2.4,2.4,2.2,2.0,1.8,2.4,2.4,2.4,2.4,2.4,2.4};   // DCA xy cut (afterburner, -1 = w/o additional cut)\r
 Float_t DCAz[numberOfSyst]   = {-1.,3.2,3.2,3.2,3.0,2.8,2.6,3.2,3.2,3.2,3.2,3.2,3.2};   // DCA z cut (afterburner, -1 = w/o additional cut)\r
 Float_t ptMin[numberOfSyst]  = {0.3,0.3,0.3,0.3,0.3,0.3,0.3,1.5,5.0,0.3,0.3,0.3,0.3};   // pt cuts\r
-Float_t ptMax[numberOfSyst]  = {1.5,1.5,1.5,1.5,1.5,1.5,1.5,5.0,10.0,10.0,1.5,1.5,1.5}; // pt cuts\r
+Float_t ptMax[numberOfSyst]  = {5.,1.5,1.5,1.5,1.5,1.5,1.5,5.0,10.0,10.0,1.5,1.5,1.5}; // pt cuts\r
 Float_t etaMin[numberOfSyst] = {-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-1.0,-0.6,-0.4}; // eta cuts\r
 Float_t etaMax[numberOfSyst] = {0.8,0.8,0.8,0.8,0.8,0.8,0.8,0.8,0.8,0.8,1.0,0.6,0.4};   // eta cuts\r
 \r
 Bool_t kUsePID = kFALSE;\r
+Bool_t kUseHBTCut = kFALSE;\r
+Bool_t kUseConversionCut = kFALSE;\r
 \r
 //______________________________________________________________________________\r
 void runBalanceFunctionPsi(\r
@@ -43,7 +45,7 @@ void runBalanceFunctionPsi(
          )\r
 {\r
     // check run type\r
-    if(runtype != "local" && runtype != "proof" && runtype != "grid"){\r
+    if(runtype != "local" && runtype != "proof" && runtype != "grid") {\r
         Printf("\n\tIncorrect run option, check first argument of run macro");\r
         Printf("\tint runtype = local, proof or grid\n");\r
         return;\r
@@ -61,6 +63,9 @@ void runBalanceFunctionPsi(
     gSystem->Load("libAOD.so");\r
     gSystem->Load("libANALYSIS.so");\r
     gSystem->Load("libANALYSISalice.so");\r
+    gSystem->Load("libEventMixing.so");\r
+    gSystem->Load("libCORRFW.so");\r
+    gSystem->Load("libPWGTools.so");\r
     gSystem->Load("libPWGCFebye.so");\r
 \r
     // additional\r
@@ -79,9 +84,35 @@ void runBalanceFunctionPsi(
     AliAnalysisManager* mgr = new AliAnalysisManager(Form("%s%i",taskname,bunchN));\r
     \r
     // create the alien handler and attach it to the manager\r
-    AliAnalysisGrid *plugin = CreateAlienHandler(bAOD,bunchN,Form("%s%i",taskname,bunchN), gridmode, proofcluster, Form("%s_%d.txt",proofdataset.Data(),bunchN)); \r
-    mgr->SetGridHandler(plugin);\r
-    \r
+    if(runtype == "grid") {\r
+      AliAnalysisGrid *plugin = CreateAlienHandler(bAOD,bunchN,Form("%s%i",taskname,bunchN), gridmode, proofcluster, Form("%s_%d.txt",proofdataset.Data(),bunchN)); \r
+      mgr->SetGridHandler(plugin);\r
+    }\r
+    else if(runtype == "local") {\r
+      TString filename;\r
+      TChain* chain = 0x0;\r
+      if((!bAOD)&&(!bMCtruth)) {\r
+       chain = new TChain("esdTree");\r
+       for(Int_t i = 0; i < 4; i++) {\r
+         filename = "/data/alice2/pchrist/HeavyIons/Data/2011/Set";\r
+         filename += i; filename += "/AliESDs.root";\r
+         chain->Add(filename.Data());\r
+       }\r
+      }\r
+      else if((bAOD)&&(!bMCtruth)) {\r
+       chain = new TChain("aodTree");\r
+       for(Int_t i = 0; i < 4; i++) {\r
+         filename = "/project/alice/pchrist/Data/2011/Set";\r
+         filename += i; filename += "/AliAOD.root";\r
+         chain->Add(filename.Data());\r
+       }\r
+      }\r
+      else if(bMCtruth) {\r
+       chain = new TChain("TE");\r
+       filename = "galice.root";\r
+       chain->Add(filename.Data());\r
+      }\r
+    }//local mode\r
 \r
     // input handler (ESD or AOD)\r
     AliVEventHandler* inputH = NULL;\r
@@ -95,12 +126,12 @@ void runBalanceFunctionPsi(
     \r
     // mc event handler\r
     if(bMCtruth) {\r
-        AliMCEventHandler* mchandler = new AliMCEventHandler();\r
-        // Not reading track references\r
-        mchandler->SetReadTR(kFALSE);\r
-        mgr->SetMCtruthEventHandler(mchandler);\r
+      AliMCEventHandler* mchandler = new AliMCEventHandler();\r
+      // Not reading track references\r
+      mchandler->SetReadTR(kFALSE);\r
+      mgr->SetMCtruthEventHandler(mchandler);\r
     }   \r
-\r
+    \r
     // AOD output handler\r
     //AliAODHandler* aodoutHandler = new AliAODHandler();\r
     //aodoutHandler->SetOutputFileName("aod.root");\r
@@ -167,7 +198,7 @@ void runBalanceFunctionPsi(
 \r
     //Add the BF task (all centralities)\r
     gROOT->LoadMacro("AddTaskBalancePsiCentralityTrain.C"); \r
-    AliAnalysisTaskBFPsi *task = AddTaskBalancePsiCentralityTrain(0,100,0,"V0M",vZ[0],DCAxy[0],DCAz[0],ptMin[0],ptMax[0],etaMin[0],etaMax[0],-1,-1,kUsePID);\r
+    AliAnalysisTaskBFPsi *task = AddTaskBalancePsiCentralityTrain(0,100,1,"V0M",vZ[0],DCAxy[0],DCAz[0],ptMin[0],ptMax[0],etaMin[0],etaMax[0],-1,-1,kUseHBTCut,kUseConversionCut,kUsePID);\r
     \r
     // enable debug printouts\r
     //mgr->SetDebugLevel(2);\r
@@ -177,7 +208,10 @@ void runBalanceFunctionPsi(
   \r
     // start analysis\r
     Printf("Starting Analysis....");\r
-    mgr->StartAnalysis(runtype,nentries,firstentry);\r
+    if(runtype == "local") \r
+      mgr->StartAnalysis("local",chain);\r
+    else\r
+      mgr->StartAnalysis(runtype,nentries,firstentry);\r
 }\r
 \r
 //______________________________________________________________________________\r