//DEFINITION OF A FEW CONSTANTS
const Double_t ymin  = -2.0 ;
const Double_t ymax  =  2.0 ;
const Double_t ptmin =  0.0 ;
const Double_t ptmax =  10.0 ;
const Int_t    mintrackrefsTPC = 2 ;
const Int_t    mintrackrefsITS = 3 ;
const Int_t    charge  = 1 ;
const Int_t    PDG = 421; 
const Int_t    minclustersTPC = 50 ;
//----------------------------------------------------

Bool_t AliCFHeavyFlavourTask()
{
	
        TString analysisMode = "local"; // "local" or "grid"
	TString inputMode    = "list"; // "list" or "xml"


	Bool_t useParFiles=kFALSE;
	
   
	//load the required libraries
	gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/LoadLibraries.C");
	gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/MakeAODInputChain.C");
	LoadLibraries(useParFiles);
	
        if(analysisMode=="grid") TGrid::Connect("alien:",0,0,"t");
	
	
	TBenchmark benchmark;
	benchmark.Start("AliCFHeavyFlavourTask");
	
	AliLog::SetGlobalDebugLevel(0);


	TChain *chainAOD = 0;
	
	if(inputMode=="list") {
	  // Local files
	  //chainAOD = MakeAODInputChain();// with this it reads ./AliAOD.root and ./AliAOD.VertexingHF.root
	  chainAOD = MakeAODInputChain("alien:///alice/cern.ch/user/r/rbala/analysis/out_lhcw/290001/",2,2);
	} else if(inputMode=="xml") {
	  // xml
	  chainAOD = MakeAODInputChain("collection_aod.xml","collection_aodHF.xml");
	}

	Info("AliCFHeavyFlavourTask",Form("CHAIN HAS %d ENTRIES",(Int_t)chainAOD->GetEntries()));

	//CONTAINER DEFINITION
	Info("AliCFHeavyFlavourTask","SETUP CONTAINER");
	//the sensitive variables (2 in this example), their indices
	UInt_t ipt = 0;
	UInt_t iy  = 1;
	//Setting up the container grid... 
	UInt_t nstep = 2 ; //number of selection steps MC 
	const Int_t nvar   = 2 ; //number of variables on the grid:pt,y
	const Int_t nbin1  = 8 ; //bins in pt
	const Int_t nbin2  = 8 ; //bins in y 
	
	//arrays for the number of bins in each dimension
	Int_t iBin[nvar];
	iBin[0]=nbin1;
	iBin[1]=nbin2;
	
	//arrays for lower bounds :
	Double_t *binLim1=new Double_t[nbin1+1];
	Double_t *binLim2=new Double_t[nbin2+1];
	
	//values for bin lower bounds
	for(Int_t i=0; i<=nbin1; i++) binLim1[i]=(Double_t)ptmin + (ptmax-ptmin)/nbin1*(Double_t)i ; 
	for(Int_t i=0; i<=nbin2; i++) binLim2[i]=(Double_t)ymin  + (ymax-ymin)  /nbin2*(Double_t)i ;

	//one "container" for MC
	AliCFContainer* container = new AliCFContainer("container","container for tracks",nstep,nvar,iBin);

	//setting the bin limits
	container -> SetBinLimits(ipt,binLim1);
	container -> SetBinLimits(iy,binLim2);
	
	//CREATE THE  CUTS -----------------------------------------------
	
	// Gen-Level kinematic cuts
	AliCFTrackKineCuts *mcKineCuts = new AliCFTrackKineCuts("mcKineCuts","MC-level kinematic cuts");
	//   mcKineCuts->SetPtRange(ptmin,ptmax);
	//   mcKineCuts->SetRapidityRange(ymin,ymax);
	//   mcKineCuts->SetChargeMC(charge);
	
	//Particle-Level cuts:  
	AliCFParticleGenCuts* mcGenCuts = new AliCFParticleGenCuts("mcGenCuts","MC particle generation cuts");
	//mcGenCuts->SetRequireIsPrimary();
	mcGenCuts->SetRequirePdgCode(PDG);
	mcGenCuts->SetAODMC(1); //special flag for reading MC in AOD tree (important)
	
	// Rec-Level kinematic cuts
	AliCFTrackKineCuts *recKineCuts = new AliCFTrackKineCuts("recKineCuts","rec-level kine cuts");
	//   recKineCuts->SetPtRange(ptmin,ptmax);
	//   recKineCuts->SetRapidityRange(ymin,ymax);
	//   recKineCuts->SetChargeRec(charge);
	
	AliCFTrackQualityCuts *recQualityCuts = new AliCFTrackQualityCuts("recQualityCuts","rec-level quality cuts");
	//recQualityCuts->SetStatus(AliESDtrack::kITSrefit);
	
	AliCFTrackIsPrimaryCuts *recIsPrimaryCuts = new AliCFTrackIsPrimaryCuts("recIsPrimaryCuts","rec-level isPrimary cuts");
	//recIsPrimaryCuts->SetAODType(AliAODTrack::kPrimary);
	
	printf("CREATE MC KINE CUTS\n");
	TObjArray* mcList = new TObjArray(0) ;
	mcList->AddLast(mcKineCuts);
	mcList->AddLast(mcGenCuts);
	
	printf("CREATE RECONSTRUCTION CUTS\n");
	TObjArray* recList = new TObjArray(0) ;
	recList->AddLast(recKineCuts);
	recList->AddLast(recQualityCuts);
	recList->AddLast(recIsPrimaryCuts);
	
	//CREATE THE INTERFACE TO CORRECTION FRAMEWORK USED IN THE TASK
	printf("CREATE INTERFACE AND CUTS\n");
	AliCFManager* man = new AliCFManager() ;
	man->SetParticleContainer     (container);
	man->SetParticleCutsList(0 , mcList); // MC
	man->SetParticleCutsList(1 , recList); // AOD
	
	//CREATE THE TASK
	printf("CREATE TASK\n");
	// create the task
	AliCFHeavyFlavourTask *task = new AliCFHeavyFlavourTask("AliCFHeavyFlavourTask");
	task->SetCFManager(man); //here is set the CF manager
	
	//SETUP THE ANALYSIS MANAGER TO READ INPUT CHAIN AND WRITE DESIRED OUTPUTS
	printf("CREATE ANALYSIS MANAGER\n");
	// Make the analysis manager
	AliAnalysisManager *mgr = new AliAnalysisManager("TestManager");
	
	mgr->SetAnalysisType(AliAnalysisManager::kLocalAnalysis);
	
	AliAODInputHandler* dataHandler = new AliAODInputHandler();
	mgr->SetInputEventHandler(dataHandler);
	
	// Create and connect containers for input/output
	
	// ------ input data ------
	AliAnalysisDataContainer *cinput0  = mgr->GetCommonInputContainer();
	
	// ----- output data -----
	
	//slot 0 : default output tree (by default handled by AliAnalysisTaskSE)
	AliAnalysisDataContainer *coutput0 = mgr->CreateContainer("ctree0", TTree::Class(),AliAnalysisManager::kOutputContainer,"outputEta.root");
	
	//now comes user's output objects :
	
	// output TH1I for event counting
	AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("chist0", TH1I::Class(),AliAnalysisManager::kOutputContainer,"output.root");
	// output Correction Framework Container (for acceptance & efficiency calculations)
	AliAnalysisDataContainer *coutput2 = mgr->CreateContainer("ccontainer0", AliCFContainer::Class(),AliAnalysisManager::kOutputContainer,"output.root");
	
//	cinput0->SetData(chainAOD);
	
	mgr->AddTask(task);
	mgr->ConnectInput(task,0,mgr->GetCommonInputContainer());
	mgr->ConnectOutput(task,0,coutput0);
	mgr->ConnectOutput(task,1,coutput1);
	mgr->ConnectOutput(task,2,coutput2);
	
	printf("READY TO RUN\n");
	//RUN !!!
	if (mgr->InitAnalysis()) {
		mgr->PrintStatus();
		mgr->StartAnalysis("local",chainAOD);
	}
	
	benchmark.Stop("AliCFHeavyFlavourTask");
	benchmark.Show("AliCFHeavyFlavourTask");
	
	return kTRUE ;
}