PWG4/macros/AddTaskJetSpectrum2.C

   1 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,Int_t iCl = 0);\r
   2 \r
   3 \r
   4 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,char* back = ""){\r
   5 \r
   6   TString cBack = back;\r
   7 \r
   8   AliAnalysisTaskJetSpectrum2 *js = 0;\r
   9   if(kUseAODMC){\r
  10     if(iFlag&(1<<0)){ // UA104\r
  11       js = AddTaskJetSpectrum2("jets","jetsAODMC_UA104",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
  12       js = AddTaskJetSpectrum2("jets","jetsAODMC2_UA104",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
  13     }\r
  14     if(iFlag&(1<<1)){ // ANTIKT 04\r
  15       js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
  16       js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC2_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag,  iEventSelectionMask,kBackground);\r
  17       // cross check MC only background subtration\r
  18       js = AddTaskJetSpectrum2("jetsAODMC2_FASTJET04","jetsAODMC_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag,  iEventSelectionMask,kBackground);\r
  19     }\r
  20     if(iFlag&(1<<2)){ // KT 04\r
  21       js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC_FASTKT04",cBack.Data(),filterMask,iPhysicsSelectionFlag,iEventSelectionMask,kBackground);\r
  22       js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC2_FASTKT04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
  23     }\r
  24     if(iFlag&(1<<3)){ // SISCONE 04\r
  25       js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","jetsAODMC_SISCONE04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
  26       js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","jetsAODMC2_SISCONE04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground);\r
  27     }\r
  28     // here can go other radii\r
  29   }\r
  30   else { // only the data ... no MC\r
  31     if(iFlag&(1<<0)){ // UA104\r
  32       js = AddTaskJetSpectrum2("jets","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground,1);\r
  33     }\r
  34     if(iFlag&(1<<1)){ // ANTIKT 04\r
  35       js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground,1);\r
  36     }\r
  37     if(iFlag&(1<<2)){ // KT 04\r
  38       js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","",cBack.Data(),filterMask,iPhysicsSelectionFlag,iEventSelectionMask,kBackground,1);\r
  39     }\r
  40     if(iFlag&(1<<3)){ // SISCONE 04\r
  41       js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask,kBackground,1);\r
  42     }\r
  43   }\r
  44   return js;\r
  45 }\r
  46 \r
  47 \r
  48 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,Int_t iCl)\r
  49 {\r
  50   // Creates a jet fider task, configures it and adds it to the analysis manager.\r
  51   // Get the pointer to the existing analysis manager via the static access method.\r
  52   //==============================================================================\r
  53   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  54   if (!mgr) {\r
  55      ::Error("AddTaskJetSpectrum2", "No analysis manager to connect to.");\r
  56     return NULL;\r
  57    }  \r
  58   \r
  59   // Check the analysis type using the event handlers connected to the analysis manager.\r
  60   //==============================================================================\r
  61   if (!mgr->GetInputEventHandler()) {\r
  62     ::Error("AddTaskJetSpectrum2", "This task requires an input event handler");\r
  63     return NULL;\r
  64   }\r
  65   \r
  66   TString type = mgr->GetInputEventHandler()->GetDataType();\r
  67   TString typeRec(bRec);\r
  68   TString typeGen(bGen);\r
  69   typeGen.ToUpper();\r
  70   typeRec.ToUpper();\r
  71   \r
  72    TString bBkgName = bBkg;\r
  73    if(kBackground&&bBkgName.Length()==0){\r
  74      // default\r
  75      if(typeRec.Contains("AODMC2"))bBkgName = "jeteventbackground_clustersAODMC2_KT06";\r
  76      else bBkgName = "jeteventbackground_clustersAOD_KT06";\r
  77    }\r
  78 \r
  79    // Create the task and configure it.\r
  80    //===========================================================================\r
  81 \r
  82    if(iFillCorrBkg)bGen = Form("SubBkg%d",iFillCorrBkg);   \r
  83    AliAnalysisTaskJetSpectrum2* pwg4spec = new  AliAnalysisTaskJetSpectrum2(Form("JetSpectrum2%s-%s_%010d_Class%02d",bRec,bGen,iEventSelectionMask,iCl));\r
  84    pwg4spec->SetFillCorrBkg(iFillCorrBkg);\r
  85    if(iCl)pwg4spec->SetEventClass(iCl);\r
  86 \r
  87    // add the filter mask for non default jets branches\r
  88    TString cAdd("");\r
  89    cAdd += Form("_Filter%05d",filterMask);\r
  90 \r
  91    if(iFillCorrBkg)pwg4spec->SetBranchGen("");  \r
  92    else {\r
  93      pwg4spec->SetBranchGen(bGen); \r
  94      if(typeGen.Contains("JETSAOD")&&!typeGen.Contains("MC"))pwg4spec->SetBranchGen(Form("%s%s",bGen,cAdd.Data())); \r
  95    }\r
  96    pwg4spec->SetBranchRec(bRec); \r
  97    if(typeRec.Contains("JETSAOD")&&!typeRec.Contains("MC"))     pwg4spec->SetBranchRec(Form("%s%s",bRec,cAdd.Data())); \r
  98 \r
  99    if(bBkgName.Length()>0)pwg4spec->SetBranchBkg(bBkgName.Data());\r
 100 \r
 101 \r
 102    pwg4spec->SetFilterMask(filterMask); \r
 103    pwg4spec->SetUseGlobalSelection(kTRUE); \r
 104    pwg4spec->SetMinJetPt(5.);\r
 105    pwg4spec->SetBkgSubtraction(kBackground);\r
 106 \r
 107 \r
 108    if(type == "AOD"){\r
 109      // Assume all jet are produced already\r
 110      pwg4spec->SetAODJetInput(kTRUE);\r
 111      pwg4spec->SetAODTrackInput(kTRUE);\r
 112      pwg4spec->SetAODMCInput(kTRUE);\r
 113    }\r
 114    else{\r
 115      if(mgr->GetMCtruthEventHandler()){\r
 116        pwg4spec-> SetAnalysisType(AliAnalysisTaskJetSpectrum2::kAnaMCESD);\r
 117      }\r
 118    }\r
 119 \r
 120    if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
 121      pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCChargedAcceptance);\r
 122    }\r
 123    else if (typeRec.Contains("AODMC2")){\r
 124      pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
 125    }\r
 126    else if (typeRec.Contains("AODMC")){\r
 127      pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
 128    }\r
 129    else { // catch akk use AOD\r
 130      pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAOD);\r
 131    }\r
 132 \r
 133    if(typeGen.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
 134      pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCChargedAcceptance);\r
 135    }\r
 136    else if (typeGen.Contains("AODMC2")){\r
 137      pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
 138    }\r
 139    else if (typeGen.Contains("AODMC")){\r
 140      pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
 141    }\r
 142    else if (typeGen.Length()>0){ // catch all use AOD\r
 143      pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAOD);\r
 144    }\r
 145 \r
 146    if(iPhysicsSelectionFlag)pwg4spec->SelectCollisionCandidates(iPhysicsSelectionFlag);\r
 147    if(iEventSelectionMask)pwg4spec->SetEventSelectionMask(iEventSelectionMask);\r
 148 \r
 149    //   pwg4spec->SetDebugLevel(10);\r
 150 \r
 151    mgr->AddTask(pwg4spec);\r
 152      \r
 153    // Create ONLY the output containers for the data produced by the task.\r
 154    // Get and connect other common input/output containers via the manager as below\r
 155    //==============================================================================\r
 156    AliAnalysisDataContainer *coutput1_Spec = mgr->CreateContainer(Form("pwg4spec2_%s_%s_%010d_Class%02d",bRec,bGen,iEventSelectionMask,iCl),TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_spec2_%s_%s_%010d_Class%02d",AliAnalysisManager::GetCommonFileName(),bRec,bGen,iEventSelectionMask,iCl));\r
 157 \r
 158    mgr->ConnectInput  (pwg4spec, 0, mgr->GetCommonInputContainer());\r
 159    mgr->ConnectOutput (pwg4spec, 0, mgr->GetCommonOutputContainer());\r
 160    mgr->ConnectOutput (pwg4spec,  1, coutput1_Spec );\r
 161    \r
 162    return pwg4spec;\r
 163 }\r
 164 \r
 165 void SetAODInput(AliAnalysisTaskJetSpectrum2 *taskJetSpectrum){\r
 166   taskJetSpectrum->SetAODJetInput(kTRUE);\r
 167   taskJetSpectrum->SetAODTrackInput(kTRUE);\r
 168   // taskJetSpectrum->SetUseGlobalSelection(kFALSE);\r
 169 }\r