]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWG4/macros/AddTaskJetSpectrum2.C
Bug fixes by Martha for HighPT, cosmetic an mem leak fix for FF (Oliver)
[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskJetSpectrum2.C
index 49478237b5af0099bb5124731292313f2dd8b17a..4649da3947d129e896bc9bf3d2e5a6aedf029112 100644 (file)
@@ -1,7 +1,82 @@
-AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2()\r
+AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(const char* bRec = "jets",const char* bGen = "jetsAODMC_UA104",const char* bBkg="",UInt_t filterMask = 32, Int_t iPhysicsSelectionFlag = AliVEvent::kMB,UInt_t iEventSelectionMask = 0,Bool_t kBackground=kTRUE,Int_t iFillCorrBkg = 0);\r
+\r
+\r
+AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2Delta(UInt_t filterMask = 32,Bool_t kUseAODMC = kFALSE,Int_t iPhysicsSelectionFlag = AliVEvent::kMB,UInt_t iFlag = 0xfffffff, UInt_t iEventSelectionMask = 0,Bool_t kBackground = kTRUE){\r
+\r
+\r
+  AliAnalysisTaskJetSpectrum2 *js = 0;\r
+\r
+  TString cBack = ""; // \r
+\r
+  if(kUseAODMC){\r
+    if(iFlag&(1<<0)){ // UA104\r
+      js = AddTaskJetSpectrum2("jets","jetsAODMC_UA104",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+      js = AddTaskJetSpectrum2("jets","jetsAODMC2_UA104",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    }\r
+    if(iFlag&(1<<1)){ // ANTIKT 04\r
+      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC2_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag,  iEventSelectionMask,kBackground);\r
+      // cross check MC only background subtration\r
+      js = AddTaskJetSpectrum2("jetsAODMC2_FASTJET04","jetsAODMC_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag,  iEventSelectionMask,kBackground);\r
+    }\r
+    if(iFlag&(1<<2)){ // KT 04\r
+      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC_FASTKT04",cBack.Data(),filterMask,iPhysicsSelectionFlag,iEventSelectionMask,kBackground);\r
+      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC2_FASTKT04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    }\r
+    if(iFlag&(1<<3)){ // SISCONE 04\r
+      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","jetsAODMC_SISCONE04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","jetsAODMC2_SISCONE04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    }\r
+    // here can go other radii\r
+  }\r
+  else { // only the data ... no MC\r
+    if(iFlag&(1<<0)){ // UA104\r
+      js = AddTaskJetSpectrum2("jets","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    }\r
+    if(iFlag&(1<<1)){ // ANTIKT 04\r
+      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    }\r
+    if(iFlag&(1<<2)){ // KT 04\r
+      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","",cBack.Data(),filterMask,iPhysicsSelectionFlag,iEventSelectionMask,kBackground);\r
+    }\r
+    if(iFlag&(1<<3)){ // SISCONE 04\r
+      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    }\r
+  }\r
+    \r
+  \r
+  /*\r
+  if(iFlag&(1<<7))js = AddTaskJetSpectrum2("jets","jetsAOD_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+  if(iFlag&(1<<8))js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+  if(iFlag&(1<<9))js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+  if(iFlag&(1<<10))js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+  \r
+  \r
+  if(iFlag&(1<<11)){\r
+    js = AddTaskJetSpectrum2("jetsAOD_UA107","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    js->SetRecEtaWindow(0.2);\r
+  }\r
+  if(iFlag&(1<<12)){\r
+    js = AddTaskJetSpectrum2("jetsAOD_FASTJET07","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    js->SetRecEtaWindow(0.2);\r
+  }\r
+  if(iFlag&(1<<13)){\r
+    js = AddTaskJetSpectrum2("jetsAOD_FASTKT07","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    js->SetRecEtaWindow(0.2);\r
+  }\r
+  if(iFlag&(1<<14)){\r
+    js = AddTaskJetSpectrum2("jetsAOD_SISCONE07","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
+    js->SetRecEtaWindow(0.2);\r
+  }\r
+  */\r
+\r
+  return js;\r
+}\r
+\r
+\r
+AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(const char* bRec,const char* bGen ,const char* bBkg,UInt_t filterMask,Int_t iPhysicsSelectionFlag,UInt_t iEventSelectionMask,Bool_t kBackground,Int_t iFillCorrBkg)\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
@@ -13,30 +88,105 @@ AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2()
    // Check the analysis type using the event handlers connected to the analysis manager.\r
    //==============================================================================\r
    if (!mgr->GetInputEventHandler()) {\r
-      ::Error("AddTaskJetSpectrum2", "This task requires an input event handler");\r
+     ::Error("AddTaskJetSpectrum2", "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
+\r
+   TString bBkgName = bBkg;\r
+   if(kBackground&&bBkgName.Length()==0){\r
+     if(typeRec.Contains("AODMC2"))bBkgName = "jeteventbackground_clustersAODMC2_KT06";\r
+     else bBkgName = "jeteventbackground_clustersAOD_KT06";\r
+   }\r
+\r
    // Create the task and configure it.\r
    //===========================================================================\r
-   \r
-   AliAnalysisTaskJetSpectrum2* pwg4spec = new  AliAnalysisTaskJetSpectrum2("Jet Spectrum");\r
-      \r
+\r
+   if(iFillCorrBkg)bGen = Form("SubBkg%d",iFillCorrBkg);   \r
+   AliAnalysisTaskJetSpectrum2* pwg4spec = new  AliAnalysisTaskJetSpectrum2(Form("JetSpectrum2%s-%s_%010d",bRec,bGen,iEventSelectionMask));\r
+   pwg4spec->SetFillCorrBkg(iFillCorrBkg);\r
    // or a config file\r
-   pwg4spec->SetAnalysisType(AliAnalysisTaskJetSpectrum2::kAnaMC);\r
-   //      if(iAODanalysis)pwg4spec->SetAODInput(kTRUE);\r
+   // pwg4spec->SetAnalysisType(AliAnalysisTaskJetSpectrum2::kAnaMC);\r
+   // if(iAODanalysis)pwg4spec->SetAODInput(kTRUE);\r
    // pwg4spec->SetDebugLevel(11); \r
-   pwg4spec->SetBranchGen("jetsMC"); \r
-   pwg4spec->SetBranchRec("jetsAOD"); \r
-   mgr->AddTask(pwg4spec);\r
 \r
 \r
 \r
-      \r
+   // add the filter mask for non default jets branches\r
+   TString cAdd("");\r
+   cAdd += Form("_Filter%05d",filterMask);\r
+\r
+   if(iFillCorrBkg)pwg4spec->SetBranchGen("");  \r
+   else {\r
+     pwg4spec->SetBranchGen(bGen); \r
+     if(typeGen.Contains("JETSAOD")&&!typeGen.Contains("MC"))     pwg4spec->SetBranchGen(Form("%s%s",bGen,cAdd.Data())); \r
+   }\r
+   pwg4spec->SetBranchRec(bRec); \r
+   if(typeRec.Contains("JETSAOD")&&!typeRec.Contains("MC"))     pwg4spec->SetBranchRec(Form("%s%s",bRec,cAdd.Data())); \r
+\r
+   if(bBkgName.Length()>0)pwg4spec->SetBranchBkg(bBkgName.Data());\r
+\r
+\r
+   pwg4spec->SetFilterMask(filterMask); \r
+   pwg4spec->SetUseGlobalSelection(kTRUE); \r
+   pwg4spec->SetMinJetPt(5.);\r
+   pwg4spec->SetBkgSubtraction(kBackground);\r
+\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
+   else{\r
+     if(mgr->GetMCtruthEventHandler()){\r
+       pwg4spec-> SetAnalysisType(AliAnalysisTaskJetSpectrum2::kAnaMCESD);\r
+     }\r
+   }\r
+\r
+   if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCChargedAcceptance);\r
+   }\r
+   else if (typeRec.Contains("AODMC2")){\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
+   }\r
+   else if (typeRec.Contains("AODMC")){\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
+   }\r
+   else { // catch akk use AOD\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAOD);\r
+   }\r
+\r
+   if(typeGen.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCChargedAcceptance);\r
+   }\r
+   else if (typeGen.Contains("AODMC2")){\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
+   }\r
+   else if (typeGen.Contains("AODMC")){\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
+   }\r
+   else if (typeGen.Length()>0){ // catch all use AOD\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAOD);\r
+   }\r
+\r
+   if(iPhysicsSelectionFlag)pwg4spec->SelectCollisionCandidates(iPhysicsSelectionFlag);\r
+   if(iEventSelectionMask)pwg4spec->SetEventSelectionMask(iEventSelectionMask);\r
+\r
+   //   pwg4spec->SetDebugLevel(10);\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("pwg4spec2", TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_spec2",AliAnalysisManager::GetCommonFileName()));\r
+   AliAnalysisDataContainer *coutput1_Spec = mgr->CreateContainer(Form("pwg4spec2_%s_%s_%010d",bRec,bGen,iEventSelectionMask),TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_spec2_%s_%s_%010d",AliAnalysisManager::GetCommonFileName(),bRec,bGen,iEventSelectionMask));\r
 \r
    mgr->ConnectInput  (pwg4spec, 0, mgr->GetCommonInputContainer());\r
    mgr->ConnectOutput (pwg4spec, 0, mgr->GetCommonOutputContainer());\r