[u/mrichter/AliRoot.git] / PWGPP / TPC / AliPerformanceTask.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
//------------------------------------------------------------------------------\r
// Implementation of the AliPerformanceTask class. It checks reconstruction performance \r
// for the reconstructed vs MC particle tracks under several conditions. For real data \r
// the control QA histograms are filled.\r
//\r
// The comparison output objects deriving from AliPerformanceObject \r
// (e.g. AliPerformanceRes, AliPerformanceEff, AliPerformanceDEdx, AliPerformanceDCA ...) \r
// are stored in the output file (details in description of these classes).\r
// \r
// Author: J.Otwinowski 01/04/2009 \r
// Changes by M.Knichel 15/10/2010\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
#include "TSystem.h"\r
\r
#include "AliAnalysisTask.h"\r
#include "AliAnalysisManager.h"\r
#include "AliESDEvent.h"\r
#include "AliESDfriend.h"\r
#include "AliMCEvent.h"\r
#include "AliESDInputHandler.h"\r
#include "AliMCEventHandler.h"\r
#include "AliESDVertex.h"\r
#include "AliMagF.h"\r
#include "AliTracker.h"\r
#include "AliGeomManager.h"\r
#include "AliCDBManager.h"\r
\r
#include "AliCentrality.h"\r
#include "AliESDVZERO.h"\r
#include "AliMultiplicity.h"\r
\r
#include "AliMCInfo.h"\r
#include "AliESDRecInfo.h"\r
#include "AliMCInfoCuts.h"\r
#include "AliRecInfoCuts.h"\r
#include "AliPerformanceObject.h"\r
#include "AliTPCPerformanceSummary.h"\r
#include "AliPerformanceTPC.h"\r
#include "AliPerformanceDEdx.h"\r
#include "AliPerformanceMatch.h"\r
#include "AliPerformanceTask.h"\r
\r
\r
using namespace std;\r
\r
ClassImp(AliPerformanceTask)\r
\r
//_____________________________________________________________________________\r
AliPerformanceTask::AliPerformanceTask() \r
  : AliAnalysisTaskSE("Performance")\r
  , fESD(0)\r
  , fESDfriend(0)\r
  , fMC(0)\r
  , fOutput(0)\r
  , fOutputSummary(0)\r
  , fPitList(0)\r
  , fCompList(0)\r
  , fUseMCInfo(kFALSE)\r
  , fUseESDfriend(kFALSE)\r
  , fUseHLT(kFALSE)\r
  , fUseTerminate(kTRUE)\r
  , fUseCentrality(0)\r
  , fUseOCDB(kTRUE)\r
  , fUseCentralityBin(0)\r
{\r
  // Dummy Constructor\r
  // should not be used\r
}\r
\r
//_____________________________________________________________________________\r
AliPerformanceTask::AliPerformanceTask(const char *name, const char */*title*/) \r
  : AliAnalysisTaskSE(name)\r
  , fESD(0)\r
  , fESDfriend(0)\r
  , fMC(0)\r
  , fOutput(0)\r
  , fOutputSummary(0)\r
  , fPitList(0)\r
  , fCompList(0)\r
  , fUseMCInfo(kFALSE)\r
  , fUseESDfriend(kFALSE)\r
  , fUseHLT(kFALSE)\r
  , fUseTerminate(kTRUE)\r
  , fUseCentrality(0)\r
  , fUseOCDB(kTRUE)\r
  , fUseCentralityBin(0)\r
{\r
  // Constructor\r
\r
  // Define input and output slots here\r
  DefineOutput(0, TTree::Class());\r
  DefineOutput(1, TList::Class());\r
\r
  // create the list for comparison objects\r
  fCompList = new TList;\r
}\r
\r
//_____________________________________________________________________________\r
AliPerformanceTask::~AliPerformanceTask()\r
{\r
  if (!AliAnalysisManager::GetAnalysisManager()->IsProofMode()) {\r
    if (fOutput)     delete fOutput;    fOutput   = 0; \r
    if (fOutputSummary) delete fOutputSummary; fOutputSummary = 0;\r
    if (fCompList)   delete fCompList;  fCompList = 0; \r
  }\r
}\r
\r
//_____________________________________________________________________________\r
Bool_t AliPerformanceTask::AddPerformanceObject(AliPerformanceObject *pObj) \r
{\r
  // add comparison 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 AliPerformanceTask::UserCreateOutputObjects()\r
{\r
  // Create histograms\r
  // Called once\r
\r
  // create output list\r
  fOutput = new TList;\r
  fOutput->SetOwner();\r
  fPitList = fOutput->MakeIterator();\r
  \r
  // create output list\r
  //fOutputSummary = new TTree;\r
  \r
  // add comparison objects to the output\r
  AliPerformanceObject *pObj=0;\r
  Int_t count=0;\r
  TIterator *pitCompList = fCompList->MakeIterator();\r
  pitCompList->Reset();\r
  while(( pObj = (AliPerformanceObject *)pitCompList->Next()) != NULL) {\r
    fOutput->Add(pObj);\r
    count++;\r
  }\r
  Printf("UserCreateOutputObjects(): Number of output comparison objects: %d \n", count);\r
  \r
  PostData(1, fOutput);  \r
  PostData(0, fOutputSummary);  \r
}\r
\r
//_____________________________________________________________________________\r
void AliPerformanceTask::UserExec(Option_t *) \r
{\r
  // Main loop\r
  // Called for each event\r
\r
\r
  // Decide whether to use HLT or Offline ESD\r
  if(fUseHLT){\r
\r
    AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> \r
      (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
    \r
    if (!esdH) {\r
      printf("ERROR: Could not get ESDInputHandler");\r
      return;\r
    } \r
    fESD = esdH->GetHLTEvent();\r
  }// end if fUseHLT\r
  else  \r
    fESD = (AliESDEvent*) (InputEvent());\r
\r
  if(fUseESDfriend)\r
    {\r
	  if (fUseHLT)\r
	  {\r
		AliESDEvent *fESDoffline;\r
	    fESDoffline = (AliESDEvent*) (InputEvent());\r
		fESDfriend = static_cast<AliESDfriend*>(fESDoffline->FindListObject("AliESDfriend"));\r
	  }\r
	  else\r
	  {\r
        fESDfriend = static_cast<AliESDfriend*>(fESD->FindListObject("AliESDfriend"));\r
	  }\r
      if(!fESDfriend) {\r
        Printf("ERROR: ESD friends not available");\r
      }\r
    }\r
  \r
  if(fUseMCInfo) {\r
      fMC = MCEvent();\r
  }  \r
\r
\r
  if (!fESD) {\r
    Printf("ERROR: ESD event not available");\r
    return;\r
  }\r
  \r
  if (fUseMCInfo && !fMC) {\r
    Printf("ERROR: MC event not available");\r
    return;\r
  }\r
\r
  if(fUseESDfriend)\r
  {\r
    if(!fESDfriend) {\r
    Printf("ERROR: ESD friends not available");\r
    }\r
  }\r
\r
  // Process analysis\r
  Bool_t process = kTRUE;\r
\r
  // Check for centrality\r
  if (fUseCentrality) {\r
    if ( CalculateCentralityBin() != fUseCentralityBin ) {\r
      process = kFALSE;\r
    }\r
  }\r
\r
  // Process comparison\r
  if (process) {\r
    AliPerformanceObject *pObj=0;\r
    fPitList->Reset();\r
    while(( pObj = (AliPerformanceObject *)fPitList->Next()) != NULL) {\r
          AliInfo(pObj->GetName());\r
          pObj->Exec(fMC,fESD,fESDfriend,fUseMCInfo,fUseESDfriend);\r
    }\r
  }\r
\r
  // Post output data.\r
  PostData(1, fOutput);\r
}\r
\r
//_____________________________________________________________________________\r
void AliPerformanceTask::Terminate(Option_t *) \r
{\r
  // Called once at the end \r
\r
  if ( !fUseTerminate )\r
    return;\r
  \r
  // check output data\r
    fOutputSummary = dynamic_cast<TTree*> (GetOutputData(0));\r
    fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
    if (!fOutput) {\r
        Printf("ERROR: AliPerformanceTask::Terminate(): fOutput data not available  ..." );\r
        return;\r
   }\r
    if (fOutputSummary) { delete fOutputSummary; fOutputSummary=0; }      \r
    AliPerformanceObject* pObj=0;\r
    AliPerformanceTPC*  pTPC = 0;\r
    AliPerformanceDEdx* pDEdx = 0;\r
    AliPerformanceMatch* pMatch = 0;\r
    AliPerformanceMatch* pPull = 0;\r
    AliPerformanceMatch* pConstrain = 0;\r
    TIterator* itOut = fOutput->MakeIterator();\r
    itOut->Reset();\r
    while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) { \r
      pObj->AnalyseFinal();\r
      /*      if (!  pTPC)  {    pTPC = dynamic_cast<AliPerformanceTPC*>(pObj); }\r
        if (! pDEdx)  {   pDEdx = dynamic_cast<AliPerformanceDEdx*>(pObj); }\r
        if (! pMatch) {  pMatch = dynamic_cast<AliPerformanceMatch*>(pObj); }\r
        if ((! pPull) && pMatch ) {  pPull = dynamic_cast<AliPerformanceMatch*>(pObj);*/\r
\r
	if (!strcmp(pObj->GetName(),"AliPerformanceTPC"))  {    pTPC = dynamic_cast<AliPerformanceTPC*>(pObj); }\r
        if (!strcmp(pObj->GetName(),"AliPerformanceDEdxTPCInner"))  {   pDEdx = dynamic_cast<AliPerformanceDEdx*>(pObj); }\r
        if (!strcmp(pObj->GetName(),"AliPerformanceMatchTPCITS")) {  pMatch = dynamic_cast<AliPerformanceMatch*>(pObj); }\r
        if (!strcmp(pObj->GetName(),"AliPerformanceMatchITSTPC")) {  pPull = dynamic_cast<AliPerformanceMatch*>(pObj);}\r
        if (!strcmp(pObj->GetName(),"AliPerformanceMatchTPCConstrain")) {  pConstrain = dynamic_cast<AliPerformanceMatch*>(pObj);}\r
    }\r
  \r
   \r
    if(!fUseOCDB)  { \r
      printf("DO NOT USE OCDB \n");\r
      return;\r
    }\r
  \r
    if (! AliCDBManager::Instance()->GetDefaultStorage()) { AliCDBManager::Instance()->SetDefaultStorage("raw://"); }\r
    TUUID uuid;\r
    TString tmpFile = gSystem->TempDirectory() + TString("/TPCQASummary.") + uuid.AsString() + TString(".root");\r
    AliTPCPerformanceSummary::WriteToFile(pTPC, pDEdx, pMatch, pPull, pConstrain, tmpFile.Data());\r
    TChain* chain = new TChain("tpcQA");\r
    if(!chain) return;\r
    chain->Add(tmpFile.Data());\r
    TTree *tree = chain->CopyTree("1");\r
    if (chain) { delete chain; chain=0; }\r
    fOutputSummary = tree;\r
      \r
     // Post output data.\r
     PostData(0, fOutputSummary);\r
\r
}\r
\r
//_____________________________________________________________________________\r
void AliPerformanceTask::FinishTaskOutput()\r
{\r
    // called once at the end of each job (on the workernode)\r
    //\r
    // projects THnSparse to TH1,2,3\r
    \r
    fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
    if (!fOutput) {\r
        Printf("ERROR: AliPerformanceTask::FinishTaskOutput(): fOutput data not available  ..." );\r
        return;\r
   }\r
\r
      AliPerformanceObject* pObj=0;\r
      TIterator* itOut = fOutput->MakeIterator();  \r
      itOut->Reset();\r
      while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) {\r
          pObj->SetRunNumber(fCurrentRunNumber);\r
          pObj->Analyse();\r
      }\r
      \r
     // Post output data.\r
     PostData(1, fOutput);\r
}\r
\r
//_____________________________________________________________________________\r
Bool_t AliPerformanceTask::Notify()\r
{\r
  static Int_t count = 0;\r
  count++;\r
  Printf("Processing %d. file", count);\r
\r
  return kTRUE;\r
}\r
\r
//________________________________________________________________________\r
Int_t AliPerformanceTask::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
7cc34f08	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	//------------------------------------------------------------------------------\r
	17	// Implementation of the AliPerformanceTask class. It checks reconstruction performance \r
	18	// for the reconstructed vs MC particle tracks under several conditions. For real data \r
	19	// the control QA histograms are filled.\r
	20	//\r
	21	// The comparison output objects deriving from AliPerformanceObject \r
	22	// (e.g. AliPerformanceRes, AliPerformanceEff, AliPerformanceDEdx, AliPerformanceDCA ...) \r
	23	// are stored in the output file (details in description of these classes).\r
	24	// \r
	25	// Author: J.Otwinowski 01/04/2009 \r
