[u/mrichter/AliRoot.git] / PWG2 / FLOW / AliAnalysisTaskCumulants.cxx

/*************************************************************************
* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  * 
**************************************************************************/

/**************************************
 * analysis task for cumulant method  * 
 *                                    * 
 * authors: Naomi van der Kolk        *
 *           (kolk@nikhef.nl)         *  
 *          Raimond Snellings         *
 *           (snelling@nikhef.nl)     * 
 *          Ante Bilandzic            *
 *           (anteb@nikhef.nl)        * 
 * ***********************************/
 
#include "Riostream.h"
#include "TChain.h"
#include "TTree.h"
#include "TFile.h"
#include "TList.h"
#include "TH1.h"
#include "TProfile.h"
#include "TProfile2D.h"
#include "TProfile3D.h"

#include "AliAnalysisTask.h"
#include "AliAnalysisDataSlot.h"
#include "AliAnalysisDataContainer.h"
#include "AliAnalysisManager.h"

#include "AliESDEvent.h"
#include "AliESDInputHandler.h"

#include "AliAODEvent.h"
#include "AliAODInputHandler.h"

#include "AliMCEventHandler.h"
#include "AliMCEvent.h"

#include "../../CORRFW/AliCFManager.h"

#include "AliAnalysisTaskCumulants.h"
#include "AliFlowEventSimpleMaker.h"
#include "AliFlowAnalysisWithCumulants.h"
#include "AliFlowCumuConstants.h"
#include "AliFlowCommonConstants.h"
#include "AliFlowCommonHistResults.h"
#include "AliCumulantsFunctions.h"

ClassImp(AliAnalysisTaskCumulants)

//================================================================================================================

AliAnalysisTaskCumulants::AliAnalysisTaskCumulants(const char *name): 
 AliAnalysisTask(name,""), 
 fESD(NULL),
 fAOD(NULL),
 fCA(NULL),//Cumulant Analysis (CA) object
 fEventMaker(NULL),
 fAnalysisType("ESD"), 
 fCFManager1(NULL),
 fCFManager2(NULL),
 fListHistos(NULL)
{
 //constructor
 cout<<"AliAnalysisTaskCumulants::AliAnalysisTaskCumulants(const char *name)"<<endl;
 //input and output slots:
 //input slot #0 works with a TChain
 DefineInput(0, TChain::Class());
 //output slot #0 writes into a TList container
 DefineOutput(0, TList::Class());  
}

AliAnalysisTaskCumulants::AliAnalysisTaskCumulants(): 
 fESD(NULL),
 fAOD(NULL), 
 fCA(NULL),//Cumulant Analysis (CA) object
 fEventMaker(NULL),
 fAnalysisType("ESD"),
 fCFManager1(NULL),
 fCFManager2(NULL),
 fListHistos(NULL)
{
 //dummy constructor
 cout<<"AliAnalysisTaskCumulants::AliAnalysisTaskCumulants()"<<endl;
}

//================================================================================================================

void AliAnalysisTaskCumulants::ConnectInputData(Option_t *) 
{
 //connect ESD or AOD (called once)
 cout<<"AliAnalysisTaskCumulants::ConnectInputData(Option_t *)"<<endl;

 TTree* tree = dynamic_cast<TTree*> (GetInputData(0));
 if (!tree) 
 {
  Printf("ERROR: Could not read chain from input slot 0");
 } 
 else 
 {
 //disable all branches and enable only the needed ones
  if (fAnalysisType == "MC") {
     // we want to process only MC
      tree->SetBranchStatus("*", kFALSE);

      AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());

      if (!esdH) {
	Printf("ERROR: Could not get ESDInputHandler");
      } else {
	fESD = esdH->GetEvent();
      }
    }
    else if (fAnalysisType == "ESD" || fAnalysisType == "ESDMC0" || fAnalysisType == "ESDMC1"  ) {
      tree->SetBranchStatus("*", kFALSE);
      tree->SetBranchStatus("Tracks.*", kTRUE);

      AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());

      if (!esdH) {
	Printf("ERROR: Could not get ESDInputHandler");
      } else
	fESD = esdH->GetEvent();
    }
    else if (fAnalysisType == "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, ESDMC0, ESDMC1, AOD and MC types are allowed!");

    }
  }
}

