[u/mrichter/AliRoot.git] / PWG0 / dNdPt / AlidNdPtTask.cxx

/**************************************************************************\r
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
*                                                                        *\r
* Author: The ALICE Off-line Project.                                    *\r
* Contributors are mentioned in the code where appropriate.              *\r
*                                                                        *\r
* Permission to use, copy, modify and distribute this software and its   *\r
* documentation strictly for non-commercial purposes is hereby granted   *\r
* without fee, provided that the above copyright notice appears in all   *\r
* copies and that both the copyright notice and this permission notice   *\r
* appear in the supporting documentation. The authors make no claims     *\r
* about the suitability of this software for any purpose. It is          *\r
* provided "as is" without express or implied warranty.                  *\r
**************************************************************************/\r
\r
#include "iostream"\r
\r
#include "TChain.h"\r
#include "TTree.h"\r
#include "TH1F.h"\r
#include "TCanvas.h"\r
#include "TList.h"\r
#include "TFile.h"\r
\r
#include "AliAnalysisTask.h"\r
#include "AliAnalysisManager.h"\r
#include "AliESDEvent.h"\r
#include "AliESDInputHandler.h"\r
#include "AliESDVertex.h"\r
#include "AliTracker.h"\r
#include "AliGeomManager.h"\r
\r
#include "AliCentrality.h"\r
#include "AliESDVZERO.h"\r
#include "AliMultiplicity.h"\r
\r
#include "AliESDtrackCuts.h"\r
#include "AliMCEventHandler.h"\r
#include "dNdPt/AlidNdPt.h"\r
#include "dNdPt/AlidNdPtEventCuts.h"\r
#include "dNdPt/AlidNdPtAcceptanceCuts.h"\r
\r
#include "dNdPt/AlidNdPtTask.h"\r
\r
using namespace std;\r
\r
ClassImp(AlidNdPtTask)\r
\r
//_____________________________________________________________________________\r
AlidNdPtTask::AlidNdPtTask(const char *name) \r
  : AliAnalysisTaskSE(name)\r
  , fESD(0)\r
  , fMC(0)\r
  , fOutput(0)\r
  , fPitList(0)\r
  , fCompList(0)\r
  , fUseMCInfo(kFALSE)\r
  , fUseCentrality(0)    // default = off\r
  , fUseCentralityBin(0)\r
{\r
  // Constructor\r
\r
  // Define input and output slots here\r
  DefineOutput(1, TList::Class());\r
\r
  // create the list for comparison objects\r
  fCompList = new TList;\r
}\r
\r
//_____________________________________________________________________________\r
AlidNdPtTask::~AlidNdPtTask()\r
{\r
  if(fOutput) delete fOutput;  fOutput =0; \r
  if(fCompList) delete fCompList;  fCompList =0; \r
}\r
\r
//____________________________________________________________________________\r
Bool_t AlidNdPtTask::Notify()\r
{\r
  static Int_t count = 0;\r
  count++;\r
  //Printf("Processing %d. file: %s", count, ((TTree*) GetInputData(0))->GetCurrentFile()->GetName());\r
  TTree *chain = (TChain*)GetInputData(0);\r
  if(chain)\r
    Printf("Processing %d. file: %s", count, chain->GetCurrentFile()->GetName());\r
\r
  /*\r
  TChain *chain = (TChain*)GetInputData(0);\r
  AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
  if (!esdH) {\r
    Printf("ERROR: Could not get ESDInputHandler");\r
    return kFALSE;\r
  } else {\r
    if(chain)\r
    Printf("chain->GetCurrentFile()->GetName() %s", chain->GetCurrentFile()->GetName());\r
  }\r
  */\r
\r
return kTRUE;\r
}\r
\r
//_____________________________________________________________________________\r
Bool_t AlidNdPtTask::AddAnalysisObject(AlidNdPt *pObj) \r
{\r
  // add analysis object to the list\r
  if(pObj == 0) {\r
    Printf("ERROR: Could not add comparison object");\r
    return kFALSE;\r
  }\r
\r
  // add object to the list\r
  fCompList->AddLast(pObj);\r
       \r
return kTRUE;\r
}\r
\r
//_____________________________________________________________________________\r
void AlidNdPtTask::UserCreateOutputObjects()\r
{\r
  // Create histograms\r
  // Called once\r
\r
  OpenFile(1, "RECREATE");\r
\r
  //\r
  // create output list\r
  //\r
  fOutput = new TList;\r
  fOutput->SetOwner();\r
  fPitList = fOutput->MakeIterator();\r
\r
  // add dNdPt analysis objects to the output\r
  AlidNdPt *pObj=0;\r
  Int_t count=0;\r
  TIterator *pitCompList = fCompList->MakeIterator();\r
  pitCompList->Reset();\r
  while(( pObj = (AlidNdPt *)pitCompList->Next()) != NULL) {\r
    fOutput->Add(pObj);\r
    count++;\r
  }\r
  Printf("UserCreateOutputObjects(): Number of output objects: %d \n", count);\r
\r
  if(fUseCentrality) {\r
    Printf("Use Centrality - Bin %d", fUseCentralityBin);\r
  }\r
}\r
\r
//_____________________________________________________________________________\r
void AlidNdPtTask::UserExec(Option_t *) \r
{\r
  //\r
  // Called for each event\r
  //\r
\r
  // ESD event\r
  fESD = (AliESDEvent*) (InputEvent());\r
  if (!fESD) {\r
    Printf("ERROR: ESD event not available");\r
    return;\r
  }\r
\r
  // MC event\r
  if(fUseMCInfo) {\r
    fMC = MCEvent();\r
    if (!fMC) {\r
      Printf("ERROR: MC event not available");\r
      return;\r
    }\r
  }\r
\r
  // Process analysis\r
  Bool_t process = kTRUE;\r
\r
  // Check fo centrality\r
  if (fUseCentrality) {\r
    if ( CalculateCentralityBin() != fUseCentralityBin ) {\r
      process = kFALSE;\r
    }\r
  }\r
\r
  if (process) {\r
    AlidNdPt *pObj = 0;\r
    fPitList->Reset();\r
    while((pObj = (AlidNdPt *)fPitList->Next()) != NULL) {\r
      pObj->Process(fESD,fMC);\r
    }\r
  }\r
\r
\r
  // Post output data.\r
  PostData(1, fOutput);\r
}\r
\r
//_____________________________________________________________________________\r
void AlidNdPtTask::Terminate(Option_t *) \r
{\r
  // Called one at the end \r
  \r
  // check output data\r
  fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
  if (!fOutput) {\r
    Printf("ERROR: AlidNdPtTask::Terminate(): Output data not avaiable GetOutputData(0)==0x0 ..." );\r
    return;\r
  }\r
}\r
\r
//________________________________________________________________________\r
Int_t AlidNdPtTask::CalculateCentralityBin(){\r
  // Get Centrality bin\r
\r
  Int_t centrality = -1;\r
  Float_t centralityF = -1;\r
\r
  if (fUseCentrality == 0)\r
    return centrality;\r
\r
  AliCentrality *esdCentrality = fESD->GetCentrality();\r
    \r
  // New : 2010-11-18 JMT \r
  if ( fUseCentrality == 1 )\r
    centralityF = esdCentrality->GetCentralityPercentile("V0M");\r
  else if ( fUseCentrality == 2 )\r
    centralityF = esdCentrality->GetCentralityPercentile("CL1");\r
  else if ( fUseCentrality == 3 )\r
    centralityF = esdCentrality->GetCentralityPercentile("TRK"); \r
  if (centralityF == 0.)\r
    centralityF = 100.;\r
\r
  if      ( centralityF >=  0. && centralityF <   5.) centrality =  0;\r
  else if ( centralityF >=  5. && centralityF <  10.) centrality =  5;\r
  else if ( centralityF >= 10. && centralityF <  20.) centrality = 10;\r
  else if ( centralityF >= 20. && centralityF <  30.) centrality = 20;\r
  else if ( centralityF >= 30. && centralityF <  40.) centrality = 30;\r
  else if ( centralityF >= 40. && centralityF <  50.) centrality = 40;\r
  else if ( centralityF >= 50. && centralityF <  60.) centrality = 50;\r
  else if ( centralityF >= 60. && centralityF <  70.) centrality = 60;\r
  else if ( centralityF >= 70. && centralityF <  80.) centrality = 70;\r
  else if ( centralityF >= 80. && centralityF <  90.) centrality = 80;\r
  else if ( centralityF >= 90. && centralityF <=100.) centrality = 90;\r
  \r
  return centrality;\r
}\r
Commit	Line	Data
0aaa8b91	1	/**************************************************************************\r
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
	3	* *\r
	4	* Author: The ALICE Off-line Project. *\r
	5	* Contributors are mentioned in the code where appropriate. *\r
	6	* *\r
	7	* Permission to use, copy, modify and distribute this software and its *\r
	8	* documentation strictly for non-commercial purposes is hereby granted *\r
	9	* without fee, provided that the above copyright notice appears in all *\r
	10	* copies and that both the copyright notice and this permission notice *\r
	11	* appear in the supporting documentation. The authors make no claims *\r
	12	* about the suitability of this software for any purpose. It is *\r
	13	* provided "as is" without express or implied warranty. *\r
	14	**************************************************************************/\r
	15	\r
	16	#include "iostream"\r
	17	\r
	18	#include "TChain.h"\r
	19	#include "TTree.h"\r
	20	#include "TH1F.h"\r
	21	#include "TCanvas.h"\r
	22	#include "TList.h"\r
	23	#include "TFile.h"\r
	24	\r
	25	#include "AliAnalysisTask.h"\r
	26	#include "AliAnalysisManager.h"\r
	27	#include "AliESDEvent.h"\r
	28	#include "AliESDInputHandler.h"\r
	29	#include "AliESDVertex.h"\r
	30	#include "AliTracker.h"\r
	31	#include "AliGeomManager.h"\r
	32	\r