814d192f	26	// Changes by M.Knichel 15/10/2010\r
7cc34f08	27	//------------------------------------------------------------------------------\r
	28	\r
	29	#include "iostream"\r
	30	\r
	31	#include "TChain.h"\r
	32	#include "TTree.h"\r
	33	#include "TH1F.h"\r
	34	#include "TCanvas.h"\r
	35	#include "TList.h"\r
	36	#include "TFile.h"\r
814d192f	37	#include "TSystem.h"\r
7cc34f08	38	\r
	39	#include "AliAnalysisTask.h"\r
	40	#include "AliAnalysisManager.h"\r
	41	#include "AliESDEvent.h"\r
	42	#include "AliESDfriend.h"\r
	43	#include "AliMCEvent.h"\r
	44	#include "AliESDInputHandler.h"\r
	45	#include "AliMCEventHandler.h"\r
	46	#include "AliESDVertex.h"\r
	47	#include "AliMagF.h"\r
	48	#include "AliTracker.h"\r
	49	#include "AliGeomManager.h"\r
52abd620	50	#include "AliCDBManager.h"\r
7cc34f08	51	\r
5da23c19	52	#include "AliCentrality.h"\r
36ace53b	53	#include "AliESDVZERO.h"\r
	54	#include "AliMultiplicity.h"\r
	55	\r
