updated

[u/mrichter/AliRoot.git] / PWGHF / hfe / macros / configs / pPb / ConfigHFEpPb.C

AliAnalysisTaskHFE* ConfigHFEpPb(Bool_t useMC, TString appendix, \r
				 UChar_t TPCcl=70, UChar_t TPCclPID = 80, \r
				 UChar_t ITScl=3, Double_t DCAxy=1000., Double_t DCAz=1000.,\r
				 Double_t* tpcdEdxcutlow=NULL,Double_t* tpcdEdxcuthigh=NULL,\r
				 Double_t TOFs=3., Int_t TOFmis=0,\r
				 Int_t itshitpixel = 0){\r
  \r
  Bool_t kAnalyseTaggedTracks = kTRUE;\r
  \r
  //***************************************//\r
  //        Setting up the HFE cuts        //\r
  //***************************************//\r
  \r
  AliHFEcuts *hfecuts = new AliHFEcuts(appendix,"HFE cuts for pPb");\r
\r
  //hfecuts->SetQAOn();\r
  hfecuts->CreateStandardCuts();\r
  hfecuts->SetMinNClustersTPC(TPCcl);\r
  hfecuts->SetMinNClustersTPCPID(TPCclPID);\r
  hfecuts->SetMinNClustersITS(ITScl);\r
  hfecuts->SetMinRatioTPCclusters(0.6);\r
  hfecuts->SetTPCmodes(AliHFEextraCuts::kFound, AliHFEextraCuts::kFoundOverFindable);\r
  hfecuts->SetCutITSpixel(itshitpixel);\r
  hfecuts->SetCheckITSLayerStatus(kFALSE);\r
  \r
  //if((iPixelAny==AliHFEextraCuts::kAny) || (iPixelAny==AliHFEextraCuts::kSecond))     \r
  //hfecuts->SetProductionVertex(0,7,0,7);\r
 \r
  hfecuts->SetMaxImpactParam(DCAxy,DCAz);\r
  hfecuts->SetUseMixedVertex(kTRUE);\r
  hfecuts->SetVertexRange(10.);\r
  // New pPb cuts (February 2013)\r
  hfecuts->SetUseCorrelationVertex();\r
  hfecuts->SetSPDVtxResolutionCut();\r
\r
  // TOF settings:\r
  Int_t usetof=0;\r
  Bool_t kTOFmis=kFALSE;\r
  if (TOFs>0.){\r
    usetof = 1;\r
    printf("CONFIGURATION FILE: TOF is used \n");\r
    hfecuts->SetTOFPIDStep(kTRUE);\r
    printf("CONFIGURATION FILE: TOF PID step is requested !!!! \n");\r
    if (TOFmis>0){\r
      kTOFmis = kTRUE;\r
      printf("CONFIGURATION FILE: TOF mismatch rejection is set ON \n");\r
    }\r
  }\r
\r
  //***************************************//\r
  //        Setting up the task            //\r
  //***************************************//\r
\r
  AliAnalysisTaskHFE *task = new AliAnalysisTaskHFE(appendix);\r
  printf("task %p\n", task);\r
  task->SetpPbAnalysis();\r
  task->SetRemoveFirstEventInChunk();\r
  task->SetRemovePileUp(kFALSE);\r
  task->SetHFECuts(hfecuts);\r
  task->GetPIDQAManager()->SetHighResolutionHistos();\r
\r
  //***************************************//\r
  //          Variable manager             //\r
  //***************************************//\r
\r
  // Define Variables\r
  Double_t ptbinning[36] = {0., 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1., 1.1, 1.2, 1.3, 1.4, 1.5, 1.75, 2., 2.25, 2.5, 2.75, 3., 3.5, 4., 4.5, 5., 5.5, 6., 7., 8., 10., 12., 14., 16., 18., 20.};\r
\r
  Double_t etabinning[9] = {-0.8, -0.6, -0.4, -0.2, 0., 0.2, 0.4, 0.6, 0.8};\r
  //Double_t etabinning[17] = {-0.8, -0.7, -0.6, -0.5, -0.4, -0.3, -0.2, -0.1, 0., 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8};\r
\r
  Int_t sizept=(sizeof(ptbinning)/sizeof(double))-1;\r
  Int_t sizeeta=(sizeof(etabinning)/sizeof(double))-1;\r
\r
  AliHFEvarManager *vm = task->GetVarManager();\r
  vm->AddVariable("pt", sizept, ptbinning);\r
  vm->AddVariable("eta", sizeeta, -0.8,0.8);\r
  vm->AddVariable("phi",21, -0, 2*TMath::Pi());\r
  vm->AddVariable("charge");\r
  vm->AddVariable("source");\r
  vm->AddVariable("centrality");\r
\r
  // For the moment, remove the part dedicated to the background subtraction.\r
  // It will be implemented in a different way, reading it from a root file.\r
\r
  //***************************************//\r
  //          Configure the PID            //\r
  //***************************************//\r
\r
  // Define PID\r
  AliHFEpid *pid = task->GetPID();\r
  if(useMC) pid->SetHasMCData(kTRUE);\r
\r
  if (usetof){\r
    pid->AddDetector("TOF", 0);\r
    pid->AddDetector("TPC", 1);\r
  } else {\r
    pid->AddDetector("TPC", 0);\r
  }\r
  \r
  // Configure TPC PID\r
  if(!useMC){\r
    Double_t paramsTPCdEdxcutlow[12] ={0.0, 0.0, 0.0,0.0,0.0,0.0,0.0,0.0,0.0,0.0,0.0,0.0};\r
    if(tpcdEdxcutlow) memcpy(paramsTPCdEdxcutlow,tpcdEdxcutlow,sizeof(paramsTPCdEdxcutlow));\r
\r
    Double_t paramsTPCdEdxcuthigh[12] ={3.0, 3.0, 3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0};\r
    if(tpcdEdxcuthigh) memcpy(paramsTPCdEdxcuthigh,tpcdEdxcuthigh,sizeof(paramsTPCdEdxcuthigh));\r
    \r
    char *cutmodel;\r
    cutmodel="pol0";\r
    \r
    for(Int_t a=0;a<11;a++)\r
    {\r
      //   cout << a << " " << paramsTPCdEdxcut[a] << endl;\r
      Double_t tpcparamlow[1]={paramsTPCdEdxcutlow[a]};\r
      Float_t tpcparamhigh=paramsTPCdEdxcuthigh[a];\r
      pid->ConfigureTPCcentralityCut(a,cutmodel,tpcparamlow,tpcparamhigh);\r
    }\r
  }\r
\r
  // Configure TOF PID\r
  if (usetof){\r
    pid->ConfigureTOF(TOFs);\r
    AliHFEpidTOF *tofpid = pid->GetDetPID(AliHFEpid::kTOFpid);\r
    if (kTOFmis){\r
      tofpid->SetRejectTOFmismatch();\r
    }\r
  }\r
  // To make different upper TOF cut to see contamination effect\r
  // The below two lines should be removed after this check\r
  //AliHFEpidTOF *tofpid = pid->GetDetPID(AliHFEpid::kTOFpid);\r
  //if(TOFs<3.) tofpid->SetTOFnSigmaBand(-3,TOFs); //only to check the assymmetric tof cut\r
\r
  //***************************************//\r
  //          V0 tagged tracks             //\r
  //***************************************//\r
  if(kAnalyseTaggedTracks){\r
    AliHFEcuts *v0trackCuts = new AliHFEcuts("V0trackCuts", "Track Cuts for tagged track Analysis");\r
    v0trackCuts->CreateStandardCuts();\r
    v0trackCuts->SetMinNClustersTPC(TPCcl);  \r
    v0trackCuts->SetMinNClustersTPCPID(TPCclPID);\r
    v0trackCuts->SetMinRatioTPCclusters(0.6);\r
    v0trackCuts->SetTPCmodes(AliHFEextraCuts::kFound, AliHFEextraCuts::kFoundOverFindable);\r
    v0trackCuts->SetMinNClustersITS(1);\r
    v0trackCuts->SetCutITSpixel(AliHFEextraCuts::kFirst);\r
    v0trackCuts->SetCheckITSLayerStatus(kFALSE);\r
    v0trackCuts->UnsetVertexRequirement();\r
    //hfecuts->SetSigmaToVertex(10);\r
    v0trackCuts->SetTOFPIDStep(kTRUE);\r
    v0trackCuts->SetQAOn();\r
\r
    task->SwitchOnPlugin(AliAnalysisTaskHFE::kTaggedTrackAnalysis);\r
    task->SetTaggedTrackCuts(v0trackCuts);\r
    task->SetCleanTaggedTrack(kTRUE);\r
  }\r
\r
  // QA\r
  printf("task %p\n", task);\r
  task->SetQAOn(AliAnalysisTaskHFE::kPIDqa);\r
  task->SetQAOn(AliAnalysisTaskHFE::kMCqa);    \r
  task->SwitchOnPlugin(AliAnalysisTaskHFE::kDEstep);\r
\r
  printf("*************************************\n");\r
  printf("Configuring standard Task:\n");\r
  task->PrintStatus();\r
  pid->PrintStatus();\r
  printf("*************************************\n"); \r
  return task;\r
}\r