[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / AliProtonFeedDownAnalysisTask.cxx

#include "TChain.h"\r
#include "TTree.h"\r
#include "TString.h"\r
#include "TList.h"\r
#include "TH2F.h"\r
\r
#include "AliAnalysisTask.h"\r
#include "AliAnalysisManager.h"\r
\r
#include "AliESDEvent.h"\r
#include "AliESDInputHandler.h"\r
#include "AliAODEvent.h"\r
#include "AliAODInputHandler.h"\r
#include "AliMCEventHandler.h"\r
#include "AliMCEvent.h"\r
#include "AliStack.h"\r
#include "AliESDVertex.h"\r
\r
#include "AliProtonFeedDownAnalysis.h"\r
#include "AliProtonAnalysisBase.h"\r
#include "AliProtonFeedDownAnalysisTask.h"\r
#include <Riostream.h>\r
\r
ClassImp(AliProtonFeedDownAnalysisTask)\r
  \r
//________________________________________________________________________ \r
AliProtonFeedDownAnalysisTask::AliProtonFeedDownAnalysisTask()\r
  : AliAnalysisTask(), fESD(0), fAOD(0), fMC(0),\r
    fList(0), fProtonAnalysis(0),fStatHist(0)\r
 {\r
  //Dummy constructor\r
  \r
}\r
\r
//________________________________________________________________________\r
AliProtonFeedDownAnalysisTask::AliProtonFeedDownAnalysisTask(const char *name) \r
  : AliAnalysisTask(name, ""), fESD(0), fAOD(0), fMC(0),\r
    fList(0), fProtonAnalysis(0),fStatHist(0)\r
    {\r
	// Constructor\r
	\r
	// Define input and output slots here\r
	// Input slot #0 works with a TChain\r
	DefineInput(0, TChain::Class());\r
	// Output slot #0 writes into a TList container\r
	DefineOutput(0, TList::Class());\r
}\r
\r
//________________________________________________________________________\r
void AliProtonFeedDownAnalysisTask::ConnectInputData(Option_t *) \r
{\r
	// Connect ESD or AOD here\r
	// Called once\r
	TString gAnalysisLevel = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); \r
	\r
	TTree* tree = dynamic_cast<TTree*> (GetInputData(0));\r
	if (!tree) \r
	{\r
		Printf("ERROR: Could not read chain from input slot 0");\r
	} \r
	else \r
	{\r
		if(gAnalysisLevel == "ESD") \r
		{\r
			AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());   \r
			if (!esdH) \r
			{\r
				Printf("ERROR: Could not get ESDInputHandler");\r
			} \r
			else\r
				fESD = esdH->GetEvent();\r
			AliMCEventHandler* mcH = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());\r
			if (!mcH) \r
			{\r
				Printf("ERROR: Could not retrieve MC event handler");\r
			}\r
			else\r
				fMC = mcH->MCEvent();\r
		}\r
		else if(gAnalysisLevel == "AOD") \r
		{\r
			AliAODInputHandler *aodH = dynamic_cast<AliAODInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());	\r
			if (!aodH) \r
			{\r
				Printf("ERROR: Could not get AODInputHandler");\r
			}\r
			 else\r
				fAOD = aodH->GetEvent();\r
		}\r
		else\r
			Printf("Wrong analysis type: Only ESD, AOD  types are allowed!");\r
	}\r
}\r
//________________________________________________________________________\r
void AliProtonFeedDownAnalysisTask::CreateOutputObjects() \r
{\r
	// Create output objects\r
	// Called once\r
	fList = new TList();\r
	fList->Add(fProtonAnalysis->GetProtonContainer());\r
	fList->Add(fProtonAnalysis->GetAntiProtonContainer());\r
	fList->Add(fProtonAnalysis->GetLambdaHist());\r
	fList->Add(fProtonAnalysis->GetLambdaweightedHist());\r
	fList->Add(fProtonAnalysis->GetAntiLambdaHist());\r
	fList->Add(fProtonAnalysis->GetAntiLambdaweightedHist());\r
	fStatHist=new TH1F("StatsHist","StatsHist",10,-0.5,9.5);\r
	fList->Add(fStatHist);\r
	fStatHist->GetXaxis()->SetBinLabel(1,"level1cut");\r
	fStatHist->GetXaxis()->SetBinLabel(2,"level2cut");\r
	fStatHist->GetXaxis()->SetBinLabel(3,"level3cut");\r
	fStatHist->GetXaxis()->SetBinLabel(4,"level4cut");\r
}\r
//________________________________________________________________________\r
void AliProtonFeedDownAnalysisTask::Exec(Option_t *) \r
{\r
	// Main loop\r
	// Called for each event\r
	TString gAnalysisLevel = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); \r
	//TString gAnalysisLevel = (fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); \r
	if(gAnalysisLevel == "ESD") \r
	{\r
		if (!fESD) \r
		{\r
			Printf("ERROR: fESD not available");\r
			return;\r
		}\r
		fStatHist->Fill(0);\r
		if(dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->IsEventTriggered(fESD,dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetTriggerMode())) \r
		{\r
			fStatHist->Fill(1);\r
			const AliESDVertex *vertex = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVertex(fESD,dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisMode(),dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVxMax(),dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVyMax(),dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVzMax());\r
			if(vertex) \r
			{\r
				fStatHist->Fill(2);\r
				Printf("Proton ESD analysis task: There are %d tracks in this event", fESD->GetNumberOfTracks());\r
				AliStack* stack1=0x0;\r
				if(!fMC)\r
					return;			\r
				stack1 = fMC->Stack();\r
				if(!stack1)\r
					return;\r
				fStatHist->Fill(3);	\r
				fProtonAnalysis->Analyze(fESD,vertex,stack1);\r
				fProtonAnalysis->Analyze(stack1);\r
\r
			}//reconstructed vertex\r
		}//triggered event\r
	}//ESD analysis              \r
	\r
	else if(gAnalysisLevel == "AOD") \r
	{\r
		if (!fAOD) \r
		{\r
			Printf("ERROR: fAOD not available");\r
			return;\r
		}\r
		Printf("Proton AOD analysis task: There are %d tracks in this event", fAOD->GetNumberOfTracks());\r
		fProtonAnalysis->Analyze(fAOD);\r
	}//AOD analysis\r
	                 \r
	\r
	// Post output data.\r
	PostData(0, fList);\r
}    \r
//__________________________________________________________________________________________\r
void AliProtonFeedDownAnalysisTask::Terminate(Option_t *) \r
{\r
\r
}\r
\r
\r
Commit	Line	Data
b3360800	1	#include "TChain.h"\r
	2	#include "TTree.h"\r
	3	#include "TString.h"\r
	4	#include "TList.h"\r
	5	#include "TH2F.h"\r
	6	\r
	7	#include "AliAnalysisTask.h"\r
	8	#include "AliAnalysisManager.h"\r
	9	\r
	10	#include "AliESDEvent.h"\r
	11	#include "AliESDInputHandler.h"\r
	12	#include "AliAODEvent.h"\r
	13	#include "AliAODInputHandler.h"\r
	14	#include "AliMCEventHandler.h"\r
	15	#include "AliMCEvent.h"\r
	16	#include "AliStack.h"\r
	17	#include "AliESDVertex.h"\r
	18	\r
	19	#include "AliProtonFeedDownAnalysis.h"\r
	20	#include "AliProtonAnalysisBase.h"\r
	21	#include "AliProtonFeedDownAnalysisTask.h"\r
	22	#include <Riostream.h>\r
	23	\r
	24	ClassImp(AliProtonFeedDownAnalysisTask)\r
	25	\r
	26	//________________________________________________________________________ \r
	27	AliProtonFeedDownAnalysisTask::AliProtonFeedDownAnalysisTask()\r
	28	: AliAnalysisTask(), fESD(0), fAOD(0), fMC(0),\r
	29	fList(0), fProtonAnalysis(0),fStatHist(0)\r
	30	{\r
	31	//Dummy constructor\r
	32	\r
	33	}\r
	34	\r
	35	//________________________________________________________________________\r
	36	AliProtonFeedDownAnalysisTask::AliProtonFeedDownAnalysisTask(const char *name) \r
	37	: AliAnalysisTask(name, ""), fESD(0), fAOD(0), fMC(0),\r
	38	fList(0), fProtonAnalysis(0),fStatHist(0)\r
	39	{\r
	40	// Constructor\r
	41	\r
	42	// Define input and output slots here\r
	43	// Input slot #0 works with a TChain\r
	44	DefineInput(0, TChain::Class());\r
	45	// Output slot #0 writes into a TList container\r
	46	DefineOutput(0, TList::Class());\r
	47	}\r
	48	\r
	49	//________________________________________________________________________\r
	50	void AliProtonFeedDownAnalysisTask::ConnectInputData(Option_t *) \r
	51	{\r
	52	// Connect ESD or AOD here\r
	53	// Called once\r
	54	TString gAnalysisLevel = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); \r
	55	\r
	56	TTree* tree = dynamic_cast<TTree*> (GetInputData(0));\r
	57	if (!tree) \r
	58	{\r
	59	Printf("ERROR: Could not read chain from input slot 0");\r
	60	} \r
	61	else \r
	62	{\r
	63	if(gAnalysisLevel == "ESD") \r
	64	{\r
65	AliESDInputHandler esdH = dynamic_cast<AliESDInputHandler> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()); \r
66	if (!esdH) \r
67	{\r
68	Printf("ERROR: Could not get ESDInputHandler");\r
69	} \r
70	else\r
71	fESD = esdH->GetEvent();\r
72	AliMCEventHandler* mcH = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());\r
73	if (!mcH) \r
74	{\r
75	Printf("ERROR: Could not retrieve MC event handler");\r
76	}\r
77	else\r
78	fMC = mcH->MCEvent();\r
79	}\r
80	else if(gAnalysisLevel == "AOD") \r
81	{\r
82	AliAODInputHandler aodH = dynamic_cast<AliAODInputHandler> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()); \r
83	if (!aodH) \r
84	{\r
85	Printf("ERROR: Could not get AODInputHandler");\r
86	}\r
87	else\r
88	fAOD = aodH->GetEvent();\r
89	}\r
90	else\r
91	Printf("Wrong analysis type: Only ESD, AOD types are allowed!");\r
92	}\r
93	}\r
94	//________________________________________________________________________\r
95	void AliProtonFeedDownAnalysisTask::CreateOutputObjects() \r
96	{\r
97	// Create output objects\r
98	// Called once\r
99	fList = new TList();\r
100	fList->Add(fProtonAnalysis->GetProtonContainer());\r
101	fList->Add(fProtonAnalysis->GetAntiProtonContainer());\r
102	fList->Add(fProtonAnalysis->GetLambdaHist());\r
103	fList->Add(fProtonAnalysis->GetLambdaweightedHist());\r
104	fList->Add(fProtonAnalysis->GetAntiLambdaHist());\r
105	fList->Add(fProtonAnalysis->GetAntiLambdaweightedHist());\r
106	fStatHist=new TH1F("StatsHist","StatsHist",10,-0.5,9.5);\r
107	fList->Add(fStatHist);\r
108	fStatHist->GetXaxis()->SetBinLabel(1,"level1cut");\r
109	fStatHist->GetXaxis()->SetBinLabel(2,"level2cut");\r
110	fStatHist->GetXaxis()->SetBinLabel(3,"level3cut");\r
111	fStatHist->GetXaxis()->SetBinLabel(4,"level4cut");\r
112	}\r
113	//________________________________________________________________________\r
114	void AliProtonFeedDownAnalysisTask::Exec(Option_t *) \r
115	{\r
116	// Main loop\r
117	// Called for each event\r
118	TString gAnalysisLevel = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); \r
119	//TString gAnalysisLevel = (fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); \r
120	if(gAnalysisLevel == "ESD") \r
121	{\r
122	if (!fESD) \r
123	{\r
124	Printf("ERROR: fESD not available");\r
125	return;\r
126	}\r
127	fStatHist->Fill(0);\r
128	if(dynamic_cast<AliProtonAnalysisBase>(fProtonAnalysis->GetProtonAnalysisBaseObject())->IsEventTriggered(fESD,dynamic_cast<AliProtonAnalysisBase>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetTriggerMode())) \r
129	{\r
130	fStatHist->Fill(1);\r
131	const AliESDVertex vertex = dynamic_cast<AliProtonAnalysisBase>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVertex(fESD,dynamic_cast<AliProtonAnalysisBase>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisMode(),dynamic_cast<AliProtonAnalysisBase>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVxMax(),dynamic_cast<AliProtonAnalysisBase>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVyMax(),dynamic_cast<AliProtonAnalysisBase>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVzMax());\r
132	if(vertex) \r
133	{\r
134	fStatHist->Fill(2);\r
135	Printf("Proton ESD analysis task: There are %d tracks in this event", fESD->GetNumberOfTracks());\r
136	AliStack* stack1=0x0;\r
137	if(!fMC)\r
138	return; \r
139	stack1 = fMC->Stack();\r
140	if(!stack1)\r
141	return;\r
142	fStatHist->Fill(3); \r
143	fProtonAnalysis->Analyze(fESD,vertex,stack1);\r
144	fProtonAnalysis->Analyze(stack1);\r
145	\r
146	}//reconstructed vertex\r
147	}//triggered event\r
148	}//ESD analysis \r
149	\r
150	else if(gAnalysisLevel == "AOD") \r
151	{\r
152	if (!fAOD) \r
153	{\r
154	Printf("ERROR: fAOD not available");\r
155	return;\r
156	}\r
157	Printf("Proton AOD analysis task: There are %d tracks in this event", fAOD->GetNumberOfTracks());\r
158	fProtonAnalysis->Analyze(fAOD);\r
159	}//AOD analysis\r
160	\r
161	\r
162	// Post output data.\r
163	PostData(0, fList);\r
164	} \r
165	//__________________________________________________________________________________________\r
166	void AliProtonFeedDownAnalysisTask::Terminate(Option_t *) \r
167	{\r
168	\r
169	}\r
170	\r
171	\r