[u/mrichter/AliRoot.git] / PWGCF / EBYE / NetParticle / macros / AddTaskNetParticle.C

/* *********************************************************************************
 * File    : AddTaskNetParticle.C  
 * Author  : Jochen Thaeder <jochen@thaeder.de>
 * *********************************************************************************
 * Configuring NetParticle Task:
 * - ARGUMENTS : 
 *     name           -> Name of the task, containing partcile type :
 *                       Currently : Proton, Pion, Kaon      
 *     isModeDist     -> Fill Distributions
 *     isModeEff      -> Fill Efficiency/Contamination ThnSparse
 *     isModeDCA      -> Fill DCA ThnSparse
 *     useQAThnSparse -> Fill QA ThnSparse
 *     isCreateCSC    -> Prepare for CrossSectionCorrection 
 *                       - requires isModeEff to be set
 *                       - Proton only
 * 
 * - OUTPUT CONTAINER : #N = 5
 *   (1) - Standard Output, Distributions
 *   (2) - Efficiency ThnSparse
 *   (3) - Contamination ThnSparse
 *   (4) - DCA ThnSparse
 *   (5) - QA ThnSparse
 *
 ********************************************************************************* */

AliAnalysisTask *AddTaskNetParticle(const Char_t * name = "jthaeder_NetProton", 
				    Bool_t isModeDist, Bool_t isModeEff, Bool_t isModeDCA, Bool_t useQAThnSparse = kFALSE,
				    Bool_t isCreateCSC = kFALSE) {

  TString sName(name);

  if (isCreateCSC && !isModeEff) {
    Error("AddTaskNetParticle", "Creating CrossSectionCorrection needs 'isModeEff' to be set.");
    return NULL;
  }

  // ----------------------------------------------
  // -- Get the current analysis manager
  // ----------------------------------------------
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
    Error("AddTaskNetParticle", "No analysis manager found.");
    return NULL;
  }

  // ----------------------------------------------
  // -- Check for MC
  // ----------------------------------------------
  Bool_t isMC = (mgr->GetMCtruthEventHandler() != NULL);
  if (isMC)
    Info("AddTaskNetParticle", "This task has MC.");
  
  // ----------------------------------------------
  // -- Create task configure it
  // ----------------------------------------------
  AliAnalysisTaskNetParticle *task = new AliAnalysisTaskNetParticle("AliAnalysisTaskNetParticle");
  if (!task) {
    Error("AddTaskNetParticle", "Task could not be created.");
    return NULL;
  }

  if (isMC) 
    task->SetIsMC();

  // -- Set particle type
  // ----------------------------------------------
  Float_t minPt, maxPt, minPtEff, maxPtEff, minPtForTOF; 

  if (sName.Contains("Proton")) {
    task->SetParticleSpecies(AliPID::kProton);
    task->SetControlParticleSpecies(3122, kTRUE, "Lambda");
    minPt    = 0.4;    maxPt    = 2.2;
    minPtEff = 0.2;    maxPtEff = 2.6;
    minPtForTOF = 0.8;
    if (isCreateCSC) {
      minPtForTOF = maxPtEff;
    }
  }
  else if (sName.Contains("Pion")) {
    task->SetParticleSpecies(AliPID::kPion);
    task->SetControlParticleSpecies(3122, kTRUE, "Lambda");  /// maybe something else ...
    minPt    = 0.3;    maxPt    = 0.9;
    minPtEff = 0.2;    maxPtEff = 1.2;
    minPtForTOF = 0.8;
  }
  else if (sName.Contains("Kaon")) {
    task->SetParticleSpecies(AliPID::kKaon);
    task->SetControlParticleSpecies(3122, kTRUE, "Lambda");  /// maybe something else ...
    minPt    = 0.2;    maxPt    = 0.4;
    minPtEff = 0.1;    maxPtEff = 0.8;
    minPtForTOF = 0.8;
  }
  else {
    Error("AddTaskNetParticle", "Unknown Particle type.");
    return NULL;
  }

  // -- Configure flags
  // ----------------------------------------------
  if (isModeEff) 
    task->SetModeEffCreation(1);     // => 1 = on    | 0 = off (default)
  if (isModeDCA)
    task->SetModeDCACreation(1);     // => 1 = on    | 0 = off (default)
  if (isModeDist)
    task->SetModeDistCreation(1);    // => 1 = on    | 0 = off (default)

  // -- Enable QA plots
  if (useQAThnSparse)
    task->SetUseQATHnSparse(kTRUE);

  // -- Configure cuts 
  // ----------------------------------------------

  // -- Set cut flags ...
  task->SetESDTrackCutMode(0);     // => 0 = clean | 1 = dirty

  // -- Set standard event cuts
  task->SetVertexZMax(10.);   
  task->SetCentralityBinMax(7);

  // -- Set track event cuts
  task->SetRapidityMax(0.5); 
  task->SetMinTrackLengthMC(70.);  
  task->SetNSigmaMaxCdd(3.); 
  task->SetNSigmaMaxCzz(3.); 

  task->SetNSigmaMaxTPC(2.5);
  task->SetNSigmaMaxTOF(2.5);
  task->SetMinPtForTOFRequired(minPtForTOF);

  // -- Set analysis ranges
  task->SetEtaMax(0.9);        
  task->SetPtRange(minPt, maxPt);           // pt cut range for the analysis
  task->SetPtRangeEff(minPtEff, maxPtEff);  // pt cut range for the correction / efficiency / contamination creation
  
  // ----------------------------------------------
  // -- Initialize and add task to the ANALYSIS manager
  // ----------------------------------------------
  if (!task->Initialize())
    mgr->AddTask(task);
  else {
    Error("Initialize failed, not adding AliAnalysistaskNetParticle !!!");
    delete task;
    return NULL;
  }

  // ----------------------------------------------
  // -- data containers - input
  // ----------------------------------------------
  AliAnalysisDataContainer *cinput  = mgr->GetCommonInputContainer();

  // ----------------------------------------------
  // -- data containers - output
  // ----------------------------------------------
  AliAnalysisDataContainer *coutput     = mgr->CreateContainer(name, TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s.root", name));
  AliAnalysisDataContainer *coutputEff  = mgr->CreateContainer(Form("%s_eff", name), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s.root", name));
  AliAnalysisDataContainer *coutputCont = mgr->CreateContainer(Form("%s_cont", name), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s.root", name));
  AliAnalysisDataContainer *coutputDca  = mgr->CreateContainer(Form("%s_dca", name), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s.root", name));

  AliAnalysisDataContainer *coutputQA   = mgr->CreateContainer(Form("%sQA", name), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%sQA.root", name));
    
  mgr->ConnectInput  (task,  0, cinput );
  mgr->ConnectOutput (task,  1, coutput);
  mgr->ConnectOutput (task,  2, coutputEff);
  mgr->ConnectOutput (task,  3, coutputCont);
  mgr->ConnectOutput (task,  4, coutputDca);
  mgr->ConnectOutput (task,  5, coutputQA);

  return task;
}
Commit	Line	Data
cb68eb1d	1	/* *********************************************************************************
	2	* File : AddTaskNetParticle.C
	3	* Author : Jochen Thaeder <jochen@thaeder.de>
	4	* *********************************************************************************
	5	* Configuring NetParticle Task:
	6	* - ARGUMENTS :
	7	* name -> Name of the task, containing partcile type :
	8	* Currently : Proton, Pion, Kaon
	9	* isModeDist -> Fill Distributions
	10	* isModeEff -> Fill Efficiency/Contamination ThnSparse
	11	* isModeDCA -> Fill DCA ThnSparse
	12	* useQAThnSparse -> Fill QA ThnSparse
	13	* isCreateCSC -> Prepare for CrossSectionCorrection
	14	* - requires isModeEff to be set
	15	* - Proton only
	16	*
	17	* - OUTPUT CONTAINER : #N = 5
	18	* (1) - Standard Output, Distributions
	19	* (2) - Efficiency ThnSparse
	20	* (3) - Contamination ThnSparse
	21	* (4) - DCA ThnSparse
	22	* (5) - QA ThnSparse
	23	*
	24	********************************************************************************* */
	25
	26	AliAnalysisTask AddTaskNetParticle(const Char_t name = "jthaeder_NetProton",
	27	Bool_t isModeDist, Bool_t isModeEff, Bool_t isModeDCA, Bool_t useQAThnSparse = kFALSE,
	28	Bool_t isCreateCSC = kFALSE) {
	29
	30	TString sName(name);
	31
	32	if (isCreateCSC && !isModeEff) {
	33	Error("AddTaskNetParticle", "Creating CrossSectionCorrection needs 'isModeEff' to be set.");
	34	return NULL;
	35	}
	36
	37	// ----------------------------------------------
	38	// -- Get the current analysis manager
	39	// ----------------------------------------------
	40	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	41	if (!mgr) {
	42	Error("AddTaskNetParticle", "No analysis manager found.");
	43	return NULL;
	44	}
	45
	46	// ----------------------------------------------
	47	// -- Check for MC
	48	// ----------------------------------------------
	49	Bool_t isMC = (mgr->GetMCtruthEventHandler() != NULL);
	50	if (isMC)
	51	Info("AddTaskNetParticle", "This task has MC.");
	52
	53	// ----------------------------------------------
	54	// -- Create task configure it
	55	// ----------------------------------------------
	56	AliAnalysisTaskNetParticle *task = new AliAnalysisTaskNetParticle("AliAnalysisTaskNetParticle");
	57	if (!task) {
	58	Error("AddTaskNetParticle", "Task could not be created.");
	59	return NULL;
	60	}
	61
	62	if (isMC)
	63	task->SetIsMC();
	64
65	// -- Set particle type
66	// ----------------------------------------------
67	Float_t minPt, maxPt, minPtEff, maxPtEff, minPtForTOF;
68
69	if (sName.Contains("Proton")) {
70	task->SetParticleSpecies(AliPID::kProton);
71	task->SetControlParticleSpecies(3122, kTRUE, "Lambda");
72	minPt = 0.4; maxPt = 2.2;
73	minPtEff = 0.2; maxPtEff = 2.6;
74	minPtForTOF = 0.8;
75	if (isCreateCSC) {
76	minPtForTOF = maxPtEff;
77	}
78	}
79	else if (sName.Contains("Pion")) {
80	task->SetParticleSpecies(AliPID::kPion);
81	task->SetControlParticleSpecies(3122, kTRUE, "Lambda"); /// maybe something else ...
82	minPt = 0.3; maxPt = 0.9;
83	minPtEff = 0.2; maxPtEff = 1.2;
84	minPtForTOF = 0.8;
85	}
86	else if (sName.Contains("Kaon")) {
87	task->SetParticleSpecies(AliPID::kKaon);
88	task->SetControlParticleSpecies(3122, kTRUE, "Lambda"); /// maybe something else ...
89	minPt = 0.2; maxPt = 0.4;
90	minPtEff = 0.1; maxPtEff = 0.8;
91	minPtForTOF = 0.8;
92	}
93	else {
94	Error("AddTaskNetParticle", "Unknown Particle type.");
95	return NULL;
96	}
97
98	// -- Configure flags
99	// ----------------------------------------------
100	if (isModeEff)
101	task->SetModeEffCreation(1); // => 1 = on \| 0 = off (default)
102	if (isModeDCA)
103	task->SetModeDCACreation(1); // => 1 = on \| 0 = off (default)
104	if (isModeDist)
105	task->SetModeDistCreation(1); // => 1 = on \| 0 = off (default)
106
107	// -- Enable QA plots
108	if (useQAThnSparse)
109	task->SetUseQATHnSparse(kTRUE);
110
111	// -- Configure cuts
112	// ----------------------------------------------
113
114	// -- Set cut flags ...
115	task->SetESDTrackCutMode(0); // => 0 = clean \| 1 = dirty
116
117	// -- Set standard event cuts
118	task->SetVertexZMax(10.);
119	task->SetCentralityBinMax(7);
120
121	// -- Set track event cuts
122	task->SetRapidityMax(0.5);
123	task->SetMinTrackLengthMC(70.);
124	task->SetNSigmaMaxCdd(3.);
125	task->SetNSigmaMaxCzz(3.);
126
127	task->SetNSigmaMaxTPC(2.5);
128	task->SetNSigmaMaxTOF(2.5);
129	task->SetMinPtForTOFRequired(minPtForTOF);
130
131	// -- Set analysis ranges
132	task->SetEtaMax(0.9);
133	task->SetPtRange(minPt, maxPt); // pt cut range for the analysis
134	task->SetPtRangeEff(minPtEff, maxPtEff); // pt cut range for the correction / efficiency / contamination creation
135
136	// ----------------------------------------------
137	// -- Initialize and add task to the ANALYSIS manager
138	// ----------------------------------------------
139	if (!task->Initialize())
140	mgr->AddTask(task);
141	else {
142	Error("Initialize failed, not adding AliAnalysistaskNetParticle !!!");
143	delete task;
144	return NULL;
145	}
146
147	// ----------------------------------------------
148	// -- data containers - input
149	// ----------------------------------------------
150	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
151
152	// ----------------------------------------------
153	// -- data containers - output
154	// ----------------------------------------------
155	AliAnalysisDataContainer *coutput = mgr->CreateContainer(name, TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s.root", name));
156	AliAnalysisDataContainer *coutputEff = mgr->CreateContainer(Form("%s_eff", name), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s.root", name));
157	AliAnalysisDataContainer *coutputCont = mgr->CreateContainer(Form("%s_cont", name), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s.root", name));
158	AliAnalysisDataContainer *coutputDca = mgr->CreateContainer(Form("%s_dca", name), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s.root", name));
159
160	AliAnalysisDataContainer *coutputQA = mgr->CreateContainer(Form("%sQA", name), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%sQA.root", name));
161
162	mgr->ConnectInput (task, 0, cinput );
163	mgr->ConnectOutput (task, 1, coutput);
164	mgr->ConnectOutput (task, 2, coutputEff);
165	mgr->ConnectOutput (task, 3, coutputCont);
166	mgr->ConnectOutput (task, 4, coutputDca);
167	mgr->ConnectOutput (task, 5, coutputQA);
168
169	return task;
170	}