//================================================================================================================

void AliAnalysisTaskCumulants::CreateOutputObjects() 
{
 //called at every worker node to initialize
 cout<<"AliAnalysisTaskCumulants::CreateOutputObjects()"<<endl;

 
 //OpenFile(0);
 

 if(!(fAnalysisType == "AOD" || fAnalysisType == "ESD" || fAnalysisType == "ESDMC0" || fAnalysisType == "ESDMC1" || fAnalysisType == "MC")) 
 {
  cout<<"WRONG ANALYSIS TYPE! only ESD, ESDMC0, ESDMC1, AOD and MC are allowed."<<endl;
  exit(1);
 }
 
 //event maker
 fEventMaker = new AliFlowEventSimpleMaker();
  
 //analyser
 fCA = new AliFlowAnalysisWithCumulants();
 fCA->CreateOutputObjects();

 if(fCA->GetHistList()) 
 {
  fListHistos = fCA->GetHistList();
  //fListHistos->Print();
 }
 else 
 {
  Printf("ERROR: Could not retrieve histogram list"); 
 }
 
 //PostData(0,fListHistos);
 
}

//================================================================================================================

void AliAnalysisTaskCumulants::Exec(Option_t *) 
{
 //main loop (called for each event)
 if (fAnalysisType == "MC") {
    // Process MC truth, therefore we receive the AliAnalysisManager and ask it for the AliMCEventHandler
    // This handler can return the current MC event

    AliMCEventHandler* eventHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
    if (!eventHandler) {
      Printf("ERROR: Could not retrieve MC event handler");
      return;
    }

    AliMCEvent* mcEvent = eventHandler->MCEvent();
    if (!mcEvent) {
      Printf("ERROR: Could not retrieve MC event");
      return;
    }

    Printf("MC particles: %d", mcEvent->GetNumberOfTracks());
    fCFManager1->SetEventInfo(mcEvent);
    fCFManager2->SetEventInfo(mcEvent);

    //cumulant analysis 
    AliFlowEventSimple* fEvent = fEventMaker->FillTracks(mcEvent,fCFManager1,fCFManager2);
    fCA->Make(fEvent);
    delete fEvent;
  }
  else if (fAnalysisType == "ESD") {
    if (!fESD) {
      Printf("ERROR: fESD not available");
      return;
    }
    Printf("There are %d tracks in this event", fESD->GetNumberOfTracks());
    
    //cumulant analysis 
    AliFlowEventSimple* fEvent = fEventMaker->FillTracks(fESD);
    fCA->Make(fEvent);
    delete fEvent;
  }
  else if (fAnalysisType == "ESDMC0") {
    if (!fESD) {
      Printf("ERROR: fESD not available");
      return;
    }
    Printf("There are %d tracks in this event", fESD->GetNumberOfTracks());
    
    AliMCEventHandler* eventHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
    if (!eventHandler) {
      Printf("ERROR: Could not retrieve MC event handler");
      return;
    }

    AliMCEvent* mcEvent = eventHandler->MCEvent();
    if (!mcEvent) {
      Printf("ERROR: Could not retrieve MC event");
      return;
    }

    fCFManager1->SetEventInfo(mcEvent);
    fCFManager2->SetEventInfo(mcEvent);

    //cumulant analysis 
    AliFlowEventSimple* fEvent=NULL;
    if (fAnalysisType == "ESDMC0") { 
      fEvent = fEventMaker->FillTracks(fESD, mcEvent, fCFManager1, fCFManager2, 0); //0 = kine from ESD, 1 = kine from MC
    } else if (fAnalysisType == "ESDMC1") {
      fEvent = fEventMaker->FillTracks(fESD, mcEvent, fCFManager1, fCFManager2, 1); //0 = kine from ESD, 1 = kine from MC
    }
    fCA->Make(fEvent);
    delete fEvent;
    //delete mcEvent;
  }
  
  else if (fAnalysisType == "AOD") {
    if (!fAOD) {
      Printf("ERROR: fAOD not available");
      return;
    }
    Printf("There are %d tracks in this event", fAOD->GetNumberOfTracks());

    // analysis 
    //For the moment don't use CF //AliFlowEventSimple* fEvent = fEventMaker->FillTracks(fAOD,fCFManager1,fCFManager2);
    AliFlowEventSimple* fEvent = fEventMaker->FillTracks(fAOD);
    fCA->Make(fEvent);
    delete fEvent;
  }


  PostData(0,fListHistos); 
}