7cc34f08	56	#include "AliMCInfo.h"\r
	57	#include "AliESDRecInfo.h"\r
	58	#include "AliMCInfoCuts.h"\r
	59	#include "AliRecInfoCuts.h"\r
7cc34f08	60	#include "AliPerformanceObject.h"\r
814d192f	61	#include "AliTPCPerformanceSummary.h"\r
	62	#include "AliPerformanceTPC.h"\r
	63	#include "AliPerformanceDEdx.h"\r
f69eb048	64	#include "AliPerformanceMatch.h"\r
7cc34f08	65	#include "AliPerformanceTask.h"\r
7cc34f08	66	\r
814d192f	67	\r
7cc34f08	68	using namespace std;\r
	69	\r
	70	ClassImp(AliPerformanceTask)\r
	71	\r
	72	//_____________________________________________________________________________\r
	73	AliPerformanceTask::AliPerformanceTask() \r
a26a028b	74	: AliAnalysisTaskSE("Performance")\r
7cc34f08	75	, fESD(0)\r
	76	, fESDfriend(0)\r
	77	, fMC(0)\r
	78	, fOutput(0)\r
814d192f	79	, fOutputSummary(0)\r
7cc34f08	80	, fPitList(0)\r
	81	, fCompList(0)\r
	82	, fUseMCInfo(kFALSE)\r
	83	, fUseESDfriend(kFALSE)\r
