ROOT logo
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;
}
 AddTaskSpectraAOD.C:1
 AddTaskSpectraAOD.C:2
 AddTaskSpectraAOD.C:3
 AddTaskSpectraAOD.C:4
 AddTaskSpectraAOD.C:5
 AddTaskSpectraAOD.C:6
 AddTaskSpectraAOD.C:7
 AddTaskSpectraAOD.C:8
 AddTaskSpectraAOD.C:9
 AddTaskSpectraAOD.C:10
 AddTaskSpectraAOD.C:11
 AddTaskSpectraAOD.C:12
 AddTaskSpectraAOD.C:13
 AddTaskSpectraAOD.C:14
 AddTaskSpectraAOD.C:15
 AddTaskSpectraAOD.C:16
 AddTaskSpectraAOD.C:17
 AddTaskSpectraAOD.C:18
 AddTaskSpectraAOD.C:19
 AddTaskSpectraAOD.C:20
 AddTaskSpectraAOD.C:21
 AddTaskSpectraAOD.C:22
 AddTaskSpectraAOD.C:23
 AddTaskSpectraAOD.C:24
 AddTaskSpectraAOD.C:25
 AddTaskSpectraAOD.C:26
 AddTaskSpectraAOD.C:27
 AddTaskSpectraAOD.C:28
 AddTaskSpectraAOD.C:29
 AddTaskSpectraAOD.C:30
 AddTaskSpectraAOD.C:31
 AddTaskSpectraAOD.C:32
 AddTaskSpectraAOD.C:33
 AddTaskSpectraAOD.C:34
 AddTaskSpectraAOD.C:35
 AddTaskSpectraAOD.C:36
 AddTaskSpectraAOD.C:37
 AddTaskSpectraAOD.C:38
 AddTaskSpectraAOD.C:39
 AddTaskSpectraAOD.C:40
 AddTaskSpectraAOD.C:41
 AddTaskSpectraAOD.C:42
 AddTaskSpectraAOD.C:43
 AddTaskSpectraAOD.C:44
 AddTaskSpectraAOD.C:45
 AddTaskSpectraAOD.C:46
 AddTaskSpectraAOD.C:47
 AddTaskSpectraAOD.C:48
 AddTaskSpectraAOD.C:49
 AddTaskSpectraAOD.C:50
 AddTaskSpectraAOD.C:51
 AddTaskSpectraAOD.C:52
 AddTaskSpectraAOD.C:53
 AddTaskSpectraAOD.C:54
 AddTaskSpectraAOD.C:55
 AddTaskSpectraAOD.C:56
 AddTaskSpectraAOD.C:57
 AddTaskSpectraAOD.C:58
 AddTaskSpectraAOD.C:59
 AddTaskSpectraAOD.C:60
 AddTaskSpectraAOD.C:61
 AddTaskSpectraAOD.C:62
 AddTaskSpectraAOD.C:63
 AddTaskSpectraAOD.C:64
 AddTaskSpectraAOD.C:65
 AddTaskSpectraAOD.C:66
 AddTaskSpectraAOD.C:67
 AddTaskSpectraAOD.C:68
 AddTaskSpectraAOD.C:69
 AddTaskSpectraAOD.C:70
 AddTaskSpectraAOD.C:71
 AddTaskSpectraAOD.C:72
 AddTaskSpectraAOD.C:73
 AddTaskSpectraAOD.C:74
 AddTaskSpectraAOD.C:75
 AddTaskSpectraAOD.C:76
 AddTaskSpectraAOD.C:77
 AddTaskSpectraAOD.C:78
 AddTaskSpectraAOD.C:79
 AddTaskSpectraAOD.C:80
 AddTaskSpectraAOD.C:81
 AddTaskSpectraAOD.C:82
 AddTaskSpectraAOD.C:83
 AddTaskSpectraAOD.C:84
 AddTaskSpectraAOD.C:85
 AddTaskSpectraAOD.C:86
 AddTaskSpectraAOD.C:87
 AddTaskSpectraAOD.C:88
 AddTaskSpectraAOD.C:89
 AddTaskSpectraAOD.C:90
 AddTaskSpectraAOD.C:91
 AddTaskSpectraAOD.C:92
 AddTaskSpectraAOD.C:93
 AddTaskSpectraAOD.C:94
 AddTaskSpectraAOD.C:95
 AddTaskSpectraAOD.C:96
 AddTaskSpectraAOD.C:97
 AddTaskSpectraAOD.C:98
 AddTaskSpectraAOD.C:99
 AddTaskSpectraAOD.C:100
 AddTaskSpectraAOD.C:101
 AddTaskSpectraAOD.C:102
 AddTaskSpectraAOD.C:103
 AddTaskSpectraAOD.C:104
 AddTaskSpectraAOD.C:105