removed printout

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

Bool_t ConfigWithFlagsJetSpectrum2();\r
AliAnalysisTaskJetSpectrum2 *jetspec = 0;\r
\r
AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(const char* bRec = "jets",const char* bGen = "jetsAODMC_UA104",const char* nonStdFile="",UInt_t filterMask = 32, Int_t iPhysicsSelectionFlag = AliVEvent::kMB,UInt_t iEventSelectionMask = 0,Int_t iCl = 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,char* back = ""){\r
\r
  TString cBack = back;\r
\r
  AliAnalysisTaskJetSpectrum2 *js = 0;\r
  if(kUseAODMC){\r
    if(iFlag&(1<<0)){ // UA104\r
      js = AddTaskJetSpectrum2("jets","jetsAODMC_UA104",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask);\r
      js = AddTaskJetSpectrum2("jets","jetsAODMC2_UA104",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask);\r
    }\r
    if(iFlag&(1<<1)){ // ANTIKT 04\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask);\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC2_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag,  iEventSelectionMask);\r
      // cross check MC only background subtration\r
      js = AddTaskJetSpectrum2("jetsAODMC2_FASTJET04","jetsAODMC_FASTJET04",cBack.Data(),filterMask,iPhysicsSelectionFlag,  iEventSelectionMask);\r
    }\r
    if(iFlag&(1<<2)){ // KT 04\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC_FASTKT04",cBack.Data(),filterMask,iPhysicsSelectionFlag,iEventSelectionMask);\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC2_FASTKT04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask);\r
    }\r
    if(iFlag&(1<<3)){ // SISCONE 04\r
      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","jetsAODMC_SISCONE04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask);\r
      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","jetsAODMC2_SISCONE04",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask);\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,1);\r
    }\r
    if(iFlag&(1<<1)){ // ANTIKT 04\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask);\r
    }\r
    if(iFlag&(1<<2)){ // KT 04\r
      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","",cBack.Data(),filterMask,iPhysicsSelectionFlag,iEventSelectionMask);\r
    }\r
    if(iFlag&(1<<3)){ // SISCONE 04\r
      js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","",cBack.Data(),filterMask,iPhysicsSelectionFlag, iEventSelectionMask);\r
    }\r
  }\r
  return js;\r
}\r
\r
\r
AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(const char* bRec,const char* bGen ,const char* nonStdFile,UInt_t filterMask,Int_t iPhysicsSelectionFlag,UInt_t iEventSelectionMask,Int_t iCl)\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
\r
   // Create the task and configure it.\r
   //===========================================================================\r
\r
   jetspec = new  AliAnalysisTaskJetSpectrum2(Form("JetSpectrum2%s-%s_%010d_Class%02d",bRec,bGen,iEventSelectionMask,iCl));\r
   if(iCl)jetspec->SetEventClass(iCl);\r
\r
   // add the filter mask for non default jets branches\r
   TString cAdd("");\r
   cAdd += Form("_Filter%05d",filterMask);\r
\r
   \r
\r
   jetspec->SetBranchGen(bGen); \r
   //  if(typeGen.Contains("JETSAOD")&&!typeGen.Contains("MC"))jetspec->SetBranchGen(Form("%s%s",bGen,cAdd.Data())); \r
\r
   jetspec->SetBranchRec(bRec); \r
   // if(typeRec.Contains("JETSAOD")&&!typeRec.Contains("MC"))     jetspec->SetBranchRec(Form("%s%s",bRec,cAdd.Data())); \r
\r
\r
\r
   if(type == "AOD"){\r
     // Assume all jets are not yet produced \r
     //     jetspec->SetAODJetInput(kTRUE);\r
     jetspec->SetAODTrackInput(kTRUE);\r
     jetspec->SetAODMCInput(kTRUE);\r
   }\r
   else{\r
     if(mgr->GetMCtruthEventHandler()){\r
       jetspec-> SetAnalysisType(AliAnalysisTaskJetSpectrum2::kAnaMCESD);\r
     }\r
   }\r
\r
   if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
     jetspec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCChargedAcceptance);\r
   }\r
   else if (typeRec.Contains("AODMC2")){\r
     jetspec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
   }\r
   else if (typeRec.Contains("AODMC")){\r
     jetspec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
   }\r
   else { // catch akk use AOD\r
     jetspec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAOD);\r
   }\r
\r
   if(typeGen.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
     jetspec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCChargedAcceptance);\r
   }\r
   else if (typeGen.Contains("AODMC2")){\r
     jetspec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
   }\r
   else if (typeGen.Contains("AODMC")){\r
     jetspec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
   }\r
   else if (typeGen.Length()>0){ // catch all use AOD\r
     jetspec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAOD);\r
   }\r