636e5a51	84	, fUseHLT(kFALSE)\r
c1b69b58	85	, fUseTerminate(kTRUE)\r
36ace53b	86	, fUseCentrality(0)\r
19563116	87	, fUseOCDB(kTRUE)\r
1e9c263c	88	, fUseCentralityBin(0)\r
7cc34f08	89	{\r
	90	// Dummy Constructor\r
	91	// should not be used\r
	92	}\r
	93	\r
	94	//_____________________________________________________________________________\r
a26a028b	95	AliPerformanceTask::AliPerformanceTask(const char name, const char /title/) \r
a26a028b	96	: AliAnalysisTaskSE(name)\r
7cc34f08	97	, fESD(0)\r
	98	, fESDfriend(0)\r
	99	, fMC(0)\r
	100	, fOutput(0)\r
814d192f	101	, fOutputSummary(0)\r
7cc34f08	102	, fPitList(0)\r
	103	, fCompList(0)\r
	104	, fUseMCInfo(kFALSE)\r
	105	, fUseESDfriend(kFALSE)\r
636e5a51	106	, fUseHLT(kFALSE)\r
c1b69b58	107	, fUseTerminate(kTRUE)\r
36ace53b	108	, fUseCentrality(0)\r
19563116	109	, fUseOCDB(kTRUE)\r
1e9c263c	110	, fUseCentralityBin(0)\r
7cc34f08	111	{\r
	112	// Constructor\r
	113	\r
	114	// Define input and output slots here\r
10d5d625	115	DefineOutput(0, TTree::Class());\r
65b3a5e1	116	DefineOutput(1, TList::Class());\r
7cc34f08	117	\r
	118	// create the list for comparison objects\r
	119	fCompList = new TList;\r
	120	}\r
	121	\r
	122	//_____________________________________________________________________________\r
	123	AliPerformanceTask::~AliPerformanceTask()\r
	124	{\r
9c12af5f	125	if (!AliAnalysisManager::GetAnalysisManager()->IsProofMode()) {\r
	126	if (fOutput) delete fOutput; fOutput = 0; \r
	127	if (fOutputSummary) delete fOutputSummary; fOutputSummary = 0;\r
	128	if (fCompList) delete fCompList; fCompList = 0; \r
	129	}\r
7cc34f08	130	}\r
7cc34f08	131	\r
7cc34f08	132	//_____________________________________________________________________________\r
	133	Bool_t AliPerformanceTask::AddPerformanceObject(AliPerformanceObject *pObj) \r
	134	{\r
	135	// add comparison object to the list\r
	136	if(pObj == 0) {\r
	137	Printf("ERROR: Could not add comparison object");\r
	138	return kFALSE;\r
	139	}\r
	140	\r
	141	// add object to the list\r
	142	fCompList->AddLast(pObj);\r
	143	\r
	144	return kTRUE;\r
	145	}\r
	146	\r
	147	//_____________________________________________________________________________\r