b93ccf1c	33	#include "AliCentrality.h"\r
0487c490	34	#include "AliESDVZERO.h"\r
	35	#include "AliMultiplicity.h"\r
	36	\r
0aaa8b91	37	#include "AliESDtrackCuts.h"\r
	38	#include "AliMCEventHandler.h"\r
	39	#include "dNdPt/AlidNdPt.h"\r
	40	#include "dNdPt/AlidNdPtEventCuts.h"\r
	41	#include "dNdPt/AlidNdPtAcceptanceCuts.h"\r
	42	\r
	43	#include "dNdPt/AlidNdPtTask.h"\r
	44	\r
	45	using namespace std;\r
	46	\r
	47	ClassImp(AlidNdPtTask)\r
	48	\r
	49	//_____________________________________________________________________________\r
	50	AlidNdPtTask::AlidNdPtTask(const char *name) \r
de93884d	51	: AliAnalysisTaskSE(name)\r
0aaa8b91	52	, fESD(0)\r
	53	, fMC(0)\r
	54	, fOutput(0)\r
	55	, fPitList(0)\r
	56	, fCompList(0)\r
	57	, fUseMCInfo(kFALSE)\r
0487c490	58	, fUseCentrality(0) // default = off\r
0487c490	59	, fUseCentralityBin(0)\r
0aaa8b91	60	{\r
	61	// Constructor\r
	62	\r
	63	// Define input and output slots here\r
de93884d	64	DefineOutput(1, TList::Class());\r
0aaa8b91	65	\r
	66	// create the list for comparison objects\r
	67	fCompList = new TList;\r
	68	}\r
	69	\r
	70	//_____________________________________________________________________________\r
	71	AlidNdPtTask::~AlidNdPtTask()\r
	72	{\r
	73	if(fOutput) delete fOutput; fOutput =0; \r
de93884d	74	if(fCompList) delete fCompList; fCompList =0; \r
0aaa8b91	75	}\r
	76	\r
	77	//____________________________________________________________________________\r
	78	Bool_t AlidNdPtTask::Notify()\r
	79	{\r
	80	static Int_t count = 0;\r
	81	count++;\r
695facdf	82	//Printf("Processing %d. file: %s", count, ((TTree*) GetInputData(0))->GetCurrentFile()->GetName());\r
	83	TTree chain = (TChain)GetInputData(0);\r
	84	if(chain)\r
	85	Printf("Processing %d. file: %s", count, chain->GetCurrentFile()->GetName());\r
0aaa8b91	86	\r
695facdf	87	/*\r
a8ac5525	88	TChain chain = (TChain)GetInputData(0);\r
	89	AliESDInputHandler esdH = dynamic_cast<AliESDInputHandler> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
	90	if (!esdH) {\r
	91	Printf("ERROR: Could not get ESDInputHandler");\r
	92	return kFALSE;\r
	93	} else {\r
a8ac5525	94	if(chain)\r
7f7dd416	95	Printf("chain->GetCurrentFile()->GetName() %s", chain->GetCurrentFile()->GetName());\r
a8ac5525	96	}\r
695facdf	97	*/\r
a8ac5525	98	\r
0aaa8b91	99	return kTRUE;\r
	100	}\r
	101	\r