//================================================================================================================

void AliAnalysisTaskCumulants::Terminate(Option_t *) 
{  
 //accessing the output list which contains the merged 2D and 3D profiles from all worker nodes
 fListHistos = (TList*)GetOutputData(0);
 //fListHistos->Print();
 
 if(fListHistos)
 {	    
  //profiles with avarage values of generating functions for int. and diff. flow
  TProfile2D *intFlowGenFun = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fIntFlowGenFun")); 
  //TProfile3D *diffFlowGenFunRe=dynamic_cast<TProfile3D*>(fListHistos->FindObject("fDiffFlowGenFunRe"));
  //TProfile3D *diffFlowGenFunIm=dynamic_cast<TProfile3D*>(fListHistos->FindObject("fDiffFlowGenFunIm"));
      
  TProfile2D *diffFlowGenFunRe0 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunRe0"));
  TProfile2D *diffFlowGenFunRe1 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunRe1")); 
  TProfile2D *diffFlowGenFunRe2 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunRe2")); 
  TProfile2D *diffFlowGenFunRe3 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunRe3")); 
  TProfile2D *diffFlowGenFunRe4 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunRe4")); 
  TProfile2D *diffFlowGenFunRe5 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunRe5")); 
  TProfile2D *diffFlowGenFunRe6 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunRe6")); 
  TProfile2D *diffFlowGenFunRe7 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunRe7")); 
  TProfile2D *diffFlowGenFunIm0 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunIm0")); 
  TProfile2D *diffFlowGenFunIm1 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunIm1")); 
  TProfile2D *diffFlowGenFunIm2 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunIm2")); 
  TProfile2D *diffFlowGenFunIm3 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunIm3")); 
  TProfile2D *diffFlowGenFunIm4 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunIm4")); 
  TProfile2D *diffFlowGenFunIm5 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunIm5")); 
  TProfile2D *diffFlowGenFunIm6 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunIm6")); 
  TProfile2D *diffFlowGenFunIm7 = dynamic_cast<TProfile2D*>(fListHistos->FindObject("fDiffFlowGenFunIm7")); 
    
  //histograms to store the final results
  TH1D *intFlowResults   = dynamic_cast<TH1D*>(fListHistos->FindObject("fIntFlowResults"));
  TH1D *diffFlowResults2 = dynamic_cast<TH1D*>(fListHistos->FindObject("fDiffFlowResults2"));
  TH1D *diffFlowResults4 = dynamic_cast<TH1D*>(fListHistos->FindObject("fDiffFlowResults4"));
  TH1D *diffFlowResults6 = dynamic_cast<TH1D*>(fListHistos->FindObject("fDiffFlowResults6"));
  TH1D *diffFlowResults8 = dynamic_cast<TH1D*>(fListHistos->FindObject("fDiffFlowResults8"));

  //profile with avarage selected multiplicity for int. flow 
  TProfile *avMult = dynamic_cast<TProfile*>(fListHistos->FindObject("fAvMultIntFlow"));
  
  //profile with avarage values of Q-vector components (1st bin: <Q_x>, 2nd bin: <Q_y>, 3rd bin: <(Q_x)^2>, 4th bin: <(Q_y)^2>) 
  TProfile *QVectorComponents = dynamic_cast<TProfile*>(fListHistos->FindObject("fQVectorComponents"));
  
  //q-distribution
  TH1D *qDist = dynamic_cast<TH1D*>(fListHistos->FindObject("fQDist"));
  
  AliCumulantsFunctions finalResults(intFlowGenFun,NULL,NULL, intFlowResults,diffFlowResults2,diffFlowResults4,diffFlowResults6,diffFlowResults8,avMult,QVectorComponents,qDist,diffFlowGenFunRe0,diffFlowGenFunRe1,diffFlowGenFunRe2,
diffFlowGenFunRe3,diffFlowGenFunRe4,diffFlowGenFunRe5,diffFlowGenFunRe6,diffFlowGenFunRe7,diffFlowGenFunIm0,diffFlowGenFunIm1,
diffFlowGenFunIm2,diffFlowGenFunIm3,diffFlowGenFunIm4,diffFlowGenFunIm5,diffFlowGenFunIm6,diffFlowGenFunIm7);
         
  finalResults.Calculate();  
 
 }
 else
 {
  cout<<"histogram list pointer is empty"<<endl;
 }
}
Commit	Line	Data
924fafb7	1	/*************************************************************************
	2	* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
2188af53	16	/**************************************
	17	* analysis task for cumulant method *
	18	* *
	19	* authors: Naomi van der Kolk *
	20	* (kolk@nikhef.nl) *
	21	* Raimond Snellings *
	22	* (snelling@nikhef.nl) *
	23	* Ante Bilandzic *
	24	* (anteb@nikhef.nl) *
	25	* ***********************************/
	26
	27	#include "Riostream.h"