a26a028b	148	void AliPerformanceTask::UserCreateOutputObjects()\r
7cc34f08	149	{\r
	150	// Create histograms\r
	151	// Called once\r
	152	\r
	153	// create output list\r
	154	fOutput = new TList;\r
	155	fOutput->SetOwner();\r
	156	fPitList = fOutput->MakeIterator();\r
814d192f	157	\r
	158	// create output list\r
	159	//fOutputSummary = new TTree;\r
	160	\r
7cc34f08	161	// add comparison objects to the output\r
	162	AliPerformanceObject *pObj=0;\r
	163	Int_t count=0;\r
	164	TIterator *pitCompList = fCompList->MakeIterator();\r
	165	pitCompList->Reset();\r
	166	while(( pObj = (AliPerformanceObject *)pitCompList->Next()) != NULL) {\r
	167	fOutput->Add(pObj);\r
	168	count++;\r
	169	}\r
783eab45	170	Printf("UserCreateOutputObjects(): Number of output comparison objects: %d \n", count);\r
814d192f	171	\r
814d192f	172	PostData(1, fOutput); \r
10d5d625	173	PostData(0, fOutputSummary); \r
7cc34f08	174	}\r
	175	\r
	176	//_____________________________________________________________________________\r
a26a028b	177	void AliPerformanceTask::UserExec(Option_t *) \r
7cc34f08	178	{\r
	179	// Main loop\r
	180	// Called for each event\r
636e5a51	181	\r
af960f39	182	\r
636e5a51	183	// Decide whether to use HLT or Offline ESD\r
	184	if(fUseHLT){\r
	185	\r
	186	AliESDInputHandler esdH = dynamic_cast<AliESDInputHandler> \r
	187	(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
	188	\r
	189	if (!esdH) {\r
	190	printf("ERROR: Could not get ESDInputHandler");\r
	191	return;\r
	192	} \r
	193	fESD = esdH->GetHLTEvent();\r
	194	}// end if fUseHLT\r
	195	else \r
	196	fESD = (AliESDEvent*) (InputEvent());\r
	197	\r
a26a028b	198	if(fUseESDfriend)\r
a26a028b	199	{\r
e95d4942	200	if (fUseHLT)\r
	201	{\r
	202	AliESDEvent *fESDoffline;\r
	203	fESDoffline = (AliESDEvent*) (InputEvent());\r
	204	fESDfriend = static_cast<AliESDfriend*>(fESDoffline->FindListObject("AliESDfriend"));\r
	205	}\r
	206	else\r
	207	{\r
	208	fESDfriend = static_cast<AliESDfriend*>(fESD->FindListObject("AliESDfriend"));\r
	209	}\r
a26a028b	210	if(!fESDfriend) {\r
	211	Printf("ERROR: ESD friends not available");\r
	212	}\r
	213	}\r
	214	\r
	215	if(fUseMCInfo) {\r
	216	fMC = MCEvent();\r
	217	} \r
7cc34f08	218	\r
7cc34f08	219	\r
	220	if (!fESD) {\r
	221	Printf("ERROR: ESD event not available");\r
	222	return;\r
	223	}\r
	224	\r
	225	if (fUseMCInfo && !fMC) {\r
	226	Printf("ERROR: MC event not available");\r
	227	return;\r
	228	}\r
	229	\r
	230	if(fUseESDfriend)\r
	231	{\r
	232	if(!fESDfriend) {\r
	233	Printf("ERROR: ESD friends not available");\r
	234	}\r
	235	}\r
	236	\r
1e9c263c	237	// Process analysis\r
1e9c263c	238	Bool_t process = kTRUE;\r
36ace53b	239	\r
af960f39	240	// Check for centrality\r
1e9c263c	241	if (fUseCentrality) {\r
	242	if ( CalculateCentralityBin() != fUseCentralityBin ) {\r
	243	process = kFALSE;\r
	244	}\r
	245	}\r
36ace53b	246	\r
7cc34f08	247	// Process comparison\r
1e9c263c	248	if (process) {\r
	249	AliPerformanceObject *pObj=0;\r
	250	fPitList->Reset();\r
	251	while(( pObj = (AliPerformanceObject *)fPitList->Next()) != NULL) {\r
53410273	252	AliInfo(pObj->GetName());\r
1e9c263c	253	pObj->Exec(fMC,fESD,fESDfriend,fUseMCInfo,fUseESDfriend);\r
36ace53b	254	}\r
7cc34f08	255	}\r
	256	\r
	257	// Post output data.\r
65b3a5e1	258	PostData(1, fOutput);\r
7cc34f08	259	}\r
	260	\r
	261	//_____________________________________________________________________________\r
	262	void AliPerformanceTask::Terminate(Option_t *) \r
	263	{\r
9ca899e5	264	// Called once at the end \r
c1b69b58	265	\r
	266	if ( !fUseTerminate )\r
	267	return;\r
1833a193	268	\r
7cc34f08	269	// check output data\r
10d5d625	270	fOutputSummary = dynamic_cast<TTree*> (GetOutputData(0));\r
814d192f	271	fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
814d192f	272	if (!fOutput) {\r
d41038d7	273	Printf("ERROR: AliPerformanceTask::Terminate(): fOutput data not available ..." );\r
814d192f	274	return;\r
	275	}\r
	276	if (fOutputSummary) { delete fOutputSummary; fOutputSummary=0; } \r
	277	AliPerformanceObject* pObj=0;\r
	278	AliPerformanceTPC* pTPC = 0;\r
	279	AliPerformanceDEdx* pDEdx = 0;\r
f69eb048	280	AliPerformanceMatch* pMatch = 0;\r
1833a193	281	AliPerformanceMatch* pPull = 0;\r
9f5fbaba	282	AliPerformanceMatch* pConstrain = 0;\r
814d192f	283	TIterator* itOut = fOutput->MakeIterator();\r
	284	itOut->Reset();\r
	285	while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) { \r
1833a193	286	pObj->AnalyseFinal();\r
1833a193	287	/* if (! pTPC) { pTPC = dynamic_cast<AliPerformanceTPC*>(pObj); }\r
f69eb048	288	if (! pDEdx) { pDEdx = dynamic_cast<AliPerformanceDEdx*>(pObj); }\r
f69eb048	289	if (! pMatch) { pMatch = dynamic_cast<AliPerformanceMatch*>(pObj); }\r
1833a193	290	if ((! pPull) && pMatch ) { pPull = dynamic_cast<AliPerformanceMatch>(pObj);/\r
	291	\r
	292	if (!strcmp(pObj->GetName(),"AliPerformanceTPC")) { pTPC = dynamic_cast<AliPerformanceTPC*>(pObj); }\r
	293	if (!strcmp(pObj->GetName(),"AliPerformanceDEdxTPCInner")) { pDEdx = dynamic_cast<AliPerformanceDEdx*>(pObj); }\r
	294	if (!strcmp(pObj->GetName(),"AliPerformanceMatchTPCITS")) { pMatch = dynamic_cast<AliPerformanceMatch*>(pObj); }\r
9f5fbaba	295	if (!strcmp(pObj->GetName(),"AliPerformanceMatchITSTPC")) { pPull = dynamic_cast<AliPerformanceMatch*>(pObj);}\r
9f5fbaba	296	if (!strcmp(pObj->GetName(),"AliPerformanceMatchTPCConstrain")) { pConstrain = dynamic_cast<AliPerformanceMatch*>(pObj);}\r
814d192f	297	}\r
9f5fbaba	298	\r
19563116	299	\r
	300	if(!fUseOCDB) { \r
	301	printf("DO NOT USE OCDB \n");\r
	302	return;\r
	303	}\r
	304	\r
d41038d7	305	if (! AliCDBManager::Instance()->GetDefaultStorage()) { AliCDBManager::Instance()->SetDefaultStorage("raw://"); }\r
814d192f	306	TUUID uuid;\r
814d192f	307	TString tmpFile = gSystem->TempDirectory() + TString("/TPCQASummary.") + uuid.AsString() + TString(".root");\r
9f5fbaba	308	AliTPCPerformanceSummary::WriteToFile(pTPC, pDEdx, pMatch, pPull, pConstrain, tmpFile.Data());\r
814d192f	309	TChain* chain = new TChain("tpcQA");\r
18e588e9	310	if(!chain) return;\r
814d192f	311	chain->Add(tmpFile.Data());\r
	312	TTree *tree = chain->CopyTree("1");\r
	313	if (chain) { delete chain; chain=0; }\r
	314	fOutputSummary = tree;\r
	315	\r
	316	// Post output data.\r
10d5d625	317	PostData(0, fOutputSummary);\r
636e5a51	318	\r
814d192f	319	}\r
	320	\r
	321	//_____________________________________________________________________________\r
	322	void AliPerformanceTask::FinishTaskOutput()\r
	323	{\r
	324	// called once at the end of each job (on the workernode)\r
	325	//\r
	326	// projects THnSparse to TH1,2,3\r
	327	\r
	328	fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
	329	if (!fOutput) {\r
	330	Printf("ERROR: AliPerformanceTask::FinishTaskOutput(): fOutput data not available ..." );\r
	331	return;\r
	332	}\r
636e5a51	333	\r
814d192f	334	AliPerformanceObject* pObj=0;\r
	335	TIterator* itOut = fOutput->MakeIterator(); \r
	336	itOut->Reset();\r
	337	while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) {\r
	338	pObj->SetRunNumber(fCurrentRunNumber);\r
	339	pObj->Analyse();\r
	340	}\r
	341	\r
	342	// Post output data.\r
	343	PostData(1, fOutput);\r
7cc34f08	344	}\r
	345	\r
	346	//_____________________________________________________________________________\r
	347	Bool_t AliPerformanceTask::Notify()\r
	348	{\r
	349	static Int_t count = 0;\r
	350	count++;\r
	351	Printf("Processing %d. file", count);\r
	352	\r
	353	return kTRUE;\r
	354	}\r
36ace53b	355	\r
36ace53b	356	//________________________________________________________________________\r
1e9c263c	357	Int_t AliPerformanceTask::CalculateCentralityBin(){\r
36ace53b	358	// Get Centrality bin\r
	359	\r
	360	Int_t centrality = -1;\r
1e9c263c	361	Float_t centralityF = -1;\r
36ace53b	362	\r
	363	if (fUseCentrality == 0)\r
	364	return centrality;\r
	365	\r
5da23c19	366	AliCentrality *esdCentrality = fESD->GetCentrality();\r
1e9c263c	367	\r
	368	// New : 2010-11-18 JMT \r
	369	if ( fUseCentrality == 1 )\r
	370	centralityF = esdCentrality->GetCentralityPercentile("V0M");\r
	371	else if ( fUseCentrality == 2 )\r
	372	centralityF = esdCentrality->GetCentralityPercentile("CL1");\r
	373	else if ( fUseCentrality == 3 )\r
	374	centralityF = esdCentrality->GetCentralityPercentile("TRK"); \r
	375	if (centralityF == 0.)\r
	376	centralityF = 100.;\r
	377	\r
	378	if ( centralityF >= 0. && centralityF < 5.) centrality = 0;\r
	379	else if ( centralityF >= 5. && centralityF < 10.) centrality = 5;\r
	380	else if ( centralityF >= 10. && centralityF < 20.) centrality = 10;\r
	381	else if ( centralityF >= 20. && centralityF < 30.) centrality = 20;\r
	382	else if ( centralityF >= 30. && centralityF < 40.) centrality = 30;\r
	383	else if ( centralityF >= 40. && centralityF < 50.) centrality = 40;\r
	384	else if ( centralityF >= 50. && centralityF < 60.) centrality = 50;\r
	385	else if ( centralityF >= 60. && centralityF < 70.) centrality = 60;\r
	386	else if ( centralityF >= 70. && centralityF < 80.) centrality = 70;\r
	387	else if ( centralityF >= 80. && centralityF < 90.) centrality = 80;\r
	388	else if ( centralityF >= 90. && centralityF <=100.) centrality = 90;\r
36ace53b	389	\r
	390	return centrality;\r
	391	}\r