PWG3/vertexingHF/macros/AddTaskCFMultiVarMultiStep.C

   1 //DEFINITION OF A FEW CONSTANTS
   2 const Double_t ymin  = -2.1 ;
   3 const Double_t ymax  =  2.1 ;
   4 // const Double_t ptmin_0_4 =  0.0 ;
   5 // const Double_t ptmax_0_4 =  4.0 ;
   6 // const Double_t ptmin_4_8 =  4.0 ;
   7 // const Double_t ptmax_4_8 =  8.0 ;
   8 // const Double_t ptmin_8_10 =  8.0 ;
   9 // const Double_t ptmax_8_10 =  10.0 ;
  10 const Double_t cosmin = -1.05;
  11 const Double_t cosmax =  1.05;
  12 const Double_t cTmin = 0;  // micron
  13 const Double_t cTmax = 500;  // micron
  14 const Double_t dcamin = 0;  // micron
  15 const Double_t dcamax = 500;  // micron
  16 const Double_t d0min = -1000;  // micron
  17 const Double_t d0max = 1000;  // micron
  18 const Double_t d0xd0min = -100000;  // micron
  19 const Double_t d0xd0max = 100000;  // micron
  20 const Double_t phimin = 0.0;
  21 //const Double_t phimax = 2Pi;  // defined in the macro!!!!!!!!!!!!!!
  22 const Int_t    mintrackrefsTPC = 2 ;
  23 const Int_t    mintrackrefsITS = 3 ;
  24 const Int_t    charge  = 1 ;
  25 const Int_t    PDG = 421;
  26 const Int_t    minclustersTPC = 50 ;
  27 // cuts
  28 const Double_t ptmin = 0.1;
  29 const Double_t ptmax = 9999.;
  30 const Double_t etamin = -0.9;
  31 const Double_t etamax = 0.9;
  32 const Double_t zmin = -15;
  33 const Double_t zmax = 15;
  34 const Int_t    minITSClusters = 5;
  35
  36 //----------------------------------------------------
  37
  38 AliCFHeavyFlavourTaskMultiVarMultiStep *AddTaskCFMultiVarMultiStep(const char* cutFile = "./D0toKpiCuts.root",Bool_t isKeepD0fromB=kFALSE, Bool_t isKeepD0fromBOnly=kFALSE)
  39 {
  40   printf("Adding CF task using cuts from file %s\n",cutFile);
  41
  42   TFile* fileCuts = new TFile(cutFile);
  43   AliRDHFCutsD0toKpi *cutsD0toKpi = (AliRDHFCutsD0toKpi*)fileCuts->Get("D0toKpiCuts");
  44
  45   // check that the fKeepD0fromB flag is set to true when the fKeepD0fromBOnly flag is true
  46   //  for now the binning is the same than for all D's
  47   if(isKeepD0fromBOnly) isKeepD0fromB = true;
  48
  49   /*
  50     Double_t ptmin_0_4;
  51     Double_t ptmax_0_4;
  52     Double_t ptmin_4_8;
  53     Double_t ptmax_4_8;
  54     Double_t ptmin_8_10;
  55     Double_t ptmax_8_10;
  56
  57     if(!isKeepD0fromB){
  58     ptmin_0_4 =  0.0 ;
  59     ptmax_0_4 =  4.0 ;
  60     ptmin_4_8 =  4.0 ;
  61     ptmax_4_8 =  8.0 ;
  62     ptmin_8_10 =  8.0 ;
  63     ptmax_8_10 =  10.0 ;
  64     } else{
  65     ptmin_0_4 =  0.0 ;
  66     ptmax_0_4 =  3.0 ;
  67     ptmin_4_8 =  3.0 ;
  68     ptmax_4_8 =  5.0 ;
  69     ptmin_8_10 =  5.0 ;
  70     ptmax_8_10 =  10.0 ;
  71     }
  72   */
  73
  74         //CONTAINER DEFINITION
  75         Info("AliCFHeavyFlavourTaskMultiVarMultiStep","SETUP CONTAINER");
  76         //the sensitive variables, their indices
  77         UInt_t ipt = 0;
  78         UInt_t iy  = 1;
  79         UInt_t icosThetaStar  = 2;
  80         UInt_t ipTpi  = 3;
  81         UInt_t ipTk  = 4;
  82         UInt_t icT  = 5;
  83         UInt_t idca  = 6;
  84         UInt_t id0pi  = 7;
  85         UInt_t id0K  = 8;
  86         UInt_t id0xd0  = 9;
  87         UInt_t ipointing  = 10;
  88         UInt_t iphi  = 11;
  89         UInt_t iz  = 12;
  90
  91         const Double_t phimax = 2*TMath::Pi();
  92
  93         //Setting up the container grid...
  94         UInt_t nstep = 10; //number of selection steps: MC with limited acceptance, MC, Acceptance, Vertex, Refit, Reco (no cuts), RecoAcceptance, RecoITSClusters (RecoAcceptance included), RecoPPR (RecoAcceptance+RecoITSCluster included), RecoPID
  95         const Int_t nvar   = 13 ; //number of variables on the grid:pt, y, cosThetaStar, pTpi, pTk, cT, dca, d0pi, d0K, d0xd0, cosPointingAngle, phi
  96 //      const Int_t nbin0_0_4  = 8 ; //bins in pt from 0 to 4 GeV
  97 //      const Int_t nbin0_4_8  = 4 ; //bins in pt from 4 to 8 GeV
  98 //      const Int_t nbin0_8_10  = 1 ; //bins in pt from 8 to 10 GeV
  99
 100 /*
 101         Int_t nbin0_0_4;
 102         Int_t nbin0_4_8;
 103         Int_t nbin0_8_10;
 104         if (!isKeepD0fromB){
 105           nbin0_0_4  = 8 ; //bins in pt from 0 to 4 GeV
 106           nbin0_4_8  = 4 ; //bins in pt from 4 to 8 GeV
 107           nbin0_8_10  = 1 ; //bins in pt from 8 to 10 GeV
 108         }else{
 109           nbin0_0_4  = 3 ; //bins in pt from 0 to 3 GeV
 110           nbin0_4_8  = 1 ; //bins in pt from 3 to 5 GeV
 111           nbin0_8_10  = 1 ; //bins in pt from 5 to 10 GeV
 112         }
 113 */
 114         const Int_t nbin0 = cutsD0toKpi->GetNPtBins(); // bins in pT
 115         printf("pT: nbin (from cuts file) = %d\n",nbin0);
 116         const Int_t nbin1  = 42 ; //bins in y
 117         const Int_t nbin2  = 42 ; //bins in cosThetaStar
 118         const Int_t nbin3_0_4  = 8 ; //bins in ptPi from 0 to 4 GeV
 119         const Int_t nbin3_4_8  = 4 ; //bins in ptPi from 4 to 8 GeV
 120         const Int_t nbin3_8_10  = 1 ; //bins in ptPi from 8 to 10 GeV
 121         const Int_t nbin4_0_4  = 8 ; //bins in ptKa from 0 to 4 GeV
 122         const Int_t nbin4_4_8  = 4 ; //bins in ptKa from 4 to 8 GeV
 123         const Int_t nbin4_8_10  = 1 ; //bins in ptKa from 8 to 10 GeV
 124         const Int_t nbin5  = 24 ; //bins in cT
 125         const Int_t nbin6  = 24 ; //bins in dca
 126         const Int_t nbin7  = 100 ; //bins in d0pi
 127         const Int_t nbin8  = 100 ; //bins in d0K
 128         const Int_t nbin9  = 80 ; //bins in d0xd0
 129         const Int_t nbin10  = 1050 ; //bins in cosPointingAngle
 130         const Int_t nbin11  = 20 ; //bins in Phi
 131         const Int_t nbin12  = 60 ; //bins in z vertex
 132
 133         //arrays for the number of bins in each dimension
 134         Int_t iBin[nvar];
 135         //iBin[0]=nbin0_0_4+nbin0_4_8+nbin0_8_10;
 136         iBin[0]=nbin0;
 137         iBin[1]=nbin1;
 138         iBin[2]=nbin2;
 139         //      iBin[3]=nbin3_0_4+nbin3_4_8+nbin3_8_10;
 140         //iBin[4]=nbin4_0_4+nbin4_4_8+nbin4_8_10;
 141         iBin[3]=nbin0;
 142         iBin[4]=nbin0;
 143         iBin[5]=nbin5;
 144         iBin[6]=nbin6;
 145         iBin[7]=nbin7;
 146         iBin[8]=nbin8;
 147         iBin[9]=nbin9;
 148         iBin[10]=nbin10;
 149         iBin[11]=nbin11;
 150         iBin[12]=nbin12;
 151
 152         //arrays for lower bounds :
 153         Double_t *binLim0=new Double_t[iBin[0]+1];
 154         Double_t *binLim1=new Double_t[iBin[1]+1];
 155         Double_t *binLim2=new Double_t[iBin[2]+1];
 156         Double_t *binLim3=new Double_t[iBin[3]+1];
 157         Double_t *binLim4=new Double_t[iBin[4]+1];
 158         Double_t *binLim5=new Double_t[iBin[5]+1];
 159         Double_t *binLim6=new Double_t[iBin[6]+1];
 160         Double_t *binLim7=new Double_t[iBin[7]+1];
 161         Double_t *binLim8=new Double_t[iBin[8]+1];
 162         Double_t *binLim9=new Double_t[iBin[9]+1];
 163         Double_t *binLim10=new Double_t[iBin[10]+1];
 164         Double_t *binLim11=new Double_t[iBin[11]+1];
 165         Double_t *binLim12=new Double_t[iBin[12]+1];
 166
 167         // checking limits
 168         /*
 169         if (ptmax_0_4 != ptmin_4_8) {
 170                 Error("AliCFHeavyFlavourTaskMultiVarMultiStep","max lim 1st range != min lim 2nd range, please check!");
 171         }
 172         if (ptmax_4_8 != ptmin_8_10) {
 173                 Error("AliCFHeavyFlavourTaskMultiVarMultiStep","max lim 2nd range != min lim 3rd range, please check!");
 174         }
 175         */
 176         // values for bin lower bounds
 177         // pt
 178         Float_t* floatbinLim0 = cutsD0toKpi->GetPtBinLimits();
 179         for (Int_t ibin0 = 0 ; ibin0<iBin[0]+1; ibin0++){
 180                 binLim0[ibin0] = (Double_t)floatbinLim0[ibin0];
 181                 binLim3[ibin0] = (Double_t)floatbinLim0[ibin0];
 182                 binLim4[ibin0] = (Double_t)floatbinLim0[ibin0];
 183         }
 184         for(Int_t i=0; i<=nbin0; i++) printf("binLim0[%d]=%f\n",i,binLim0[i]);
 185
 186         /*
 187         for(Int_t i=0; i<=nbin0_0_4; i++) binLim0[i]=(Double_t)ptmin_0_4 + (ptmax_0_4-ptmin_0_4)/nbin0_0_4*(Double_t)i ;
 188         if (binLim0[nbin0_0_4] != ptmin_4_8)  {
 189                 Error("AliCFHeavyFlavourTaskMultiVarMultiStep","Calculated bin lim for pt - 1st range - differs from expected!\n");
 190         }
 191         for(Int_t i=0; i<=nbin0_4_8; i++) binLim0[i+nbin0_0_4]=(Double_t)ptmin_4_8 + (ptmax_4_8-ptmin_4_8)/nbin0_4_8*(Double_t)i ;
 192         if (binLim0[nbin0_0_4+nbin0_4_8] != ptmin_8_10)  {
 193                 Error("AliCFHeavyFlavourTaskMultiVarMultiStep","Calculated bin lim for pt - 2nd range - differs from expected!\n");
 194         }
 195         for(Int_t i=0; i<=nbin0_8_10; i++) binLim0[i+nbin0_0_4+nbin0_4_8]=(Double_t)ptmin_8_10 + (ptmax_8_10-ptmin_8_10)/nbin0_8_10*(Double_t)i ;
 196         */
 197
 198         // y
 199         for(Int_t i=0; i<=nbin1; i++) binLim1[i]=(Double_t)ymin  + (ymax-ymin)  /nbin1*(Double_t)i ;
 200
 201         // cosThetaStar
 202         for(Int_t i=0; i<=nbin2; i++) binLim2[i]=(Double_t)cosmin  + (cosmax-cosmin)  /nbin2*(Double_t)i ;
 203
 204         /*
 205         // ptPi
 206         for(Int_t i=0; i<=nbin3_0_4; i++) binLim3[i]=(Double_t)ptmin_0_4 + (ptmax_0_4-ptmin_0_4)/nbin3_0_4*(Double_t)i ;
 207         if (binLim3[nbin3_0_4] != ptmin_4_8)  {
 208                 Error("AliCFHeavyFlavourTaskMultiVarMultiStep","Calculated bin lim for ptPi - 1st range - differs from expected!");
 209         }
 210         for(Int_t i=0; i<=nbin3_4_8; i++) binLim3[i+nbin3_0_4]=(Double_t)ptmin_4_8 + (ptmax_4_8-ptmin_4_8)/nbin3_4_8*(Double_t)i ;
 211         if (binLim3[nbin3_0_4+nbin3_4_8] != ptmin_8_10)  {
 212                 Error("AliCFHeavyFlavourTaskMultiVarMultiStep","Calculated bin lim for ptPi - 2nd range - differs from expected!\n");
 213         }
 214         for(Int_t i=0; i<=nbin3_8_10; i++) binLim3[i+nbin3_0_4+nbin3_4_8]=(Double_t)ptmin_8_10 + (ptmax_8_10-ptmin_8_10)/nbin3_8_10*(Double_t)i ;
 215
 216         // ptKa
 217         for(Int_t i=0; i<=nbin4_0_4; i++) binLim4[i]=(Double_t)ptmin_0_4 + (ptmax_0_4-ptmin_0_4)/nbin4_0_4*(Double_t)i ;
 218         if (binLim4[nbin4_0_4] != ptmin_4_8)  {
 219                 Error("AliCFHeavyFlavourTaskMultiVarMultiStep","Calculated bin lim for ptKa - 1st range - differs from expected!");
 220         }
 221         for(Int_t i=0; i<=nbin4_4_8; i++) binLim4[i+nbin4_0_4]=(Double_t)ptmin_4_8 + (ptmax_4_8-ptmin_4_8)/nbin4_4_8*(Double_t)i ;
 222         if (binLim4[nbin4_0_4+nbin4_4_8] != ptmin_8_10)  {
 223                 Error("AliCFHeavyFlavourTaskMultiVarMultiStep","Calculated bin lim for ptKa - 2nd range - differs from expected!\n");
 224         }
 225         for(Int_t i=0; i<=nbin4_8_10; i++) binLim4[i+nbin4_0_4+nbin4_4_8]=(Double_t)ptmin_8_10 + (ptmax_8_10-ptmin_8_10)/nbin4_8_10*(Double_t)i ;
 226         */
 227         // cT
 228         for(Int_t i=0; i<=nbin5; i++) binLim5[i]=(Double_t)cTmin  + (cTmax-cTmin)  /nbin5*(Double_t)i ;
 229
 230         // dca
 231         for(Int_t i=0; i<=nbin6; i++) binLim6[i]=(Double_t)dcamin  + (dcamax-dcamin)  /nbin6*(Double_t)i ;
 232
 233         // d0pi
 234         for(Int_t i=0; i<=nbin7; i++) binLim7[i]=(Double_t)d0min  + (d0max-d0min)  /nbin7*(Double_t)i ;
 235
 236         // d0K
 237         for(Int_t i=0; i<=nbin8; i++) binLim8[i]=(Double_t)d0min  + (d0max-d0min)  /nbin8*(Double_t)i ;
 238
 239         // d0xd0
 240         for(Int_t i=0; i<=nbin9; i++) binLim9[i]=(Double_t)d0xd0min  + (d0xd0max-d0xd0min)  /nbin9*(Double_t)i ;
 241
 242         // cosPointingAngle
 243         for(Int_t i=0; i<=nbin10; i++) binLim10[i]=(Double_t)cosmin  + (cosmax-cosmin)  /nbin10*(Double_t)i ;
 244
 245         // Phi
 246         for(Int_t i=0; i<=nbin11; i++) binLim11[i]=(Double_t)phimin  + (phimax-phimin)  /nbin11*(Double_t)i ;
 247
 248         // z Primary Vertex
 249         for(Int_t i=0; i<=nbin12; i++) {
 250                 binLim12[i]=(Double_t)zmin  + (zmax-zmin)  /nbin12*(Double_t)i ;
 251                 //              Info("AliCFHeavyFlavourTaskMultiVarMultiStep",Form("i-th bin, lower limit = %f", binLim12[i]));
 252         }
 253
 254         // debugging printings
 255         //Info("AliCFHeavyFlavourTaskMultiVarMultiStep","Printing lower limits for bins in pt");
 256         //for (Int_t i =0; i<= iBin[0]; i++){
 257         //      Info("AliCFHeavyFlavourTaskMultiVarMultiStep",Form("i-th bin, lower limit = %f", binLim0[i]));
 258         //}
 259         //Info("Printing lower limits for bins in ptPi");
 260         //for (Int_t i =0; i<= iBin[3]; i++){
 261         //      Info("AliCFHeavyFlavourTaskMultiVarMultiStep",Form("i-th bin, lower limit = %f", binLim3[i]));
 262         //}
 263         //Info("Printing lower limits for bins in ptKa");
 264         //for (Int_t i =0; i<= iBin[4]; i++){
 265         //      Info("AliCFHeavyFlavourTaskMultiVarMultiStep",Form("i-th bin, lower limit = %f", binLim4[i]));
 266         //      }
 267
 268         //one "container" for MC
 269         TString nameContainer="";
 270         if(!isKeepD0fromB) {
 271           nameContainer="CFHFccontainer0";
 272         }
 273         else if(isKeepD0fromBOnly){
 274           nameContainer="CFHFccontainer0D0fromB";
 275         }
 276         else  {
 277           nameContainer="CFHFccontainer0allD0";
 278
 279         }
 280
 281         AliCFContainer* container = new AliCFContainer(nameContainer,"container for tracks",nstep,nvar,iBin);
 282         //setting the bin limits
 283         printf("pt\n");
 284         container -> SetBinLimits(ipt,binLim0);
 285         printf("y\n");
 286         container -> SetBinLimits(iy,binLim1);
 287         printf("cts\n");
 288         container -> SetBinLimits(icosThetaStar,binLim2);
 289         printf("ptPi\n");
 290         container -> SetBinLimits(ipTpi,binLim3);
 291         printf("ptK\n");
 292         container -> SetBinLimits(ipTk,binLim4);
 293         printf("cT\n");
 294         container -> SetBinLimits(icT,binLim5);
 295         printf("dca\n");
 296         container -> SetBinLimits(idca,binLim6);
 297         printf("d0Pi\n");
 298         container -> SetBinLimits(id0pi,binLim7);
 299         printf("d0K\n");
 300         container -> SetBinLimits(id0K,binLim8);
 301         printf("d0xd0\n");
 302         container -> SetBinLimits(id0xd0,binLim9);
 303         printf("pointing\n");
 304         container -> SetBinLimits(ipointing,binLim10);
 305         printf("phi\n");
 306         container -> SetBinLimits(iphi,binLim11);
 307         printf("z\n");
 308         container -> SetBinLimits(iz,binLim12);
 309
 310         container -> SetStepTitle(0, "MCLimAcc");
 311         container -> SetStepTitle(1, "MC");
 312         container -> SetStepTitle(2, "MCAcc");
 313         container -> SetStepTitle(3, "RecoVertex");
 314         container -> SetStepTitle(4, "RecoRefit");
 315         container -> SetStepTitle(5, "Reco");
 316         container -> SetStepTitle(6, "RecoAcc");
 317         container -> SetStepTitle(7, "RecoITSCluster");
 318         container -> SetStepTitle(8, "RecoCuts");
 319         container -> SetStepTitle(9, "RecoPID");
 320
 321         container -> SetVarTitle(ipt,"pt");
 322         container -> SetVarTitle(iy,"y");
 323         container -> SetVarTitle(icosThetaStar, "cosThetaStar");
 324         container -> SetVarTitle(ipTpi, "ptpi");
 325         container -> SetVarTitle(ipTk, "ptK");
 326         container -> SetVarTitle(icT, "ct");
 327         container -> SetVarTitle(idca, "dca");
 328         container -> SetVarTitle(id0pi, "d0pi");
 329         container -> SetVarTitle(id0K, "d0K");
 330         container -> SetVarTitle(id0xd0, "d0xd0");
 331         container -> SetVarTitle(ipointing, "piointing");
 332         container -> SetVarTitle(iphi, "phi");
 333         container -> SetVarTitle(iz, "z");
 334
 335
 336         //CREATE THE  CUTS -----------------------------------------------
 337
 338         // Gen-Level kinematic cuts
 339         AliCFTrackKineCuts *mcKineCuts = new AliCFTrackKineCuts("mcKineCuts","MC-level kinematic cuts");
 340
 341         //Particle-Level cuts:
 342         AliCFParticleGenCuts* mcGenCuts = new AliCFParticleGenCuts("mcGenCuts","MC particle generation cuts");
 343         mcGenCuts->SetRequirePdgCode(PDG, kTRUE);  // kTRUE set in order to include D0_bar
 344         mcGenCuts->SetAODMC(1); //special flag for reading MC in AOD tree (important)
 345
 346         // Acceptance cuts:
 347         AliCFAcceptanceCuts* accCuts = new AliCFAcceptanceCuts("accCuts", "Acceptance cuts");
 348         AliCFTrackKineCuts *kineAccCuts = new AliCFTrackKineCuts("kineAccCuts","Kine-Acceptance cuts");
 349         kineAccCuts->SetPtRange(ptmin,ptmax);
 350         kineAccCuts->SetEtaRange(etamin,etamax);
 351
 352         // Rec-Level kinematic cuts
 353         AliCFTrackKineCuts *recKineCuts = new AliCFTrackKineCuts("recKineCuts","rec-level kine cuts");
 354
 355         AliCFTrackQualityCuts *recQualityCuts = new AliCFTrackQualityCuts("recQualityCuts","rec-level quality cuts");
 356
 357         AliCFTrackIsPrimaryCuts *recIsPrimaryCuts = new AliCFTrackIsPrimaryCuts("recIsPrimaryCuts","rec-level isPrimary cuts");
 358
 359         printf("CREATE MC KINE CUTS\n");
 360         TObjArray* mcList = new TObjArray(0) ;
 361         mcList->AddLast(mcKineCuts);
 362         mcList->AddLast(mcGenCuts);
 363
 364         printf("CREATE ACCEPTANCE CUTS\n");
 365         TObjArray* accList = new TObjArray(0) ;
 366         accList->AddLast(kineAccCuts);
 367
 368         printf("CREATE RECONSTRUCTION CUTS\n");
 369         TObjArray* recList = new TObjArray(0) ;   // not used!!
 370         recList->AddLast(recKineCuts);
 371         recList->AddLast(recQualityCuts);
 372         recList->AddLast(recIsPrimaryCuts);
 373
 374         TObjArray* emptyList = new TObjArray(0);
 375
 376         //CREATE THE INTERFACE TO CORRECTION FRAMEWORK USED IN THE TASK
 377         printf("CREATE INTERFACE AND CUTS\n");
 378         AliCFManager* man = new AliCFManager() ;
 379         man->SetParticleContainer     (container);
 380         man->SetParticleCutsList(0 , mcList); // MC, Limited Acceptance
 381         man->SetParticleCutsList(1 , mcList); // MC
 382         man->SetParticleCutsList(2 , accList); // Acceptance
 383         man->SetParticleCutsList(3 , emptyList); // Vertex
 384         man->SetParticleCutsList(4 , emptyList); // Refit
 385         man->SetParticleCutsList(5 , emptyList); // AOD
 386         man->SetParticleCutsList(6 , emptyList); // AOD in Acceptance
 387         man->SetParticleCutsList(7 , emptyList); // AOD with required n. of ITS clusters
 388         man->SetParticleCutsList(8 , emptyList); // AOD Reco (PPR cuts implemented in Task)
 389         man->SetParticleCutsList(9 , emptyList); // AOD Reco PID
 390
 391         // Get the pointer to the existing analysis manager via the static access method.
 392         //==============================================================================
 393         AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
 394         if (!mgr) {
 395           ::Error("AddTaskCompareHF", "No analysis manager to connect to.");
 396           return NULL;
 397         }
 398         //CREATE THE TASK
 399         printf("CREATE TASK\n");
 400         // create the task
 401         AliCFHeavyFlavourTaskMultiVarMultiStep *task = new AliCFHeavyFlavourTaskMultiVarMultiStep("AliCFHeavyFlavourTaskMultiVarMultiStep",cutsD0toKpi);
 402         task->SetFillFromGenerated(kFALSE);
 403         task->SetMinITSClusters(minITSClusters);
 404         task->SetCFManager(man); //here is set the CF manager
 405         task->SetKeepD0fromB(isKeepD0fromB);
 406         task->SetKeepD0fromBOnly(isKeepD0fromBOnly);
 407
 408         //-----------------------------------------------------------//
 409         //   create correlation matrix for unfolding - only eta-pt   //
 410         //-----------------------------------------------------------//
 411
 412         Bool_t AcceptanceUnf = kTRUE; // unfold at acceptance level, otherwise PPR
 413
 414         Int_t thnDim[4];
 415
 416         //first half  : reconstructed
 417         //second half : MC
 418
 419         thnDim[0] = iBin[0];
 420         thnDim[2] = iBin[0];
 421         thnDim[1] = iBin[1];
 422         thnDim[3] = iBin[1];
 423
 424         TString nameCorr="";
 425         if(!isKeepD0fromB) {
 426           nameCorr="CFHFcorr0";
 427         }
 428         else if(isKeepD0fromBOnly){
 429           nameCorr= "CFHFcorr0KeepD0fromBOnly";
 430         }
 431         else  {
 432           nameCorr="CFHFcorr0allD0";
 433
 434         }
 435
 436         THnSparseD* correlation = new THnSparseD(nameCorr,"THnSparse with correlations",4,thnDim);
 437         Double_t** binEdges = new Double_t[2];
 438
 439         // set bin limits
 440
 441         binEdges[0]= binLim0;
 442         binEdges[1]= binLim1;
 443
 444         correlation->SetBinEdges(0,binEdges[0]);
 445         correlation->SetBinEdges(2,binEdges[0]);
 446
 447         correlation->SetBinEdges(1,binEdges[1]);
 448         correlation->SetBinEdges(3,binEdges[1]);
 449
 450         correlation->Sumw2();
 451
 452         // correlation matrix ready
 453         //------------------------------------------------//
 454
 455         task->SetCorrelationMatrix(correlation); // correlation matrix for unfolding
 456
 457         // Create and connect containers for input/output
 458
 459         // ------ input data ------
 460         AliAnalysisDataContainer *cinput0  = mgr->GetCommonInputContainer();
 461
 462         // ----- output data -----
 463
 464         TString outputfile = AliAnalysisManager::GetCommonFileName();
 465         TString output1name="", output2name="", output3name="";
 466         output2name=nameContainer;
 467         output3name=nameCorr;
 468         if(!isKeepD0fromB) {
 469           outputfile += ":PWG3_D2H_CFtaskD0toKpi";
 470           output1name="CFHFchist0";
 471         }
 472         else if(isKeepD0fromBOnly){
 473           outputfile += ":PWG3_D2H_CFtaskD0toKpiKeepD0fromBOnly";
 474           output1name="CFHFchist0D0fromB";
 475         }
 476         else  {
 477           outputfile += ":PWG3_D2H_CFtaskD0toKpiKeepD0fromB";
 478           output1name="CFHFchist0allD0";
 479
 480         }
 481
 482         //now comes user's output objects :
 483         // output TH1I for event counting
 484         AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(output1name, TH1I::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());
 485         // output Correction Framework Container (for acceptance & efficiency calculations)
 486         AliAnalysisDataContainer *coutput2 = mgr->CreateContainer(output2name, AliCFContainer::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());
 487         // Unfolding - correlation matrix
 488         AliAnalysisDataContainer *coutput3 = mgr->CreateContainer(output3name, THnSparseD::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());
 489
 490         mgr->AddTask(task);
 491
 492         mgr->ConnectInput(task,0,mgr->GetCommonInputContainer());
 493         mgr->ConnectOutput(task,1,coutput1);
 494         mgr->ConnectOutput(task,2,coutput2);
 495         mgr->ConnectOutput(task,3,coutput3);
 496         return task;
 497 }
 498