0aaa8b91	102	//_____________________________________________________________________________\r
	103	Bool_t AlidNdPtTask::AddAnalysisObject(AlidNdPt *pObj) \r
	104	{\r
	105	// add analysis object to the list\r
	106	if(pObj == 0) {\r
	107	Printf("ERROR: Could not add comparison object");\r
	108	return kFALSE;\r
	109	}\r
	110	\r
	111	// add object to the list\r
	112	fCompList->AddLast(pObj);\r
	113	\r
	114	return kTRUE;\r
	115	}\r
	116	\r
	117	//_____________________________________________________________________________\r
de93884d	118	void AlidNdPtTask::UserCreateOutputObjects()\r
0aaa8b91	119	{\r
	120	// Create histograms\r
	121	// Called once\r
	122	\r
de93884d	123	OpenFile(1, "RECREATE");\r
0aaa8b91	124	\r
	125	//\r
	126	// create output list\r
	127	//\r
	128	fOutput = new TList;\r
	129	fOutput->SetOwner();\r
	130	fPitList = fOutput->MakeIterator();\r
	131	\r
	132	// add dNdPt analysis objects to the output\r
	133	AlidNdPt *pObj=0;\r
	134	Int_t count=0;\r
	135	TIterator *pitCompList = fCompList->MakeIterator();\r
	136	pitCompList->Reset();\r
	137	while(( pObj = (AlidNdPt *)pitCompList->Next()) != NULL) {\r
	138	fOutput->Add(pObj);\r
	139	count++;\r
	140	}\r
de93884d	141	Printf("UserCreateOutputObjects(): Number of output objects: %d \n", count);\r
0487c490	142	\r
	143	if(fUseCentrality) {\r
	144	Printf("Use Centrality - Bin %d", fUseCentralityBin);\r
	145	}\r
0aaa8b91	146	}\r
	147	\r
	148	//_____________________________________________________________________________\r
