PWG2/EBYE/macros/AddTaskBalanceCentralityTrain.C

   1 // now in options\r
   2 //=============================================//\r
   3 //const char* centralityEstimator = "V0M";\r
   4 //const char* centralityEstimator = "CL1";\r
   5 //const char* centralityEstimator = "TRK";\r
   6 //=============================================//\r
   7 //Bool_t gRunShuffling = kFALSE;\r
   8 //Bool_t gRunShuffling = kTRUE;\r
   9 //=============================================//\r
  10 //_________________________________________________________//\r
  11 AliAnalysisTaskBF *AddTaskBalanceCentralityTrain(Double_t centrMin=0.,\r
  12                                                  Double_t centrMax=100.,\r
  13                                                  Bool_t gRunShuffling=kFALSE,\r
  14                                                  TString centralityEstimator="V0M",\r
  15                                                  Double_t vertexZ=10.,\r
  16                                                  Double_t DCAxy=-1,\r
  17                                                  Double_t DCAz=-1,\r
  18                                                  Double_t ptMin=0.3,\r
  19                                                  Double_t ptMax=1.5,\r
  20                                                  Double_t etaMin=-0.8,\r
  21                                                  Double_t etaMax=0.8,\r
  22                                                  Double_t maxTPCchi2 = -1, \r
  23                                                  Int_t minNClustersTPC = -1,\r
  24                                                  Int_t AODfilterBit = 128,\r
  25                                                  TString fileNameBase="AnalysisResults") {\r
  26 \r
  27   // Creates a balance function analysis task and adds it to the analysis manager.\r
  28   // Get the pointer to the existing analysis manager via the static access method.\r
  29   TString centralityName("");\r
  30   centralityName+=Form("%.0f",centrMin);\r
  31   centralityName+="-";\r
  32   centralityName+=Form("%.0f",centrMax);\r
  33   centralityName+="_";\r
  34   centralityName+=Form("%s",centralityEstimator.Data());\r
  35   centralityName+="_";\r
  36   centralityName+=Form("vZ%.1f",vertexZ);\r
  37   centralityName+="_";\r
  38   centralityName+=Form("DCAxy%.1f",DCAxy);\r
  39   centralityName+="_";\r
  40   centralityName+=Form("DCAz%.1f",DCAz);\r
  41   centralityName+="_Pt";\r
  42   centralityName+=Form("%.1f",ptMin);\r
  43   centralityName+="-";\r
  44   centralityName+=Form("%.1f",ptMax);\r
  45   centralityName+="_Eta";\r
  46   centralityName+=Form("%.1f",etaMin);\r
  47   centralityName+="-";\r
  48   centralityName+=Form("%.1f",etaMax);\r
  49   centralityName+="_Chi";\r
  50   centralityName+=Form("%.1f",maxTPCchi2);\r
  51   centralityName+="_nClus";\r
  52   centralityName+=Form("%d",minNClustersTPC);\r
  53   centralityName+="_Bit";\r
  54   centralityName+=Form("%d",AODfilterBit);\r
  55 \r
  56 \r
  57 \r
  58 \r
  59   TString outputFileName(fileNameBase);\r
  60   outputFileName.Append(".root");\r
  61 \r
  62   //===========================================================================\r
  63   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  64   if (!mgr) {\r
  65     ::Error("AddTaskBF", "No analysis manager to connect to.");\r
  66     return NULL;\r
  67   }\r
  68 \r
  69   // Check the analysis type using the event handlers connected to the analysis manager.\r
  70   //===========================================================================\r
  71   if (!mgr->GetInputEventHandler()) {\r
  72     ::Error("AddTaskBF", "This task requires an input event handler");\r
  73     return NULL;\r
  74   }\r
  75   TString analysisType = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"\r
  76   if(dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler())) analysisType = "MC";\r
  77 \r
  78   // for local changed BF configuration\r
  79   //gROOT->LoadMacro("./configBalanceFunctionAnalysis.C");\r
  80   gROOT->LoadMacro("$ALICE_ROOT/PWG2/EBYE/macros/configBalanceFunctionAnalysis.C");\r
  81   AliBalance *bf  = 0;  // Balance Function object\r
  82   AliBalance *bfs = 0;  // shuffled Balance function object\r
  83 \r
  84   if (analysisType=="ESD"){\r
  85     bf  = GetBalanceFunctionObject("ESD",centralityEstimator,centrMin,centrMax);\r
  86     if(gRunShuffling) bfs = GetBalanceFunctionObject("ESD",centralityEstimator,centrMin,centrMax,kTRUE);\r
  87   }\r
  88   else if (analysisType=="AOD"){\r
  89     bf  = GetBalanceFunctionObject("AOD",centralityEstimator,centrMin,centrMax);\r
  90     if(gRunShuffling) bfs = GetBalanceFunctionObject("AOD",centralityEstimator,centrMin,centrMax,kTRUE);\r
  91   }\r
  92   else if (analysisType=="MC"){\r
  93     bf  = GetBalanceFunctionObject("MC",centralityEstimator,centrMin,centrMax);\r
  94     if(gRunShuffling) bfs = GetBalanceFunctionObject("MC",centralityEstimator,centrMin,centrMax,kTRUE);\r
  95   }\r
  96   else{\r
  97     ::Error("AddTaskBF", "analysis type NOT known.");\r
  98     return NULL;\r
  99   }\r
 100 \r
 101   // Create the task, add it to manager and configure it.\r
 102   //===========================================================================\r
 103   AliAnalysisTaskBF *taskBF = new AliAnalysisTaskBF("TaskBF");\r
 104   taskBF->SetAnalysisObject(bf);\r
 105   if(gRunShuffling) taskBF->SetShufflingObject(bfs);\r
 106 \r
 107   taskBF->SetCentralityPercentileRange(centrMin,centrMax);\r
 108   if(analysisType == "ESD") {\r
 109     AliESDtrackCuts *trackCuts = GetTrackCutsObject(ptMin,ptMax,etaMin,etaMax,maxTPCchi2,DCAxy,DCAz,minNClustersTPC);\r
 110     taskBF->SetAnalysisCutObject(trackCuts);\r
 111     \r
 112   }\r
 113   else if(analysisType == "AOD") {\r
 114     // pt and eta cut (pt_min, pt_max, eta_min, eta_max)\r
 115     taskBF->SetAODtrackCutBit(AODfilterBit);\r
 116     taskBF->SetKinematicsCutsAOD(ptMin,ptMax,etaMin,etaMax);\r
 117 \r
 118     // set extra DCA cuts (-1 no extra cut)\r
 119     taskBF->SetExtraDCACutsAOD(DCAxy,DCAz);\r
 120 \r
 121     // set extra TPC chi2 / nr of clusters cut\r
 122     taskBF->SetExtraTPCCutsAOD(maxTPCchi2, minNClustersTPC);\r
 123     \r
 124   }\r
 125   else if(analysisType == "MC") {\r
 126     taskBF->SetKinematicsCutsAOD(ptMin,ptMax,etaMin,etaMax); \r
 127   }\r
 128 \r
 129   // offline trigger selection (AliVEvent.h)\r
 130   // taskBF->UseOfflineTrigger(); // NOT used (selection is done with the AliAnalysisTaskSE::SelectCollisionCandidates()) \r
 131   // with this only selected events are analyzed (first 2 bins in event QA histogram are the same))\r
 132   // documentation in https://twiki.cern.ch/twiki/bin/viewauth/ALICE/PWG1EvSelDocumentation\r
 133   taskBF->SelectCollisionCandidates(AliVEvent::kMB);\r
 134   \r
 135   // centrality estimator (default = V0M)\r
 136   taskBF->SetCentralityEstimator(centralityEstimator);\r
 137   \r
 138   // vertex cut (x,y,z)\r
 139   taskBF->SetVertexDiamond(.3,.3,vertexZ);\r
 140   \r
 141 \r
 142 \r
 143   //bf->PrintAnalysisSettings();\r
 144   mgr->AddTask(taskBF);\r
 145   \r
 146   // Create ONLY the output containers for the data produced by the task.\r
 147   // Get and connect other common input/output containers via the manager as below\r
 148   //==============================================================================\r
 149   TString outputFileName = AliAnalysisManager::GetCommonFileName();\r
 150   outputFileName += ":PWG2EbyE.outputBalanceFunctionAnalysis";\r
 151   AliAnalysisDataContainer *coutQA = mgr->CreateContainer(Form("listQA_%s",centralityName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName.Data());\r
 152   AliAnalysisDataContainer *coutBF = mgr->CreateContainer(Form("listBF_%s",centralityName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName.Data());\r
 153   if(gRunShuffling) AliAnalysisDataContainer *coutBFS= mgr->CreateContainer(Form("listBFShuffled_%s",centralityName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName.Data());\r
 154   mgr->ConnectInput(taskBF, 0, mgr->GetCommonInputContainer());\r
 155   mgr->ConnectOutput(taskBF, 1, coutQA);\r
 156   mgr->ConnectOutput(taskBF, 2, coutBF);\r
 157   if(gRunShuffling) mgr->ConnectOutput(taskBF, 3, coutBFS);\r
 158 \r
 159   return taskBF;\r
 160 }\r