]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWG4/macros/AddTaskJetCluster.C
updates for differnt jet configs (including background), reduced number of UA1 Jets
[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskJetCluster.C
index 2638bbffaba9e3f821f88f242140b99de3914ecd..5520747012950415d88a315b2e13585402831ea9 100644 (file)
@@ -1,64 +1,73 @@
-AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec = "AOD",char* bGen = "",UInt_t filterMask = 16, Int_t iPhysicsSelection = 1,Char_t *jf = "KT", Float_t radius = 0.4,Int_t nSkip = 0);\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&&false){// do not use the MC info yet\r
-    if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.00001);\r
-    if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.1);\r
-    if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.2);\r
-    if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.4);\r
-    if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.6);\r
-    if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelection,jf,0.8);\r
-    if(iFlag&(1<<10))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&(1<<1))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.1);\r
-    if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.2);\r
-    if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.4);\r
-    if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.6);\r
-    if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelection,jf,0.8);\r
-    if(iFlag&(1<<10))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,Int_t nSkip)\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
+ AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec = "AOD",char* bGen = "",UInt_t filterMask = 16, UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t *jf = "KT", Float_t radius = 0.4,Int_t nSkip = 0,Int_t kWriteAOD = kFALSE,char* deltaFile = "",Float_t ptTrackCut = 0.15);\r
+\r
+ Int_t kBackGroundMode = 0;\r
+ Float_t kPtTrackCut = 0.15;\r
+\r
+ AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t *jf = "KT", UInt_t iFlag){\r
+   AliAnalysisTaskJetCluster *js = 0;\r
+   if(kUseAODMC&&false){// do not use the MC info yet\r
+     if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.00001);\r
+     if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.1);\r
+     if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.2);\r
+     if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.4);\r
+     if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.6);\r
+     if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.8);\r
+     if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,1.0);\r
+   }\r
+   else{\r
+     if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.00001);\r
+     if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.1);\r
+     if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.2);\r
+     if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.4);\r
+     if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.6);\r
+     if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.8);\r
+     if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,1.0);\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
-\r
-   char *cRadius = "";\r
-   if(radius>0)cRadius = Form("%02d",(int)((radius+0.01)*10.));\r
-\r
-   \r
-   AliAnalysisTaskJetCluster* pwg4spec = new  AliAnalysisTaskJetCluster(Form("JetCluster_%s_%s_skip%d",jf,cRadius,nSkip));\r
+   return js;\r
+ }\r
+\r
+\r
+ AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filterMask,UInt_t iPhysicsSelectionFlag,Char_t *jf,Float_t radius,Int_t nSkip,Int_t kWriteAOD,char *deltaFile,Float_t ptTrackCut)\r
+ {\r
+ // Creates a jet fider task, configures it and adds it to the analysis manager.\r
+   kPtTrackCut = ptTrackCut;\r
+\r
+   TString outputFile(deltaFile);\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
+\r
+\r
+\r
+    TString cAdd = "";\r
+    cAdd += Form("%02d_",(int)((radius+0.01)*10.));\r
+    cAdd += Form("B%d",(int)kBackgroundMode);\r
+    cAdd += Form("_Filter%05d",filterMask);\r
+    cAdd += Form("_Cut%05d",(int)(1000.*kPtTrackCut));\r
+    cAdd += Form("_Skip%02d",nSkip);\r
+    Printf("%s %E",cAdd.Data(),kPtTrackCut);\r
+    AliAnalysisTaskJetCluster* pwg4spec = new  AliAnalysisTaskJetCluster(Form("JetCluster_%s%s",jf,cAdd.Data()));\r
       \r
    // or a config file\r
    // pwg4spec->SetAnalysisType(AliAnalysisTaskJetCluster::kAnaMC);\r
@@ -84,7 +93,7 @@ AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filte
    }\r
    else if (typeRec.Contains("AOD")) {\r
      pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAOD);\r
-     pwg4spec->SetTrackPtCut(0.15);\r
+     pwg4spec->SetTrackPtCut(kPtTrackCut);\r
    }\r
 \r
 \r
@@ -121,21 +130,22 @@ AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filte
    }\r
 \r
    \r
-   if(TMath::Abs(radius-0.4)<0.01){\r
-     pwg4spec->SetJetOutputBranch(Form("clusters%s_%s%s",bRec,jf,cRadius));\r
-     pwg4spec->SetJetOutputMinPt(1); // store only jets / clusters above a certain threshold\r
+   if(kWriteAOD){\r
+     if(outputFile.Length())pwg4spec->SetJetOutputFile(outputFile);\r
+     pwg4spec->SetJetOutputBranch(Form("clusters%s_%s%s",bRec,jf,cAdd.Data()));\r
+     pwg4spec->SetJetOutputMinPt(0); // store only jets / clusters above a certain threshold\r
    }\r
 \r
    pwg4spec->SetNSkipLeadingRan(nSkip);\r
 \r
-   if(iPhysicsSelection)pwg4spec->SelectCollisionCandidates();\r
+   if(iPhysicsSelectionFlag)pwg4spec->SelectCollisionCandidates(iPhysicsSelectionFlag);\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_skip%d",bRec,bGen,jf,cRadius,nSkip), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_cluster_%s_%s_%s_%s_skip%d",AliAnalysisManager::GetCommonFileName(),bRec,bGen,jf,cRadius,nSkip));\r
+   AliAnalysisDataContainer *coutput1_Spec = mgr->CreateContainer(Form("pwg4cluster_%s_%s_%s%s",bRec,bGen,jf,cAdd.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_cluster_%s_%s_%s%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen,jf,cAdd.Data()));\r
 \r
    mgr->ConnectInput  (pwg4spec, 0, mgr->GetCommonInputContainer());\r
    mgr->ConnectOutput (pwg4spec, 0, mgr->GetCommonOutputContainer());\r