PWG2/FEMTOSCOPY/macros/Train/PionsFemtoscopy/ConfigFemtoAnalysis.C

   1 /*********************************************************************
   2  *                                                                   *
   3  * Configfemtoanalysis.C - configuration macro for the femtoscopic   *
   4  * analysis, meant as a QA process for two-particle effects          *
   5  *                                                                   *
   6  * Author: Adam Kisiel (Adam.Kisiel@cern.ch)                         *
   7  *                                                                   *
   8  *********************************************************************/
   9
  10 #if !defined(__CINT__) || defined(__MAKECINT_)
  11 #include "AliFemtoManager.h"
  12 #include "AliFemtoEventReaderESDChain.h"
  13 #include "AliFemtoEventReaderESDChainKine.h"
  14 #include "AliFemtoEventReaderAODChain.h"
  15 #include "AliFemtoSimpleAnalysis.h"
  16 #include "AliFemtoBasicEventCut.h"
  17 #include "AliFemtoESDTrackCut.h"
  18 #include "AliFemtoCorrFctn.h"
  19 #include "AliFemtoCutMonitorParticleYPt.h"
  20 #include "AliFemtoCutMonitorParticleVertPos.h"
  21 #include "AliFemtoCutMonitorParticleMomRes.h"
  22 #include "AliFemtoCutMonitorParticlePID.h"
  23 #include "AliFemtoCutMonitorEventMult.h"
  24 #include "AliFemtoCutMonitorEventVertex.h"
  25 #include "AliFemtoShareQualityTPCEntranceSepPairCut.h"
  26 #include "AliFemtoPairCutAntiGamma.h"
  27 #include "AliFemtoPairCutRadialDistance.h"
  28 #include "AliFemtoQinvCorrFctn.h"
  29 #include "AliFemtoCorrFctnNonIdDR.h"
  30 #include "AliFemtoShareQualityCorrFctn.h"
  31 #include "AliFemtoTPCInnerCorrFctn.h"
  32 #include "AliFemtoVertexMultAnalysis.h"
  33 #include "AliFemtoCorrFctn3DSpherical.h"
  34 #include "AliFemtoChi2CorrFctn.h"
  35 #include "AliFemtoCorrFctnTPCNcls.h"
  36 #include "AliFemtoBPLCMS3DCorrFctn.h"
  37 #include "AliFemtoCorrFctn3DLCMSSym.h"
  38 #include "AliFemtoModelBPLCMSCorrFctn.h"
  39 #include "AliFemtoModelCorrFctn3DSpherical.h"
  40 #include "AliFemtoModelGausLCMSFreezeOutGenerator.h"
  41 #include "AliFemtoModelGausRinvFreezeOutGenerator.h"
  42 #include "AliFemtoModelManager.h"
  43 #include "AliFemtoModelWeightGeneratorBasic.h"
  44 #include "AliFemtoModelWeightGeneratorLednicky.h"
  45 #include "AliFemtoCorrFctnDirectYlm.h"
  46 #include "AliFemtoModelCorrFctnDirectYlm.h"
  47 #include "AliFemtoModelCorrFctnSource.h"
  48 #include "AliFemtoCutMonitorParticlePtPDG.h"
  49 #include "AliFemtoKTPairCut.h"
  50 #endif
  51
  52 //________________________________________________________________________
  53 AliFemtoManager* ConfigFemtoAnalysis() {
  54
  55   double PionMass = 0.13956995;
  56   double KaonMass = 0.493677;
  57   double ProtonMass = 0.938272013;
  58
  59
  60   int runmults[10] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1};
  61   int multbins[11] = {0.001, 50, 100, 200, 300, 400, 500, 600, 700, 800, 900};
  62
  63   int runch[3] = {1, 1, 1};
  64   const char *chrgs[3] = { "pip", "pim", "pimpip" };
  65
  66
  67   int runktdep = 1;
  68   double ktrng[3] = {0.01, 0.7, 100.0};
  69   //  double ktrng[8] = {0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 1.0};
  70
  71   int numOfMultBins = 10;
  72   int numOfChTypes = 3;
  73   int numOfkTbins = 2;
  74
  75   int runqinv = 1;
  76   int runshlcms = 1;// 0:PRF(PAP), 1:LCMS(PP,APAP)
  77
  78   int runtype = 2; // Types 0 - global, 1 - ITS only, 2 - TPC Inner
  79   int isrealdata = 1;
  80
  81   int gammacut = 1;
  82
  83   double shqmax = 1.2;
  84   int nbinssh = 100;
  85
  86   AliFemtoEventReaderAODChain *Reader = new AliFemtoEventReaderAODChain();
  87   Reader->SetFilterBit(7);
  88   Reader->SetCentralityPreSelection(0.001, 910);
  89
  90   AliFemtoManager* Manager = new AliFemtoManager();
  91   Manager->SetEventReader(Reader);
  92
  93   AliFemtoVertexMultAnalysis    *anetaphitpc[10*3];
  94   AliFemtoBasicEventCut         *mecetaphitpc[10*3];
  95   AliFemtoCutMonitorEventMult   *cutPassEvMetaphitpc[10*3];
  96   AliFemtoCutMonitorEventMult   *cutFailEvMetaphitpc[10*3];
  97   AliFemtoCutMonitorEventVertex *cutPassEvVetaphitpc[10*3];
  98   AliFemtoCutMonitorEventVertex *cutFailEvVetaphitpc[10*3];
  99   AliFemtoESDTrackCut           *dtc1etaphitpc[10*3];
 100   AliFemtoESDTrackCut           *dtc2etaphitpc[10*3];
 101   AliFemtoCutMonitorParticleYPt *cutPass1YPtetaphitpc[10*3];
 102   AliFemtoCutMonitorParticleYPt *cutFail1YPtetaphitpc[10*3];
 103   AliFemtoCutMonitorParticlePID *cutPass1PIDetaphitpc[10*3];
 104   AliFemtoCutMonitorParticlePID *cutFail1PIDetaphitpc[10*3];
 105   AliFemtoCutMonitorParticleYPt *cutPass2YPtetaphitpc[10*3];
 106   AliFemtoCutMonitorParticleYPt *cutFail2YPtetaphitpc[10*3];
 107   AliFemtoCutMonitorParticlePID *cutPass2PIDetaphitpc[10*3];
 108   AliFemtoCutMonitorParticlePID *cutFail2PIDetaphitpc[10*3];
 109   //   AliFemtoPairCutAntiGamma      *sqpcetaphitpcdiff[10*3];
 110   //   AliFemtoShareQualityTPCEntranceSepPairCut      *sqpcetaphitpcsame[10*3];
 111   AliFemtoPairCutAntiGamma      *sqpcetaphitpc[10*3];
 112   //  AliFemtoPairCutRadialDistance      *sqpcetaphitpc[10*3];
 113   //  AliFemtoChi2CorrFctn          *cchiqinvetaphitpc[20*2];
 114   AliFemtoKTPairCut             *ktpcuts[10*3*2];
 115   AliFemtoCorrFctnDirectYlm     *cylmkttpc[10*3];
 116   //AliFemtoCorrFctnDirectYlm     *cylmetaphitpc[10*3];
 117   AliFemtoQinvCorrFctn          *cqinvkttpc[10*3*2];
 118   AliFemtoQinvCorrFctn          *cqinvtpc[10*3];
 119   AliFemtoCorrFctnNonIdDR       *ckstartpc[10*3];
 120   AliFemtoCorrFctnDEtaDPhi      *cdedpetaphi[10*3];
 121
 122   //   AliFemtoCorrFctn3DLCMSSym     *cq3dlcmskttpc[20*2];
 123   //   AliFemtoCorrFctnTPCNcls       *cqinvnclstpc[20];
 124   //   AliFemtoShareQualityCorrFctn  *cqinvsqtpc[20*10];
 125   //   AliFemtoChi2CorrFctn          *cqinvchi2tpc[20];
 126   //   AliFemtoTPCInnerCorrFctn      *cqinvinnertpc[20*10];
 127
 128   // *** Third QA task - HBT analysis with all pair cuts off, TPC only ***
 129   // *** Begin pion-pion (positive) analysis ***
 130   int aniter = 0;
 131
 132   for (int imult = 0; imult < numOfMultBins; imult++) {
 133     if (runmults[imult]) {
 134
 135       for (int ichg = 0; ichg < numOfChTypes; ichg++) {
 136         if (runch[ichg]) {
 137
 138           aniter = ichg * numOfMultBins + imult;
 139           anetaphitpc[aniter] = new AliFemtoVertexMultAnalysis(8, -8.0, 8.0, 4, multbins[imult], multbins[imult+1]);
 140           anetaphitpc[aniter]->SetNumEventsToMix(10);
 141           anetaphitpc[aniter]->SetMinSizePartCollection(1);
 142
 143           mecetaphitpc[aniter] = new AliFemtoBasicEventCut();
 144           mecetaphitpc[aniter]->SetEventMult(0.001,100000);
 145           mecetaphitpc[aniter]->SetVertZPos(-8,8);
 146
 147           if (isrealdata)
 148             mecetaphitpc[aniter]->SetAcceptOnlyPhysics(kTRUE);
 149
 150           cutPassEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutPass%stpcM%i", chrgs[ichg], imult));
 151           cutFailEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutFail%stpcM%i", chrgs[ichg], imult));
 152           mecetaphitpc[aniter]->AddCutMonitor(cutPassEvMetaphitpc[aniter], cutFailEvMetaphitpc[aniter]);
 153
 154           cutPassEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutPass%stpcM%i", chrgs[ichg], imult));
 155           cutFailEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutFail%stpcM%i", chrgs[ichg], imult));
 156           mecetaphitpc[aniter]->AddCutMonitor(cutPassEvVetaphitpc[aniter], cutFailEvVetaphitpc[aniter]);
 157
 158           dtc1etaphitpc[aniter] = new AliFemtoESDTrackCut();
 159           dtc2etaphitpc[aniter] = new AliFemtoESDTrackCut();
 160
 161           if (ichg == 0)
 162             dtc1etaphitpc[aniter]->SetCharge(1.0);
 163           else if (ichg == 1)
 164             dtc1etaphitpc[aniter]->SetCharge(-1.0);
 165           else if (ichg == 2) {
 166             dtc1etaphitpc[aniter]->SetCharge(-1.0);
 167             dtc2etaphitpc[aniter]->SetCharge(1.0);
 168           }
 169
 170           dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
 171           dtc1etaphitpc[aniter]->SetMass(PionMass);
 172           dtc1etaphitpc[aniter]->SetMostProbablePion();
 173
 174           if (ichg == 2) {
 175             dtc2etaphitpc[aniter]->SetEta(-0.8,0.8);
 176             dtc2etaphitpc[aniter]->SetMass(PionMass);
 177             dtc2etaphitpc[aniter]->SetMostProbablePion();
 178           }
 179
 180           // Track quality cuts
 181
 182           if (runtype == 0) {
 183             dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
 184             //      dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit);
 185             //    dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit);
 186             dtc1etaphitpc[aniter]->SetminTPCncls(80);
 187             dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 188             dtc1etaphitpc[aniter]->SetLabel(kFALSE);
 189             //    dtc1etaphitpc[aniter]->SetMaxITSChiNdof(6.0);
 190             dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 191             dtc1etaphitpc[aniter]->SetMaxImpactXY(0.2);
 192             //            dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
 193             dtc1etaphitpc[aniter]->SetMaxImpactZ(0.15);
 194             //      dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
 195           }
 196           else if (runtype == 1) {
 197             //      dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
 198             //    dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit);
 199             //      dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit|AliESDtrack::kITSpureSA);
 200             //      dtc1etaphitpc[aniter]->SetminTPCncls(70);
 201             dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit);
 202             dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 203             dtc1etaphitpc[aniter]->SetLabel(kFALSE);
 204             //    dtc1etaphitpc[aniter]->SetMaxITSChiNdof(6.0);
 205             //      dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(6.0);
 206             dtc1etaphitpc[aniter]->SetMaxImpactXY(0.2);
 207             dtc1etaphitpc[aniter]->SetMaxImpactZ(0.25);
 208             //      dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
 209           }
 210           else if (runtype == 2) {
 211             dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin);
 212             dtc1etaphitpc[aniter]->SetminTPCncls(80);
 213             dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 214             dtc1etaphitpc[aniter]->SetLabel(kFALSE);
 215             dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 216             dtc1etaphitpc[aniter]->SetMaxImpactXY(2.4);
 217             dtc1etaphitpc[aniter]->SetMaxImpactZ(3.0);
 218
 219             if (ichg == 2) {
 220               dtc2etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin);
 221               dtc2etaphitpc[aniter]->SetminTPCncls(80);
 222               dtc2etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 223               dtc2etaphitpc[aniter]->SetLabel(kFALSE);
 224               dtc2etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 225               dtc2etaphitpc[aniter]->SetMaxImpactXY(0.25);
 226               dtc2etaphitpc[aniter]->SetMaxImpactZ(0.15);
 227             }
 228
 229           }
 230
 231           cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult),PionMass);
 232           cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult),PionMass);
 233           dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
 234
 235           cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),2);//0-pion,1-kaon,2-proton
 236           cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),2);
 237           dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
 238
 239           if (ichg == 2){
 240             cutPass2PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass2%stpcM%i", chrgs[ichg], imult),2);//0-pion,1-kaon,2-proton
 241             cutFail2PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail2%stpcM%i", chrgs[ichg], imult),2);
 242             dtc2etaphitpc[aniter]->AddCutMonitor(cutPass2PIDetaphitpc[aniter], cutFail2PIDetaphitpc[aniter]);
 243           }
 244
 245           sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
 246
 247           if (runtype == 0) {
 248             sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
 249             sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
 250             sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
 251             // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
 252             // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
 253             //      sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
 254             //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
 255             //      sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
 256           }
 257           else if (runtype == 1) {
 258             sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
 259             sqpcetaphitpc[aniter]->SetShareFractionMax(1.05);
 260             sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
 261             // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.002);
 262             // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.008);
 263             //      sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
 264             //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
 265             //      sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
 266           }
 267           else if (runtype == 2) {
 268             sqpcetaphitpc[aniter]->SetUseAOD(kTRUE);
 269             sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
 270             sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
 271             sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
 272
 273             if (gammacut == 0) {
 274               sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
 275               sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
 276             }
 277             else if (gammacut == 1) {
 278               sqpcetaphitpc[aniter]->SetMaxEEMinv(0.002);
 279               sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.008);
 280             }
 281
 282             //phi-star cut - values from Johana
 283             //      sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.012);
 284             //      sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.017);
 285
 286           }
 287
 288           anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
 289
 290           if (ichg == 2) {
 291             anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
 292             anetaphitpc[aniter]->SetSecondParticleCut(dtc2etaphitpc[aniter]);
 293           }
 294           else {
 295             anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
 296             anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
 297           }
 298
 299           anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
 300
 301           if (ichg == 2) {
 302             ckstartpc[aniter] = new AliFemtoCorrFctnNonIdDR(Form("ckstar%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 303             anetaphitpc[aniter]->AddCorrFctn(ckstartpc[aniter]);
 304           }
 305           else {
 306             cqinvtpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 307             anetaphitpc[aniter]->AddCorrFctn(cqinvtpc[aniter]);
 308           }
 309
 310           cylmkttpc[aniter] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%i", chrgs[ichg], imult),3,nbinssh, 0.0,shqmax,runshlcms);
 311           anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[aniter]);
 312
 313           if (runktdep) {
 314             int ktm;
 315             for (int ikt=0; ikt<numOfkTbins; ikt++) {
 316
 317               ktm = aniter * numOfkTbins + ikt;
 318               ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
 319
 320               cylmkttpc[ktm] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%ikT%i", chrgs[ichg], imult, ikt),3,nbinssh,0.0,shqmax,0);
 321               cylmkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 322               anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[ktm]);
 323
 324               cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,(imult>6)?shqmax*2.5:shqmax);
 325               cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 326               anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);
 327
 328               //              cqinvsqtpc[ktm] = new AliFemtoShareQualityCorrFctn(Form("cqinvsq%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
 329               //              cqinvsqtpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 330               //              anetaphitpc[aniter]->AddCorrFctn(cqinvsqtpc[ktm]);
 331
 332               //              cqinvinnertpc[ktm] = new AliFemtoTPCInnerCorrFctn(Form("cqinvinner%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
 333               //              cqinvinnertpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 334               //              cqinvinnertpc[ktm]->SetRadius(1.2);
 335               //              anetaphitpc[aniter]->AddCorrFctn(cqinvinnertpc[ktm]);
 336
 337               //              if (run3d) {
 338               //                cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),60,(imult>3)?((imult>6)?((imult>7)?0.6:0.4):0.25):0.15);
 339               //                cq3dlcmskttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 340               //                anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[ktm]);
 341               //              }
 342             }
 343           }
 344
 345          // cdedpetaphi[aniter] = new AliFemtoCorrFctnDEtaDPhi(Form("cdedp%stpcM%i", chrgs[ichg], imult),24, 24);
 346          // anetaphitpc[aniter]->AddCorrFctn(cdedpetaphi[aniter]);
 347
 348           Manager->AddAnalysis(anetaphitpc[aniter]);
 349         }
 350       }
 351     }
 352   }
 353   // *** End pion-pion (positive) analysis
 354
 355   return Manager;
 356 }