924fafb7	28	#include "TChain.h"
	29	#include "TTree.h"
	30	#include "TFile.h"
aaebd73d	31	#include "TList.h"
aaebd73d	32	#include "TH1.h"
aaebd73d	33	#include "TProfile.h"
	34	#include "TProfile2D.h"
	35	#include "TProfile3D.h"
924fafb7	36
aaebd73d	37	#include "AliAnalysisTask.h"
	38	#include "AliAnalysisDataSlot.h"
	39	#include "AliAnalysisDataContainer.h"
924fafb7	40	#include "AliAnalysisManager.h"
	41
	42	#include "AliESDEvent.h"
	43	#include "AliESDInputHandler.h"
	44
	45	#include "AliAODEvent.h"
	46	#include "AliAODInputHandler.h"
	47
	48	#include "AliMCEventHandler.h"
	49	#include "AliMCEvent.h"
	50
aaebd73d	51	#include "../../CORRFW/AliCFManager.h"
aaebd73d	52
924fafb7	53	#include "AliAnalysisTaskCumulants.h"
	54	#include "AliFlowEventSimpleMaker.h"
	55	#include "AliFlowAnalysisWithCumulants.h"
aaebd73d	56	#include "AliFlowCumuConstants.h"
	57	#include "AliFlowCommonConstants.h"
	58	#include "AliFlowCommonHistResults.h"
aaebd73d	59	#include "AliCumulantsFunctions.h"
924fafb7	60
924fafb7	61	ClassImp(AliAnalysisTaskCumulants)
924fafb7	62
2188af53	63	//================================================================================================================
	64
	65	AliAnalysisTaskCumulants::AliAnalysisTaskCumulants(const char *name):
	66	AliAnalysisTask(name,""),
	67	fESD(NULL),
	68	fAOD(NULL),
	69	fCA(NULL),//Cumulant Analysis (CA) object
	70	fEventMaker(NULL),
	71	fAnalysisType("ESD"),
	72	fCFManager1(NULL),
	73	fCFManager2(NULL),
	74	fListHistos(NULL)
924fafb7	75	{
2188af53	76	//constructor
	77	cout<<"AliAnalysisTaskCumulants::AliAnalysisTaskCumulants(const char *name)"<<endl;
	78	//input and output slots:
	79	//input slot #0 works with a TChain
	80	DefineInput(0, TChain::Class());
	81	//output slot #0 writes into a TList container
	82	DefineOutput(0, TList::Class());
924fafb7	83	}
2188af53	84
	85	AliAnalysisTaskCumulants::AliAnalysisTaskCumulants():
	86	fESD(NULL),
	87	fAOD(NULL),
	88	fCA(NULL),//Cumulant Analysis (CA) object
	89	fEventMaker(NULL),
	90	fAnalysisType("ESD"),
	91	fCFManager1(NULL),
	92	fCFManager2(NULL),
	93	fListHistos(NULL)
aaebd73d	94	{
2188af53	95	//dummy constructor
	96	cout<<"AliAnalysisTaskCumulants::AliAnalysisTaskCumulants()"<<endl;
	97	}
	98
	99	//================================================================================================================
