PWGLF/SPECTRA/PiKaPr/TestAOD/AddTaskSpectraAOD.C

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