PWGCF/FEMTOSCOPY/macros/Train/FemtoQA/pp_MC/ConfigFemtoAnalysis.C

   1
   2 /*********************************************************************
   3  *                                                                   *
   4  * ConfigFemtoAnalysis.C - configuration macro for the femtoscopic   *
   5  * analysis, meant as a QA process for two-particle effects          *
   6  *                                                                   *
   7  * Author: Adam Kisiel (Adam.Kisiel@cern.ch)                         *
   8  *                                                                   *
   9  *********************************************************************/
  10
  11 #if !defined(__CINT__) || defined(__MAKECINT_)
  12 #include "AliFemtoManager.h"
  13 #include "AliFemtoEventReaderESDChain.h"
  14 #include "AliFemtoEventReaderESDChainKine.h"
  15 #include "AliFemtoEventReaderAODChain.h"
  16 #include "AliFemtoSimpleAnalysis.h"
  17 #include "AliFemtoBasicEventCut.h"
  18 #include "AliFemtoESDTrackCut.h"
  19 #include "AliFemtoCorrFctn.h"
  20 #include "AliFemtoCutMonitorParticleYPt.h"
  21 #include "AliFemtoCutMonitorParticleVertPos.h"
  22 #include "AliFemtoCutMonitorParticleMomRes.h"
  23 #include "AliFemtoCutMonitorParticlePID.h"
  24 #include "AliFemtoCutMonitorEventMult.h"
  25 #include "AliFemtoCutMonitorEventVertex.h"
  26 #include "AliFemtoShareQualityTPCEntranceSepPairCut.h"
  27 #include "AliFemtoPairCutAntiGamma.h"
  28 #include "AliFemtoPairCutRadialDistance.h"
  29 #include "AliFemtoQinvCorrFctn.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 #include "AliFemtoCutMonitorCollections.h"
  51 #endif
  52
  53 //________________________________________________________________________
  54 AliFemtoManager* ConfigFemtoAnalysis() {
  55
  56   double PionMass = 0.13956995;
  57   double KaonMass = 0.493677;
  58
  59   //multiplicity bins
  60   const int numOfMultBins = 5;
  61         int runmults[numOfMultBins] = {1, 0, 0, 0, 0};
  62         int multbins[numOfMultBins+1] = {50, 150, 50,150,2,150};
  63
  64   int runch[2] = {0, 1};
  65   const char *chrgs[2] = { "pip", "pim" };
  66
  67         const int numOfkTbins = 4;
  68         int runktdep[numOfkTbins] = {1,0,0,1};
  69   double ktrng[numOfkTbins+1] = {0.2, 0.3,0.5, 0.6, 0.7};
  70
  71   int run3d = 0; // Do 3D cartesian analysis?
  72   int runshlcms = 1;
  73
  74   //PhysicsSelection set only in runBatch with trigger + Physics Selection Task
  75
  76   double shqmax;
  77   int nbinssh = 200;
  78
  79   if (runshlcms) shqmax = 2.0;
  80   else shqmax = 0.9;
  81
  82   AliFemtoEventReaderAODChain *Reader = new AliFemtoEventReaderAODChain();
  83   Reader->SetFilterBit(96);
  84   Reader->SetDCAglobalTrack(kTRUE);
  85         Reader->SetUseMultiplicity(AliFemtoEventReaderAOD::kReference);
  86         Reader->SetMinPlpContribSPD(3);
  87         Reader->SetIsPileUpEvent(kTRUE);
  88
  89   //Reader->SetCentralityPreSelection(500, 950);
  90
  91   //AliFemtoEventReaderESDChainKine* Reader=new AliFemtoEventReaderESDChainKine();
  92   //Reader->SetConstrained(true);
  93   // Reader->SetUseTPCOnly(false);
  94
  95   //AliFemtoEventReaderESDChain *Reader = new AliFemtoEventReaderESDChain();
  96   //Reader->SetUseMultiplicity(AliFemtoEventReaderESDChain::kV0Centrality);
  97   //Reader->SetReadTrackType(AliFemtoEventReaderESDChain::kGlobal);
  98
  99
 100   AliFemtoManager* Manager=new AliFemtoManager();
 101   Manager->SetEventReader(Reader);
 102
 103   AliFemtoVertexMultAnalysis    *anetaphitpc[20];
 104   AliFemtoBasicEventCut         *mecetaphitpc[20];
 105   AliFemtoCutMonitorEventMult   *cutPassEvMetaphitpc[20];
 106   AliFemtoCutMonitorEventMult   *cutFailEvMetaphitpc[20];
 107   AliFemtoCutMonitorEventVertex *cutPassEvVetaphitpc[20];
 108   AliFemtoCutMonitorEventVertex *cutFailEvVetaphitpc[20];
 109   AliFemtoCutMonitorCollections   *cutPassColletaphitpc[20];
 110   AliFemtoCutMonitorCollections   *cutFailColletaphitpc[20];
 111   AliFemtoESDTrackCut           *dtc1etaphitpc[20];
 112   AliFemtoESDTrackCut           *dtc2etaphitpc[20];
 113   AliFemtoCutMonitorParticleYPt *cutPass1YPtetaphitpc[20];
 114   AliFemtoCutMonitorParticleYPt *cutFail1YPtetaphitpc[20];
 115   AliFemtoCutMonitorParticlePID *cutPass1PIDetaphitpc[20];
 116   AliFemtoCutMonitorParticlePID *cutFail1PIDetaphitpc[20];
 117   AliFemtoCutMonitorParticleYPt *cutPass2YPtetaphitpc[20];
 118   AliFemtoCutMonitorParticleYPt *cutFail2YPtetaphitpc[20];
 119   AliFemtoCutMonitorParticlePID *cutPass2PIDetaphitpc[20];
 120   AliFemtoCutMonitorParticlePID *cutFail2PIDetaphitpc[20];
 121   //  AliFemtoPairCutAntiGamma      *sqpcetaphitpc[20];
 122   AliFemtoShareQualityTPCEntranceSepPairCut      *sqpcetaphitpc[20];
 123   //AliFemtoPairCutRadialDistance      *sqpcetaphitpc[20];
 124   AliFemtoCorrFctnDirectYlm     *cylmetaphitpc[20];
 125   AliFemtoCorrFctnDEtaDPhi      *cdedpetaphi[20];
 126   AliFemtoChi2CorrFctn          *cchiqinvetaphitpc[20];
 127   AliFemtoKTPairCut             *ktpcuts[20*7];
 128   AliFemtoCorrFctnDirectYlm     *cylmkttpc[20*7];
 129   AliFemtoQinvCorrFctn          *cqinvkttpc[20*7];
 130   AliFemtoCorrFctn3DLCMSSym     *cq3dlcmskttpc[20*7];
 131   AliFemtoCorrFctnTPCNcls       *cqinvnclstpc[20];
 132   AliFemtoShareQualityCorrFctn  *cqinvsqtpc[20*10];
 133   AliFemtoChi2CorrFctn          *cqinvchi2tpc[20];
 134   AliFemtoTPCInnerCorrFctn      *cqinvinnertpc[20*10];
 135
 136   // *** Begin pion-pion analysis ***
 137   int aniter = 0;
 138
 139   for (int imult=0; imult<numOfMultBins; imult++) {
 140     if (runmults[imult]) {
 141       for (int ichg=0; ichg<2; ichg++) {
 142         if (runch[ichg]) {
 143           aniter = ichg*numOfMultBins+imult;
 144
 145           anetaphitpc[aniter] = new AliFemtoVertexMultAnalysis(10, -10.0, 10.0, 4, multbins[imult], multbins[imult+1]);
 146           anetaphitpc[aniter]->SetNumEventsToMix(5);
 147           anetaphitpc[aniter]->SetMinSizePartCollection(1);
 148           anetaphitpc[aniter]->SetVerboseMode(kFALSE);
 149
 150           mecetaphitpc[aniter] = new AliFemtoBasicEventCut();
 151           mecetaphitpc[aniter]->SetEventMult(0,10000);
 152           mecetaphitpc[aniter]->SetVertZPos(-10,10);
 153
 154           // cutPassEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutPass%stpcM%i", chrgs[ichg], imult),500);
 155           // cutFailEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutFail%stpcM%i", chrgs[ichg], imult),500);
 156           // mecetaphitpc[aniter]->AddCutMonitor(cutPassEvMetaphitpc[aniter], cutFailEvMetaphitpc[aniter]);
 157
 158           // cutPassEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutPass%stpcM%i", chrgs[ichg], imult));
 159           // cutFailEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutFail%stpcM%i", chrgs[ichg], imult));
 160           // mecetaphitpc[aniter]->AddCutMonitor(cutPassEvVetaphitpc[aniter], cutFailEvVetaphitpc[aniter]);
 161
 162           // cutPassColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutPass%stpcM%i", chrgs[ichg], imult));
 163           // cutFailColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutFail%stpcM%i", chrgs[ichg], imult));
 164           // mecetaphitpc[aniter]->AddCutMonitor(cutPassColletaphitpc[aniter], cutFailColletaphitpc[aniter]);
 165
 166           dtc1etaphitpc[aniter] = new AliFemtoESDTrackCut();
 167
 168           if (ichg == 0)
 169             dtc1etaphitpc[aniter]->SetCharge(1.0);
 170           else if (ichg == 1)
 171             dtc1etaphitpc[aniter]->SetCharge(-1.0);
 172
 173           dtc1etaphitpc[aniter]->SetPt(0.12,4.0);
 174           dtc1etaphitpc[aniter]->SetEta(-1.2,1.2);
 175
 176           //PID method
 177           dtc1etaphitpc[aniter]->SetMass(PionMass);
 178           dtc1etaphitpc[aniter]->SetMostProbablePion();
 179           //dtc1etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
 180
 181           //Track quality cuts
 182           dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
 183           dtc1etaphitpc[aniter]->SetminTPCncls(50);
 184           dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
 185
 186
 187           dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
 188           dtc1etaphitpc[aniter]->SetMaxITSChiNdof(36);
 189           dtc1etaphitpc[aniter]->SetLabel(kFALSE);
 190
 191           //primary particles: hits in ITS + DCA cut
 192           dtc1etaphitpc[aniter]->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
 193                                                           AliESDtrackCuts::kAny);
 194           dtc1etaphitpc[aniter]->SetMaxImpactZ(2.0);
 195           //dtc1etaphitpc[aniter]->SetMaxImpactXY(2.4);
 196           dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0105, 0.0350, -1.1);
 197           //dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
 198           //dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
 199
 200           // cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult), 0.13957);
 201           // cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult), 0.13957);
 202           // dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
 203
 204           // cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),0);
 205           // cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),0);
 206           // dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
 207
 208           //sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
 209           sqpcetaphitpc[aniter] = new AliFemtoShareQualityTPCEntranceSepPairCut();
 210           //sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistance();
 211           sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
 212           sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
 213           sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
 214           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
 215           //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
 216           //runtype==0
 217           // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
 218           // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
 219           // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
 220           // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
 221           // sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
 222           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
 223           //runtype==1
 224           //        sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
 225           //        sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
 226           //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
 227           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
 228           //runtype==2
 229           //        sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.0);
 230           //        sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
 231           //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.045);
 232           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
 233           //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
 234
 235           anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
 236           anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
 237           anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
 238           anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
 239
 240           //Correlation functions
 241
 242           // //Spherical harmonics (without kT bins)
 243           // cylmetaphitpc[aniter] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%i", chrgs[ichg], imult),3,nbinssh,0.0,shqmax,runshlcms);
 244           // anetaphitpc[aniter]->AddCorrFctn(cylmetaphitpc[aniter]);
 245
 246           // //Qinv (without kT bins)
 247           // cqinvkttpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 248           // anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[aniter]);
 249
 250           // //3D cartesian (without kT bins)
 251           // if(run3d){
 252           //   cq3dlcmskttpc[aniter] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%i", chrgs[ichg], imult),60,0.5);
 253           //   anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[aniter]);
 254           // }
 255
 256           // cqinvnclstpc[aniter] = new AliFemtoCorrFctnTPCNcls(Form("cqinvncls%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 257           // anetaphitpc[aniter]->AddCorrFctn(cqinvnclstpc[aniter]);
 258
 259           // cqinvchi2tpc[aniter] = new AliFemtoChi2CorrFctn(Form("cqinvchi2%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
 260           // anetaphitpc[aniter]->AddCorrFctn(cqinvchi2tpc[aniter]);
 261
 262           int ktm;
 263           for (int ikt=0; ikt<numOfkTbins; ikt++) {
 264             if (runktdep[ikt]) {
 265
 266               ktm = aniter*numOfkTbins + ikt;
 267               ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
 268
 269               // cylmkttpc[ktm] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%ikT%i", chrgs[ichg], imult, ikt),3,
 270               //                                                nbinssh, 0.0, shqmax, runshlcms);
 271               // cylmkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 272               // anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[ktm]);
 273
 274               cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0, shqmax);
 275               cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 276               anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);
 277
 278               // cqinvsqtpc[ktm] = new AliFemtoShareQualityCorrFctn(Form("cqinvsq%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
 279               // cqinvsqtpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 280               // anetaphitpc[aniter]->AddCorrFctn(cqinvsqtpc[ktm]);
 281
 282               cqinvinnertpc[ktm] = new AliFemtoTPCInnerCorrFctn(Form("PhistarEta%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
 283               cqinvinnertpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 284               cqinvinnertpc[ktm]->SetRadius(1.2);
 285               anetaphitpc[aniter]->AddCorrFctn(cqinvinnertpc[ktm]);
 286
 287               // if (run3d) {
 288               //   //           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);
 289               //   cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),60,0.5);
 290               //   cq3dlcmskttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
 291               //   anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[ktm]);
 292               // }
 293
 294             }
 295           }
 296
 297           // cdedpetaphi[aniter] = new AliFemtoCorrFctnDEtaDPhi(Form("cdedp%stpcM%i", chrgs[ichg], imult),39, 39);
 298           // anetaphitpc[aniter]->AddCorrFctn(cdedpetaphi[aniter]);
 299
 300           Manager->AddAnalysis(anetaphitpc[aniter]);
 301         }
 302       }
 303     }
 304   }
 305   // *** End pion-pion analysis
 306
 307   return Manager;
 308 }