924fafb7	100
924fafb7	101	void AliAnalysisTaskCumulants::ConnectInputData(Option_t *)
924fafb7	102	{
2188af53	103	//connect ESD or AOD (called once)
	104	cout<<"AliAnalysisTaskCumulants::ConnectInputData(Option_t *)"<<endl;
	105
	106	TTree* tree = dynamic_cast<TTree*> (GetInputData(0));
	107	if (!tree)
	108	{
	109	Printf("ERROR: Could not read chain from input slot 0");
	110	}
	111	else
	112	{
	113	//disable all branches and enable only the needed ones
	114	if (fAnalysisType == "MC") {
	115	// we want to process only MC
924fafb7	116	tree->SetBranchStatus("*", kFALSE);
	117
	118	AliESDInputHandler esdH = dynamic_cast<AliESDInputHandler> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	119
	120	if (!esdH) {
	121	Printf("ERROR: Could not get ESDInputHandler");
	122	} else {
	123	fESD = esdH->GetEvent();
	124	}
	125	}
	126	else if (fAnalysisType == "ESD" \|\| fAnalysisType == "ESDMC0" \|\| fAnalysisType == "ESDMC1" ) {
	127	tree->SetBranchStatus("*", kFALSE);
	128	tree->SetBranchStatus("Tracks.*", kTRUE);
	129
	130	AliESDInputHandler esdH = dynamic_cast<AliESDInputHandler> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	131
	132	if (!esdH) {
	133	Printf("ERROR: Could not get ESDInputHandler");
	134	} else
	135	fESD = esdH->GetEvent();
	136	}
	137	else if (fAnalysisType == "AOD") {
	138	AliAODInputHandler aodH = dynamic_cast<AliAODInputHandler> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	139
	140	if (!aodH) {
	141	Printf("ERROR: Could not get AODInputHandler");
	142	}
	143	else {
	144	fAOD = aodH->GetEvent();
	145	}
	146	}
	147	else {
	148	Printf("Wrong analysis type: Only ESD, ESDMC0, ESDMC1, AOD and MC types are allowed!");
	149
	150	}
	151	}
	152	}
	153
2188af53	154	//================================================================================================================
2188af53	155
924fafb7	156	void AliAnalysisTaskCumulants::CreateOutputObjects()
924fafb7	157	{
2188af53	158	//called at every worker node to initialize
2188af53	159	cout<<"AliAnalysisTaskCumulants::CreateOutputObjects()"<<endl;
924fafb7	160
aaebd73d	161
2188af53	162	//OpenFile(0);
	163
	164
	165	if(!(fAnalysisType == "AOD" \|\| fAnalysisType == "ESD" \|\| fAnalysisType == "ESDMC0" \|\| fAnalysisType == "ESDMC1" \|\| fAnalysisType == "MC"))
	166	{
	167	cout<<"WRONG ANALYSIS TYPE! only ESD, ESDMC0, ESDMC1, AOD and MC are allowed."<<endl;
	168	exit(1);
	169	}
	170
	171	//event maker
	172	fEventMaker = new AliFlowEventSimpleMaker();
	173
	174	//analyser
	175	fCA = new AliFlowAnalysisWithCumulants();
	176	fCA->CreateOutputObjects();
	177
	178	if(fCA->GetHistList())
	179	{
	180	fListHistos = fCA->GetHistList();
	181	//fListHistos->Print();
aaebd73d	182	}
2188af53	183	else
	184	{
	185	Printf("ERROR: Could not retrieve histogram list");
	186	}
	187
	188	//PostData(0,fListHistos);
	189
	190	}
	191
	192	//================================================================================================================