de93884d	149	void AlidNdPtTask::UserExec(Option_t *) \r
0aaa8b91	150	{\r
de93884d	151	//\r
0aaa8b91	152	// Called for each event\r
de93884d	153	//\r
	154	\r
	155	// ESD event\r
	156	fESD = (AliESDEvent*) (InputEvent());\r
0aaa8b91	157	if (!fESD) {\r
	158	Printf("ERROR: ESD event not available");\r
	159	return;\r
	160	}\r
	161	\r
de93884d	162	// MC event\r
	163	if(fUseMCInfo) {\r
	164	fMC = MCEvent();\r
	165	if (!fMC) {\r
	166	Printf("ERROR: MC event not available");\r
	167	return;\r
	168	}\r
	169	}\r
	170	\r
0aaa8b91	171	// Process analysis\r
0487c490	172	Bool_t process = kTRUE;\r
	173	\r
	174	// Check fo centrality\r
	175	if (fUseCentrality) {\r
	176	if ( CalculateCentralityBin() != fUseCentralityBin ) {\r
	177	process = kFALSE;\r
	178	}\r
0aaa8b91	179	}\r
0aaa8b91	180	\r
0487c490	181	if (process) {\r
	182	AlidNdPt *pObj = 0;\r
	183	fPitList->Reset();\r
	184	while((pObj = (AlidNdPt *)fPitList->Next()) != NULL) {\r
	185	pObj->Process(fESD,fMC);\r
	186	}\r
	187	}\r
	188	\r
	189	\r
0aaa8b91	190	// Post output data.\r
de93884d	191	PostData(1, fOutput);\r
0aaa8b91	192	}\r
	193	\r
	194	//_____________________________________________________________________________\r
	195	void AlidNdPtTask::Terminate(Option_t *) \r
	196	{\r
	197	// Called one at the end \r
	198	\r
	199	// check output data\r
de93884d	200	fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
0aaa8b91	201	if (!fOutput) {\r
	202	Printf("ERROR: AlidNdPtTask::Terminate(): Output data not avaiable GetOutputData(0)==0x0 ..." );\r
	203	return;\r
	204	}\r
	205	}\r
