[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / AntiprotonToProton / AliProtonFeedDownAnalysisTask.cxx

#include "TChain.h"
#include "TTree.h"
#include "TString.h"
#include "TList.h"
#include "TH2F.h"

#include "AliAnalysisTask.h"
#include "AliAnalysisManager.h"

#include "AliESDEvent.h"
#include "AliESDInputHandler.h"
#include "AliAODEvent.h"
#include "AliAODInputHandler.h"
#include "AliMCEventHandler.h"
#include "AliMCEvent.h"
#include "AliStack.h"
#include "AliESDVertex.h"

#include "AliProtonFeedDownAnalysis.h"
#include "AliProtonAnalysisBase.h"
#include "AliProtonFeedDownAnalysisTask.h"
#include <Riostream.h>

ClassImp(AliProtonFeedDownAnalysisTask)
  
//________________________________________________________________________ 
AliProtonFeedDownAnalysisTask::AliProtonFeedDownAnalysisTask()
  : AliAnalysisTask(), fESD(0), fAOD(0), fMC(0),
    fList(0), fProtonAnalysis(0),fStatHist(0)
 {
  //Dummy constructor
  
}

//________________________________________________________________________
AliProtonFeedDownAnalysisTask::AliProtonFeedDownAnalysisTask(const char *name) 
  : AliAnalysisTask(name, ""), fESD(0), fAOD(0), fMC(0),
    fList(0), fProtonAnalysis(0),fStatHist(0)
    {
	// Constructor
	
	// Define input and output slots here
	// Input slot #0 works with a TChain
	DefineInput(0, TChain::Class());
	// Output slot #0 writes into a TList container
	DefineOutput(0, TList::Class());
}

//________________________________________________________________________
void AliProtonFeedDownAnalysisTask::ConnectInputData(Option_t *) 
{
	// Connect ESD or AOD here
	// Called once
	TString gAnalysisLevel = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); 
	
	TTree* tree = dynamic_cast<TTree*> (GetInputData(0));
	if (!tree) 
	{
		Printf("ERROR: Could not read chain from input slot 0");
	} 
	else 
	{
		if(gAnalysisLevel == "ESD") 
		{
			AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());   
			if (!esdH) 
			{
				Printf("ERROR: Could not get ESDInputHandler");
			} 
			else
				fESD = esdH->GetEvent();
			AliMCEventHandler* mcH = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
			if (!mcH) 
			{
				Printf("ERROR: Could not retrieve MC event handler");
			}
			else
				fMC = mcH->MCEvent();
		}
		else if(gAnalysisLevel == "AOD") 
		{
			AliAODInputHandler *aodH = dynamic_cast<AliAODInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());	
			if (!aodH) 
			{
				Printf("ERROR: Could not get AODInputHandler");
			}
			 else
				fAOD = aodH->GetEvent();
		}
		else
			Printf("Wrong analysis type: Only ESD, AOD  types are allowed!");
	}
}
//________________________________________________________________________
void AliProtonFeedDownAnalysisTask::CreateOutputObjects() 
{
	// Create output objects
	// Called once
	fList = new TList();
	fList->Add(fProtonAnalysis->GetProtonContainer());
	fList->Add(fProtonAnalysis->GetAntiProtonContainer());
	fList->Add(fProtonAnalysis->GetLambdaHist());
	fList->Add(fProtonAnalysis->GetLambdaweightedHist());
	fList->Add(fProtonAnalysis->GetAntiLambdaHist());
	fList->Add(fProtonAnalysis->GetAntiLambdaweightedHist());
	fStatHist=new TH1F("StatsHist","StatsHist",10,-0.5,9.5);
	fList->Add(fStatHist);
	fStatHist->GetXaxis()->SetBinLabel(1,"level1cut");
	fStatHist->GetXaxis()->SetBinLabel(2,"level2cut");
	fStatHist->GetXaxis()->SetBinLabel(3,"level3cut");
	fStatHist->GetXaxis()->SetBinLabel(4,"level4cut");
}
//________________________________________________________________________
void AliProtonFeedDownAnalysisTask::Exec(Option_t *) 
{
	// Main loop
	// Called for each event
	TString gAnalysisLevel = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); 
	//TString gAnalysisLevel = (fProtonAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisLevel(); 
	if(gAnalysisLevel == "ESD") 
	{
		if (!fESD) 
		{
			Printf("ERROR: fESD not available");
			return;
		}
		fStatHist->Fill(0);
		if(dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->IsEventTriggered(fESD,dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetTriggerMode())) 
		{
			fStatHist->Fill(1);
			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());
			if(vertex) 
			{
				fStatHist->Fill(2);
				Printf("Proton ESD analysis task: There are %d tracks in this event", fESD->GetNumberOfTracks());
				AliStack* stack1=0x0;
				if(!fMC)
					return;			
				stack1 = fMC->Stack();
				if(!stack1)
					return;
				fStatHist->Fill(3);	
				fProtonAnalysis->Analyze(fESD,vertex,stack1);
				fProtonAnalysis->Analyze(stack1);

			}//reconstructed vertex
		}//triggered event
	}//ESD analysis              
	
	else if(gAnalysisLevel == "AOD") 
	{
		if (!fAOD) 
		{
			Printf("ERROR: fAOD not available");
			return;
		}
		Printf("Proton AOD analysis task: There are %d tracks in this event", fAOD->GetNumberOfTracks());
		fProtonAnalysis->Analyze(fAOD);
	}//AOD analysis
	                 
	
	// Post output data.
	PostData(0, fList);
}    
//__________________________________________________________________________________________
void AliProtonFeedDownAnalysisTask::Terminate(Option_t *) 
{

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