[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / PiKaPr / TestAOD / AddTaskSpectraAOD.C

AliAnalysisTaskSpectraAOD* AddTaskSpectraAOD(Bool_t mc=kFALSE,
					     Double_t CentCutMin=0,
					     Double_t CentCutMax=100,
					     Double_t QvecCutMin=0,
					     Double_t QvecCutMax=100,
					     Double_t EtaMin=-0.8,
					     Double_t EtaMax=0.8,
					     Double_t Nsigmapid=3.,
					     Double_t pt=5.,
					     Double_t p=5.,
					     Double_t y=.5,
					     Double_t ptTofMatch=.6,
					     UInt_t trkbit=1,
					     UInt_t trkbitQVector=1,
					     Double_t DCA=100000,
					     UInt_t minNclsTPC=70,
					     Int_t nrebin=0,
					     TString opt=""){
  
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) 
    {
      ::Error("AddAliAnalysisTaskSpectraAOD", "No analysis manager to connect to.");
      return NULL;
    }   
  
  // Check the analysis type using the event handlers connected to the analysis manager.
  //==============================================================================
  if (!mgr->GetInputEventHandler()) 
    {
      ::Error("AddTaskITSsaTracks", "This task requires an input event handler");
      return NULL;
    }   
  
  TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
  if(type.Contains("ESD"))
    {
      ::Error("AddTaskITSsaTracks", "This task requires to run on AOD");
      return NULL;
    }
  
  using namespace AliSpectraNameSpace;
  
  AliSpectraAODPID *pid = new AliSpectraAODPID(); 
  pid->SetNSigmaCut(Nsigmapid);
  
  AliSpectraAODTrackCuts  * trcuts = new AliSpectraAODTrackCuts("Track Cuts");  
  trcuts->SetDCA(DCA);
  trcuts->SetTrackBits(trkbit);
  trcuts->SetPt(pt);
  trcuts->SetP(p);
  trcuts->SetY(y);
  trcuts->SetPtTOFMatching(ptTofMatch);   
  trcuts->SetEta(EtaMin,EtaMax);
  trcuts->SetMinTPCcls(minNclsTPC);
  trcuts->PrintCuts();
  
  AliSpectraAODEventCuts * evcuts = new AliSpectraAODEventCuts("Event Cuts");
  evcuts->SetQVectorCut(QvecCutMin,QvecCutMax);
  evcuts->SetCentralityCutMax(CentCutMax);  
  evcuts->SetCentralityCutMin(CentCutMin);
  evcuts->SetTrackBits(trkbitQVector);
  if(mc==1)evcuts->SetIsMC(kTRUE);
  evcuts->PrintCuts();
  
  AliAnalysisTaskSpectraAOD *task = new AliAnalysisTaskSpectraAOD(Form("TaskAODSpectraCent%.0fto%.0f_QVec%.1fto%.1f_Eta%.1fto%.1f_%.1fSigmaPID_TrBit%d%s",	
  								       CentCutMin,
  								       CentCutMax,
  								       QvecCutMin,
  								       QvecCutMax,
  								       EtaMin,
  								       EtaMax,
  								       Nsigmapid,
  								       trkbit,
								       opt.Data()));
  task->SetPID(pid);  
  task->SetEventCuts(evcuts);
  task->SetTrackCuts(trcuts);
  task->SetNRebin(nrebin);
  if(mc==1)task->SetIsMC(kTRUE);
  
  TString outputFileName = AliAnalysisManager::GetCommonFileName();
  
  TString typeofdata=mc?"MC":"Data";
  
  outputFileName += Form(":OutputAODSpectraTask_%s_Cent%.0fto%.0f_QVec%.1fto%.1f_Eta%.1fto%.1f_%.1fSigmaPID_TrBit%d%s",typeofdata.Data(),evcuts->GetCentralityMin(),evcuts->GetCentralityMax(),evcuts->GetQVectorCutMin(), evcuts->GetQVectorCutMax(),trcuts->GetEtaMin(),trcuts->GetEtaMax(),pid->GetNSigmaCut(),trcuts->GetTrackType(),opt.Data());
  
  cout<<"outputFileName:  "<<outputFileName<<endl;
  AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();      
  AliAnalysisDataContainer *coutputpt1 = mgr->CreateContainer("chistpt", AliSpectraAODHistoManager::Class(),  AliAnalysisManager::kOutputContainer,outputFileName);
  AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer("cvcutpt", AliSpectraAODEventCuts::Class(),    AliAnalysisManager::kOutputContainer,outputFileName);
  AliAnalysisDataContainer *coutputpt3 = mgr->CreateContainer("ctcutpt", AliSpectraAODTrackCuts::Class(),     AliAnalysisManager::kOutputContainer, outputFileName);
  AliAnalysisDataContainer *coutputpt4 = mgr->CreateContainer("cpidpt",  AliSpectraAODPID::Class(),     AliAnalysisManager::kOutputContainer,outputFileName);

  mgr->AddTask(task);
  
  mgr->ConnectInput(task, 0, cinput);
  mgr->ConnectOutput(task, 1, coutputpt1);
  mgr->ConnectOutput(task, 2, coutputpt2);
  mgr->ConnectOutput(task, 3, coutputpt3);
  mgr->ConnectOutput(task, 4, coutputpt4);

  return task;
}
Commit	Line	Data
c683985a	1	AliAnalysisTaskSpectraAOD* AddTaskSpectraAOD(Bool_t mc=kFALSE,
	2	Double_t CentCutMin=0,
	3	Double_t CentCutMax=100,
	4	Double_t QvecCutMin=0,
	5	Double_t QvecCutMax=100,
	6	Double_t EtaMin=-0.8,
	7	Double_t EtaMax=0.8,
	8	Double_t Nsigmapid=3.,
	9	Double_t pt=5.,
	10	Double_t p=5.,
	11	Double_t y=.5,
	12	Double_t ptTofMatch=.6,
	13	UInt_t trkbit=1,
	14	UInt_t trkbitQVector=1,
	15	Double_t DCA=100000,
	16	UInt_t minNclsTPC=70,
	17	Int_t nrebin=0,
	18	TString opt=""){
	19
	20	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	21	if (!mgr)
	22	{
	23	::Error("AddAliAnalysisTaskSpectraAOD", "No analysis manager to connect to.");
	24	return NULL;
	25	}
	26
	27	// Check the analysis type using the event handlers connected to the analysis manager.
	28	//==============================================================================
	29	if (!mgr->GetInputEventHandler())
	30	{
	31	::Error("AddTaskITSsaTracks", "This task requires an input event handler");
	32	return NULL;
	33	}
	34
	35	TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
	36	if(type.Contains("ESD"))
	37	{
	38	::Error("AddTaskITSsaTracks", "This task requires to run on AOD");
	39	return NULL;
	40	}
	41
	42	using namespace AliSpectraNameSpace;
	43
	44	AliSpectraAODPID *pid = new AliSpectraAODPID();
	45	pid->SetNSigmaCut(Nsigmapid);
	46
	47	AliSpectraAODTrackCuts * trcuts = new AliSpectraAODTrackCuts("Track Cuts");
	48	trcuts->SetDCA(DCA);
	49	trcuts->SetTrackBits(trkbit);
	50	trcuts->SetPt(pt);
	51	trcuts->SetP(p);
	52	trcuts->SetY(y);
	53	trcuts->SetPtTOFMatching(ptTofMatch);
	54	trcuts->SetEta(EtaMin,EtaMax);
	55	trcuts->SetMinTPCcls(minNclsTPC);
	56	trcuts->PrintCuts();
	57
	58	AliSpectraAODEventCuts * evcuts = new AliSpectraAODEventCuts("Event Cuts");
	59	evcuts->SetQVectorCut(QvecCutMin,QvecCutMax);
	60	evcuts->SetCentralityCutMax(CentCutMax);
	61	evcuts->SetCentralityCutMin(CentCutMin);
	62	evcuts->SetTrackBits(trkbitQVector);
	63	if(mc==1)evcuts->SetIsMC(kTRUE);
	64	evcuts->PrintCuts();
65
66	AliAnalysisTaskSpectraAOD *task = new AliAnalysisTaskSpectraAOD(Form("TaskAODSpectraCent%.0fto%.0f_QVec%.1fto%.1f_Eta%.1fto%.1f_%.1fSigmaPID_TrBit%d%s",
67	CentCutMin,
68	CentCutMax,
69	QvecCutMin,
70	QvecCutMax,
71	EtaMin,
72	EtaMax,
73	Nsigmapid,
74	trkbit,
75	opt.Data()));
76	task->SetPID(pid);
77	task->SetEventCuts(evcuts);
78	task->SetTrackCuts(trcuts);
79	task->SetNRebin(nrebin);
80	if(mc==1)task->SetIsMC(kTRUE);
81
82	TString outputFileName = AliAnalysisManager::GetCommonFileName();
83
84	TString typeofdata=mc?"MC":"Data";
85
86	outputFileName += Form(":OutputAODSpectraTask_%s_Cent%.0fto%.0f_QVec%.1fto%.1f_Eta%.1fto%.1f_%.1fSigmaPID_TrBit%d%s",typeofdata.Data(),evcuts->GetCentralityMin(),evcuts->GetCentralityMax(),evcuts->GetQVectorCutMin(), evcuts->GetQVectorCutMax(),trcuts->GetEtaMin(),trcuts->GetEtaMax(),pid->GetNSigmaCut(),trcuts->GetTrackType(),opt.Data());
87
88	cout<<"outputFileName: "<<outputFileName<<endl;
89	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
90	AliAnalysisDataContainer *coutputpt1 = mgr->CreateContainer("chistpt", AliSpectraAODHistoManager::Class(), AliAnalysisManager::kOutputContainer,outputFileName);
91	AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer("cvcutpt", AliSpectraAODEventCuts::Class(), AliAnalysisManager::kOutputContainer,outputFileName);
92	AliAnalysisDataContainer *coutputpt3 = mgr->CreateContainer("ctcutpt", AliSpectraAODTrackCuts::Class(), AliAnalysisManager::kOutputContainer, outputFileName);
93	AliAnalysisDataContainer *coutputpt4 = mgr->CreateContainer("cpidpt", AliSpectraAODPID::Class(), AliAnalysisManager::kOutputContainer,outputFileName);
94
95	mgr->AddTask(task);
96
97	mgr->ConnectInput(task, 0, cinput);
98	mgr->ConnectOutput(task, 1, coutputpt1);
99	mgr->ConnectOutput(task, 2, coutputpt2);
100	mgr->ConnectOutput(task, 3, coutputpt3);
101	mgr->ConnectOutput(task, 4, coutputpt4);
102
103	return task;
104	}