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


/*************************************************************************************************
***  Add Fragmentation Function Task ***
**************************************************************************************************
The fragmenation function task expects an ESD filter and jet finder running before this task. 
Or it runs on delta-AODs filled with filtered tracks and jets before.

** Parameters **
(char) recJetsBranch: branch in AOD for (reconstructed) jets
(char) genJetsBranch: branch in AOD for (generated) jets
(char) jetType: "AOD"   jets from recJetsBranch
                "AODMC" jets from genJetsBranch
                "KINE"  jets from PYCELL
                 +"b" (e.g. "AODb") jets with acceptance cuts
(char) trackType: "AOD"     reconstructed tracks from AOD filled by ESD filter (choose filter mask!)
                  "AODMC"   MC tracks from AOD filled by kine filter
                  "KINE"    kine particles from MC event 
                  +"2" (e.g. "AOD2")  charged tracks only
                  +"b" (e.g. "AOD2b") with acceptance cuts
(UInt_t) filterMask: select filter bit of ESD filter task

***************************************************************************************************/


AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(UInt_t iFlag=1, UInt_t filterMask=16){
        
        AliAnalysisTaskFragmentationFunction *ff=0;

        // UA1 Jets
        // only reconstructed (default)
	if(iFlag&(1<<0)) ff = AddTaskFragmentationFunction("jets", "", "", "", filterMask);
        // charged MC tracks and jets
	if(iFlag&(1<<1)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA104", "AODMC", "AODMC2", filterMask);
        // charged MC tracks and jets with acceptance cuts
	if(iFlag&(1<<2)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA104", "AODMCb", "AODMC2b", filterMask);
        // kine tracks in acceptance, pythia jets in acceptance
	if(iFlag&(1<<3)) ff = AddTaskFragmentationFunction("jets", "", "KINEb", "KINEb", filterMask);
        // reconstructed charged tracks after cuts, MC jets in acceptance 
	if(iFlag&(1<<4)) ff = AddTaskFragmentationFunction("jets", "jetsMC2b", "AODMCb", "AOD2b", filterMask);

        // kt jets
        // only reconstructed 
	if(iFlag&(1<<10)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "", "", "", filterMask);
        // charged MC tracks and jets
	if(iFlag&(1<<11)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "jetsAODMC2_FASTKT04", "AODMC", "AODMC2", filterMask);
        // charged MC tracks and jets with acceptance cuts
	if(iFlag&(1<<12)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "jetsAODMC2_FASTKT04", "AODMCb", "AODMC2b", filterMask);

        // anti-kt jets
        // only reconstructed 
	if(iFlag&(1<<20)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "", "", "", filterMask);
        // charged MC tracks and jets
	if(iFlag&(1<<21)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "jetsAODMC2_FASTJET04", "AODMC", "AODMC2", filterMask);
        // charged MC tracks and jets with acceptance cuts
	if(iFlag&(1<<22)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "jetsAODMC2_FASTJET04", "AODMCb", "AODMC2b", filterMask);


	return ff;
}

// _______________________________________________________________________________________

AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(
        const char* recJetsBranch,
	const char* genJetsBranch,
	const char* jetType,
	const char* trackType,
	UInt_t filterMask)
{
   // Creates a fragmentation function task,
   // configures it and adds it to the analysis manager.

   //******************************************************************************
   //*** Configuration Parameter **************************************************
   //******************************************************************************

   // space for configuration parameter: histo bin, cuts, ...
   // so far only default parameter used

   Int_t debug = -1; // debug level, -1: not set here

   //******************************************************************************


   // Get the pointer to the existing analysis manager via the static access method.
   //==============================================================================
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   if (!mgr) {
	  ::Error("AddTaskFragmentationFunction", "No analysis manager to connect to.");
	  return NULL;
   }
   
   // Check the analysis type using the event handlers connected to the analysis manager.
   //==============================================================================
   if (!mgr->GetInputEventHandler()) {
	 ::Error("AddTaskFragmentationFunction", "This task requires an input event handler");
	  return NULL;
   }

   TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
   Printf("Data Type: %s", type.Data());

   TString branchRecJets(recJetsBranch);
   TString branchGenJets(genJetsBranch);
   TString typeJets(jetType);
   TString typeTracks(trackType);

   if(branchRecJets.Length()==0) branchRecJets = "noRecJets";
   if(branchGenJets.Length()==0) branchGenJets = "noGenJets";
   if(typeTracks.Length()==0) typeTracks = "trackTypeUndef";
   if(typeJets.Length()==0)   typeJets   = "jetTypeUndef";
   
   // Create the task and configure it.
   //===========================================================================

   AliAnalysisTaskFragmentationFunction *task = new AliAnalysisTaskFragmentationFunction(
        Form("Fragmenation Function %s %s %s %s", branchRecJets.Data(), branchGenJets.Data(), typeJets.Data(), typeTracks.Data()));
   
   if(debug>=0) task->SetDebugLevel(debug);
   
   Printf("Rec Jets %s", branchRecJets.Data());
   Printf("Gen Jets %s", branchGenJets.Data());
   Printf("Jet Type %s", typeJets.Data());
   Printf("Track Type %s", typeTracks.Data());
   
   if(!branchRecJets.Contains("noRecJets")) task->SetBranchRecJets(branchRecJets);
   if(!branchGenJets.Contains("noGenJets")) task->SetBranchGenJets(branchGenJets);


   if(typeTracks.Contains("AODMC2b"))      task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCChargedAcceptance);
   else if(typeTracks.Contains("AODMC2"))  task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCCharged);
   else if(typeTracks.Contains("AODMC"))   task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCAll);
   else if(typeTracks.Contains("KINE2b"))  task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineChargedAcceptance);
   else if(typeTracks.Contains("KINE2"))   task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineCharged);
   else if(typeTracks.Contains("KINE"))    task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineAll);
   else if(typeTracks.Contains("AODb"))    task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODCuts);
   else if(typeTracks.Contains("AOD"))     task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAOD);
   else if(typeTracks.Contains("trackTypeUndef")) task->SetTrackTypeGen(0); // undefined
   else Printf("trackType %s not found", typeTracks.Data());

   if(typeJets.Contains("AODMCb"))         task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance);
   else if(typeJets.Contains("AODMC"))     task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGen);
   else if(typeJets.Contains("KINEb"))     task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKineAcceptance);
   else if(typeJets.Contains("KINE"))      task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKine);
   else if(typeJets.Contains("AODb"))      task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance);
   else if(typeJets.Contains("AOD"))       task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRec);
   else if(typeJets.Contains("jetTypeUndef")) task->SetJetTypeGen(0); // undefined
   else Printf("jetType %s not found", typeJets.Data());
   
   if(typeJets.Contains("AODMCb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance); // kJetsRecAcceptance
   else if(typeJets.Contains("AODb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance); 
   else task->SetJetTypeRecEff(0);

   task->SetFilterMask(filterMask);
  
   // Set default parameters 
   // Cut selection 
   task->SetTrackCuts();       // default : pt > 0.150 GeV, |eta|<0.9, full phi acc
   task->SetJetCuts();         // default: jet pt > 5 GeV, |eta|<0.5, full phi acc
   task->SetDiJetCuts();       // default: type of cut = 1 (cut in deltaPhi), deltaPhi = 0., cdf = 0.5, fraction of pt = 0.6
   task->SetKindSlices();      // default: kindSlice = 1 (inv mass)
   task->SetFFRadius();        // default: R = 0.4
   task->SetHighPtThreshold(); // default: pt > 5 Gev
   // Define histo bins
   task->SetFFHistoBins();
   task->SetQAJetHistoBins();
   task->SetQATrackHistoBins();
   task->SetIJHistoBins();
   task->SetDiJetHistoBins();
   task->SetQADiJetHistoBins();

   mgr->AddTask(task);

   // Create ONLY the output containers for the data produced by the task.
   // Get and connect other common input/output containers via the manager as below
   //==============================================================================

   AliAnalysisDataContainer *coutput_FragFunc = mgr->CreateContainer(
      Form("fracfunc_%s_%s_%s_%s", branchRecJets.Data(), branchGenJets.Data(), typeTracks.Data(), typeJets.Data()),
      TList::Class(),
      AliAnalysisManager::kOutputContainer,
      Form("%s:PWG4_FragmentationFunction_%s_%s_%s_%s", 
         AliAnalysisManager::GetCommonFileName(), branchRecJets.Data(), branchGenJets. Data(), typeTracks.Data(), typeJets.Data()));

   mgr->ConnectInput  (task, 0, mgr->GetCommonInputContainer());
   mgr->ConnectOutput (task, 0, mgr->GetCommonOutputContainer());
   mgr->ConnectOutput (task, 1, coutput_FragFunc);
   
   return task;
}
Commit	Line	Data
1db1733e	1
	2	/*************************************************************************************************
	3	* Add Fragmentation Function Task *
	4	**************************************************************************************************
	5	The fragmenation function task expects an ESD filter and jet finder running before this task.
	6	Or it runs on delta-AODs filled with filtered tracks and jets before.
	7
	8	Parameters
	9	(char) recJetsBranch: branch in AOD for (reconstructed) jets
	10	(char) genJetsBranch: branch in AOD for (generated) jets
	11	(char) jetType: "AOD" jets from recJetsBranch
	12	"AODMC" jets from genJetsBranch
	13	"KINE" jets from PYCELL
	14	+"b" (e.g. "AODb") jets with acceptance cuts
	15	(char) trackType: "AOD" reconstructed tracks from AOD filled by ESD filter (choose filter mask!)
	16	"AODMC" MC tracks from AOD filled by kine filter
	17	"KINE" kine particles from MC event
	18	+"2" (e.g. "AOD2") charged tracks only
	19	+"b" (e.g. "AOD2b") with acceptance cuts
	20	(UInt_t) filterMask: select filter bit of ESD filter task
	21
	22	***************************************************************************************************/
	23
	24
	25
	26
	27	AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(UInt_t iFlag=1, UInt_t filterMask=16){
	28
	29	AliAnalysisTaskFragmentationFunction *ff=0;
	30
c2658535	31	// UA1 Jets
1db1733e	32	// only reconstructed (default)
1db1733e	33	if(iFlag&(1<<0)) ff = AddTaskFragmentationFunction("jets", "", "", "", filterMask);
ac017ff3	34	// charged MC tracks and jets
492fbd39	35	if(iFlag&(1<<1)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA104", "AODMC", "AODMC2", filterMask);
ac017ff3	36	// charged MC tracks and jets with acceptance cuts
492fbd39	37	if(iFlag&(1<<2)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA104", "AODMCb", "AODMC2b", filterMask);
1db1733e	38	// kine tracks in acceptance, pythia jets in acceptance
ac017ff3	39	if(iFlag&(1<<3)) ff = AddTaskFragmentationFunction("jets", "", "KINEb", "KINEb", filterMask);
1db1733e	40	// reconstructed charged tracks after cuts, MC jets in acceptance
ac017ff3	41	if(iFlag&(1<<4)) ff = AddTaskFragmentationFunction("jets", "jetsMC2b", "AODMCb", "AOD2b", filterMask);
c2658535	42
	43	// kt jets
	44	// only reconstructed
	45	if(iFlag&(1<<10)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "", "", "", filterMask);
	46	// charged MC tracks and jets
	47	if(iFlag&(1<<11)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "jetsAODMC2_FASTKT04", "AODMC", "AODMC2", filterMask);
	48	// charged MC tracks and jets with acceptance cuts
	49	if(iFlag&(1<<12)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "jetsAODMC2_FASTKT04", "AODMCb", "AODMC2b", filterMask);
	50
	51	// anti-kt jets
	52	// only reconstructed
	53	if(iFlag&(1<<20)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "", "", "", filterMask);
	54	// charged MC tracks and jets
	55	if(iFlag&(1<<21)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "jetsAODMC2_FASTJET04", "AODMC", "AODMC2", filterMask);
	56	// charged MC tracks and jets with acceptance cuts
	57	if(iFlag&(1<<22)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "jetsAODMC2_FASTJET04", "AODMCb", "AODMC2b", filterMask);
	58
	59
1db1733e	60
	61	return ff;
	62	}
	63
	64	// _______________________________________________________________________________________
	65
	66	AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(
	67	const char* recJetsBranch,
	68	const char* genJetsBranch,
	69	const char* jetType,
	70	const char* trackType,
	71	UInt_t filterMask)
6bec8236	72	{
1db1733e	73	// Creates a fragmentation function task,
	74	// configures it and adds it to the analysis manager.
	75
	76	//******************************************************************************
	77	//* Configuration Parameter ************************************************
	78	//******************************************************************************
	79
	80	// space for configuration parameter: histo bin, cuts, ...
	81	// so far only default parameter used
	82
	83	Int_t debug = -1; // debug level, -1: not set here
6bec8236	84
1db1733e	85	//******************************************************************************
	86
	87
	88
6bec8236	89	// Get the pointer to the existing analysis manager via the static access method.
	90	//==============================================================================
	91	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	92	if (!mgr) {
1db1733e	93	::Error("AddTaskFragmentationFunction", "No analysis manager to connect to.");
	94	return NULL;
	95	}
6bec8236	96
	97	// Check the analysis type using the event handlers connected to the analysis manager.
	98	//==============================================================================
	99	if (!mgr->GetInputEventHandler()) {
1db1733e	100	::Error("AddTaskFragmentationFunction", "This task requires an input event handler");
1db1733e	101	return NULL;
6bec8236	102	}
6bec8236	103
1db1733e	104	TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
	105	Printf("Data Type: %s", type.Data());
	106
	107	TString branchRecJets(recJetsBranch);
	108	TString branchGenJets(genJetsBranch);
	109	TString typeJets(jetType);
	110	TString typeTracks(trackType);
	111
	112	if(branchRecJets.Length()==0) branchRecJets = "noRecJets";
	113	if(branchGenJets.Length()==0) branchGenJets = "noGenJets";
	114	if(typeTracks.Length()==0) typeTracks = "trackTypeUndef";
	115	if(typeJets.Length()==0) typeJets = "jetTypeUndef";
	116
6bec8236	117	// Create the task and configure it.
6bec8236	118	//===========================================================================
1db1733e	119
	120	AliAnalysisTaskFragmentationFunction *task = new AliAnalysisTaskFragmentationFunction(
	121	Form("Fragmenation Function %s %s %s %s", branchRecJets.Data(), branchGenJets.Data(), typeJets.Data(), typeTracks.Data()));
	122
	123	if(debug>=0) task->SetDebugLevel(debug);
6bec8236	124
1db1733e	125	Printf("Rec Jets %s", branchRecJets.Data());
	126	Printf("Gen Jets %s", branchGenJets.Data());
	127	Printf("Jet Type %s", typeJets.Data());
	128	Printf("Track Type %s", typeTracks.Data());
6bec8236	129
1db1733e	130	if(!branchRecJets.Contains("noRecJets")) task->SetBranchRecJets(branchRecJets);
1db1733e	131	if(!branchGenJets.Contains("noGenJets")) task->SetBranchGenJets(branchGenJets);
6bec8236	132
6bec8236	133
1db1733e	134	if(typeTracks.Contains("AODMC2b")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCChargedAcceptance);
	135	else if(typeTracks.Contains("AODMC2")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCCharged);
	136	else if(typeTracks.Contains("AODMC")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCAll);
	137	else if(typeTracks.Contains("KINE2b")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineChargedAcceptance);
	138	else if(typeTracks.Contains("KINE2")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineCharged);
	139	else if(typeTracks.Contains("KINE")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineAll);
	140	else if(typeTracks.Contains("AODb")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODCuts);
	141	else if(typeTracks.Contains("AOD")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAOD);
	142	else if(typeTracks.Contains("trackTypeUndef")) task->SetTrackTypeGen(0); // undefined
	143	else Printf("trackType %s not found", typeTracks.Data());
	144
	145	if(typeJets.Contains("AODMCb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance);
	146	else if(typeJets.Contains("AODMC")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGen);
	147	else if(typeJets.Contains("KINEb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKineAcceptance);
	148	else if(typeJets.Contains("KINE")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKine);
	149	else if(typeJets.Contains("AODb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance);
	150	else if(typeJets.Contains("AOD")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRec);
	151	else if(typeJets.Contains("jetTypeUndef")) task->SetJetTypeGen(0); // undefined
	152	else Printf("jetType %s not found", typeJets.Data());
	153
7d9e8b32	154	if(typeJets.Contains("AODMCb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance); // kJetsRecAcceptance
	155	else if(typeJets.Contains("AODb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance);
	156	else task->SetJetTypeRecEff(0);
	157
1db1733e	158	task->SetFilterMask(filterMask);
1db1733e	159
f3564520	160	// Set default parameters
f3564520	161	// Cut selection
1db1733e	162	task->SetTrackCuts(); // default : pt > 0.150 GeV, \|eta\|<0.9, full phi acc
1db1733e	163	task->SetJetCuts(); // default: jet pt > 5 GeV, \|eta\|<0.5, full phi acc
f3564520	164	task->SetDiJetCuts(); // default: type of cut = 1 (cut in deltaPhi), deltaPhi = 0., cdf = 0.5, fraction of pt = 0.6
f3564520	165	task->SetKindSlices(); // default: kindSlice = 1 (inv mass)
1db1733e	166	task->SetFFRadius(); // default: R = 0.4
1db1733e	167	task->SetHighPtThreshold(); // default: pt > 5 Gev
f3564520	168	// Define histo bins
1db1733e	169	task->SetFFHistoBins();
	170	task->SetQAJetHistoBins();
	171	task->SetQATrackHistoBins();
f3564520	172	task->SetIJHistoBins();
	173	task->SetDiJetHistoBins();
	174	task->SetQADiJetHistoBins();
1db1733e	175
1db1733e	176	mgr->AddTask(task);
6bec8236	177
6bec8236	178	// Create ONLY the output containers for the data produced by the task.
	179	// Get and connect other common input/output containers via the manager as below
	180	//==============================================================================
6bec8236	181
1db1733e	182	AliAnalysisDataContainer *coutput_FragFunc = mgr->CreateContainer(
	183	Form("fracfunc_%s_%s_%s_%s", branchRecJets.Data(), branchGenJets.Data(), typeTracks.Data(), typeJets.Data()),
	184	TList::Class(),
	185	AliAnalysisManager::kOutputContainer,
	186	Form("%s:PWG4_FragmentationFunction_%s_%s_%s_%s",
	187	AliAnalysisManager::GetCommonFileName(), branchRecJets.Data(), branchGenJets. Data(), typeTracks.Data(), typeJets.Data()));
	188
	189	mgr->ConnectInput (task, 0, mgr->GetCommonInputContainer());
	190	mgr->ConnectOutput (task, 0, mgr->GetCommonOutputContainer());
	191	mgr->ConnectOutput (task, 1, coutput_FragFunc);
6bec8236	192
1db1733e	193	return task;
6bec8236	194	}