[u/mrichter/AliRoot.git] / PWGCF / Correlations / macros / AddTaskLV0Correlation.C

AliLeadingV0Correlation* AddTaskLV0Correlation(TString  fListName                = "LV0Correlation",
											   TString	fCollisiontype           = "PP",
											   Bool_t	fAnalysisMC              = 0,
											   Int_t    fCase                    = 2,
											   Bool_t   fRemovePileUP            = 0,
											   Bool_t   fRemoveAutoCorr          = 0,
											   Double_t	fPVzCut                  = 10,
											   UInt_t   fFilterBit               = 128,
											   Double_t	fRapidityCut             = 0.75,
											   Double_t	fEtaCut                  = 0.9,
											   Int_t	fmaxEventsinPool         = 2000,
											   Int_t	fmaxTracksinPool         = 1000,
											   Int_t    fMinEventsToMix          = 5,
											   Double_t	fV0radius				 = 0.3,
											   Double_t	fV0PostoPVz				 = 0.060,
											   Double_t	fV0NegtoPVz				 = 0.060,
											   Double_t	fDCAV0Daughters			 = 1.00,
											   Double_t	fCPAK0					 = 0.960,
											   Double_t	fCPALam					 = 0.997,
											   Double_t	fRejectLamK0			 = 0.005,
											   Double_t	fRejectK0Lam			 = 0.010,
											   Double_t fSigmaPID                = 3.0,
											   Double_t fCutCTK0				 = 20.0,
											   Double_t fCutCTLa				 = 30.0,
											   Double_t fMassCutK0               = 0.0105,             
											   Double_t fMassCutLa               = 0.0105) 
{
	// Get the current analysis manager.
    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
    if (!mgr) {Error("AddTaskLV0Correlation.C", "No Analysis Manager ");return 0;}
	
	//PVz Binning for pool PP or PbPb	
	Double_t pvzbinlimits[] = {-10,-9,-8,-7,-6,-5,-4,-3,-2,-1,0,1,2,3,4,5,6,7,8,9,10};
	Int_t pvzbinnumb = sizeof(pvzbinlimits)/sizeof(Double_t) - 1;
	
	//Mult Binning for pool	pp 
	Double_t cent_mult_binlimitsPP[] = {  0, 10, 20, 30, 40, 50, 60, 70, 80, 90,
		100,110,120,130,140,150,160,170,180,190,
		200,210,220,230,240,250,260,270,280,290,
		300,310,320,330,340,350,360,370,380,390,
		400,410,420,430,440,450,460,470,480,490,
		500,520,540,560,580,600,620,640,660,680,
		700,650,800,900,1000,1200,1400,1600,1800,
		2000};
	
	Int_t cent_mult_bin_numbPP = sizeof(cent_mult_binlimitsPP)/sizeof(Double_t) - 1;
	
	//Cent Binning for pool	PbPb 
	Double_t cent_mult_binlimitsPbPb[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,
		14,15,16,17,18,19,20,21,22,23,24,25,26,27,
		28,29,30,31,32,33,34,35,36,37,38,39,40,42,
		44,46,48,50,52,54,56,58,60,65,70,75,80,90};
	
	Int_t cent_mult_bin_numbPbPb = sizeof(cent_mult_binlimitsPbPb)/sizeof(Double_t) - 1;
	//Correlation task
	AliLeadingV0Correlation *myTask = new AliLeadingV0Correlation(fListName.Data());
	
	myTask->SetCollidingSystem(fCollisiontype);
	myTask->SetMCAnalysis(fAnalysisMC);
	myTask->SetCase(fCase);
	myTask->SetRemovePileUp(fRemovePileUP);
	myTask->SetRemoveAutoCorr(fRemoveAutoCorr);
	myTask->SetMaxNEventsInPool(fmaxEventsinPool);
	myTask->SetMinNTracksInPool(fmaxTracksinPool);
	myTask->SetMinEventsToMix(fMinEventsToMix);
	myTask->SetPoolPVzBinLimits(pvzbinnumb,pvzbinlimits);
	if(fCollisiontype=="PP")myTask->SetPoolCentBinLimits(cent_mult_bin_numbPP,cent_mult_binlimitsPP);
	if(fCollisiontype=="PbPb2010"||fCollisiontype=="PbPb2011")myTask->SetPoolCentBinLimits(cent_mult_bin_numbPbPb,cent_mult_binlimitsPbPb);
	myTask->SetPrimeryVertexCut(fPVzCut);
	myTask->SetEatCut(fEtaCut);
	myTask->SetFilterBit(fFilterBit);
	myTask->SetCutRap(fRapidityCut);
	myTask->SetV0Radius(fV0radius);
	myTask->SetV0PostoPVz(fV0PostoPVz);
	myTask->SetV0NegtoPVz(fV0NegtoPVz);
	myTask->SetDCAV0Daughters(fDCAV0Daughters);
	myTask->SetCPAK0(fCPAK0);
	myTask->SetCPALam(fCPALam);
	myTask->SetRejectLamK0(fRejectLamK0);
	myTask->SetRejectK0Lam(fRejectK0Lam);
	myTask->SetSigmaPID(fSigmaPID);
	myTask->SetCTK0(fCutCTK0);
	myTask->SetCTLa(fCutCTLa);
	myTask->SetMassCutK0(fMassCutK0);
	myTask->SetMassCutLa(fMassCutLa);
	
	mgr->AddTask(myTask);
	
	// Create containers for input/output
	TString outputFileName = AliAnalysisManager::GetCommonFileName();
	AliAnalysisDataContainer *cinput  = mgr->GetCommonInputContainer();
	AliAnalysisDataContainer *coutput = mgr->CreateContainer(fListName.Data(), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName);
	
	//____________________________________________//
	mgr->ConnectInput(myTask,0,cinput);	
	mgr->ConnectOutput(myTask,1,coutput);
	
return myTask;
}
Commit	Line	Data
da6ac661	1	AliLeadingV0Correlation* AddTaskLV0Correlation(TString fListName = "LV0Correlation",
	2	TString fCollisiontype = "PP",
	3	Bool_t fAnalysisMC = 0,
	4	Int_t fCase = 2,
	5	Bool_t fRemovePileUP = 0,
	6	Bool_t fRemoveAutoCorr = 0,
	7	Double_t fPVzCut = 10,
	8	UInt_t fFilterBit = 128,
	9	Double_t fRapidityCut = 0.75,
	10	Double_t fEtaCut = 0.9,
	11	Int_t fmaxEventsinPool = 2000,
	12	Int_t fmaxTracksinPool = 1000,
	13	Int_t fMinEventsToMix = 5,
	14	Double_t fV0radius = 0.3,
	15	Double_t fV0PostoPVz = 0.060,
	16	Double_t fV0NegtoPVz = 0.060,
	17	Double_t fDCAV0Daughters = 1.00,
	18	Double_t fCPAK0 = 0.960,
	19	Double_t fCPALam = 0.997,
	20	Double_t fRejectLamK0 = 0.005,
	21	Double_t fRejectK0Lam = 0.010,
	22	Double_t fSigmaPID = 3.0,
	23	Double_t fCutCTK0 = 20.0,
	24	Double_t fCutCTLa = 30.0,
	25	Double_t fMassCutK0 = 0.0105,
	26	Double_t fMassCutLa = 0.0105)