\r
   if(iEventSelectionMask)jetspec->SetEventSelectionMask(iEventSelectionMask);\r
\r
   //   jetspec->SetDebugLevel(10);\r
   if(!ConfigWithFlagsJetSpectrum2())return 0;\r
\r
   jetspec->SetUseGlobalSelection(kTRUE); \r
   jetspec->SetMinJetPt(-1.);//keep all jets \r
\r
   // to fetch the AOD from the AOD extension ouput \r
   if(strlen(nonStdFile))jetspec->SetNonStdFile(nonStdFile);\r
   mgr->AddTask(jetspec);\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("pwgje_spec2_%s_%s_%010d_Class%02d",bRec,bGen,iEventSelectionMask,iCl),TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWGJE_spec2_%s_%s_%010d_Class%02d",AliAnalysisManager::GetCommonFileName(),bRec,bGen,iEventSelectionMask,iCl));\r
\r
   mgr->ConnectInput  (jetspec, 0, mgr->GetCommonInputContainer());\r
   mgr->ConnectOutput (jetspec, 0, mgr->GetCommonOutputContainer());\r
   mgr->ConnectOutput (jetspec,  1, coutput1_Spec );\r
   \r
   return jetspec;\r
}\r
\r
void SetAODInput(AliAnalysisTaskJetSpectrum2 *taskJetSpectrum){\r
  taskJetSpectrum->SetAODJetInput(kTRUE);\r
  taskJetSpectrum->SetAODTrackInput(kTRUE);\r
  // taskJetSpectrum->SetUseGlobalSelection(kFALSE);\r
}\r
\r
Bool_t ConfigWithFlagsJetSpectrum2(){\r
    \r
  Bool_t bGood1 = kFALSE;\r
  Bool_t bGood2 = kFALSE;\r
\r
  \r
  Int_t nTrigger = AliAnalysisManager::GetGlobalInt("kNTrigger",bGood1);\r
  \r
  if(bGood1){\r
    jetspec->SetNTrigger(nTrigger);\r
    for(int it = 0;it < nTrigger;it++){\r
      jetspec->SetTrigger(it,\r
			  AliAnalysisManager::GetGlobalInt(Form("kTriggerBit%d",it),bGood1),\r
			  AliAnalysisManager::GetGlobalStr(Form("kTriggerName%d",it),bGood2));\r
    }\r
   }\r
   else {\r
     Printf("%s%d: kNTrigger not defined",(char*)__FILE__,__LINE__); return kFALSE; \r
   }\r
\r
     \r
   AliAnalysisManager::GetGlobalInt("kPhysicsSelectionFlag",bGood1);if(bGood1){\r
     jetspec->SelectCollisionCandidates(AliAnalysisManager::GetGlobalInt("kPhysicsSelectionFlag",bGood1));\r
   }\r
   else {Printf("%s%d: kPhysicsSelectionFlag not defined",(char*)__FILE__,__LINE__); return kFALSE; }\r
\r
\r
   AliAnalysisManager::GetGlobalStr("kDeltaAODJetName",bGood1);if(bGood1)jetspec->SetNonStdFile(AliAnalysisManager::GetGlobalStr("kDeltaAODJetName",bGood1));\r
   else {Printf("%s%d: kDeltaAODJetName not defined",(char*)__FILE__,__LINE__); return kFALSE; }\r
\r
   AliAnalysisManager::GetGlobalDbl("kTrackEtaWindow",bGood1);if(bGood1)jetspec->SetTrackEtaWindow(AliAnalysisManager::GetGlobalDbl("kTrackEtaWindow",bGood1));\r
   else {Printf("%s%d: kTrackEtaWindow not defined",(char*)__FILE__,__LINE__); return kFALSE; }\r
\r
   AliAnalysisManager::GetGlobalDbl("kJetEtaWindow",bGood1);if(bGood1)jetspec->SetJetEtaWindow(AliAnalysisManager::GetGlobalDbl("kJetEtaWindow",bGood1));\r
   else {Printf("%s%d: kJetEtaWindow not defined",(char*)__FILE__,__LINE__); return kFALSE; }\r
\r
   \r
   AliAnalysisManager::GetGlobalInt("kHighPtFilterMask",bGood1);if(bGood1)jetspec->SetFilterMask(AliAnalysisManager::GetGlobalInt("kHighPtFilterMask",bGood1));\r
   else {Printf("%s%d: kHighPtFilterMask not defined",(char*)__FILE__,__LINE__); return kFALSE; }   \r
   \r
   return kTRUE;\r
\r
\r
}\r