PWG4/macros/AddTaskPWG4HighPtSpectra.C

   1 //DEFINITION OF A FEW CONSTANTS
   2 #define round(x) ((x)>=0?(long)((x)+0.5):(long)((x)-0.5))
   3
   4 const Float_t phimin = 0.;
   5 const Float_t phimax = 2.*TMath::Pi();
   6 const Float_t etamin = -0.9;
   7 const Float_t etamax = 0.9;
   8
   9 const Int_t   mintrackrefsTPC = 0;
  10 const Int_t   mintrackrefsITS = 0;
  11
  12 void AddTaskPWG4HighPtSpectraAll(char *prodType = "LHC10h",Bool_t isPbPb=kTRUE, Int_t iAODanalysis = 0)
  13 {
  14   int cent = 10;
  15
  16   AliPWG4HighPtSpectra *taskSpectra00cent10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,0,0);
  17   AliPWG4HighPtSpectra *taskSpectra01cent10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,0,1);
  18   //  AliPWG4HighPtSpectra *taskSpectra02cent10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,0,2);
  19   //  AliPWG4HighPtSpectra *taskSpectra10cent10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,1,0);
  20   //  AliPWG4HighPtSpectra *taskSpectra20cent10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,2,0);
  21   AliPWG4HighPtSpectra *taskSpectra70cent10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,7,0);
  22   AliPWG4HighPtSpectra *taskSpectra71cent10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,7,1);
  23   AliPWG4HighPtSpectra *taskSpectra72cent10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,7,2);
  24
  25   if(isPbPb) {
  26     for(cent=0; cent<4; cent++) {
  27       AliPWG4HighPtSpectra *taskSpectra00 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,0,0);
  28       AliPWG4HighPtSpectra *taskSpectra01 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,0,1);
  29       //      AliPWG4HighPtSpectra *taskSpectra02 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,0,2);
  30       //      AliPWG4HighPtSpectra *taskSpectra10 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,1,0);
  31       //      AliPWG4HighPtSpectra *taskSpectra20 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,2,0);
  32       AliPWG4HighPtSpectra *taskSpectra70 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,7,0);
  33       AliPWG4HighPtSpectra *taskSpectra71 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,7,1);
  34       AliPWG4HighPtSpectra *taskSpectra72 = AddTaskPWG4HighPtSpectra(prodType,isPbPb,cent,7,2);
  35     }
  36   }
  37
  38 }
  39
  40
  41 AliPWG4HighPtSpectra* AddTaskPWG4HighPtSpectra(char *prodType = "LHC10e14", Bool_t isPbPb=kTRUE,Int_t centClass = 0, Int_t trackType = 0, Int_t cuts = 0)
  42 {
  43
  44   /*
  45     trackType: 0 = global
  46                1 = TPC stand alone
  47                2 = TPC stand alone constrained to SPD vertex
  48     cuts:      0 (global) = standard ITSTPC2010 a la RAA analysis
  49                1 (global) = ITSrefit, no SPD requirements -> standard for jet analysis
  50                2 (global) = ITSrefit + no hits in SPD
  51                3 (global) = standard ITS tight cuts with nCrossed rows cut for hybrid tracks
  52                0 (TPC)    = standard TPC + NClusters>70
  53                1 (TPC)    = standard TPC + NClusters>0 --> to study new TPC QA recommendations
  54    */
  55
  56   // Creates HighPtSpectra analysis task and adds it to the analysis manager.
  57
  58   //Load common track cut class
  59   gROOT->LoadMacro("$ALICE_ROOT/PWG4/macros/CreateTrackCutsPWG4.C");
  60
  61   // A. Get the pointer to the existing analysis manager via the static access method.
  62   //==============================================================================
  63   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  64   if (!mgr) {
  65     Error("AddTaskPWG4HighPtSpectra", "No analysis manager to connect to.");
  66     return NULL;
  67   }
  68
  69   // B. Check the analysis type using the event handlers connected to the analysis
  70   //    manager. The availability of MC handler can also be checked here.
  71   //==============================================================================
  72   if (!mgr->GetInputEventHandler()) {
  73     ::Error("AddTaskPWG4HighPtSpectra", "This task requires an input event handler");
  74     return NULL;
  75   }
  76   TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
  77   const char *analysisType = "ESD";//"TPC"
  78
  79   // C. Create the task, add it to manager.
  80   //===========================================================================
  81  //CONTAINER DEFINITION
  82   Info("AliPWG4HighPtSpectra","SETUP CONTAINER");
  83   //the sensitive variables, their indices
  84   UInt_t ipt   = 0;
  85   UInt_t iphi  = 1;
  86   UInt_t ieta  = 2;
  87   UInt_t incls = 3;
  88
  89   //Setting up the container grid...
  90   UInt_t nstep = 4; //Steps/Modes for containers
  91   Int_t kStepReconstructed          = 0;
  92   Int_t kStepSecondaries            = 1;
  93   Int_t kStepReconstructedMC        = 2;
  94   Int_t kStepMCAcceptance           = 3;
  95
  96   //redefine pt ranges in case of Jet-Jet production
  97   Float_t ptBinEdges[4][4];
  98   Float_t ptmin =  0.0 ;
  99   Float_t ptmax =  100.0 ;
 100
 101   ptBinEdges[0][0] = 2.;
 102   ptBinEdges[0][1] = 0.2;
 103   ptBinEdges[1][0] = 6.;
 104   ptBinEdges[1][1] = 0.5;
 105   ptBinEdges[2][0] = 20.;
 106   ptBinEdges[2][1] = 2.;
 107   ptBinEdges[3][0] = 100.;
 108   ptBinEdges[3][1] = 5.;
 109
 110   const Int_t nvar   = 4; //number of variables on the grid: pt:phi:eta:NClsIter1
 111
 112   const Int_t nbin11 = round((ptBinEdges[0][0]-ptmin)/ptBinEdges[0][1]);
 113   const Int_t nbin12 = round((ptBinEdges[1][0]-ptBinEdges[0][0])/ptBinEdges[1][1])+nbin11;
 114   const Int_t nbin13 = round((ptBinEdges[2][0]-ptBinEdges[1][0])/ptBinEdges[2][1])+nbin12;
 115   const Int_t nbin14 = round((ptBinEdges[3][0]-ptBinEdges[2][0])/ptBinEdges[3][1])+nbin13;
 116
 117   const Int_t nbin1  = nbin14; //bins in pt
 118   const Int_t nbin2  =  18;    //bins in phi
 119   const Int_t nbin3  =  2;     //bins in eta
 120   const Int_t nbin4  =  6;     //bins in NClsIter1: 0 70 80 90 100 120
 121
 122
 123   //arrays for the number of bins in each dimension
 124   Int_t iBin[nvar];
 125   iBin[0]=nbin1;
 126   iBin[1]=nbin2;
 127   iBin[2]=nbin3;
 128   iBin[3]=nbin4;
 129
 130   //arrays for lower bounds :
 131   Double_t *binLim1=new Double_t[nbin1+1];
 132   Double_t *binLim2=new Double_t[nbin2+1];
 133   Double_t *binLim3=new Double_t[nbin3+1];
 134   Double_t *binLim4=new Double_t[nbin4+1];
 135
 136   //values for bin lower bounds
 137   for(Int_t i=0; i<=nbin1; i++) {
 138     if(i<=nbin11) binLim1[i]=(Double_t)ptmin + (ptBinEdges[0][0]-ptmin)/nbin11*(Double_t)i ;
 139     if(i<=nbin12 && i>nbin11) binLim1[i]=(Double_t)ptBinEdges[0][0] + (ptBinEdges[1][0]-ptBinEdges[0][0])/(nbin12-nbin11)*((Double_t)i-(Double_t)nbin11) ;
 140     if(i<=nbin13 && i>nbin12) binLim1[i]=(Double_t)ptBinEdges[1][0] + (ptBinEdges[2][0]-ptBinEdges[1][0])/(nbin13-nbin12)*((Double_t)i-(Double_t)nbin12) ;
 141     if(i<=nbin14 && i>nbin13) binLim1[i]=(Double_t)ptBinEdges[2][0] + (ptBinEdges[3][0]-ptBinEdges[2][0])/(nbin14-nbin13)*((Double_t)i-(Double_t)nbin13) ;
 142   }
 143   for(Int_t i=0; i<=nbin2; i++) binLim2[i]=(Double_t)phimin + (phimax-phimin)/nbin2*(Double_t)i ;
 144   for(Int_t i=0; i<=nbin3; i++) binLim3[i]=(Double_t)etamin + (etamax-etamin)/nbin3*(Double_t)i ;
 145   binLim4[0] = 0.;
 146   binLim4[1] = 70.;
 147   binLim4[2] = 80.;
 148   binLim4[3] = 90.;
 149   binLim4[4] = 100.;
 150   binLim4[5] = 120.;
 151   binLim4[6] = 160.;
 152
 153
 154   AliCFContainer* containerPos = new AliCFContainer("containerPos","container for positive tracks",nstep,nvar,iBin);
 155   //setting the bin limits
 156   containerPos -> SetBinLimits(ipt,binLim1);
 157   containerPos -> SetBinLimits(iphi,binLim2);
 158   containerPos -> SetBinLimits(ieta,binLim3);
 159   containerPos -> SetBinLimits(incls,binLim4);
 160
 161   AliCFContainer* containerNeg = new AliCFContainer("containerNeg","container for negative tracks",nstep,nvar,iBin);
 162   //setting the bin limits
 163   containerNeg -> SetBinLimits(ipt,binLim1);
 164   containerNeg -> SetBinLimits(iphi,binLim2);
 165   containerNeg -> SetBinLimits(ieta,binLim3);
 166   containerNeg -> SetBinLimits(incls,binLim4);
 167
 168   //CREATE THE  CUTS -----------------------------------------------
 169   //Use AliESDtrackCuts
 170   AliESDtrackCuts *trackCuts = new AliESDtrackCuts("AliESDtrackCuts","Standard Cuts");
 171   AliESDtrackCuts *trackCutsReject = 0x0;
 172   //Standard Cuts
 173   //Set track cuts for global tracks
 174   if(trackType==0 && cuts==0) {
 175     // tight global tracks - RAA analysis
 176     trackCuts = CreateTrackCutsPWG4(1000);
 177   }
 178   if(trackType==0 && cuts==1) {
 179     //Cuts global tracks with ITSrefit requirement and SPDrequirement for jet analysis
 180     trackCuts = CreateTrackCutsPWG4(10001005);
 181   }
 182   if(trackType==0 && cuts==2) {
 183     //Cuts global tracks with ITSrefit requirement but without SPD
 184     trackCuts = CreateTrackCutsPWG4(10011005);
 185   }
 186   if(trackType==7 && cuts==0) {
 187     // no requirements on SPD and ITSrefit failed
 188     trackCuts = CreateTrackCutsPWG4(10041005);   //no ITSrefit requirement
 189     trackCutsReject = CreateTrackCutsPWG4(1005); //ITSrefit requirement
 190     trackCutsReject->SetEtaRange(-etamin,etamax);
 191     trackCutsReject->SetPtRange(0.15, 1e10);
 192   }
 193   if(trackType==7 && cuts==1) {
 194     //Cuts global tracks with ITSrefit requirement but without SPD
 195     trackCuts = CreateTrackCutsPWG4(10011005);
 196   }
 197   if(trackType==7 && cuts==2) {
 198     // no requirements on SPD and ITSrefit failed
 199     trackCuts = CreateTrackCutsPWG4(10041005);       //no ITSrefit requirement filter 256
 200     trackCutsReject = CreateTrackCutsPWG4(10001005); //ITSrefit requirement filter 16
 201     trackCutsReject->SetEtaRange(etamin,etamax);
 202     trackCutsReject->SetPtRange(0.15, 1e10);
 203   }
 204
 205   if(trackType==1 && cuts==0) {
 206     //Set track cuts for TPConly tracks
 207     trackCuts = CreateTrackCutsPWG4(2001);
 208   }
 209   if(trackType==2 && cuts==0) {
 210      //       Set track cuts for TPConly constrained tracks
 211     trackCuts = CreateTrackCutsPWG4(2001);
 212   }
 213   trackCuts->SetEtaRange(etamin,etamax);
 214   trackCuts->SetPtRange(0.15, 1e10);
 215
 216   // Gen-Level kinematic cuts
 217   AliCFTrackKineCuts *mcKineCuts = new AliCFTrackKineCuts("mcKineCuts","MC-level kinematic cuts");
 218   mcKineCuts->SetPtRange(0.15,1e10);
 219   mcKineCuts->SetEtaRange(etamin,etamax);//-0.5,0.5);
 220   mcKineCuts->SetRequireIsCharged(kTRUE);
 221
 222   //Acceptance Cuts
 223   //AliCFAcceptanceCuts *mcAccCuts = new AliCFAcceptanceCuts("mcAccCuts","MC acceptance cuts");
 224   // mcAccCuts->SetMinNHitITS(mintrackrefsITS);
 225   //mcAccCuts->SetMinNHitTPC(mintrackrefsTPC);
 226
 227   TObjArray* recList = new TObjArray(0);
 228   TObjArray* secList = new TObjArray(0) ;
 229   TObjArray* recMCList = new TObjArray(0);
 230
 231   printf("CREATE MC KINE CUTS\n");
 232   TObjArray* mcList = new TObjArray(0) ;
 233   mcList->AddLast(mcKineCuts);
 234   //mcList->AddLast(mcAccCuts);
 235
 236   //CREATE THE INTERFACE TO CORRECTION FRAMEWORK USED IN THE TASK
 237   printf("CREATE INTERFACE AND CUTS\n");
 238   AliCFManager* manPos = new AliCFManager("manPos","Manager for Positive tracks") ;
 239   manPos->SetParticleContainer(containerPos);
 240   manPos->SetParticleCutsList(kStepReconstructed,recList);
 241   manPos->SetParticleCutsList(kStepSecondaries,secList);
 242   manPos->SetParticleCutsList(kStepReconstructedMC,recMCList);
 243   manPos->SetParticleCutsList(kStepMCAcceptance,mcList);
 244
 245   AliCFManager* manNeg = new AliCFManager("manNeg","Manager for Negative tracks") ;
 246   manNeg->SetParticleContainer(containerNeg);
 247   manNeg->SetParticleCutsList(kStepReconstructed,recList);
 248   manNeg->SetParticleCutsList(kStepSecondaries,secList);
 249   manNeg->SetParticleCutsList(kStepReconstructedMC,recMCList);
 250   manNeg->SetParticleCutsList(kStepMCAcceptance,mcList);
 251
 252
 253   AliPWG4HighPtSpectra *taskPWG4HighPtSpectra = new AliPWG4HighPtSpectra(Form("AliPWG4HighPtSpectraCent%dTrackType%dCuts%d",centClass,trackType,cuts));
 254   taskPWG4HighPtSpectra->SetTrackType(trackType);
 255   taskPWG4HighPtSpectra->SetCuts(trackCuts);
 256   taskPWG4HighPtSpectra->SetCutsReject(trackCutsReject);
 257   taskPWG4HighPtSpectra->SetCFManagerPos(manPos); //here is set the CF manager +
 258   taskPWG4HighPtSpectra->SetCFManagerNeg(manNeg); //here is set the CF manager -
 259
 260   if(isPbPb) {
 261     taskPWG4HighPtSpectra->SetIsPbPb(kTRUE);
 262     taskPWG4HighPtSpectra->SetCentralityClass(centClass);
 263   }
 264   //  taskPWG4HighPtSpectra->SetSigmaConstrainedMax(5.);
 265
 266
 267   // E. Create ONLY the output containers for the data produced by the task.
 268   // Get and connect other common input/output containers via the manager as below
 269   //==============================================================================
 270
 271   //------ output containers ------
 272   TString outputfile = AliAnalysisManager::GetCommonFileName();
 273   outputfile += Form(":PWG4_HighPtSpectraCent%dTrackType%dCuts%d",centClass,trackType,cuts);
 274
 275   AliAnalysisDataContainer *coutput0 = mgr->CreateContainer(Form("chist0HighPtSpectraCent%dTrackType%dCuts%d",centClass,trackType,cuts), TList::Class(),AliAnalysisManager::kOutputContainer,outputfile);
 276   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(Form("ccontainer0HighPtSpectraCent%dTrackType%dCuts%d",centClass,trackType,cuts), AliCFContainer::Class(),AliAnalysisManager::kOutputContainer,outputfile);
 277   AliAnalysisDataContainer *coutput2 = mgr->CreateContainer(Form("ccontainer1HighPtSpectraCent%dTrackType%dCuts%d",centClass,trackType,cuts), AliCFContainer::Class(),AliAnalysisManager::kOutputContainer,outputfile);
 278   AliAnalysisDataContainer *cout_cuts0 = mgr->CreateContainer(Form("qa_SpectraTrackCutsCent%dTrackType%dCuts%d",centClass,trackType,cuts), AliESDtrackCuts::Class(), AliAnalysisManager::kParamContainer,outputfile);
 279
 280   mgr->AddTask(taskPWG4HighPtSpectra);
 281
 282   mgr->ConnectInput(taskPWG4HighPtSpectra,0,mgr->GetCommonInputContainer());
 283   mgr->ConnectOutput(taskPWG4HighPtSpectra,0,coutput0);
 284   mgr->ConnectOutput(taskPWG4HighPtSpectra,1,coutput1);
 285   mgr->ConnectOutput(taskPWG4HighPtSpectra,2,coutput2);
 286   mgr->ConnectOutput(taskPWG4HighPtSpectra,3,cout_cuts0);
 287
 288   // Return task pointer at the end
 289   return taskPWG4HighPtSpectra;
 290 }