0487c490	206	\r
	207	//________________________________________________________________________\r
	208	Int_t AlidNdPtTask::CalculateCentralityBin(){\r
	209	// Get Centrality bin\r
	210	\r
	211	Int_t centrality = -1;\r
515964cb	212	Float_t centralityF = -1;\r
0487c490	213	\r
	214	if (fUseCentrality == 0)\r
	215	return centrality;\r
	216	\r
b93ccf1c	217	AliCentrality *esdCentrality = fESD->GetCentrality();\r
515964cb	218	\r
	219	// New : 2010-11-18 JMT \r
	220	if ( fUseCentrality == 1 )\r
	221	centralityF = esdCentrality->GetCentralityPercentile("V0M");\r
	222	else if ( fUseCentrality == 2 )\r
	223	centralityF = esdCentrality->GetCentralityPercentile("CL1");\r
f4f8cae4	224	else if ( fUseCentrality == 3 )\r
f4f8cae4	225	centralityF = esdCentrality->GetCentralityPercentile("TRK"); \r
515964cb	226	if (centralityF == 0.)\r
	227	centralityF = 100.;\r
	228	\r
	229	if ( centralityF >= 0. && centralityF < 5.) centrality = 0;\r
	230	else if ( centralityF >= 5. && centralityF < 10.) centrality = 5;\r
	231	else if ( centralityF >= 10. && centralityF < 20.) centrality = 10;\r
	232	else if ( centralityF >= 20. && centralityF < 30.) centrality = 20;\r
	233	else if ( centralityF >= 30. && centralityF < 40.) centrality = 30;\r
	234	else if ( centralityF >= 40. && centralityF < 50.) centrality = 40;\r
	235	else if ( centralityF >= 50. && centralityF < 60.) centrality = 50;\r
	236	else if ( centralityF >= 60. && centralityF < 70.) centrality = 60;\r
	237	else if ( centralityF >= 70. && centralityF < 80.) centrality = 70;\r
	238	else if ( centralityF >= 80. && centralityF < 90.) centrality = 80;\r
	239	else if ( centralityF >= 90. && centralityF <=100.) centrality = 90;\r
0487c490	240	\r
	241	return centrality;\r
	242	}\r