PWGHF/hfe/macros/AddTaskHFEpPb.C

   1 AliAnalysisTask *AddTaskHFEpPb(){\r
   2   // Switches for the TOF-TPC analysis\r
   3   Bool_t kTPC_Only                   = kTRUE;\r
   4   Bool_t kTPCTOF_Ref                 = kTRUE;\r
   5   Bool_t kTPCTOF_Sys                 = kFALSE;\r
   6   Bool_t kTRD_Test                   = kFALSE;\r
   7   //get the current analysis manager\r
   8   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
   9   if (!mgr) {\r
  10     Error("AddTask_hfe_HFE", "No analysis manager found.");\r
  11     return 0;\r
  12   }\r
  13   \r
  14   Bool_t MCthere=kTRUE;\r
  15   AliMCEventHandler *mcH = dynamic_cast<AliMCEventHandler*>(mgr->GetMCtruthEventHandler());\r
  16   if(!mcH){\r
  17     MCthere=kFALSE;\r
  18   }\r
  19   \r
  20   AliAnalysisDataContainer *cinput  = mgr->GetCommonInputContainer();\r
  21   \r
  22   //@@ 0 @@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@\r
  23   \r
  24   // Default settings \r
  25   const int kDefITScl = 3;\r
  26   const int kDefTPCcl = 110;\r
  27   const int kDefTPCclPID = 80;\r
  28   const double kDefDCAr = 1.;\r
  29   const double kDefDCAz = 2.;\r
  30   // For pPb, new fits by Martin, 07.02.2013\r
  31   const double kDefTPCs = 0.045;\r
  32   // it seems the sigma is 1 and not 0.88 (3 sigma = 2.685)\r
  33   const double kDefTPCu = 3.045;   \r
  34   const double kDefTOFs = 3.;\r
  35 \r
  36   Double_t dEdxlm[12] = {0.045,0.045,0.045,0.045,0.045,0.045,0.045,0.045,0.045,0.045,0.045,0.045};  //  50%\r
  37   Double_t dEdxhm[12] = {2.685,2.685,2.685,2.685,2.685,2.685,2.685,2.685,2.685,2.685,2.685,2.685};\r
  38   // For systematics:\r
  39   Double_t tpcl0[12]  = {-0.835,-0.835,-0.835,-0.835,-0.835,-0.835,-0.835,-0.835,-0.835,-0.835,-0.835,-0.835};  //  84%\r
  40   Double_t tpcl1[12]  = {-0.184,-0.184,-0.184,-0.184,-0.184,-0.184,-0.184,-0.184,-0.184,-0.184,-0.184,-0.184};  //  60%\r
  41   Double_t tpcl2[12]  = {0.265,0.265,0.265,0.265,0.265,0.265,0.265,0.265,0.265,0.265,0.265,0.265};  //   40%\r
  42 \r
  43   \r
  44   // For TPC only\r
  45   const double kDefTPCs1 = 0.034;\r
  46   const double kDefTPCu1 = 2.674;\r
  47   \r
  48   if(kTPC_Only){\r
  49     // for the moment (09.02.2013) use the same as TOF-TPC. Refine later on\r
  50     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,0,0,AliHFEextraCuts::kBoth);\r
  51   }\r
  52   \r
  53   //------------------------------//\r
  54   //   TPC-TOF analysis\r
  55   //------------------------------//\r
  56   \r
  57   if(kTPCTOF_Ref){\r
  58     // Reference task\r
  59      // Fits by Martin 07.02.2013\r
  60     // mean = 0.045\r
  61     // sigma = 0.88\r
  62     // 50%\r
  63     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  64     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,0.295,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  65 \r
  66     // add centrality\r
  67     RegisterTask(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,&dEdxlm[0],&dEdxhm[0],kDefTOFs,0,AliHFEextraCuts::kBoth); // 50%\r
  68     //if (!MCthere){\r
  69     //  RegisterTask(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,&tpcl0[0],&dEdxhm[0],kDefTOFs,0,AliHFEextraCuts::kBoth);  // 84%\r
  70     //}\r
  71   }\r
  72   if(kTPCTOF_Sys){\r
  73 \r
  74     // TPC PID\r
  75     if (!MCthere){\r
  76       // 84%\r
  77       RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,-0.835,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  78       // 60%\r
  79       RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,-0.184,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  80       // 40%\r
  81       RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,0.265,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  82     }\r
  83 \r
  84     // TOF PID\r
  85     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,2.0,0,AliHFEextraCuts::kBoth);\r
  86     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,4.0,0,AliHFEextraCuts::kBoth);\r
  87     // TOF latest mismatch - helps nothing\r
  88     //RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,1,AliHFEextraCuts::kBoth);\r
  89 \r
  90     // ITS hits and SPD request\r
  91     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,4,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  92     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,5,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  93     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,3,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kFirst);\r
  94     RegisterTaskPID2(MCthere,kDefTPCcl,kDefTPCclPID,4,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kFirst);\r
  95 \r
  96     // TPC clusters\r
  97     RegisterTaskPID2(MCthere,100,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  98     RegisterTaskPID2(MCthere,120,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
  99     RegisterTaskPID2(MCthere,130,kDefTPCclPID,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
 100     // TPC clusters PID\r
 101     RegisterTaskPID2(MCthere,kDefTPCcl,100,kDefITScl,kDefDCAr,kDefDCAz,kDefTPCs,kDefTPCu,kDefTOFs,0,AliHFEextraCuts::kBoth);\r
 102 \r
 103   }\r
 104   \r
 105   return NULL;\r
 106 }\r
 107 \r
 108 \r
 109 //===============================================================================\r
 110 \r
 111 AliAnalysisTask *RegisterTask(Bool_t useMC, Int_t tpcCls=120, Int_t tpcClsPID = 80, \r
 112                               Int_t itsCls=4, Double_t dcaxy=1.0, Double_t dcaz=2.0, \r
 113                               Double_t *tpcdEdxcutlow=NULL, \r
 114                               Double_t *tpcdEdxcuthigh=NULL, \r
 115                               //Double_t tpcs=-0.0113, Double_t tpcu=3.09, \r
 116                               Double_t tofs=3., Int_t tofm=0,\r
 117                               Int_t itshitpixel = AliHFEextraCuts::kBoth, \r
 118                               Bool_t withetacorrection = kFALSE){\r
 119 \r
 120   Int_t idcaxy = (Int_t)(dcaxy*10.);\r
 121   Int_t idcaz = (Int_t)(dcaz*10.);\r
 122   Int_t tpclow = 0;\r
 123   if(tpcdEdxcutlow) tpclow = (Int_t) (tpcdEdxcutlow[0]*1000.);\r
 124   Int_t itofs = (Int_t)(tofs*10.);\r
 125   Int_t ipixelany = itshitpixel;\r
 126   Int_t ietacorr = 0;\r
 127   if(withetacorrection) ietacorr = 1;\r
 128   \r
 129   TString appendix(TString::Format("centTPCc%dTPCp%dITS%dDCAr%dz%dTPCs%dTOFs%dm%ipa%d",tpcCls,\r
 130                                    tpcClsPID,itsCls,idcaxy,idcaz,tpclow,itofs,tofm,ipixelany));\r
 131   printf("Add macro appendix %s\n", appendix.Data());\r
 132 \r
 133 \r
 134   gROOT->LoadMacro("$ALICE_ROOT/PWGHF/hfe/macros/configs/pPb/ConfigHFEpPb.C");\r
 135   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
 136   AliAnalysisDataContainer *cinput  = mgr->GetCommonInputContainer();\r
 137   AliAnalysisTaskHFE *task = ConfigHFEpPb(useMC, appendix, tpcCls, tpcClsPID, itsCls, dcaxy, dcaz, \r
 138                                           tpcdEdxcutlow,tpcdEdxcuthigh,\r
 139                                           tofs,tofm,itshitpixel);\r
 140 \r
 141   task->SetESDAnalysis();\r
 142   if (useMC)\r
 143     task->SetHasMCData(kTRUE);\r
 144   else{\r
 145     task->SetHasMCData(kFALSE);\r
 146   }\r
 147   task->SelectCollisionCandidates(AliVEvent::kINT7);\r
 148 \r
 149   TString containerName = mgr->GetCommonFileName();\r
 150   containerName += ":HFEpPbcent";\r
 151   containerName += appendix.Data();\r
 152   printf("container name: %s\n", containerName.Data());\r
 153  \r
 154   //create data containers\r
 155   task->ConnectOutput(1, mgr->CreateContainer(Form("HFE_Results_%s", appendix.Data()), \r
 156                                               TList::Class(), AliAnalysisManager::kOutputContainer, \r
 157                                               containerName.Data()));\r
 158   task->ConnectOutput(2, mgr->CreateContainer(Form("HFE_QA_%s", appendix.Data()), TList::Class(), \r
 159                                               AliAnalysisManager::kOutputContainer, \r
 160                                               containerName.Data()));\r
 161   mgr->ConnectInput(task,  0, cinput );\r
 162 \r
 163   mgr->AddTask(task);\r
 164   return NULL;\r
 165 }\r
 166 \r
 167 //===============================================================================\r
 168 \r
 169 AliAnalysisTask *RegisterTaskPID2(Bool_t useMC, Int_t tpcCls=120, Int_t tpcClsPID = 80, \r
 170                                   Int_t itsCls=4, Double_t dcaxy=1.0, Double_t dcaz=2.0, \r
 171                                   Double_t tpcs=-0.0113, Double_t tpcu=3.09, Double_t tofs=3., \r
 172                                   Int_t tofm=0,\r
 173                                   Int_t itshitpixel = AliHFEextraCuts::kBoth, \r
 174                                   Bool_t withetacorrection = kTRUE){\r
 175   gROOT->LoadMacro("$ALICE_ROOT/PWGHF/hfe/macros/configs/pPb/ConfigHFEmbpPb.C");\r
 176   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
 177   AliAnalysisDataContainer *cinput  = mgr->GetCommonInputContainer();\r
 178   AliAnalysisTaskHFE *task = ConfigHFEmbpPb(useMC, tpcCls, tpcClsPID, itsCls, dcaxy, dcaz, \r
 179                                              tpcs,tpcu,tofs,tofm,3.,kFALSE,kTRUE,kFALSE,itshitpixel,withetacorrection);\r
 180   task->SetESDAnalysis();\r
 181   if (useMC)\r
 182     task->SetHasMCData(kTRUE);\r
 183   else{\r
 184     task->SetHasMCData(kFALSE);\r
 185   }\r
 186   task->SelectCollisionCandidates(AliVEvent::kINT7);\r
 187  \r
 188   Int_t idcaxy = (Int_t)(dcaxy*10.);\r
 189   Int_t idcaz = (Int_t)(dcaz*10.);\r
 190   Int_t itpcs = (Int_t)(tpcs*1000.);\r
 191   Int_t itofs = (Int_t)(tofs*10.);\r
 192   Int_t ipixelany = itshitpixel;\r
 193   Int_t ietacorr = 0;\r
 194   if(withetacorrection) ietacorr = 1;\r
 195   \r
 196   TString appendix(TString::Format("TPCc%dTPCp%dITS%dDCAr%dz%dTPCs%dTOFs%dm%ipa%d",tpcCls,\r
 197                                    tpcClsPID,itsCls,idcaxy,idcaz,itpcs,itofs,tofm,ipixelany));\r
 198   printf("Add macro appendix %s\n", appendix.Data());\r
 199 \r
 200   TString containerName = mgr->GetCommonFileName();\r
 201   containerName += ":HFEpPbmb";\r
 202   containerName += appendix.Data();\r
 203   printf("container name: %s\n", containerName.Data());\r
 204 \r
 205   //create data containers\r
 206   task->ConnectOutput(1, mgr->CreateContainer(Form("HFE_Results_%s", appendix.Data()), \r
 207                                               TList::Class(), AliAnalysisManager::kOutputContainer, \r
 208                                               containerName.Data()));\r
 209   task->ConnectOutput(2, mgr->CreateContainer(Form("HFE_QA_%s", appendix.Data()), TList::Class(), \r
 210                                               AliAnalysisManager::kOutputContainer, \r
 211                                               containerName.Data()));\r
 212   mgr->ConnectInput(task,  0, cinput );\r
 213 \r
 214   mgr->AddTask(task);\r
 215   return NULL;\r
 216 }\r
 217 \r
 218 \r
 219 //=========================================================================\r