ff0805a7	27	{
	28	// Get the current analysis manager.
	29	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	30	if (!mgr) {Error("AddTaskLV0Correlation.C", "No Analysis Manager ");return 0;}
	31
da6ac661	32	//PVz Binning for pool PP or PbPb
ff0805a7	33	Double_t pvzbinlimits[] = {-10,-9,-8,-7,-6,-5,-4,-3,-2,-1,0,1,2,3,4,5,6,7,8,9,10};
	34	Int_t pvzbinnumb = sizeof(pvzbinlimits)/sizeof(Double_t) - 1;
	35
da6ac661	36	//Mult Binning for pool pp
	37	Double_t cent_mult_binlimitsPP[] = { 0, 10, 20, 30, 40, 50, 60, 70, 80, 90,
	38	100,110,120,130,140,150,160,170,180,190,
	39	200,210,220,230,240,250,260,270,280,290,
	40	300,310,320,330,340,350,360,370,380,390,
	41	400,410,420,430,440,450,460,470,480,490,
	42	500,520,540,560,580,600,620,640,660,680,
	43	700,650,800,900,1000,1200,1400,1600,1800,
	44	2000};
	45
	46	Int_t cent_mult_bin_numbPP = sizeof(cent_mult_binlimitsPP)/sizeof(Double_t) - 1;
	47
	48	//Cent Binning for pool PbPb
	49	Double_t cent_mult_binlimitsPbPb[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,
	50	14,15,16,17,18,19,20,21,22,23,24,25,26,27,
	51	28,29,30,31,32,33,34,35,36,37,38,39,40,42,
	52	44,46,48,50,52,54,56,58,60,65,70,75,80,90};
	53
	54	Int_t cent_mult_bin_numbPbPb = sizeof(cent_mult_binlimitsPbPb)/sizeof(Double_t) - 1;
ff0805a7	55	//Correlation task
	56	AliLeadingV0Correlation *myTask = new AliLeadingV0Correlation(fListName.Data());
	57
	58	myTask->SetCollidingSystem(fCollisiontype);
	59	myTask->SetMCAnalysis(fAnalysisMC);
da6ac661	60	myTask->SetCase(fCase);
	61	myTask->SetRemovePileUp(fRemovePileUP);
	62	myTask->SetRemoveAutoCorr(fRemoveAutoCorr);
9bcef488	63	myTask->SetMaxNEventsInPool(fmaxEventsinPool);
9bcef488	64	myTask->SetMinNTracksInPool(fmaxTracksinPool);
da6ac661	65	myTask->SetMinEventsToMix(fMinEventsToMix);
ff0805a7	66	myTask->SetPoolPVzBinLimits(pvzbinnumb,pvzbinlimits);
da6ac661	67	if(fCollisiontype=="PP")myTask->SetPoolCentBinLimits(cent_mult_bin_numbPP,cent_mult_binlimitsPP);
	68	if(fCollisiontype=="PbPb2010"\|\|fCollisiontype=="PbPb2011")myTask->SetPoolCentBinLimits(cent_mult_bin_numbPbPb,cent_mult_binlimitsPbPb);
	69	myTask->SetPrimeryVertexCut(fPVzCut);
	70	myTask->SetEatCut(fEtaCut);
	71	myTask->SetFilterBit(fFilterBit);
	72	myTask->SetCutRap(fRapidityCut);
	73	myTask->SetV0Radius(fV0radius);
	74	myTask->SetV0PostoPVz(fV0PostoPVz);
	75	myTask->SetV0NegtoPVz(fV0NegtoPVz);
	76	myTask->SetDCAV0Daughters(fDCAV0Daughters);
	77	myTask->SetCPAK0(fCPAK0);
	78	myTask->SetCPALam(fCPALam);
	79	myTask->SetRejectLamK0(fRejectLamK0);
	80	myTask->SetRejectK0Lam(fRejectK0Lam);
	81	myTask->SetSigmaPID(fSigmaPID);
	82	myTask->SetCTK0(fCutCTK0);
	83	myTask->SetCTLa(fCutCTLa);
	84	myTask->SetMassCutK0(fMassCutK0);
	85	myTask->SetMassCutLa(fMassCutLa);
ff0805a7	86
	87	mgr->AddTask(myTask);
	88
	89	// Create containers for input/output
	90	TString outputFileName = AliAnalysisManager::GetCommonFileName();
9bcef488	91	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
ff0805a7	92	AliAnalysisDataContainer *coutput = mgr->CreateContainer(fListName.Data(), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName);
	93
	94	//____________________________________________//
	95	mgr->ConnectInput(myTask,0,cinput);
	96	mgr->ConnectOutput(myTask,1,coutput);
	97
	98	return myTask;
	99	}
	100
	101