adding TPC cosntrained tracks back in for comparison

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

\r
AliAnalysisTaskFragFuncBB *AddTaskFragFuncBB(UInt_t filterMask=16, Int_t iPhysicsSelection=1){\r
	AddTaskFragFuncBB("jets", "", filterMask, iPhysicsSelection);\r
}\r
\r
AliAnalysisTaskFragFuncBB *AddTaskFragFuncBB(UInt_t filterMask=16, Bool_t kUseAODMC=kFALSE, Int_t iPhysicsSelection=1, UInt_t iFlag){\r
    AliAnalysisTaskFragFuncBB *ff=0;\r
	if(kUseAODMC){\r
	    if(iFlag&(1<<0)) ff = AddTaskFragFuncBB("jets", "jetsAODMC2b_UA104", filterMask, iPhysicsSelection);\r
		if(iFlag&(1<<1)) ff = AddTaskFragFuncBB("jetsAOD_UA107", "jetsAODMC2b_UA107", filterMask, iPhysicsSelection);\r
	}\r
	if(iFlag&(1<<2)) ff = AddTaskFragFuncBB("jets", "", filterMask, iPhysicsSelection);\r
	\r
	return ff;\r
}\r
\r
// _______________________________________________________________________________________\r
\r
AliAnalysisTaskFragFuncBB *AddTaskFragFuncBB(\r
	const char* bRecJets,\r
	const char* bGenJets,\r
	UInt_t filterMask,\r
	Int_t iPhysicsSelection)\r
{\r
   // Creates a fragmentation function task,\r
   // 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("AddTaskFragFuncBB", "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("AddTaskFragFunc", "This task requires an input event handler");\r
	  return NULL;\r
   }\r
\r
   TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"\r
   TString typeRec(bRecJets);\r
   TString typeGen(bGenJets);\r
   typeRec.ToUpper();\r
   typeGen.ToUpper();\r
   \r
   // Create the task and configure it.\r
   //===========================================================================\r
\r
   AliAnalysisTaskFragFuncBB *fragfunc = new AliAnalysisTaskFragFuncBB(Form("Fragmenation Function %s %s", bRecJets, bGenJets));\r
   \r
   // pwg4fragfunc->SetAnalysisType(AliAnalysisTaskFragFunc::kAnaMC);\r
   // if(iAODanalysis) pwg4fragfunc->SetAODInput(kTRUE);\r
   // pwg4fragfunc->SetDebugLevel(11);\r
   \r
   Printf("Rec Jets %s", bRecJets);\r
   Printf("Gen Jets %s", bGenJets);\r
   \r
   fragfunc->SetBranchGenJets(bGenJets);\r
   fragfunc->SetBranchRecJets(bRecJets);\r
   \r
   fragfunc->SetFilterMask(filterMask);\r
   //fragfunc->SetUseGlobalSelection(kTRUE);\r
   \r
   if(type == "AOD"){\r
	  // Assume all jets are produced already\r
	  fragfunc->SetAODJetInput(kTRUE);\r
	  fragfunc->SetAODTrackInput(kTRUE);\r
	  fragfunc->SetAODMCInput(kTRUE);\r
   }\r
	\r
   if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
	 fragfunc->SetTrackTypeRec(AliAnalysisTaskFragFunc::kTrackAODMCChargedAcceptance);\r
   }\r
   else if (typeRec.Contains("AODMC2")){\r
	 fragfunc->SetTrackTypeRec(AliAnalysisTaskFragFunc::kTrackAODMCCharged);\r
   }\r
   else if (typeRec.Contains("AODMC")){\r
	 fragfunc->SetTrackTypeRec(AliAnalysisTaskFragFunc::kTrackAODMCAll);\r
   }\r
   else { // catch akk use AOD\r
	 fragfunc->SetTrackTypeRec(AliAnalysisTaskFragFunc::kTrackAOD);\r
   }\r
\r
   if(typeGen.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
	 fragfunc->SetTrackTypeGen(AliAnalysisTaskFragFunc::kTrackAODMCChargedAcceptance);\r
   }\r
   else if (typeGen.Contains("AODMC2")){\r
	 fragfunc->SetTrackTypeGen(AliAnalysisTaskFragFunc::kTrackAODMCCharged);\r
   }\r
   else if (typeGen.Contains("AODMC")){\r
	 fragfunc->SetTrackTypeGen(AliAnalysisTaskFragFunc::kTrackAODMCAll);\r
   }\r
   else if (typeGen.Length()>0){ // catch all use AOD\r
	 fragfunc->SetTrackTypeGen(AliAnalysisTaskFragFunc::kTrackAOD);\r
   }\r
   else {  //\r
	  fragfunc->SetTrackTypeGen(AliAnalysisTaskFragFunc::kTrackKineCharged);\r
   }\r
\r
   //if(iPhysicsSelection)fragfunc->SelectCollisionCandidates();\r
\r
   mgr->AddTask(fragfunc);\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_FragFunc = mgr->CreateContainer(Form("fracfuncBB_%s_%s",bRecJets,bGenJets), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_fragfuncBB_%s_%s",AliAnalysisManager::GetCommonFileName(),bRecJets,bGenJets));\r
\r
   mgr->ConnectInput  (fragfunc, 0, mgr->GetCommonInputContainer());\r
   mgr->ConnectOutput (fragfunc, 0, mgr->GetCommonOutputContainer());\r
   mgr->ConnectOutput (fragfunc, 1, coutput1_FragFunc);\r
   \r
   return fragfunc;\r
}