From 8bd000efc03419957b4397595eb50c22e077ced9 Mon Sep 17 00:00:00 2001
From: gconesab <gconesab@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Tue, 8 May 2012 10:08:58 +0000
Subject: [PATCH] fix task to not depend on data type for standard cases, add
 simple counting task configuration

---
 .../macros/AddTaskCaloTrackCorr.C             | 18 ++--
 .../macros/AddTaskCounter.C                   | 83 +++++++++++++++++++
 2 files changed, 94 insertions(+), 7 deletions(-)
 create mode 100644 PWGGA/CaloTrackCorrelations/macros/AddTaskCounter.C

diff --git a/PWGGA/CaloTrackCorrelations/macros/AddTaskCaloTrackCorr.C b/PWGGA/CaloTrackCorrelations/macros/AddTaskCaloTrackCorr.C
index d2927d2855e..553a30f3a56 100644
--- a/PWGGA/CaloTrackCorrelations/macros/AddTaskCaloTrackCorr.C
+++ b/PWGGA/CaloTrackCorrelations/macros/AddTaskCaloTrackCorr.C
@@ -10,7 +10,7 @@ Int_t   kYears         = 2011;
 TString kCollisions    = "pp";
 TString kTrig          = "EMC7" ;
 TString kClusterArray  = "";
-TString kData          = "ESD";
+TString kData          = ""; // MC or deltaAOD
 TString kInputDataType = "ESD";
 TString kCalorimeter   = "EMCAL";
 Bool_t  kTM            = kTRUE;
@@ -25,7 +25,7 @@ Bool_t  kCalibE        = kTRUE;
 Bool_t  kCalibT        = kTRUE;
 Bool_t  kBadMap        = kTRUE;
 
-AliAnalysisTaskCaloTrackCorrelation *AddTaskCaloTrackCorr(const TString  data          = "AOD",
+AliAnalysisTaskCaloTrackCorrelation *AddTaskCaloTrackCorr(const TString  data          = "",
                                                           const TString  calorimeter   = "EMCAL", 
                                                           const Bool_t   simulation    = kFALSE,
                                                           const Bool_t   eventsel      = kFALSE,
@@ -303,10 +303,14 @@ AliCaloTrackReader * ConfigureReader()
 {
   
   AliCaloTrackReader * reader = 0;
-  if     (kData.Contains("AOD"))   reader = new AliCaloTrackAODReader();
-  else if(kData=="ESD")            reader = new AliCaloTrackESDReader();
-  else if(kData=="MC" && 
-          kInputDataType == "ESD") reader = new AliCaloTrackMCReader();
+  if     (kInputDataType == "ESD"&& kData=="MC" ) 
+    reader = new AliCaloTrackMCReader();
+  else if(kInputDataType=="AOD" || kData.Contains("AOD"))   
+    reader = new AliCaloTrackAODReader();
+  else if(kInputDataType=="ESD")            
+    reader = new AliCaloTrackESDReader();
+  else 
+    printf("AliCaloTrackReader::ConfigureReader() - Data combination not known kData=%s, kInputData=%s\n",kData.Data(),kInputDataType.Data());
   
   reader->SetDebug(kDebug);//10 for lots of messages
   
@@ -559,7 +563,7 @@ AliAnaPhoton* ConfigurePhotonAnalysis()
   //PHOS
   caloPID->SetPHOSDispersionCut(2.5);
   caloPID->SetPHOSRCut(2.);
-  if(kData=="AOD") caloPID->SetPHOSRCut(2000.); // Open cut since dX, dZ not stored
+  if(kInputData=="AOD") caloPID->SetPHOSRCut(2000.); // Open cut since dX, dZ not stored
       
   ana->SwitchOffFillShowerShapeHistograms();  // Filled before photon shower shape selection
   
diff --git a/PWGGA/CaloTrackCorrelations/macros/AddTaskCounter.C b/PWGGA/CaloTrackCorrelations/macros/AddTaskCounter.C
new file mode 100644
index 00000000000..0b6cd7d5972
--- /dev/null
+++ b/PWGGA/CaloTrackCorrelations/macros/AddTaskCounter.C
@@ -0,0 +1,83 @@
+//_______________________________________________
+void AddTaskCounter(const TString trigger = "MB")
+{
+  
+  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
+  
+  AliAnalysisTaskCounter * counter =  new AliAnalysisTaskCounter(Form("Counter%s",trigger.Data()));
+  //if(kRun > 140000 && kRun < 146900) counter ->RejectFastCluster();
+  //if     (kCollision=="pp"  )   counter->SetZVertexCut(10.);  //Open cut
+  //else if(kCollision=="PbPb")   counter->SetZVertexCut(10.);  //Centrality defined in this range.
+  
+  if(trigger=="EMC7")
+  {
+    printf("counter trigger EMC7\n");
+    counter->SelectCollisionCandidates(AliVEvent::kEMC7);
+  }
+  else if (trigger=="INT7")
+  {
+    printf("counter trigger INT7\n");
+    counter->SelectCollisionCandidates(AliVEvent::kINT7);
+  }
+  if(trigger=="EMC1")
+  {
+    printf("counter trigger EMC1\n");
+    counter->SelectCollisionCandidates(AliVEvent::kEMC1);
+  }
+  else if(trigger=="MB")
+  {
+    printf("counter trigger MB\n");
+    counter->SelectCollisionCandidates(AliVEvent::kMB);
+  }
+  else if(trigger=="PHOS")
+  {
+    printf("counter trigger PHOS\n");
+    counter->SelectCollisionCandidates(AliVEvent::kPHI7);
+  }
+  else if(trigger=="PHOSPb")
+  {
+    printf("counter trigger PHOSPb\n");
+    counter->SelectCollisionCandidates(AliVEvent::kPHOSPb);
+  }
+  else if(trigger=="AnyINT")
+  {
+    printf("counter trigger AnyINT\n");
+    counter->SelectCollisionCandidates(AliVEvent::kAnyINT);
+  }  
+  else if(trigger=="INT")
+  {
+    printf("counter trigger AnyINT\n");
+    counter->SelectCollisionCandidates(AliVEvent::kAny);
+  }
+  else if(trigger=="EMCEGA")
+  {
+    printf("counter trigger EMC Gamma\n");
+    counter->SelectCollisionCandidates(AliVEvent::kEMCEGA);
+  } 
+  else if(trigger=="EMCEJE")
+  {
+    printf("counter trigger EMC Jet\n");
+    counter->SelectCollisionCandidates(AliVEvent::kEMCEJE);
+  }
+  else if(trigger=="Central")
+  {
+    printf("counter trigger Central\n");
+    counter->SelectCollisionCandidates(AliVEvent::kCentral);
+  } 
+  else if(trigger=="SemiCentral")
+  {
+    printf("counter trigger SemiCentral\n");
+    counter->SelectCollisionCandidates(AliVEvent::kSemiCentral);
+  }
+  
+  TString outputFile = AliAnalysisManager::GetCommonFileName(); 
+  AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
+  
+  AliAnalysisDataContainer *coutput = 
+  mgr->CreateContainer(Form("Counter%s",trigger.Data()), TList::Class(), AliAnalysisManager::kOutputContainer,  outputFile.Data());
+  mgr->AddTask(counter);
+  mgr->ConnectInput  (counter, 0, cinput1);
+  mgr->ConnectOutput (counter, 1, coutput);
+  
+}
+
-- 
2.43.0