Macro to display the output of SDD task in QA-train

[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskJetSpectrum2.C

AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(const char* bRec = "jets",const char* bGen = "jetsAODMC_UA104",const char* bBkg="",UInt_t filterMask = 32, Int_t iPhysicsSelection = 1,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 iPhysicsSelection = 1,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,iPhysicsSelection, iEventSelectionMask,kBackground);\r
      js = AddTaskJetSpectrum2("jets","jetsAODMC2_UA104",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
    }\r
    if(iFlag&(1<<1)){ // ANTIKT 04\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC_FASTJET04",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC2_FASTJET04",cBack.Data(),filterMask,iPhysicsSelection,  iEventSelectionMask,kBackground);\r
      // cross check MC only background subtration\r
      js = AddTaskJetSpectrum2("jetsAODMC2_FASTJET04","jetsAODMC_FASTJET04",cBack.Data(),filterMask,iPhysicsSelection,  iEventSelectionMask,kBackground);\r
    }\r
    if(iFlag&(1<<2)){ // KT 04\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC_FASTKT04",cBack.Data(),filterMask,iPhysicsSelection,iEventSelectionMask,kBackground);\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC2_FASTKT04",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
    }\r
    if(iFlag&(1<<3)){ // SISCONE 04\r
      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","jetsAODMC_SISCONE04",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","jetsAODMC2_SISCONE04",cBack.Data(),filterMask,iPhysicsSelection, 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,iPhysicsSelection, iEventSelectionMask,kBackground);\r
    }\r
    if(iFlag&(1<<1)){ // ANTIKT 04\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
    }\r
    if(iFlag&(1<<2)){ // KT 04\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","",cBack.Data(),filterMask,iPhysicsSelection,iEventSelectionMask,kBackground);\r
    }\r
    if(iFlag&(1<<3)){ // SISCONE 04\r
      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
    }\r
  }\r
    \r
  \r
  /*\r
  if(iFlag&(1<<7))js = AddTaskJetSpectrum2("jets","jetsAOD_FASTJET04",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
  if(iFlag&(1<<8))js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
  if(iFlag&(1<<9))js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
  if(iFlag&(1<<10))js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
  \r
  \r
  if(iFlag&(1<<11)){\r
    js = AddTaskJetSpectrum2("jetsAOD_UA107","",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
    js->SetRecEtaWindow(0.2);\r
  }\r
  if(iFlag&(1<<12)){\r
    js = AddTaskJetSpectrum2("jetsAOD_FASTJET07","",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
    js->SetRecEtaWindow(0.2);\r
  }\r
  if(iFlag&(1<<13)){\r
    js = AddTaskJetSpectrum2("jetsAOD_FASTKT07","",cBack.Data(),filterMask,iPhysicsSelection, iEventSelectionMask,kBackground);\r
    js->SetRecEtaWindow(0.2);\r
  }\r
  if(iFlag&(1<<14)){\r
    js = AddTaskJetSpectrum2("jetsAOD_SISCONE07","",cBack.Data(),filterMask,iPhysicsSelection, 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 iPhysicsSelection,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
   // Get the pointer to the existing analysis manager via the static access method.\r
   //==============================================================================\r
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
   if (!mgr) {\r
      ::Error("AddTaskJetSpectrum2", "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("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
   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->SetDebugLevel(11); \r
\r
   if(iFillCorrBkg)pwg4spec->SetBranchGen("");  \r
   else   pwg4spec->SetBranchGen(bGen); \r
   pwg4spec->SetBranchRec(bRec); \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(iPhysicsSelection)pwg4spec->SelectCollisionCandidates();\r
   if(iEventSelectionMask)pwg4spec->SetEventSelectionMask(iEventSelectionMask);\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("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
   mgr->ConnectOutput (pwg4spec,  1, coutput1_Spec );\r
   \r
   return pwg4spec;\r
}\r