add tasks for jet like cluster analysis

diff --git a/PWG4/macros/AddTaskJetCluster.C b/PWG4/macros/AddTaskJetCluster.C
new file mode 100644 (file)
index 0000000..c8704c7
--- /dev/null
+++ b/PWG4/macros/AddTaskJetCluster.C
@@ -0,0 +1,135 @@
+AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec = "AOD",char* bGen = "",UInt_t filterMask = 16, Int_t iPhysicsSelection = 1,Char_t *jf = "KT", Float_t radius = 0.4);\r
+\r
+\r
+AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,Int_t iPhysicsSelection = 1,Char_t *jf = "KT", UInt_t iFlag){\r
+  AliAnalysisTaskJetCluster *js = 0;\r
+  if(kUseAODMC){\r
+    if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.00001);\r
+    if(iFlag&(2<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.1);\r
+    if(iFlag&(3<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.2);\r
+    if(iFlag&(4<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.4);\r
+    if(iFlag&(5<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.6);\r
+    if(iFlag&(6<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.8);\r
+    if(iFlag&(7<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,1.0);\r
+  }\r
+  else{\r
+    if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.00001);\r
+    if(iFlag&(2<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.1);\r
+    if(iFlag&(3<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.2);\r
+    //    if(iFlag&(4<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.4);\r
+    if(iFlag&(5<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.6);\r
+    if(iFlag&(6<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.8);\r
+    if(iFlag&(7<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,1.0);\r
+  }\r
+  \r
+  return js;\r
+}\r
+\r
+\r
+AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filterMask,Int_t iPhysicsSelection,Char_t *jf,Float_t radius)\r
+{\r
+// Creates a jet fider task, configures it and adds it to the analysis manager.\r
+\r
+   // Get the pointer to the existing analysis manager via the static access method.\r
+   //==============================================================================\r
+   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
+   if (!mgr) {\r
+      ::Error("AddTaskJetCluster", "No analysis manager to connect to.");\r
+      return NULL;\r
+   }  \r
+   \r
+   // Check the analysis type using the event handlers connected to the analysis manager.\r
+   //==============================================================================\r
+   if (!mgr->GetInputEventHandler()) {\r
+     ::Error("AddTaskJetCluster", "This task requires an input event handler");\r
+      return NULL;\r
+   }\r
+\r
+   TString type = mgr->GetInputEventHandler()->GetDataType();\r
+   TString typeRec(bRec);\r
+   TString typeGen(bGen);\r
+   typeGen.ToUpper();\r
+   typeRec.ToUpper();\r
+   // Create the task and configure it.\r
+   //===========================================================================\r
+   \r
+   AliAnalysisTaskJetCluster* pwg4spec = new  AliAnalysisTaskJetCluster(Form("Jet Spectrum %s %s",bRec,bGen));\r
+      \r
+   // or a config file\r
+   // pwg4spec->SetAnalysisType(AliAnalysisTaskJetCluster::kAnaMC);\r
+   // if(iAODanalysis)pwg4spec->SetAODInput(kTRUE);\r
+   // pwg4spec->SetDebugLevel(11); \r
+   pwg4spec->SetFilterMask(filterMask); \r
+   //   pwg4spec->SetUseGlobalSelection(kTRUE); \r
+\r
+   if(type == "AOD"){\r
+     // Assume all jet are produced already\r
+     pwg4spec->SetAODJetInput(kTRUE);\r
+     pwg4spec->SetAODTrackInput(kTRUE);\r
+     pwg4spec->SetAODMCInput(kTRUE);\r
+   }\r
+\r
+   if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance);\r
+   }\r
+   else if (typeRec.Contains("AODMC2")){\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCCharged);\r
+   }\r
+   else if (typeRec.Contains("AODMC")){\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCAll);\r
+   }\r
+   else if (typeRec.Contains("AOD")) {\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAOD);\r
+   }\r
+\r
+   if(typeGen.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance);\r
+   }\r
+   else if (typeGen.Contains("AODMC2")){\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetCluster::kTrackAODMCCharged);\r
+   }\r
+   else if (typeGen.Contains("AODMC")){\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetCluster::kTrackAODMCAll);\r
+   }\r
+   else if (typeGen.Contains("AOD")) {\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetCluster::kTrackAOD);\r
+   }\r
+\r
+   char *cRadius = "";\r
+   if(radius>0)cRadius = Form("%02d",(int)((radius+0.01)*10.));\r
+\r
+\r
+\r
+   pwg4spec->SetRparam(radius);\r
+\r
+   switch (jf) {\r
+   case "ANTIKT":\r
+     pwg4spec->SetAlgorithm(2); // antikt from fastjet/JetDefinition.hh\r
+     break;\r
+   case "CA":\r
+     pwg4spec->SetAlgorithm(1); // kt from fastjet/JetDefinition.hh\r
+     break;\r
+   case "KT":\r
+     pwg4spec->SetAlgorithm(0); // kt from fastjet/JetDefinition.hh\r
+     break;\r
+   default:\r
+     ::Error("AddTaskJetCluster", "Wrong jet finder selected\n");\r
+     return 0;\r
+   }\r
+\r
+   \r
+   if(iPhysicsSelection)pwg4spec->SelectCollisionCandidates();\r
+\r
+   mgr->AddTask(pwg4spec);\r
+     \r
+   // Create ONLY the output containers for the data produced by the task.\r
+   // Get and connect other common input/output containers via the manager as below\r
+   //==============================================================================\r
+   AliAnalysisDataContainer *coutput1_Spec = mgr->CreateContainer(Form("pwg4cluster_%s_%s_%s_%s",bRec,bGen,jf,cRadius), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_cluster_%s_%s_%s_%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen,jf,cRadius));\r
+\r
+   mgr->ConnectInput  (pwg4spec, 0, mgr->GetCommonInputContainer());\r
+   mgr->ConnectOutput (pwg4spec, 0, mgr->GetCommonOutputContainer());\r
+   mgr->ConnectOutput (pwg4spec,  1, coutput1_Spec );\r
+   \r
+   return pwg4spec;\r
+}\r