924fafb7	193
924fafb7	194	void AliAnalysisTaskCumulants::Exec(Option_t *)
924fafb7	195	{
2188af53	196	//main loop (called for each event)
2188af53	197	if (fAnalysisType == "MC") {
924fafb7	198	// Process MC truth, therefore we receive the AliAnalysisManager and ask it for the AliMCEventHandler
	199	// This handler can return the current MC event
	200
	201	AliMCEventHandler* eventHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
	202	if (!eventHandler) {
	203	Printf("ERROR: Could not retrieve MC event handler");
	204	return;
	205	}
	206
	207	AliMCEvent* mcEvent = eventHandler->MCEvent();
	208	if (!mcEvent) {
	209	Printf("ERROR: Could not retrieve MC event");
	210	return;
	211	}
	212
	213	Printf("MC particles: %d", mcEvent->GetNumberOfTracks());
aaebd73d	214	fCFManager1->SetEventInfo(mcEvent);
aaebd73d	215	fCFManager2->SetEventInfo(mcEvent);
924fafb7	216
aaebd73d	217	//cumulant analysis
aaebd73d	218	AliFlowEventSimple* fEvent = fEventMaker->FillTracks(mcEvent,fCFManager1,fCFManager2);
2188af53	219	fCA->Make(fEvent);
924fafb7	220	delete fEvent;
	221	}
	222	else if (fAnalysisType == "ESD") {
	223	if (!fESD) {
	224	Printf("ERROR: fESD not available");
	225	return;
	226	}
	227	Printf("There are %d tracks in this event", fESD->GetNumberOfTracks());
	228
2188af53	229	//cumulant analysis
924fafb7	230	AliFlowEventSimple* fEvent = fEventMaker->FillTracks(fESD);
2188af53	231	fCA->Make(fEvent);
924fafb7	232	delete fEvent;
	233	}
	234	else if (fAnalysisType == "ESDMC0") {
	235	if (!fESD) {
	236	Printf("ERROR: fESD not available");
	237	return;
	238	}
	239	Printf("There are %d tracks in this event", fESD->GetNumberOfTracks());
	240
	241	AliMCEventHandler* eventHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
	242	if (!eventHandler) {
	243	Printf("ERROR: Could not retrieve MC event handler");
	244	return;
	245	}
	246
	247	AliMCEvent* mcEvent = eventHandler->MCEvent();
	248	if (!mcEvent) {
	249	Printf("ERROR: Could not retrieve MC event");
	250	return;
	251	}
	252
aaebd73d	253	fCFManager1->SetEventInfo(mcEvent);
aaebd73d	254	fCFManager2->SetEventInfo(mcEvent);
924fafb7	255
2188af53	256	//cumulant analysis
aaebd73d	257	AliFlowEventSimple* fEvent=NULL;
	258	if (fAnalysisType == "ESDMC0") {
	259	fEvent = fEventMaker->FillTracks(fESD, mcEvent, fCFManager1, fCFManager2, 0); //0 = kine from ESD, 1 = kine from MC
	260	} else if (fAnalysisType == "ESDMC1") {
	261	fEvent = fEventMaker->FillTracks(fESD, mcEvent, fCFManager1, fCFManager2, 1); //0 = kine from ESD, 1 = kine from MC
	262	}
2188af53	263	fCA->Make(fEvent);
924fafb7	264	delete fEvent;
	265	//delete mcEvent;
	266	}
aaebd73d	267
924fafb7	268	else if (fAnalysisType == "AOD") {
	269	if (!fAOD) {
	270	Printf("ERROR: fAOD not available");
	271	return;
	272	}
	273	Printf("There are %d tracks in this event", fAOD->GetNumberOfTracks());
	274
aaebd73d	275	// analysis
aaebd73d	276	//For the moment don't use CF //AliFlowEventSimple* fEvent = fEventMaker->FillTracks(fAOD,fCFManager1,fCFManager2);
924fafb7	277	AliFlowEventSimple* fEvent = fEventMaker->FillTracks(fAOD);
2188af53	278	fCA->Make(fEvent);
924fafb7	279	delete fEvent;
924fafb7	280	}
aaebd73d	281
	282
	283	PostData(0,fListHistos);
924fafb7	284	}
924fafb7	285
2188af53	286	//================================================================================================================
2188af53	287
924fafb7	288	void AliAnalysisTaskCumulants::Terminate(Option_t *)
aaebd73d	289	{
2188af53	290	//accessing the output list which contains the merged 2D and 3D profiles from all worker nodes
	291	fListHistos = (TList*)GetOutputData(0);
	292	//fListHistos->Print();
	293
	294	if(fListHistos)
	295	{
	296	//profiles with avarage values of generating functions for int. and diff. flow
	297	TProfile2D intFlowGenFun = dynamic_cast<TProfile2D>(fListHistos->FindObject("fIntFlowGenFun"));
	298	//TProfile3D diffFlowGenFunRe=dynamic_cast<TProfile3D>(fListHistos->FindObject("fDiffFlowGenFunRe"));
	299	//TProfile3D diffFlowGenFunIm=dynamic_cast<TProfile3D>(fListHistos->FindObject("fDiffFlowGenFunIm"));
aaebd73d	300
2188af53	301	TProfile2D diffFlowGenFunRe0 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunRe0"));
	302	TProfile2D diffFlowGenFunRe1 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunRe1"));
	303	TProfile2D diffFlowGenFunRe2 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunRe2"));
	304	TProfile2D diffFlowGenFunRe3 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunRe3"));
	305	TProfile2D diffFlowGenFunRe4 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunRe4"));
	306	TProfile2D diffFlowGenFunRe5 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunRe5"));
	307	TProfile2D diffFlowGenFunRe6 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunRe6"));
	308	TProfile2D diffFlowGenFunRe7 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunRe7"));
	309	TProfile2D diffFlowGenFunIm0 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunIm0"));
	310	TProfile2D diffFlowGenFunIm1 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunIm1"));
	311	TProfile2D diffFlowGenFunIm2 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunIm2"));
	312	TProfile2D diffFlowGenFunIm3 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunIm3"));
	313	TProfile2D diffFlowGenFunIm4 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunIm4"));
	314	TProfile2D diffFlowGenFunIm5 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunIm5"));
	315	TProfile2D diffFlowGenFunIm6 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunIm6"));
	316	TProfile2D diffFlowGenFunIm7 = dynamic_cast<TProfile2D>(fListHistos->FindObject("fDiffFlowGenFunIm7"));
	317
	318	//histograms to store the final results
	319	TH1D intFlowResults = dynamic_cast<TH1D>(fListHistos->FindObject("fIntFlowResults"));
	320	TH1D diffFlowResults2 = dynamic_cast<TH1D>(fListHistos->FindObject("fDiffFlowResults2"));
	321	TH1D diffFlowResults4 = dynamic_cast<TH1D>(fListHistos->FindObject("fDiffFlowResults4"));
	322	TH1D diffFlowResults6 = dynamic_cast<TH1D>(fListHistos->FindObject("fDiffFlowResults6"));
	323	TH1D diffFlowResults8 = dynamic_cast<TH1D>(fListHistos->FindObject("fDiffFlowResults8"));
	324
	325	//profile with avarage selected multiplicity for int. flow
	326	TProfile avMult = dynamic_cast<TProfile>(fListHistos->FindObject("fAvMultIntFlow"));
	327
	328	//profile with avarage values of Q-vector components (1st bin: <Q_x>, 2nd bin: <Q_y>, 3rd bin: <(Q_x)^2>, 4th bin: <(Q_y)^2>)
	329	TProfile QVectorComponents = dynamic_cast<TProfile>(fListHistos->FindObject("fQVectorComponents"));
	330
	331	//q-distribution
	332	TH1D qDist = dynamic_cast<TH1D>(fListHistos->FindObject("fQDist"));
	333
	334	AliCumulantsFunctions finalResults(intFlowGenFun,NULL,NULL, intFlowResults,diffFlowResults2,diffFlowResults4,diffFlowResults6,diffFlowResults8,avMult,QVectorComponents,qDist,diffFlowGenFunRe0,diffFlowGenFunRe1,diffFlowGenFunRe2,
	335	diffFlowGenFunRe3,diffFlowGenFunRe4,diffFlowGenFunRe5,diffFlowGenFunRe6,diffFlowGenFunRe7,diffFlowGenFunIm0,diffFlowGenFunIm1,
	336	diffFlowGenFunIm2,diffFlowGenFunIm3,diffFlowGenFunIm4,diffFlowGenFunIm5,diffFlowGenFunIm6,diffFlowGenFunIm7);
	337
	338	finalResults.Calculate();
	339
	340	}
	341	else
	342	{
	343	cout<<"histogram list pointer is empty"<<endl;
	344	}
aaebd73d	345	}
	346
	347
	348
	349
	350
	351
	352
	353
	354
	355
	356
	357
	358
	359
	360
	361
924fafb7	362
924fafb7	363
924fafb7	364
924fafb7	365
c75fdbdc	366